vtkSlicerGeneralizedReformatLogic.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.
 
==============================================================================*/
 
// .NAME vtkSlicerGeneralizedReformatLogic - slicer logic class for volumes manipulation
// .SECTION Description
// This class manages the logic associated with reading, saving,
// and changing propertied of the volumes
 
#ifndef __vtkSlicerGeneralizedReformatLogic_h
#define __vtkSlicerGeneralizedReformatLogic_h
 
// Slicer includes
#include "vtkSlicerModuleLogic.h"
 
// GeneralizedReformat MRML includes
#include <vtkMRMLAbstractVolumeResampler.h>
 
// MRML includes
class vtkMRMLMarkupsCurveNode;
class vtkMRMLScalarVolumeNode;
 
// VTK includes
class vtkMatrix4x4;
class vtkPoints;
#include <vtkSmartPointer.h>
 
// STD includes
#include <cstdlib>
#include <map>
 
#include "vtkSlicerGeneralizedReformatModuleLogicExport.h"
 
class VTK_SLICER_GENERALIZEDREFORMAT_MODULE_LOGIC_EXPORT vtkSlicerGeneralizedReformatLogic : public vtkSlicerModuleLogic
{
public:
  static vtkSlicerGeneralizedReformatLogic* New();
  vtkTypeMacro(vtkSlicerGeneralizedReformatLogic, vtkSlicerModuleLogic);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  bool GetPointsProjectedToPlane(vtkPoints* pointsArrayIn, vtkMatrix4x4* transformWorldToPlane, vtkPoints* pointsArrayOut);

  bool ComputeStraighteningTransform(vtkMRMLTransformNode* transformToStraightenedNode,
                                     vtkMRMLMarkupsCurveNode* curveNode,
                                     const double sliceSizeMm[2],
                                     double outputSpacingMm,
                                     bool stretching = false,
                                     double rotationDeg = 0.0,
                                     vtkMRMLModelNode* reslicingPlanesModelNode = nullptr);

  bool StraightenVolume(vtkMRMLScalarVolumeNode* outputStraightenedVolume,
                        vtkMRMLScalarVolumeNode* volumeNode,
                        const double outputStraightenedVolumeSpacing[3],
                        vtkMRMLTransformNode* straighteningTransformNode);

  bool ProjectVolume(vtkMRMLScalarVolumeNode* outputProjectedVolume, vtkMRMLScalarVolumeNode* inputStraightenedVolume, int projectionAxisIndex = 0);

  bool ResampleVolume(std::string& resamplerName,
                      vtkMRMLVolumeNode* inputVolume,
                      vtkMRMLVolumeNode* outputVolume,
                      vtkMRMLTransformNode* resamplingTransform,
                      vtkMRMLVolumeNode* referenceVolume = nullptr,
                      int interpolationType = vtkMRMLAbstractVolumeResampler::InterpolationTypeLinear,
                      const vtkMRMLAbstractVolumeResampler::ResamplingParameters& resamplingParameters = vtkMRMLAbstractVolumeResampler::ResamplingParameters());

  void RegisterVolumeResampler(const std::string& resamplerName, vtkMRMLAbstractVolumeResampler* resampler);
  void UnregisterVolumeResampler(const std::string& resamplerName);
  bool IsVolumeResamplerRegistered(const std::string& resamplerName);
  vtkMRMLAbstractVolumeResampler* GetVolumeResampler(const std::string& resamplerName);
 
protected:
  vtkSlicerGeneralizedReformatLogic();
  ~vtkSlicerGeneralizedReformatLogic() override;
 
  void SetMRMLSceneInternal(vtkMRMLScene* newScene) override;
  void RegisterNodes() override;
  void UpdateFromMRMLScene() override;
  void OnMRMLSceneNodeAdded(vtkMRMLNode* node) override;
  void OnMRMLSceneNodeRemoved(vtkMRMLNode* node) override;
 
  double TransformSpacingFactor{ 5.0 };
 
  std::map<std::string, vtkSmartPointer<vtkMRMLAbstractVolumeResampler>> Resamplers;
 
private:
  vtkSlicerGeneralizedReformatLogic(const vtkSlicerGeneralizedReformatLogic&); // Not implemented
  void operator=(const vtkSlicerGeneralizedReformatLogic&);                    // Not implemented
};
 
#endif