Salome HOME
updated copyright message
[modules/shaper.git] / src / PartSet / PartSet_Tools.cpp
1 // Copyright (C) 2014-2023  CEA, EDF
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 #include <PartSet_Tools.h>
21 #include <PartSet_Module.h>
22 #include <PartSet_SketcherMgr.h>
23
24 #include <ModelAPI_Data.h>
25 #include <ModelAPI_AttributeDouble.h>
26 #include <ModelAPI_AttributeRefList.h>
27 #include <ModelAPI_Document.h>
28 #include <ModelAPI_Session.h>
29 #include <ModelAPI_ResultConstruction.h>
30 #include <ModelAPI_Events.h>
31 #include <ModelAPI_Validator.h>
32 #include <ModelAPI_Tools.h>
33
34 #include <ModuleBase_IViewWindow.h>
35
36 #include <ModelGeomAlgo_Point2D.h>
37
38 #include <Events_Loop.h>
39 #include <Events_InfoMessage.h>
40
41 #include <SketcherPrs_Tools.h>
42
43 #include <XGUI_ModuleConnector.h>
44 #include <XGUI_Displayer.h>
45 #include <XGUI_Workshop.h>
46 #include <XGUI_SelectionMgr.h>
47 #include <XGUI_Selection.h>
48
49 #include <GeomDataAPI_Point.h>
50 #include <GeomDataAPI_Dir.h>
51 #include <GeomDataAPI_Point2D.h>
52 #include <GeomDataAPI_Point2DArray.h>
53 #include <GeomAPI_Pln.h>
54 #include <GeomAPI_Pnt2d.h>
55 #include <GeomAPI_Pnt.h>
56 #include <GeomAPI_Edge.h>
57 #include <GeomAPI_Vertex.h>
58 #include <GeomAPI_ShapeExplorer.h>
59
60 #include <GeomAPI_Dir.h>
61 #include <GeomAPI_XYZ.h>
62
63 #include <SketchPlugin_Feature.h>
64 #include <SketchPlugin_Sketch.h>
65 #include <SketchPlugin_ConstraintCoincidence.h>
66 #include <SketchPlugin_ConstraintDistance.h>
67 #include <SketchPlugin_ConstraintLength.h>
68 #include <SketchPlugin_ConstraintRadius.h>
69 #include <SketchPlugin_ConstraintRigid.h>
70 #include <SketchPlugin_Constraint.h>
71 #include <SketchPlugin_Circle.h>
72 #include <SketchPlugin_Arc.h>
73 #include <SketchPlugin_Line.h>
74 #include <SketchPlugin_Point.h>
75 #include <SketchPlugin_Projection.h>
76 #include <SketchPlugin_IntersectionPoint.h>
77
78 #include <ModuleBase_IWorkshop.h>
79 #include <ModuleBase_ViewerPrs.h>
80 #include <ModuleBase_Tools.h>
81
82 #include <V3d_View.hxx>
83 #include <gp_Pln.hxx>
84 #include <gp_Circ.hxx>
85 #include <ProjLib.hxx>
86 #include <ElSLib.hxx>
87 #include <Geom_Line.hxx>
88 #include <GeomAPI_ProjectPointOnCurve.hxx>
89 #include <BRep_Tool.hxx>
90 #include <TopoDS.hxx>
91 #include <TopoDS_Edge.hxx>
92 #include <TopoDS_Vertex.hxx>
93 #include <AIS_InteractiveObject.hxx>
94 #include <StdSelect_BRepOwner.hxx>
95 #include <SelectMgr_IndexedMapOfOwner.hxx>
96
97 #include <QMouseEvent>
98
99 #ifdef _DEBUG
100 #include <QDebug>
101 #endif
102
103 const double PRECISION_TOLERANCE = 0.000001;
104 const int AIS_DEFAULT_WIDTH = 2;
105 const bool SKETCH_PROJECTION_INCLUDE_INTO_RESULT = false; // by default, it is not presented
106
107 int PartSet_Tools::getAISDefaultWidth()
108 {
109   return AIS_DEFAULT_WIDTH;
110 }
111
112 gp_Pnt PartSet_Tools::convertClickToPoint(QPoint thePoint, Handle(V3d_View) theView)
113 {
114   if (theView.IsNull())
115     return gp_Pnt();
116
117   Standard_Real XAt, YAt, ZAt;
118   theView->At(XAt, YAt, ZAt);
119   gp_Pnt AtPoint(XAt, YAt, ZAt);
120
121   double aX, aY, aZ;
122   theView->Proj(aX, aY, aZ);
123   gp_Dir EyeDir(aX, aY, aZ);
124   EyeDir.Reverse();
125
126   gp_Pln PlaneOfTheView = gp_Pln(AtPoint, EyeDir);
127   Standard_Real X, Y, Z;
128   theView->Convert(thePoint.x(), thePoint.y(), X, Y, Z);
129   gp_Pnt ConvertedPoint(X, Y, Z);
130
131   gp_Pnt2d ConvertedPointOnPlane = ProjLib::Project(PlaneOfTheView, ConvertedPoint);
132   return ElSLib::Value(ConvertedPointOnPlane.X(), ConvertedPointOnPlane.Y(),
133                                      PlaneOfTheView);
134 }
135
136 void PartSet_Tools::convertTo2D(const gp_Pnt& thePoint, FeaturePtr theSketch,
137 Handle(V3d_View) theView,
138                                 double& theX, double& theY)
139 {
140   if (!theSketch)
141     return;
142
143   AttributeDoublePtr anAttr;
144   std::shared_ptr<ModelAPI_Data> aData = theSketch->data();
145
146   std::shared_ptr<GeomDataAPI_Point> anOrigin = std::dynamic_pointer_cast<GeomDataAPI_Point>(
147       aData->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
148
149   std::shared_ptr<GeomDataAPI_Dir> aX = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
150       aData->attribute(SketchPlugin_Sketch::DIRX_ID()));
151   std::shared_ptr<GeomDataAPI_Dir> aNorm = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
152       aData->attribute(SketchPlugin_Sketch::NORM_ID()));
153   std::shared_ptr<GeomAPI_XYZ> anY = aNorm->xyz()->cross(aX->xyz());
154
155   gp_Pnt anOriginPnt(anOrigin->x(), anOrigin->y(), anOrigin->z());
156   gp_Vec aVec(anOriginPnt, thePoint);
157
158   if (!theView.IsNull()) {
159     Standard_Real XEye, YEye, ZEye, XAt, YAt, ZAt;
160     theView->Eye(XEye, YEye, ZEye);
161
162     theView->At(XAt, YAt, ZAt);
163     gp_Pnt EyePoint(XEye, YEye, ZEye);
164     gp_Pnt AtPoint(XAt, YAt, ZAt);
165
166     if (EyePoint.Distance(AtPoint) > gp::Resolution()) {
167       gp_Vec anEyeVec(EyePoint, AtPoint);
168       anEyeVec.Normalize();
169
170       std::shared_ptr<GeomDataAPI_Dir> aNormal = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
171          aData->attribute(SketchPlugin_Sketch::NORM_ID()));
172       gp_Vec aNormalVec(aNormal->x(), aNormal->y(), aNormal->z());
173
174       double aDen = anEyeVec * aNormalVec;
175       double aLVec = aDen != 0 ? aVec * aNormalVec / aDen : DBL_MAX;
176
177       gp_Vec aDeltaVec = anEyeVec * aLVec;
178       aVec = aVec - aDeltaVec;
179     }
180   }
181   theX = aVec.X() * aX->x() + aVec.Y() * aX->y() + aVec.Z() * aX->z();
182   theY = aVec.X() * anY->x() + aVec.Y() * anY->y() + aVec.Z() * anY->z();
183 }
184
185 std::shared_ptr<GeomAPI_Pnt2d> PartSet_Tools::convertTo2D(FeaturePtr theSketch,
186                                                     const std::shared_ptr<GeomAPI_Pnt>& thePnt)
187 {
188   std::shared_ptr<GeomAPI_Pnt2d> aRes;
189   if (theSketch->getKind() != SketchPlugin_Sketch::ID())
190     return aRes;
191   std::shared_ptr<GeomDataAPI_Point> aC = std::dynamic_pointer_cast<GeomDataAPI_Point>(
192       theSketch->data()->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
193   std::shared_ptr<GeomDataAPI_Dir> aNorm = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
194       theSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
195   std::shared_ptr<GeomDataAPI_Dir> aX = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
196       theSketch->data()->attribute(SketchPlugin_Sketch::DIRX_ID()));
197   std::shared_ptr<GeomAPI_Dir> aY(new GeomAPI_Dir(aNorm->dir()->cross(aX->dir())));
198   return thePnt->to2D(aC->pnt(), aX->dir(), aY);
199 }
200
201
202 std::shared_ptr<GeomAPI_Pnt> PartSet_Tools::convertTo3D(const double theX, const double theY,
203                                                         FeaturePtr theSketch)
204 {
205   std::shared_ptr<ModelAPI_Data> aData = theSketch->data();
206
207   std::shared_ptr<GeomDataAPI_Point> aC = std::dynamic_pointer_cast<GeomDataAPI_Point>(
208       aData->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
209   std::shared_ptr<GeomDataAPI_Dir> aX = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
210       aData->attribute(SketchPlugin_Sketch::DIRX_ID()));
211   std::shared_ptr<GeomDataAPI_Dir> aNorm = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
212       aData->attribute(SketchPlugin_Sketch::NORM_ID()));
213   std::shared_ptr<GeomAPI_Dir> aY(new GeomAPI_Dir(aNorm->dir()->cross(aX->dir())));
214
215   std::shared_ptr<GeomAPI_Pnt2d> aPnt2d =
216     std::shared_ptr<GeomAPI_Pnt2d>(new GeomAPI_Pnt2d(theX, theY));
217
218   return aPnt2d->to3D(aC->pnt(), aX->dir(), aY);
219 }
220
221 std::shared_ptr<ModelAPI_Document> PartSet_Tools::document()
222 {
223   return ModelAPI_Session::get()->moduleDocument();
224 }
225
226 void PartSet_Tools::setFeatureValue(FeaturePtr theFeature, double theValue,
227                                     const std::string& theAttribute)
228 {
229   if (!theFeature)
230     return;
231   std::shared_ptr<ModelAPI_Data> aData = theFeature->data();
232   AttributeDoublePtr anAttribute = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
233       aData->attribute(theAttribute));
234   if (anAttribute)
235     anAttribute->setValue(theValue);
236 }
237
238 double PartSet_Tools::featureValue(FeaturePtr theFeature, const std::string& theAttribute,
239                                    bool& isValid)
240 {
241   isValid = false;
242   double aValue = 0;
243   if (theFeature) {
244     std::shared_ptr<ModelAPI_Data> aData = theFeature->data();
245     AttributeDoublePtr anAttribute = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
246         aData->attribute(theAttribute));
247     if (anAttribute) {
248       aValue = anAttribute->value();
249       isValid = true;
250     }
251   }
252   return aValue;
253 }
254
255 FeaturePtr PartSet_Tools::feature(FeaturePtr theFeature, const std::string& theAttribute,
256                                   const std::string& theKind)
257 {
258   FeaturePtr aFeature;
259   if (!theFeature)
260     return aFeature;
261
262   std::shared_ptr<ModelAPI_Data> aData = theFeature->data();
263   std::shared_ptr<ModelAPI_AttributeRefAttr> anAttr = std::dynamic_pointer_cast<
264       ModelAPI_AttributeRefAttr>(aData->attribute(theAttribute));
265   if (anAttr) {
266     aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(anAttr->object());
267     if (!theKind.empty() && aFeature && aFeature->getKind() != theKind) {
268       aFeature = FeaturePtr();
269     }
270   }
271   return aFeature;
272 }
273
274 void PartSet_Tools::createConstraint(CompositeFeaturePtr theSketch,
275                                      std::shared_ptr<GeomDataAPI_Point2D> thePoint1,
276                                      std::shared_ptr<GeomDataAPI_Point2D> thePoint2)
277 {
278   FeaturePtr aFeature;
279   if (theSketch) {
280     aFeature = theSketch->addFeature(SketchPlugin_ConstraintCoincidence::ID());
281   } else {
282     std::shared_ptr<ModelAPI_Document> aDoc = document();
283     aFeature = aDoc->addFeature(SketchPlugin_ConstraintCoincidence::ID());
284   }
285
286   std::shared_ptr<ModelAPI_Data> aData = aFeature->data();
287
288   std::shared_ptr<ModelAPI_AttributeRefAttr> aRef1 = std::dynamic_pointer_cast<
289       ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_A()));
290   aRef1->setAttr(thePoint1);
291
292   std::shared_ptr<ModelAPI_AttributeRefAttr> aRef2 = std::dynamic_pointer_cast<
293       ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_B()));
294   aRef2->setAttr(thePoint2);
295
296   // we need to flush created signal in order to coincidence is processed by solver
297   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
298 }
299
300 std::shared_ptr<GeomAPI_Pln> PartSet_Tools::sketchPlane(CompositeFeaturePtr theSketch)
301 {
302   std::shared_ptr<GeomAPI_Pln> aPlane;
303
304   std::shared_ptr<GeomDataAPI_Point> anOrigin = std::dynamic_pointer_cast<GeomDataAPI_Point>(
305       theSketch->data()->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
306   std::shared_ptr<GeomDataAPI_Dir> aNormal = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
307       theSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
308
309   if (aNormal.get() && aNormal->isInitialized() &&
310       anOrigin.get() && anOrigin->isInitialized())
311     aPlane = std::shared_ptr<GeomAPI_Pln>(new GeomAPI_Pln(anOrigin->pnt(), aNormal->dir()));
312
313   return aPlane;
314 }
315
316 void PartSet_Tools::nullifySketchPlane(CompositeFeaturePtr theSketch)
317 {
318   std::shared_ptr<GeomDataAPI_Point> anOrigin = std::dynamic_pointer_cast<GeomDataAPI_Point>(
319     theSketch->data()->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
320   std::shared_ptr<GeomDataAPI_Dir> aNormal = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
321     theSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
322
323   aNormal->reset();
324   anOrigin->reset();
325 }
326
327 std::shared_ptr<GeomAPI_Pnt> PartSet_Tools::point3D(std::shared_ptr<GeomAPI_Pnt2d> thePoint2D,
328                                                       CompositeFeaturePtr theSketch)
329 {
330   std::shared_ptr<GeomAPI_Pnt> aPoint;
331   if (!theSketch || !thePoint2D)
332     return aPoint;
333
334   DataPtr aData = theSketch->data();
335   std::shared_ptr<GeomDataAPI_Point> aC = std::dynamic_pointer_cast<GeomDataAPI_Point>(
336       aData->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
337   std::shared_ptr<GeomDataAPI_Dir> aX = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
338       aData->attribute(SketchPlugin_Sketch::DIRX_ID()));
339   std::shared_ptr<GeomDataAPI_Dir> aNorm = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
340       aData->attribute(SketchPlugin_Sketch::NORM_ID()));
341   std::shared_ptr<GeomAPI_Dir> aY(new GeomAPI_Dir(aNorm->dir()->cross(aX->dir())));
342
343   return thePoint2D->to3D(aC->pnt(), aX->dir(), aY);
344 }
345
346 ResultPtr PartSet_Tools::findFixedObjectByExternal(const TopoDS_Shape& theShape,
347                                                    const ObjectPtr& theObject,
348                                                    CompositeFeaturePtr theSketch)
349 {
350   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
351   if (!aResult.get())
352     return ResultPtr();
353
354   for (int i = 0, aNbSubs = theSketch->numberOfSubs(); i < aNbSubs; i++) {
355     FeaturePtr aFeature = theSketch->subFeature(i);
356     if (aFeature->getKind() != SketchPlugin_Projection::PROJECTED_FEATURE_ID())
357       continue;
358     if (aFeature->lastResult() == aResult)
359       return aResult;
360   }
361   return ResultPtr();
362 }
363
364 ResultPtr PartSet_Tools::createFixedObjectByExternal(
365                                    const std::shared_ptr<GeomAPI_Shape>& theShape,
366                                    const ObjectPtr& theObject,
367                                    CompositeFeaturePtr theSketch,
368                                    const bool theTemporary,
369                                    FeaturePtr& theCreatedFeature)
370 {
371   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
372   if (!aResult.get())
373     return ResultPtr();
374
375   FeaturePtr aProjectionFeature = theSketch->addFeature(SketchPlugin_Projection::ID());
376   theCreatedFeature = aProjectionFeature;
377   AttributeSelectionPtr anExternalAttr = std::dynamic_pointer_cast<ModelAPI_AttributeSelection>(
378                  aProjectionFeature->attribute(SketchPlugin_Projection::EXTERNAL_FEATURE_ID()));
379   anExternalAttr->setValue(aResult, theShape);
380
381   AttributeBooleanPtr anIntoResult = std::dynamic_pointer_cast<ModelAPI_AttributeBoolean>
382     (aProjectionFeature->data()->attribute(SketchPlugin_Projection::INCLUDE_INTO_RESULT()));
383   anIntoResult->setValue(SKETCH_PROJECTION_INCLUDE_INTO_RESULT);
384   aProjectionFeature->execute();
385
386   ModelAPI_ValidatorsFactory* aValidators = ModelAPI_Session::get()->validators();
387   bool isValid = aValidators->validate(aProjectionFeature);
388
389   // if projection feature has not been created, exit
390   AttributeRefAttrPtr aRefAttr = aProjectionFeature->data()->refattr(
391     SketchPlugin_Projection::PROJECTED_FEATURE_ID());
392   if (!isValid || !aRefAttr || !aRefAttr->isInitialized())
393   {
394     // remove external feature if the attribute is not filled
395     std::set<FeaturePtr> aFeatures;
396     aFeatures.insert(aProjectionFeature);
397     ModelAPI_Tools::removeFeaturesAndReferences(aFeatures);
398     return ResultPtr();
399   }
400
401   FeaturePtr aProjection = ModelAPI_Feature::feature(aRefAttr->object());
402   if (aProjection.get() && aProjection->lastResult().get())
403     return aProjection->lastResult();
404
405   return ResultPtr();
406 }
407
408 bool PartSet_Tools::isContainPresentation(const QList<ModuleBase_ViewerPrsPtr>& theSelected,
409                                           const ModuleBase_ViewerPrsPtr& thePrs)
410 {
411   foreach (ModuleBase_ViewerPrsPtr aPrs, theSelected) {
412     if (aPrs->object() == thePrs->object())
413       return true;
414   }
415   return false;
416 }
417
418 GeomShapePtr PartSet_Tools::findShapeBy2DPoint(const AttributePtr& theAttribute,
419                                                        ModuleBase_IWorkshop* theWorkshop)
420 {
421   GeomShapePtr aShape;
422   XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(theWorkshop);
423   XGUI_Displayer* aDisplayer = aConnector->workshop()->displayer();
424
425   // 2. find visualized vertices of the attribute and if the attribute of the vertex is
426   // the same, return it
427   FeaturePtr anAttributeFeature = ModelAPI_Feature::feature(theAttribute->owner());
428   // 2.1 get visualized results of the feature
429   const std::list<ResultPtr>& aResList = anAttributeFeature->results();
430   std::list<ResultPtr>::const_iterator anIt = aResList.begin(), aLast = aResList.end();
431   for (; anIt != aLast; anIt++) {
432     AISObjectPtr aAISObj = aDisplayer->getAISObject(*anIt);
433     if (aAISObj.get() != NULL) {
434       Handle(AIS_InteractiveObject) anAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
435       // 2.2 find selected owners of a visualizedd object
436       SelectMgr_IndexedMapOfOwner aSelectedOwners;
437       aConnector->workshop()->selector()->selection()->entityOwners(anAISIO, aSelectedOwners);
438       for (Standard_Integer i = 1, n = aSelectedOwners.Extent(); i <= n; i++) {
439         Handle(SelectMgr_EntityOwner) anOwner = aSelectedOwners(i);
440         if (!anOwner.IsNull()) {
441           Handle(StdSelect_BRepOwner) aBRepOwner = Handle(StdSelect_BRepOwner)::DownCast(anOwner);
442           if (!aBRepOwner.IsNull() && aBRepOwner->HasShape()) {
443             const TopoDS_Shape& aBRepShape = aBRepOwner->Shape();
444             if (aBRepShape.ShapeType() == TopAbs_VERTEX) {
445               // 2.3 if the owner is vertex and an attribute of the vertex is equal to the initial
446               // attribute, returns the shape
447               PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(theWorkshop->module());
448               PartSet_SketcherMgr* aSketchMgr = aModule->sketchMgr();
449               std::pair<AttributePtr, int> aPntAttrIndex = PartSet_Tools::findAttributeBy2dPoint(
450                   anAttributeFeature, aBRepShape, aSketchMgr->activeSketch());
451               if (aPntAttrIndex.first.get() != NULL && aPntAttrIndex.first == theAttribute) {
452                 aShape = std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape);
453                 aShape->setImpl(new TopoDS_Shape(aBRepShape));
454                 break;
455               }
456             }
457           }
458         }
459       }
460     }
461   }
462   return aShape;
463 }
464
465 std::shared_ptr<GeomAPI_Pnt2d> PartSet_Tools::getPoint(
466                                                   std::shared_ptr<ModelAPI_Feature>& theFeature,
467                                                   const std::string& theAttribute)
468 {
469   std::shared_ptr<GeomDataAPI_Point2D> aPointAttr = ModelGeomAlgo_Point2D::getPointOfRefAttr(
470                                           theFeature.get(), theAttribute, SketchPlugin_Point::ID(),
471                                           SketchPlugin_Point::COORD_ID());
472   if (aPointAttr.get() != NULL)
473     return aPointAttr->pnt();
474   return std::shared_ptr<GeomAPI_Pnt2d>();
475 }
476
477 std::shared_ptr<GeomAPI_Pnt2d> PartSet_Tools::getPnt2d(QMouseEvent* theEvent,
478                                                 ModuleBase_IViewWindow* theWindow,
479                                                 const FeaturePtr& theSketch)
480 {
481   gp_Pnt aPnt = PartSet_Tools::convertClickToPoint(theEvent->pos(), theWindow->v3dView());
482   double aX, anY;
483   Handle(V3d_View) aView = theWindow->v3dView();
484   PartSet_Tools::convertTo2D(aPnt, theSketch, aView, aX, anY);
485
486   return std::shared_ptr<GeomAPI_Pnt2d>(new GeomAPI_Pnt2d(aX, anY));
487 }
488
489 std::shared_ptr<GeomAPI_Pnt2d> PartSet_Tools::getPnt2d(const Handle(V3d_View)& theView,
490                                                        const TopoDS_Shape& theShape,
491                                                        const FeaturePtr& theSketch)
492 {
493   GeomPnt2dPtr aPoint2D;
494   if (!theShape.IsNull() && theShape.ShapeType() == TopAbs_VERTEX) {
495     const TopoDS_Vertex& aVertex = TopoDS::Vertex(theShape);
496     if (!aVertex.IsNull()) {
497       // the case when the point is taken from the existing vertex
498       gp_Pnt aPoint = BRep_Tool::Pnt(aVertex);
499       double aX, aY;
500       PartSet_Tools::convertTo2D(aPoint, theSketch, theView, aX, aY);
501       aPoint2D.reset(new GeomAPI_Pnt2d(aX, aY));
502     }
503   }
504   return aPoint2D;
505 }
506
507 FeaturePtr findFirstCoincidenceByData(const DataPtr& theData,
508                                       std::shared_ptr<GeomAPI_Pnt2d> thePoint)
509 {
510   FeaturePtr aCoincident;
511
512   const std::set<AttributePtr>& aRefsList = theData->refsToMe();
513   std::set<AttributePtr>::const_iterator aIt;
514   for (aIt = aRefsList.cbegin(); aIt != aRefsList.cend(); ++aIt) {
515     std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
516     FeaturePtr aConstrFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
517     if (aConstrFeature->getKind() == SketchPlugin_ConstraintCoincidence::ID()) {
518       std::shared_ptr<GeomAPI_Pnt2d> a2dPnt =
519         PartSet_Tools::getPoint(aConstrFeature, SketchPlugin_ConstraintCoincidence::ENTITY_A());
520       if (a2dPnt.get() && thePoint->isEqual(a2dPnt)) {
521         aCoincident = aConstrFeature;
522         break;
523       } else {
524         a2dPnt = PartSet_Tools::getPoint(aConstrFeature,
525                                           SketchPlugin_ConstraintCoincidence::ENTITY_B());
526         if (a2dPnt.get() && thePoint->isEqual(a2dPnt)) {
527           aCoincident = aConstrFeature;
528           break;
529         }
530       }
531     }
532   }
533   return aCoincident;
534 }
535
536 FeaturePtr PartSet_Tools::findFirstCoincidence(const FeaturePtr& theFeature,
537                                                std::shared_ptr<GeomAPI_Pnt2d> thePoint)
538 {
539   FeaturePtr aCoincident;
540   if (theFeature.get() == NULL)
541     return aCoincident;
542
543   const std::set<AttributePtr>& aRefsList = theFeature->data()->refsToMe();
544   std::set<AttributePtr>::const_iterator aIt;
545   for (aIt = aRefsList.cbegin(); aIt != aRefsList.cend(); ++aIt) {
546     std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
547     FeaturePtr aConstrFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
548     if (aConstrFeature && aConstrFeature->getKind() == SketchPlugin_ConstraintCoincidence::ID()) {
549       std::shared_ptr<GeomAPI_Pnt2d> a2dPnt =
550         PartSet_Tools::getPoint(aConstrFeature, SketchPlugin_ConstraintCoincidence::ENTITY_A());
551       if (a2dPnt.get() && thePoint->isEqual(a2dPnt)) {
552         aCoincident = aConstrFeature;
553         break;
554       } else {
555         a2dPnt = PartSet_Tools::getPoint(aConstrFeature,
556                                           SketchPlugin_ConstraintCoincidence::ENTITY_B());
557         if (a2dPnt.get() && thePoint->isEqual(a2dPnt)) {
558           aCoincident = aConstrFeature;
559           break;
560         }
561       }
562     }
563   }
564   /// Find by result
565   if (!aCoincident.get()) {
566     std::list<ResultPtr> aResults = theFeature->results();
567     std::list<ResultPtr>::const_iterator aResIt;
568     for (aResIt = aResults.cbegin(); aResIt != aResults.cend(); ++aResIt) {
569       ResultPtr aResult = *aResIt;
570       aCoincident = findFirstCoincidenceByData(aResult->data(), thePoint);
571       if (aCoincident.get())
572         break;
573     }
574   }
575   return aCoincident;
576 }
577
578 void PartSet_Tools::findCoincidences(FeaturePtr theStartCoin, QList<FeaturePtr>& theList,
579                                      QList<FeaturePtr>& theCoincidencies,
580                                      std::string theAttr, QList<bool>& theIsAttributes)
581 {
582   std::shared_ptr<GeomAPI_Pnt2d> aOrig = getCoincedencePoint(theStartCoin);
583   if (aOrig.get() == NULL)
584     return;
585
586   AttributeRefAttrPtr aPnt = theStartCoin->refattr(theAttr);
587   if (!aPnt)
588     return;
589   ObjectPtr aObj = aPnt->object();
590   FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
591   if (aFeature.get()) {
592     if (!theList.contains(aFeature)) {
593       theList.append(aFeature);
594       theCoincidencies.append(theStartCoin);
595       theIsAttributes.append(true); // point attribute on a feature
596       const std::set<AttributePtr>& aRefsList = aFeature->data()->refsToMe();
597       std::set<AttributePtr>::const_iterator aIt;
598       for (aIt = aRefsList.cbegin(); aIt != aRefsList.cend(); ++aIt) {
599         std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
600         FeaturePtr aConstrFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
601         if (aConstrFeature->getKind() == SketchPlugin_ConstraintCoincidence::ID()) {
602           if (!theCoincidencies.contains(aConstrFeature)) {
603             std::shared_ptr<GeomAPI_Pnt2d> aCoincPnt = getCoincedencePoint(aConstrFeature);
604             if (aCoincPnt.get() && aOrig->isEqual(aCoincPnt)) {
605               findCoincidences(aConstrFeature, theList, theCoincidencies,
606                 SketchPlugin_ConstraintCoincidence::ENTITY_A(), theIsAttributes);
607               findCoincidences(aConstrFeature, theList, theCoincidencies,
608                 SketchPlugin_ConstraintCoincidence::ENTITY_B(), theIsAttributes);
609             }
610           }
611         }
612       }
613     }
614   } else {
615     // Find by Results
616     ResultConstructionPtr aResult = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aObj);
617     if (aResult.get()) {
618       FeaturePtr aFeat = ModelAPI_Feature::feature(aPnt->object());
619       if (!theList.contains(aFeat))
620         theList.append(aFeat);
621       theCoincidencies.append(theStartCoin);
622       theIsAttributes.append(false); // point attribute on a feature
623
624       const std::set<AttributePtr>& aRefsList = aResult->data()->refsToMe();
625       std::set<AttributePtr>::const_iterator aIt;
626       for (aIt = aRefsList.cbegin(); aIt != aRefsList.cend(); ++aIt) {
627         std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
628         FeaturePtr aConstrFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
629         if (aConstrFeature->getKind() == SketchPlugin_ConstraintCoincidence::ID()) {
630           if (!theCoincidencies.contains(aConstrFeature)) {
631             std::shared_ptr<GeomAPI_Pnt2d> aCoincPnt = getCoincedencePoint(aConstrFeature);
632             if (aCoincPnt.get() && aOrig->isEqual(aCoincPnt)) {
633               findCoincidences(aConstrFeature, theList, theCoincidencies,
634                 SketchPlugin_ConstraintCoincidence::ENTITY_A(), theIsAttributes);
635               findCoincidences(aConstrFeature, theList, theCoincidencies,
636                 SketchPlugin_ConstraintCoincidence::ENTITY_B(), theIsAttributes);
637             }
638           }
639         }
640       }
641     }
642   }
643 }
644
645 std::shared_ptr<GeomAPI_Pnt2d> PartSet_Tools::getCoincedencePoint(FeaturePtr theStartCoin)
646 {
647   std::shared_ptr<GeomAPI_Pnt2d> aPnt = SketcherPrs_Tools::getPoint(theStartCoin.get(),
648                                                         SketchPlugin_Constraint::ENTITY_A());
649   if (aPnt.get() == NULL)
650     aPnt = SketcherPrs_Tools::getPoint(theStartCoin.get(), SketchPlugin_Constraint::ENTITY_B());
651   return aPnt;
652 }
653
654 class PointWrapper
655 {
656 public:
657   PointWrapper(AttributePtr theAttribute)
658     : myPoint(std::dynamic_pointer_cast<GeomDataAPI_Point2D>(theAttribute)),
659       myArray(std::dynamic_pointer_cast<GeomDataAPI_Point2DArray>(theAttribute))
660   {}
661
662   int size() const { return myPoint.get() ? 1 : (myArray.get() ? myArray->size() : 0); }
663
664   GeomPointPtr point(int theIndex, FeaturePtr theSketch)
665   {
666     GeomPnt2dPtr aP2d;
667     if (myPoint.get())
668       aP2d = myPoint->pnt();
669     else if (myArray.get())
670       aP2d = myArray->pnt(theIndex);
671
672     GeomPointPtr aP3d;
673     if (aP2d.get())
674       aP3d = PartSet_Tools::convertTo3D(aP2d->x(), aP2d->y(), theSketch);
675     return aP3d;
676   }
677
678   bool isArray() const { return myArray.get(); }
679
680 private:
681   AttributePoint2DPtr myPoint;
682   AttributePoint2DArrayPtr myArray;
683 };
684
685 std::pair<AttributePtr, int> PartSet_Tools::findAttributeBy2dPoint(ObjectPtr theObj,
686                                                                    const TopoDS_Shape theShape,
687                                                                    FeaturePtr theSketch)
688 {
689
690   AttributePtr anAttribute;
691   int aPointIndex = -1;
692   FeaturePtr aFeature = ModelAPI_Feature::feature(theObj);
693   if (aFeature) {
694     if (theShape.ShapeType() == TopAbs_VERTEX) {
695       const TopoDS_Vertex& aVertex = TopoDS::Vertex(theShape);
696       if (!aVertex.IsNull())  {
697         gp_Pnt aPoint = BRep_Tool::Pnt(aVertex);
698         std::shared_ptr<GeomAPI_Pnt> aValue = std::shared_ptr<GeomAPI_Pnt>(
699             new GeomAPI_Pnt(aPoint.X(), aPoint.Y(), aPoint.Z()));
700
701         // find the given point in the feature attributes
702         std::list<AttributePtr> anAttiributes =
703           aFeature->data()->attributes(GeomDataAPI_Point2D::typeId());
704         std::list<AttributePtr> anArrays =
705           aFeature->data()->attributes(GeomDataAPI_Point2DArray::typeId());
706         anAttiributes.insert(anAttiributes.end(), anArrays.begin(), anArrays.end());
707
708         std::list<AttributePtr>::const_iterator anIt = anAttiributes.begin(),
709                                                 aLast = anAttiributes.end();
710         double aMinDistance = 1.e-6; // searching for point with minimal distance and < 1.e-6
711         for (; anIt != aLast && !anAttribute; anIt++) {
712           PointWrapper aWrapper(*anIt);
713           for (int anIndex = 0, aSize = aWrapper.size(); anIndex < aSize; ++anIndex) {
714             std::shared_ptr<GeomAPI_Pnt> aPnt = aWrapper.point(anIndex, theSketch);
715             if (aPnt) {
716               double aDistance = aPnt->distance(aValue);
717               if (aDistance < aMinDistance) {
718                 anAttribute = *anIt;
719                 if (aWrapper.isArray())
720                   aPointIndex = anIndex;
721                 aMinDistance = aPnt->distance(aValue);
722               }
723             }
724           }
725         }
726       }
727     }
728   }
729   return std::pair<AttributePtr, int>(anAttribute, aPointIndex);
730 }
731
732 void PartSet_Tools::sendSubFeaturesEvent(const CompositeFeaturePtr& theComposite,
733                                          const Events_ID theEventId)
734 {
735   if (!theComposite.get())
736     return;
737
738   int aNumberOfSubs = theComposite->numberOfSubs();
739   for (int i = 0; i < aNumberOfSubs; i++) {
740     FeaturePtr aSubFeature = theComposite->subFeature(i);
741     static const ModelAPI_EventCreator* aECreator = ModelAPI_EventCreator::get();
742     aECreator->sendUpdated(aSubFeature, theEventId);
743   }
744   Events_Loop::loop()->flush(theEventId);
745 }
746
747 bool PartSet_Tools::isAuxiliarySketchEntity(const ObjectPtr& theObject)
748 {
749   bool isAuxiliaryFeature = false;
750
751   FeaturePtr anObjectFeature = ModelAPI_Feature::feature(theObject);
752   std::string anAuxiliaryAttribute = SketchPlugin_SketchEntity::AUXILIARY_ID();
753   AttributeBooleanPtr anAuxiliaryAttr = std::dynamic_pointer_cast<ModelAPI_AttributeBoolean>(
754                                     anObjectFeature->data()->attribute(anAuxiliaryAttribute));
755   if (anAuxiliaryAttr.get())
756     isAuxiliaryFeature = anAuxiliaryAttr->value();
757
758
759   return isAuxiliaryFeature;
760 }
761
762 bool PartSet_Tools::isIncludeIntoSketchResult(const ObjectPtr& theObject)
763 {
764   // check the feature is neither Projection nor IntersectionPoint feature
765   FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
766   if (aFeature->getKind() == SketchPlugin_Projection::ID() ||
767       aFeature->getKind() == SketchPlugin_IntersectionPoint::ID())
768     return false;
769
770   // go through the references to the feature to check
771   // if it was created by Projection or Intersection
772   const std::set<AttributePtr>& aRefs = theObject->data()->refsToMe();
773   for (std::set<AttributePtr>::const_iterator aRefIt = aRefs.begin();
774        aRefIt != aRefs.end(); ++aRefIt) {
775     AttributePtr anAttr = *aRefIt;
776     std::string anIncludeToResultAttrName;
777     if (anAttr->id() == SketchPlugin_Projection::PROJECTED_FEATURE_ID())
778       anIncludeToResultAttrName = SketchPlugin_Projection::INCLUDE_INTO_RESULT();
779     else if (anAttr->id() == SketchPlugin_IntersectionPoint::INTERSECTION_POINTS_ID())
780       anIncludeToResultAttrName = SketchPlugin_IntersectionPoint::INCLUDE_INTO_RESULT();
781
782     if (!anIncludeToResultAttrName.empty()) {
783       // check "include into result" flag
784       FeaturePtr aParent = ModelAPI_Feature::feature(anAttr->owner());
785       return aParent->boolean(anIncludeToResultAttrName)->value();
786     }
787   }
788   return true;
789 }
790
791
792 ResultPtr PartSet_Tools::createFixedByExternalCenter(
793     const ObjectPtr& theObject,
794     const std::shared_ptr<GeomAPI_Edge>& theEdge,
795     ModelAPI_AttributeSelection::CenterType theType,
796     const CompositeFeaturePtr& theSketch,
797     bool theTemporary,
798     FeaturePtr& theCreatedFeature)
799 {
800   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
801   if (!aResult.get())
802     return ResultPtr();
803
804   FeaturePtr aProjectionFeature = theSketch->addFeature(SketchPlugin_Projection::ID());
805   theCreatedFeature = aProjectionFeature;
806   AttributeSelectionPtr anExternalAttr = std::dynamic_pointer_cast<ModelAPI_AttributeSelection>(
807                  aProjectionFeature->attribute(SketchPlugin_Projection::EXTERNAL_FEATURE_ID()));
808   anExternalAttr->setValueCenter(aResult, theEdge, theType, theTemporary);
809
810   AttributeBooleanPtr anIntoResult = std::dynamic_pointer_cast<ModelAPI_AttributeBoolean>
811     (aProjectionFeature->data()->attribute(SketchPlugin_Projection::INCLUDE_INTO_RESULT()));
812   anIntoResult->setValue(SKETCH_PROJECTION_INCLUDE_INTO_RESULT);
813   aProjectionFeature->execute();
814
815   // if projection feature has not been created, exit
816   AttributeRefAttrPtr aRefAttr = aProjectionFeature->data()->refattr(
817     SketchPlugin_Projection::PROJECTED_FEATURE_ID());
818   if (!aRefAttr || !aRefAttr->isInitialized())
819     return ResultPtr();
820
821   FeaturePtr aProjection = ModelAPI_Feature::feature(aRefAttr->object());
822   if (aProjection.get() && aProjection->lastResult().get())
823     return aProjection->lastResult();
824
825   return ResultPtr();
826 }
827
828 void PartSet_Tools::getFirstAndLastIndexInFolder(const ObjectPtr& theFolder,
829   int& theFirst, int& theLast)
830 {
831   theFirst = -1;
832   theLast = -1;
833
834   DocumentPtr aDoc = theFolder->document();
835   FolderPtr aFolder = std::dynamic_pointer_cast<ModelAPI_Folder>(theFolder);
836   if (!aFolder.get())
837     return;
838
839   AttributeReferencePtr aFirstFeatAttr =
840     aFolder->data()->reference(ModelAPI_Folder::FIRST_FEATURE_ID());
841   if (!aFirstFeatAttr.get())
842     return;
843   FeaturePtr aFirstFeatureInFolder = ModelAPI_Feature::feature(aFirstFeatAttr->value());
844   if (!aFirstFeatureInFolder.get())
845     return;
846
847   FeaturePtr aLastFeatureInFolder = aFolder->lastVisibleFeature();
848   if (!aLastFeatureInFolder.get())
849     return;
850
851   theFirst = aDoc->index(aFirstFeatureInFolder);
852   theLast = aDoc->index(aLastFeatureInFolder);
853 }
854
855
856 void PartSet_Tools::getDefaultColor(ObjectPtr theObject, const bool isEmptyColorValid,
857   std::vector<int>& theColor)
858 {
859   theColor.clear();
860   // get default color from the preferences manager for the given result
861   std::string aSection, aName, aDefault;
862   theObject->colorConfigInfo(aSection, aName, aDefault);
863   if (!aSection.empty() && !aName.empty()) {
864     theColor = Config_PropManager::color(aSection, aName);
865   }
866   if (!isEmptyColorValid && theColor.empty()) {
867     // all AIS objects, where the color is not set, are in black.
868     // The color should be defined in XML or set in the attribute
869     theColor = Config_PropManager::color("Visualization", "object_default_color");
870     Events_InfoMessage("PartSet_Tools",
871       "A default color is not defined in the preferences for this result type").send();
872   }
873 }
874
875 double PartSet_Tools::getDefaultDeflection(const ObjectPtr& theObject)
876 {
877   double aDeflection = -1;
878   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
879   if (aResult.get()) {
880     bool isConstruction = false;
881
882     std::string aResultGroup = aResult->groupName();
883     if (aResultGroup == ModelAPI_ResultConstruction::group())
884       isConstruction = true;
885     else if (aResultGroup == ModelAPI_ResultBody::group()) {
886       GeomShapePtr aGeomShape = aResult->shape();
887       if (aGeomShape.get()) {
888         // if the shape could not be exploded on faces, it contains only wires, edges, and vertices
889         // correction of deviation for them should not influence to the application performance
890         GeomAPI_ShapeExplorer anExp(aGeomShape, GeomAPI_Shape::FACE);
891         isConstruction = !anExp.more();
892       }
893     }
894     if (isConstruction)
895       aDeflection = Config_PropManager::real("Visualization", "construction_deflection");
896     else
897       aDeflection = Config_PropManager::real("Visualization", "body_deflection");
898   }
899   return aDeflection;
900 }
901
902
903 double PartSet_Tools::getDefaultTransparency()
904 {
905   return Config_PropManager::integer("Visualization", "shaper_default_transparency") / 100.;
906 }
907
908 QCursor PartSet_Tools::getOperationCursor()
909 {
910   int aId = Config_PropManager::integer(SKETCH_TAB_NAME, "operation_cursor");
911   switch (aId) {
912   case 0:
913     return QCursor(Qt::ArrowCursor);
914   case 1:
915     return QCursor(Qt::CrossCursor);
916   case 2:
917     return QCursor(Qt::PointingHandCursor);
918   }
919   return QCursor();
920 }