itkSpatialObjectWriter.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright Insight Software 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
 *
 *         http://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 itkSpatialObjectWriter_h
#define itkSpatialObjectWriter_h

#include "itkMetaSceneConverter.h"
#include "itkMetaConverterBase.h"
#include "itkSpatialObject.h"

namespace itk
{
template< unsigned int NDimensions = 3,
          typename PixelType = unsigned char,
          typename TMeshTraits = DefaultStaticMeshTraits< PixelType,
                                                          NDimensions,
                                                          NDimensions >
          >
class ITK_TEMPLATE_EXPORT SpatialObjectWriter:public Object
{
public:
  ITK_DISALLOW_COPY_AND_ASSIGN(SpatialObjectWriter);

  using Self = SpatialObjectWriter;
  using Pointer = SmartPointer< Self >;

  using Superclass = Object;

  using SpatialObjectType = SpatialObject< NDimensions >;
  using SpatialObjectPointer = typename SpatialObjectType::Pointer;
  using SceneType = GroupSpatialObject< NDimensions >;

  using MetaConverterBaseType = MetaConverterBase< NDimensions >;

  using MetaSceneConverterType
    = MetaSceneConverter< NDimensions, PixelType, TMeshTraits >;

  itkNewMacro(Self);

  itkTypeMacro(SpatialObjectWriter, Object);

  void Update();

  itkSetStringMacro(FileName);

  itkGetStringMacro(FileName);

  void SetInput(SpatialObjectType *input){ m_SpatialObject = input; }

  void SetInput(SceneType *input){ m_Scene = input; }

  itkSetMacro(BinaryPoints, bool);
  itkGetConstMacro(BinaryPoints, bool);

  void SetTransformPrecision(unsigned int precision);

  unsigned int GetTransformPrecision();

  itkSetMacro(WriteImagesInSeparateFile, bool);
  itkGetConstMacro(WriteImagesInSeparateFile, bool);

  void RegisterMetaConverter(const char *metaTypeName,
                             const char *spatialObjectTypeName,
                             MetaConverterBaseType *converter);

protected:
  std::string m_FileName;
  bool        m_BinaryPoints;
  bool        m_WriteImagesInSeparateFile;

  SpatialObjectWriter();
  ~SpatialObjectWriter() override = default;

private:

  SpatialObjectPointer m_SpatialObject;
  SceneType *          m_Scene;

  typename MetaSceneConverterType::Pointer m_MetaToSpatialConverter;
};
} // namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkSpatialObjectWriter.hxx"
#endif

#endif // itkSpatialObjectWriter_h