]> SALOME platform Git repositories - modules/shaper.git/blob - src/PartSet/PartSet_Module.cpp
Salome HOME
5214dd6e1d10dbaeecb3c991a5a4342ccd6fd109
[modules/shaper.git] / src / PartSet / PartSet_Module.cpp
1 #include <PartSet_Module.h>
2 #include <PartSet_OperationSketch.h>
3 #include <PartSet_OperationSketchLine.h>
4 #include <PartSet_OperationEditLine.h>
5 #include <ModuleBase_Operation.h>
6 #include <ModuleBase_OperationDescription.h>
7 #include <PartSet_Listener.h>
8
9 #include <ModuleBase_Operation.h>
10
11 #include <XGUI_MainWindow.h>
12 #include <XGUI_Displayer.h>
13 #include <XGUI_Viewer.h>
14 #include <XGUI_Workshop.h>
15 #include <XGUI_OperationMgr.h>
16 #include <XGUI_SelectionMgr.h>
17 #include <XGUI_ViewPort.h>
18 #include <XGUI_ActionsMgr.h>
19 #include <XGUI_ViewerProxy.h>
20
21 #include <Config_PointerMessage.h>
22 #include <Config_ModuleReader.h>
23 #include <Config_WidgetReader.h>
24 #include <Events_Loop.h>
25 #include <Events_Message.h>
26 #include <Events_Error.h>
27
28 #include <GeomAPI_Shape.h>
29
30 #include <AIS_ListOfInteractive.hxx>
31
32 #include <QObject>
33 #include <QMouseEvent>
34 #include <QString>
35
36 #ifdef _DEBUG
37 #include <QDebug>
38 #endif
39
40
41 /*!Create and return new instance of XGUI_Module*/
42 extern "C" PARTSET_EXPORT XGUI_Module* createModule(XGUI_Workshop* theWshop)
43 {
44   return new PartSet_Module(theWshop);
45 }
46
47 PartSet_Module::PartSet_Module(XGUI_Workshop* theWshop)
48 {
49   myWorkshop = theWshop;
50   myListener = new PartSet_Listener(this);
51
52   XGUI_OperationMgr* anOperationMgr = myWorkshop->operationMgr();
53
54   connect(anOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)),
55           this, SLOT(onOperationStopped(ModuleBase_Operation*)));
56
57   connect(myWorkshop->viewer(), SIGNAL(mousePress(QMouseEvent*)),
58           this, SLOT(onMousePressed(QMouseEvent*)));
59   connect(myWorkshop->viewer(), SIGNAL(mouseRelease(QMouseEvent*)),
60           this, SLOT(onMouseReleased(QMouseEvent*)));
61   connect(myWorkshop->viewer(), SIGNAL(mouseMove(QMouseEvent*)),
62           this, SLOT(onMouseMoved(QMouseEvent*)));
63   connect(myWorkshop->viewer(), SIGNAL(keyRelease(QKeyEvent*)),
64           this, SLOT(onKeyRelease(QKeyEvent*)));
65 }
66
67 PartSet_Module::~PartSet_Module()
68 {
69 }
70
71 XGUI_Workshop* PartSet_Module::workshop() const
72 {
73   return myWorkshop;
74 }
75
76 void PartSet_Module::createFeatures()
77 {
78   Config_ModuleReader aXMLReader = Config_ModuleReader();
79   aXMLReader.readAll();
80   myFeaturesInFiles = aXMLReader.featuresInFiles();
81 }
82
83 void PartSet_Module::featureCreated(XGUI_Command* theFeature)
84 {
85   theFeature->connectTo(this, SLOT(onFeatureTriggered()));
86 }
87
88 QStringList PartSet_Module::nestedFeatures(QString)
89 {
90   return QStringList();
91 }
92
93 std::string PartSet_Module::featureFile(const std::string& theFeatureId)
94 {
95   return myFeaturesInFiles[theFeatureId];
96 }
97
98 /*
99  *
100  */
101 void PartSet_Module::onFeatureTriggered()
102 {
103   XGUI_Command* aCmd = dynamic_cast<XGUI_Command*>(sender());
104   //Do nothing on uncheck
105   if(aCmd->isCheckable() && !aCmd->isChecked())
106     return;
107   launchOperation(aCmd->id());
108 }
109   
110 void PartSet_Module::launchOperation(const QString& theCmdId)
111 {
112   ModuleBase_Operation* anOperation = createOperation(theCmdId.toStdString());
113   sendOperation(anOperation);
114 }
115
116 void PartSet_Module::onOperationStopped(ModuleBase_Operation* theOperation)
117 {
118   if (!theOperation)
119     return;
120   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(theOperation);
121   if (aPreviewOp) {
122   }
123 }
124
125 void PartSet_Module::onMousePressed(QMouseEvent* theEvent)
126 {
127   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(
128                                        myWorkshop->operationMgr()->currentOperation());
129   if (aPreviewOp)
130   {
131     std::list<XGUI_ViewerPrs> aPresentations = myWorkshop->displayer()->GetViewerPrs();
132     aPreviewOp->mousePressed(theEvent, myWorkshop->viewer()->activeView(), aPresentations);
133   }
134 }
135
136 void PartSet_Module::onMouseReleased(QMouseEvent* theEvent)
137 {
138   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(
139                                        myWorkshop->operationMgr()->currentOperation());
140   if (aPreviewOp)
141   {
142     std::list<XGUI_ViewerPrs> aPresentations = myWorkshop->displayer()->GetViewerPrs();
143     aPreviewOp->mouseReleased(theEvent, myWorkshop->viewer()->activeView(), aPresentations);
144   }
145 }
146
147 void PartSet_Module::onMouseMoved(QMouseEvent* theEvent)
148 {
149   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(
150                                        myWorkshop->operationMgr()->currentOperation());
151   if (aPreviewOp)
152     aPreviewOp->mouseMoved(theEvent, myWorkshop->viewer()->activeView());
153 }
154
155 void PartSet_Module::onKeyRelease(QKeyEvent* theEvent)
156 {
157   ModuleBase_Operation* anOperation = myWorkshop->operationMgr()->currentOperation();
158   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
159   if (aPreviewOp) {
160     aPreviewOp->keyReleased(theEvent->key());
161   }
162 }
163
164 void PartSet_Module::onPlaneSelected(double theX, double theY, double theZ)
165 {
166   myWorkshop->viewer()->setViewProjection(theX, theY, theZ);
167
168   ModuleBase_Operation* anOperation = myWorkshop->operationMgr()->currentOperation();
169   if (anOperation) {
170     PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
171     if (aPreviewOp) {
172       visualizePreview(aPreviewOp->feature(), false);
173     }
174   }
175
176   myWorkshop->actionsMgr()->setNestedActionsEnabled(true);
177 }
178
179 void PartSet_Module::onLaunchOperation(std::string theName, boost::shared_ptr<ModelAPI_Feature> theFeature)
180 {
181   ModuleBase_Operation* anOperation = createOperation(theName.c_str());
182   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
183   if (aPreviewOp)
184   {
185     std::list<XGUI_ViewerPrs> aPresentations = myWorkshop->displayer()->GetViewerPrs();
186     aPreviewOp->init(theFeature, aPresentations);
187   }
188   myWorkshop->actionsMgr()->setActionChecked(anOperation->getDescription()->operationId(), true);
189   sendOperation(anOperation);
190 }
191
192 void PartSet_Module::onMultiSelectionEnabled(bool theEnabled)
193 {
194   XGUI_ViewerProxy* aViewer = myWorkshop->viewer();
195   aViewer->enableMultiselection(theEnabled);
196 }
197
198 void PartSet_Module::onStopSelection(const std::list<XGUI_ViewerPrs>& theFeatures, const bool isStop,
199                                      const bool isToSelect)
200 {
201   XGUI_Displayer* aDisplayer = myWorkshop->displayer();
202   aDisplayer->StopSelection(theFeatures, isStop, false);
203   if (!isStop) {
204     std::list<XGUI_ViewerPrs>::const_iterator anIt = theFeatures.begin(), aLast = theFeatures.end();
205     boost::shared_ptr<ModelAPI_Feature> aFeature;
206     for (; anIt != aLast; anIt++) {
207       activateFeature((*anIt).feature(), false);
208     }
209   }
210   if (isToSelect)
211     aDisplayer->SetSelected(theFeatures, false);
212   aDisplayer->UpdateViewer();
213 }
214
215 void PartSet_Module::onFeatureConstructed(boost::shared_ptr<ModelAPI_Feature> theFeature,
216                                           int theMode)
217 {
218   bool isDisplay = theMode != PartSet_OperationSketchBase::FM_Abort;
219   visualizePreview(theFeature, isDisplay, false);
220
221   if (theMode == PartSet_OperationSketchBase::FM_Activation ||
222       theMode == PartSet_OperationSketchBase::FM_Deactivation)
223     activateFeature(theFeature, true);
224 }
225
226 ModuleBase_Operation* PartSet_Module::createOperation(const std::string& theCmdId)
227 {
228   // get operation xml description
229   std::string aStdCmdId = theCmdId;
230   if (aStdCmdId == PartSet_OperationEditLine::Type())
231     aStdCmdId = PartSet_OperationSketchLine::Type();
232   std::string aPluginFileName = featureFile(aStdCmdId);
233   Config_WidgetReader aWdgReader = Config_WidgetReader(aPluginFileName);
234   aWdgReader.readAll();
235   std::string aXmlCfg = aWdgReader.featureWidgetCfg(aStdCmdId);
236   std::string aDescription = aWdgReader.featureDescription(aStdCmdId);
237
238   // create the operation
239   ModuleBase_Operation* anOperation;
240   if (theCmdId == PartSet_OperationSketch::Type()) {
241     anOperation = new PartSet_OperationSketch(theCmdId.c_str(), this);
242   }
243   else if(theCmdId == PartSet_OperationSketchLine::Type() ||
244           theCmdId == PartSet_OperationEditLine::Type()) {
245     ModuleBase_Operation* aCurOperation = myWorkshop->operationMgr()->currentOperation();
246     boost::shared_ptr<ModelAPI_Feature> aSketchFeature;
247     if (aCurOperation)
248       aSketchFeature = aCurOperation->feature();
249     if (theCmdId == PartSet_OperationSketchLine::Type())
250       anOperation = new PartSet_OperationSketchLine(theCmdId.c_str(), this, aSketchFeature);
251     else
252       anOperation = new PartSet_OperationEditLine(theCmdId.c_str(), this, aSketchFeature);
253   }
254   else {
255     anOperation = new ModuleBase_Operation(theCmdId.c_str(), this);
256   }
257   anOperation->getDescription()->setXmlRepresentation(QString::fromStdString(aXmlCfg));
258   anOperation->getDescription()->setDescription(QString::fromStdString(aDescription));
259
260   // connect the operation
261   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
262   if (aPreviewOp) {
263     connect(aPreviewOp, SIGNAL(featureConstructed(boost::shared_ptr<ModelAPI_Feature>, int)),
264             this, SLOT(onFeatureConstructed(boost::shared_ptr<ModelAPI_Feature>, int)));
265     connect(aPreviewOp, SIGNAL(launchOperation(std::string, boost::shared_ptr<ModelAPI_Feature>)),
266             this, SLOT(onLaunchOperation(std::string, boost::shared_ptr<ModelAPI_Feature>)));
267     connect(aPreviewOp, SIGNAL(multiSelectionEnabled(bool)),
268             this, SLOT(onMultiSelectionEnabled(bool)));
269
270     connect(aPreviewOp, SIGNAL(multiSelectionEnabled(bool)),
271             this, SLOT(onMultiSelectionEnabled(bool)));
272     connect(aPreviewOp, SIGNAL(stopSelection(const std::list<XGUI_ViewerPrs>&, const bool, const bool)),
273             this, SLOT(onStopSelection(const std::list<XGUI_ViewerPrs>&, const bool, const bool)));
274
275     PartSet_OperationSketch* aSketchOp = dynamic_cast<PartSet_OperationSketch*>(aPreviewOp);
276     if (aSketchOp) {
277       connect(aSketchOp, SIGNAL(planeSelected(double, double, double)),
278               this, SLOT(onPlaneSelected(double, double, double)));
279     }
280   }
281
282   return anOperation;
283 }
284
285 void PartSet_Module::sendOperation(ModuleBase_Operation* theOperation)
286 {
287   //TODO(sbh): Implement static method to extract event id [SEID]
288   static Events_ID aModuleEvent = Events_Loop::eventByName("PartSetModuleEvent");
289   Config_PointerMessage aMessage(aModuleEvent, this);
290   aMessage.setPointer(theOperation);
291   Events_Loop::loop()->send(aMessage);
292 }
293
294 void PartSet_Module::visualizePreview(boost::shared_ptr<ModelAPI_Feature> theFeature, bool isDisplay,
295                                       const bool isUpdateViewer)
296 {
297   ModuleBase_Operation* anOperation = myWorkshop->operationMgr()->currentOperation();
298   if (!anOperation)
299     return;
300
301   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
302   if (!aPreviewOp)
303     return;
304
305   XGUI_Displayer* aDisplayer = myWorkshop->displayer();
306   if (isDisplay) {
307     boost::shared_ptr<GeomAPI_Shape> aPreview = aPreviewOp->preview(theFeature);
308     aDisplayer->Redisplay(theFeature,
309                           aPreview ? aPreview->impl<TopoDS_Shape>() : TopoDS_Shape(), false);
310   }
311   else
312     aDisplayer->Erase(anOperation->feature(), false);
313
314   if (isUpdateViewer)
315     aDisplayer->UpdateViewer();
316 }
317
318 void PartSet_Module::activateFeature(boost::shared_ptr<ModelAPI_Feature> theFeature,
319                                      const bool isUpdateViewer)
320 {
321   ModuleBase_Operation* anOperation = myWorkshop->operationMgr()->currentOperation();
322   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
323   if (aPreviewOp) {
324     XGUI_Displayer* aDisplayer = myWorkshop->displayer();
325     aDisplayer->ActivateInLocalContext(theFeature, aPreviewOp->getSelectionModes(theFeature),
326                                        isUpdateViewer);
327   }
328 }
329
330 void PartSet_Module::updateCurrentPreview(const std::string& theCmdId)
331 {
332   ModuleBase_Operation* anOperation = myWorkshop->operationMgr()->currentOperation();
333   if (!anOperation)
334     return;
335
336   PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
337   if (!aPreviewOp)
338     return;
339
340   boost::shared_ptr<ModelAPI_Feature> aFeature = aPreviewOp->feature();
341   if (!aFeature || aFeature->getKind() != theCmdId)
342     return;
343
344   std::map<boost::shared_ptr<ModelAPI_Feature>, boost::shared_ptr<GeomAPI_Shape> >
345                                                                      aList = aPreviewOp->preview();
346   XGUI_Displayer* aDisplayer = myWorkshop->displayer();
347   std::list<int> aModes = aPreviewOp->getSelectionModes(aPreviewOp->feature());
348
349   std::map<boost::shared_ptr<ModelAPI_Feature>, boost::shared_ptr<GeomAPI_Shape> >::const_iterator
350                                                          anIt = aList.begin(), aLast = aList.end();
351   for (; anIt != aLast; anIt++) {
352     boost::shared_ptr<ModelAPI_Feature> aFeature = (*anIt).first;
353     boost::shared_ptr<GeomAPI_Shape> aPreview = (*anIt).second;
354     aDisplayer->Redisplay(aFeature,
355                           aPreview ? aPreview->impl<TopoDS_Shape>() : TopoDS_Shape(), false);
356     aDisplayer->ActivateInLocalContext(aFeature, aModes, false);
357   }
358   aDisplayer->UpdateViewer();
359 }
360