mitkLRDensityEstimation.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef mitkLRDensityEstimation_h
#define mitkLRDensityEstimation_h

#include <vigra/matrix.hxx>
#include <vigra/random_forest.hxx>

#include <mitkDataCollection.h>

#include <MitkCLImportanceWeightingExports.h>

namespace mitk
{
  class MITKCLIMPORTANCEWEIGHTING_EXPORT LRDensityEstimation
  {
  public:
    typedef vigra::MultiArray<2, double> MatrixType;

    void SetCollection(DataCollection::Pointer data);
    DataCollection::Pointer GetCollection();

    void SetTestMask(std::string name);
    std::string GetTestMask();

    void SetTrainMask(std::string name);
    std::string GetTrainMask();

    void SetWeightName(std::string name);
    std::string GetWeightName();

    void Update();
    void WeightsForAll(mitk::DataCollection::Pointer train, mitk::DataCollection::Pointer test);
    void SetModalities(std::vector<std::string> modalities);
    std::vector<std::string> GetModalities();

    void Predict();

  private:
    DataCollection::Pointer m_Collection;
    MatrixType m_Weights;
    std::string m_TestMask;
    std::string m_TrainMask;
    std::string m_WeightName;
    std::vector<std::string> m_Modalities;

    std::string m_WeightFileName;
  }; // class LRDensityEstimation
} //namespace mitk

#endif //mitkLRDensityEstimation_h