]> Creatis software - creaVtk.git/blob - bbtk_creaVtk_PKG/src/bbcreaVtkMeshDeformation.cxx
#3490 MeshManager
[creaVtk.git] / bbtk_creaVtk_PKG / src / bbcreaVtkMeshDeformation.cxx
1 //===== 
2 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
3 //===== 
4 #include "bbcreaVtkMeshDeformation.h"
5 #include "bbcreaVtkPackage.h"
6 namespace bbcreaVtk
7 {
8
9 BBTK_ADD_BLACK_BOX_TO_PACKAGE(creaVtk,MeshDeformation)
10 BBTK_BLACK_BOX_IMPLEMENTATION(MeshDeformation,bbtk::AtomicBlackBox);
11 //===== 
12 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
13 //===== 
14 void MeshDeformation::Process()
15 {
16
17 // THE MAIN PROCESSING METHOD BODY
18 //   Here we simply set the input 'In' value to the output 'Out'
19 //   And print out the output value
20 // INPUT/OUTPUT ACCESSORS ARE OF THE FORM :
21 //    void bbSet{Input|Output}NAME(const TYPE&)
22 //    const TYPE& bbGet{Input|Output}NAME() const 
23 //    Where :
24 //    * NAME is the name of the input/output
25 //      (the one provided in the attribute 'name' of the tag 'input')
26 //    * TYPE is the C++ type of the input/output
27 //      (the one provided in the attribute 'type' of the tag 'input')
28 //    bbSetOutputOut( bbGetInputIn() );
29 //    std::cout << "Output value = " <<bbGetOutputOut() << std::endl;
30
31     std::vector<double> lstCenter = bbGetInputCenter();
32
33     double  s   = bbGetInputS();
34     bool    ok  = false;
35     if (bbGetInputTypeIn()==0) // direction
36     {
37         if (bbGetInputDirection().size()==3)
38         {
39             ok = !( (bbGetInputDirection()[0]==0) && (bbGetInputDirection()[1]==0) && (bbGetInputDirection()[2]==0) );
40         }
41     } // if TypeIn 0
42     
43     if (bbGetInputTypeIn()==1)  // center
44     {
45         ok = ( lstCenter.size()==3 );
46     } // if TypeIn 1
47
48     if ( (bbGetInputIn()!=NULL) && (ok==true) && (bbGetInputEdgeId()>=0) && (bbGetInputActive()==true) )
49     {
50         vtkPoints* points=bbGetInputIn()->GetPoints();
51         long    i,size=points->GetNumberOfPoints();
52         double  p[3];  // point
53         double  pb[3]; // point base
54         double  np[3]; // new point
55         double  sx,sy,sz;
56         sx = s*2;
57         sy = sx;
58         sz = sy;
59
60         double displcement_x = 0;
61         double displcement_y = 0;
62         double displcement_z = 0;
63         
64         if (bbGetInputTypeIn()==0) // Direction
65         {
66             displcement_x = bbGetInputDirection()[0];
67             displcement_y = bbGetInputDirection()[1];
68             displcement_z = bbGetInputDirection()[2];
69         } // if TypeIn 0 Direction
70                 
71                 printf(" EED MeshDeformation::Process   %ld   %ld  -   %f %f %f \n", EdgeIdBack, bbGetInputEdgeId() , lstCenter[0],lstCenter[1],lstCenter[2] );
72                 
73         if (bbGetInputTypeIn()==1) // Center
74         {
75             if (EdgeIdBack==bbGetInputEdgeId() )
76             {
77                 displcement_x = (lstCenter[0]-backLstCenter[0])/1.0;
78                 displcement_y = (lstCenter[1]-backLstCenter[1])/1.0;
79                 displcement_z = (lstCenter[2]-backLstCenter[2])/1.0;
80             } // if EdgeIdBack!=bbGetInputEdgeId()
81             backLstCenter[0] = lstCenter[0];
82             backLstCenter[1] = lstCenter[1];
83             backLstCenter[2] = lstCenter[2];
84         } // if TypeIn 1 Center
85         points->GetPoint( bbGetInputEdgeId() , pb );
86         if (EdgeIdBack!=bbGetInputEdgeId() )
87         {
88             EdgeIdBack = bbGetInputEdgeId();
89             voiIdPoints.clear();
90             for ( i=0 ; i<size ; i++)
91             {
92                 points->GetPoint(i,p);
93                 if ( (p[0]>(pb[0]-sx)) && (p[0]<(pb[0]+sx)) &&
94                      (p[1]>(pb[1]-sy)) && (p[1]<(pb[1]+sy)) &&
95                      (p[2]>(pb[2]-sz)) && (p[2]<(pb[2]+sz)) )
96                 {
97                     voiIdPoints.push_back( i );
98                 } // if
99             } // for i
100                         backLstCenter[0] = lstCenter[0];
101                         backLstCenter[1] = lstCenter[1];
102                         backLstCenter[2] = lstCenter[2];
103         } // if EdgeIdBack
104
105         if ( !((displcement_x==0) &&(displcement_y==0) && (displcement_z==0)) )
106         {
107             size=voiIdPoints.size();
108             for (i=0;i<size;i++)
109             {
110                 points->GetPoint( voiIdPoints[i] , p );
111                 double x = p[0] - pb[0]; //distance between a point and seed point
112                 double y = p[1] - pb[1];
113                 double z = p[2] - pb[2];
114                 double distance = std::sqrt(std::pow(x, 2) + std::pow(y, 2) + std::pow(z, 2));
115
116     //        https://en.wikipedia.org/wiki/Generalized_normal_distribution
117     // with Gama function
118                 
119                 double weight = exp(-distance * 1. / s);
120                 if (weight > 0.0001)
121                 {
122                     np[0] = p[0] + weight*displcement_x;
123                     np[1] = p[1] + weight*displcement_y;
124                     np[2] = p[2] + weight*displcement_z;
125                     points->SetPoint( voiIdPoints[i] , np );
126                 } // if
127             } // for
128             points->Modified();
129             bbGetInputIn()->Modified();
130         } // if distance_x y z  != 0
131     } // In != NULL    ok    active
132 }
133 //===== 
134 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
135 //===== 
136 void MeshDeformation::bbUserSetDefaultValues()
137 {
138
139 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX 
140 //    Here we initialize the input 'In' to 0
141     bbSetInputActive(true);
142     bbSetInputTypeIn(0);
143     bbSetInputIn(NULL);
144     std::vector<double> direction;
145     direction.push_back(1);
146     direction.push_back(0);
147     direction.push_back(0);
148     bbSetInputDirection(direction);
149     EdgeIdBack=-1;
150     bbSetInputEdgeId(EdgeIdBack);
151     bbSetInputS(10);
152     
153     backLstCenter.push_back(0);
154     backLstCenter.push_back(0);
155     backLstCenter.push_back(0);
156 }
157 //===== 
158 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
159 //===== 
160 void MeshDeformation::bbUserInitializeProcessing()
161 {
162
163 //  THE INITIALIZATION METHOD BODY :
164 //    Here does nothing 
165 //    but this is where you should allocate the internal/output pointers 
166 //    if any 
167
168   
169 }
170 //===== 
171 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
172 //===== 
173 void MeshDeformation::bbUserFinalizeProcessing()
174 {
175
176 //  THE FINALIZATION METHOD BODY :
177 //    Here does nothing 
178 //    but this is where you should desallocate the internal/output pointers 
179 //    if any
180   
181 }
182 }
183 // EO namespace bbcreaVtk
184
185