mitkLinearSplitting.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 mitkLinearSplitting_h
#define mitkLinearSplitting_h
 
#include <vigra/multi_array.hxx>
#include <vigra/random_forest.hxx>
#include <mitkAdditionalRFData.h>
 
namespace mitk
{
  template<class TLossAccumulator>
  class LinearSplitting
  {
  public:
      typedef typename TLossAccumulator::WeightContainerType TWeightContainer;
      typedef TWeightContainer WeightContainerType;
 
      LinearSplitting();
 
      template <class T>
      LinearSplitting(vigra::ProblemSpec<T> const &ext);
 
      void UsePointWeights(bool pointWeight);
      bool IsUsingPointWeights();
 
      void UseRandomSplit(bool randomSplit);
      bool IsUsingRandomSplit();
 
      void SetPointWeights(WeightContainerType weight);
      WeightContainerType GetPointWeights();
 
      void SetAdditionalData(AdditionalRFDataAbstract* data);
      AdditionalRFDataAbstract* GetAdditionalData() const;
 
      template <class T>
      void set_external_parameters(vigra::ProblemSpec<T> const &ext);
 
      template <class TDataSourceFeature,
                class TDataSourceLabel,
                class TDataIterator,
                class TArray>
      void operator()(TDataSourceFeature const &column,
                      TDataSourceLabel const &labels,
                      TDataIterator &begin,
                      TDataIterator &end,
                      TArray const &regionResponse);
 
      template <class TDataSourceLabel,
                class TDataIterator,
                class TArray>
      double LossOfRegion(TDataSourceLabel const & labels,
                          TDataIterator &begin,
                          TDataIterator &end,
                          TArray const & regionResponse);
 
      double GetMinimumLoss()
      {
          return m_MinimumLoss;
      }
 
      double GetMinimumThreshold()
      {
          return m_MinimumThreshold;
      }
 
      std::ptrdiff_t GetMinimumIndex()
      {
          return m_MinimumIndex;
      }
 
      vigra::ArrayVector<double>* GetBestCurrentCounts()
      {
          return m_BestCurrentCounts;
      }
 
  private:
      bool m_UsePointWeights;
      bool m_UseRandomSplit;
      WeightContainerType m_PointWeights;
      // From original code
      vigra::ArrayVector<double> m_ClassWeights;
      vigra::ArrayVector<double> m_BestCurrentCounts[2];
      double m_MinimumLoss;
      double m_MinimumThreshold;
      std::ptrdiff_t m_MinimumIndex;
      vigra::ProblemSpec<> m_ExtParameter;
      AdditionalRFDataAbstract* m_AdditionalData;
  };
}
 
#include <../src/Splitter/mitkLinearSplitting.cpp>
#endif