Salome HOME
Issue #2027 Sketcher Trim Feature
[modules/shaper.git] / src / PartSet / PartSet_WidgetFeaturePointSelector.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
2
3 // File:        PartSet_WidgetFeaturePointSelector.cpp
4 // Created:     28 Feb 2017
5 // Author:      Natalia ERMOLAEVA
6
7 #include <Config_WidgetAPI.h>
8
9 #include <Events_Loop.h>
10
11 #include <GeomDataAPI_Point2D.h>
12 #include <GeomDataAPI_Point.h>
13 #include <GeomAPI_Edge.h>
14 #include <GeomAPI_Pnt2d.h>
15 #include <GeomAlgoAPI_ShapeTools.h>
16
17 #include <ModuleBase_ISelection.h>
18 #include <ModuleBase_ViewerPrs.h>
19
20 #include <ModelAPI_Events.h>
21 #include <ModelAPI_Feature.h>
22 #include <ModelAPI_Tools.h>
23
24 #include <ModuleBase_IViewWindow.h>
25 #include <ModuleBase_IWorkshop.h>
26 #include <ModuleBase_IModule.h>
27
28 #include "PartSet_WidgetFeaturePointSelector.h"
29 #include "PartSet_Tools.h"
30
31 //#include <SketchPlugin_ConstraintCoincidence.h>
32 //#include <SketchPlugin_Constraint.h>
33 #include <SketchPlugin_Point.h>
34 #include <SketchPlugin_Trim.h>
35
36 #include <XGUI_Tools.h>
37 #include <XGUI_Workshop.h>
38 #include <XGUI_Displayer.h>
39 #include <XGUI_ViewerProxy.h>
40
41 #include <QWidget>
42 #include <QMouseEvent>
43
44 #define HIGHLIGHT_STAYS_PROBLEM
45 #ifdef HIGHLIGHT_STAYS_PROBLEM
46 #include <Quantity_Color.hxx>
47 #define SKETCH_ENTITY_COLOR "225,0,0"
48 #endif
49
50 PartSet_WidgetFeaturePointSelector::PartSet_WidgetFeaturePointSelector(QWidget* theParent,
51                                                          ModuleBase_IWorkshop* theWorkshop,
52                                                          const Config_WidgetAPI* theData)
53 : ModuleBase_WidgetShapeSelector(theParent, theWorkshop, theData)
54 {
55 }
56
57 PartSet_WidgetFeaturePointSelector::~PartSet_WidgetFeaturePointSelector()
58 {
59 }
60
61 //********************************************************************
62 bool PartSet_WidgetFeaturePointSelector::isValidSelection(
63                                         const std::shared_ptr<ModuleBase_ViewerPrs>& theValue)
64 {
65   return true;
66 }
67
68 //********************************************************************
69 void PartSet_WidgetFeaturePointSelector::activateCustom()
70 {
71   ModuleBase_WidgetShapeSelector::activateCustom();
72
73   myWorkshop->module()->activateCustomPrs(myFeature,
74                             ModuleBase_IModule::CustomizeHighlightedObjects, true);
75
76   Handle(AIS_InteractiveContext) aContext =
77                           XGUI_Tools::workshop(myWorkshop)->viewer()->AISContext();
78
79   std::vector<int> aColors;
80   aColors = Config_PropManager::color("Visualization", "sketch_entity_color",
81                                      SKETCH_ENTITY_COLOR);
82   Quantity_Color aColor(aColors[0] / 255., aColors[1] / 255., aColors[2] / 255., Quantity_TOC_RGB);
83
84 #ifdef HIGHLIGHT_STAYS_PROBLEM
85   Handle(Graphic3d_HighlightStyle) aHStyle = aContext->HighlightStyle();
86   myHighlightColor = aHStyle->Color();
87   aHStyle->SetColor(aColor);
88   aContext->SetHighlightStyle(aHStyle);
89
90   Handle(Graphic3d_HighlightStyle) aSStyle = aContext->SelectionStyle();
91   mySelectionColor = aSStyle->Color();
92   aSStyle->SetColor(aColor);
93   aContext->SetSelectionStyle(aSStyle);
94 #endif
95 }
96
97 //********************************************************************
98 void PartSet_WidgetFeaturePointSelector::deactivate()
99 {
100   ModuleBase_WidgetShapeSelector::deactivate();
101
102   Handle(AIS_InteractiveContext) aContext =
103                           XGUI_Tools::workshop(myWorkshop)->viewer()->AISContext();
104
105 #ifdef HIGHLIGHT_STAYS_PROBLEM
106   Handle(Graphic3d_HighlightStyle) aHStyle = aContext->HighlightStyle();
107   aHStyle->SetColor(myHighlightColor);
108   aContext->SetHighlightStyle(aHStyle);
109
110   Handle(Graphic3d_HighlightStyle) aSStyle = aContext->SelectionStyle();
111   aSStyle->SetColor(mySelectionColor);
112   aContext->SetSelectionStyle(aSStyle);
113 #endif
114 }
115
116 //********************************************************************
117 void PartSet_WidgetFeaturePointSelector::mouseMoved(ModuleBase_IViewWindow* theWindow,
118                                                     QMouseEvent* theEvent)
119 {
120   ModuleBase_ISelection* aSelect = myWorkshop->selection();
121   QList<ModuleBase_ViewerPrsPtr> aHighlighted = aSelect->getHighlighted();
122
123   if (!aHighlighted.empty()) {
124     ModuleBase_ViewerPrsPtr aPrs = aHighlighted.first();
125     fillFeature(aPrs, theWindow, theEvent);
126   }
127 }
128
129 //********************************************************************
130 void PartSet_WidgetFeaturePointSelector::mouseReleased(ModuleBase_IViewWindow* theWindow,
131                                                        QMouseEvent* theEvent)
132 {
133   // the contex menu release by the right button should not be processed by this widget
134   if (theEvent->button() != Qt::LeftButton)
135     return;
136
137   emit focusOutWidget(this);
138 }
139
140 //********************************************************************
141 bool PartSet_WidgetFeaturePointSelector::fillFeature(
142                             const std::shared_ptr<ModuleBase_ViewerPrs>& theSelectedPrs,
143                             ModuleBase_IViewWindow* theWindow,
144                             QMouseEvent* theEvent)
145 {
146   bool aFilled = false;
147   if (theSelectedPrs.get() && theSelectedPrs->object().get()) {
148     ObjectPtr anObject = theSelectedPrs->object();
149     gp_Pnt aPnt = PartSet_Tools::convertClickToPoint(theEvent->pos(), theWindow->v3dView());
150     double aX, anY;
151     Handle(V3d_View) aView = theWindow->v3dView();
152     PartSet_Tools::convertTo2D(aPnt, mySketch, aView, aX, anY);
153
154     std::shared_ptr<ModelAPI_AttributeReference> aRef =
155                             std::dynamic_pointer_cast<ModelAPI_AttributeReference>(
156                             feature()->data()->attribute(SketchPlugin_Trim::BASE_OBJECT()));
157     aRef->setValue(anObject);
158
159     std::shared_ptr<GeomDataAPI_Point2D> anAttributePoint =
160                     std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
161                     feature()->data()->attribute(SketchPlugin_Trim::ENTITY_POINT()));
162     anAttributePoint->setValue(aX, anY);
163     // redisplay AIS presentation in viewer
164 #ifndef HIGHLIGHT_STAYS_PROBLEM
165     // an attempt to clear highlighted item in the viewer: but of OCCT
166     XGUI_Tools::workshop(myWorkshop)->displayer()->clearSelected(true);
167 #endif
168     updateObject(feature());
169     aFilled = true;
170   }
171   return aFilled;
172 }
173
174 //********************************************************************
175 QList<ModuleBase_ViewerPrsPtr> PartSet_WidgetFeaturePointSelector::getAttributeSelection() const
176 {
177   return QList<ModuleBase_ViewerPrsPtr>();
178 }
179
180
181 //********************************************************************
182 bool PartSet_WidgetFeaturePointSelector::setSelection(
183                                           QList<std::shared_ptr<ModuleBase_ViewerPrs>>& theValues,
184                                           const bool theToValidate)
185 {
186   // false is returned to do not emit focus out widget by selected sub-shape
187   return false;
188 }
189
190 void PartSet_WidgetFeaturePointSelector::setPreSelection(
191                                   const std::shared_ptr<ModuleBase_ViewerPrs>& thePreSelected,
192                                   ModuleBase_IViewWindow* theWnd,
193                                   QMouseEvent* theEvent)
194 {
195   if (fillFeature(thePreSelected, theWnd, theEvent))
196     mouseReleased(theWnd, theEvent);
197 }