1 #include "XGUI_Module.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 "ModuleBase_WidgetFactory.h"
12 #include "XGUI_SelectionMgr.h"
13 #include "XGUI_ObjectsBrowser.h"
14 #include "XGUI_Displayer.h"
15 #include "XGUI_OperationMgr.h"
16 #include "XGUI_SalomeConnector.h"
17 #include "XGUI_ActionsMgr.h"
19 #include <ModelAPI_PluginManager.h>
20 #include <ModelAPI_Feature.h>
21 #include <ModelAPI_Data.h>
22 #include <ModelAPI_AttributeDocRef.h>
24 #include <Events_Loop.h>
25 #include <ModuleBase_PropPanelOperation.h>
26 #include <ModuleBase_Operation.h>
27 #include <Config_FeatureMessage.h>
28 #include <Config_PointerMessage.h>
30 #include <QApplication>
31 #include <QFileDialog>
32 #include <QMessageBox>
33 #include <QMdiSubWindow>
34 #include <QPushButton>
35 #include <QDockWidget>
48 XGUI_Workshop::XGUI_Workshop(XGUI_SalomeConnector* theConnector)
50 myCurrentFile(QString()),
51 myPartSetModule(NULL),
52 mySalomeConnector(theConnector),
53 myPropertyPanelDock(0),
57 myMainWindow = mySalomeConnector? 0 : new XGUI_MainWindow();
59 // In SALOME viewer is accessible only when module is initialized
60 // and in SALOME mode myDisplayer object has to be created later
61 // So, displayer will be created on demand.
63 mySelector = new XGUI_SelectionMgr(this);
64 myOperationMgr = new XGUI_OperationMgr(this);
65 myActionsMgr = new XGUI_ActionsMgr(this);
66 connect(myOperationMgr, SIGNAL(operationStarted()), this, SLOT(onOperationStarted()));
67 connect(myOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)),
68 this, SLOT(onOperationStopped(ModuleBase_Operation*)));
71 //******************************************************
72 XGUI_Workshop::~XGUI_Workshop(void)
76 //******************************************************
77 void XGUI_Workshop::startApplication()
80 //Initialize event listening
81 Events_Loop* aLoop = Events_Loop::loop();
82 //TODO(sbh): Implement static method to extract event id [SEID]
83 Events_ID aFeatureId = aLoop->eventByName("FeatureEvent");
84 aLoop->registerListener(this, aFeatureId);
85 Events_ID aPartSetId = aLoop->eventByName("PartSetModuleEvent");
86 aLoop->registerListener(this, aPartSetId);
90 updateCommandStatus();
93 // Testing of document creation
94 //boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
95 //boost::shared_ptr<ModelAPI_Feature> aPoint1 = aMgr->rootDocument()->addFeature("Point");
96 //boost::shared_ptr<ModelAPI_Feature> aPart = aMgr->rootDocument()->addFeature("Part");
98 //aMgr->setCurrentDocument(aPart->data()->docRef("PartDocument")->value());
99 //boost::shared_ptr<ModelAPI_Feature> aPoint2 = aMgr->rootDocument()->addFeature("Point");
100 //aPoint2 = aMgr->rootDocument()->addFeature("Point");
102 //aPart = aMgr->rootDocument()->addFeature("Part");
106 //******************************************************
107 void XGUI_Workshop::initMenu()
109 if (isSalomeMode()) {
110 // Create only Undo, Redo commands
111 salomeConnector()->addEditCommand("UNDO_CMD",
112 tr("Undo"), tr("Undo last command"),
113 QIcon(":pictures/undo.png"),
114 false, this, SLOT(onUndo()),
116 salomeConnector()->addEditCommand("REDO_CMD",
117 tr("Redo"), tr("Redo last command"),
118 QIcon(":pictures/redo.png"),
119 false, this, SLOT(onRedo()),
121 salomeConnector()->addEditMenuSeparator();
124 XGUI_Workbench* aPage = myMainWindow->menuObject()->generalPage();
126 // File commands group
127 XGUI_MenuGroupPanel* aGroup = aPage->addGroup("Default");
129 XGUI_Command* aCommand;
131 aCommand = aGroup->addFeature("SAVE_CMD", tr("Save..."), tr("Save the document"),
132 QIcon(":pictures/save.png"), QKeySequence::Save);
133 aCommand->connectTo(this, SLOT(onSave()));
134 //aCommand->disable();
136 aCommand = aGroup->addFeature("UNDO_CMD", tr("Undo"), tr("Undo last command"),
137 QIcon(":pictures/undo.png"), QKeySequence::Undo);
138 aCommand->connectTo(this, SLOT(onUndo()));
140 aCommand = aGroup->addFeature("REDO_CMD", tr("Redo"), tr("Redo last command"),
141 QIcon(":pictures/redo.png"), QKeySequence::Redo);
142 aCommand->connectTo(this, SLOT(onRedo()));
144 aCommand = aGroup->addFeature("REBUILD_CMD", tr("Rebuild"), tr("Rebuild data objects"),
145 QIcon(":pictures/rebuild.png"));
147 aCommand = aGroup->addFeature("SAVEAS_CMD", tr("Save as..."), tr("Save the document into a file"),
148 QIcon(":pictures/save.png"));
149 aCommand->connectTo(this, SLOT(onSaveAs()));
150 //aCommand->disable();
152 aCommand = aGroup->addFeature("OPEN_CMD", tr("Open..."), tr("Open a new document"),
153 QIcon(":pictures/open.png"), QKeySequence::Open);
154 aCommand->connectTo(this, SLOT(onOpen()));
156 //aCommand = aGroup->addFeature("NEW_CMD", tr("New"), tr("Create a new document"),
157 // QIcon(":pictures/new.png"), QKeySequence::New);
158 //aCommand->connectTo(this, SLOT(onNew()));
160 aCommand = aGroup->addFeature("EXIT_CMD", tr("Exit"), tr("Exit application"),
161 QIcon(":pictures/close.png"), QKeySequence::Close);
162 aCommand->connectTo(this, SLOT(onExit()));
166 //******************************************************
167 XGUI_Workbench* XGUI_Workshop::addWorkbench(const QString& theName)
169 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
170 return aMenuBar->addWorkbench(theName);
173 //******************************************************
174 void XGUI_Workshop::processEvent(const Events_Message* theMessage)
176 static Events_ID aFeatureId = Events_Loop::loop()->eventByName("FeatureEvent");
177 if (theMessage->eventID() == aFeatureId) {
178 const Config_FeatureMessage* aFeatureMsg =
179 dynamic_cast<const Config_FeatureMessage*>(theMessage);
180 addFeature(aFeatureMsg);
183 const Config_PointerMessage* aPartSetMsg =
184 dynamic_cast<const Config_PointerMessage*>(theMessage);
186 ModuleBase_PropPanelOperation* anOperation =
187 (ModuleBase_PropPanelOperation*)(aPartSetMsg->pointer());
189 if (myOperationMgr->startOperation(anOperation)) {
190 if (anOperation->xmlRepresentation().isEmpty()) {
191 anOperation->commit();
192 updateCommandStatus();
199 qDebug() << "XGUI_Workshop::ProcessEvent: "
200 << "Catch message, but it can not be processed.";
205 //******************************************************
206 void XGUI_Workshop::onOperationStarted()
208 ModuleBase_PropPanelOperation* aOperation =
209 (ModuleBase_PropPanelOperation*)(myOperationMgr->currentOperation());
211 if(!aOperation->xmlRepresentation().isEmpty()) { //!< No need for property panel
212 connectWithOperation(aOperation);
213 QWidget* aPropWidget = myPropertyPanelDock->findChild<QWidget*>(XGUI::PROP_PANEL_WDG);
214 qDeleteAll(aPropWidget->children());
218 ModuleBase_WidgetFactory aFactory = ModuleBase_WidgetFactory(aOperation);
219 aFactory.createWidget(aPropWidget);
220 setPropertyPannelTitle(aOperation->description());
224 //******************************************************
225 void XGUI_Workshop::onOperationStopped(ModuleBase_Operation* theOperation)
227 ModuleBase_PropPanelOperation* aOperation =
228 (ModuleBase_PropPanelOperation*)(myOperationMgr->currentOperation());
230 if(aOperation->xmlRepresentation().isEmpty()) { //!< No need for property panel
231 updateCommandStatus();
234 updateCommandStatus();
237 myActionsMgr->restoreCommandState();
245 void XGUI_Workshop::addFeature(const Config_FeatureMessage* theMessage)
249 qDebug() << "XGUI_Workshop::addFeature: NULL message.";
253 //Find or create Workbench
254 QString aWchName = QString::fromStdString(theMessage->workbenchId());
255 if (isSalomeMode()) {
256 salomeConnector()->addFeature(aWchName,
257 QString::fromStdString(theMessage->id()),
258 QString::fromStdString(theMessage->text()),
259 QString::fromStdString(theMessage->tooltip()),
260 QIcon(theMessage->icon().c_str()),
262 SLOT(onFeatureTriggered()), QKeySequence());
264 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
265 XGUI_Workbench* aPage = aMenuBar->findWorkbench(aWchName);
267 aPage = addWorkbench(aWchName);
269 //Find or create Group
270 QString aGroupName = QString::fromStdString(theMessage->groupId());
271 XGUI_MenuGroupPanel* aGroup = aPage->findGroup(aGroupName);
273 aGroup = aPage->addGroup(aGroupName);
275 bool isUsePropPanel = theMessage->isUseInput();
277 XGUI_Command* aCommand = aGroup->addFeature(QString::fromStdString(theMessage->id()),
278 QString::fromStdString(theMessage->text()),
279 QString::fromStdString(theMessage->tooltip()),
280 QIcon(theMessage->icon().c_str()),
281 QKeySequence(), isUsePropPanel);
282 myActionsMgr->addCommand(aCommand);
283 myPartSetModule->featureCreated(aCommand);
288 * Makes a signal/slot connections between Property Panel
289 * and given operation. The given operation becomes a
290 * current operation and previous operation if exists
292 void XGUI_Workshop::connectWithOperation(ModuleBase_Operation* theOperation)
294 QPushButton* aOkBtn = myPropertyPanelDock->findChild<QPushButton*>(XGUI::PROP_PANEL_OK);
295 connect(aOkBtn, SIGNAL(clicked()), theOperation, SLOT(commit()));
296 QPushButton* aCancelBtn = myPropertyPanelDock->findChild<QPushButton*>(XGUI::PROP_PANEL_CANCEL);
297 connect(aCancelBtn, SIGNAL(clicked()), theOperation, SLOT(abort()));
299 QAction* aCommand = 0;
300 if (isSalomeMode()) {
301 aCommand = salomeConnector()->command(theOperation->operationId());
303 XGUI_MainMenu* aMenu = myMainWindow->menuObject();
304 aCommand = aMenu->feature(theOperation->operationId());
306 //Abort operation on uncheck the command
307 connect(aCommand, SIGNAL(triggered(bool)), theOperation, SLOT(setRunning(bool)));
311 * Saves document with given name.
313 void XGUI_Workshop::saveDocument(QString theName)
315 QApplication::restoreOverrideCursor();
316 boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
317 boost::shared_ptr<ModelAPI_Document> aDoc = aMgr->rootDocument();
318 aDoc->save(theName.toLatin1().constData());
319 QApplication::restoreOverrideCursor();
322 //******************************************************
323 void XGUI_Workshop::onExit()
325 boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
326 boost::shared_ptr<ModelAPI_Document> aDoc = aMgr->rootDocument();
327 if(aDoc->isModified()) {
328 int anAnswer = QMessageBox::question(
329 myMainWindow, tr("Save current file"),
330 tr("The document is modified, save before exit?"),
331 QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
332 if(anAnswer == QMessageBox::Save) {
334 } else if (anAnswer == QMessageBox::Cancel) {
341 //******************************************************
342 void XGUI_Workshop::onNew()
344 QApplication::setOverrideCursor(Qt::WaitCursor);
345 if (objectBrowser() == 0) {
347 mySelector->connectObjectBrowser(objectBrowser());
350 if (!isSalomeMode()) {
351 myMainWindow->showPythonConsole();
352 QMdiSubWindow* aWnd = myMainWindow->viewer()->createView();
353 aWnd->showMaximized();
354 updateCommandStatus();
356 QApplication::restoreOverrideCursor();
359 //******************************************************
360 void XGUI_Workshop::onOpen()
362 //save current file before close if modified
363 boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
364 boost::shared_ptr<ModelAPI_Document> aDoc = aMgr->rootDocument();
365 if(aDoc->isModified()) {
366 //TODO(sbh): re-launch the app?
367 int anAnswer = QMessageBox::question(
368 myMainWindow, tr("Save current file"),
369 tr("The document is modified, save before opening another?"),
370 QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
371 if(anAnswer == QMessageBox::Save) {
373 } else if (anAnswer == QMessageBox::Cancel) {
380 //show file dialog, check if readable and open
381 myCurrentFile = QFileDialog::getOpenFileName(mainWindow());
382 if(myCurrentFile.isEmpty())
384 QFileInfo aFileInfo(myCurrentFile);
385 if(!aFileInfo.exists() || !aFileInfo.isReadable()) {
386 QMessageBox::critical(myMainWindow, tr("Warning"), tr("Unable to open the file."));
390 QApplication::setOverrideCursor(Qt::WaitCursor);
391 aDoc->load(myCurrentFile.toLatin1().constData());
392 QApplication::restoreOverrideCursor();
393 updateCommandStatus();
396 //******************************************************
397 void XGUI_Workshop::onSave()
399 if(myCurrentFile.isEmpty()) {
403 saveDocument(myCurrentFile);
404 updateCommandStatus();
407 //******************************************************
408 void XGUI_Workshop::onSaveAs()
410 QString aTemp = myCurrentFile;
411 myCurrentFile = QFileDialog::getSaveFileName(mainWindow());
412 if(myCurrentFile.isEmpty()) {
413 myCurrentFile = aTemp;
416 QFileInfo aFileInfo(myCurrentFile);
417 if(aFileInfo.exists() && !aFileInfo.isWritable()) {
418 QMessageBox::critical(myMainWindow, tr("Warning"), tr("Unable to save the file."));
424 //******************************************************
425 void XGUI_Workshop::onUndo()
427 objectBrowser()->setCurrentIndex(QModelIndex());
428 boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
429 boost::shared_ptr<ModelAPI_Document> aDoc = aMgr->rootDocument();
431 updateCommandStatus();
434 //******************************************************
435 void XGUI_Workshop::onRedo()
437 objectBrowser()->setCurrentIndex(QModelIndex());
438 boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
439 boost::shared_ptr<ModelAPI_Document> aDoc = aMgr->rootDocument();
441 updateCommandStatus();
444 //******************************************************
445 XGUI_Module* XGUI_Workshop::loadModule(const QString& theModule)
447 QString libName = library(theModule);
448 if (libName.isEmpty()) {
450 qPrintable( tr( "Information about module \"%1\" doesn't exist." ).arg( theModule ) ));
455 CREATE_FUNC crtInst = 0;
459 HINSTANCE modLib = ::LoadLibrary((LPTSTR) qPrintable(libName));
463 FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
464 0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
465 QString aMsg((char*) &lpMsgBuf);
466 err = QString("Failed to load %1. %2").arg(libName).arg(aMsg);
467 ::LocalFree(lpMsgBuf);
469 crtInst = (CREATE_FUNC) ::GetProcAddress(modLib, CREATE_MODULE);
473 FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM
474 | FORMAT_MESSAGE_IGNORE_INSERTS,
475 0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
476 QString aMsg((char*) &lpMsgBuf);
477 err = QString("Failed to find %1 function. %2").arg( CREATE_MODULE).arg(aMsg);
478 ::LocalFree(lpMsgBuf);
482 void* modLib = dlopen( libName.toLatin1(), RTLD_LAZY );
484 err = QString( "Can not load library %1. %2" ).arg( libName ).arg( dlerror() );
487 crtInst = (CREATE_FUNC)dlsym( modLib, CREATE_MODULE );
489 err = QString( "Failed to find function %1. %2" ).arg( CREATE_MODULE ).arg( dlerror() );
493 XGUI_Module* aModule = crtInst ? crtInst(this) : 0;
495 if (!err.isEmpty()) {
496 if (mainWindow() && mainWindow()->isVisible())
497 QMessageBox::warning(mainWindow(), tr("Error"), err);
499 qWarning( qPrintable( err ));
504 //******************************************************
505 bool XGUI_Workshop::activateModule()
507 myPartSetModule = loadModule("PartSet");
508 if (!myPartSetModule)
510 myPartSetModule->createFeatures();
514 //******************************************************
515 void XGUI_Workshop::updateCommandStatus()
517 if (isSalomeMode()) // TODO: update commands in SALOME
519 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
521 QList<XGUI_Command*> aCommands = aMenuBar->features();
522 QList<XGUI_Command*>::const_iterator aIt;
524 boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
525 if (aMgr->hasRootDocument()) {
526 XGUI_Command* aUndoCmd;
527 XGUI_Command* aRedoCmd;
528 for (aIt = aCommands.constBegin(); aIt != aCommands.constEnd(); ++aIt) {
529 if ((*aIt)->id() == "UNDO_CMD")
531 else if ((*aIt)->id() == "REDO_CMD")
533 else // Enable all commands
536 boost::shared_ptr<ModelAPI_Document> aDoc = aMgr->rootDocument();
537 aUndoCmd->setEnabled(aDoc->canUndo());
538 aRedoCmd->setEnabled(aDoc->canRedo());
540 for (aIt = aCommands.constBegin(); aIt != aCommands.constEnd(); ++aIt) {
541 if ((*aIt)->id() == "NEW_CMD")
543 else if ((*aIt)->id() == "EXIT_CMD")
551 //******************************************************
552 QDockWidget* XGUI_Workshop::createObjectBrowser(QWidget* theParent)
554 QDockWidget* aObjDock = new QDockWidget(theParent);
555 aObjDock->setAllowedAreas(Qt::LeftDockWidgetArea | Qt::RightDockWidgetArea);
556 aObjDock->setWindowTitle(tr("Object browser"));
557 myObjectBrowser = new XGUI_ObjectsBrowser(aObjDock);
558 aObjDock->setWidget(myObjectBrowser);
562 //******************************************************
563 QDockWidget* XGUI_Workshop::createPropertyPanel(QWidget* theParent)
565 QDockWidget* aPropPanel = new QDockWidget(theParent);
566 aPropPanel->setWindowTitle(tr("Property Panel"));
567 QAction* aViewAct = aPropPanel->toggleViewAction();
568 aPropPanel->setObjectName(XGUI::PROP_PANEL);
570 QWidget* aContent = new QWidget(aPropPanel);
571 QVBoxLayout* aMainLay = new QVBoxLayout(aContent);
572 aMainLay->setContentsMargins(3, 3, 3, 3);
573 aPropPanel->setWidget(aContent);
575 QFrame* aFrm = new QFrame(aContent);
576 aFrm->setFrameStyle(QFrame::Sunken);
577 aFrm->setFrameShape(QFrame::Panel);
578 QHBoxLayout* aBtnLay = new QHBoxLayout(aFrm);
579 aBtnLay->setContentsMargins(0, 0, 0, 0);
580 aMainLay->addWidget(aFrm);
582 QPushButton* aBtn = new QPushButton(QIcon(":pictures/button_help.png"), "", aFrm);
584 aBtnLay->addWidget(aBtn);
585 aBtnLay->addStretch(1);
586 aBtn = new QPushButton(QIcon(":pictures/button_ok.png"), "", aFrm);
587 aBtn->setObjectName(XGUI::PROP_PANEL_OK);
589 aBtnLay->addWidget(aBtn);
590 aBtn = new QPushButton(QIcon(":pictures/button_cancel.png"), "", aFrm);
591 aBtn->setObjectName(XGUI::PROP_PANEL_CANCEL);
593 aBtnLay->addWidget(aBtn);
595 QWidget* aCustomWidget = new QWidget(aContent);
596 aCustomWidget->setObjectName(XGUI::PROP_PANEL_WDG);
597 aMainLay->addWidget(aCustomWidget);
598 aMainLay->addStretch(1);
603 //******************************************************
604 void XGUI_Workshop::setPropertyPannelTitle(const QString& theTitle)
606 myPropertyPanelDock->setWindowTitle(theTitle);
609 //******************************************************
611 * Creates dock widgets, places them in corresponding area
612 * and tabifies if necessary.
614 void XGUI_Workshop::createDockWidgets()
616 QMainWindow* aDesktop = isSalomeMode()? salomeConnector()->desktop() :
618 QDockWidget* aObjDock = createObjectBrowser(aDesktop);
619 aDesktop->addDockWidget(Qt::LeftDockWidgetArea, aObjDock);
620 myPropertyPanelDock = createPropertyPanel(aDesktop);
621 aDesktop->addDockWidget(Qt::LeftDockWidgetArea, myPropertyPanelDock);
622 hidePropertyPanel(); //<! Invisible by default
624 aDesktop->tabifyDockWidget(aObjDock, myPropertyPanelDock);
627 //******************************************************
628 void XGUI_Workshop::showPropertyPanel()
630 QAction* aViewAct = myPropertyPanelDock->toggleViewAction();
631 //<! Restore ability to close panel from the window's menu
632 aViewAct->setEnabled(true);
633 myPropertyPanelDock->show();
634 myPropertyPanelDock->raise();
637 //******************************************************
638 void XGUI_Workshop::hidePropertyPanel()
640 QAction* aViewAct = myPropertyPanelDock->toggleViewAction();
641 //<! Do not allow to show empty property panel
642 aViewAct->setEnabled(false);
643 myPropertyPanelDock->hide();
646 //******************************************************
647 void XGUI_Workshop::showObjectBrowser()
649 myObjectBrowser->parentWidget()->show();
652 //******************************************************
653 void XGUI_Workshop::hideObjectBrowser()
655 myObjectBrowser->parentWidget()->hide();
658 //******************************************************
659 void XGUI_Workshop::onFeatureTriggered()
661 QAction* aCmd = dynamic_cast<QAction*>(sender());
663 QString aId = salomeConnector()->commandId(aCmd);
665 myPartSetModule->launchOperation(aId);
669 //******************************************************
670 XGUI_Displayer* XGUI_Workshop::displayer() const
672 // In SALOME viewer is accessible only when module is initialized
673 // and in SALOME mode myDisplayer object has to be created later (on demand)
675 XGUI_Workshop* that = (XGUI_Workshop*)this;
676 that->myDisplayer = isSalomeMode() ?
677 new XGUI_Displayer(salomeConnector()->AISContext()):
678 new XGUI_Displayer(myMainWindow->viewer()->AISContext());