mitkEuclideanSkeletonFeatureExtraction.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 __mitkEuclideanSkeletonFeatureExtraction_h
#define __mitkEuclideanSkeletonFeatureExtraction_h

#include "mitkSegmentationIncludes.h"
#include "mitkFilter.h"
#include "mitkICVolume.h"

#include "mitkEuclideanDistanceTransformFilter.h"

class MITK_SEGMENTATION_API mitkEuclideanSkeletonFeature : public mitkICVolume
{
public:
    
    struct Descriptor
    {
        unsigned skeletonPos;
        float radius;
        float size;
    };
    typedef std::vector<Descriptor> DescriptorList;
    MITK_TYPE(mitkEuclideanSkeletonFeature,mitkICVolume)
    mitkEuclideanSkeletonFeature();
    void GetSkeletonPosition(unsigned skeletonIndex,int &posX,int &posY,int &posZ);
    void GetSkeletonPosition(unsigned skeletonIndex,int pos[3]);

    unsigned AddDescriptor(const Descriptor& descriptor)
    {
        m_dList.push_back(descriptor);
        return m_dList.size()-1;
    }
    unsigned GetNumberOfDescriptors()
    {
        return m_dList.size();
    }
    Descriptor& GetDescriptor(unsigned index)
    {
        return m_dList[index];
    }
    void RemoveAllDescriptors()
    {
        m_dList.clear();
    }
protected:
    virtual ~mitkEuclideanSkeletonFeature();

private:
    mitkEuclideanSkeletonFeature(const mitkEuclideanSkeletonFeature&);
    void operator = (const mitkEuclideanSkeletonFeature&);
    
    DescriptorList m_dList; 
};

class MITK_SEGMENTATION_API mitkEuclideanSkeletonFeatureExtraction : public mitkFilter
{
public:
    MITK_TYPE(mitkEuclideanSkeletonFeatureExtraction,mitkFilter)
    mitkEuclideanSkeletonFeatureExtraction();

    void SetLowThreshValue(double d){m_dtFilter->SetLowThreshValue(d);}
    int GetLowThreshValue(){return m_dtFilter->GetLowThreshValue();}
    void SetHighThreshValue(double d){m_dtFilter->SetHighThreshValue(d);}
    int GetHighThreshValue(){return m_dtFilter->GetHighThreshValue();}
    void SetInput(mitkVolume *inData) {m_dtFilter->SetInput(inData);}
    mitkVolume* GetInput() {return m_dtFilter->GetInput();} 
    mitkEuclideanSkeletonFeature* GetEuclideanSkeletonFeature();

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

    mitkEuclideanDistanceTransformFilter *m_dtFilter;
    mitkVolume* m_DistanceData;
    mitkEuclideanSkeletonFeature *m_ESF;

private:
    mitkEuclideanSkeletonFeatureExtraction(const mitkEuclideanSkeletonFeatureExtraction&);
    void operator = (const mitkEuclideanSkeletonFeatureExtraction&);
};

#endif

---