File: PickMe.cpp

package info (click to toggle)
camitk 5.2.0-5
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 358,388 kB
  • sloc: cpp: 86,984; xml: 1,295; sh: 1,280; ansic: 142; makefile: 112; perl: 84; sed: 20
file content (91 lines) | stat: -rw-r--r-- 3,621 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
/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2024 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$
 ****************************************************************************/

// CamiTK includes
#include "PickMe.h"

#include <vtkCell.h>

using namespace camitk;

// --------------- Constructor -------------------
PickMe::PickMe(const QString& file) : VtkMeshComponent(file) {
    // initially all the point data have a value of 0.0 (everything is blue)
    // when the user pick a cell or a point in 3D, the point data are modified
    // using the point index (for point picking) or average of cell's point index
    // (for cell picking)
    // -> the mesh starts to show colors!
    initPointData();
}

// --------------- initPointData -------------------
void PickMe::initPointData() {
    // create a new vtkDataArray to store the demo value
    demoPointData = vtkSmartPointer<vtkDoubleArray>::New();

    int numberOfPoints = getPointSet()->GetNumberOfPoints();
    demoPointData->SetNumberOfValues(numberOfPoints);
    demoPointData->SetName("Pick Me Data");
    demoPointData->FillComponent(0, 0.0);

    // add the data to the mesh
    addPointData(demoPointData->GetName(), demoPointData);
}


// -------------------- pointPicked --------------------
void PickMe::pointPicked(vtkIdType pointId, bool sel/* picking does not change the selection state, do not bother with the 2nd parameter*/) {
    MeshComponent::pointPicked(pointId, sel);

    // picking show demo point data
    demoPointData->SetValue(pointId, 1.0 - double (pointId) / double (getPointSet()->GetNumberOfPoints()));

    refreshPointData();
}

// -------------------- cellPicked --------------------
void PickMe::cellPicked(vtkIdType cellId, bool  sel /* picking does not change the selection state, do not bother with the 2nd parameter*/) {
    MeshComponent::cellPicked(cellId, sel);

    // change data for all cell points
    vtkSmartPointer<vtkCell> c = getPointSet()->GetCell(cellId);
    for (vtkIdType i = 0; i < c->GetNumberOfPoints(); i++) {
        vtkIdType pointId = c->GetPointId(i);
        demoPointData->SetValue(pointId, double (pointId) / double (getPointSet()->GetNumberOfPoints()));
    }

    refreshPointData();
}

// -------------------- refreshPointData --------------------
void PickMe::refreshPointData() {
    // force the modification flag of the vtkDoubleArray so that the range is going to be recomputed
    // This is needed so that VTK can "discover" that the data have changed (calls to SetValue do not modify the internal time stamp)
    demoPointData->Modified();

    // force the data visualization to be updated
    setDataRepresentationVisibility(MeshComponent::POINTS, demoPointData->GetName(), true);
}