rtkConvexShape.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright RTK Consortium
 *
 *  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 rtkConvexShape_h
#define rtkConvexShape_h

#include <itkMatrix.h>
#include <itkPoint.h>
#include <itkDataObject.h>
#include <itkObjectFactory.h>

#include "RTKExport.h"
#include "rtkMacro.h"

namespace rtk
{
class RTK_EXPORT ConvexShape : public itk::DataObject
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(ConvexShape);

  using Self = ConvexShape;
  using Superclass = itk::DataObject;
  using Pointer = itk::SmartPointer<Self>;
  using ConstPointer = itk::SmartPointer<const Self>;

  static constexpr unsigned int Dimension = 3;
  using ScalarType = double;
  using PointType = itk::Vector<ScalarType, Dimension>;
  using VectorType = itk::Vector<ScalarType, Dimension>;
  using RotationMatrixType = itk::Matrix<ScalarType, Dimension, Dimension>;

  itkNewMacro(Self);

  itkOverrideGetNameOfClassMacro(ConvexShape);

  virtual bool
  IsInside(const PointType & point) const;

  virtual bool
  IsIntersectedByRay(const PointType &  rayOrigin,
                     const VectorType & rayDirection,
                     ScalarType &       nearDist,
                     ScalarType &       farDist) const;

  virtual void
  Rescale(const VectorType & r);

  virtual void
  Translate(const VectorType & t);

  virtual void
  Rotate(const RotationMatrixType & r);

  void
  AddClipPlane(const VectorType & dir, const ScalarType & pos);
  void
  SetClipPlanes(const std::vector<VectorType> & dir, const std::vector<ScalarType> & pos);

  itkSetMacro(Density, ScalarType);
  itkGetConstMacro(Density, ScalarType);
  itkGetMacro(Density, ScalarType);

  itkGetConstReferenceMacro(PlaneDirections, std::vector<VectorType>);
  itkGetConstReferenceMacro(PlanePositions, std::vector<ScalarType>);

protected:
  ConvexShape();
  bool
  ApplyClipPlanes(const PointType &  rayOrigin,
                  const VectorType & rayDirection,
                  ScalarType &       nearDist,
                  ScalarType &       farDist) const;
  bool
  ApplyClipPlanes(const PointType & point) const;
  itk::LightObject::Pointer
  InternalClone() const override;

private:
  ScalarType              m_Density{ 0. };
  std::vector<VectorType> m_PlaneDirections;
  std::vector<ScalarType> m_PlanePositions;
};

} // namespace rtk
#endif