0032121: Draw Harness, ViewerTest - implement -reset option for vlight command
[occt.git] / src / AIS / AIS_LightSource.cxx
1 // Created on: 2020-09-07
2 // Created by: Maria KRYLOVA
3 // Copyright (c) 2020 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_LightSource.hxx>
17
18 #include <AIS_InteractiveContext.hxx>
19 #include <Graphic3d_ArrayOfPoints.hxx>
20 #include <Graphic3d_ArrayOfSegments.hxx>
21 #include <Graphic3d_ArrayOfTriangles.hxx>
22 #include <Graphic3d_CView.hxx>
23 #include <Graphic3d_Group.hxx>
24 #include <Prs3d_ArrowAspect.hxx>
25 #include <Prs3d_PointAspect.hxx>
26 #include <Prs3d_Text.hxx>
27 #include <Prs3d_ToolCylinder.hxx>
28 #include <Prs3d_ToolSphere.hxx>
29 #include <Select3D_SensitivePoint.hxx>
30 #include <V3d_View.hxx>
31
32 IMPLEMENT_STANDARD_RTTIEXT(AIS_LightSource, AIS_InteractiveObject)
33 IMPLEMENT_STANDARD_RTTIEXT(AIS_LightSourceOwner, SelectMgr_EntityOwner)
34
35 // =======================================================================
36 // function : AIS_LightSourceOwner
37 // purpose  :
38 // =======================================================================
39 AIS_LightSourceOwner::AIS_LightSourceOwner (const Handle(AIS_LightSource)& theObject,
40                                             Standard_Integer thePriority)
41 : SelectMgr_EntityOwner ((const Handle(SelectMgr_SelectableObject)&)theObject, thePriority)
42 {
43   //
44 }
45
46 // =======================================================================
47 // function : HandleMouseClick
48 // purpose  :
49 // =======================================================================
50 Standard_Boolean AIS_LightSourceOwner::HandleMouseClick (const Graphic3d_Vec2i& ,
51                                                          Aspect_VKeyMouse theKey,
52                                                          Aspect_VKeyFlags theFlags,
53                                                          bool )
54 {
55   AIS_LightSource* aLightSource = dynamic_cast<AIS_LightSource*>(mySelectable);
56   if (aLightSource != NULL
57    && aLightSource->ToSwitchOnClick()
58    && theKey == Aspect_VKeyMouse_LeftButton
59    && theFlags == Aspect_VKeyFlags_NONE)
60   {
61     aLightSource->Light()->SetEnabled (!aLightSource->Light()->IsEnabled());
62     aLightSource->updateLightAspects();
63     return true;
64   }
65   return false;
66 }
67
68 // =======================================================================
69 // function : Constructor
70 // purpose  :
71 // =======================================================================
72 AIS_LightSource::AIS_LightSource (const Handle(Graphic3d_CLight)& theLight)
73 : myLightSource (theLight),
74   myCodirMarkerType (Aspect_TOM_X),
75   myOpposMarkerType (Aspect_TOM_O_POINT),
76   mySize (50),
77   myNbArrows (5),
78   myNbSplitsQuadric (theLight->Type() == Graphic3d_TypeOfLightSource_Ambient ? 10 : 30),
79   myNbSplitsArrow (20),
80   myIsZoomable (theLight->Type() == Graphic3d_TypeOfLightSource_Positional
81              || theLight->Type() == Graphic3d_TypeOfLightSource_Spot),
82   myToDisplayName (true),
83   myToDisplayRange (true),
84   myToSwitchOnClick (true)
85 {
86   myMarkerTypes[0] = Aspect_TOM_O_X;
87   myMarkerTypes[1] = Aspect_TOM_O_POINT;
88
89   myInfiniteState = true;
90
91   const Quantity_Color aColor = theLight->Color();
92   myDrawer->SetPointAspect (new Prs3d_PointAspect (myMarkerTypes[1], aColor, 3.0f));
93   myDisabledMarkerAspect = new Graphic3d_AspectMarker3d (Aspect_TOM_EMPTY, aColor, 3.0f);
94
95   Graphic3d_MaterialAspect aMat (Graphic3d_NameOfMaterial_UserDefined);
96   aMat.SetColor (aColor);
97   myDrawer->SetArrowAspect (new Prs3d_ArrowAspect());
98   myDrawer->ArrowAspect()->SetColor (aColor);
99   myDrawer->ArrowAspect()->Aspect()->SetShadingModel (Graphic3d_TOSM_UNLIT);
100   myDrawer->ArrowAspect()->Aspect()->ChangeFrontMaterial() = aMat;
101   myDrawer->ArrowAspect()->Aspect()->SetMarkerType (Aspect_TOM_EMPTY);
102   myDrawer->ArrowAspect()->Aspect()->SetMarkerScale (2.0f);
103   myArrowLineAspectShadow = new Graphic3d_AspectLine3d (Quantity_NOC_BLACK, Aspect_TOL_SOLID,
104                                                         theLight->Type() != Graphic3d_TypeOfLightSource_Ambient ? 3.0f : 1.0f);
105
106   myDrawer->SetupOwnShadingAspect();
107   myDrawer->ShadingAspect()->SetColor (aColor);
108   myDrawer->ShadingAspect()->SetMaterial (aMat);
109   myDrawer->ShadingAspect()->SetTransparency (0.5f);
110   myDrawer->ShadingAspect()->Aspect()->SetShadingModel (Graphic3d_TOSM_UNLIT);
111
112   myDrawer->SetTextAspect (new Prs3d_TextAspect());
113   myDrawer->TextAspect()->Aspect()->SetDisplayType (Aspect_TODT_SHADOW);
114   myDrawer->TextAspect()->Aspect()->SetColorSubTitle (Quantity_NOC_BLACK);
115   myDrawer->TextAspect()->SetHorizontalJustification (Graphic3d_HTA_LEFT);
116   myDrawer->TextAspect()->SetVerticalJustification (Graphic3d_VTA_TOPFIRSTLINE);
117
118   updateLightTransformPersistence();
119
120   myDrawer->SetDisplayMode (0);
121   myDynHilightDrawer = new Prs3d_Drawer();
122   myDynHilightDrawer->Link (myDrawer);
123   myDynHilightDrawer->SetDisplayMode (1);
124   myDynHilightDrawer->SetColor (Quantity_NOC_CYAN1);
125
126   if (!myTransformPersistence.IsNull()
127     && myTransformPersistence->IsTrihedronOr2d())
128   {
129     myDrawer->SetZLayer (Graphic3d_ZLayerId_Topmost);
130     myDynHilightDrawer->SetZLayer (Graphic3d_ZLayerId_Topmost);
131     myDrawer->TextAspect()->SetHorizontalJustification (Graphic3d_HTA_CENTER);
132     myDrawer->TextAspect()->SetVerticalJustification (Graphic3d_VTA_TOP);
133   }
134 }
135
136 // =======================================================================
137 // function : updateLightAspects
138 // purpose  :
139 // =======================================================================
140 void AIS_LightSource::updateLightAspects()
141 {
142   const Quantity_Color aBaseColor = myLightSource->Color();
143   const Quantity_Color aDimColor (aBaseColor.Rgb() * 0.3f);
144   const Quantity_Color aColor = myLightSource->IsEnabled() ? aBaseColor : aDimColor;
145   myDrawer->PointAspect()->SetColor (aColor);
146   myDrawer->PointAspect()->Aspect()->SetMarkerType (MarkerType (myLightSource->IsEnabled()));
147   myDrawer->PointAspect()->Aspect()->SetMarkerImage(MarkerImage(myLightSource->IsEnabled()));
148
149   myDisabledMarkerAspect->SetColor (aColor);
150   myDisabledMarkerAspect->SetMarkerScale(myDrawer->PointAspect()->Aspect()->MarkerScale());
151   myDisabledMarkerAspect->SetMarkerType (myLightSource->IsEnabled() ? Aspect_TOM_EMPTY : MarkerType (false));
152   myDisabledMarkerAspect->SetMarkerImage(MarkerImage (false));
153
154   myDrawer->ShadingAspect()->SetColor (aColor);
155   myDrawer->ArrowAspect()  ->SetColor (aColor);
156   myDrawer->ArrowAspect()->Aspect()->ChangeFrontMaterial().SetColor (aColor);
157
158   if (myLightSource->Type() == Graphic3d_TypeOfLightSource_Directional)
159   {
160     const Standard_Real anAngleTol = 2.0 * M_PI / 180.0;
161     Aspect_TypeOfMarker aDirMark = Aspect_TOM_EMPTY;
162     if (myLightSource->IsEnabled()
163      && myLightSource->IsHeadlight()
164      && myLightSource->Direction().IsParallel (gp::DZ(), anAngleTol))
165     {
166       aDirMark = myLightSource->Direction().IsOpposite (-gp::DZ(), anAngleTol) ? myOpposMarkerType : myCodirMarkerType;
167     }
168     myDrawer->ArrowAspect()->Aspect()->SetMarkerType (aDirMark);
169   }
170   SynchronizeAspects();
171 }
172
173 // =======================================================================
174 // function : updateLightTransformPersistence
175 // purpose  :
176 // =======================================================================
177 void AIS_LightSource::updateLightTransformPersistence()
178 {
179   Handle(Graphic3d_TransformPers) aTrsfPers = myTransformPersistence;
180   switch (myLightSource->Type())
181   {
182     case Graphic3d_TypeOfLightSource_Ambient:
183     {
184       if (!myIsZoomable)
185       {
186         if (aTrsfPers.IsNull() || !aTrsfPers->IsTrihedronOr2d())
187         {
188           aTrsfPers = new Graphic3d_TransformPers (Graphic3d_TMF_TriedronPers, Aspect_TOTP_LEFT_UPPER, Graphic3d_Vec2i(50));
189         }
190       }
191       else
192       {
193         aTrsfPers.Nullify();
194       }
195       break;
196     }
197     case Graphic3d_TypeOfLightSource_Directional:
198     {
199       Graphic3d_TransModeFlags aMode = myLightSource->IsHeadlight() ? Graphic3d_TMF_2d : Graphic3d_TMF_TriedronPers;
200       if (myIsZoomable)
201       {
202         aMode = myLightSource->IsHeadlight() ? Graphic3d_TMF_CameraPers : Graphic3d_TMF_None;
203       }
204       if (aMode != Graphic3d_TMF_None)
205       {
206         if (aTrsfPers.IsNull() || aTrsfPers->Mode() != aMode)
207         {
208           if (aMode == Graphic3d_TMF_CameraPers)
209           {
210             aTrsfPers = new Graphic3d_TransformPers (Graphic3d_TMF_CameraPers);
211           }
212           else
213           {
214             aTrsfPers = new Graphic3d_TransformPers (aMode, Aspect_TOTP_LEFT_UPPER, Graphic3d_Vec2i(50));
215           }
216         }
217       }
218       else
219       {
220         aTrsfPers.Nullify();
221       }
222       break;
223     }
224     case Graphic3d_TypeOfLightSource_Positional:
225     case Graphic3d_TypeOfLightSource_Spot:
226     {
227       Graphic3d_TransModeFlags aMode = myLightSource->IsHeadlight()
228                                      ? Graphic3d_TMF_CameraPers
229                                      : (!myIsZoomable ? Graphic3d_TMF_ZoomPers : Graphic3d_TMF_None);
230       if (aMode != Graphic3d_TMF_None)
231       {
232         if (aTrsfPers.IsNull() || aTrsfPers->Mode() != aMode)
233         {
234           if (aMode == Graphic3d_TMF_CameraPers)
235           {
236             aTrsfPers = new Graphic3d_TransformPers (Graphic3d_TMF_CameraPers);
237           }
238           else
239           {
240             aTrsfPers = new Graphic3d_TransformPers (aMode, myLightSource->Position());
241           }
242         }
243         if (aMode == Graphic3d_TMF_ZoomPers)
244         {
245           aTrsfPers->SetAnchorPoint (myLightSource->Position());
246         }
247       }
248       else
249       {
250         aTrsfPers.Nullify();
251       }
252       break;
253     }
254   }
255
256   SetTransformPersistence (aTrsfPers);
257 }
258
259 // =======================================================================
260 // function : updateLightLocalTransformation
261 // purpose  :
262 // =======================================================================
263 void AIS_LightSource::updateLightLocalTransformation()
264 {
265   myLocalTransformation.Nullify();
266   switch (myLightSource->Type())
267   {
268     case Graphic3d_TypeOfLightSource_Ambient:
269     {
270       if (myIsZoomable)
271       {
272         gp_Trsf aTrsf;
273         aTrsf.SetTranslation (gp::Origin(), myLightSource->Position());
274         myLocalTransformation = new TopLoc_Datum3D (aTrsf);
275       }
276       break;
277     }
278     case Graphic3d_TypeOfLightSource_Directional:
279     {
280       const gp_Pnt aLightPos = (myIsZoomable && !myLightSource->IsHeadlight())
281                              ? myLightSource->DisplayPosition()
282                              : gp::Origin();
283       gp_Trsf aTrsf;
284       const gp_Ax2 anAx2 (aLightPos, -myLightSource->Direction());
285       aTrsf.SetTransformation (anAx2, gp_Ax3());
286       myLocalTransformation = new TopLoc_Datum3D (aTrsf);
287       break;
288     }
289     case Graphic3d_TypeOfLightSource_Positional:
290     {
291       if (myIsZoomable)
292       {
293         gp_Trsf aTrsf;
294         aTrsf.SetTranslation (gp::Origin(), myLightSource->Position());
295         myLocalTransformation = new TopLoc_Datum3D (aTrsf);
296       }
297       break;
298     }
299     case Graphic3d_TypeOfLightSource_Spot:
300     {
301       gp_Trsf aTrsf;
302       const gp_Ax2 anAx2 (myIsZoomable ? myLightSource->Position() : gp::Origin(), -myLightSource->Direction());
303       aTrsf.SetTransformation (anAx2, gp_Ax3());
304       myLocalTransformation = new TopLoc_Datum3D (aTrsf);
305       break;
306     }
307   }
308   UpdateTransformation();
309 }
310
311 // =======================================================================
312 // function : setLocalTransformation
313 // purpose  :
314 // =======================================================================
315 void AIS_LightSource::setLocalTransformation (const Handle(TopLoc_Datum3D)& theTrsf)
316 {
317   const gp_Trsf aTrsf = theTrsf->Transformation();
318   switch (myLightSource->Type())
319   {
320     case Graphic3d_TypeOfLightSource_Ambient:
321     {
322       break;
323     }
324     case Graphic3d_TypeOfLightSource_Directional:
325     {
326       gp_Dir aNewDir = (-gp::DZ()).Transformed (aTrsf);
327       myLightSource->SetDirection (aNewDir);
328       if (myIsZoomable)
329       {
330         gp_Pnt aNewPos = gp::Origin().Transformed (aTrsf);
331         myLightSource->SetDisplayPosition (aNewPos);
332       }
333       break;
334     }
335     case Graphic3d_TypeOfLightSource_Positional:
336     {
337       gp_Pnt aNewPos = gp::Origin().Transformed (aTrsf);
338       myLightSource->SetPosition (aNewPos);
339       break;
340     }
341     case Graphic3d_TypeOfLightSource_Spot:
342     {
343       gp_Pnt aNewPos = gp::Origin().Transformed (aTrsf);
344       myLightSource->SetPosition (aNewPos);
345
346       gp_Dir aNewDir = (-gp::DZ()).Transformed (aTrsf);
347       myLightSource->SetDirection (aNewDir);
348       break;
349     }
350   }
351
352   base_type::setLocalTransformation (new TopLoc_Datum3D (aTrsf));
353
354   updateLightAspects();
355   updateLightTransformPersistence();
356 }
357
358 // =======================================================================
359 // function : Compute
360 // purpose  :
361 // =======================================================================
362 void AIS_LightSource::Compute (const Handle(PrsMgr_PresentationManager)& ,
363                                const Handle(Prs3d_Presentation)& thePrs,
364                                const Standard_Integer theMode)
365 {
366   thePrs->SetInfiniteState (myInfiniteState);
367   if (theMode != 0
368    && theMode != 1)
369   {
370     return;
371   }
372
373   if (theMode == 0)
374   {
375     updateLightAspects();
376     updateLightTransformPersistence();
377     updateLightLocalTransformation();
378   }
379
380   switch (myLightSource->Type())
381   {
382     case Graphic3d_TypeOfLightSource_Ambient:     computeAmbient    (thePrs, theMode); break;
383     case Graphic3d_TypeOfLightSource_Directional: computeDirectional(thePrs, theMode); break;
384     case Graphic3d_TypeOfLightSource_Positional:  computePositional (thePrs, theMode); break;
385     case Graphic3d_TypeOfLightSource_Spot:        computeSpot       (thePrs, theMode); break;
386   }
387
388   if (myToDisplayName)
389   {
390     TCollection_AsciiString aPrefix = !myTransformPersistence.IsNull()
391                                     && myTransformPersistence->IsTrihedronOr2d()
392                                     ? "\n" : "   ";
393     TCollection_AsciiString aName = aPrefix + myLightSource->Name();
394     Prs3d_Text::Draw (thePrs->NewGroup(), myDrawer->TextAspect(), aName, gp::Origin());
395   }
396 }
397
398 // =======================================================================
399 // function : computeAmbient
400 // purpose  :
401 // =======================================================================
402 void AIS_LightSource::computeAmbient (const Handle(Prs3d_Presentation)& thePrs,
403                                       const Standard_Integer theMode)
404 {
405   const gp_XYZ aLightPos = gp::Origin().XYZ();
406   if (theMode == 0)
407   {
408     Handle(Graphic3d_ArrayOfTriangles) aSphereArray = Prs3d_ToolSphere::Create (mySize * 0.25, myNbSplitsQuadric, myNbSplitsQuadric, gp_Trsf());
409     Handle(Graphic3d_Group) aSphereGroup = thePrs->NewGroup();
410     aSphereGroup->SetClosed (true);
411     aSphereGroup->SetGroupPrimitivesAspect (myDrawer->ShadingAspect()->Aspect());
412     aSphereGroup->AddPrimitiveArray (aSphereArray);
413   }
414   if (theMode == 0
415    || theMode == 1)
416   {
417     const Standard_Real aLen = mySize * 0.25;
418     const Standard_Integer aNbArrows = 6;
419     const gp_Dir aDirList[6] = { -gp::DX(), gp::DX(), -gp::DY(), gp::DY(), -gp::DZ(), gp::DZ() };
420
421     const Prs3d_ToolCylinder aCylTool (mySize * 0.1, 0.0, mySize * 0.2, myNbSplitsArrow, myNbSplitsArrow);
422     Handle(Graphic3d_ArrayOfTriangles) aTrisArray = new Graphic3d_ArrayOfTriangles (aNbArrows * aCylTool.VerticesNb(),
423                                                                                     aNbArrows * aCylTool.TrianglesNb() * 3,
424                                                                                     Graphic3d_ArrayFlags_VertexNormal);
425     Handle(Graphic3d_ArrayOfSegments) aLineArray = new Graphic3d_ArrayOfSegments (aNbArrows * 2);
426     for (Standard_Integer anArrIter = 0; anArrIter < aNbArrows; ++anArrIter)
427     {
428       const gp_Dir& aDir = aDirList[anArrIter];
429       const gp_XYZ  aPnt = aLightPos + aDir.XYZ() * aLen;
430       if (!aLineArray.IsNull())
431       {
432         aLineArray->AddVertex (aPnt + aDir.XYZ() * aLen * 0.5);
433         aLineArray->AddVertex (aPnt + aDir.XYZ() * aLen * 1.5);
434       }
435       if (!aTrisArray.IsNull())
436       {
437         const gp_Ax3 aSystem (aPnt + aDir.XYZ() * aLen, -aDir);
438         gp_Trsf aTrsfCone;
439         aTrsfCone.SetTransformation (aSystem, gp_Ax3());
440         aCylTool.FillArray (aTrisArray, aTrsfCone);
441       }
442     }
443
444     if (!aLineArray.IsNull())
445     {
446       Handle(Graphic3d_Group) aDirGroupShadow = thePrs->NewGroup();
447       aDirGroupShadow->SetGroupPrimitivesAspect (myArrowLineAspectShadow);
448       aDirGroupShadow->AddPrimitiveArray (aLineArray);
449     }
450     if (!aTrisArray.IsNull())
451     {
452       Handle(Graphic3d_Group) anArrowGroup = thePrs->NewGroup();
453       anArrowGroup->SetClosed (true);
454       anArrowGroup->SetGroupPrimitivesAspect (myDrawer->ArrowAspect()->Aspect());
455       anArrowGroup->AddPrimitiveArray (aTrisArray);
456     }
457   }
458
459   {
460     Handle(Graphic3d_ArrayOfPoints) aPoints = new Graphic3d_ArrayOfPoints (1);
461     aPoints->AddVertex (aLightPos);
462     Handle(Graphic3d_Group) aGroup = thePrs->NewGroup();
463     aGroup->SetGroupPrimitivesAspect (theMode == 1 ? myDrawer->PointAspect()->Aspect() : myDisabledMarkerAspect);
464     aGroup->AddPrimitiveArray (aPoints);
465   }
466 }
467
468 // =======================================================================
469 // function : computeDirectional
470 // purpose  :
471 // =======================================================================
472 void AIS_LightSource::computeDirectional (const Handle(Prs3d_Presentation)& thePrs,
473                                           const Standard_Integer theMode)
474 {
475   const Standard_Real aDistance = mySize * 0.5;
476   const Standard_Real aStep = aDistance * 0.5;
477
478   // light source direction is set to local transformation
479   const gp_Dir aLightDir = -gp::DZ();
480   const gp_XYZ aLightPos = -aStep * aLightDir.XYZ();
481
482   Standard_Integer aNbArrows = 1;
483   if      (myNbArrows >= 9) { aNbArrows = 9; }
484   else if (myNbArrows >= 5) { aNbArrows = 5; }
485   else if (myNbArrows >= 3) { aNbArrows = 3; }
486   TColgp_Array1OfPnt aPoints (1, aNbArrows);
487   {
488     const gp_Ax2 anAxes (gp::Origin(), aLightDir);
489     const gp_XYZ aDY = anAxes.YDirection().XYZ() * aStep;
490     const gp_XYZ aDX = anAxes.XDirection().XYZ() * aStep;
491     const gp_XYZ aDXY = aDX + aDY;
492     switch (aNbArrows)
493     {
494       case 9:
495       {
496         aPoints.SetValue (6, aLightPos + aDY);
497         aPoints.SetValue (7, aLightPos + aDX);
498         aPoints.SetValue (8, aLightPos - aDY);
499         aPoints.SetValue (9, aLightPos - aDX);
500       }
501       Standard_FALLTHROUGH
502       case 5:
503       {
504         aPoints.SetValue (4, aLightPos - aDY + aDX);
505         aPoints.SetValue (5, aLightPos + aDY - aDX);
506       }
507       Standard_FALLTHROUGH
508       case 3:
509       {
510         aPoints.SetValue (2, aLightPos + aDXY);
511         aPoints.SetValue (3, aLightPos - aDXY);
512       }
513       Standard_FALLTHROUGH
514       case 1:
515       {
516         aPoints.SetValue (1, aLightPos);
517         break;
518       }
519     }
520   }
521
522   const Prs3d_ToolCylinder aCylTool (aDistance * 0.1, 0.0, aDistance * 0.2, myNbSplitsArrow, myNbSplitsArrow);
523   Handle(Graphic3d_ArrayOfTriangles) aTrisArray;
524   if (theMode == 0)
525   {
526     aTrisArray = new Graphic3d_ArrayOfTriangles (aNbArrows * aCylTool.VerticesNb(),
527                                                  aNbArrows * aCylTool.TrianglesNb() * 3,
528                                                  Graphic3d_ArrayFlags_VertexNormal);
529   }
530   Handle(Graphic3d_ArrayOfPoints) aPntArray = new Graphic3d_ArrayOfPoints (aNbArrows);
531   Handle(Graphic3d_ArrayOfSegments) aLineArray = new Graphic3d_ArrayOfSegments (aNbArrows * 2);
532   for (Standard_Integer aPntIter = aPoints.Lower(); aPntIter <= aPoints.Upper(); ++aPntIter)
533   {
534     const gp_Pnt aPnt = aPoints.Value (aPntIter);
535     if (!aPntArray.IsNull())
536     {
537       aPntArray->AddVertex (aPnt);
538     }
539     if (!aLineArray.IsNull())
540     {
541       aLineArray->AddVertex (aPnt);
542       aLineArray->AddVertex (gp_Pnt (aPnt.XYZ() + aLightDir.XYZ() * aDistance));
543     }
544     if (!aTrisArray.IsNull())
545     {
546       const gp_Ax3 aSystem (aPnt.XYZ() + aLightDir.XYZ() * aDistance, aLightDir);
547       gp_Trsf aTrsfCone;
548       aTrsfCone.SetTransformation (aSystem, gp_Ax3());
549       aCylTool.FillArray (aTrisArray, aTrsfCone);
550     }
551   }
552
553   if (!aLineArray.IsNull() && theMode == 0)
554   {
555     Handle(Graphic3d_Group) aDirGroupShadow = thePrs->NewGroup();
556     aDirGroupShadow->SetGroupPrimitivesAspect (myArrowLineAspectShadow);
557     aDirGroupShadow->AddPrimitiveArray (aLineArray);
558   }
559   if (!aLineArray.IsNull())
560   {
561     Handle(Graphic3d_Group) aDirGroup = thePrs->NewGroup();
562     aDirGroup->SetGroupPrimitivesAspect (myDrawer->ArrowAspect()->Aspect());
563     aDirGroup->AddPrimitiveArray (aLineArray);
564   }
565   if (!aTrisArray.IsNull())
566   {
567     Handle(Graphic3d_Group) anArrowGroup = thePrs->NewGroup();
568     anArrowGroup->SetClosed (true);
569     anArrowGroup->SetGroupPrimitivesAspect (myDrawer->ArrowAspect()->Aspect());
570     anArrowGroup->AddPrimitiveArray (aTrisArray);
571   }
572   if (!aPntArray.IsNull())
573   {
574     Handle(Graphic3d_Group) aGroup = thePrs->NewGroup();
575     aGroup->SetGroupPrimitivesAspect (myDrawer->ArrowAspect()->Aspect());
576     aGroup->AddPrimitiveArray (aPntArray);
577   }
578   {
579     Handle(Graphic3d_ArrayOfPoints) aPntArray2 = new Graphic3d_ArrayOfPoints (1);
580     aPntArray2->AddVertex (aLightPos);
581     Handle(Graphic3d_Group) aGroup = thePrs->NewGroup();
582     aGroup->SetGroupPrimitivesAspect (myDisabledMarkerAspect);
583     aGroup->AddPrimitiveArray (aPntArray2);
584   }
585 }
586
587 // =======================================================================
588 // function : computePositional
589 // purpose  :
590 // =======================================================================
591 void AIS_LightSource::computePositional (const Handle(Prs3d_Presentation)& thePrs,
592                                          const Standard_Integer theMode)
593 {
594   // light source position is set to local transformation
595   const gp_XYZ aLightPos = gp::Origin().XYZ();
596   const Standard_Real aRadius = (myIsZoomable && myLightSource->HasRange()) ? myLightSource->Range() : 0.0;
597   if (theMode == 0
598    && aRadius > 0.0
599    && myToDisplayRange)
600   {
601     Handle(Graphic3d_ArrayOfTriangles) aPosRangeArray = Prs3d_ToolSphere::Create (aRadius, myNbSplitsQuadric, myNbSplitsQuadric, gp_Trsf());
602     Handle(Graphic3d_Group) aRangeGroup = thePrs->NewGroup();
603     aRangeGroup->SetClosed (true);
604     aRangeGroup->SetGroupPrimitivesAspect (myDrawer->ShadingAspect()->Aspect());
605     aRangeGroup->AddPrimitiveArray (aPosRangeArray);
606   }
607   {
608     Handle(Graphic3d_ArrayOfPoints) aPoints = new Graphic3d_ArrayOfPoints (1);
609     aPoints->AddVertex (aLightPos);
610     Handle(Graphic3d_Group) aGroup = thePrs->NewGroup();
611     aGroup->SetGroupPrimitivesAspect (myDrawer->PointAspect()->Aspect());
612     aGroup->AddPrimitiveArray (aPoints);
613   }
614 }
615
616 // =======================================================================
617 // function : computeSpot
618 // purpose  :
619 // =======================================================================
620 void AIS_LightSource::computeSpot (const Handle(Prs3d_Presentation)& thePrs,
621                                    const Standard_Integer theMode)
622 {
623   // light source position and direction are set to local transformation
624   const gp_Dir aLightDir = -gp::DZ();
625   const gp_XYZ aLightPos = gp::Origin().XYZ();
626   const Standard_Real aDistance = (myIsZoomable && myLightSource->HasRange()) ? myLightSource->Range() : mySize;
627   {
628     Handle(Graphic3d_ArrayOfPoints) aPoints = new Graphic3d_ArrayOfPoints (1);
629     aPoints->AddVertex (aLightPos);
630
631     Handle(Graphic3d_Group) aGroup = thePrs->NewGroup();
632     aGroup->SetGroupPrimitivesAspect (myDrawer->PointAspect()->Aspect());
633     aGroup->AddPrimitiveArray (aPoints);
634   }
635
636   {
637     Handle(Graphic3d_ArrayOfSegments) aDirArray = new Graphic3d_ArrayOfSegments (2);
638     aDirArray->AddVertex (aLightPos);
639     aDirArray->AddVertex (gp_Pnt (aLightPos + aLightDir.XYZ() * aDistance));
640
641     Handle(Graphic3d_Group) aDirGroupShadow = thePrs->NewGroup();
642     aDirGroupShadow->SetClosed (true);
643     aDirGroupShadow->SetGroupPrimitivesAspect (myArrowLineAspectShadow);
644     aDirGroupShadow->AddPrimitiveArray (aDirArray);
645
646     Handle(Graphic3d_Group) aDirGroup = thePrs->NewGroup();
647     aDirGroup->SetClosed (true);
648     aDirGroup->SetGroupPrimitivesAspect (myDrawer->ArrowAspect()->Aspect());
649     aDirGroup->AddPrimitiveArray (aDirArray);
650   }
651
652   if (theMode == 0
653    && myToDisplayRange)
654   {
655     const Standard_ShortReal aHalfAngle = myLightSource->Angle() / 2.0f;
656     const Standard_Real aRadius = aDistance * Tan (aHalfAngle);
657     gp_Ax3  aSystem (aLightPos + aLightDir.XYZ() * aDistance, -aLightDir);
658     gp_Trsf aTrsfCone;
659     aTrsfCone.SetTransformation (aSystem, gp_Ax3());
660     Handle(Graphic3d_ArrayOfTriangles) aSpotRangeArray = Prs3d_ToolCylinder::Create (aRadius, 0.0, aDistance,
661                                                                                      myNbSplitsQuadric, myNbSplitsQuadric, aTrsfCone);
662
663     Handle(Graphic3d_Group) aRangeGroup = thePrs->NewGroup();
664     aRangeGroup->SetClosed (true);
665     aRangeGroup->SetGroupPrimitivesAspect (myDrawer->ShadingAspect()->Aspect());
666     aRangeGroup->AddPrimitiveArray (aSpotRangeArray);
667   }
668 }
669
670 // =======================================================================
671 // function : ComputeSelection
672 // purpose  :
673 // =======================================================================
674 void AIS_LightSource::ComputeSelection (const Handle(SelectMgr_Selection)& theSel,
675                                         const Standard_Integer theMode)
676 {
677   if (theMode != 0)
678   {
679     return;
680   }
681
682   Handle(AIS_LightSourceOwner) anEntityOwner = new AIS_LightSourceOwner (this, 15);
683   {
684     Handle(Select3D_SensitivePoint) aSensPosition = new Select3D_SensitivePoint (anEntityOwner, gp::Origin());
685     aSensPosition->SetSensitivityFactor (12);
686     if (!myTransformPersistence.IsNull()
687       && myTransformPersistence->IsTrihedronOr2d())
688     {
689       aSensPosition->SetSensitivityFactor (Max (12, Standard_Integer (mySize * 0.5)));
690     }
691     theSel->Add (aSensPosition);
692   }
693 }