vtkITKImageSequenceWriter.h

Go to the documentation of this file.

/*==========================================================================
 
  Copyright (c) Ebatinca S.L., Las Palmas de Gran Canaria, Spain
 
  See COPYRIGHT.txt
  or http://www.slicer.org/copyright/copyright.txt for details.
 
  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.
 
  This file was originally developed by Csaba Pinter, Ebatinca, funded
  by the grant GRT-00000485 of Children's Hospital of Philadelphia, USA.
 
==========================================================================*/
 
#ifndef __vtkITKImageSequenceWriter_h
#define __vtkITKImageSequenceWriter_h
 
#include "vtkImageAlgorithm.h"
#include "vtkImageData.h"
#include "vtkObjectFactory.h"
#include "vtkMatrix4x4.h"
 
#include "vtkITK.h"
#include "vtkITKImageWriter.h"
#include "itkImageIOBase.h"
 
class vtkStringArray;
class AttributeMapType;
class AxisInfoMapType;
 
class VTK_ITK_EXPORT vtkITKImageSequenceWriter : public vtkImageAlgorithm
{
public:
  static vtkITKImageSequenceWriter* New();
  vtkTypeMacro(vtkITKImageSequenceWriter, vtkImageAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  vtkGetMacro(UseCompression, int);
  vtkSetMacro(UseCompression, int);
  vtkBooleanMacro(UseCompression, int);

  vtkGetStringMacro(FileName);
  vtkSetStringMacro(FileName);

  vtkGetStringMacro(ImageIOClassName);
  vtkSetStringMacro(ImageIOClassName);

  void Write();

  void SetRasToIJKMatrix(vtkMatrix4x4* mat) { RasToIJKMatrix = mat; }

  vtkSetMacro(VoxelVectorType, int);
  vtkGetMacro(VoxelVectorType, int);

  vtkSetStringMacro(IntentCode);
  vtkGetStringMacro(IntentCode);

  vtkGetMacro(SequenceAxisSpacing, double);

  vtkGetMacro(SequenceAxisOrigin, double);

  void SetAttribute(const std::string& name, const std::string& value);
  AttributeMapType* GetAttributes() const { return this->Attributes; }

  void SetAxisLabel(unsigned int axis, const char* label);
  AxisInfoMapType* GetAxisLabels() const { return this->AxisLabels; }

  void SetAxisUnit(unsigned int axis, const char* unit);
  AxisInfoMapType* GetAxisUnits() const { return this->AxisUnits; }
 
protected:
  vtkITKImageSequenceWriter();
  ~vtkITKImageSequenceWriter() override;

  int FillInputPortInformation(int port, vtkInformation* info) override;

  bool IsNiftiFile();

  void UpdateNiftiSequenceAxis(int numberOfFrames);
 
protected:
  char* FileName{ nullptr };
  vtkMatrix4x4* RasToIJKMatrix{ nullptr };
  int UseCompression{ 0 };
  char* ImageIOClassName{ nullptr };
  int VoxelVectorType{ vtkITKImageWriter::VoxelVectorTypeUndefined };
  char* IntentCode{ nullptr };
  double SequenceAxisSpacing{ 1.0 };
  double SequenceAxisOrigin{ 0.0 };
 
  AttributeMapType* Attributes;
  AxisInfoMapType* AxisLabels;
  AxisInfoMapType* AxisUnits;
 
private:
  vtkITKImageSequenceWriter(const vtkITKImageSequenceWriter&) = delete;
  void operator=(const vtkITKImageSequenceWriter&) = delete;
};
 
#endif