mitkPlanarFigureWriter.h

Go to the documentation of this file.

/*============================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

============================================================================*/

#ifndef _MITK_PlanarFigure_WRITER__H_
#define _MITK_PlanarFigure_WRITER__H_

#include <MitkPlanarFigureExports.h>
#include <itkProcessObject.h>
#include <mitkFileWriterWithInformation.h>
#include <mitkPlanarFigure.h>

class TiXmlElement;
namespace mitk
{
  class MITKPLANARFIGURE_EXPORT PlanarFigureWriter : public mitk::FileWriterWithInformation
  {
  public:
    mitkClassMacro(PlanarFigureWriter, mitk::FileWriter);

    mitkWriterMacro;

    itkFactorylessNewMacro(Self);

    itkCloneMacro(Self);

      typedef mitk::PlanarFigure InputType;

    typedef InputType::Pointer InputTypePointer;

    itkSetStringMacro(FileName);

    itkGetStringMacro(FileName);

    itkSetStringMacro(FilePrefix);

    itkGetStringMacro(FilePrefix);

    itkSetStringMacro(FilePattern);

    itkGetStringMacro(FilePattern);

    using Superclass::SetInput;

    void SetInput(InputType *input);

    void SetInput(const unsigned int &num, InputType *input);

    PlanarFigure *GetInput();

    PlanarFigure *GetInput(const unsigned int &num);

    std::vector<std::string> GetPossibleFileExtensions() override;

    std::string GetFileExtension() override;

    bool CanWriteDataType(DataNode *) override;

    std::string GetWritenMIMEType() override;

    virtual void SetInput(DataNode *);

    std::string GetSupportedBaseData() const override;

    itkGetConstMacro(Success, bool);

    const char *GetDefaultFilename() override { return "PlanarFigure.pf"; }
    const char *GetFileDialogPattern() override { return "Planar Figure Files (*.pf)"; }
    const char *GetDefaultExtension() override { return ".pf"; }
    bool CanWriteBaseDataType(BaseData::Pointer data) override
    {
      return dynamic_cast<mitk::PlanarFigure *>(data.GetPointer());
    }
    void DoWrite(BaseData::Pointer data) override
    {
      if (CanWriteBaseDataType(data))
      {
        this->SetInput(dynamic_cast<mitk::PlanarFigure *>(data.GetPointer()));
        this->Update();
      }
    }

    void ReleaseMemory() override;

  protected:
    PlanarFigureWriter();

    ~PlanarFigureWriter() override;

    void GenerateData() override;

    virtual void ResizeInputs(const unsigned int &num);

    TiXmlElement *CreateXMLVectorElement(const char *name, itk::FixedArray<mitk::ScalarType, 3> v);

    std::string m_FileName;
    std::string m_FilePrefix;
    std::string m_FilePattern;
    std::string m_Extension;
    std::string m_MimeType;
    bool m_Success;
  };
}

#endif