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 "XGUI_WidgetFactory.h"
12 #include "XGUI_SelectionMgr.h"
13 #include "XGUI_ObjectsBrowser.h"
14 #include "XGUI_Displayer.h"
15 #include "XGUI_OperationMgr.h"
17 #include <ModelAPI_PluginManager.h>
18 #include <ModelAPI_Feature.h>
19 #include <ModelAPI_Data.h>
20 #include <ModelAPI_AttributeDocRef.h>
22 #include <Event_Loop.h>
23 #include <ModuleBase_PropPanelOperation.h>
24 #include <ModuleBase_Operation.h>
25 #include <Config_FeatureMessage.h>
26 #include <Config_PointerMessage.h>
28 #include <QApplication>
29 #include <QFileDialog>
30 #include <QMessageBox>
31 #include <QMdiSubWindow>
32 #include <QPushButton>
33 #include <QDockWidget>
45 XGUI_Workshop::XGUI_Workshop()
49 myMainWindow = new XGUI_MainWindow();
50 mySelector = new XGUI_SelectionMgr(this);
51 myDisplayer = new XGUI_Displayer(myMainWindow->viewer());
52 myOperationMgr = new XGUI_OperationMgr(this);
53 connect(myOperationMgr, SIGNAL(operationStarted()), this, SLOT(onOperationStarted()));
54 connect(myOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)),
55 this, SLOT(onOperationStopped(ModuleBase_Operation*)));
58 //******************************************************
59 XGUI_Workshop::~XGUI_Workshop(void)
63 //******************************************************
64 void XGUI_Workshop::startApplication()
67 //Initialize event listening
68 Event_Loop* aLoop = Event_Loop::loop();
69 //TODO(sbh): Implement static method to extract event id [SEID]
70 Event_ID aFeatureId = aLoop->eventByName("FeatureEvent");
71 aLoop->registerListener(this, aFeatureId);
72 Event_ID aPartSetId = aLoop->eventByName("PartSetModuleEvent");
73 aLoop->registerListener(this, aPartSetId);
77 updateCommandStatus();
79 // Testing of document creation
80 //boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
81 //boost::shared_ptr<ModelAPI_Feature> aPoint1 = aMgr->rootDocument()->addFeature("Point");
82 //boost::shared_ptr<ModelAPI_Feature> aPart = aMgr->rootDocument()->addFeature("Part");
84 //aMgr->setCurrentDocument(aPart->data()->docRef("PartDocument")->value());
85 //boost::shared_ptr<ModelAPI_Feature> aPoint2 = aMgr->rootDocument()->addFeature("Point");
86 //aPoint2 = aMgr->rootDocument()->addFeature("Point");
88 //aPart = aMgr->rootDocument()->addFeature("Part");
92 //******************************************************
93 void XGUI_Workshop::initMenu()
95 XGUI_Workbench* aPage = myMainWindow->menuObject()->generalPage();
97 // File commands group
98 XGUI_MenuGroupPanel* aGroup = aPage->addGroup("Default");
100 XGUI_Command* aCommand;
102 aCommand = aGroup->addFeature("SAVE_CMD", tr("Save..."), tr("Save the document"),
103 QIcon(":pictures/save.png"), QKeySequence::Save);
104 aCommand->connectTo(this, SLOT(onSave()));
105 //aCommand->disable();
107 aCommand = aGroup->addFeature("UNDO_CMD", tr("Undo"), tr("Undo last command"),
108 QIcon(":pictures/undo.png"), QKeySequence::Undo);
109 aCommand->connectTo(this, SLOT(onUndo()));
111 aCommand = aGroup->addFeature("REDO_CMD", tr("Redo"), tr("Redo last command"),
112 QIcon(":pictures/redo.png"), QKeySequence::Redo);
113 aCommand->connectTo(this, SLOT(onRedo()));
115 aCommand = aGroup->addFeature("REBUILD_CMD", tr("Rebuild"), tr("Rebuild data objects"),
116 QIcon(":pictures/rebuild.png"));
118 aCommand = aGroup->addFeature("SAVEAS_CMD", tr("Save as..."), tr("Save the document into a file"),
119 QIcon(":pictures/save.png"));
120 aCommand->connectTo(this, SLOT(onSaveAs()));
121 //aCommand->disable();
123 aCommand = aGroup->addFeature("OPEN_CMD", tr("Open..."), tr("Open a new document"),
124 QIcon(":pictures/open.png"), QKeySequence::Open);
125 aCommand->connectTo(this, SLOT(onOpen()));
127 //aCommand = aGroup->addFeature("NEW_CMD", tr("New"), tr("Create a new document"),
128 // QIcon(":pictures/new.png"), QKeySequence::New);
129 //aCommand->connectTo(this, SLOT(onNew()));
131 aCommand = aGroup->addFeature("EXIT_CMD", tr("Exit"), tr("Exit application"),
132 QIcon(":pictures/close.png"), QKeySequence::Close);
133 aCommand->connectTo(this, SLOT(onExit()));
137 //******************************************************
138 XGUI_Workbench* XGUI_Workshop::addWorkbench(const QString& theName)
140 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
141 return aMenuBar->addWorkbench(theName);
144 //******************************************************
145 void XGUI_Workshop::processEvent(const Event_Message* theMessage)
147 static Event_ID aFeatureId = Event_Loop::loop()->eventByName("FeatureEvent");
148 if (theMessage->eventID() == aFeatureId) {
149 const Config_FeatureMessage* aFeatureMsg =
150 dynamic_cast<const Config_FeatureMessage*>(theMessage);
151 addFeature(aFeatureMsg);
154 const Config_PointerMessage* aPartSetMsg =
155 dynamic_cast<const Config_PointerMessage*>(theMessage);
157 ModuleBase_PropPanelOperation* anOperation =
158 (ModuleBase_PropPanelOperation*)(aPartSetMsg->pointer());
160 if (myOperationMgr->startOperation(anOperation)) {
161 if (anOperation->isPerformedImmediately()) {
162 anOperation->commit();
163 updateCommandStatus();
170 qDebug() << "XGUI_Workshop::ProcessEvent: "
171 << "Catch message, but it can not be processed.";
176 //******************************************************
177 void XGUI_Workshop::onOperationStarted()
179 ModuleBase_PropPanelOperation* aOperation =
180 (ModuleBase_PropPanelOperation*)(myOperationMgr->currentOperation());
182 if(aOperation->xmlRepresentation().isEmpty()) { //!< No need for property panel
184 connectWithOperation(aOperation);
185 QWidget* aPropWidget = myMainWindow->findChild<QWidget*>(XGUI::PROP_PANEL_WDG);
186 qDeleteAll(aPropWidget->children());
188 myMainWindow->showPropertyPanel();
190 XGUI_WidgetFactory aFactory = XGUI_WidgetFactory(aOperation);
191 aFactory.createWidget(aPropWidget);
192 myMainWindow->setPropertyPannelTitle(aOperation->description());
196 //******************************************************
197 void XGUI_Workshop::onOperationStopped(ModuleBase_Operation* theOperation)
199 myMainWindow->hidePropertyPanel();
200 updateCommandStatus();
202 XGUI_MainMenu* aMenu = myMainWindow->menuObject();
203 aMenu->restoreCommandState();
209 void XGUI_Workshop::addFeature(const Config_FeatureMessage* theMessage)
213 qDebug() << "XGUI_Workshop::addFeature: NULL message.";
217 //Find or create Workbench
218 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
219 QString aWchName = QString::fromStdString(theMessage->workbenchId());
220 XGUI_Workbench* aPage = aMenuBar->findWorkbench(aWchName);
222 aPage = addWorkbench(aWchName);
224 //Find or create Group
225 QString aGroupName = QString::fromStdString(theMessage->groupId());
226 XGUI_MenuGroupPanel* aGroup = aPage->findGroup(aGroupName);
228 aGroup = aPage->addGroup(aGroupName);
230 bool isUsePropPanel = theMessage->isUseInput();
232 XGUI_Command* aCommand = aGroup->addFeature(QString::fromStdString(theMessage->id()),
233 QString::fromStdString(theMessage->text()),
234 QString::fromStdString(theMessage->tooltip()),
235 QIcon(theMessage->icon().c_str()),
236 QKeySequence(), isUsePropPanel);
238 connect(aCommand, SIGNAL(toggled(bool)),
239 myMainWindow->menuObject(), SLOT(onFeatureChecked(bool)));
240 myPartSetModule->featureCreated(aCommand);
244 * Makes a signal/slot connections between Property Panel
245 * and given operation. The given operation becomes a
246 * current operation and previous operation if exists
248 void XGUI_Workshop::connectWithOperation(ModuleBase_Operation* theOperation)
250 QDockWidget* aPanel = myMainWindow->findChild<QDockWidget*>(XGUI::PROP_PANEL);
251 QPushButton* aOkBtn = aPanel->findChild<QPushButton*>(XGUI::PROP_PANEL_OK);
252 connect(aOkBtn, SIGNAL(clicked()), theOperation, SLOT(commit()));
253 QPushButton* aCancelBtn = aPanel->findChild<QPushButton*>(XGUI::PROP_PANEL_CANCEL);
254 connect(aCancelBtn, SIGNAL(clicked()), theOperation, SLOT(abort()));
256 XGUI_MainMenu* aMenu = myMainWindow->menuObject();
257 XGUI_Command* aCommand = aMenu->feature(theOperation->operationId());
258 //Abort operation on uncheck the command
259 connect(aCommand, SIGNAL(toggled(bool)), theOperation, SLOT(setRunning(bool)));
263 //******************************************************
264 void XGUI_Workshop::onExit()
269 //******************************************************
270 void XGUI_Workshop::onNew()
272 QApplication::setOverrideCursor(Qt::WaitCursor);
273 if (myMainWindow->objectBrowser() == 0) {
274 myMainWindow->createDockWidgets();
275 mySelector->connectObjectBrowser(myMainWindow->objectBrowser());
277 myMainWindow->showObjectBrowser();
278 myMainWindow->showPythonConsole();
279 QMdiSubWindow* aWnd = myMainWindow->viewer()->createView();
280 aWnd->showMaximized();
281 updateCommandStatus();
282 QApplication::restoreOverrideCursor();
285 //******************************************************
286 void XGUI_Workshop::onOpen()
288 //QString aFileName = QFileDialog::getOpenFileName(mainWindow());
289 updateCommandStatus();
292 //******************************************************
293 void XGUI_Workshop::onSave()
295 updateCommandStatus();
298 //******************************************************
299 void XGUI_Workshop::onSaveAs()
301 //QString aFileName = QFileDialog::getSaveFileName(mainWindow());
302 updateCommandStatus();
305 //******************************************************
306 void XGUI_Workshop::onUndo()
308 myMainWindow->objectBrowser()->setCurrentIndex(QModelIndex());
309 boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
310 boost::shared_ptr<ModelAPI_Document> aDoc = aMgr->rootDocument();
312 updateCommandStatus();
315 //******************************************************
316 void XGUI_Workshop::onRedo()
318 myMainWindow->objectBrowser()->setCurrentIndex(QModelIndex());
319 boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
320 boost::shared_ptr<ModelAPI_Document> aDoc = aMgr->rootDocument();
322 updateCommandStatus();
325 //******************************************************
326 XGUI_Module* XGUI_Workshop::loadModule(const QString& theModule)
328 QString libName = library(theModule);
329 if (libName.isEmpty()) {
331 qPrintable( tr( "Information about module \"%1\" doesn't exist." ).arg( theModule ) ));
336 CREATE_FUNC crtInst = 0;
340 HINSTANCE modLib = ::LoadLibrary((LPTSTR) qPrintable(libName));
344 FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
345 0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
346 QString aMsg((char*) &lpMsgBuf);
347 err = QString("Failed to load %1. %2").arg(libName).arg(aMsg);
348 ::LocalFree(lpMsgBuf);
350 crtInst = (CREATE_FUNC) ::GetProcAddress(modLib, CREATE_MODULE);
354 FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM
355 | FORMAT_MESSAGE_IGNORE_INSERTS,
356 0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
357 QString aMsg((char*) &lpMsgBuf);
358 err = QString("Failed to find %1 function. %2").arg( CREATE_MODULE).arg(aMsg);
359 ::LocalFree(lpMsgBuf);
363 void* modLib = dlopen( libName.toLatin1(), RTLD_LAZY );
365 err = QString( "Can not load library %1. %2" ).arg( libName ).arg( dlerror() );
368 crtInst = (CREATE_FUNC)dlsym( modLib, CREATE_MODULE );
370 err = QString( "Failed to find function %1. %2" ).arg( CREATE_MODULE ).arg( dlerror() );
374 XGUI_Module* aModule = crtInst ? crtInst(this) : 0;
376 if (!err.isEmpty()) {
377 if (mainWindow() && mainWindow()->isVisible())
378 QMessageBox::warning(mainWindow(), tr("Error"), err);
380 qWarning( qPrintable( err ));
385 //******************************************************
386 bool XGUI_Workshop::activateModule()
388 myPartSetModule = loadModule("PartSet");
389 if (!myPartSetModule)
391 myPartSetModule->createFeatures();
395 //******************************************************
396 void XGUI_Workshop::updateCommandStatus()
398 XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
400 QList<XGUI_Command*> aCommands = aMenuBar->features();
401 QList<XGUI_Command*>::const_iterator aIt;
403 boost::shared_ptr<ModelAPI_PluginManager> aMgr = ModelAPI_PluginManager::get();
404 if (aMgr->hasRootDocument()) {
405 XGUI_Command* aUndoCmd;
406 XGUI_Command* aRedoCmd;
407 for (aIt = aCommands.constBegin(); aIt != aCommands.constEnd(); ++aIt) {
408 if ((*aIt)->id() == "UNDO_CMD")
410 else if ((*aIt)->id() == "REDO_CMD")
412 else // Enable all commands
415 boost::shared_ptr<ModelAPI_Document> aDoc = aMgr->rootDocument();
416 aUndoCmd->setEnabled(aDoc->canUndo());
417 aRedoCmd->setEnabled(aDoc->canRedo());
419 for (aIt = aCommands.constBegin(); aIt != aCommands.constEnd(); ++aIt) {
420 if ((*aIt)->id() == "NEW_CMD")
422 else if ((*aIt)->id() == "EXIT_CMD")