Rasterize_STL/Code.cxx at master · sergeykhrykov/Rasterize_STL · GitHub

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
#include "itkMesh.h"
#include "itkMeshFileReader.h"

#include "itkImage.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"

#include "itkCastImageFilter.h"
#include "itkTriangleMeshToBinaryImageFilter.h"

int main( int argc, char* argv[] )
{
  if( argc < 2 )
    {
    std::cerr << "Usage: "<< std::endl;
    std::cerr << argv[0];
	std::cerr << " <InputMeshName> [<InputMeshName> <InputMeshName> ... ]";
    std::cerr << std::endl;
    return EXIT_FAILURE;
    }

  const unsigned int Dimension = 3;
  typedef double MeshPixelType;

  typedef itk::Mesh< MeshPixelType, Dimension > MeshType;

  typedef itk::MeshFileReader< MeshType >  MeshReaderType;


  typedef unsigned char                           InputPixelType;
  typedef itk::Image< InputPixelType, Dimension > InputImageType;
  typedef itk::ImageFileReader< InputImageType >  ImageReaderType;

  //ImageReaderType::Pointer imageReader = ImageReaderType::New();
  //imageReader->SetFileName( inputImageName );

  typedef unsigned char                             OutputPixelType;
  typedef itk::Image< OutputPixelType, Dimension >  OutputImageType;

  typedef itk::TriangleMeshToBinaryImageFilter< MeshType, OutputImageType > FilterType;


  for (int i = 1; i < argc; ++i)
  {
	  const char * inputMeshName = argv[i];
	  std::string outputImageName = argv[i];
	  outputImageName += ".mha";

	  MeshReaderType::Pointer meshReader = MeshReaderType::New();
	  meshReader->SetFileName(inputMeshName);

	  FilterType::Pointer filter = FilterType::New();

	  InputImageType::SizeType size;
	  size[0] = 500;
	  size[1] = 500;
	  size[2] = 500;
	  filter->SetSize(size);

	  InputImageType::SpacingType spacing;
	  spacing[0] = .2;
	  spacing[1] = .2;
	  spacing[2] = .2;
	  filter->SetSpacing(spacing);

	  InputImageType::PointType origin;
	  origin[0] = -46;
	  origin[1] = -17;
	  origin[2] = -2;
	  filter->SetOrigin(origin);
	  filter->SetInput(meshReader->GetOutput());
	  filter->SetInsideValue(itk::NumericTraits< OutputPixelType >::max());

	  std::cout << "Rasterizing file: " << inputMeshName << "...";
	  try
	  {
		  filter->Update();
	  }
	  catch (itk::ExceptionObject & error)
	  {
		  std::cerr << "Error: " << error << std::endl;
		  return EXIT_FAILURE;
	  }

	  std::cout << "Done" << std::endl;

	  std::cout << "Writing...";
	  typedef itk::ImageFileWriter< OutputImageType > WriterType;
	  WriterType::Pointer writer = WriterType::New();
	  writer->SetFileName(outputImageName.c_str());
	  writer->SetInput(filter->GetOutput());
	  try
	  {
		  writer->Update();
	  }
	  catch (itk::ExceptionObject & error)
	  {
		  std::cerr << "Error: " << error << std::endl;
		  return EXIT_FAILURE;
	  }
	  std::cout << "Done" << std::endl;

  }

  return EXIT_SUCCESS;
}