ENH: Ingest ITKMeshToPolyData into Modules/Filtering

Modules/Filtering/MeshToPolyData/CMakeLists.txt

@@ -0,0 +1,11 @@
+cmake_minimum_required(VERSION 3.16.3)
+project(MeshToPolyData)
+
+if(NOT ITK_SOURCE_DIR)
+  find_package(ITK REQUIRED)
+  list(APPEND CMAKE_MODULE_PATH ${ITK_CMAKE_DIR})
+  include(ITKModuleExternal)
+else()
+  set(ITK_DIR ${CMAKE_BINARY_DIR})
+  itk_module_impl()
+endif()

Modules/Filtering/MeshToPolyData/include/itkImageToPointSetFilter.h

@@ -0,0 +1,93 @@
+/*=========================================================================
+ *
+ *  Copyright NumFOCUS
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *         https://www.apache.org/licenses/LICENSE-2.0.txt
+ *
+ *  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 itkImageToPointSetFilter_h
+#define itkImageToPointSetFilter_h
+
+#include "itkImageToMeshFilter.h"
+
+namespace itk
+{
+/** \class ImageToPointSetFilter
+ *
+ * \brief Convert an Image to a PointSet.
+ *
+ * This class provides default where every pixel in an Image is converted to a
+ * point in the output PointSet.
+ *
+ * This class inherits from ImageToMeshFilter, but no topological information is
+ * added to the mesh.
+ *
+ * \ingroup MeshToPolyData
+ */
+template <typename TInputImage, typename TOutputMesh>
+class ITK_TEMPLATE_EXPORT ImageToPointSetFilter : public ImageToMeshFilter<TInputImage, TOutputMesh>
+{
+public:
+  ITK_DISALLOW_COPY_AND_MOVE(ImageToPointSetFilter);
+
+  /** Standard class type alias. */
+  using Self = ImageToPointSetFilter;
+  using Superclass = ImageToMeshFilter<TInputImage, TOutputMesh>;
+  using Pointer = SmartPointer<Self>;
+  using ConstPointer = SmartPointer<const Self>;
+
+  /** Method for creation through the object factory. */
+  itkNewMacro(Self);
+
+  /** Run-time type information (and related methods). */
+  itkOverrideGetNameOfClassMacro(ImageToPointSetFilter);
+
+  /** Some convenient type alias. */
+  using InputImageType = TInputImage;
+  using InputImageConstPointer = typename InputImageType::ConstPointer;
+  using InputImageRegionType = typename InputImageType::RegionType;
+  using InputImagePixelType = typename InputImageType::PixelType;
+
+  /** Some type alias associated with the output mesh. */
+  using OutputMeshType = TOutputMesh;
+  using PointType = typename OutputMeshType::PointType;
+  using OutputMeshPointer = typename OutputMeshType::Pointer;
+  using PointsContainer = typename OutputMeshType::PointsContainer;
+  using PointIdentifier = typename OutputMeshType::PointIdentifier;
+  using PointsContainerPointer = typename PointsContainer::Pointer;
+  using PointsContainerIterator = typename PointsContainer::Iterator;
+  using PointDataContainer = typename OutputMeshType::PointDataContainer;
+  using PointDataContainerPointer = typename PointDataContainer::Pointer;
+  using PointDataContainerIterator = typename PointDataContainer::Iterator;
+
+  /** The dimension of the output mesh. */
+  static constexpr unsigned int PointDimension = TOutputMesh::PointDimension;
+
+  /** ImageDimension constant */
+  static constexpr unsigned int ImageDimension = TInputImage::ImageDimension;
+
+protected:
+  ImageToPointSetFilter() = default;
+  ~ImageToPointSetFilter() override = default;
+
+  void
+  GenerateData() override;
+};
+
+} // end namespace itk
+
+#ifndef ITK_MANUAL_INSTANTIATION
+#  include "itkImageToPointSetFilter.hxx"
+#endif
+
+#endif

Modules/Filtering/MeshToPolyData/include/itkImageToPointSetFilter.hxx

