mitkEuclideanSkeletonGraphCut.h

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

  Program:   3DMed
  Date:      $Date: 2014-02-25 18:30:00 +0800 $
  Version:   $Version: 4.6.0 $
  Copyright: MIPG, Institute of Automation, Chinese Academy of Sciences

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

#ifndef __mitkEuclideanSkeletonGraphCut_h
#define __mitkEuclideanSkeletonGraphCut_h

#include "mitkSegmentationIncludes.h"
#include "mitkEuclideanSkeletonFeatureExtraction.h"
#include "mitkVolumeToVolumeFilter.h"

class MITK_SEGMENTATION_API mitkEuclideanSkeletonGraphCut : public mitkVolumeToVolumeFilter
{
public:
    MITK_TYPE(mitkEuclideanSkeletonGraphCut,mitkFilter)
    mitkEuclideanSkeletonGraphCut();

    void SetEuclideanSkeletonFeature(mitkEuclideanSkeletonFeature *inData) {m_ESF = inData;}
    mitkEuclideanSkeletonFeature * GetEuclideanSkeletonFeature() {return m_ESF;}
    void SetUserSelection(unsigned char* userSelection);
    void SetIntensityFactor(float intFac) { m_IntensityFactor=intFac;}
    void SetRadiusFactor(float radFac) { m_RadiusFactor=radFac;}

    typedef enum
    {
        OM_NoOutput,
        OM_Mask,
        OM_A,
        OM_B,
        OM_CA,
        OM_CB,
        OM_BIN_A,
        OM_BIN_B
    }OutputMode;
    void SetOutputMode (OutputMode om) { m_OutputMode=om;}
    void GetCutResult(unsigned char* cr);
    bool GenerateOutput(OutputMode om);

protected:
    virtual ~mitkEuclideanSkeletonGraphCut();
    virtual bool Execute();

    mitkRCPtr<mitkEuclideanSkeletonFeature> m_ESF;
    unsigned char *m_UserSelection;

    float m_IntensityFactor,m_RadiusFactor;

    OutputMode m_OutputMode;
    unsigned char *m_CutResult;

private:
    mitkEuclideanSkeletonGraphCut(const mitkEuclideanSkeletonGraphCut&);
    void operator = (const mitkEuclideanSkeletonGraphCut&);

};

#endif

---