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>

// Export
#include "vtkSlicerMarkupsModuleMRMLExport.h"

class vtkCallbackCommand;

class VTK_SLICER_MARKUPS_MODULE_MRML_EXPORT vtkCurveMeasurementsCalculator : public vtkPolyDataAlgorithm
{
public:
  vtkTypeMacro(vtkCurveMeasurementsCalculator, vtkPolyDataAlgorithm);
  static vtkCurveMeasurementsCalculator* New();

  void PrintSelf(ostream& os, vtkIndent indent) override;

  void SetMeasurements(vtkCollection* measurements);
  vtkCollection* GetMeasurements();

  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"; };

  vtkMTimeType GetMTime() override;

  vtkGetMacro(CurvatureUnits, std::string);
  vtkSetMacro(CurvatureUnits, std::string);

  static bool InterpolateArray(vtkDoubleArray* inputValues, vtkDoubleArray* interpolatedValues,
    vtkDoubleArray* pedigreeIdsArray, double pedigreeIdsValueScale=1.0);

protected:
  bool CalculatePolyDataCurvature(vtkPolyData* polyData);
  bool InterpolateControlPointMeasurementToPolyData(vtkPolyData* outputPolyData);

  static void OnControlPointArrayModified(vtkObject* caller, unsigned long eid, void* clientData, void* callData);

protected:
  vtkWeakPointer<vtkCollection> Measurements;

  bool CurveIsClosed{false};

  bool CalculateCurvature{false};

  vtkCallbackCommand* ControlPointArrayModifiedCallbackCommand;
  vtkCollection* ObservedControlPointArrays;

  std::string CurvatureUnits;

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