1 #include "ModuleBase_IModule.h"
2 #include "XGUI_Constants.h"
3 #include "XGUI_Command.h"
4 #include "XGUI_MainMenu.h"
5 #include "XGUI_MainWindow.h"
6 #include "XGUI_MenuGroupPanel.h"
7 #include "XGUI_Tools.h"
8 #include "XGUI_Workbench.h"
9 #include "XGUI_Workshop.h"
10 #include "XGUI_Viewer.h"
11 #include "XGUI_SelectionMgr.h"
12 #include "XGUI_Selection.h"
13 #include "XGUI_ObjectsBrowser.h"
14 #include "XGUI_Displayer.h"
15 #include "XGUI_OperationMgr.h"
16 #include "XGUI_SalomeConnector.h"
17 #include "XGUI_SalomeViewer.h"
18 #include "XGUI_ActionsMgr.h"
19 #include "XGUI_ErrorDialog.h"
20 #include "XGUI_ViewerProxy.h"
21 #include "XGUI_PropertyPanel.h"
22 #include "XGUI_ContextMenuMgr.h"
23 #include "XGUI_ModuleConnector.h"
24 #include "XGUI_Preferences.h"
25 #include <XGUI_QtEvents.h>
27 #include <ModelAPI_Events.h>
28 #include <ModelAPI_Session.h>
29 #include <ModelAPI_Feature.h>
30 #include <ModelAPI_Data.h>
31 #include <ModelAPI_AttributeDocRef.h>
32 #include <ModelAPI_Object.h>
33 #include <ModelAPI_Validator.h>
34 #include <ModelAPI_ResultConstruction.h>
35 #include <ModelAPI_ResultBody.h>
37 #include <PartSetPlugin_Part.h>
39 #include <Events_Loop.h>
40 #include <Events_Error.h>
41 #include <Events_LongOp.h>
43 #include <ModuleBase_Operation.h>
44 #include <ModuleBase_Operation.h>
45 #include <ModuleBase_OperationDescription.h>
46 #include <ModuleBase_SelectionValidator.h>
47 #include "ModuleBase_WidgetFactory.h"
49 #include <Config_Common.h>
50 #include <Config_FeatureMessage.h>
51 #include <Config_PointerMessage.h>
52 #include <Config_ModuleReader.h>
54 #include <QApplication>
55 #include <QFileDialog>
56 #include <QMessageBox>
57 #include <QMdiSubWindow>
58 #include <QPushButton>
59 #include <QDockWidget>
75 QMap<QString, QString> XGUI_Workshop::myIcons;
77 QString XGUI_Workshop::featureIcon(const std::string& theId)
79 QString aId(theId.c_str());
80 if (myIcons.contains(aId))
85 XGUI_Workshop::XGUI_Workshop(XGUI_SalomeConnector* theConnector)
87 myCurrentDir(QString()),
89 mySalomeConnector(theConnector),
94 myPartActivating(false)
96 myMainWindow = mySalomeConnector ? 0 : new XGUI_MainWindow();
98 myDisplayer = new XGUI_Displayer(this);
100 mySelector = new XGUI_SelectionMgr(this);
101 //connect(mySelector, SIGNAL(selectionChanged()), this, SLOT(updateModuleCommands()));
103 myOperationMgr = new XGUI_OperationMgr(this);
104 myActionsMgr = new XGUI_ActionsMgr(this);
105 myErrorDlg = new XGUI_ErrorDialog(myMainWindow);
106 myContextMenuMgr = new XGUI_ContextMenuMgr(this);
107 connect(myContextMenuMgr, SIGNAL(actionTriggered(const QString&, bool)), this,
108 SLOT(onContextMenuCommand(const QString&, bool)));
110 myViewerProxy = new XGUI_ViewerProxy(this);
111 connect(myViewerProxy, SIGNAL(selectionChanged()), this, SLOT(updateCommandsOnViewSelection()));
113 myModuleConnector = new XGUI_ModuleConnector(this);
115 connect(myOperationMgr, SIGNAL(operationStarted()), SLOT(onOperationStarted()));
116 connect(myOperationMgr, SIGNAL(operationResumed()), SLOT(onOperationStarted()));
117 connect(myOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)),
118 SLOT(onOperationStopped(ModuleBase_Operation*)));
119 connect(myMainWindow, SIGNAL(exitKeySequence()), SLOT(onExit()));
120 connect(myOperationMgr, SIGNAL(operationStarted()), myActionsMgr, SLOT(update()));
121 connect(myOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)), myActionsMgr,
123 connect(this, SIGNAL(errorOccurred(const QString&)), myErrorDlg, SLOT(addError(const QString&)));
126 //******************************************************
127 XGUI_Workshop::~XGUI_Workshop(void)
131 //******************************************************
132 void XGUI_Workshop::startApplication()
135 //Initialize event listening
136 Events_Loop* aLoop = Events_Loop::loop();
137 aLoop->registerListener(this, Events_Error::errorID()); //!< Listening application errors.
138 //TODO(sbh): Implement static method to extract event id [SEID]
139 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_LOADED));
140 // TODO Is it good to use non standard event within workshop?
141 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OPERATION_LAUNCHED));
142 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
143 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_CREATED));
144 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
145 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_DELETED));
146 aLoop->registerListener(this, Events_Loop::eventByName("LongOperation"));
147 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_PLUGIN_LOADED));
148 aLoop->registerListener(this, Events_Loop::eventByName("CurrentDocumentChanged"));
150 registerValidators();
153 myMainWindow->show();
154 updateCommandStatus();
156 XGUI_Preferences::loadCustomProps();
160 //******************************************************
161 void XGUI_Workshop::initMenu()
163 myContextMenuMgr->createActions();
165 if (isSalomeMode()) {
166 // Create only Undo, Redo commands
167 QAction* aAction = salomeConnector()->addEditCommand("UNDO_CMD", tr("Undo"),
168 tr("Undo last command"),
169 QIcon(":pictures/undo.png"),
170 QKeySequence::Undo, false);
171 connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onUndo()));
172 aAction = salomeConnector()->addEditCommand("REDO_CMD", tr("Redo"), tr("Redo last command"),
173 QIcon(":pictures/redo.png"), QKeySequence::Redo,
175 connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onRedo()));
176 salomeConnector()->addEditMenuSeparator();
179 // File commands group
180 XGUI_MenuGroupPanel* aGroup = myMainWindow->menuObject()->generalPage();
182 XGUI_Command* aCommand;
184 aCommand = aGroup->addFeature("SAVE_CMD", tr("Save..."), tr("Save the document"),
185 QIcon(":pictures/save.png"), QKeySequence::Save);
186 aCommand->connectTo(this, SLOT(onSave()));
187 //aCommand->disable();
189 aCommand = aGroup->addFeature("UNDO_CMD", tr("Undo"), tr("Undo last command"),
190 QIcon(":pictures/undo.png"), QKeySequence::Undo);
191 aCommand->connectTo(this, SLOT(onUndo()));
193 aCommand = aGroup->addFeature("REDO_CMD", tr("Redo"), tr("Redo last command"),
194 QIcon(":pictures/redo.png"), QKeySequence::Redo);
195 aCommand->connectTo(this, SLOT(onRedo()));
197 aCommand = aGroup->addFeature("REBUILD_CMD", tr("Rebuild"), tr("Rebuild data objects"),
198 QIcon(":pictures/rebuild.png"));
199 aCommand->connectTo(this, SLOT(onRebuild()));
201 aCommand = aGroup->addFeature("SAVEAS_CMD", tr("Save as..."), tr("Save the document into a file"),
202 QIcon(":pictures/save.png"));
203 aCommand->connectTo(this, SLOT(onSaveAs()));
204 //aCommand->disable();
206 aCommand = aGroup->addFeature("OPEN_CMD", tr("Open..."), tr("Open a new document"),
207 QIcon(":pictures/open.png"), QKeySequence::Open);
208 aCommand->connectTo(this, SLOT(onOpen()));
210 //aCommand = aGroup->addFeature("NEW_CMD", tr("New"), tr("Create a new document"),
211 // QIcon(":pictures/new.png"), QKeySequence::New);
212 //aCommand->connectTo(this, SLOT(onNew()));
214 aCommand = aGroup->addFeature("PREF_CMD", tr("Preferences"), tr("Edit preferences"),
215 QIcon(":pictures/preferences.png"), QKeySequence::Preferences);
216 aCommand->connectTo(this, SLOT(onPreferences()));
218 aCommand = aGroup->addFeature("EXIT_CMD", tr("Exit"), tr("Exit application"),
219 QIcon(":pictures/close.png"), QKeySequence::Close);
220 aCommand->connectTo(this, SLOT(onExit()));
221 //FIXME: SBH's test action. Can be used for some GUI tests.
223 // aCommand = aGroup->addFeature("TEST_CMD", "Test!", "Private debug button",
224 // QIcon(":pictures/close.png"), QKeySequence(), true);
225 // aCommand->connectTo(myMainWindow, SLOT(dockPythonConsole()));
229 //******************************************************
230 XGUI_Workbench* XGUI_Workshop::addWorkbench(const QString& theName)
232 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
233 return aMenuBar->addWorkbench(theName);
236 //******************************************************
237 void XGUI_Workshop::processEvent(const boost::shared_ptr<Events_Message>& theMessage)
239 if (QApplication::instance()->thread() != QThread::currentThread()) {
241 std::cout << "XGUI_Workshop::processEvent: " << "Working in another thread." << std::endl;
243 SessionPtr aMgr = ModelAPI_Session::get();
244 PostponeMessageQtEvent* aPostponeEvent = new PostponeMessageQtEvent(theMessage);
245 QApplication::postEvent(this, aPostponeEvent);
249 //A message to start feature creation received.
250 if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_LOADED)) {
251 boost::shared_ptr<Config_FeatureMessage> aFeatureMsg =
252 boost::dynamic_pointer_cast<Config_FeatureMessage>(theMessage);
253 if (!aFeatureMsg->isInternal()) {
254 addFeature(aFeatureMsg);
258 // Process creation of Part
259 else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_CREATED)) {
260 boost::shared_ptr<ModelAPI_ObjectUpdatedMessage> aUpdMsg =
261 boost::dynamic_pointer_cast<ModelAPI_ObjectUpdatedMessage>(theMessage);
262 onFeatureCreatedMsg(aUpdMsg);
264 if (mySalomeConnector)
265 mySalomeConnector->createPreferences();
266 myUpdatePrefs = false;
269 else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_PLUGIN_LOADED)) {
270 myUpdatePrefs = true;
274 else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY)) {
275 boost::shared_ptr<ModelAPI_ObjectUpdatedMessage> aUpdMsg =
276 boost::dynamic_pointer_cast<ModelAPI_ObjectUpdatedMessage>(theMessage);
277 onFeatureRedisplayMsg(aUpdMsg);
280 //Update property panel on corresponding message. If there is no current operation (no
281 //property panel), or received message has different feature to the current - do nothing.
282 else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED)) {
283 boost::shared_ptr<ModelAPI_ObjectUpdatedMessage> anUpdateMsg =
284 boost::dynamic_pointer_cast<ModelAPI_ObjectUpdatedMessage>(theMessage);
285 onFeatureUpdatedMsg(anUpdateMsg);
288 else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED)) {
289 boost::shared_ptr<ModelAPI_ObjectDeletedMessage> aDelMsg =
290 boost::dynamic_pointer_cast<ModelAPI_ObjectDeletedMessage>(theMessage);
291 onObjectDeletedMsg(aDelMsg);
294 else if (theMessage->eventID() == Events_LongOp::eventID()) {
295 if (Events_LongOp::isPerformed())
296 QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
298 QApplication::restoreOverrideCursor();
301 //An operation passed by message. Start it, process and commit.
302 else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OPERATION_LAUNCHED)) {
303 boost::shared_ptr<Config_PointerMessage> aPartSetMsg =
304 boost::dynamic_pointer_cast<Config_PointerMessage>(theMessage);
305 //myPropertyPanel->cleanContent();
306 ModuleBase_Operation* anOperation = (ModuleBase_Operation*) aPartSetMsg->pointer();
308 if (myOperationMgr->startOperation(anOperation)) {
309 myPropertyPanel->updateContentWidget(anOperation->feature());
310 if (!anOperation->getDescription()->hasXmlRepresentation()) {
311 if (anOperation->commit())
312 updateCommandStatus();
316 else if (theMessage->eventID() == Events_Loop::loop()->eventByName("CurrentDocumentChanged")) {
317 // Find and Activate active part
318 if (myPartActivating)
320 SessionPtr aMgr = ModelAPI_Session::get();
321 DocumentPtr aActiveDoc = aMgr->activeDocument();
322 DocumentPtr aDoc = aMgr->moduleDocument();
323 if (aActiveDoc == aDoc) {
324 activatePart(ResultPartPtr());
327 std::string aGrpName = ModelAPI_ResultPart::group();
328 for (int i = 0; i < aDoc->size(aGrpName); i++) {
329 ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aDoc->object(aGrpName, i));
330 if (aPart->partDoc() == aActiveDoc) {
331 activatePart(aPart); // Activate a part which corresponds to active Doc
335 // If not found then activate global document
336 activatePart(ResultPartPtr());
339 //Show error dialog if error message received.
340 boost::shared_ptr<Events_Error> anAppError = boost::dynamic_pointer_cast<Events_Error>(theMessage);
342 emit errorOccurred(QString::fromLatin1(anAppError->description()));
345 if (!isSalomeMode()) {
346 SessionPtr aMgr = ModelAPI_Session::get();
347 if (aMgr->isModified() != myMainWindow->isModifiedState())
348 myMainWindow->setModifiedState(aMgr->isModified());
353 //******************************************************
354 void XGUI_Workshop::onStartWaiting()
356 if (Events_LongOp::isPerformed()) {
357 QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
361 //******************************************************
362 void XGUI_Workshop::onFeatureUpdatedMsg(const boost::shared_ptr<ModelAPI_ObjectUpdatedMessage>& theMsg)
364 std::set<ObjectPtr> aFeatures = theMsg->objects();
365 if (myOperationMgr->hasOperation()) {
366 FeaturePtr aCurrentFeature = myOperationMgr->currentOperation()->feature();
367 std::set<ObjectPtr>::const_iterator aIt;
368 for (aIt = aFeatures.begin(); aIt != aFeatures.end(); ++aIt) {
369 ObjectPtr aNewFeature = (*aIt);
370 if (aNewFeature == aCurrentFeature) {
371 myPropertyPanel->updateContentWidget(aCurrentFeature);
376 myOperationMgr->onValidateOperation();
378 myObjectBrowser->processEvent(theMsg);
381 //******************************************************
382 void XGUI_Workshop::onFeatureRedisplayMsg(const boost::shared_ptr<ModelAPI_ObjectUpdatedMessage>& theMsg)
384 std::set<ObjectPtr> aObjects = theMsg->objects();
385 std::set<ObjectPtr>::const_iterator aIt;
386 for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
387 ObjectPtr aObj = (*aIt);
388 if (!aObj->data() || !aObj->data()->isValid())
389 myDisplayer->erase(aObj, false);
391 if (myDisplayer->isVisible(aObj)) // TODO VSV: Correction sketch drawing
392 myDisplayer->display(aObj, false); // In order to update presentation
394 if (myOperationMgr->hasOperation()) {
395 ModuleBase_Operation* aOperation = myOperationMgr->currentOperation();
396 if (aOperation->hasObject(aObj)) { // Display only current operation results
397 myDisplayer->display(aObj, false);
403 myDisplayer->updateViewer();
406 //******************************************************
407 void XGUI_Workshop::onFeatureCreatedMsg(const boost::shared_ptr<ModelAPI_ObjectUpdatedMessage>& theMsg)
409 std::set<ObjectPtr> aObjects = theMsg->objects();
411 std::set<ObjectPtr>::const_iterator aIt;
412 bool aHasPart = false;
413 bool isDisplayed = false;
414 for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
415 ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(*aIt);
418 // If a feature is created from the aplication's python console
419 // it doesn't stored in the operation mgr and doesn't displayed
420 } else if (myOperationMgr->hasOperation()) {
421 ModuleBase_Operation* aOperation = myOperationMgr->currentOperation();
422 if (aOperation->hasObject(*aIt)) { // Display only current operation results
423 myDisplayer->display(*aIt, false);
429 myObjectBrowser->processEvent(theMsg);
431 myDisplayer->updateViewer();
437 //******************************************************
438 void XGUI_Workshop::onObjectDeletedMsg(const boost::shared_ptr<ModelAPI_ObjectDeletedMessage>& theMsg)
441 myObjectBrowser->processEvent(theMsg);
442 //std::set<ObjectPtr> aFeatures = theMsg->objects();
445 //******************************************************
446 void XGUI_Workshop::onOperationStarted()
448 ModuleBase_Operation* aOperation = myOperationMgr->currentOperation();
449 if (this->isSalomeMode())
450 aOperation->setNestedFeatures(mySalomeConnector->nestedActions(aOperation->id()));
452 aOperation->setNestedFeatures(myActionsMgr->nestedCommands(aOperation->id()));
454 if (aOperation->getDescription()->hasXmlRepresentation()) { //!< No need for property panel
455 connectWithOperation(aOperation);
458 QString aXmlRepr = aOperation->getDescription()->xmlRepresentation();
459 ModuleBase_WidgetFactory aFactory = ModuleBase_WidgetFactory(aXmlRepr.toStdString(),
462 myPropertyPanel->cleanContent();
463 aFactory.createWidget(myPropertyPanel->contentWidget());
465 QList<ModuleBase_ModelWidget*> aWidgets = aFactory.getModelWidgets();
466 QList<ModuleBase_ModelWidget*>::const_iterator anIt = aWidgets.begin(), aLast = aWidgets.end();
467 ModuleBase_ModelWidget* aWidget;
468 for (; anIt != aLast; anIt++) {
470 aWidget->setFeature(aOperation->feature());
471 QObject::connect(aWidget, SIGNAL(valuesChanged()), this, SLOT(onWidgetValuesChanged()));
472 // Init default values
473 if (!aOperation->isEditOperation() && !aWidget->isComputedDefault()) {
474 aWidget->storeValue();
478 myPropertyPanel->setModelWidgets(aWidgets);
479 myPropertyPanel->onActivateNextWidget(NULL);
480 myPropertyPanel->setWindowTitle(aOperation->getDescription()->description());
482 updateCommandStatus();
485 //******************************************************
486 void XGUI_Workshop::onOperationStopped(ModuleBase_Operation* theOperation)
488 //!< No need for property panel
489 updateCommandStatus();
491 myPropertyPanel->cleanContent();
494 bool XGUI_Workshop::event(QEvent * theEvent)
496 PostponeMessageQtEvent* aPostponedEv = dynamic_cast<PostponeMessageQtEvent*>(theEvent);
499 std::cout << "XGUI_Workshop::event " << "Got PostponeMessageQtEvent" << std::endl;
500 bool isMyThread = (QApplication::instance()->thread() == QThread::currentThread());
501 std::cout << "XGUI_Workshop::event " << "I am in the Qt's thread: "
502 << isMyThread << std::endl;
504 boost::shared_ptr<Events_Message> aEventPtr = aPostponedEv->postponedMessage();
505 processEvent(aEventPtr);
514 void XGUI_Workshop::addFeature(const boost::shared_ptr<Config_FeatureMessage>& theMessage)
518 qDebug() << "XGUI_Workshop::addFeature: NULL message.";
522 // Remember features icons
523 myIcons[QString::fromStdString(theMessage->id())] = QString::fromStdString(theMessage->icon());
525 //Find or create Workbench
526 QString aWchName = QString::fromStdString(theMessage->workbenchId());
527 QString aNestedFeatures = QString::fromStdString(theMessage->nestedFeatures());
528 bool isUsePropPanel = theMessage->isUseInput();
529 QString aFeatureId = QString::fromStdString(theMessage->id());
530 if (isSalomeMode()) {
531 QAction* aAction = salomeConnector()->addFeature(aWchName, aFeatureId,
532 QString::fromStdString(theMessage->text()),
533 QString::fromStdString(theMessage->tooltip()),
534 QIcon(theMessage->icon().c_str()),
535 QKeySequence(), isUsePropPanel);
536 salomeConnector()->setNestedActions(aFeatureId, aNestedFeatures.split(" ", QString::SkipEmptyParts));
537 myActionsMgr->addCommand(aAction);
538 myModule->featureCreated(aAction);
541 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
542 XGUI_Workbench* aPage = aMenuBar->findWorkbench(aWchName);
544 aPage = addWorkbench(aWchName);
546 //Find or create Group
547 QString aGroupName = QString::fromStdString(theMessage->groupId());
548 XGUI_MenuGroupPanel* aGroup = aPage->findGroup(aGroupName);
550 aGroup = aPage->addGroup(aGroupName);
552 // Check if hotkey sequence is already defined:
553 QKeySequence aHotKey = myActionsMgr->registerShortcut(
554 QString::fromStdString(theMessage->keysequence()));
556 XGUI_Command* aCommand = aGroup->addFeature(aFeatureId,
557 QString::fromStdString(theMessage->text()),
558 QString::fromStdString(theMessage->tooltip()),
559 QIcon(theMessage->icon().c_str()), aHotKey,
561 aCommand->setNestedCommands(aNestedFeatures.split(" ", QString::SkipEmptyParts));
562 myActionsMgr->addCommand(aCommand);
563 myModule->featureCreated(aCommand);
568 * Makes a signal/slot connections between Property Panel
569 * and given operation. The given operation becomes a
570 * current operation and previous operation if exists
572 void XGUI_Workshop::connectWithOperation(ModuleBase_Operation* theOperation)
574 QAction* aCommand = 0;
575 if (isSalomeMode()) {
576 aCommand = salomeConnector()->command(theOperation->getDescription()->operationId());
578 XGUI_MainMenu* aMenu = myMainWindow->menuObject();
579 FeaturePtr aFeature = theOperation->feature();
581 aCommand = aMenu->feature(QString::fromStdString(aFeature->getKind()));
583 //Abort operation on uncheck the command
585 connect(aCommand, SIGNAL(triggered(bool)), theOperation, SLOT(setRunning(bool)));
590 * Saves document with given name.
592 void XGUI_Workshop::saveDocument(const QString& theName, std::list<std::string>& theFileNames)
594 QApplication::restoreOverrideCursor();
595 SessionPtr aMgr = ModelAPI_Session::get();
596 aMgr->save(theName.toLatin1().constData(), theFileNames);
597 QApplication::restoreOverrideCursor();
600 bool XGUI_Workshop::isActiveOperationAborted()
602 return myOperationMgr->abortAllOperations();
605 //******************************************************
606 void XGUI_Workshop::onExit()
608 SessionPtr aMgr = ModelAPI_Session::get();
609 if (aMgr->isModified()) {
610 int anAnswer = QMessageBox::question(
611 myMainWindow, tr("Save current file"), tr("The document is modified, save before exit?"),
612 QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
613 if (anAnswer == QMessageBox::Save) {
614 bool saved = onSave();
618 } else if (anAnswer == QMessageBox::Cancel) {
625 //******************************************************
626 void XGUI_Workshop::onNew()
628 QApplication::setOverrideCursor(Qt::WaitCursor);
629 if (objectBrowser() == 0) {
631 mySelector->connectViewers();
633 myViewerProxy->connectToViewer();
635 if (!isSalomeMode()) {
636 myMainWindow->showPythonConsole();
637 QMdiSubWindow* aWnd = myMainWindow->viewer()->createView();
638 aWnd->showMaximized();
639 updateCommandStatus();
641 myContextMenuMgr->connectViewer();
642 QApplication::restoreOverrideCursor();
645 //******************************************************
646 void XGUI_Workshop::onOpen()
648 if(!isActiveOperationAborted())
650 //save current file before close if modified
651 SessionPtr aSession = ModelAPI_Session::get();
652 if (aSession->isModified()) {
653 //TODO(sbh): re-launch the app?
654 int anAnswer = QMessageBox::question(
655 myMainWindow, tr("Save current file"),
656 tr("The document is modified, save before opening another?"),
657 QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
658 if (anAnswer == QMessageBox::Save) {
660 } else if (anAnswer == QMessageBox::Cancel) {
663 aSession->moduleDocument()->close();
667 //show file dialog, check if readable and open
668 myCurrentDir = QFileDialog::getExistingDirectory(mainWindow());
669 if (myCurrentDir.isEmpty())
671 QFileInfo aFileInfo(myCurrentDir);
672 if (!aFileInfo.exists() || !aFileInfo.isReadable()) {
673 QMessageBox::critical(myMainWindow, tr("Warning"), tr("Unable to open the file."));
677 QApplication::setOverrideCursor(Qt::WaitCursor);
678 aSession->load(myCurrentDir.toLatin1().constData());
679 myObjectBrowser->rebuildDataTree();
681 updateCommandStatus();
682 QApplication::restoreOverrideCursor();
685 //******************************************************
686 bool XGUI_Workshop::onSave()
688 if(!isActiveOperationAborted())
690 if (myCurrentDir.isEmpty()) {
693 std::list<std::string> aFiles;
694 saveDocument(myCurrentDir, aFiles);
695 updateCommandStatus();
696 myMainWindow->setModifiedState(false);
700 //******************************************************
701 bool XGUI_Workshop::onSaveAs()
703 if(!isActiveOperationAborted())
705 QFileDialog dialog(mainWindow());
706 dialog.setWindowTitle(tr("Select directory to save files..."));
707 dialog.setFileMode(QFileDialog::Directory);
708 dialog.setFilter(tr("Folders (*)"));
709 dialog.setOptions(QFileDialog::HideNameFilterDetails | QFileDialog::ShowDirsOnly);
710 dialog.setViewMode(QFileDialog::Detail);
712 if (!dialog.exec()) {
715 QString aTempDir = dialog.selectedFiles().first();
717 if (aDir.exists() && !aDir.entryInfoList(QDir::NoDotAndDotDot | QDir::AllEntries).isEmpty()) {
718 int answer = QMessageBox::question(
720 //: Title of the dialog which asks user if he wants to save study in existing non-empty folder
722 tr("The folder already contains some files, save anyway?"),
723 QMessageBox::Save | QMessageBox::Cancel);
724 if (answer == QMessageBox::Cancel) {
728 myCurrentDir = aTempDir;
729 if (!isSalomeMode()) {
730 myMainWindow->setCurrentDir(myCurrentDir, false);
731 myMainWindow->setModifiedState(false);
736 //******************************************************
737 void XGUI_Workshop::onUndo()
739 objectBrowser()->treeView()->setCurrentIndex(QModelIndex());
740 SessionPtr aMgr = ModelAPI_Session::get();
741 if (aMgr->isOperation())
742 operationMgr()->onAbortOperation();
744 updateCommandStatus();
747 //******************************************************
748 void XGUI_Workshop::onRedo()
750 objectBrowser()->treeView()->setCurrentIndex(QModelIndex());
751 SessionPtr aMgr = ModelAPI_Session::get();
752 if (aMgr->isOperation())
753 operationMgr()->onAbortOperation();
755 updateCommandStatus();
758 //******************************************************
759 void XGUI_Workshop::onRebuild()
761 static const Events_ID aRebuildEvent = Events_Loop::loop()->eventByName("Rebuild");
762 Events_Loop::loop()->send(boost::shared_ptr<Events_Message>(
763 new Events_Message(aRebuildEvent, this)));
766 //******************************************************
767 void XGUI_Workshop::onPreferences()
770 XGUI_Preferences::editPreferences(aModif);
771 if (aModif.size() > 0) {
773 foreach (XGUI_Pref aPref, aModif)
775 aSection = aPref.first;
776 if (aSection == XGUI_Preferences::VIEWER_SECTION) {
778 myMainWindow->viewer()->updateFromResources();
779 } else if (aSection == XGUI_Preferences::MENU_SECTION) {
781 myMainWindow->menuObject()->updateFromResources();
787 //******************************************************
788 ModuleBase_IModule* XGUI_Workshop::loadModule(const QString& theModule)
790 QString libName = QString::fromStdString(library(theModule.toStdString()));
791 if (libName.isEmpty()) {
792 qWarning(qPrintable(tr("Information about module \"%1\" doesn't exist.").arg(theModule)));
797 CREATE_FUNC crtInst = 0;
800 HINSTANCE modLib = ::LoadLibrary((LPTSTR) qPrintable(libName));
804 FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
805 0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
806 QString aMsg((char*) &lpMsgBuf);
807 err = QString("Failed to load %1. %2").arg(libName).arg(aMsg);
808 ::LocalFree(lpMsgBuf);
810 crtInst = (CREATE_FUNC) ::GetProcAddress(modLib, CREATE_MODULE);
814 FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM
815 | FORMAT_MESSAGE_IGNORE_INSERTS,
816 0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
817 QString aMsg((char*) &lpMsgBuf);
818 err = QString("Failed to find %1 function. %2").arg( CREATE_MODULE).arg(aMsg);
819 ::LocalFree(lpMsgBuf);
823 void* modLib = dlopen( libName.toLatin1(), RTLD_LAZY | RTLD_GLOBAL );
825 err = QString( "Can not load library %1. %2" ).arg( libName ).arg( dlerror() );
827 crtInst = (CREATE_FUNC)dlsym( modLib, CREATE_MODULE );
829 err = QString( "Failed to find function %1. %2" ).arg( CREATE_MODULE ).arg( dlerror() );
834 ModuleBase_IModule* aModule = crtInst ? crtInst(this) : 0;
836 if (!err.isEmpty()) {
838 QMessageBox::warning(mainWindow(), tr("Error"), err);
840 qWarning(qPrintable(err));
846 //******************************************************
847 bool XGUI_Workshop::activateModule()
849 Config_ModuleReader aModuleReader;
850 QString moduleName = QString::fromStdString(aModuleReader.getModuleName());
851 myModule = loadModule(moduleName);
854 myModule->createFeatures();
855 myActionsMgr->update();
859 //******************************************************
860 void XGUI_Workshop::updateCommandStatus()
862 QList<QAction*> aCommands;
863 if (isSalomeMode()) { // update commands in SALOME mode
864 aCommands = salomeConnector()->commandList();
866 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
867 foreach (XGUI_Command* aCmd, aMenuBar->features())
868 aCommands.append(aCmd);
870 SessionPtr aMgr = ModelAPI_Session::get();
871 if (aMgr->hasModuleDocument()) {
874 foreach(QAction* aCmd, aCommands) {
875 QString aId = aCmd->data().toString();
876 if (aId == "UNDO_CMD")
878 else if (aId == "REDO_CMD")
881 // Enable all commands
882 aCmd->setEnabled(true);
884 aUndoCmd->setEnabled(aMgr->canUndo());
885 aRedoCmd->setEnabled(aMgr->canRedo());
887 foreach(QAction* aCmd, aCommands) {
888 QString aId = aCmd->data().toString();
889 if (aId == "NEW_CMD")
890 aCmd->setEnabled(true);
891 else if (aId == "EXIT_CMD")
892 aCmd->setEnabled(true);
894 aCmd->setEnabled(false);
897 myActionsMgr->update();
900 //******************************************************
901 QList<QAction*> XGUI_Workshop::getModuleCommands() const
903 QList<QAction*> aCommands;
904 if (isSalomeMode()) { // update commands in SALOME mode
905 aCommands = salomeConnector()->commandList();
907 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
908 foreach(XGUI_Command* aCmd, aMenuBar->features())
910 aCommands.append(aCmd);
916 //******************************************************
917 QDockWidget* XGUI_Workshop::createObjectBrowser(QWidget* theParent)
919 QDockWidget* aObjDock = new QDockWidget(theParent);
920 aObjDock->setAllowedAreas(Qt::LeftDockWidgetArea | Qt::RightDockWidgetArea | Qt::BottomDockWidgetArea);
921 aObjDock->setWindowTitle(tr("Object browser"));
922 aObjDock->setStyleSheet(
923 "::title { position: relative; padding-left: 5px; text-align: left center }");
924 myObjectBrowser = new XGUI_ObjectsBrowser(aObjDock);
925 connect(myObjectBrowser, SIGNAL(activePartChanged(ObjectPtr)), this,
926 SLOT(changeCurrentDocument(ObjectPtr)));
927 aObjDock->setWidget(myObjectBrowser);
929 myContextMenuMgr->connectObjectBrowser();
933 //******************************************************
935 * Creates dock widgets, places them in corresponding area
936 * and tabifies if necessary.
938 void XGUI_Workshop::createDockWidgets()
940 QMainWindow* aDesktop = isSalomeMode() ? salomeConnector()->desktop() : myMainWindow;
941 QDockWidget* aObjDock = createObjectBrowser(aDesktop);
942 aDesktop->addDockWidget(Qt::LeftDockWidgetArea, aObjDock);
943 myPropertyPanel = new XGUI_PropertyPanel(aDesktop);
944 myPropertyPanel->setAllowedAreas(Qt::LeftDockWidgetArea | Qt::RightDockWidgetArea | Qt::BottomDockWidgetArea);
945 aDesktop->addDockWidget(Qt::LeftDockWidgetArea, myPropertyPanel);
946 hidePropertyPanel(); //<! Invisible by default
948 aDesktop->tabifyDockWidget(aObjDock, myPropertyPanel);
950 QPushButton* aOkBtn = myPropertyPanel->findChild<QPushButton*>(XGUI::PROP_PANEL_OK);
951 connect(aOkBtn, SIGNAL(clicked()), myOperationMgr, SLOT(onCommitOperation()));
952 QPushButton* aCancelBtn = myPropertyPanel->findChild<QPushButton*>(XGUI::PROP_PANEL_CANCEL);
953 connect(aCancelBtn, SIGNAL(clicked()), myOperationMgr, SLOT(onAbortOperation()));
954 connect(myPropertyPanel, SIGNAL(keyReleased(QKeyEvent*)), myOperationMgr,
955 SLOT(onKeyReleased(QKeyEvent*)));
956 connect(myPropertyPanel, SIGNAL(widgetActivated(ModuleBase_ModelWidget*)), myOperationMgr,
957 SLOT(onWidgetActivated(ModuleBase_ModelWidget*)));
958 connect(myOperationMgr, SIGNAL(activateNextWidget(ModuleBase_ModelWidget*)), myPropertyPanel,
959 SLOT(onActivateNextWidget(ModuleBase_ModelWidget*)));
960 connect(myOperationMgr, SIGNAL(operationValidated(bool)), myPropertyPanel,
961 SLOT(setAcceptEnabled(bool)));
965 //******************************************************
966 void XGUI_Workshop::showPropertyPanel()
968 QAction* aViewAct = myPropertyPanel->toggleViewAction();
969 //<! Restore ability to close panel from the window's menu
970 aViewAct->setEnabled(true);
971 myPropertyPanel->show();
972 myPropertyPanel->raise();
975 //******************************************************
976 void XGUI_Workshop::hidePropertyPanel()
978 QAction* aViewAct = myPropertyPanel->toggleViewAction();
979 //<! Do not allow to show empty property panel
980 aViewAct->setEnabled(false);
981 myPropertyPanel->hide();
984 //******************************************************
985 void XGUI_Workshop::showObjectBrowser()
987 myObjectBrowser->parentWidget()->show();
990 //******************************************************
991 void XGUI_Workshop::hideObjectBrowser()
993 myObjectBrowser->parentWidget()->hide();
996 //******************************************************
997 void XGUI_Workshop::onFeatureTriggered()
999 QAction* aCmd = dynamic_cast<QAction*>(sender());
1001 QString aId = salomeConnector()->commandId(aCmd);
1003 myModule->launchOperation(aId);
1007 //******************************************************
1008 void XGUI_Workshop::changeCurrentDocument(ObjectPtr theObj)
1010 SessionPtr aMgr = ModelAPI_Session::get();
1012 ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(theObj);
1014 DocumentPtr aPartDoc = aPart->partDoc();
1016 aMgr->setActiveDocument(aPartDoc);
1021 aMgr->setActiveDocument(aMgr->moduleDocument());
1024 //******************************************************
1025 void XGUI_Workshop::salomeViewerSelectionChanged()
1027 emit salomeViewerSelection();
1030 //**************************************************************
1031 XGUI_SalomeViewer* XGUI_Workshop::salomeViewer() const
1033 return mySalomeConnector->viewer();
1036 //**************************************************************
1037 void XGUI_Workshop::onContextMenuCommand(const QString& theId, bool isChecked)
1039 QList<ObjectPtr> aObjects = mySelector->selection()->selectedObjects();
1040 if ((theId == "ACTIVATE_PART_CMD") && (aObjects.size() > 0)) {
1041 ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aObjects.first());
1042 activatePart(aPart);
1043 } else if (theId == "DEACTIVATE_PART_CMD")
1044 activatePart(ResultPartPtr());
1045 else if (theId == "DELETE_CMD")
1046 deleteObjects(aObjects);
1047 else if (theId == "SHOW_CMD")
1048 showObjects(aObjects, true);
1049 else if (theId == "HIDE_CMD")
1050 showObjects(aObjects, false);
1051 else if (theId == "SHOW_ONLY_CMD")
1052 showOnlyObjects(aObjects);
1055 //**************************************************************
1056 void XGUI_Workshop::onWidgetValuesChanged()
1058 ModuleBase_Operation* anOperation = myOperationMgr->currentOperation();
1059 FeaturePtr aFeature = anOperation->feature();
1061 ModuleBase_ModelWidget* aSenderWidget = dynamic_cast<ModuleBase_ModelWidget*>(sender());
1063 // aCustom->storeValue(aFeature);
1065 const QList<ModuleBase_ModelWidget*>& aWidgets = myPropertyPanel->modelWidgets();
1066 QList<ModuleBase_ModelWidget*>::const_iterator anIt = aWidgets.begin(), aLast = aWidgets.end();
1067 for (; anIt != aLast; anIt++) {
1068 ModuleBase_ModelWidget* aCustom = *anIt;
1069 if (aCustom && (/*!aCustom->isInitialized(aFeature) ||*/aCustom == aSenderWidget)) {
1070 //aCustom->storeValue(aFeature);
1071 aCustom->storeValue();
1076 //**************************************************************
1077 void XGUI_Workshop::activatePart(ResultPartPtr theFeature)
1079 if (!myPartActivating) {
1080 myPartActivating = true;
1082 theFeature->activate();
1083 changeCurrentDocument(theFeature);
1084 myObjectBrowser->activatePart(theFeature);
1085 myPartActivating = false;
1089 //**************************************************************
1090 void XGUI_Workshop::activateLastPart()
1092 SessionPtr aMgr = ModelAPI_Session::get();
1093 DocumentPtr aDoc = aMgr->moduleDocument();
1094 std::string aGrpName = ModelAPI_ResultPart::group();
1095 ObjectPtr aLastPart = aDoc->object(aGrpName, aDoc->size(aGrpName) - 1);
1096 ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aLastPart);
1098 activatePart(aPart);
1102 //**************************************************************
1103 void XGUI_Workshop::deleteObjects(const QList<ObjectPtr>& theList)
1105 QMainWindow* aDesktop = isSalomeMode() ? salomeConnector()->desktop() : myMainWindow;
1106 QMessageBox::StandardButton aRes = QMessageBox::warning(
1107 aDesktop, tr("Delete features"), tr("Seleted features will be deleted. Continue?"),
1108 QMessageBox::No | QMessageBox::Yes, QMessageBox::No);
1109 // ToDo: definbe deleting method
1110 if (aRes == QMessageBox::Yes) {
1111 SessionPtr aMgr = ModelAPI_Session::get();
1112 aMgr->startOperation();
1113 foreach (ObjectPtr aObj, theList)
1115 ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aObj);
1117 DocumentPtr aDoc = aPart->document();
1118 if (aDoc == aMgr->activeDocument()) {
1121 //aMgr->moduleDocument()->removeFeature(aPart->owner());
1123 FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
1125 aObj->document()->removeFeature(aFeature);
1128 myDisplayer->updateViewer();
1129 aMgr->finishOperation();
1133 //**************************************************************
1134 void XGUI_Workshop::showObjects(const QList<ObjectPtr>& theList, bool isVisible)
1136 foreach (ObjectPtr aObj, theList)
1138 ResultPtr aRes = boost::dynamic_pointer_cast<ModelAPI_Result>(aObj);
1141 myDisplayer->display(aRes, false);
1143 myDisplayer->erase(aRes, false);
1147 myDisplayer->updateViewer();
1150 //**************************************************************
1151 void XGUI_Workshop::showOnlyObjects(const QList<ObjectPtr>& theList)
1153 myDisplayer->eraseAll(false);
1154 showObjects(theList, true);
1158 //**************************************************************
1159 void XGUI_Workshop::updateCommandsOnViewSelection()
1161 XGUI_Selection* aSelection = mySelector->selection();
1162 if (aSelection->getSelected().size() == 0)
1165 // Restrict validators to manage only nested (child) features
1166 // of the current feature i.e. if current feature is Sketch -
1167 // Sketch Features & Constraints can be validated.
1168 QStringList aNestedIds;
1169 if(myOperationMgr->hasOperation()) {
1170 FeaturePtr aFeature = myOperationMgr->currentOperation()->feature();
1172 aNestedIds << myActionsMgr->nestedCommands(QString::fromStdString(aFeature->getKind()));
1175 SessionPtr aMgr = ModelAPI_Session::get();
1176 ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
1177 QList<QAction*> aActions = getModuleCommands();
1178 foreach(QAction* aAction, aActions) {
1179 QString aId = aAction->data().toString();
1180 if(!aNestedIds.contains(aId))
1182 std::list<ModelAPI_Validator*> aValidators;
1183 std::list<std::list<std::string> > anArguments;
1184 aFactory->validators(aId.toStdString(), aValidators, anArguments);
1185 std::list<ModelAPI_Validator*>::iterator aValidator = aValidators.begin();
1186 for (; aValidator != aValidators.end(); aValidator++) {
1188 const ModuleBase_SelectionValidator* aSelValidator =
1189 dynamic_cast<const ModuleBase_SelectionValidator*>(*aValidator);
1190 if (aSelValidator) {
1191 aAction->setEnabled(aSelValidator->isValid(aSelection));
1198 //**************************************************************
1199 void XGUI_Workshop::registerValidators() const
1201 SessionPtr aMgr = ModelAPI_Session::get();
1202 ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
1205 //**************************************************************
1206 void XGUI_Workshop::displayAllResults()
1208 SessionPtr aMgr = ModelAPI_Session::get();
1209 DocumentPtr aRootDoc = aMgr->moduleDocument();
1210 displayDocumentResults(aRootDoc);
1211 for (int i = 0; i < aRootDoc->size(ModelAPI_ResultPart::group()); i++) {
1212 ObjectPtr aObject = aRootDoc->object(ModelAPI_ResultPart::group(), i);
1213 ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aObject);
1214 displayDocumentResults(aPart->partDoc());
1216 myDisplayer->updateViewer();
1219 //**************************************************************
1220 void XGUI_Workshop::displayDocumentResults(DocumentPtr theDoc)
1224 displayGroupResults(theDoc, ModelAPI_ResultConstruction::group());
1225 displayGroupResults(theDoc, ModelAPI_ResultBody::group());
1228 //**************************************************************
1229 void XGUI_Workshop::displayGroupResults(DocumentPtr theDoc, std::string theGroup)
1231 for (int i = 0; i < theDoc->size(theGroup); i++)
1232 myDisplayer->display(theDoc->object(theGroup, i), false);