itkSpatialObjectReader.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  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 itkSpatialObjectReader_h
#define itkSpatialObjectReader_h
 
#include "itkMetaSceneConverter.h"
#include "itkMetaConverterBase.h"
#include "itkSpatialObject.h"
#include "itkGroupSpatialObject.h"
#include "itkProcessObject.h"
 
namespace itk
{
template <unsigned int VDimension = 3,
          typename PixelType = unsigned char,
          typename TMeshTraits = DefaultStaticMeshTraits<PixelType, VDimension, VDimension>>
class ITK_TEMPLATE_EXPORT SpatialObjectReader : public Object
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(SpatialObjectReader);

  using Self = SpatialObjectReader;
  using Pointer = SmartPointer<Self>;
  using SpatialObjectType = SpatialObject<VDimension>;
  using SpatialObjectPointer = typename SpatialObjectType::Pointer;
  using GroupType = GroupSpatialObject<VDimension>;
  using GroupPointer = typename GroupType::Pointer;

  using MetaConverterBaseType = MetaConverterBase<VDimension>;
  using MetaSceneConverterType = MetaSceneConverter<VDimension, PixelType, TMeshTraits>;

  itkNewMacro(Self);

  using Superclass = Object;
  itkOverrideGetNameOfClassMacro(SpatialObjectReader);

  void
  Update();

  void
  SetMetaIOVersion(unsigned int ver);
  unsigned int
  GetMetaIOVersion() const;

  itkSetStringMacro(FileName);

  itkGetStringMacro(FileName);

  SpatialObjectPointer
  GetOutput()
  {
    return m_SpatialObject;
  }

  GroupPointer
  GetGroup()
  {
    if (m_Group == nullptr)
    {
      if (m_SpatialObject->GetTypeName() == "GroupSpatialObject")
      {
        m_Group = static_cast<GroupType *>(m_SpatialObject.GetPointer());
      }
      else
      {
        m_Group = GroupType::New();
        m_Group->AddChild(m_SpatialObject);
      }
    }
    return m_Group;
  }


  const MetaEvent *
  GetEvent()
  {
    return m_MetaToSpatialConverter->GetEvent();
  }
 
  void
  SetEvent(MetaEvent * event)
  {
    m_MetaToSpatialConverter->SetEvent(event);
  }

  void
  RegisterMetaConverter(const char *            metaTypeName,
                        const char *            spatialObjectTypeName,
                        MetaConverterBaseType * converter);
 
protected:
  std::string m_FileName{};
 
  SpatialObjectReader();
  ~SpatialObjectReader() override = default;
 
private:
  GroupPointer m_Group{};
 
  SpatialObjectPointer m_SpatialObject{};
 
  typename MetaSceneConverterType::Pointer m_MetaToSpatialConverter{};
};
} // namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#  include "itkSpatialObjectReader.hxx"
#endif
 
#endif // itkSpatialObjectReader_h