File: ItkImageComponent.cpp

package info (click to toggle)
camitk 6.0.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 389,496 kB
  • sloc: cpp: 103,476; sh: 2,448; python: 1,618; xml: 984; makefile: 128; perl: 84; sed: 20
file content (130 lines) | stat: -rw-r--r-- 4,272 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
130
/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2025 Univ. Grenoble Alpes, CNRS, Grenoble INP - UGA, TIMC, 38000 Grenoble, France
 *
 * 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>
#include <AbortException.h>

using namespace camitk;

//-- Qt
#include <QFileInfo>

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

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

// include Log.h last in order to solve MSVC wingdi.h macro redefinition of ERROR
#include <Log.h>

// -------------------- constructor --------------------
ItkImageComponent::ItkImageComponent(const QString& fileName)
    : ImageComponent(fileName, QFileInfo(fileName).baseName()) {
    readImageData(fileName);
}

// -------------------- readImageData --------------------
void ItkImageComponent::readImageData(const QString& filename) {
    if (!filename.isEmpty()) {
        vtkSmartPointer<vtkImageData> img = readVolume(filename);
        if (img) {
            setImageData(img, false);
        }
        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 = nullptr;
    vtkSmartPointer<vtkImageData> newImage = nullptr;

    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::stringstream buffer;
        err.Print(buffer);
        CAMITK_ERROR(tr("ITK exception caught while reading the volume:\n%1\n%2").arg(err.what(), buffer.str().c_str()))
        return nullptr;
    }

    oldPointer = nullptr;

    return newImage;
}