File: AnyDecoration.cpp

package info (click to toggle)
camitk 3.3.2-2
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 391,128 kB
  • ctags: 9,706
  • sloc: cpp: 76,454; xml: 2,665; sh: 287; ansic: 142; makefile: 99; perl: 84; sed: 20
file content (127 lines) | stat: -rw-r--r-- 4,562 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
/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2014 UJF-Grenoble 1, CNRS, TIMC-IMAG UMR 5525 (GMCAO)
 *
 * 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 <pml/StructuralComponent.h>
#include <pml/Atom.h>
#include <pml/PhysicalModel.h>
#include <pml/Component.h>
#include <pml/StructuralComponent.h>
#include <pml/MultiComponent.h>

// camitk
#include <InteractiveViewer.h>
#include <Geometry.h>
using namespace camitk;

#include "AnyDecoration.h"
#include "StructuralComponentDC.h"
#include "PMManagerDC.h"

#include <vtkUnstructuredGrid.h>
#include <vtkActor.h>

// ------------- constructor --------------
AnyDecoration::AnyDecoration(StructuralComponentDC * scdc) : Decoration(scdc) {
  mySC = scdc->getSC();

  // build 3D representation

  myGeometry = new Geometry(scdc->getName(),
                            StructuralComponentDC::structuralComponentToVtk(mySC, &atomMap),
                            scdc->getPMManagerDC()->toDCRenderingMode(mySC->getMode()));

  // set the visible mode
  myGeometry->setRenderingModes(scdc->getPMManagerDC()->toDCRenderingMode(mySC->getMode()));
  // surface /points color = SC color
  myGeometry->setActorColor(InterfaceGeometry::Surface, mySC->getColor());
  myGeometry->setActorColor(InterfaceGeometry::Points, mySC->getColor());
  // wireframe color = black
  double rgb[4];
  for (unsigned int i = 0;i < 4;i++)
    rgb[i] = 1.0;
  myGeometry->setActorColor(InterfaceGeometry::Wireframe, rgb);
}

// ------------- destructor --------------
AnyDecoration::~AnyDecoration() {
  show(false);
  dc->removeProp("AnyDecoration Surface");
  dc->removeProp("AnyDecoration Points");
  dc->removeProp("AnyDecoration Wireframe");
  delete myGeometry;
  myGeometry = NULL;
}

// ------------- update --------------
void AnyDecoration::update() {
  StructuralComponent *atoms = mySC->getAtoms();
  for (unsigned int i = 0;i < atoms->getNumberOfStructures(); i++) {
    Atom *a = (Atom *) atoms->getStructure(i);
    // find a in the map
    std::AtomVtkPointsIndexMapIterator result;
    result = atomMap.find(a);
    // get the new atom position
    if (result != atomMap.end()) {
      double pos[3];
      a->getPosition(pos);
      // set the new position in my geometry
      myGeometry->setPointPosition(result->second, pos[0], pos[1], pos[2]);
    }
  }
}

// ------------- show --------------
void AnyDecoration::show(const bool d) {
  if (d) {
    // make sure that all asked rendering are shown
    if (myGeometry->getRenderingModes() & InterfaceGeometry::Surface) {
      dc->addProp("AnyDecoration Surface", myGeometry->getActor(InterfaceGeometry::Surface));
      dc->getProp("AnyDecoration Surface")->VisibilityOn();
    }
    if (myGeometry->getRenderingModes() & InterfaceGeometry::Points) {
      dc->addProp("AnyDecoration Points", myGeometry->getActor(InterfaceGeometry::Points));
      dc->getProp("AnyDecoration Points")->VisibilityOn();
    }
    if (myGeometry->getRenderingModes() & InterfaceGeometry::Wireframe) {
      dc->addProp("AnyDecoration Wireframe", myGeometry->getActor(InterfaceGeometry::Wireframe));
      dc->getProp("AnyDecoration Wireframe")->VisibilityOn();
    }
  }
  else {
    // remove all actors
    if (dc->getProp("AnyDecoration Surface"))
      dc->getProp("AnyDecoration Surface")->VisibilityOff();
    if (dc->getProp("AnyDecoration Points"))
      dc->getProp("AnyDecoration Points")->VisibilityOff();
    if (dc->getProp("AnyDecoration Wireframe"))
      dc->getProp("AnyDecoration Wireframe")->VisibilityOff();
  }
}


// ------------- setColor --------------
void AnyDecoration::setColor(const double r, const double g, const double b) {
  myGeometry->setColor(r, g, b);
}