@@ -0,0 +1,76 @@
+/*=========================================================================
+ *
+ *  Copyright NumFOCUS
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *         https://www.apache.org/licenses/LICENSE-2.0.txt
+ *
+ *  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 itkImageToPointSetFilter_hxx
+#define itkImageToPointSetFilter_hxx
+
+#include "itkImageToPointSetFilter.h"
+
+#include "itkImageRegionConstIteratorWithIndex.h"
+#include "itkProgressReporter.h"
+
+namespace itk
+{
+
+template <typename TInputImage, typename TOutputMesh>
+void
+ImageToPointSetFilter<TInputImage, TOutputMesh>::GenerateData()
+{
+  // " from itkBinaryMask3DMeshSource:
+  // This indicates that the current BufferedRegion is equal to the
+  // requested region. This action prevents useless rexecutions of
+  // the pipeline.
+  this->GetOutput()->SetBufferedRegion(this->GetOutput()->GetRequestedRegion());
+
+  OutputMeshPointer      mesh = this->GetOutput();
+  PointsContainerPointer points = mesh->GetPoints();
+  InputImageConstPointer image = this->GetInput(0);
+
+  PointDataContainerPointer pointData;
+  if (mesh->GetPointData())
+  {
+    pointData = mesh->GetPointData();
+  }
+  else
+  { // Create
+    pointData = PointDataContainer::New();
+  }
+
+  const SizeValueType numberOfPixels = image->GetRequestedRegion().GetNumberOfPixels();
+  ProgressReporter    progress(this, 0, numberOfPixels);
+  points->Reserve(numberOfPixels);
+  pointData->Reserve(numberOfPixels);
+  mesh->SetPointData(pointData.GetPointer());
+
+  typename itk::ImageRegionConstIteratorWithIndex<InputImageType> imageIt(image, image->GetRequestedRegion());
+  imageIt.GoToBegin();
+  PointsContainerIterator    pointsIt = points->Begin();
+  PointDataContainerIterator pointDataIt = pointData->Begin();
+  while (!imageIt.IsAtEnd())
+  {
+    image->TransformIndexToPhysicalPoint(imageIt.GetIndex(), pointsIt.Value());
+    (pointDataIt.Value()) = imageIt.Get();
+    ++pointsIt;
+    ++pointDataIt;
+    ++imageIt;
+    progress.CompletedPixel();
+  }
+}
+
+} // end namespace itk
+
+#endif

Modules/Filtering/MeshToPolyData/include/itkMeshToPolyDataFilter.h

@@ -0,0 +1,132 @@
+/*=========================================================================
+ *
+ *  Copyright NumFOCUS
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *         https://www.apache.org/licenses/LICENSE-2.0.txt
+ *
+ *  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 itkMeshToPolyDataFilter_h
+#define itkMeshToPolyDataFilter_h
+
+#include "itkProcessObject.h"
+#include "itkPolyData.h"
+
+#include <type_traits>
+
+namespace itk
+{
+
+template <typename TInputType, typename = void>
+struct HasCellTraits : std::false_type
+{};
+
+template <typename TInputType>
+struct HasCellTraits<TInputType, std::void_t<typename TInputType::CellTraits>> : std::true_type
+{};
+
+/** \class MeshToPolyDataFilter
+ *
+ * \brief Convert a PointSet or Mesh to PolyData
+ *
+ * Convert an itk::PointSet or itk::Mesh to an itk::PolyData for visualization
+ * with vtk.js.
+ *
+ * Currently support ITK cell types:
+ *
+ * - itk::VertexCell
+ * - itk::LineCell
+ * - itk::TriangleCell
+ * - itk::QuadrilateralCell
+ * - itk::PolygonCell
+ *
+ * \ingroup MeshToPolyData
+ *
+ */
+template <typename TInputMesh>
+class MeshToPolyDataFilter : public ProcessObject
+{
+public:
+  ITK_DISALLOW_COPY_AND_MOVE(MeshToPolyDataFilter);
+
+  static constexpr unsigned int PointDimension = TInputMesh::PointDimension;
+
+  using InputMeshType = TInputMesh;
+
+  /** Standard class typedefs. */
+  using Self = MeshToPolyDataFilter<InputMeshType>;
+  using Superclass = ProcessObject;
+  using Pointer = SmartPointer<Self>;
+  using ConstPointer = SmartPointer<const Self>;
+
+  using PolyDataType = PolyData<typename InputMeshType::PixelType>;
+
+  /** Run-time type information. */
+  itkOverrideGetNameOfClassMacro(MeshToPolyDataFilter);
+
+  /** Standard New macro. */
+  itkNewMacro(Self);
+
+  /** Set the mesh input of this process object.  */
+  using Superclass::SetInput;
+  void
+  SetInput(const InputMeshType * input);
+
+  /** Get the mesh input of this process object.  */
+  const InputMeshType *
+  GetInput() const;
+
+  const InputMeshType *
+  GetInput(unsigned int idx) const;
+
+  PolyDataType *
+  GetOutput();
+  const PolyDataType *
+  GetOutput() const;
+
+  PolyDataType *
+  GetOutput(unsigned int idx);
+
+protected:
+  MeshToPolyDataFilter();
+  ~MeshToPolyDataFilter() override = default;
+
+  void
+  PrintSelf(std::ostream & os, Indent indent) const override;
+
+  void
+  GenerateData() override;
+
+  template <typename TInputMeshDispatch,
+            typename std::enable_if<!HasCellTraits<TInputMeshDispatch>::value, int>::type = 0>
+  void
+  GenerateDataDispatch();
+
+  template <typename TInputMeshDispatch,
+            typename std::enable_if<HasCellTraits<TInputMeshDispatch>::value, int>::type = 0>
+  void
+  GenerateDataDispatch();
+
+  ProcessObject::DataObjectPointer
+  MakeOutput(ProcessObject::DataObjectPointerArraySizeType idx) override;
+  ProcessObject::DataObjectPointer
+  MakeOutput(const ProcessObject::DataObjectIdentifierType &) override;
+
+private:
+};
+} // namespace itk
+
+#ifndef ITK_MANUAL_INSTANTIATION
+#  include "itkMeshToPolyDataFilter.hxx"
+#endif
+
+#endif // itkMeshToPolyDataFilter