File: TestPhantomSegmentationUsingLSTK.cxx

package info (click to toggle)
volview 3.4-3
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 25,204 kB
  • sloc: cpp: 132,585; ansic: 11,612; tcl: 236; sh: 64; makefile: 25; xml: 8
file content (54 lines) | stat: -rw-r--r-- 1,849 bytes parent folder | download
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
/*=========================================================================

  Copyright (c) Kitware, Inc.
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/VolViewCopyright.htm 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 notice for more information.

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

#include "vtkType.h"
#include "itkPhantomSegmentationImageFilter.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include <iostream>

int main( int argc, char *argv[] )
{
  if (argc < 8)
    {
    std::cerr << "Usage args : PhantomCTData.mha ForeGroundDensity BackgroundDensity Center_x center_y center_z Radius segmentation" << std::endl;
    return EXIT_FAILURE;
    }

  typedef itk::Image< short, 3 > ImageType;
  typedef itk::PhantomSegmentationImageFilter< ImageType > SegmenterType;

  itk::ImageFileReader< ImageType >::Pointer reader = 
                itk::ImageFileReader< ImageType >::New();
  reader->SetFileName( argv[1] );
  reader->Update();

  SegmenterType::Pointer segmenter = SegmenterType::New();
  segmenter->SetInput( reader->GetOutput() );
  segmenter->SetForegroundDensity( atof( argv[2] ) );
  segmenter->SetBackgroundDensity( atof( argv[3] ) );
  double c[3] = { atof(argv[4]), atof(argv[5]), atof(argv[6]) };
  segmenter->SetCenter( c );
  segmenter->SetRadius( atoi( argv[7] ) );

  segmenter->Update();

  typedef SegmenterType::OutputImageType OutputImageType;
  typedef itk::ImageFileWriter< OutputImageType > WriterType;
  WriterType::Pointer writer = WriterType::New();
  writer->SetFileName( argv[8] );
  writer->SetInput( segmenter->GetOutput() );
  writer->Update();

  return EXIT_SUCCESS;
}