niftkFitPlaneToPointCloudWrapper.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). 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 in the top level directory for details.

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

#ifndef niftkPlaneToPointCloudWrapper_h
#define niftkPlaneToPointCloudWrapper_h

#include "niftkPCLExports.h"
#include <string>
#include <ostream>
#include <mitkCommon.h>
#include <mitkPointSet.h>
#include <itkObject.h>
#include <itkObjectFactory.h>
#include <itkObjectFactoryBase.h>

// forward declaration to avoid pulling in truckloads of pcl headers.
namespace pcl
{
struct ModelCoefficients;
}

namespace niftk
{

// i dont think this should be called ...Wrapper. suggestions for a better name?
class NIFTKPCL_EXPORT FitPlaneToPointCloudWrapper : public itk::Object
{
public:
  mitkClassMacroItkParent(FitPlaneToPointCloudWrapper, itk::Object)
  itkNewMacro(FitPlaneToPointCloudWrapper)


protected:
  FitPlaneToPointCloudWrapper();
  virtual ~FitPlaneToPointCloudWrapper();

  FitPlaneToPointCloudWrapper(const FitPlaneToPointCloudWrapper&);
  FitPlaneToPointCloudWrapper& operator=(const FitPlaneToPointCloudWrapper&);


public:
  void FitPlane(const std::string& filename);

  void FitPlane(const mitk::PointSet::Pointer& pointset);
  void PrintOutput(std::ostream& log) const;

  void GetParameters(float& a, float& b, float& c, float& d) const;


private:
  pcl::ModelCoefficients*  m_PlaneCoefficients;
  float                    m_MinPlaneDistance;
  float                    m_MaxPlaneDistance;
  float                    m_AvgPlaneDistance;
  float                    m_RmsPlaneDistance;
};

} // end namespace

#endif