File: RawImageComponent.cpp

package info (click to toggle)
camitk 6.0.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 389,508 kB
  • sloc: cpp: 103,476; sh: 2,448; python: 1,618; xml: 984; makefile: 128; perl: 84; sed: 20
file content (121 lines) | stat: -rw-r--r-- 4,140 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
/*****************************************************************************
 * $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 "RawImageComponent.h"

#include <AbortException.h>

#include <QFileInfo>

// -- VTK stuff
// disable warning generated by clang about the surrounded headers
#include <CamiTKDisableWarnings>
#include <vtkImageReader2.h>
#include <CamiTKReEnableWarnings>

using namespace camitk;

// -------------------- constructor --------------------
RawImageComponent::RawImageComponent(const QString& fileName)
    : ImageComponent(fileName, QFileInfo(fileName).baseName()) {
    if (!fileName.isNull()) {
        // restore the normal cursor
        QApplication::restoreOverrideCursor();

        myDialog = new RawDataDialog(fileName);
        if (myDialog->exec() == QDialog::Accepted) {
            readImageData();
        }
        else {
            throw AbortException("Opening cancelled by user\n");
        }
    }
    else {
        throw AbortException("No file name\n");
    }
}

// -------------------- destructor --------------------
RawImageComponent::~RawImageComponent() {
    if (myDialog) {
        delete (myDialog);
    }
}

// -------------------- readImageData --------------------
void RawImageComponent::readImageData() {
    QString filename       = myDialog->getFilename();
    int dimX               = myDialog->getDimX();
    int dimY               = myDialog->getDimY();
    int dimZ               = myDialog->getDimZ();
    int voxelType          = myDialog->getVoxelType();
    bool bigEndianType     = myDialog->isBigEndian();
    int nbScalarComponents = myDialog->getNbScalarComponents();
    double voxelSizeX      = myDialog->getVoxelSizeX();
    double voxelSizeY      = myDialog->getVoxelSizeY();
    double voxelSizeZ      = myDialog->getVoxelSizeZ();
    double Ox              = myDialog->getOriginX();
    double Oy              = myDialog->getOriginY();
    double Oz              = myDialog->getOriginZ();
    int headerSize         = myDialog->getHeaderSize();
    bool lowerLeftOrigin   = myDialog->isLowerLeftOrigin();
    orientation            = myDialog->getOrientation();

    // Reader and image variables
    vtkSmartPointer<vtkImageReader2> reader = vtkSmartPointer<vtkImageReader2>::New();
    vtkSmartPointer<vtkImageData> image = nullptr;

    reader->SetFileName(filename.toStdString().c_str());
    if (dimZ > 1) {
        reader->SetFileDimensionality(3);
    }
    reader->SetDataExtent(0, dimX - 1, 0, dimY - 1, 0, dimZ - 1);
    reader->SetDataScalarType(voxelType);
    if (bigEndianType) {
        reader->SetDataByteOrderToBigEndian();
    }
    reader->SetNumberOfScalarComponents(nbScalarComponents);
    reader->SetDataSpacing(voxelSizeX, voxelSizeY, voxelSizeZ);
    reader->SetDataOrigin(Ox, Oy, Oz);
    if (lowerLeftOrigin) {
        reader->FileLowerLeftOn();
    }
    else {
        reader->FileLowerLeftOff();
    }
    reader->SetHeaderSize(headerSize);
    image = reader->GetOutput();

    try {
        reader->Update();
    }
    catch (...) {
        throw AbortException("VTK was unable to read file " + filename.toStdString());
    }

    setImageData(image, false, orientation);
}