1 #include <PartSet_Module.h>
2 #include <PartSet_OperationSketch.h>
3 #include <PartSet_OperationSketchLine.h>
4 #include <PartSet_OperationEditLine.h>
5 #include <PartSet_OperationConstraint.h>
6 #include <ModuleBase_Operation.h>
7 #include <ModuleBase_OperationDescription.h>
8 #include <PartSet_Listener.h>
9 #include <PartSet_TestOCC.h>
10 #include <PartSet_Presentation.h>
12 #include <ModuleBase_Operation.h>
13 #include <ModelAPI_Object.h>
15 #include <XGUI_MainWindow.h>
16 #include <XGUI_Displayer.h>
17 #include <XGUI_Viewer.h>
18 #include <XGUI_Workshop.h>
19 #include <XGUI_OperationMgr.h>
20 #include <XGUI_SelectionMgr.h>
21 #include <XGUI_ViewPort.h>
22 #include <XGUI_ActionsMgr.h>
23 #include <XGUI_ViewerProxy.h>
24 #include <XGUI_ContextMenuMgr.h>
25 #include <XGUI_PropertyPanel.h>
26 #include <XGUI_Tools.h>
28 #include <Config_PointerMessage.h>
29 #include <Config_ModuleReader.h>
30 #include <Config_WidgetReader.h>
31 #include <Events_Loop.h>
32 #include <Events_Message.h>
33 #include <Events_Error.h>
35 #include <GeomAPI_Shape.h>
37 #include <AIS_ListOfInteractive.hxx>
38 #include <AIS_DimensionSelectionMode.hxx>
41 #include <QMouseEvent>
48 /*!Create and return new instance of XGUI_Module*/
49 extern "C" PARTSET_EXPORT XGUI_Module* createModule(XGUI_Workshop* theWshop)
51 return new PartSet_Module(theWshop);
54 PartSet_Module::PartSet_Module(XGUI_Workshop* theWshop)
56 myWorkshop = theWshop;
57 myListener = new PartSet_Listener(this);
59 XGUI_OperationMgr* anOperationMgr = myWorkshop->operationMgr();
61 connect(anOperationMgr, SIGNAL(operationStarted()),
62 this, SLOT(onOperationStarted()));
64 connect(anOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)),
65 this, SLOT(onOperationStopped(ModuleBase_Operation*)));
67 XGUI_ContextMenuMgr* aContextMenuMgr = myWorkshop->contextMenuMgr();
68 connect(aContextMenuMgr, SIGNAL(actionTriggered(const QString&, bool)),
69 this, SLOT(onContextMenuCommand(const QString&, bool)));
71 connect(myWorkshop->viewer(), SIGNAL(mousePress(QMouseEvent*)),
72 this, SLOT(onMousePressed(QMouseEvent*)));
73 connect(myWorkshop->viewer(), SIGNAL(mouseRelease(QMouseEvent*)),
74 this, SLOT(onMouseReleased(QMouseEvent*)));
75 connect(myWorkshop->viewer(), SIGNAL(mouseMove(QMouseEvent*)),
76 this, SLOT(onMouseMoved(QMouseEvent*)));
77 connect(myWorkshop->viewer(), SIGNAL(keyRelease(QKeyEvent*)),
78 this, SLOT(onKeyRelease(QKeyEvent*)));
81 PartSet_Module::~PartSet_Module()
85 XGUI_Workshop* PartSet_Module::workshop() const
90 void PartSet_Module::createFeatures()
92 Config_ModuleReader aXMLReader = Config_ModuleReader();
94 myFeaturesInFiles = aXMLReader.featuresInFiles();
97 void PartSet_Module::featureCreated(QAction* theFeature)
99 connect(theFeature, SIGNAL(triggered(bool)), this, SLOT(onFeatureTriggered()));
102 QStringList PartSet_Module::nestedFeatures(QString)
104 return QStringList();
107 std::string PartSet_Module::featureFile(const std::string& theFeatureId)
109 return myFeaturesInFiles[theFeatureId];
115 void PartSet_Module::onFeatureTriggered()
117 //PartSet_TestOCC::local_selection_change_shape(myWorkshop->viewer()->AISContext(),
118 // myWorkshop->viewer()->activeView());
120 //PartSet_TestOCC::local_selection_erase(myWorkshop->viewer()->AISContext(),
121 // myWorkshop->viewer()->activeView());
122 QAction* aCmd = dynamic_cast<QAction*>(sender());
123 //Do nothing on uncheck
124 if(aCmd->isCheckable() && !aCmd->isChecked())
126 launchOperation(aCmd->data().toString());
129 void PartSet_Module::launchOperation(const QString& theCmdId)
131 ModuleBase_Operation* anOperation = createOperation(theCmdId.toStdString());
132 sendOperation(anOperation);
135 void PartSet_Module::onOperationStarted()
137 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(
138 myWorkshop->operationMgr()->currentOperation());
140 XGUI_PropertyPanel* aPropPanel = myWorkshop->propertyPanel();
141 connect(aPreviewOp, SIGNAL(focusActivated(const std::string&)),
142 aPropPanel, SLOT(onFocusActivated(const std::string&)));
146 void PartSet_Module::onOperationStopped(ModuleBase_Operation* theOperation)
150 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(theOperation);
152 XGUI_PropertyPanel* aPropPanel = myWorkshop->propertyPanel();
153 disconnect(aPreviewOp, SIGNAL(focusActivated(const std::string&)),
154 aPropPanel, SLOT(onFocusActivated(const std::string&)));
158 void PartSet_Module::onContextMenuCommand(const QString& theId, bool isChecked)
160 QFeatureList aFeatures = myWorkshop->selector()->selectedFeatures();
161 if (theId == "EDIT_CMD" && (aFeatures.size() > 0)) {
162 editFeature(aFeatures.first());
166 void PartSet_Module::onMousePressed(QMouseEvent* theEvent)
168 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(
169 myWorkshop->operationMgr()->currentOperation());
172 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
173 std::list<XGUI_ViewerPrs> aSelected = aDisplayer->getSelected();
174 std::list<XGUI_ViewerPrs> aHighlighted = aDisplayer->getHighlighted();
176 aPreviewOp->mousePressed(theEvent, myWorkshop->viewer()->activeView(), aSelected, aHighlighted);
180 void PartSet_Module::onMouseReleased(QMouseEvent* theEvent)
182 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(
183 myWorkshop->operationMgr()->currentOperation());
186 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
187 std::list<XGUI_ViewerPrs> aSelected = aDisplayer->getSelected();
188 std::list<XGUI_ViewerPrs> aHighlighted = aDisplayer->getHighlighted();
190 aPreviewOp->mouseReleased(theEvent, myWorkshop->viewer()->activeView(), aSelected, aHighlighted);
194 void PartSet_Module::onMouseMoved(QMouseEvent* theEvent)
196 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(
197 myWorkshop->operationMgr()->currentOperation());
199 aPreviewOp->mouseMoved(theEvent, myWorkshop->viewer()->activeView());
202 void PartSet_Module::onKeyRelease(QKeyEvent* theEvent)
204 ModuleBase_Operation* anOperation = myWorkshop->operationMgr()->currentOperation();
205 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
207 aPreviewOp->keyReleased(theEvent->key());
211 void PartSet_Module::onPlaneSelected(double theX, double theY, double theZ)
213 myWorkshop->viewer()->setViewProjection(theX, theY, theZ);
214 myWorkshop->actionsMgr()->update();
216 //PartSet_TestOCC::testSelection(myWorkshop);
219 void PartSet_Module::onFitAllView()
221 myWorkshop->viewer()->fitAll();
224 void PartSet_Module::onLaunchOperation(std::string theName, boost::shared_ptr<ModelAPI_Feature> theFeature)
226 ModuleBase_Operation* anOperation = createOperation(theName.c_str());
227 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
230 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
231 // refill the features list with avoiding of the features, obtained only by vertex shape (TODO)
232 std::list<XGUI_ViewerPrs> aSelected = aDisplayer->getSelected(TopAbs_VERTEX);
233 std::list<XGUI_ViewerPrs> aHighlighted = aDisplayer->getHighlighted(TopAbs_VERTEX);
234 aPreviewOp->init(theFeature, aSelected, aHighlighted);
236 anOperation->setFeature(theFeature);
238 sendOperation(anOperation);
239 myWorkshop->actionsMgr()->updateCheckState();
242 void PartSet_Module::onMultiSelectionEnabled(bool theEnabled)
244 XGUI_ViewerProxy* aViewer = myWorkshop->viewer();
245 aViewer->enableMultiselection(theEnabled);
248 void PartSet_Module::onStopSelection(const std::list<XGUI_ViewerPrs>& theFeatures, const bool isStop)
250 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
252 std::list<XGUI_ViewerPrs>::const_iterator anIt = theFeatures.begin(), aLast = theFeatures.end();
253 boost::shared_ptr<ModelAPI_Feature> aFeature;
254 for (; anIt != aLast; anIt++) {
255 activateFeature((*anIt).feature(), false);
258 aDisplayer->stopSelection(theFeatures, isStop, false);
260 XGUI_ViewerProxy* aViewer = myWorkshop->viewer();
261 aViewer->enableSelection(!isStop);
263 aDisplayer->updateViewer();
266 void PartSet_Module::onSetSelection(const std::list<XGUI_ViewerPrs>& theFeatures)
268 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
269 aDisplayer->setSelected(theFeatures, false);
270 aDisplayer->updateViewer();
273 void PartSet_Module::onCloseLocalContext()
275 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
276 aDisplayer->closeLocalContexts();
279 void PartSet_Module::onFeatureConstructed(boost::shared_ptr<ModelAPI_Feature> theFeature,
282 bool isDisplay = theMode != PartSet_OperationSketchBase::FM_Hide;
283 visualizePreview(theFeature, isDisplay, false);
285 ModuleBase_Operation* aCurOperation = myWorkshop->operationMgr()->currentOperation();
286 boost::shared_ptr<ModelAPI_Feature> aSketch;
287 PartSet_OperationSketchBase* aPrevOp = dynamic_cast<PartSet_OperationSketchBase*>(aCurOperation);
289 std::map<boost::shared_ptr<ModelAPI_Feature>, boost::shared_ptr<GeomAPI_Shape> >
290 aList = aPrevOp->subPreview();
291 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
292 std::list<int> aModes = aPrevOp->getSelectionModes(aPrevOp->feature());
294 std::map<boost::shared_ptr<ModelAPI_Feature>, boost::shared_ptr<GeomAPI_Shape> >::const_iterator
295 anIt = aList.begin(), aLast = aList.end();
296 for (; anIt != aLast; anIt++) {
297 boost::shared_ptr<ModelAPI_Feature> aFeature = (*anIt).first;
298 visualizePreview(aFeature, false, false);
300 aDisplayer->updateViewer();
304 if (theMode == PartSet_OperationSketchBase::FM_Activation ||
305 theMode == PartSet_OperationSketchBase::FM_Deactivation)
306 activateFeature(theFeature, true);
309 ModuleBase_Operation* PartSet_Module::createOperation(const std::string& theCmdId)
311 // get operation xml description
312 std::string aStdCmdId = theCmdId;
313 if (aStdCmdId == PartSet_OperationEditLine::Type())
314 aStdCmdId = PartSet_OperationSketchLine::Type();
315 std::string aPluginFileName = featureFile(aStdCmdId);
316 Config_WidgetReader aWdgReader = Config_WidgetReader(aPluginFileName);
317 aWdgReader.readAll();
318 std::string aXmlCfg = aWdgReader.featureWidgetCfg(aStdCmdId);
319 std::string aDescription = aWdgReader.featureDescription(aStdCmdId);
321 // create the operation
322 ModuleBase_Operation* anOperation = 0;
323 if (theCmdId == PartSet_OperationSketch::Type()) {
324 anOperation = new PartSet_OperationSketch(theCmdId.c_str(), this);
327 ModuleBase_Operation* aCurOperation = myWorkshop->operationMgr()->currentOperation();
328 boost::shared_ptr<ModelAPI_Feature> aSketch;
329 PartSet_OperationSketchBase* aPrevOp = dynamic_cast<PartSet_OperationSketchBase*>(aCurOperation);
331 aSketch = aPrevOp->sketch();
332 if (theCmdId == PartSet_OperationSketchLine::Type())
333 anOperation = new PartSet_OperationSketchLine(theCmdId.c_str(), this, aSketch);
334 else if (theCmdId == PartSet_OperationEditLine::Type())
335 anOperation = new PartSet_OperationEditLine(theCmdId.c_str(), this, aSketch);
336 else if (theCmdId == PartSet_OperationConstraint::Type())
337 anOperation = new PartSet_OperationConstraint(theCmdId.c_str(), this, aSketch);
341 anOperation = new ModuleBase_Operation(theCmdId.c_str(), this);
343 anOperation->getDescription()->setXmlRepresentation(QString::fromStdString(aXmlCfg));
344 anOperation->getDescription()->setDescription(QString::fromStdString(aDescription));
346 // connect the operation
347 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
349 connect(aPreviewOp, SIGNAL(featureConstructed(boost::shared_ptr<ModelAPI_Feature>, int)),
350 this, SLOT(onFeatureConstructed(boost::shared_ptr<ModelAPI_Feature>, int)));
351 connect(aPreviewOp, SIGNAL(launchOperation(std::string, boost::shared_ptr<ModelAPI_Feature>)),
352 this, SLOT(onLaunchOperation(std::string, boost::shared_ptr<ModelAPI_Feature>)));
353 connect(aPreviewOp, SIGNAL(multiSelectionEnabled(bool)),
354 this, SLOT(onMultiSelectionEnabled(bool)));
356 connect(aPreviewOp, SIGNAL(multiSelectionEnabled(bool)),
357 this, SLOT(onMultiSelectionEnabled(bool)));
358 connect(aPreviewOp, SIGNAL(stopSelection(const std::list<XGUI_ViewerPrs>&, const bool)),
359 this, SLOT(onStopSelection(const std::list<XGUI_ViewerPrs>&, const bool)));
360 connect(aPreviewOp, SIGNAL(setSelection(const std::list<XGUI_ViewerPrs>&)),
361 this, SLOT(onSetSelection(const std::list<XGUI_ViewerPrs>&)));
363 connect(aPreviewOp, SIGNAL(closeLocalContext()),
364 this, SLOT(onCloseLocalContext()));
366 PartSet_OperationSketch* aSketchOp = dynamic_cast<PartSet_OperationSketch*>(aPreviewOp);
368 connect(aSketchOp, SIGNAL(planeSelected(double, double, double)),
369 this, SLOT(onPlaneSelected(double, double, double)));
370 connect(aSketchOp, SIGNAL(fitAllView()),
371 this, SLOT(onFitAllView()));
378 void PartSet_Module::sendOperation(ModuleBase_Operation* theOperation)
380 //TODO(sbh): Implement static method to extract event id [SEID]
381 static Events_ID aModuleEvent = Events_Loop::eventByName("PartSetModuleEvent");
382 Config_PointerMessage aMessage(aModuleEvent, this);
383 aMessage.setPointer(theOperation);
384 Events_Loop::loop()->send(aMessage);
387 void PartSet_Module::visualizePreview(boost::shared_ptr<ModelAPI_Feature> theFeature, bool isDisplay,
388 const bool isUpdateViewer)
390 ModuleBase_Operation* anOperation = myWorkshop->operationMgr()->currentOperation();
394 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
398 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
400 boost::shared_ptr<GeomAPI_Shape> aPreview = aPreviewOp->preview(theFeature);
401 Handle(AIS_InteractiveObject) anAIS = PartSet_Presentation::createPresentation(
402 theFeature, aPreviewOp->sketch(),
403 aPreview ? aPreview->impl<TopoDS_Shape>() : TopoDS_Shape(),
404 aDisplayer->getAISObject(theFeature));
406 int aSelectionMode = -1;
407 if (theFeature->getKind() == "SketchConstraintLength") {
408 aSelectionMode = AIS_DSM_Text;
410 aDisplayer->redisplay(theFeature, anAIS, aSelectionMode, false);
413 aDisplayer->erase(theFeature, false);
416 aDisplayer->updateViewer();
419 void PartSet_Module::activateFeature(boost::shared_ptr<ModelAPI_Feature> theFeature,
420 const bool isUpdateViewer)
422 ModuleBase_Operation* anOperation = myWorkshop->operationMgr()->currentOperation();
423 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
425 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
426 aDisplayer->activateInLocalContext(theFeature, aPreviewOp->getSelectionModes(theFeature),
431 void PartSet_Module::updateCurrentPreview(const std::string& theCmdId)
433 ModuleBase_Operation* anOperation = myWorkshop->operationMgr()->currentOperation();
437 PartSet_OperationSketchBase* aPreviewOp = dynamic_cast<PartSet_OperationSketchBase*>(anOperation);
441 boost::shared_ptr<ModelAPI_Feature> aFeature = aPreviewOp->feature();
442 if (!aFeature || aFeature->getKind() != theCmdId)
445 std::map<boost::shared_ptr<ModelAPI_Feature>, boost::shared_ptr<GeomAPI_Shape> >
446 aList = aPreviewOp->subPreview();
447 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
448 std::list<int> aModes = aPreviewOp->getSelectionModes(aPreviewOp->feature());
450 std::map<boost::shared_ptr<ModelAPI_Feature>, boost::shared_ptr<GeomAPI_Shape> >::const_iterator
451 anIt = aList.begin(), aLast = aList.end();
452 for (; anIt != aLast; anIt++) {
453 boost::shared_ptr<ModelAPI_Feature> aFeature = (*anIt).first;
454 boost::shared_ptr<GeomAPI_Shape> aPreview = (*anIt).second;
455 Handle(AIS_InteractiveObject) anAIS = PartSet_Presentation::createPresentation(
456 aFeature, aPreviewOp->sketch(),
457 aPreview ? aPreview->impl<TopoDS_Shape>() : TopoDS_Shape(),
458 aDisplayer->getAISObject(aFeature));
460 aDisplayer->redisplay(aFeature, anAIS, -1, false);
461 aDisplayer->activateInLocalContext(aFeature, aModes, false);
463 aDisplayer->updateViewer();
466 void PartSet_Module::editFeature(FeaturePtr theFeature)
471 // if (theFeature->getKind() == SKETCH_KIND) {
472 FeaturePtr aFeature = theFeature;
473 if (XGUI_Tools::isModelObject(aFeature)) {
474 ObjectPtr aObject = boost::dynamic_pointer_cast<ModelAPI_Object>(aFeature);
475 aFeature = aObject->featureRef();
479 onLaunchOperation(aFeature->getKind(), aFeature);
480 updateCurrentPreview(aFeature->getKind());