vtkCurveMeasurementsCalculator.h

Go to the documentation of this file.

/*==============================================================================
 
  Program: 3D Slicer
 
  Portions (c) Copyright Brigham and Women's Hospital (BWH) All Rights Reserved.
 
  See COPYRIGHT.txt
  or http://www.slicer.org/copyright/copyright.txt for details.
 
  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
 
==============================================================================*/
 
#ifndef __vtkCurveMeasurementsCalculator_h
#define __vtkCurveMeasurementsCalculator_h
 
// VTK includes
#include <vtkCollection.h>
#include <vtkPolyData.h>
#include <vtkPolyDataAlgorithm.h>
#include <vtkSetGet.h>
#include <vtkWeakPointer.h>
 
// Markups MRML includes
#include <vtkMRMLMarkupsNode.h>
 
// Export
#include "vtkMRMLExport.h"
 
class vtkCallbackCommand;

class VTK_MRML_EXPORT vtkCurveMeasurementsCalculator : public vtkPolyDataAlgorithm
{
public:
  vtkTypeMacro(vtkCurveMeasurementsCalculator, vtkPolyDataAlgorithm);
  static vtkCurveMeasurementsCalculator* New();
 
  void PrintSelf(ostream& os, vtkIndent indent) override;


  void SetInputMarkupsMRMLNode(vtkMRMLMarkupsNode* node);
  vtkMRMLMarkupsNode* GetInputMarkupsMRMLNode();


  vtkSetMacro(CurveIsClosed, bool);
  vtkGetMacro(CurveIsClosed, bool);
  vtkBooleanMacro(CurveIsClosed, bool);


  vtkSetMacro(CalculateCurvature, bool);
  vtkGetMacro(CalculateCurvature, bool);
  vtkBooleanMacro(CalculateCurvature, bool);

  static const char* GetMeanCurvatureName() { return "curvature mean"; };
  static const char* GetMaxCurvatureName() { return "curvature max"; };
  static const char* GetCurvatureArrayName() { return "Curvature"; };
 
  vtkGetMacro(CurvatureUnits, std::string);
  vtkSetMacro(CurvatureUnits, std::string);


  vtkSetMacro(CalculateTorsion, bool);
  vtkGetMacro(CalculateTorsion, bool);
  vtkBooleanMacro(CalculateTorsion, bool);

  static const char* GetMeanTorsionName() { return "torsion mean"; };
  static const char* GetMaxTorsionName() { return "torsion max"; };
  static const char* GetTorsionArrayName() { return "Torsion"; };
 
  vtkGetMacro(TorsionUnits, std::string);
  vtkSetMacro(TorsionUnits, std::string);
 
  vtkMTimeType GetMTime() override;

  static bool InterpolateArray(vtkDoubleArray* inputValues,
                               bool closedCurve,
                               vtkDoubleArray* interpolatedValues,
                               vtkDoubleArray* pedigreeIdsArray,
                               double pedigreeIdsValueScale = 1.0);
 
protected:
  bool CalculatePolyDataCurvature(vtkPolyData* polyData);
  bool CalculatePolyDataTorsion(vtkPolyData* polyData);
  bool InterpolateControlPointMeasurementToPolyData(vtkPolyData* outputPolyData);

  static void OnControlPointArrayModified(vtkObject* caller, unsigned long eid, void* clientData, void* callData);
 
protected:
  vtkWeakPointer<vtkMRMLMarkupsNode> InputMarkupsMRMLNode;

  bool CurveIsClosed{ false };

  bool CalculateCurvature{ false };

  bool CalculateTorsion{ false };

  vtkCallbackCommand* ControlPointArrayModifiedCallbackCommand;
  vtkCollection* ObservedControlPointArrays;
 
  std::string CurvatureUnits{ "mm-1" };
  std::string TorsionUnits{ "mm-1" };
 
protected:
  int FillInputPortInformation(int port, vtkInformation* info) override;
  int RequestData(vtkInformation* request, vtkInformationVector** inputVector, vtkInformationVector* outputVector) override;
 
protected:
  vtkCurveMeasurementsCalculator();
  ~vtkCurveMeasurementsCalculator() override;
  vtkCurveMeasurementsCalculator(const vtkCurveMeasurementsCalculator&) = delete;
  void operator=(const vtkCurveMeasurementsCalculator&) = delete;
};
 
#endif