67038d8fb8002a1db935752b4d2ba311304ad7a9
[occt.git] / src / AIS / AIS_PointCloud.cxx
1 // Created on: 2014-08-13
2 // Created by: Maxim GLIBIN
3 // Copyright (c) 2014 OPEN CASCADE SAS
4 //
5 // This file is part of Open CASCADE Technology software library.
6 //
7 // This library is free software; you can redistribute it and/or modify it under
8 // the terms of the GNU Lesser General Public License version 2.1 as published
9 // by the Free Software Foundation, with special exception defined in the file
10 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
11 // distribution for complete text of the license and disclaimer of any warranty.
12 //
13 // Alternatively, this file may be used under the terms of Open CASCADE
14 // commercial license or contractual agreement.
15
16 #include <AIS_PointCloud.hxx>
17
18 #include <AIS_GraphicTool.hxx>
19 #include <Graphic3d_AspectFillArea3d.hxx>
20 #include <Graphic3d_AspectMarker3d.hxx>
21 #include <Graphic3d_Group.hxx>
22 #include <Prs3d_Drawer.hxx>
23 #include <Prs3d_PointAspect.hxx>
24 #include <Prs3d_Presentation.hxx>
25 #include <Prs3d_Root.hxx>
26 #include <Prs3d_ShadingAspect.hxx>
27 #include <PrsMgr_ModedPresentation.hxx>
28 #include <PrsMgr_Presentations.hxx>
29 #include <Select3D_SensitiveBox.hxx>
30 #include <Select3D_SensitivePrimitiveArray.hxx>
31 #include <SelectMgr_EntityOwner.hxx>
32 #include <SelectMgr_Selection.hxx>
33 #include <StdPrs_BndBox.hxx>
34
35
36 IMPLEMENT_STANDARD_RTTIEXT(AIS_PointCloud,AIS_InteractiveObject)
37
38 //==================================================
39 // Function: AIS_PointCloud
40 // Purpose : Constructor
41 //==================================================
42 AIS_PointCloud::AIS_PointCloud()
43 {
44   // override default point style to Aspect_TOM_POINT
45   myDrawer->SetPointAspect (new Prs3d_PointAspect (Aspect_TOM_POINT, Quantity_NOC_YELLOW, 1.0));
46
47   SetDisplayMode (AIS_PointCloud::DM_Points);
48   SetHilightMode (AIS_PointCloud::DM_BndBox);
49 }
50
51 //=======================================================================
52 //function : GetPoints
53 //purpose  :
54 //=======================================================================
55 const Handle(Graphic3d_ArrayOfPoints) AIS_PointCloud::GetPoints() const
56 {
57   return myPoints;
58 }
59
60 //=======================================================================
61 //function : GetBoundingBox
62 //purpose  :
63 //=======================================================================
64 Bnd_Box AIS_PointCloud::GetBoundingBox() const
65 {
66   return myBndBox;
67 }
68
69 //! Auxiliary method
70 static inline Bnd_Box getBoundingBox (const Handle(Graphic3d_ArrayOfPoints)& thePoints)
71 {
72   Bnd_Box aBndBox;
73   if (thePoints.IsNull())
74   {
75     return aBndBox;
76   }
77
78   const Standard_Integer aNbVertices = thePoints->VertexNumber();
79   for (Standard_Integer aVertIter = 1; aVertIter <= aNbVertices; ++aVertIter)
80   {
81     aBndBox.Add (thePoints->Vertice (aVertIter));
82   }
83   return aBndBox;
84 }
85
86 //=======================================================================
87 //function : SetPoints
88 //purpose  :
89 //=======================================================================
90 void AIS_PointCloud::SetPoints (const Handle(Graphic3d_ArrayOfPoints)& thePoints)
91 {
92   myPoints = thePoints;
93   myBndBox = getBoundingBox (thePoints);
94 }
95
96 //=======================================================================
97 //function : SetPoints
98 //purpose  :
99 //=======================================================================
100 void AIS_PointCloud::SetPoints (const Handle(TColgp_HArray1OfPnt)&     theCoords,
101                                 const Handle(Quantity_HArray1OfColor)& theColors,
102                                 const Handle(TColgp_HArray1OfDir)&     theNormals)
103 {
104   myPoints.Nullify();
105   myBndBox.SetVoid();
106   if (theCoords.IsNull())
107   {
108     return;
109   }
110
111   const Standard_Integer aNbPoints = theCoords->Length();
112   if ((!theNormals.IsNull() && theNormals->Length() != aNbPoints)
113    || (!theColors.IsNull()  && theColors->Length()  != aNbPoints))
114   {
115     // invalid input
116     return;
117   }
118
119   const Standard_Boolean hasColors  = !theColors.IsNull()  && theColors->Length()  == aNbPoints;
120   const Standard_Boolean hasNormals = !theNormals.IsNull() && theNormals->Length() == aNbPoints;
121
122   const Standard_Integer aDiffColors  = hasColors  ? (theColors->Lower()  - theCoords->Lower()) : 0;
123   const Standard_Integer aDiffNormals = hasNormals ? (theNormals->Lower() - theCoords->Lower()) : 0;
124
125   myPoints = new Graphic3d_ArrayOfPoints (aNbPoints, hasColors, hasNormals);
126   for (Standard_Integer aPntIter = theCoords->Lower(); aPntIter <= theCoords->Upper(); ++aPntIter)
127   {
128     myPoints->AddVertex (theCoords->Value (aPntIter));
129     if (hasColors)
130     {
131       myPoints->SetVertexColor (myPoints->VertexNumber(),
132                                 theColors->Value (aPntIter + aDiffColors));
133     }
134     if (hasNormals)
135     {
136       myPoints->SetVertexNormal (myPoints->VertexNumber(),
137                                  theNormals->Value (aPntIter + aDiffNormals));
138     }
139   }
140   myBndBox = getBoundingBox (myPoints);
141 }
142
143 //=======================================================================
144 //function : SetColor
145 //purpose  :
146 //=======================================================================
147 void AIS_PointCloud::SetColor (const Quantity_NameOfColor theColor)
148 {
149   SetColor (Quantity_Color (theColor));
150 }
151
152 //=======================================================================
153 //function : SetColor
154 //purpose  :
155 //=======================================================================
156 void AIS_PointCloud::SetColor (const Quantity_Color& theColor)
157 {
158   AIS_InteractiveObject::SetColor(theColor);
159
160   if (!myDrawer->HasOwnPointAspect())
161   {
162     myDrawer->SetPointAspect (new Prs3d_PointAspect (Aspect_TOM_POINT, theColor, 1.0));
163     if (myDrawer->HasLink())
164     {
165       *myDrawer->PointAspect()->Aspect() = *myDrawer->Link()->PointAspect()->Aspect();
166     }
167   }
168   if (!myDrawer->HasOwnShadingAspect())
169   {
170     myDrawer->SetShadingAspect (new Prs3d_ShadingAspect());
171     if (myDrawer->HasLink())
172     {
173       *myDrawer->ShadingAspect()->Aspect() = *myDrawer->Link()->ShadingAspect()->Aspect();
174     }
175   }
176
177   // Override color
178   myDrawer->ShadingAspect()->SetColor (theColor);
179   myDrawer->PointAspect()  ->SetColor (theColor);
180
181   const PrsMgr_Presentations&        aPrsList     = Presentations();
182   Handle(Graphic3d_AspectMarker3d)   aPointAspect = myDrawer->PointAspect()->Aspect();
183   Handle(Graphic3d_AspectFillArea3d) anAreaAspect = myDrawer->ShadingAspect()->Aspect();
184   for (Standard_Integer aPrsIt = 1; aPrsIt <= aPrsList.Length(); ++aPrsIt)
185   {
186     const PrsMgr_ModedPresentation& aPrsModed = aPrsList.Value (aPrsIt);
187     if (aPrsModed.Mode() != AIS_PointCloud::DM_Points)
188     {
189       continue;
190     }
191
192     const Handle(Prs3d_Presentation)& aPrs = aPrsModed.Presentation()->Presentation();
193     for (Graphic3d_SequenceOfGroup::Iterator aGroupIt (aPrs->Groups()); aGroupIt.More(); aGroupIt.Next())
194     {
195       const Handle(Graphic3d_Group)& aGroup = aGroupIt.Value();
196       if (aGroup->IsGroupPrimitivesAspectSet (Graphic3d_ASPECT_MARKER))
197       {
198         aGroup->SetGroupPrimitivesAspect (aPointAspect);
199       }
200       if (aGroup->IsGroupPrimitivesAspectSet (Graphic3d_ASPECT_FILL_AREA))
201       {
202         aGroup->SetGroupPrimitivesAspect (anAreaAspect);
203       }
204     }
205   }
206 }
207
208 //=======================================================================
209 //function : UnsetColor
210 //purpose  :
211 //=======================================================================
212 void AIS_PointCloud::UnsetColor()
213 {
214   if (!HasColor())
215   {
216     return;
217   }
218
219   AIS_InteractiveObject::UnsetColor();
220
221   if (!HasWidth())
222   {
223     myDrawer->SetPointAspect (Handle(Prs3d_PointAspect)());
224   }
225   else
226   {
227     myDrawer->PointAspect()->SetColor (myDrawer->HasLink()
228                                      ? myDrawer->Link()->PointAspect()->Aspect()->Color()
229                                      : Quantity_Color (Quantity_NOC_YELLOW));
230   }
231
232   if (HasMaterial()
233    || IsTransparent())
234   {
235     Graphic3d_MaterialAspect aDefaultMat (Graphic3d_NOM_BRASS);
236     Graphic3d_MaterialAspect aMat = aDefaultMat;
237     if (HasMaterial() || myDrawer->HasLink())
238     {
239       aMat = AIS_GraphicTool::GetMaterial (HasMaterial() ? myDrawer : myDrawer->Link());
240     }
241     if (HasMaterial())
242     {
243       Quantity_Color aColor = aDefaultMat.AmbientColor();
244       if (myDrawer->HasLink())
245       {
246         aColor = myDrawer->Link()->ShadingAspect()->Color (myCurrentFacingModel);
247       }
248       aMat.SetColor (aColor);
249     }
250     if (IsTransparent())
251     {
252       Standard_Real aTransp = myDrawer->ShadingAspect()->Transparency (myCurrentFacingModel);
253       aMat.SetTransparency (aTransp);
254     }
255     myDrawer->ShadingAspect()->SetMaterial (aMat, myCurrentFacingModel);
256   }
257   else
258   {
259     myDrawer->SetShadingAspect (Handle(Prs3d_ShadingAspect)());
260   }
261   myDrawer->SetPointAspect (Handle(Prs3d_PointAspect)());
262
263   // modify shading presentation without re-computation
264   const PrsMgr_Presentations&        aPrsList   = Presentations();
265   Handle(Graphic3d_AspectFillArea3d) anAreaAsp  = myDrawer->Link()->ShadingAspect()->Aspect();
266   Handle(Graphic3d_AspectMarker3d)   aMarkerAsp = myDrawer->Link()->PointAspect()->Aspect();
267   for (Standard_Integer aPrsIt = 1; aPrsIt <= aPrsList.Length(); ++aPrsIt)
268   {
269     const PrsMgr_ModedPresentation& aPrsModed = aPrsList.Value (aPrsIt);
270     if (aPrsModed.Mode() != AIS_PointCloud::DM_Points)
271     {
272       continue;
273     }
274
275     const Handle(Prs3d_Presentation)& aPrs = aPrsModed.Presentation()->Presentation();
276     for (Graphic3d_SequenceOfGroup::Iterator aGroupIt (aPrs->Groups()); aGroupIt.More(); aGroupIt.Next())
277     {
278       const Handle(Graphic3d_Group)& aGroup = aGroupIt.Value();
279
280       // Check if aspect of given type is set for the group,
281       // because setting aspect for group with no already set aspect
282       // can lead to loss of presentation data
283       if (aGroup->IsGroupPrimitivesAspectSet (Graphic3d_ASPECT_FILL_AREA))
284       {
285         aGroup->SetGroupPrimitivesAspect (anAreaAsp);
286       }
287       if (aGroup->IsGroupPrimitivesAspectSet (Graphic3d_ASPECT_MARKER))
288       {
289         aGroup->SetGroupPrimitivesAspect (aMarkerAsp);
290       }
291     }
292   }
293 }
294
295 //=======================================================================
296 //function : SetMaterial
297 //purpose  :
298 //=======================================================================
299 void AIS_PointCloud::SetMaterial (const Graphic3d_NameOfMaterial theMatName)
300 {
301   SetMaterial (Graphic3d_MaterialAspect (theMatName));
302 }
303
304 //=======================================================================
305 //function : SetMaterial
306 //purpose  :
307 //=======================================================================
308 void AIS_PointCloud::SetMaterial (const Graphic3d_MaterialAspect& theMat)
309 {
310   if (!myDrawer->HasOwnShadingAspect())
311   {
312     myDrawer->SetShadingAspect (new Prs3d_ShadingAspect());
313     if (myDrawer->HasLink())
314     {
315       *myDrawer->ShadingAspect()->Aspect() = *myDrawer->Link()->ShadingAspect()->Aspect();
316     }
317   }
318   hasOwnMaterial = Standard_True;
319
320   myDrawer->ShadingAspect()->SetMaterial (theMat, myCurrentFacingModel);
321   if (HasColor())
322   {
323     myDrawer->ShadingAspect()->SetColor (myOwnColor, myCurrentFacingModel);
324   }
325   myDrawer->ShadingAspect()->SetTransparency (myTransparency, myCurrentFacingModel);
326
327   // modify shading presentation without re-computation
328   const PrsMgr_Presentations&        aPrsList  = Presentations();
329   Handle(Graphic3d_AspectFillArea3d) anAreaAsp = myDrawer->ShadingAspect()->Aspect();
330   for (Standard_Integer aPrsIt = 1; aPrsIt <= aPrsList.Length(); ++aPrsIt)
331   {
332     const PrsMgr_ModedPresentation& aPrsModed = aPrsList.Value (aPrsIt);
333     if (aPrsModed.Mode() != AIS_PointCloud::DM_Points)
334     {
335       continue;
336     }
337
338     const Handle(Prs3d_Presentation)& aPrs = aPrsModed.Presentation()->Presentation();
339     for (Graphic3d_SequenceOfGroup::Iterator aGroupIt (aPrs->Groups()); aGroupIt.More(); aGroupIt.Next())
340     {
341       const Handle(Graphic3d_Group)& aGroup = aGroupIt.Value();
342       if (aGroup->IsGroupPrimitivesAspectSet (Graphic3d_ASPECT_FILL_AREA))
343       {
344         aGroup->SetGroupPrimitivesAspect (anAreaAsp);
345       }
346     }
347   }
348 }
349
350 //=======================================================================
351 //function : UnsetMaterial
352 //purpose  :
353 //=======================================================================
354 void AIS_PointCloud::UnsetMaterial()
355 {
356   if (!HasMaterial())
357   {
358     return;
359   }
360
361   if (HasColor()
362    || IsTransparent())
363   {
364     Graphic3d_MaterialAspect aDefaultMat (Graphic3d_NOM_BRASS);
365     myDrawer->ShadingAspect()->SetMaterial (myDrawer->HasLink() ?
366                                             myDrawer->Link()->ShadingAspect()->Material (myCurrentFacingModel) :
367                                             aDefaultMat,
368                                             myCurrentFacingModel);
369     if (HasColor())
370     {
371       myDrawer->ShadingAspect()->SetColor        (myOwnColor,     myCurrentFacingModel);
372       myDrawer->ShadingAspect()->SetTransparency (myTransparency, myCurrentFacingModel);
373     }
374   }
375   else
376   {
377     myDrawer->SetShadingAspect (Handle(Prs3d_ShadingAspect)());
378   }
379   hasOwnMaterial = Standard_False;
380
381   // modify shading presentation without re-computation
382   const PrsMgr_Presentations&        aPrsList  = Presentations();
383   Handle(Graphic3d_AspectFillArea3d) anAreaAsp = myDrawer->ShadingAspect()->Aspect();
384   for (Standard_Integer aPrsIt = 1; aPrsIt <= aPrsList.Length(); ++aPrsIt)
385   {
386     const PrsMgr_ModedPresentation& aPrsModed = aPrsList.Value (aPrsIt);
387     if (aPrsModed.Mode() != AIS_PointCloud::DM_Points)
388     {
389       continue;
390     }
391
392     const Handle(Prs3d_Presentation)& aPrs = aPrsModed.Presentation()->Presentation();
393     for (Graphic3d_SequenceOfGroup::Iterator aGroupIt (aPrs->Groups()); aGroupIt.More(); aGroupIt.Next())
394     {
395       const Handle(Graphic3d_Group)& aGroup = aGroupIt.Value();
396       if (aGroup->IsGroupPrimitivesAspectSet (Graphic3d_ASPECT_FILL_AREA))
397       {
398         aGroup->SetGroupPrimitivesAspect (anAreaAsp);
399       }
400     }
401   }
402 }
403
404 //=======================================================================
405 //function : Compute
406 //purpose  :
407 //=======================================================================
408 void AIS_PointCloud::Compute (const Handle(PrsMgr_PresentationManager3d)& /*thePrsMgr*/,
409                               const Handle(Prs3d_Presentation)&           thePrs,
410                               const Standard_Integer                      theMode)
411 {
412   switch (theMode)
413   {
414     case AIS_PointCloud::DM_Points:
415     {
416       const Handle(Graphic3d_ArrayOfPoints) aPoints = GetPoints();
417       if (aPoints.IsNull())
418       {
419         return;
420       }
421
422       Handle(Graphic3d_Group) aGroup = Prs3d_Root::CurrentGroup (thePrs);
423       aGroup->SetGroupPrimitivesAspect (myDrawer->PointAspect()->Aspect());
424       aGroup->SetGroupPrimitivesAspect (myDrawer->ShadingAspect()->Aspect());
425       aGroup->AddPrimitiveArray (aPoints);
426       break;
427     }
428     case AIS_PointCloud::DM_BndBox:
429     {
430       Bnd_Box aBndBox = GetBoundingBox();
431       if (aBndBox.IsVoid())
432       {
433         return;
434       }
435
436       StdPrs_BndBox::Add (thePrs, aBndBox, myDrawer);
437       break;
438     }
439   }
440 }
441
442 //=======================================================================
443 //function : ComputeSelection
444 //purpose  :
445 //=======================================================================
446 void AIS_PointCloud::ComputeSelection (const Handle(SelectMgr_Selection)& theSelection,
447                                        const Standard_Integer             theMode)
448 {
449   Handle(SelectMgr_EntityOwner) anOwner = new SelectMgr_EntityOwner (this);
450   switch (theMode)
451   {
452     case SM_Points:
453     {
454       const Handle(Graphic3d_ArrayOfPoints) aPoints = GetPoints();
455       if (!aPoints.IsNull()
456        && !aPoints->Attributes().IsNull())
457       {
458         Handle(Select3D_SensitivePrimitiveArray) aSensitive = new Select3D_SensitivePrimitiveArray (anOwner);
459         aSensitive->SetSensitivityFactor (8);
460         aSensitive->InitPoints (aPoints->Attributes(), aPoints->Indices(), TopLoc_Location());
461         aSensitive->BVH();
462         theSelection->Add (aSensitive);
463         return;
464       }
465       break;
466     }
467     case SM_BndBox:
468     {
469       break;
470     }
471     default:
472     {
473       return;
474     }
475   }
476
477   Bnd_Box aBndBox = GetBoundingBox();
478   if (aBndBox.IsVoid())
479   {
480     return;
481   }
482   Handle(Select3D_SensitiveBox) aSensBox = new Select3D_SensitiveBox (anOwner, aBndBox);
483   theSelection->Add (aSensBox);
484 }