0029748: Samples - Inspector tool - use recently opened files in TInspectorEXE
[occt.git] / tools / DFBrowser / DFBrowser_Window.cxx
1 // Created on: 2017-06-16
2 // Created by: Natalia ERMOLAEVA
3 // Copyright (c) 2017 OPEN CASCADE SAS
4 //
5 // This file is part of Open CASCADE Technology software library.
6 //
7 // This library is free software; you can redistribute it and/or modify it under
8 // the terms of the GNU Lesser General Public License version 2.1 as published
9 // by the Free Software Foundation, with special exception defined in the file
10 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
11 // distribution for complete text of the license and disclaimer of any warranty.
12 //
13 // Alternatively, this file may be used under the terms of Open CASCADE
14 // commercial license or contractual agreement. 
15
16 #include <inspector/DFBrowser_Window.hxx>
17
18 #include <AIS_InteractiveContext.hxx>
19 #include <AIS_InteractiveObject.hxx>
20 #include <AIS_ListOfInteractive.hxx>
21
22 #include <CDF_Session.hxx>
23
24 #include <inspector/DFBrowserPane_AttributePaneAPI.hxx>
25
26 #include <inspector/DFBrowser_AttributePaneStack.hxx>
27 #include <inspector/DFBrowser_AttributePaneType.hxx>
28 #include <inspector/DFBrowser_DumpView.hxx>
29 #include <inspector/DFBrowser_Item.hxx>
30 #include <inspector/DFBrowser_ItemApplication.hxx>
31 #include <inspector/DFBrowser_Module.hxx>
32 #include <inspector/DFBrowser_OpenApplication.hxx>
33 #include <inspector/DFBrowser_PropertyPanel.hxx>
34 #include <inspector/DFBrowser_SearchLine.hxx>
35 #include <inspector/DFBrowser_SearchView.hxx>
36 #include <inspector/DFBrowser_Thread.hxx>
37 #include <inspector/DFBrowser_ThreadItemSearch.hxx>
38 #include <inspector/DFBrowser_Tools.hxx>
39 #include <inspector/DFBrowser_TreeLevelLine.hxx>
40 #include <inspector/DFBrowser_TreeLevelView.hxx>
41 #include <inspector/DFBrowser_TreeModel.hxx>
42
43 #include <inspector/DFBrowserPane_AttributePaneSelector.hxx>
44 #include <inspector/DFBrowserPane_SelectionKind.hxx>
45 #include <inspector/DFBrowserPane_Tools.hxx>
46
47 #include <inspector/TreeModel_ContextMenu.hxx>
48 #include <inspector/TreeModel_Tools.hxx>
49
50 #include <inspector/ViewControl_TreeView.hxx>
51
52 #include <inspector/View_Tools.hxx>
53
54 #include <OSD_Directory.hxx>
55 #include <OSD_Environment.hxx>
56 #include <OSD_Protection.hxx>
57
58 #include <inspector/View_Displayer.hxx>
59 #include <inspector/View_ToolBar.hxx>
60 #include <inspector/View_Viewer.hxx>
61 #include <inspector/View_Widget.hxx>
62 #include <inspector/View_Window.hxx>
63
64 #include <TDF_Tool.hxx>
65 #include <inspector/ViewControl_MessageDialog.hxx>
66 #include <inspector/ViewControl_Tools.hxx>
67
68 #include <Standard_WarningsDisable.hxx>
69 #include <QAction>
70 #include <QApplication>
71 #include <QComboBox>
72 #include <QDir>
73 #include <QDockWidget>
74 #include <QGridLayout>
75 #include <QList>
76 #include <QMainWindow>
77 #include <QItemSelectionModel>
78 #include <QTabWidget>
79 #include <QToolBar>
80 #include <QTreeView>
81 #include <QMenu>
82 #include <QMessageBox>
83 #include <QStatusBar>
84 #include <QWidget>
85 #if QT_VERSION < 0x050000
86 #include <QWindowsStyle>
87 #else
88 #include <QStyleFactory>
89 #endif
90 #include <Standard_WarningsRestore.hxx>
91
92 const int DFBROWSER_DEFAULT_WIDTH = 650;
93 const int DFBROWSER_DEFAULT_HEIGHT = 350;
94 const int DFBROWSER_DEFAULT_TREE_VIEW_WIDTH = 325;
95 const int DFBROWSER_DEFAULT_TREE_VIEW_HEIGHT = 500;
96 const int DFBROWSER_DEFAULT_VIEW_WIDTH = 400;
97 const int DFBROWSER_DEFAULT_VIEW_HEIGHT = 300;
98
99 const int DFBROWSER_DEFAULT_POSITION_X = 200;
100 const int DFBROWSER_DEFAULT_POSITION_Y = 60;
101
102 const int DEFAULT_PROPERTY_PANEL_HEIGHT = 200;
103 const int DEFAULT_BROWSER_HEIGHT = 800;
104
105 // =======================================================================
106 // function : Constructor
107 // purpose :
108 // =======================================================================
109 DFBrowser_Window::DFBrowser_Window()
110 : myModule (0), myParent (0), myPropertyPanel (0), myExportToShapeViewDialog (0)
111 {
112   myMainWindow = new QMainWindow (0);
113
114   // tree view
115   myTreeView = new ViewControl_TreeView (myMainWindow);
116   myTreeView->setContextMenuPolicy (Qt::CustomContextMenu);
117   connect (myTreeView, SIGNAL (customContextMenuRequested (const QPoint&)),
118            this, SLOT (onTreeViewContextMenuRequested (const QPoint&)));
119   new TreeModel_ContextMenu (myTreeView);
120   ((ViewControl_TreeView*)myTreeView)->SetPredefinedSize (QSize (DFBROWSER_DEFAULT_TREE_VIEW_WIDTH,
121                                                                  DFBROWSER_DEFAULT_TREE_VIEW_HEIGHT));
122   myTreeView->setHeaderHidden (true);
123   myTreeView->setSortingEnabled (Standard_False);
124   myMainWindow->setCentralWidget (myTreeView);
125
126 #if QT_VERSION < 0x050000
127   myTreeView->setStyle (new QWindowsStyle);
128 #else
129   myTreeView->setStyle (QStyleFactory::create("Windows"));
130 #endif
131
132   myTreeLevelLine = new DFBrowser_TreeLevelLine (myMainWindow);
133   connect (myTreeLevelLine->GetSearchLine(), SIGNAL (searchActivated()), this, SLOT (onSearchActivated()));
134   connect (myTreeLevelLine, SIGNAL (indexSelected (const QModelIndex&)),
135           this, SLOT (onTreeLevelLineSelected (const QModelIndex&)));
136   connect (myTreeLevelLine, SIGNAL (updateClicked()), this, SLOT (onUpdateClicked()));
137
138   QDockWidget* aTreeLineDockWidget = new QDockWidget (tr ("Tree Level Line"), myMainWindow);
139   aTreeLineDockWidget->setObjectName (aTreeLineDockWidget->windowTitle());
140   aTreeLineDockWidget->setTitleBarWidget (new QWidget(myMainWindow));
141   aTreeLineDockWidget->setWidget (myTreeLevelLine->GetControl());
142   myMainWindow->addDockWidget (Qt::TopDockWidgetArea, aTreeLineDockWidget);
143
144   // property panel
145   myPropertyPanel = new DFBrowser_PropertyPanel (myMainWindow);
146   DFBrowser_AttributePaneStack* anAttributePaneStack = myPropertyPanel->GetAttributesStack();
147   anAttributePaneStack->GetSearchView()->SetSearchLine (myTreeLevelLine->GetSearchLine());
148
149   connect (anAttributePaneStack->GetPaneSelector(),
150            SIGNAL (tableSelectionChanged (const QItemSelection&, const QItemSelection&, QItemSelectionModel*)),
151            this, SLOT (onPaneSelectionChanged (const QItemSelection&, const QItemSelection&, QItemSelectionModel*)));
152
153   DFBrowser_SearchView* aSearchView = anAttributePaneStack->GetSearchView();
154   connect (aSearchView, SIGNAL (pathSelected (const QStringList&, const QString&)),
155            this, SLOT (onSearchPathSelected (const QStringList&, const QString&)));
156   connect (aSearchView, SIGNAL (pathDoubleClicked (const QStringList&, const QString&)),
157            this, SLOT (onSearchPathDoubleClicked (const QStringList&, const QString&)));
158
159   DFBrowser_TreeLevelView* aLevelView = anAttributePaneStack->GetTreeLevelView();
160   connect (aLevelView, SIGNAL (indexSelected (const QModelIndex&)), this, SLOT (onLevelSelected (const QModelIndex&)));
161   connect (aLevelView, SIGNAL (indexDoubleClicked (const QModelIndex&)),
162            this, SLOT (onLevelDoubleClicked (const QModelIndex&)));
163
164   QDockWidget* aPropertyPanelWidget = new QDockWidget (tr ("PropertyPanel"), myMainWindow);
165   aPropertyPanelWidget->setObjectName (aPropertyPanelWidget->windowTitle());
166   aPropertyPanelWidget->setTitleBarWidget (new QWidget(myMainWindow));
167   aPropertyPanelWidget->setWidget (myPropertyPanel->GetControl());
168   myMainWindow->addDockWidget (Qt::RightDockWidgetArea, aPropertyPanelWidget);
169
170   // dump view window
171   QWidget* aDumpWidget = new QWidget(myMainWindow);
172   QVBoxLayout* aDumpLay = new QVBoxLayout(aDumpWidget);
173   aDumpLay->setMargin(0);
174   myDumpView = new DFBrowser_DumpView(aDumpWidget);
175   aDumpLay->addWidget(myDumpView->GetControl());
176   QDockWidget* aDumpDockWidget = new QDockWidget(tr("Dump"), myMainWindow);
177   aDumpDockWidget->setObjectName(aDumpDockWidget->windowTitle());
178
179   aDumpDockWidget->setWidget(aDumpWidget);
180   myMainWindow->addDockWidget(Qt::RightDockWidgetArea, aDumpDockWidget);
181
182   // view
183   myViewWindow = new View_Window (myMainWindow);
184   myViewWindow->GetView()->SetPredefinedSize (DFBROWSER_DEFAULT_VIEW_WIDTH, DFBROWSER_DEFAULT_VIEW_HEIGHT);
185
186   QDockWidget* aViewDockWidget = new QDockWidget (tr ("View"), myMainWindow);
187   aViewDockWidget->setObjectName (aViewDockWidget->windowTitle());
188   aViewDockWidget->setTitleBarWidget (myViewWindow->GetViewToolBar()->GetControl());
189   aViewDockWidget->setWidget (myViewWindow);
190   myMainWindow->addDockWidget (Qt::RightDockWidgetArea, aViewDockWidget);
191
192   QColor aHColor (229, 243, 255);
193   myViewWindow->GetDisplayer()->SetAttributeColor (Quantity_Color(aHColor.red() / 255., aHColor.green() / 255.,
194                                                    aHColor.blue() / 255., Quantity_TOC_RGB), View_PresentationType_Additional);
195
196   myMainWindow->splitDockWidget(aPropertyPanelWidget, aViewDockWidget, Qt::Vertical);
197   myMainWindow->tabifyDockWidget (aDumpDockWidget, aViewDockWidget);
198
199   myTreeView->resize (DFBROWSER_DEFAULT_TREE_VIEW_WIDTH, DFBROWSER_DEFAULT_TREE_VIEW_HEIGHT);
200
201   myThread = new DFBrowser_Thread (this);
202 }
203
204 // =======================================================================
205 // function : Destructor
206 // purpose :
207 // =======================================================================
208 DFBrowser_Window::~DFBrowser_Window()
209 {
210   delete myModule;
211 }
212
213 // =======================================================================
214 // function : SetParent
215 // purpose :
216 // =======================================================================
217 void DFBrowser_Window::SetParent (void* theParent)
218 {
219   myParent = (QWidget*)theParent;
220   if (myParent)
221   {
222     QLayout* aLayout = myParent->layout();
223     if (aLayout)
224       aLayout->addWidget (GetMainWindow());
225
226     if (!myOpenedFileName.isEmpty())
227       myParent->setObjectName(myOpenedFileName);
228   }
229 }
230
231 // =======================================================================
232 // function : FillActionsMenu
233 // purpose :
234 // =======================================================================
235 void DFBrowser_Window::FillActionsMenu (void* theMenu)
236 {
237   QMenu* aMenu = (QMenu*)theMenu;
238   QList<QDockWidget*> aDockwidgets = myMainWindow->findChildren<QDockWidget*>();
239   for (QList<QDockWidget*>::iterator it = aDockwidgets.begin(); it != aDockwidgets.end(); ++it)
240   {
241     QDockWidget* aDockWidget = *it;
242     if (aDockWidget->parentWidget() == myMainWindow)
243       aMenu->addAction (aDockWidget->toggleViewAction());
244   }
245 }
246
247 // =======================================================================
248 // function : GetPreferences
249 // purpose :
250 // =======================================================================
251 void DFBrowser_Window::GetPreferences (TInspectorAPI_PreferencesDataMap& theItem)
252 {
253   theItem.Bind ("geometry",  TreeModel_Tools::ToString (myMainWindow->saveState()).toStdString().c_str());
254
255   QMap<QString, QString> anItems;
256   TreeModel_Tools::SaveState (myTreeView, anItems);
257   View_Tools::SaveState(myViewWindow, anItems);
258
259   for (QMap<QString, QString>::const_iterator anItemsIt = anItems.begin(); anItemsIt != anItems.end(); anItemsIt++)
260     theItem.Bind (anItemsIt.key().toStdString().c_str(), anItemsIt.value().toStdString().c_str());
261 }
262
263 // =======================================================================
264 // function : SetPreferences
265 // purpose :
266 // =======================================================================
267 void DFBrowser_Window::SetPreferences (const TInspectorAPI_PreferencesDataMap& theItem)
268 {
269   if (theItem.IsEmpty())
270   {
271     TreeModel_Tools::SetDefaultHeaderSections (myTreeView);
272     return;
273   }
274
275   for (TInspectorAPI_IteratorOfPreferencesDataMap anItemIt (theItem); anItemIt.More(); anItemIt.Next())
276   {
277     if (anItemIt.Key().IsEqual ("geometry"))
278       myMainWindow->restoreState (TreeModel_Tools::ToByteArray (anItemIt.Value().ToCString()));
279     else if (TreeModel_Tools::RestoreState (myTreeView, anItemIt.Key().ToCString(), anItemIt.Value().ToCString()))
280       continue;
281     else if (View_Tools::RestoreState(myViewWindow, anItemIt.Key().ToCString(), anItemIt.Value().ToCString()))
282       continue;
283   }
284 }
285
286 // =======================================================================
287 // function : UpdateContent
288 // purpose :
289 // =======================================================================
290 void DFBrowser_Window::UpdateContent()
291 {
292   TCollection_AsciiString aName = "TKDFBrowser";
293
294   if (myParameters->FindParameters (aName))
295     Init(myParameters->Parameters (aName));
296   else
297     Init(NCollection_List<Handle(Standard_Transient)>());
298
299   if (myParameters->FindFileNames(aName))
300   {
301     NCollection_List<TCollection_AsciiString> aFileNames = myParameters->FileNames (aName);
302     if (aFileNames.Extent() > 0) // only one document file might be opened
303       OpenFile (aFileNames.First());
304     myParameters->SetFileNames(aName, NCollection_List<TCollection_AsciiString>());
305   }
306   onUpdateClicked();
307
308   // make parameter items selected if defined
309   if (myParameters->FindSelectedNames(aName))
310   {
311     const NCollection_List<TCollection_AsciiString>& aSelected = myParameters->GetSelectedNames (aName);
312
313     DFBrowser_TreeModel* aTreeModel = dynamic_cast<DFBrowser_TreeModel*> (myTreeView->model());
314     Handle(TDocStd_Application) anApplication = aTreeModel->GetTDocStdApplication();
315
316     QItemSelectionModel* aSelectionModel = myTreeView->selectionModel();
317     aSelectionModel->clear();
318
319     NCollection_List<TCollection_AsciiString>::Iterator aSelectedIt (aSelected);
320     if (aSelectedIt.More())
321     {
322       TCollection_AsciiString aLabelEntry = aSelectedIt.Value();
323
324       TDF_Label aLabel;
325       for (Standard_Integer aDocId = 1, aNbDoc = anApplication->NbDocuments(); aDocId <= aNbDoc; aDocId++)
326       {
327         Handle(TDocStd_Document) aDocument;
328         anApplication->GetDocument (aDocId, aDocument);
329
330         TDF_Tool::Label(aDocument->GetData(), aLabelEntry.ToCString(), aLabel, Standard_False);
331         if (!aLabel.IsNull())
332           break;
333       }
334       if (!aLabel.IsNull())
335       {
336         QModelIndex anIndexToBeSelected = aTreeModel->FindIndex (aLabel);
337
338         TCollection_AsciiString anAttributeType;
339         aSelectedIt.Next();
340         // find attribute by attribute type on the given label
341         if  (aSelectedIt.More())
342         {
343           anAttributeType = aSelectedIt.Value();
344
345           TreeModel_ItemBasePtr aLabelItem = TreeModel_ModelBase::GetItemByIndex (anIndexToBeSelected);
346           //DFBrowser_ItemPtr anItem = itemDynamicCast<DFBrowser_Item> (anItemBase);
347           for (int aChildId = 0, aNbChildren = aLabelItem->rowCount(); aChildId < aNbChildren; aChildId++)
348           {
349             QModelIndex anIndex = aTreeModel->index (aChildId, 0, anIndexToBeSelected);
350             TreeModel_ItemBasePtr anItemBase = TreeModel_ModelBase::GetItemByIndex (anIndex);
351             DFBrowser_ItemPtr anItem = itemDynamicCast<DFBrowser_Item> (anItemBase);
352             if (anItem->HasAttribute())
353             {
354               // processing attribute in theValue
355               DFBrowser_ItemApplicationPtr aRootAppItem = itemDynamicCast<DFBrowser_ItemApplication>(aTreeModel->RootItem (0));
356               QString anAttributeInfo = DFBrowser_Module::GetAttributeInfo (anItem->GetAttribute(), aRootAppItem->GetModule(),
357                                                       Qt::DisplayRole, 0).toString();
358               if (anAttributeInfo == anAttributeType.ToCString())
359               {
360                 anIndexToBeSelected = anIndex;
361                 break;
362               }
363             }
364           }
365         }
366         aSelectionModel->select (anIndexToBeSelected, QItemSelectionModel::Select);
367         myTreeView->scrollTo (anIndexToBeSelected);
368       }
369     }
370
371     myParameters->SetSelectedNames(aName, NCollection_List<TCollection_AsciiString>());
372   }
373 }
374
375 // =======================================================================
376 // function : Init
377 // purpose :
378 // =======================================================================
379 void DFBrowser_Window::Init (const NCollection_List<Handle(Standard_Transient)>& theParameters)
380 {
381   Handle(TDocStd_Application) anApplication;
382   if (myModule)
383   {
384     DFBrowser_TreeModel* anOCAFViewModel = dynamic_cast<DFBrowser_TreeModel*> (myModule->GetOCAFViewModel());
385     if (anOCAFViewModel)
386       anApplication = anOCAFViewModel->GetTDocStdApplication();
387   }
388   Handle(AIS_InteractiveContext) aContext;
389   if (myModule)
390     aContext = myModule->GetExternalContext();
391
392   bool aSameApplication = !anApplication.IsNull(), aSameContext = !aContext.IsNull();
393   for (NCollection_List<Handle(Standard_Transient)>::Iterator aParametersIt (theParameters);
394        aParametersIt.More(); aParametersIt.Next())
395   {
396     Handle(Standard_Transient) anObject = aParametersIt.Value();
397     // check if the object is an application
398     Handle(TDocStd_Application) anIApplication = Handle(TDocStd_Application)::DownCast (anObject);
399     if (!anIApplication.IsNull())
400     {
401       aSameApplication = anApplication == anIApplication;
402       if (!aSameApplication)
403         anApplication = anIApplication;
404     }
405     // check if the object is an interactive context
406     Handle(AIS_InteractiveContext) anIContext = Handle(AIS_InteractiveContext)::DownCast (anObject);
407     if (!anIContext.IsNull())
408     {
409       aSameContext = aContext == anIContext;
410       if (!aSameContext)
411         aContext = anIContext;
412     }
413   }
414   if (aSameApplication)
415   {
416     if (!aSameContext && !aContext.IsNull())
417     {
418       myModule->SetExternalContext (aContext);
419       myViewWindow->SetContext (View_ContextType_External, aContext);
420     }
421     return;
422   }
423
424   myModule = new DFBrowser_Module();
425   myModule->CreateViewModel (myMainWindow);
426
427   myPropertyPanel->GetAttributesStack()->SetModule (myModule);
428
429   // model should be set after the attribute pane stack is initialized by module
430   QAbstractItemModel* aModel = myModule->GetOCAFViewModel();
431   setOCAFModel (aModel);
432   myModule->SetOCAFViewSelectionModel (myTreeView->selectionModel());
433   myTreeLevelLine->GetSearchLine()->SetModule (myModule);
434   myPropertyPanel->GetAttributesStack()->GetSearchView()->InitModels();
435
436   connect (myModule, SIGNAL (beforeUpdateTreeModel()), this, SLOT (onBeforeUpdateTreeModel()));
437
438   if (!aContext.IsNull())
439   {
440     myModule->SetExternalContext (aContext);
441     myViewWindow->SetContext (View_ContextType_External, aContext);
442   }
443
444   myModule->SetApplication (anApplication);
445   //! expand first three levels: CUSTOM
446   QModelIndex aParentIndex = aModel->index (0, 0);
447   setExpandedLevels (myTreeView, aParentIndex, 3/*levels*/);
448
449   myThread->ProcessApplication();
450   myModule->SetInitialTreeViewSelection();
451 }
452
453 // =======================================================================
454 // function : OpenFile
455 // purpose :
456 // =======================================================================
457 void DFBrowser_Window::OpenFile (const TCollection_AsciiString& theFileName)
458 {
459   QApplication::setOverrideCursor (Qt::WaitCursor);
460   myThread->TerminateThread();
461
462   myTreeLevelLine->ClearHistory();
463   QItemSelectionModel* aSelectionModel = myModule->GetOCAFViewSelectionModel();
464   if (aSelectionModel)
465   {
466     aSelectionModel->clearSelection();
467     QModelIndex anIndex;
468     aSelectionModel->select (anIndex, QItemSelectionModel::ClearAndSelect);
469   }
470   ClearThreadCache();
471
472   myTreeLevelLine->ClearHistory();
473
474   DFBrowser_TreeModel* anOCAFViewModel = dynamic_cast<DFBrowser_TreeModel*> (myModule->GetOCAFViewModel());
475   anOCAFViewModel->Reset();
476
477   //! close previous documents to open new document
478   Handle(TDocStd_Application) anApplication;
479   if (CDF_Session::Exists())
480   {
481     Handle(CDF_Session) aSession = CDF_Session::CurrentSession();
482     if (!aSession.IsNull())
483     {
484       anApplication = Handle(TDocStd_Application)::DownCast (CDF_Session::CurrentSession()->CurrentApplication());
485       if (!anApplication.IsNull())
486       {
487         for (int aDocId = 1, aNbDocuments = anApplication->NbDocuments(); aDocId <= aNbDocuments; aDocId++)
488         {
489           Handle(TDocStd_Document) aDocument;
490           anApplication->GetDocument (aDocId, aDocument);
491           if (!aDocument.IsNull())
492             anApplication->Close (aDocument);
493         }
494       }
495     }
496   }
497   //! open new document
498   bool isSTEPFileName = false;
499   anApplication = DFBrowser_OpenApplication::OpenApplication (theFileName, isSTEPFileName);
500   
501   if (myParent)
502     myParent->setObjectName (isSTEPFileName ? QString (TCollection_AsciiString (theFileName).ToCString()) : getWindowTitle());
503   else
504     myOpenedFileName = isSTEPFileName ? QString(TCollection_AsciiString(theFileName).ToCString()) : getWindowTitle();
505
506   if (anApplication.IsNull())
507   {
508     QApplication::restoreOverrideCursor();
509     QMessageBox::information (0, "Error", QString ("File %1 can't be opened by OCAF application")
510                                                    .arg(TCollection_AsciiString (theFileName).ToCString()));
511   }
512   else {
513     myModule->SetApplication (anApplication);
514     //! expand first three levels: CUSTOM
515     QModelIndex aParentIndex = anOCAFViewModel->index (0, 0);
516     setExpandedLevels (myTreeView, aParentIndex, 3/*levels*/);
517
518     myThread->ProcessApplication();
519     myModule->SetInitialTreeViewSelection();
520     QApplication::restoreOverrideCursor();
521   }
522 }
523
524 // =======================================================================
525 // function : getWindowTitle
526 // purpose :
527 // =======================================================================
528 QString DFBrowser_Window::getWindowTitle() const
529 {
530   DFBrowser_TreeModel* anOCAFViewModel = dynamic_cast<DFBrowser_TreeModel*> (myModule->GetOCAFViewModel());
531   if (!anOCAFViewModel)
532     return "";
533
534   Handle(TDocStd_Application) anApplication = anOCAFViewModel->GetTDocStdApplication();
535   if (anApplication.IsNull() || anApplication->NbDocuments() == 0)
536     return "";
537
538   Handle(TDocStd_Document) aDocument;
539   anApplication->GetDocument (1, aDocument);
540   if (aDocument.IsNull() || !aDocument->IsSaved())
541     return "";
542
543   return DFBrowserPane_Tools::ToString (aDocument->GetPath());
544 }
545
546 // =======================================================================
547 // function : setExpandedLevels
548 // purpose :
549 // =======================================================================
550 void DFBrowser_Window::setExpandedLevels (QTreeView* theTreeView, const QModelIndex& theParentIndex, const int theLevels)
551 {
552   if (theLevels <= 0)
553     return;
554
555   QAbstractItemModel* aModel = theTreeView->model();
556   if (!aModel)
557     return;
558
559   theTreeView->setExpanded (theParentIndex, true);
560   for (int aRowId = 0, aRows = aModel->rowCount (theParentIndex); aRowId < aRows; aRowId++)
561     setExpandedLevels (theTreeView, aModel->index (aRowId, 0, theParentIndex), theLevels - 1);
562 }
563
564 // =======================================================================
565 // function : setOCAFModel
566 // purpose :
567 // =======================================================================
568 void DFBrowser_Window::setOCAFModel (QAbstractItemModel* theModel)
569 {
570   myTreeView->setModel (theModel);
571
572   QItemSelectionModel* aSelectionModel = new QItemSelectionModel (theModel);
573   myTreeView->setSelectionModel (aSelectionModel);
574
575   connect (aSelectionModel, SIGNAL (selectionChanged (const QItemSelection&, const QItemSelection&)),
576            myTreeLevelLine, SLOT (OnTreeViewSelectionChanged (const QItemSelection&, const QItemSelection&)));
577   connect(aSelectionModel, SIGNAL (selectionChanged (const QItemSelection&, const QItemSelection&)),
578           myDumpView, SLOT (OnTreeViewSelectionChanged (const QItemSelection&, const QItemSelection&)));
579   connect (aSelectionModel, SIGNAL (selectionChanged (const QItemSelection&, const QItemSelection&)),
580            this, SLOT (onTreeViewSelectionChanged (const QItemSelection&, const QItemSelection&)));
581 }
582
583 // =======================================================================
584 // function : onBeforeUpdateTreeModel
585 // purpose :
586 // =======================================================================
587 void DFBrowser_Window::onBeforeUpdateTreeModel()
588 {
589   myTreeLevelLine->ClearHistory();
590   ClearThreadCache();
591   myThread->ProcessApplication();
592 }
593
594 // =======================================================================
595 // function : ClearThreadCache
596 // purpose :
597 // =======================================================================
598 void DFBrowser_Window::ClearThreadCache()
599 {
600   DFBrowser_ThreadItemSearch::ClearValues (GetTreeLevelLine()->GetSearchLine());
601 }
602
603 // =======================================================================
604 // function : TmpDirectory
605 // purpose :
606 // =======================================================================
607 TCollection_AsciiString DFBrowser_Window::TmpDirectory()
608 {
609   TCollection_AsciiString aTmpDir;
610 #ifdef _WIN32
611   OSD_Environment anEnvironment ("TEMP");
612   aTmpDir = anEnvironment.Value();
613   if (aTmpDir.IsEmpty() )
614   {
615     anEnvironment.SetName("TMP");
616     aTmpDir = anEnvironment.Value();
617     if (aTmpDir.IsEmpty())
618       aTmpDir = "C:\\";
619   }
620   OSD_Path aTmpPath (aTmpDir);
621   OSD_Directory aTmpDirectory (aTmpPath);
622   if (!aTmpDirectory.Exists())
623     aTmpDirectory.Build(OSD_Protection());
624 #else
625   OSD_Directory aTmpDirectory = OSD_Directory::BuildTemporary();
626   OSD_Path aTmpPath;
627   aTmpDirectory.Path (aTmpPath);
628   aTmpPath.SystemName(aTmpDir);
629 #endif
630
631   return aTmpDir;
632 }
633
634 // =======================================================================
635 // function : onTreeViewContextMenuRequested
636 // purpose :
637 // =======================================================================
638 void DFBrowser_Window::onTreeViewContextMenuRequested (const QPoint& thePosition)
639 {
640   QMenu* aMenu = new QMenu(GetMainWindow());
641   aMenu->addAction (ViewControl_Tools::CreateAction (tr ("Expand"), SLOT (onExpand()), GetMainWindow(), this));
642   aMenu->addAction (ViewControl_Tools::CreateAction (tr ("Expand All"), SLOT (onExpandAll()), GetMainWindow(), this));
643   aMenu->addAction (ViewControl_Tools::CreateAction (tr ("Collapse All"), SLOT (onCollapseAll()), GetMainWindow(), this));
644
645   QPoint aPoint = myTreeView->mapToGlobal (thePosition);
646   aMenu->exec (aPoint);
647 }
648
649 // =======================================================================
650 // function : onExpand
651 // purpose :
652 // =======================================================================
653 void DFBrowser_Window::onExpand()
654 {
655   QApplication::setOverrideCursor (Qt::WaitCursor);
656
657   QItemSelectionModel* aSelectionModel = myTreeView->selectionModel();
658   QModelIndexList aSelectedIndices = aSelectionModel->selectedIndexes();
659   for (int aSelectedId = 0, aSize = aSelectedIndices.size(); aSelectedId < aSize; aSelectedId++)
660   {
661     int aLevels = 2;
662     setExpanded (myTreeView, aSelectedIndices[aSelectedId], true, aLevels);
663   }
664   QApplication::restoreOverrideCursor();
665 }
666
667 // =======================================================================
668 // function : onExpandAll
669 // purpose :
670 // =======================================================================
671 void DFBrowser_Window::onExpandAll()
672 {
673   QApplication::setOverrideCursor (Qt::WaitCursor);
674
675   QItemSelectionModel* aSelectionModel = myTreeView->selectionModel();
676   QModelIndexList aSelectedIndices = aSelectionModel->selectedIndexes();
677   for (int  aSelectedId = 0, aSize = aSelectedIndices.size(); aSelectedId < aSize; aSelectedId++)
678   {
679     int aLevels = -1;
680     setExpanded (myTreeView, aSelectedIndices[aSelectedId], true, aLevels);
681   }
682   QApplication::restoreOverrideCursor();
683 }
684
685 // =======================================================================
686 // function : onCollapseAll
687 // purpose :
688 // =======================================================================
689 void DFBrowser_Window::onCollapseAll()
690 {
691   QItemSelectionModel* aSelectionModel = myTreeView->selectionModel();
692   QModelIndexList aSelectedIndices = aSelectionModel->selectedIndexes();
693   for (int aSelectedId = 0, aSize = aSelectedIndices.size(); aSelectedId < aSize; aSelectedId++) {
694     int aLevels = -1;
695     setExpanded (myTreeView, aSelectedIndices[aSelectedId], false, aLevels);
696   }
697 }
698
699 // =======================================================================
700 // function : onTreeViewSelectionChanged
701 // purpose :
702 // =======================================================================
703 void DFBrowser_Window::onTreeViewSelectionChanged (const QItemSelection& theSelected,
704                                                    const QItemSelection& theDeselected)
705 {
706   if (!myModule)
707     return;
708   // previuos selection should be cleared in the panel selectors
709   DFBrowser_AttributePaneStack* anAttributePaneStack = myPropertyPanel->GetAttributesStack();
710   anAttributePaneStack->GetPaneSelector()->ClearSelected();
711
712   myPropertyPanel->UpdateBySelectionChanged (theSelected, theDeselected);
713   anAttributePaneStack->GetTreeLevelView()->UpdateByTreeSelectionChanged (theSelected, theDeselected);
714
715   QModelIndexList aSelectedIndices = theSelected.indexes();
716   QModelIndex aSelectedIndex = TreeModel_ModelBase::SingleSelected (aSelectedIndices, 0);
717
718   myTreeView->scrollTo (aSelectedIndex);
719   View_Displayer* aDisplayer = myViewWindow->GetDisplayer();
720   
721   aDisplayer->ErasePresentations (View_PresentationType_Additional, false);
722   aDisplayer->DisplayPresentation (findPresentation (aSelectedIndex), View_PresentationType_Main);
723 }
724
725 // =======================================================================
726 // function : onSearchActivated
727 // purpose :
728 // =======================================================================
729 void DFBrowser_Window::onSearchActivated()
730 {
731   myPropertyPanel->GetAttributesStack()->SetPaneMode ((myTreeLevelLine->GetSearchLine()->Text().isEmpty()
732                         ? DFBrowser_AttributePaneType_ItemView : DFBrowser_AttributePaneType_SearchView));
733 }
734
735 // =======================================================================
736 // function : onPaneSelectionChanged
737 // purpose :
738 // =======================================================================
739 void DFBrowser_Window::onPaneSelectionChanged (const QItemSelection&,
740                                                const QItemSelection&,
741                                                QItemSelectionModel* theModel)
742 {
743   DFBrowserPane_AttributePaneAPI* anAttributePane = myPropertyPanel->GetAttributesStack()->GetCurrentPane();
744   switch (anAttributePane->GetSelectionKind (theModel))
745   {
746     case DFBrowserPane_SelectionKind_ExportToShapeViewer:
747     {
748       QItemSelectionModel* aSelectionModel = theModel;
749       QModelIndexList aSelectedIndices = aSelectionModel->selectedIndexes();
750       if (aSelectedIndices.size() != 1)
751         return;
752
753       TCollection_AsciiString aPluginName ("TKShapeView");
754       NCollection_List<Handle(Standard_Transient)> aParameters;
755       if (myParameters->FindParameters (aPluginName))
756         aParameters = myParameters->Parameters (aPluginName);
757
758       NCollection_List<TCollection_AsciiString> anItemNames;
759       if (myParameters->FindSelectedNames (aPluginName))
760         anItemNames = myParameters->GetSelectedNames (aPluginName);
761
762       int aParametersCount = aParameters.Extent();
763       anAttributePane->GetSelectionParameters (aSelectionModel, aParameters, anItemNames);
764       if (aParametersCount != aParameters.Extent()) // some TShapes are added
765       {
766         TCollection_AsciiString aPluginShortName = aPluginName.SubString (3, aPluginName.Length());
767         QString aMessage = QString ("TShape %1 is sent to %2.")
768           .arg (DFBrowserPane_Tools::GetPointerInfo (aParameters.Last()).ToCString())
769           .arg (aPluginShortName.ToCString());
770         QString aQuestion = QString ("Would you like to activate %1 immediately?\n")
771           .arg (aPluginShortName.ToCString()).toStdString().c_str();
772         if (!myExportToShapeViewDialog)
773           myExportToShapeViewDialog = new ViewControl_MessageDialog (myParent, aMessage, aQuestion);
774         else
775           myExportToShapeViewDialog->SetInformation (aMessage);
776         myExportToShapeViewDialog->Start();
777
778         myParameters->SetSelectedNames (aPluginName, anItemNames);
779         myParameters->SetParameters (aPluginName, aParameters, myExportToShapeViewDialog->IsAccepted());
780       }
781       return;
782     }
783     case DFBrowserPane_SelectionKind_ExportToBREP:
784     case DFBrowserPane_SelectionKind_LabelReferences:
785     case DFBrowserPane_SelectionKind_AttributeReferences:
786     default: break;
787   }
788
789   QItemSelectionModel* aSelectionModel = myTreeView->selectionModel();
790   QModelIndexList aSelectedIndices = aSelectionModel->selectedIndexes();
791   if (aSelectedIndices.size() != 1)
792     return;
793
794   // make the shape visualized
795   QModelIndex aSelectedIndex = aSelectedIndices.first();
796   View_Displayer* aDisplayer = myViewWindow->GetDisplayer();
797   aDisplayer->DisplayPresentation (findPresentation (aSelectedIndex), View_PresentationType_Main);
798
799   // highlight and scroll to the referenced item if it exists
800   Handle(TDF_Attribute) anAttribute = myModule->FindAttribute (aSelectedIndex);
801   NCollection_List<TDF_Label> aReferences;
802   Handle(Standard_Transient) aPresentation;
803   anAttributePane->GetReferences (anAttribute, aReferences, aPresentation);
804   QModelIndexList anIndices;
805   DFBrowser_TreeModel* aTreeModel = dynamic_cast<DFBrowser_TreeModel*> (myTreeView->model());
806   if (!aReferences.IsEmpty())
807     aTreeModel->ConvertToIndices (aReferences, anIndices);
808   else {
809     NCollection_List<Handle(TDF_Attribute)> anAttributeReferences;
810     anAttributePane->GetAttributeReferences (anAttribute, anAttributeReferences, aPresentation);
811     aTreeModel->ConvertToIndices (anAttributeReferences, anIndices);
812   }
813   highlightIndices (anIndices);
814   // display either the reference presentation of the panel or find a presentation if the reference
815   // is an attribute
816   if (!aPresentation.IsNull())
817     aDisplayer->DisplayPresentation (aPresentation, View_PresentationType_Additional);
818   else {
819     aDisplayer->ErasePresentations (View_PresentationType_Additional, false);
820     AIS_ListOfInteractive aDisplayed;
821     findPresentations (anIndices, aDisplayed);
822     for (AIS_ListIteratorOfListOfInteractive aDisplayedIt (aDisplayed); aDisplayedIt.More(); aDisplayedIt.Next())
823       aDisplayer->DisplayPresentation (aDisplayedIt.Value(), View_PresentationType_Additional, false);
824
825     aDisplayer->UpdateViewer();
826   }
827 }
828
829 // =======================================================================
830 // function : onTreeLevelLineSelected
831 // purpose :
832 // =======================================================================
833 void DFBrowser_Window::onTreeLevelLineSelected (const QModelIndex& theIndex)
834 {
835   QItemSelectionModel* aSelectionModel = myTreeView->selectionModel();
836   if (theIndex.isValid())
837     aSelectionModel->select (theIndex, QItemSelectionModel::ClearAndSelect);
838   else
839     aSelectionModel->clearSelection();
840 }
841
842 // =======================================================================
843 // function : onUpdateClicked
844 // purpose :
845 // =======================================================================
846 void DFBrowser_Window::onUpdateClicked()
847 {
848   if (myModule)
849     myModule->UpdateTreeModel();
850 }
851
852 // =======================================================================
853 // function : onSearchPathSelected
854 // purpose :
855 // =======================================================================
856 void DFBrowser_Window::onSearchPathSelected (const QStringList& thePath, const QString& theValue)
857 {
858   DFBrowser_TreeModel* aDFBrowserModel = dynamic_cast<DFBrowser_TreeModel*> (myTreeView->model());
859   const QModelIndex& anIndex = aDFBrowserModel->FindIndexByPath (thePath, theValue);
860
861   if (anIndex.isValid())
862   {
863     QModelIndexList anIndices;
864     anIndices.append (anIndex);
865     highlightIndices (anIndices);
866   }
867 }
868
869 // =======================================================================
870 // function : onSearchPathDoubleClicked
871 // purpose :
872 // =======================================================================
873 void DFBrowser_Window::onSearchPathDoubleClicked (const QStringList& thePath, const QString& theValue)
874 {
875   DFBrowser_TreeModel* aDFBrowserModel = dynamic_cast<DFBrowser_TreeModel*> (myTreeView->model());
876   const QModelIndex& anIndex = aDFBrowserModel->FindIndexByPath (thePath, theValue);
877
878   QItemSelectionModel* aSelectionModel = myTreeView->selectionModel();
879   if (anIndex.isValid())
880     aSelectionModel->select (anIndex, QItemSelectionModel::ClearAndSelect);
881   else
882     aSelectionModel->clearSelection();
883 }
884
885 // =======================================================================
886 // function : onLevelSelected
887 // purpose :
888 // =======================================================================
889 void DFBrowser_Window::onLevelSelected (const QModelIndex& theIndex)
890 {
891   if (!theIndex.isValid())
892     return;
893
894   QModelIndexList anIndices;
895   anIndices.append (theIndex);
896   highlightIndices (anIndices);
897   View_Displayer* aDisplayer = myViewWindow->GetDisplayer();
898   aDisplayer->ErasePresentations (View_PresentationType_Additional, false);
899   aDisplayer->DisplayPresentation (findPresentation (theIndex), View_PresentationType_Main);
900 }
901
902 // =======================================================================
903 // function : onLevelDoubleClicked
904 // purpose :
905 // =======================================================================
906 void DFBrowser_Window::onLevelDoubleClicked (const QModelIndex& theIndex)
907 {
908   QItemSelectionModel* aSelectionModel = myTreeView->selectionModel();
909   if (theIndex.isValid())
910     aSelectionModel->select (theIndex, QItemSelectionModel::ClearAndSelect);
911   else
912     aSelectionModel->clearSelection();
913 }
914
915 // =======================================================================
916 // function : highlightIndices
917 // purpose :
918 // =======================================================================
919 void DFBrowser_Window::highlightIndices (const QModelIndexList& theIndices)
920 {
921   QAbstractItemModel* aModel = myTreeView->model();
922   if (!aModel)
923     return;
924
925   DFBrowser_TreeModel* aTreeModel = dynamic_cast<DFBrowser_TreeModel*> (aModel);
926   if (!aTreeModel)
927     return;
928
929   aTreeModel->SetHighlighted (theIndices);
930
931   QModelIndex anIndexToScroll;
932   if (!theIndices.isEmpty())
933     anIndexToScroll = theIndices.last(); // scroll to last selected index
934   else
935   {
936     // scroll to tree selected item
937     QItemSelectionModel* aSelectionModel = myTreeView->selectionModel();
938     QModelIndexList aSelectedIndices = aSelectionModel->selectedIndexes();
939     if (aSelectedIndices.size() == 1)
940       anIndexToScroll = aSelectedIndices.first();
941   }
942   if (anIndexToScroll.isValid())
943     myTreeView->scrollTo (anIndexToScroll);
944
945   if (theIndices.isEmpty())
946     myTreeView->setFocus(); // to see the selected item in active palette color
947
948   aTreeModel->EmitLayoutChanged();
949 }
950
951 // =======================================================================
952 // function : findPresentation
953 // purpose :
954 // =======================================================================
955 Handle(AIS_InteractiveObject) DFBrowser_Window::findPresentation (const QModelIndex& theIndex)
956 {
957   Handle(AIS_InteractiveObject) aPresentation;
958
959   QModelIndexList anIndices;
960   anIndices.append (theIndex);
961   AIS_ListOfInteractive aDisplayed;
962   findPresentations (anIndices, aDisplayed);
963   if (!aDisplayed.IsEmpty())
964     aPresentation = aDisplayed.First();
965
966   return aPresentation;
967 }
968
969 // =======================================================================
970 // function : findPresentations
971 // purpose :
972 // =======================================================================
973 void DFBrowser_Window::findPresentations (const QModelIndexList& theIndices, AIS_ListOfInteractive& thePresentations)
974 {
975   for (int anIndexId = 0, aCount = theIndices.size(); anIndexId < aCount; anIndexId++)
976   {
977     Handle(AIS_InteractiveObject) aPresentation;
978     Handle(TDF_Attribute) anAttribute = myModule->FindAttribute (theIndices[anIndexId]);
979     if (anAttribute.IsNull())
980       continue;
981     DFBrowserPane_AttributePaneAPI* anAttributePane = myModule->GetAttributePane (anAttribute);
982     if (!anAttributePane)
983       continue;
984     aPresentation = Handle(AIS_InteractiveObject)::DownCast (anAttributePane->GetPresentation (anAttribute));
985     if (aPresentation.IsNull())
986       continue;
987
988     thePresentations.Append (aPresentation);
989   }
990 }
991
992 // =======================================================================
993 // function : setExpanded
994 // purpose :
995 // =======================================================================
996 void DFBrowser_Window::setExpanded (QTreeView* theTreeView, const QModelIndex& theIndex, const bool isExpanded,
997                                     int& theLevels)
998 {
999   bool isToExpand = theLevels == -1 || theLevels > 0;
1000   if (!isToExpand)
1001     return;
1002
1003   theTreeView->setExpanded (theIndex, isExpanded);
1004   if (theLevels != -1)
1005     theLevels--;
1006
1007   QAbstractItemModel* aModel = theTreeView->model();
1008   for (int aRowId = 0, aRows = aModel->rowCount (theIndex); aRowId < aRows; aRowId++)
1009   {
1010     int aLevels = theLevels;
1011     setExpanded (theTreeView, aModel->index (aRowId, 0, theIndex), isExpanded, aLevels);
1012   }
1013 }