mitkCostFunctionBase.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 mitkCostFunctionBase_h
#define mitkCostFunctionBase_h

//Which one do we really need?
//#include <berryISelectionListener.h>
#include <itkImage.h>

#include <vector>
#include "mitkCommon.h"
#include "MitkConnectomicsExports.h"

//To use pair
#include <utility>

namespace mitk
{
  template <typename TPixel, unsigned int VImageDimension>
  class CostFunctionBase
  {
  public:
    typedef itk::ImageRegion<3> RegionType;
    typedef itk::Image< TPixel, VImageDimension > ImageType;
    //Set Functions
    void SetImage(itk::Image<TPixel, VImageDimension> *);
    void SetRegion(std::pair<RegionType, int>);
    //Main Function
    int CalculateCost();
    int MaximalValue();

  private:
    ImageType * m_Image;
    std::pair<RegionType, int> m_RegionPair;
    //Functions to calculate cost
    int RegionIsFullEnough ();
    int SidesAreNotTooLong ();
    int COMLiesWithinParcel();
    std::vector<double> GetCenterOfMass();

    std::vector<int> m_weight;
  };
}

#include "mitkCostFunctionBase.cpp"

#endif /* mitkCostFunctionBase_h */