AnisotropicDiffusion.h

#pragma once

 
#include "DefinitionsImageProcessing.h"

namespace FLImaging
{
    namespace ImageProcessing
    {
        class CInternalAnisotropicDiffusion;

        class FL_EXPORT CAnisotropicDiffusion : public CAlgorithmFeature_SR_SP_DI_DR_DP_BLK_FBLK_BAT
        {
        public:

            enum EDiffusionCoefficientMode
            {
                EDiffusionCoefficientMode_Parabolic = 0,

                EDiffusionCoefficientMode_PeronaMalik,

                EDiffusionCoefficientMode_Weickert,
            };

            CAnisotropicDiffusion();

            CAnisotropicDiffusion(const CAnisotropicDiffusion& anisotropicDiffusion);

            CAnisotropicDiffusion(const CAnisotropicDiffusion* pAnisotropicDiffusion);

            virtual ~CAnisotropicDiffusion();

            const CAnisotropicDiffusion& operator=(const CAnisotropicDiffusion& anisotropicDiffusion);

            virtual const FLImaging::CResult Assign(const CAnisotropicDiffusion& anisotropicDiffusion);

            virtual const FLImaging::CResult Assign(const CAnisotropicDiffusion* pAnisotropicDiffusion);

            virtual const FLImaging::CResult Execute();

            virtual const FLImaging::CResult Clear() override;

            virtual const FLImaging::CResult SetDiffusionMode(EDiffusionCoefficientMode eDiffMode = EDiffusionCoefficientMode_Weickert);

            virtual EDiffusionCoefficientMode GetDiffusionMode();

            virtual const FLImaging::CResult SetIteration(int64_t i64Iteration = 1);

            virtual int64_t GetIteration();

            virtual const FLImaging::CResult SetContrast(double f64Contrast = 5.);

            virtual double GetContrast();

            virtual const FLImaging::CResult SetTheta(double f64Theta = 10.);

            virtual double GetTheta();

            virtual const CResult SetPaddingMethod(EPaddingMethod ePaddingMethod = EPaddingMethod_DecreasingKernel);

            virtual EPaddingMethod GetPaddingMethod();

            virtual const CResult SetPaddingValue(const Base::CMultiVar<double>& mvPaddingValue);

            virtual const Base::CMultiVar<double> GetPaddingValue();
 
            virtual Foundation::EProcessingUnitFeatures GetOptimalProcessingUnit() const override;
 
            virtual const CResult RunWarmUpCuda();
 
            virtual int64_t GetRequiredGPUMemory() const override;
 
            virtual bool IsLicenseAvailable() const override;
 
            DeclareGetClassType();
            SupportToDuplicateObject(CAnisotropicDiffusion, *this);
 
        protected:

            EDiffusionCoefficientMode m_eDiffusionMode;

            int64_t m_i64Iteration;

            double m_f64Contrast;

            double m_f64Theta;

            EPaddingMethod m_ePaddingMethod;

            Base::CMultiVar<double> m_mvPaddingValue;
 
        protected:
            friend class CInternalAnisotropicDiffusion;
            CInternalAnisotropicDiffusion* m_pInternal;
        };
    }
}