ObjectAugmentationParameters.h

#pragma once
 
#include "DefinitionsAI.h"
#include "ObjectAugmentationPresetParameters.h"
 
#ifdef _M_X64

 
#pragma pack(push, 1)

namespace FLImaging
{
    namespace AI
    {
        class FL_EXPORT CObjectAugmentationParameters : public Base::CFLBase
        {
        public:
            
            enum EObjectAugmentation_OperationSource
            {
                EObjectAugmentation_OperationSource_Self = 0,
                EObjectAugmentation_OperationSource_LearningDataset,
                EObjectAugmentation_OperationSource_UserDefined,
            };
 
        public:

            CObjectAugmentationParameters();
            CObjectAugmentationParameters(const CObjectAugmentationParameters& ap);
            CObjectAugmentationParameters(const CObjectAugmentationParameters* pAp);

            virtual ~CObjectAugmentationParameters();

            virtual const CResult Assign(const CObjectAugmentationParameters& ap);
            virtual const CResult Assign(const CObjectAugmentationParameters* pAp);

            const CObjectAugmentationParameters& operator=(const CObjectAugmentationParameters& ap);
            virtual const CResult Clear();

            virtual const CResult EnableAugmentation(bool bEnableAugmentation);
            virtual bool IsAugmentationEnabled() const;

            virtual const CResult SetObjectAugmentationOperationSourceMethod(EObjectAugmentation_OperationSource eObjectAugmentationOperationSourceMethod = EObjectAugmentation_OperationSource_LearningDataset);
            virtual const CResult GetObjectAugmentationOperationSourceMethod(EObjectAugmentation_OperationSource& eObjectAugmentationOperationSourceMethod) const;

            virtual const CResult SetObjectAugmentationUserDefinedImage(const Base::CFLImage& fliObjectAugmentationUserDefinedImage);
            virtual const CResult SetObjectAugmentationUserDefinedImage(const Base::CFLImage* pFliObjectAugmentationUserDefinedImage);
            virtual const Base::CFLImage* GetObjectAugmentationUserDefinedImage() const;
            
            virtual const CResult SetObjectCountParam(int64_t i64MinCount = 1, int64_t i64MaxCount = 2);
            virtual const CResult GetObjectCountParam(int64_t& i64MinCount, int64_t& i64MaxCount) const;

            virtual const CResult AddContainmentClassPair(int32_t i32ParentClassNum, int32_t i32ChildClassNum);

            virtual const CResult RemoveContainmentClassPair(int32_t i32ParentClassNum, int32_t i32ChildClassNum);

            virtual const CResult ClearContainmentClassPairs();

            virtual const Base::CFLArray<std::pair<int32_t, int32_t>>& GetContainmentClassPairs() const;

            virtual const CResult SetLearningAugmentationPreset(const Base::CFLArray<CFLBase*>& flaObjectAugPreset);
            virtual const CResult SetLearningAugmentationPreset(int32_t i32Index, const CObjectAugmentationPresetParameters& augPreset);
            virtual const CResult AddLearningAugmentationPreset(const Base::CFLArray<CFLBase*>& flaAugPreset);
            virtual const CResult AddLearningAugmentationPreset(const CObjectAugmentationPresetParameters& augPreset);
            virtual const CObjectAugmentationPresetParameters* GetLearningAugmentationPreset(int32_t i32Index) const;
            virtual const CResult GetLearningAugmentationPreset(int32_t i32Index, CObjectAugmentationPresetParameters& augPreset) const;
            virtual const Base::CFLArray<CFLBase*>* GetLearningAugmentationPresets() const;
            virtual const CResult RemoveLearningAugmentationPreset(int32_t i32Index);
            virtual const CResult RemoveAllLearningAugmentationPreset();
            virtual int64_t GetLearningAugmentationPresetCount() const;
 
            virtual const CResult GetBinaryData(Base::CFLData& fldBinary, int32_t i32Version = -1) const;
            virtual const CResult GetBinaryData(Base::CFLData* pFldBinary, int32_t i32Version = -1) const;
 
            virtual const CResult SetBinaryData(const Base::CFLData& fldBinary);
            virtual const CResult SetBinaryData(const Base::CFLData* pFldBinary);
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CObjectAugmentationParameters, *this);
 
        protected:
            bool m_bAugmentation;

            Base::CFLArray<std::pair<int32_t, int32_t>> m_flaContainmentClassNumbers;
            EObjectAugmentation_OperationSource m_eObjectAugmentationOperationSourceMethod;

            Base::CFLImage* m_pFliObjectAugmentationUserDefinedImage;

            int64_t m_i64MinCount;
            int64_t m_i64MaxCount;

            Base::CFLArray<CFLBase*> m_flaObjectAugmentationPreset;
        };
    }
}
 
#pragma pack(pop)
#endif