ComputationalGraphAugmentationForClassifier.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "ComputationalGraphAugmentationBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        class CAugmentationParameters;
 
        template<typename T>
        class CInternalAugmentationForClassifier;
 
        template <typename T>
        class FL_EXPORT CComputationalGraphAugmentationForClassifier : public CComputationalGraphAugmentationBase<T>
        {
        protected:
            CComputationalGraphAugmentationForClassifier();
            CComputationalGraphAugmentationForClassifier(const CComputationalGraphAugmentationForClassifier<T>& cg);
 
        public:
 
            CComputationalGraphAugmentationForClassifier(const CComputationalBase<T>& cbOperand, const CAugmentationParameters& apParams, const Base::CFLArray<Base::CFLArray<int32_t>>& flaClassNum, const Base::CFLArray<Base::CFLBase*>& flaAugmnetaitonParams);
            CComputationalGraphAugmentationForClassifier(const CComputationalBase<T>& cbOperand, const CAugmentationParameters* pApParams, const Base::CFLArray<Base::CFLArray<int32_t>>* pFlaClassNum, const Base::CFLArray<Base::CFLBase*>* pFlaAugmentationParams);
            CComputationalGraphAugmentationForClassifier(const CComputationalBase<T>& cbOperand);
        
            virtual ~CComputationalGraphAugmentationForClassifier();
 
            virtual const CResult SetIndividualClassesAugmentationParameters(const Base::CFLArray<Base::CFLArray<int32_t>>& flaClassNum, const Base::CFLArray<Base::CFLBase*>& flaAugmnetaiton);
            virtual const CResult GetIndividualClassesAugmentationParameters(Base::CFLArray<Base::CFLArray<int32_t>>& flaClassNum, Base::CFLArray<Base::CFLBase*>& flaAugmnetaitonParams) const;
            virtual const CResult GetIndividualClassesAugmentationParameters(std::vector<std::vector<int32_t>>& vctClassNum, std::vector<CAugmentationParameters>& vctAp) const;
            
            virtual CComputationalBase<T>* Clone() const override;
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CComputationalGraphAugmentationForClassifier, *this);
 
        protected:
        public:
            DeclareGetSignletonObject(CComputationalGraphAugmentationForClassifier);
 
            Base::CFLArray<Base::CFLArray<int32_t>> m_flaIndividualClassesAugmentationClassNum;
            Base::CFLArray<Base::CFLBase*> m_flaIndividualClassesAugmentationParams;
 
        protected:
            std::map<int32_t, CAugmentationParameters>& m_mapAugParam;
            std::vector<int32_t>& m_vctIndividualClassesAugmentationClassNum;
            std::vector<CAugmentationParameters>& m_vctIndividualClassesAugmentationParams;
 
        private:
            template <typename T> friend class CInternalAugmentationForClassifier;
        };
 
        #define CCGFAugmentationCF(...) (*(new CComputationalGraphAugmentationForClassifier<float>(__VA_ARGS__)))
        #define CCGDAugmentationCF(...) (*(new CComputationalGraphAugmentationForClassifier<double>(__VA_ARGS__)))
 
        #define CCGTAugmentationCF(T, ...) (*(new CComputationalGraphAugmentationForClassifier<T>(__VA_ARGS__)))
    }
}
 
#endif