File: ItkImageComponent.cpp

package info (click to toggle)
camitk 4.0.4-2
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 343,588 kB
  • sloc: cpp: 78,476; xml: 1,210; sh: 723; ansic: 142; makefile: 101; perl: 84; sed: 20
file content (129 lines) | stat: -rw-r--r-- 4,225 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
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
/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2016 Univ. Grenoble Alpes, CNRS, TIMC-IMAG UMR 5525 (GMCAO)
 *
 * Visit http://camitk.imag.fr for more information
 *
 * This file is part of CamiTK.
 *
 * CamiTK is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * only, as published by the Free Software Foundation.
 *
 * CamiTK is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License version 3 for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * version 3 along with CamiTK.  If not, see <http://www.gnu.org/licenses/>.
 *
 * $CAMITK_LICENCE_END$
 ****************************************************************************/

#include "ItkImageComponent.h"
#include "ImageComponent.h"

using namespace camitk;

//-- Qt
#include <QString>
#include <QTextStream>
#include <QFileInfo>
#include <QMessageBox>

//-- Vtk
#include <vtkImageData.h>

//-- Itk
#include <itkImageFileReader.h>
#include <itkOrientImageFilter.h>
#include <itkImageToVTKImageFilter.h>

// -------------------- constructor --------------------
ItkImageComponent::ItkImageComponent(const QString& fileName) throw(AbortException)
    : ImageComponent(fileName) {
    createComponent(fileName);
}

// -------------------- createComponent --------------------
void ItkImageComponent::createComponent(const QString & filename) {
    if (!filename.isEmpty()) {
        vtkSmartPointer<vtkImageData> img = readVolume(filename);
        if (img) {
            setImageData(img, false);
            // use file basename as default name
            setName(QFileInfo(filename).baseName());
        } else {
            throw (AbortException("Could not open file\n"));
        }
    } else {
        throw (AbortException("No filename found\n"));
    }

}

// -------------------- readVolume --------------------
vtkSmartPointer<vtkImageData> ItkImageComponent::readVolume(const QString& filename) {
    // Let itk ImageIO factory do all the job...
    // see itk users guide  chapter 7 "Rading and Writing Images"
    typedef short                               PixelType;
    const   unsigned int                    Dimension = 3;
    typedef itk::Image< PixelType, Dimension >  ImageType;
    typedef itk::ImageFileReader< ImageType >  ReaderType;
    typedef itk::ImageToVTKImageFilter< ImageType > ConnectorType;
    //  typedef itk::OrientImageFilter<ImageType,ImageType> OrientType;

    // To copy data
    int dims[3];
    int extent[6];
    double origin[3];
    double spacing[3];

    vtkSmartPointer<vtkImageData> oldPointer = NULL;
    vtkSmartPointer<vtkImageData> newImage = NULL;

    ReaderType::Pointer reader = ReaderType::New();
    // To convert itkImages to vtkImages
    ConnectorType::Pointer connector = ConnectorType::New();

    reader->SetFileName( filename.toStdString().c_str() );
    connector->SetInput(reader->GetOutput());
    try {
        connector->Update();

        // Copy the output image because the filters will be removed
        // and we want to keep the image for other vtk pipelines
        oldPointer = connector->GetOutput();
        newImage = vtkSmartPointer<vtkImageData>::New();

        oldPointer->GetDimensions(dims);
        oldPointer->GetExtent(extent);
        oldPointer->GetOrigin(origin);
        oldPointer->GetSpacing(spacing);

        newImage->SetDimensions(dims);
        newImage->SetExtent(extent);
        newImage->SetOrigin(origin);
        newImage->SetSpacing(spacing);

        newImage->DeepCopy(oldPointer);

        // set the original slices. Other orientation are build when/if required only.
    } catch ( itk::ExceptionObject & err ) {
        std::cerr << "ExceptionObject caught !" << std::endl;
        std::cerr << err << std::endl;
        return NULL;
    }

    //if (reader)
    // reader->Delete();

    oldPointer = NULL;

    return newImage;
}