File: ComputeNeighborValueAction.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 (119 lines) | stat: -rw-r--r-- 4,377 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
/*****************************************************************************
 * $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 "ComputeNeighborValueAction.h"

#include <Property.h>
#include <Log.h>

using namespace camitk;


// --------------- Constructor -------------------
ComputeNeighborValueAction::ComputeNeighborValueAction(ActionExtension* extension) : Action(extension) {
    // Setting name, description and input component
    setName(tr("Compute Average Neighbor Value"));
    setDescription("Compute the average voxels value in the neiborhoods of the selected voxel. Show results on the log console.");
    setComponentClassName("ImageComponent");

    // Setting classification family and tags
    setFamily(tr("Tutorial"));
    addTag(tr("Neighborhood"));
    addTag(tr("Voxel Value"));

    // Setting the action's parameters
    addParameter(new Property(tr("Radius"), QVariant(5.0), tr("The radius that defines the distance around which the average value is computed"), tr("Number of Voxels")));

}

// --------------- destructor -------------------
ComputeNeighborValueAction::~ComputeNeighborValueAction() {
    // Do not do anything yet.
    // Delete stuff if you create stuff
    // (except if you use smart pointers of course !!)
}

// --------------- apply -------------------
Action::ApplyStatus ComputeNeighborValueAction::apply() {

    for (Component* comp : getTargets()) {
        ImageComponent* input = dynamic_cast<ImageComponent*>(comp);
        process(input);
    }

    return SUCCESS;
}

// --------------- process -------------------
void ComputeNeighborValueAction::process(ImageComponent* comp) {
    // Get the parameters
    double r = getParameterValue("Radius").toDouble();

    // get the last picked pixel index
    int x0, y0, z0;
    comp->getLastPixelPicked(&x0, &y0, &z0);

    // get the image dimensions
    // Getting the input image
    vtkSmartPointer<vtkImageData> inputImage  = comp->getImageData();
    int dims[3];
    // int numberOfScalarComponents = inputImage->GetNumberOfScalarComponents();
    inputImage->GetDimensions(dims);

    double avgValue = 0.0;
    int nbPixel = 0;

    // check the pixel has been selected
    if (x0 != -1 && y0 != -1 && z0 != -1) {

        for (int z = 0; z < dims[2]; z++) {
            for (int y = 0; y < dims[1]; y++) {
                for (int x = 0; x < dims[0]; x++) {
                    // test pixel is in the neighborood of the selected pixel
                    if (distance(x0, x, y0, y, z0, z) < r) {
                        avgValue += comp->getImageData()->GetScalarComponentAsDouble(x, y, z, 0);
                        nbPixel++;
                    }
                }
            }
        }

        avgValue = avgValue / (double) nbPixel;

        // display the result on the console
        CAMITK_INFO(tr("Pixel selected: (%1,%2,%3)\n"
                       "Average pixel values in the neighborhood of %4 pixels: %5").arg(QString("%1").arg(x0, 3), QString("%1").arg(y0, 3), QString("%1").arg(z0, 3), QString::number(r), QString("%1").arg(avgValue, 5)))

    }
    refreshApplication();
}

// --------------- distance -------------------
double ComputeNeighborValueAction::distance(int x1, int x2, int y1, int y2, int z1, int z2) {
    return sqrt((double)((x2 - x1) * (x2 - x1)) +
                (double)((y2 - y1) * (y2 - y1)) +
                (double)((z2 - z1) * (z2 - z1))
               );
}