File: main.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 (103 lines) | stat: -rw-r--r-- 4,192 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
/*****************************************************************************
 * $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$
 ****************************************************************************/

// -- Core stuff
#include <ExtensionManager.h>
#include <QApplication>
#include <Core.h>
#include <Application.h>
#include <Action.h>
#include <ImageComponent.h>
#include <Log.h>

using namespace camitk;

int main(int argc, char* argv[]) {
    // Example of a non-gui program that
    // - open a MHA image file
    // - resample it to half its size using the "Resample" action
    // - save the resampled image in the current working directory
    //
    // note: it still requires a graphics renderer even if nothing is displayed
    // even if no event loop are started.
    //
    Log::getLogger()->setLogLevel(InterfaceLogger::INFO);
    Application app("nogui", argc, argv);

    // open an image volume
    ComponentExtension* cp = ExtensionManager::getComponentExtension("mha");
    Component* image = cp->open(QFileInfo(Core::getTestDataDir() + "/brain.mha").absoluteFilePath());

    // get the Resample action
    Action* resampleAction = Application::getAction("Resample");

    if (resampleAction != nullptr) {
        CAMITK_INFO_ALT(QString("Applying action: %1").arg(resampleAction->getName()))

        // set the input
        resampleAction->setInputComponent(image);

        //app.getMainWindow()->setCentralWidget(resampleAction->getWidget());
        //app.exec();

        // set the parameter
        ImageComponent* inputImage = dynamic_cast<ImageComponent*>(image);
        // Note: as the Resample action takes ImageComponent as inputs, the dynamic_cast is
        // guaranty to work perfectly well.
        int* dims = inputImage->getImageData()->GetDimensions();
        double factor = 0.5;
        resampleAction->setParameterValue("New Image X Dimension", int(dims[0]*factor));
        resampleAction->setParameterValue("New Image Y Dimension", int(dims[1]*factor));
        resampleAction->setParameterValue("New Image Z Dimension", std::max(1, int(dims[2]*factor)));

        // apply the action
        if (resampleAction->applyInPipeline() != Action::SUCCESS) {
            CAMITK_WARNING_ALT("Error while applying the Resample action");
            return EXIT_FAILURE;
        }

        // get the last instantiated top level component, i.e., the result of the resampleAction
        Component* resampled = Application::getTopLevelComponents().last();

        // save the resampled image into the current directory
        QString outputFileName = QDir::currentPath() + "/" + QFileInfo(image->getFileName()).completeBaseName() + "-output." + QFileInfo(image->getFileName()).suffix();

        resampled->setFileName(outputFileName);

        if (Application::save(resampled)) {
            CAMITK_INFO_ALT(QString("Resampled image succesfully saved to %1").arg(resampled->getFileName()))
            return EXIT_SUCCESS;
        }
        else {
            CAMITK_WARNING_ALT("Error while saving the resampled image");
            return EXIT_FAILURE;
        }
    }
    else {
        CAMITK_WARNING_ALT("Cannot find Action \"Resample\". Please make sure the \"ResampleExtension\" is present in your action extension installation path.")
        return EXIT_FAILURE;
    }
}