Salome HOME
Issue #1315 Middle point constraint problem: classes creation
[modules/shaper.git] / src / SketcherPrs / SketcherPrs_Collinear.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
2
3 // File:        SketcherPrs_Collinear.cpp
4 // Created:     29 February 2016
5 // Author:      Natalia ERMOLAEVA
6
7 #include "SketcherPrs_Collinear.h"
8 #include "SketcherPrs_Tools.h"
9 #include "SketcherPrs_PositionMgr.h"
10
11 #include <SketchPlugin_Constraint.h>
12
13 #include "Events_Error.h"
14
15 #include <Graphic3d_AspectLine3d.hxx>
16 #include <Prs3d_Root.hxx>
17
18
19 IMPLEMENT_STANDARD_HANDLE(SketcherPrs_Collinear, SketcherPrs_SymbolPrs);
20 IMPLEMENT_STANDARD_RTTIEXT(SketcherPrs_Collinear, SketcherPrs_SymbolPrs);
21
22 static Handle(Image_AlienPixMap) MyPixMap;
23
24 SketcherPrs_Collinear::SketcherPrs_Collinear(ModelAPI_Feature* theConstraint, 
25                                      const std::shared_ptr<GeomAPI_Ax3>& thePlane) 
26  : SketcherPrs_SymbolPrs(theConstraint, thePlane)
27 {
28   myPntArray = new Graphic3d_ArrayOfPoints(2);
29   myPntArray->AddVertex(0., 0., 0.);
30   myPntArray->AddVertex(0., 0., 0.);
31 }  
32
33 bool SketcherPrs_Collinear::IsReadyToDisplay(ModelAPI_Feature* theConstraint,
34                                          const std::shared_ptr<GeomAPI_Ax3>&/* thePlane*/)
35 {
36   bool aReadyToDisplay = false;
37
38   ObjectPtr aObj1 = SketcherPrs_Tools::getResult(theConstraint, SketchPlugin_Constraint::ENTITY_A());
39   ObjectPtr aObj2 = SketcherPrs_Tools::getResult(theConstraint, SketchPlugin_Constraint::ENTITY_B());
40
41   aReadyToDisplay = SketcherPrs_Tools::getShape(aObj1).get() != NULL &&
42                     SketcherPrs_Tools::getShape(aObj2).get() != NULL;
43
44   return aReadyToDisplay;
45 }
46
47 bool SketcherPrs_Collinear::updatePoints(double theStep) const
48 {
49   if (!IsReadyToDisplay(myConstraint, myPlane))
50     return false;
51
52   ObjectPtr aObj1 = SketcherPrs_Tools::getResult(myConstraint, SketchPlugin_Constraint::ENTITY_A());
53   ObjectPtr aObj2 = SketcherPrs_Tools::getResult(myConstraint, SketchPlugin_Constraint::ENTITY_B());
54
55   // Set points of the presentation
56   SketcherPrs_PositionMgr* aMgr = SketcherPrs_PositionMgr::get();
57   gp_Pnt aP1 = aMgr->getPosition(aObj1, this, theStep);
58   gp_Pnt aP2 = aMgr->getPosition(aObj2, this, theStep);
59   myPntArray->SetVertice(1, aP1);
60   myPntArray->SetVertice(2, aP2);
61   return true;
62 }
63
64 void SketcherPrs_Collinear::drawLines(const Handle(Prs3d_Presentation)& thePrs, Quantity_Color theColor) const
65 {
66   Handle(Graphic3d_Group) aGroup = Prs3d_Root::NewGroup(thePrs);
67
68   Handle(Graphic3d_AspectLine3d) aLineAspect = new Graphic3d_AspectLine3d(theColor, Aspect_TOL_SOLID, 2);
69   aGroup->SetPrimitivesAspect(aLineAspect);
70
71   // Draw first line
72   ObjectPtr aObj = SketcherPrs_Tools::getResult(myConstraint, SketchPlugin_Constraint::ENTITY_A());
73   std::shared_ptr<GeomAPI_Shape> aLine = SketcherPrs_Tools::getShape(aObj);
74   if (aLine.get() == NULL)
75     return;
76   drawShape(aLine, thePrs);
77
78   // Draw second line
79   aObj = SketcherPrs_Tools::getResult(myConstraint, SketchPlugin_Constraint::ENTITY_B());
80   aLine = SketcherPrs_Tools::getShape(aObj);
81   if (aLine.get() == NULL)
82     return;
83   drawShape(aLine, thePrs);
84 }
85