mitkPointSetEstimator.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 __mitkPointSetEstimator_h
#define __mitkPointSetEstimator_h

#include "mitkProcessObject.h"
#include "mitkTransform.h"
#include "mitkPointSet.h"
#include "mitkRCPtr.h"


class MITK_REGISTRATION_API mitkPointSetEstimator : public mitkProcessObject
{
public:
    MITK_TYPE(mitkPointSetEstimator, mitkProcessObject) 
    virtual void PrintSelf(ostream &os);

    mitkPointSetEstimator();

    virtual ~mitkPointSetEstimator();   
    
    virtual void SetFixedPointSet(mitkPointSet *fixedPointSet) {m_FixedPointSet = fixedPointSet;}

    virtual void SetMovingPointSet(mitkPointSet *movingPointSet) {m_MovingPointSet = movingPointSet;}

    virtual void SetFixedPointSet(VectorParameterType& p);

    virtual void SetMovingPointSet(VectorParameterType& p);

    mitkPointSet* GetFixedPointSet() { return m_FixedPointSet; }

    mitkPointSet* GetMovingPointSet() { return m_MovingPointSet; }

    unsigned int GetNumberOfPoints() {return m_NumberOfPoints;}

    void SetPointSetMaskFlag(bool flag);

    void SetTransform(mitkTransform* transform) {m_Transform = transform;}

    mitkTransform* GetTransform();

    mitkMatrixType* GetTransformMatrix();

    const VectorParameterType& GetTransformParameters() const;

    void SetSpaceDimension(unsigned int d) { m_SpaceDimension = d; }

    unsigned int GetSpaceDimension() { return m_SpaceDimension; }

    virtual bool IsTransformSupported(mitkTransform* transform = NULL); 

    void SwapInputs();

protected:
    virtual bool Execute();

    mitkRCPtr<mitkTransform>    m_Transform;
    mitkMatrixType*             m_TransformMatrix;
    VectorParameterType         m_TransformParameters; 

    mitkPointSet*               m_FixedPointSet;
    mitkPointSet*               m_MovingPointSet;
    unsigned int                m_NumberOfPoints;
    unsigned int                m_SpaceDimension;

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

};


//#define DEFINED_mitkPointSetEstimator



#endif

---