1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
|
/*=========================================================================
Program: Advanced Normalization Tools
Module: $RCSfile: itkLabeledPointSetFileWriter.h,v $
Language: C++
Date: $Date: 2009/03/04 23:10:58 $
Version: $Revision: 1.17 $
Copyright (c) ConsortiumOfANTS. All rights reserved.
See accompanying COPYING.txt or
http://sourceforge.net/projects/advants/files/ANTS/ANTSCopyright.txt for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#ifndef __itkLabeledPointSetFileWriter_h
#define __itkLabeledPointSetFileWriter_h
#include "itkMesh.h"
#include "itkArray.h"
#include "itkImage.h"
#include "itkVectorContainer.h"
namespace itk
{
/** \class LabeledPointSetFileWriter
* \brief
* Writes an itkMesh to a file in various txt file formats.
*
*/
template <class TInputMesh>
class LabeledPointSetFileWriter : public Object
{
public:
/** Standard "Self" typedef. */
typedef LabeledPointSetFileWriter Self;
typedef Object Superclass;
typedef SmartPointer<Self> Pointer;
typedef SmartPointer<const Self> ConstPointer;
/** Method for creation through the object factory */
itkNewMacro( Self );
/** Write the Input mesh to the Output file.
* Use either Update() or Write(). */
void Update();
void Write();
/** Extract dimension from the output mesh. */
itkStaticConstMacro( Dimension, unsigned int,
TInputMesh::PointType::Dimension );
/** Run-time type information (and related methods). */
itkTypeMacro( LabeledPointSetFileWriter, Object );
/** Hold on to the type information specified by the template parameters. */
typedef TInputMesh InputMeshType;
typedef typename TInputMesh::Pointer InputMeshPointer;
typedef typename InputMeshType::MeshTraits MeshTraits;
typedef typename InputMeshType::Superclass PointSetType;
typedef typename InputMeshType::PointType PointType;
typedef typename MeshTraits::PixelType PixelType;
typedef Array<PixelType> MultiComponentScalarType;
typedef Array<unsigned long> LineType;
typedef VectorContainer<long,
MultiComponentScalarType> MultiComponentScalarSetType;
typedef VectorContainer<long, LineType> LineSetType;
typedef Image<PixelType,
itkGetStaticConstMacro( Dimension )> LabeledPointSetImageType;
typedef typename
LabeledPointSetImageType::SizeType ImageSizeType;
typedef typename
LabeledPointSetImageType::PointType ImageOriginType;
typedef typename
LabeledPointSetImageType::SpacingType ImageSpacingType;
typedef typename
LabeledPointSetImageType::DirectionType ImageDirectionType;
/** Set the Input */
void SetInput( InputMeshType * input );
/** Set/Get the name of the file where data are written. */
itkSetStringMacro( FileName );
itkGetStringMacro( FileName );
/** Specify other attributes */
itkSetMacro( Lines, typename LineSetType::Pointer );
itkSetMacro( MultiComponentScalars,
typename MultiComponentScalarSetType::Pointer );
/** Specify image attributes if output is an image. */
itkSetMacro( ImageSize, ImageSizeType );
itkGetConstMacro( ImageSize, ImageSizeType );
itkSetMacro( ImageOrigin, ImageOriginType );
itkGetConstMacro( ImageOrigin, ImageOriginType );
itkSetMacro( ImageSpacing, ImageSpacingType );
itkGetConstMacro( ImageSpacing, ImageSpacingType );
itkSetMacro( ImageDirection, ImageDirectionType );
itkGetConstMacro( ImageDirection, ImageDirectionType );
protected:
LabeledPointSetFileWriter();
virtual ~LabeledPointSetFileWriter();
virtual void GenerateData();
std::string m_FileName;
InputMeshPointer m_Input;
typename MultiComponentScalarSetType::Pointer m_MultiComponentScalars;
typename LineSetType::Pointer m_Lines;
/**
* If output is an image type, the attributes must be specified.
*/
ImageSizeType m_ImageSize;
ImageSpacingType m_ImageSpacing;
ImageOriginType m_ImageOrigin;
ImageDirectionType m_ImageDirection;
void PrintSelf(std::ostream& os, Indent indent) const ITK_OVERRIDE;
private:
LabeledPointSetFileWriter(const Self &); // purposely not implemented
void operator=(const Self &); // purposely not implemented
void WritePointsToAvantsFile();
void WritePointsToImageFile();
void WriteVTKFile();
void WritePointsToVTKFile();
void WriteScalarsToVTKFile();
void WriteLinesToVTKFile();
};
} // namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkLabeledPointSetFileWriter.hxx"
#endif
#endif
|