include/optimizer/iPenaltyMedianRootPrior.hh

Go to the documentation of this file.

#ifndef IPENALTYMEDIANROOTPRIOR_HH
#define IPENALTYMEDIANROOTPRIOR_HH 1

#include "vPenalty.hh"
#include "sAddonManager.hh"
#include "sOutputManager.hh"

#define MRF_NEIGHBORHOOD_SPHERE 0
#define MRF_NEIGHBORHOOD_BOX 1
#define MRF_NEIGHBORHOOD_6_NEAREST 2

#define MRF_NOT_DEFINED -1

#define MRF_NEIGHBOR_X 0
#define MRF_NEIGHBOR_Y 1
#define MRF_NEIGHBOR_Z 2

class iPenaltyMedianRootPrior : public vPenalty
{
  // -------------------------------------------------------------------
  // Constructor & Destructor
  public:

    iPenaltyMedianRootPrior();
    ~iPenaltyMedianRootPrior();

  // -------------------------------------------------------------------
  // Public member functions
  public:
    // Function for automatic insertion (put the class name as the parameter and do not add semi-column at the end of the line)
    FUNCTION_PENALTY(iPenaltyMedianRootPrior)
    int ReadConfigurationFile(const string& a_configurationFile);
    int ReadOptionsList(const string& a_optionsList);
    int LocalPreProcessingStep(int a_tbf, int a_rbf, int a_cbf, INTNB a_voxel, int a_th);
    FLTNB ComputePenaltyValue(FLTNB* ap_image, INTNB a_voxel, int a_th);
    FLTNB ComputeFirstDerivative(FLTNB* ap_image, INTNB a_voxel, int a_th);
    FLTNB ComputeSecondDerivative(FLTNB* ap_image, INTNB a_voxel, int a_th);

  // -------------------------------------------------------------------
  // Private member functions
  private:
    void ShowHelpSpecific();
    int CheckSpecificParameters();
    int InitializeSpecific();
    int BuildNeighborhoodKernel();
    int BuildSpecificNeighborhood(INTNB a_voxel, int a_th);
    int ComputeMedianInNeighborhood(int a_tbf, int a_rbf, int a_cbf, INTNB a_voxel, int a_th);

  // -------------------------------------------------------------------
  // Data members
  private:
    // Neighborhood
    int m_neighborhoodShape;             
    FLTNB m_neighborhoodSphereRadius;    
    int m_neighborhoodBoxOrder;          
    int m_neighborhoodBoxExcludeCorners; 
    INTNB m_neighborhoodMaxNbVoxels;     
    INTNB** m2p_neighborhoodKernel;      
    INTNB* mp_neighborhoodNbVoxels;      
    INTNB** m2p_neighborhoodIndices;     
    // Specific
    FLTNB* mp_medianValue;               
};

// Class for automatic insertion (set here the visible optimizer's name as the first parameter,
// put the class name as the second parameter and do NOT add semi-colon at the end of the line)
CLASS_PENALTY(MRP,iPenaltyMedianRootPrior)

#endif