0026936: Drawbacks of inlining in new type system in OCCT 7.0 -- automatic
[occt.git] / samples / mfc / standard / Common / User_Cylinder.cxx
1 #include <stdafx.h>
2
3 #include <User_Cylinder.hxx>
4
5 // Implementation of Handle and type mgt
6 //
7
8 IMPLEMENT_STANDARD_RTTIEXT(User_Cylinder,AIS_InteractiveObject)
9
10 #include "ColoredMeshDlg.h"
11
12 #include <Graphic3d_ArrayOfTriangles.hxx>
13 #include <Graphic3d_StructureManager.hxx>
14 #include <PrsMgr_PresentationManager3d.hxx>
15 #include <StdPrs_ToolTriangulatedShape.hxx>
16 #include <Poly_Connect.hxx>
17 #include <TColgp_Array1OfPnt.hxx>
18 #include <Poly_Triangulation.hxx>
19 #include <TColgp_Array1OfDir.hxx>
20 #include <GProp_PGProps.hxx>
21 #include <Graphic3d_Array1OfVertex.hxx>
22 #include <Quantity_Color.hxx>
23 #include <BRepMesh_IncrementalMesh.hxx>
24 #include <Prs3d.hxx>
25
26 #include <AIS_GraphicTool.hxx>
27
28 // Constructors implementation
29 //
30
31 User_Cylinder::User_Cylinder(const Standard_Real R, const Standard_Real H) :
32 AIS_InteractiveObject(PrsMgr_TOP_ProjectorDependant)
33 {
34   BRepPrimAPI_MakeCylinder S(R,H);
35   myShape = S.Shape();
36   SetHilightMode(0);
37   myDrawer->SetShadingAspect(new Prs3d_ShadingAspect());
38   myPlanarFaceColor = Quantity_NOC_FIREBRICK3;
39   myCylindricalFaceColor = Quantity_NOC_GRAY;
40 }
41
42 User_Cylinder::User_Cylinder(const gp_Ax2 CylAx2, const Standard_Real R, const Standard_Real H) :
43 AIS_InteractiveObject(PrsMgr_TOP_ProjectorDependant)
44
45 {
46   BRepPrimAPI_MakeCylinder S(CylAx2,R,H);
47   BRepBuilderAPI_NurbsConvert aNurbsConvert(S.Shape());
48   myShape = aNurbsConvert.Shape();
49   SetHilightMode(0);
50   myDrawer->SetShadingAspect(new Prs3d_ShadingAspect());
51   myPlanarFaceColor = Quantity_NOC_FIREBRICK3;
52   myCylindricalFaceColor = Quantity_NOC_KHAKI4;
53 }
54
55 void User_Cylinder::Compute(const Handle(PrsMgr_PresentationManager3d)& /*aPresentationManager*/,
56                             const Handle(Prs3d_Presentation)& aPresentation,
57                             const Standard_Integer aMode ) 
58 {
59   switch (aMode) {
60 case 0:
61   StdPrs_WFShape::Add(aPresentation,myShape, myDrawer );
62   break;
63 case 1:
64   {
65
66     Standard_Real aTransparency = Transparency();
67     Graphic3d_NameOfMaterial aMaterial = Material();
68     myDrawer->SetShadingAspectGlobal(Standard_False);
69     TopExp_Explorer Ex;
70     Handle(Geom_Surface) Surface;
71
72     for (Ex.Init(myShape,TopAbs_FACE); Ex.More(); Ex.Next())
73     {
74
75       Surface = BRep_Tool::Surface(TopoDS::Face(Ex.Current()));
76       myDrawer->ShadingAspect()->SetMaterial(aMaterial);
77       if (Surface->IsKind(STANDARD_TYPE(Geom_Plane)))
78         myDrawer->ShadingAspect()->SetColor(myPlanarFaceColor);
79       else
80         myDrawer->ShadingAspect()->SetColor(myCylindricalFaceColor);
81
82
83       myDrawer->ShadingAspect()->SetTransparency (aTransparency);
84       StdPrs_ShadedShape::Add(aPresentation,Ex.Current(), myDrawer);
85     }
86     break;
87
88   }
89 case 6: //color
90   {
91     BRepTools::Clean(myShape);
92     BRepTools::Update(myShape);
93
94     Handle(Graphic3d_StructureManager) aStrucMana = GetContext()->MainPrsMgr()->StructureManager();
95
96     Handle(Graphic3d_Group) mygroup = Prs3d_Root::CurrentGroup(aPresentation);
97     myAspect = (new Prs3d_ShadingAspect())->Aspect();
98     Graphic3d_MaterialAspect material = myAspect->FrontMaterial();
99     material.SetReflectionModeOff(Graphic3d_TOR_AMBIENT);
100     material.SetReflectionModeOff(Graphic3d_TOR_DIFFUSE);
101     material.SetReflectionModeOff(Graphic3d_TOR_SPECULAR);
102     material.SetReflectionModeOff(Graphic3d_TOR_EMISSION);
103     myAspect->SetFrontMaterial(material);
104
105     mygroup->SetPrimitivesAspect(myAspect);
106     myAspect->SetEdgeOn();
107
108     myDeflection = Prs3d::GetDeflection(myShape,myDrawer);
109     BRepMesh_IncrementalMesh(myShape,myDeflection);
110
111     myX1OnOff = Standard_False;
112     myXBlueOnOff = Standard_False;
113     myXGreenOnOff =Standard_False;
114     myXRedOnOff = Standard_False;
115     myY1OnOff = Standard_False;
116     myYBlueOnOff = Standard_False;
117     myYGreenOnOff = Standard_False;
118     myYRedOnOff = Standard_False;
119     myZ1OnOff = Standard_False;
120     myZBlueOnOff =Standard_False;
121     myZGreenOnOff = Standard_False;
122     myZRedOnOff = Standard_False;
123
124     CColoredMeshDlg Dlg(NULL);
125     Dlg.DoModal();
126
127     myX1OnOff = Dlg.X1OnOff;
128
129     myXBlueOnOff = Dlg.m_CheckXBlueOnOff;
130     myXGreenOnOff = Dlg.m_CheckXGreenOnOff;
131     myXRedOnOff = Dlg.m_CheckXRedOnOff;
132
133     myY1OnOff = Dlg.Y1OnOff;
134
135     myYBlueOnOff = Dlg.m_CheckYBlueOnOff;
136     myYGreenOnOff = Dlg.m_CheckYGreenOnOff;
137     myYRedOnOff = Dlg.m_CheckYRedOnOff;
138
139     myZ1OnOff = Dlg.Z1OnOff;
140
141     myZBlueOnOff = Dlg.m_CheckZBlueOnOff;
142     myZGreenOnOff = Dlg.m_CheckZGreenOnOff;
143     myZRedOnOff = Dlg.m_CheckZRedOnOff;
144
145     // Adds a triangulation of the shape myShape to its topological data structure.
146     // This triangulation is computed with the deflection myDeflection.
147
148 #ifdef DEBUG
149     cout <<"Deflection = " << myDeflection << "\n" << endl;
150 #endif
151
152     Standard_Integer NumFace;
153     TopExp_Explorer ExpFace;
154
155     //khr -->
156
157     gp_Pnt H (0,0,0);
158     gp_Pnt B (0,0,1000000000);
159     for( NumFace=0,ExpFace.Init(myShape,TopAbs_FACE); ExpFace.More(); ExpFace.Next(),NumFace++ )
160     {
161       TopoDS_Face myFace = TopoDS::Face(ExpFace.Current());
162       TopLoc_Location myLocation = myFace.Location();
163
164 #ifdef DEBUG
165       cout << "J\'explore actuellement la face " << NumFace << "\n" << endl;
166 #endif
167       Handle(Poly_Triangulation) myT = BRep_Tool::Triangulation(myFace, myLocation);
168       // Returns the Triangulation of the face. It is a null handle if there is no triangulation.
169
170       if (myT.IsNull())
171       {
172 #ifdef DEBUG
173         // cout << "Triangulation of the face "<< i <<" is null \n"<< endl;
174 #endif
175         return;
176       }
177
178       const TColgp_Array1OfPnt& Nodes= myT->Nodes();
179
180       const Poly_Array1OfTriangle& triangles = myT->Triangles();
181
182       Standard_Integer nnn = myT->NbTriangles(); // nnn : nombre de triangles
183       Standard_Integer nt, n1, n2, n3 = 0;// nt : triangle courant
184       // ni : sommet i du triangle courant
185       //recherche du pt "haut" et du pt "bas
186       for (nt = 1; nt <= nnn; nt++)
187       {
188         // triangles(nt).Get(n1,n2,n3); // le triangle est n1,n2,n3
189
190         if (myFace.Orientation() == TopAbs_REVERSED) // si la face est "reversed"
191           triangles(nt).Get(n1,n3,n2); // le triangle est n1,n3,n2
192         else
193           triangles(nt).Get(n1,n2,n3); // le triangle est n1,n2,n3
194
195         if (TriangleIsValid (Nodes(n1),Nodes(n2),Nodes(n3)) )
196         { // Associates a vertexNT to each node
197           gp_Pnt p = Nodes(n1).Transformed(myLocation.Transformation());
198           gp_Pnt q = Nodes(n2).Transformed(myLocation.Transformation());
199           gp_Pnt r = Nodes(n3).Transformed(myLocation.Transformation());
200
201           if (p.Z() > H.Z()) H=p;
202           if (q.Z() > H.Z()) H=q;
203           if (r.Z() > H.Z()) H=r;
204           if (p.Z() < B.Z()) B=p;
205           if (q.Z() < B.Z()) B=q;
206           if (r.Z() < B.Z()) B=r;
207         }
208       }
209     }
210
211     //khr <--
212
213
214     for( NumFace=0,ExpFace.Init(myShape,TopAbs_FACE); ExpFace.More(); ExpFace.Next(),NumFace++ )
215     {
216       TopoDS_Face myFace = TopoDS::Face(ExpFace.Current());
217       TopLoc_Location myLocation = myFace.Location();
218
219 #ifdef DEBUG
220       cout << "J\'explore actuellement la face " << NumFace << "\n" << endl;
221 #endif
222       Handle(Poly_Triangulation) myT = BRep_Tool::Triangulation(myFace, myLocation);
223       // Returns the Triangulation of the face. It is a null handle if there is no triangulation.
224
225       if (myT.IsNull())
226       {
227 #ifdef DEBUG
228         //cout << "Triangulation of the face "<< i <<" is null \n"<< endl;
229 #endif
230         return;
231       }
232       Poly_Connect pc(myT);
233       const TColgp_Array1OfPnt& Nodes= myT->Nodes();
234       BAR = GProp_PGProps::Barycentre(Nodes);
235
236
237       //const TColgp_Array1OfPnt2d& UVNodes = myT->UVNodes();
238       const Poly_Array1OfTriangle& triangles = myT->Triangles();
239       TColgp_Array1OfDir myNormal(Nodes.Lower(), Nodes.Upper());
240
241       StdPrs_ToolTriangulatedShape::Normal(myFace, pc, myNormal);
242       BRepTools::UVBounds(myFace,Umin, Umax, Vmin, Vmax);
243       dUmax = (Umax - Umin);
244       dVmax = (Vmax - Vmin);
245
246       Standard_Integer nnn = myT->NbTriangles(); // nnn : nombre de triangles
247       Standard_Integer nt, n1, n2, n3 = 0;// nt : triangle courant
248       // ni : sommet i du triangle courant
249
250       //recherche du pt "haut" et du pt "bas
251       // gp_Pnt H (0,0,0);
252       // gp_Pnt B (0,0,1000000000);
253
254       for (nt = 1; nt <= nnn; nt++)
255       {
256         // triangles(nt).Get(n1,n2,n3); // le triangle est n1,n2,n3
257         if (myFace.Orientation() == TopAbs_REVERSED) // si la face est "reversed"
258           triangles(nt).Get(n1,n3,n2); // le triangle est n1,n3,n2
259         else
260           triangles(nt).Get(n1,n2,n3); // le triangle est n1,n2,n3
261
262         if (TriangleIsValid (Nodes(n1),Nodes(n2),Nodes(n3)) )
263         { // Associates a vertexNT to each node
264           gp_Pnt p = Nodes(n1).Transformed(myLocation.Transformation());
265           gp_Pnt q = Nodes(n2).Transformed(myLocation.Transformation());
266           gp_Pnt r = Nodes(n3).Transformed(myLocation.Transformation());
267         }
268       }
269
270       Handle(Graphic3d_ArrayOfTriangles) aOP = new Graphic3d_ArrayOfTriangles(3 * nnn, 0, Standard_True, Standard_True);
271
272       for (nt = 1; nt <= nnn; nt++)
273       {
274 #ifdef DEBUG
275         cout << "On traite actuellement le triangle : "<< nt <<"\n";
276 #endif
277         if (myFace.Orientation() == TopAbs_REVERSED) // si la face est "reversed"
278           triangles(nt).Get(n1,n3,n2); // le triangle est n1,n3,n2
279         else
280           triangles(nt).Get(n1,n2,n3); // le triangle est n1,n2,n3
281
282         if (TriangleIsValid (Nodes(n1),Nodes(n2),Nodes(n3)) )
283         { // Associates a vertexNT to each node
284
285           TColgp_Array1OfPnt Points(1,3);
286
287           gp_Pnt p = Nodes(n1).Transformed(myLocation.Transformation());
288           gp_Pnt q = Nodes(n2).Transformed(myLocation.Transformation());
289           gp_Pnt r = Nodes(n3).Transformed(myLocation.Transformation());
290
291           Points(1).SetCoord(p.X(), p.Y(), p.Z());
292           Points(2).SetCoord(q.X(), q.Y(), q.Z());
293           Points(3).SetCoord(r.X(), r.Y(), r.Z());
294
295           aOP->AddVertex(Points(1), myNormal(n1), Color(p,B.Z(),H.Z(),Dlg.Colorization));
296           aOP->AddVertex(Points(2), myNormal(n2), Color(q,B.Z(),H.Z(),Dlg.Colorization));
297           aOP->AddVertex(Points(3), myNormal(n3), Color(r,B.Z(),H.Z(),Dlg.Colorization));
298         } // end of "if the triangle is valid
299       } // end of the "parcours" of the triangles
300
301       Prs3d_Root::CurrentGroup (aPresentation)->AddPrimitiveArray (aOP);
302
303       mygroup->SetGroupPrimitivesAspect(myAspect);
304     }// end of the exploration of the shape in faces
305
306     break;
307   }
308   }
309 }
310
311 void User_Cylinder::Compute(const Handle(Prs3d_Projector)& aProjector,
312                             const Handle(Prs3d_Presentation)& aPresentation)
313 {
314   myDrawer->EnableDrawHiddenLine();
315   StdPrs_HLRPolyShape::Add(aPresentation,myShape,myDrawer,aProjector);
316 }
317
318 void User_Cylinder::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
319                                     const Standard_Integer aMode)
320 {
321   switch(aMode)
322   {
323   case 0:
324     StdSelect_BRepSelectionTool::Load(aSelection,this,myShape,TopAbs_SHAPE, 0.01, 0.1);
325     break;
326   case 4:
327     StdSelect_BRepSelectionTool::Load(aSelection,this,myShape,TopAbs_FACE, 0.01, 0.1);
328     break;
329   }
330 }
331
332 Standard_Integer User_Cylinder::NbPossibleSelection() const
333 {
334   return 2;
335 }
336
337 Standard_Boolean User_Cylinder::AcceptShapeDecomposition() const
338 {
339   return Standard_True;
340 }
341
342 void User_Cylinder::SetPlanarFaceColor(const Quantity_Color acolor)
343 {
344   myPlanarFaceColor = acolor;
345 }
346
347 void User_Cylinder::SetCylindricalFaceColor(const Quantity_Color acolor)
348 {
349   myCylindricalFaceColor = acolor;
350 }
351
352 Standard_Boolean User_Cylinder::TriangleIsValid(const gp_Pnt& P1, const gp_Pnt& P2, const gp_Pnt& P3) const
353
354   gp_Vec V1(P1,P2);// V1=(P1,P2)
355   gp_Vec V2(P2,P3);// V2=(P2,P3)
356   gp_Vec V3(P3,P1);// V3=(P3,P1)
357
358   if ((V1.SquareMagnitude() > 1.e-10) && (V2.SquareMagnitude() > 1.e-10) && (V3.SquareMagnitude() > 1.e-10))
359   {
360     V1.Cross(V2);// V1 = Normal 
361     if (V1.SquareMagnitude() > 1.e-10)
362       return Standard_True;
363     else
364       return Standard_False;
365   }
366   else
367     return Standard_False;
368 }
369
370 Quantity_Color User_Cylinder::Color(gp_Pnt& thePoint,Standard_Real AltMin,Standard_Real AltMax,
371                                     const Standard_Integer ColorizationMode) 
372 {
373   red =1; //initializing colors parameters
374   green=1;
375   blue =1;
376   switch ( ColorizationMode)
377   {
378   case 0 : //normal, vert/maron
379     {
380       Standard_Real Alt= thePoint.Z();
381
382       Standard_Real AltDelta;
383
384       AltDelta = AltMax-AltMin;
385
386       red = 0.5- ((0.5*(AltMax-Alt))/(AltDelta));
387       //Standard_Real A = 7*Alt-7*AltMin;
388       green = (3*AltMax-AltMin)/(3*AltMax-AltMin+(7*Alt-7*AltMin));
389       blue = 0 ;
390
391       Quantity_Color color;
392       color.SetValues(red,green,blue, Quantity_TOC_RGB);
393       return color;
394       break;
395     }//end case 0
396
397   case 1 : //mer-neige
398     {
399       Standard_Real Alt= thePoint.Z();
400
401       Standard_Real b =AltMax-AltMin;
402       Standard_Real a= AltMax-thePoint.Z();
403
404       red =1;
405       green=1;
406       blue =1;
407       if (0<a && a <= (b/5))
408       {
409         red = 1;
410         green = 1;
411         blue = 1;
412       }
413       else if ((b/5)<a && a <= (2*b/5))
414       {
415         red = .55;
416         green = 0.3;
417         blue = 0;
418       }
419       else if ((2*b/5)<a && a <= (18*b/20))
420       {
421         green =1/(((7/(3*AltMax-AltMin))*Alt)+(1-(7*AltMin/(3*AltMax-AltMin))));
422         red = 1/(((1000/(AltMax-AltMin))*Alt)+1000*(1-(AltMin/(AltMax-AltMin))));
423         blue = 0;
424       }
425       else if ((18*b/20)<a && a <= (18.5*b/20))
426       {
427         red = 0.75;
428         green = 0.66;
429         blue = 0;
430       }
431
432       else if ((18.5*b/20)<a && a <= b)
433       {
434         red = 0.25;
435         green = .66;
436         blue = 1;
437       }
438       Quantity_Color color;
439       color.SetValues(red,green,blue, Quantity_TOC_RGB);
440       return color;
441       break;
442     }//end case 1
443
444   case 2 :
445     {
446       gp_Pnt P (85.,0.,-105.);
447       gp_Vec TheVect ( P, thePoint);
448       Standard_Real CoordX;
449       Standard_Real CoordY;
450       Standard_Real CoordZ;
451
452       CoordX = TheVect.X();
453       CoordY = TheVect.Y();
454       CoordZ = TheVect.Z();
455
456       Standard_Real Distance = BAR.Distance ( P);
457
458       Standard_Real a =fabs(CoordX);
459       Standard_Real b =fabs(CoordY);
460       Standard_Real c =fabs(CoordZ);
461
462       Standard_Real xx = a / Max(Distance,a); //(Max (Distance, Maxi));
463       Standard_Real yy = b / Max(Distance,b); //(Max (Distance, Maxi));
464       Standard_Real zz = c / Max(Distance,c); //(Max (Distance, Maxi));
465
466
467       if (myXRedOnOff)
468         red = xx;
469       else if (myXGreenOnOff)
470         green =xx;
471       else if (myXBlueOnOff)
472         blue=xx;
473
474       if (myYRedOnOff)
475         red = yy;
476       else if (myYGreenOnOff)
477         green = yy;
478       else if (myYBlueOnOff)
479         blue = yy;
480
481       if (myZRedOnOff)
482         red = zz;
483       else if (myZGreenOnOff)
484         green = zz;
485       else if (myZBlueOnOff)
486         blue = zz;
487       Quantity_Color color;
488       color.SetValues(red,green,blue, Quantity_TOC_RGB);
489       return color;
490       break;
491     }//end case 2
492   }//end switch
493
494   Quantity_Color c;
495   return c;
496 }
497
498 void User_Cylinder::SetColor(const Quantity_Color &aColor)
499 {
500   AIS_InteractiveObject::SetColor(aColor);
501   SetPlanarFaceColor(aColor);
502   SetCylindricalFaceColor(aColor);
503 }