mitkThresholdSplit.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkThresholdSplit_h
#define mitkThresholdSplit_h
 
#include <vigra/multi_array.hxx>
#include <vigra/random_forest.hxx>
#include <mitkAdditionalRFData.h>
 
namespace mitk
{
    template<class TColumnDecisionFunctor, class TFeatureCalculator, class TTag = vigra::ClassificationTag>
    class ThresholdSplit: public vigra::SplitBase<TTag>
    {
    public:
        ThresholdSplit();
//        ThresholdSplit(const ThresholdSplit & other);
 
        void SetFeatureCalculator(TFeatureCalculator processor);
        TFeatureCalculator GetFeatureCalculator() const;
 
        void SetCalculatingFeature(bool calculate);
        bool IsCalculatingFeature() const;
 
        void UsePointBasedWeights(bool weightsOn);
        bool IsUsingPointBasedWeights() const;
 
        void UseRandomSplit(bool split) {m_UseRandomSplit = split;}
        bool IsUsingRandomSplit() const { return m_UseRandomSplit; }
 
        void SetPrecision(double value);
        double GetPrecision() const;
 
        void SetMaximumTreeDepth(int value);
        int GetMaximumTreeDepth() const override;
 
        void SetAdditionalData(AdditionalRFDataAbstract* data);
        AdditionalRFDataAbstract* GetAdditionalData() const;
 
        void SetWeights(vigra::MultiArrayView<2, double> weights);
        vigra::MultiArrayView<2, double> GetWeights() const;
 
        // From vigra::ThresholdSplit
        double minGini() const;
        int bestSplitColumn() const;
        double bestSplitThreshold() const;
 
        template<class T>
        void set_external_parameters(vigra::ProblemSpec<T> const & in);
 
        template<class T, class C, class T2, class C2, class Region, class Random>
        int findBestSplit(vigra::MultiArrayView<2, T, C> features,
                          vigra::MultiArrayView<2, T2, C2> labels,
                          Region & region,
                          vigra::ArrayVector<Region>& childRegions,
                          Random & randint);
 
        double region_gini_;
 
    private:
 
        // From vigra::ThresholdSplit
        typedef vigra::SplitBase<TTag> SB;
 
       // splitter parameters (used by copy constructor)
        bool m_CalculatingFeature;
        bool m_UseWeights;
        bool m_UseRandomSplit;
        double m_Precision;
        int m_MaximumTreeDepth;
        TFeatureCalculator m_FeatureCalculator;
        vigra::MultiArrayView<2, double> m_Weights;
 
        // variabels to work with
        vigra::ArrayVector<vigra::Int32> splitColumns;
        TColumnDecisionFunctor bgfunc;
        vigra::ArrayVector<double> min_gini_;
        vigra::ArrayVector<std::ptrdiff_t> min_indices_;
        vigra::ArrayVector<double> min_thresholds_;
        int bestSplitIndex;
        AdditionalRFDataAbstract* m_AdditionalData;
 
    };
}
 
 
#include <../src/Splitter/mitkThresholdSplit.cpp>
#endif