KMeansCluster.h

#pragma once
 
namespace FLImaging
{
    namespace Foundation
    {
        template<typename T>
        class FL_EXPORT CKMeansCluster : public Base::CFLBase
        {
        public:
            enum EInitializeMethod :int32_t
            {
                EInitializeMethod_Auto = 0,
 
                EInitializeMethod_Random,
 
                EInitializeMethod_Manual,
            };
 
            CKMeansCluster();
 
            virtual ~CKMeansCluster();
 
            const CResult SetNumOfCluster(int64_t i64NumOfCluster);
 
            int64_t GetNumOfCluster();
 
            int64_t GetNumOfDimension();
 
            const CResult AssignData(T* pArrTData, int64_t i64NumOfDimension, int64_t i64NumOfData);
 
            const CResult SetData(T* pArrTData, int64_t i64NumOfDimension, int64_t i64NumOfData);
 
            const CResult ClearData();
 
            const CResult SetMaximumIteration(int64_t i64MaximumIteration);
 
            int64_t GetMaximumIteration();
 
            const CResult SetInitializeCentroidMethod(EInitializeMethod eInitializeMethod);
 
            EInitializeMethod GetInitializeCentroidMethod();
 
            const CResult SetInitCentroid(int64_t i64Index, Base::CFLArray<T>* pFlaCentroid);
 
            const CResult GetResultCentroid(int64_t i64Index, Base::CFLArray<T>* pFlaCentroid);
 
            const CResult ClearCentroid();
 
            const CResult GetResultGroupIndex(int64_t i64DataIndex, int64_t* pI64GroupIndex);
 
            const CResult GetResultGroupIndex(int64_t i64DataIndex, int64_t* pI64GroupIndex, Base::CFLArray<T>* pFlaData);
 
            const CResult GetResultDataListOfGroup(int64_t i64GroupIndex, Base::CFLArray<Base::CFLArray<T>>* pFlaList);
 
            const CResult GetResultDataListOfGroup(int64_t i64GroupIndex, Base::CFLArray<int64_t>* pFlaList);
 
            const CResult Calculate();
 
 
            SupportToDuplicateObject(CKMeansCluster, *this);
            DeclareGetClassType();
 
        protected:
            double InternalEuclideanDistance(T* pData1, T* pData2, int64_t i64Dimension);
 
        protected:
            int64_t m_i64NumOfDimension;
            int64_t m_i64NumOfData;
            int64_t m_i64NumOfCluster;
 
            T* m_pTVectors;
            T** m_pTVectorsYCache;
            bool m_bBufferCreated;
 
            T* m_pTCentroid;
            T** m_pTCentroidYCache;
 
            int64_t* m_pTResultDataGroup;
 
            int64_t m_i64MaximumIteration;
            EInitializeMethod m_eInitializeMethod;
        };
    }
}