Salome HOME
Update copyrights
[modules/shaper.git] / src / XGUI / XGUI_ViewerProxy.cpp
1 // Copyright (C) 2014-2019  CEA/DEN, EDF R&D
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 #include "XGUI_ViewerProxy.h"
21 #include "XGUI_Workshop.h"
22 #include "XGUI_SalomeConnector.h"
23 #include "XGUI_Displayer.h"
24
25 #ifndef HAVE_SALOME
26   #include <AppElements_MainWindow.h>
27   #include <AppElements_ViewPort.h>
28   #include <AppElements_ViewWindow.h>
29   #include <AppElements_Viewer.h>
30 #endif
31
32 #include <ModuleBase_IViewWindow.h>
33 #include <GeomAPI_Shape.h>
34 #include <ModelAPI_ResultConstruction.h>
35
36 #include <Config_PropManager.h>
37
38 #include <AIS_Shape.hxx>
39 #include <StdSelect_BRepOwner.hxx>
40
41 #include <QEvent>
42
43
44 #define HIGHLIGHT_COLOR Quantity_NOC_YELLOW
45
46 XGUI_ViewerProxy::XGUI_ViewerProxy(XGUI_Workshop* theParent)
47     : ModuleBase_IViewer(theParent),
48       myWorkshop(theParent)
49 {
50 }
51
52 void XGUI_ViewerProxy::connectViewProxy()
53 {
54 #ifdef HAVE_SALOME
55   connect(myWorkshop->salomeConnector()->viewer(), SIGNAL(trihedronVisibilityChanged(bool)),
56          SIGNAL(trihedronVisibilityChanged(bool)));
57 #else
58   connect(myWorkshop->mainWindow()->viewer(), SIGNAL(trihedronVisibilityChanged(bool)),
59          SIGNAL(trihedronVisibilityChanged(bool)));
60 #endif
61 }
62
63 Handle(AIS_InteractiveContext) XGUI_ViewerProxy::AISContext() const
64 {
65   Handle(AIS_InteractiveContext) aContext;
66 #ifdef HAVE_SALOME
67   aContext = myWorkshop->salomeConnector()->viewer()->AISContext();
68 #else
69   aContext = myWorkshop->mainWindow()->viewer()->AISContext();
70 #endif
71   return aContext;
72 }
73
74 Handle(AIS_Trihedron) XGUI_ViewerProxy::trihedron() const
75 {
76 #ifdef HAVE_SALOME
77   return myWorkshop->salomeConnector()->viewer()->trihedron();
78 #else
79   return myWorkshop->mainWindow()->viewer()->trihedron();
80 #endif
81 }
82
83 Handle(V3d_Viewer) XGUI_ViewerProxy::v3dViewer() const
84 {
85 #ifdef HAVE_SALOME
86   return myWorkshop->salomeConnector()->viewer()->v3dViewer();
87 #else
88   return myWorkshop->mainWindow()->viewer()->v3dViewer();
89 #endif
90 }
91
92 Handle(V3d_View) XGUI_ViewerProxy::activeView() const
93 {
94 #ifdef HAVE_SALOME
95   return myWorkshop->salomeConnector()->viewer()->activeView();
96 #else
97   AppElements_Viewer* aViewer = myWorkshop->mainWindow()->viewer();
98   return (aViewer->activeViewWindow()) ?
99     aViewer->activeViewWindow()->viewPortApp()->getView() :
100     Handle(V3d_View)();
101 #endif
102 }
103
104 QWidget* XGUI_ViewerProxy::activeViewPort() const
105 {
106 #ifdef HAVE_SALOME
107   return myWorkshop->salomeConnector()->viewer()->activeViewPort();
108 #else
109   AppElements_Viewer* aViewer = myWorkshop->mainWindow()->viewer();
110   return (aViewer->activeViewWindow()) ?
111          aViewer->activeViewWindow()->viewPortApp(): 0;
112 #endif
113 }
114
115 void XGUI_ViewerProxy::setViewProjection(double theX, double theY, double theZ, double theTwist)
116 {
117   Handle(V3d_View) aView3d = activeView();
118   if (!aView3d.IsNull()) {
119     aView3d->SetProj(theX, theY, theZ);
120     aView3d->SetTwist( theTwist );
121     aView3d->FitAll(0.01, false);
122     aView3d->SetZSize(0.);
123     if (aView3d->Depth() < 0.1)
124       aView3d->DepthFitAll();
125   }
126 }
127
128 void XGUI_ViewerProxy::fitAll()
129 {
130 #ifdef HAVE_SALOME
131   myWorkshop->salomeConnector()->viewer()->fitAll();
132 #else
133   AppElements_Viewer* aViewer = myWorkshop->mainWindow()->viewer();
134   if (aViewer->activeViewWindow())
135     aViewer->activeViewWindow()->viewPortApp()->fitAll();
136 #endif
137 }
138
139 void XGUI_ViewerProxy::eraseAll()
140 {
141 #ifdef HAVE_SALOME
142   myWorkshop->salomeConnector()->viewer()->eraseAll();
143 #else
144 #endif
145 }
146
147 void XGUI_ViewerProxy::connectToViewer()
148 {
149 #ifdef HAVE_SALOME
150   ModuleBase_IViewer* aViewer = myWorkshop->salomeConnector()->viewer();
151
152   connect(aViewer, SIGNAL(lastViewClosed()), this, SIGNAL(lastViewClosed()));
153   connect(aViewer, SIGNAL(tryCloseView(ModuleBase_IViewWindow*)),
154     this, SIGNAL(tryCloseView(ModuleBase_IViewWindow*)));
155
156   connect(aViewer, SIGNAL(deleteView(ModuleBase_IViewWindow*)),
157     this, SIGNAL(deleteView(ModuleBase_IViewWindow*)));
158
159   connect(aViewer, SIGNAL(viewCreated(ModuleBase_IViewWindow*)),
160     this, SLOT(onViewCreated(ModuleBase_IViewWindow*)));
161
162   connect(aViewer, SIGNAL(activated(ModuleBase_IViewWindow*)),
163     this, SIGNAL(activated(ModuleBase_IViewWindow*)));
164
165   connect(aViewer, SIGNAL(mousePress(ModuleBase_IViewWindow*, QMouseEvent*)),
166     this, SIGNAL(mousePress(ModuleBase_IViewWindow*, QMouseEvent*)));
167
168   connect(aViewer, SIGNAL(mouseRelease(ModuleBase_IViewWindow*, QMouseEvent*)),
169     this, SIGNAL(mouseRelease(ModuleBase_IViewWindow*, QMouseEvent*)));
170
171   connect(aViewer, SIGNAL(mouseDoubleClick(ModuleBase_IViewWindow*, QMouseEvent*)), this,
172           SIGNAL(mouseDoubleClick(ModuleBase_IViewWindow*, QMouseEvent*)));
173
174   connect(aViewer, SIGNAL(mouseMove(ModuleBase_IViewWindow*, QMouseEvent*)),
175     this, SLOT(onMouseMove(ModuleBase_IViewWindow*, QMouseEvent*)));
176
177   connect(aViewer, SIGNAL(keyPress(ModuleBase_IViewWindow*, QKeyEvent*)),
178     this, SIGNAL(keyPress(ModuleBase_IViewWindow*, QKeyEvent*)));
179
180   connect(aViewer, SIGNAL(keyRelease(ModuleBase_IViewWindow*, QKeyEvent*)),
181     this, SIGNAL(keyRelease(ModuleBase_IViewWindow*, QKeyEvent*)));
182
183   connect(aViewer, SIGNAL(selectionChanged()), this, SIGNAL(selectionChanged()));
184
185   connect(aViewer, SIGNAL(viewTransformed(int)), this, SIGNAL(viewTransformed(int)));
186
187   connect(aViewer, SIGNAL(contextMenuRequested(QContextMenuEvent*)), this,
188           SIGNAL(contextMenuRequested(QContextMenuEvent*)));
189 #else
190   AppElements_Viewer* aViewer = myWorkshop->mainWindow()->viewer();
191
192   connect(aViewer, SIGNAL(lastViewClosed()), this, SIGNAL(lastViewClosed()));
193
194   connect(aViewer, SIGNAL(tryCloseView(AppElements_ViewWindow*)),
195           this, SLOT(onTryCloseView(AppElements_ViewWindow*)));
196
197   connect(aViewer, SIGNAL(deleteView(AppElements_ViewWindow*)),
198           this, SLOT(onDeleteView(AppElements_ViewWindow*)));
199
200   connect(aViewer, SIGNAL(viewCreated(AppElements_ViewWindow*)),
201           this, SLOT(onViewCreated(AppElements_ViewWindow*)));
202
203   connect(aViewer, SIGNAL(activated(AppElements_ViewWindow*)),
204           this, SLOT(onActivated(AppElements_ViewWindow*)));
205
206   connect(aViewer, SIGNAL(mousePress(AppElements_ViewWindow*, QMouseEvent*)), this,
207           SLOT(onMousePress(AppElements_ViewWindow*, QMouseEvent*)));
208
209   connect(aViewer, SIGNAL(mouseRelease(AppElements_ViewWindow*, QMouseEvent*)), this,
210           SLOT(onMouseRelease(AppElements_ViewWindow*, QMouseEvent*)));
211
212   connect(aViewer, SIGNAL(mouseDoubleClick(AppElements_ViewWindow*, QMouseEvent*)), this,
213           SLOT(onMouseDoubleClick(AppElements_ViewWindow*, QMouseEvent*)));
214
215   connect(aViewer, SIGNAL(mouseMove(AppElements_ViewWindow*, QMouseEvent*)), this,
216           SLOT(onMouseMove(AppElements_ViewWindow*, QMouseEvent*)));
217
218   connect(aViewer, SIGNAL(keyPress(AppElements_ViewWindow*, QKeyEvent*)), this,
219           SLOT(onKeyPress(AppElements_ViewWindow*, QKeyEvent*)));
220
221   connect(aViewer, SIGNAL(keyRelease(AppElements_ViewWindow*, QKeyEvent*)), this,
222           SLOT(onKeyRelease(AppElements_ViewWindow*, QKeyEvent*)));
223
224   connect(aViewer, SIGNAL(selectionChanged()), this, SIGNAL(selectionChanged()));
225   connect(aViewer, SIGNAL(contextMenuRequested(QContextMenuEvent*)), this,
226           SIGNAL(contextMenuRequested(QContextMenuEvent*)));
227 #endif
228 }
229
230 bool XGUI_ViewerProxy::eventFilter(QObject *theObject, QEvent *theEvent)
231 {
232   if (theEvent->type() == QEvent::Enter) {
233     emit enterViewPort();
234   }
235   else if (theEvent->type() == QEvent::Leave) {
236     emit leaveViewPort();
237   }
238   return ModuleBase_IViewer::eventFilter(theObject, theEvent);
239 }
240
241 void XGUI_ViewerProxy::onViewCreated(ModuleBase_IViewWindow* theWnd)
242 {
243   theWnd->viewPort()->installEventFilter(this);
244
245   myWindowScale.insert (theWnd->v3dView(), theWnd->v3dView()->Camera()->Scale());
246
247   emit viewCreated(theWnd);
248 }
249
250 #ifndef HAVE_SALOME
251 void XGUI_ViewerProxy::onTryCloseView(AppElements_ViewWindow* theWnd)
252 {
253   emit tryCloseView(theWnd);
254 }
255
256 void XGUI_ViewerProxy::onDeleteView(AppElements_ViewWindow* theWnd)
257 {
258   if (myWindowScale.contains(theWnd->v3dView()))
259     myWindowScale.remove (theWnd->v3dView());
260   emit deleteView(theWnd);
261 }
262
263 void XGUI_ViewerProxy::onViewCreated(AppElements_ViewWindow* theWnd)
264 {
265   theWnd->viewPort()->installEventFilter(this);
266
267   connect(theWnd, SIGNAL(vpTransformationFinished(AppElements_ViewWindow::OperationType)),
268     this, SLOT(onViewTransformed(AppElements_ViewWindow::OperationType)));
269
270   myWindowScale.insert (theWnd->v3dView(), theWnd->v3dView()->Camera()->Scale());
271
272   emit viewCreated(theWnd);
273 }
274
275 void XGUI_ViewerProxy::onActivated(AppElements_ViewWindow* theWnd)
276 {
277   emit activated(theWnd);
278 }
279
280 void XGUI_ViewerProxy::onMousePress(AppElements_ViewWindow* theWnd, QMouseEvent* theEvent)
281 {
282   emit mousePress(theWnd, theEvent);
283 }
284
285 void XGUI_ViewerProxy::onMouseRelease(AppElements_ViewWindow* theWnd, QMouseEvent* theEvent)
286 {
287   emit mouseRelease(theWnd, theEvent);
288 }
289
290 void XGUI_ViewerProxy::onMouseDoubleClick(AppElements_ViewWindow* theWnd, QMouseEvent* theEvent)
291 {
292   emit mouseDoubleClick(theWnd, theEvent);
293 }
294
295 void XGUI_ViewerProxy::onMouseMove(AppElements_ViewWindow* theWnd, QMouseEvent* theEvent)
296 {
297   updateHighlight();
298   emit mouseMove(theWnd, theEvent);
299 }
300
301 void XGUI_ViewerProxy::onKeyPress(AppElements_ViewWindow* theWnd, QKeyEvent* theEvent)
302 {
303   emit keyPress(theWnd, theEvent);
304 }
305
306 void XGUI_ViewerProxy::onKeyRelease(AppElements_ViewWindow* theWnd, QKeyEvent* theEvent)
307 {
308   emit keyRelease(theWnd, theEvent);
309 }
310
311 void XGUI_ViewerProxy::onViewTransformed(AppElements_ViewWindow::OperationType theType)
312 {
313   emit viewTransformed((int) theType);
314 }
315
316 #endif
317
318
319 //***************************************
320 void XGUI_ViewerProxy::enableSelection(bool isEnabled)
321 {
322 #ifdef HAVE_SALOME
323   myWorkshop->salomeConnector()->viewer()->enableSelection(isEnabled);
324 #else
325   myWorkshop->mainWindow()->viewer()->setSelectionEnabled(isEnabled);
326 #endif
327 }
328
329 //***************************************
330 bool XGUI_ViewerProxy::isSelectionEnabled() const
331 {
332 #ifdef HAVE_SALOME
333   return myWorkshop->salomeConnector()->viewer()->isSelectionEnabled();
334 #else
335   return myWorkshop->mainWindow()->viewer()->isSelectionEnabled();
336 #endif
337 }
338
339 //***************************************
340 void XGUI_ViewerProxy::enableMultiselection(bool isEnable)
341 {
342 #ifdef HAVE_SALOME
343   myWorkshop->salomeConnector()->viewer()->enableMultiselection(isEnable);
344 #else
345   myWorkshop->mainWindow()->viewer()->setMultiSelectionEnabled(isEnable);
346 #endif
347 }
348
349 //***************************************
350 bool XGUI_ViewerProxy::isMultiSelectionEnabled() const
351 {
352 #ifdef HAVE_SALOME
353   return myWorkshop->salomeConnector()->viewer()->isMultiSelectionEnabled();
354 #else
355   return myWorkshop->mainWindow()->viewer()->isMultiSelectionEnabled();
356 #endif
357 }
358
359 //***************************************
360 bool XGUI_ViewerProxy::enableDrawMode(bool isEnabled)
361 {
362 #ifdef HAVE_SALOME
363   return myWorkshop->salomeConnector()->viewer()->enableDrawMode(isEnabled);
364 #else
365   return myWorkshop->mainWindow()->viewer()->enableDrawMode(isEnabled);
366 #endif
367 }
368
369 //***************************************
370 void XGUI_ViewerProxy::addSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
371 {
372   myWorkshop->displayer()->addSelectionFilter(theFilter);
373 }
374
375 //***************************************
376 void XGUI_ViewerProxy::removeSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
377 {
378   myWorkshop->displayer()->removeSelectionFilter(theFilter);
379 }
380
381 //***************************************
382 bool XGUI_ViewerProxy::hasSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
383 {
384   return myWorkshop->displayer()->hasSelectionFilter(theFilter);
385 }
386
387 //***************************************
388 void XGUI_ViewerProxy::clearSelectionFilters()
389 {
390   myWorkshop->displayer()->removeFilters();
391 }
392
393 //***************************************
394 void XGUI_ViewerProxy::update()
395 {
396   myWorkshop->displayer()->updateViewer();
397 }
398
399 //***************************************
400 bool XGUI_ViewerProxy::canDragByMouse() const
401 {
402   if (myWorkshop->isSalomeMode()) {
403     ModuleBase_IViewer* aViewer = myWorkshop->salomeConnector()->viewer();
404     return aViewer->canDragByMouse();
405   } else {
406     return true;
407   }
408 }
409
410 //***************************************
411 void XGUI_ViewerProxy::displayHighlight(FeaturePtr theFeature, const TopoDS_Shape& theIgnoreShape)
412 {
413   Handle(AIS_InteractiveContext) aContext = AISContext();
414
415   double aDeflection;
416   if (myResult->groupName() == ModelAPI_ResultConstruction::group()) {
417     //FeaturePtr aFeature = ModelAPI_Feature::feature(myResult);
418     if (theFeature.get()) {
419       std::list<ResultPtr> aResults = theFeature->results();
420       std::list<ResultPtr>::const_iterator aIt;
421       ResultPtr aRes;
422       Handle(AIS_Shape) aAis;
423       for (aIt = aResults.cbegin(); aIt != aResults.cend(); aIt++) {
424         aRes = (*aIt);
425         TopoDS_Shape aTShape = aRes->shape()->impl<TopoDS_Shape>();
426         if (!aTShape.IsSame(theIgnoreShape)) {
427           aAis = new AIS_Shape(aTShape);
428           aAis->SetColor(HIGHLIGHT_COLOR);
429           aAis->SetZLayer(Graphic3d_ZLayerId_Top); //Graphic3d_ZLayerId_Topmost
430           aDeflection = Config_PropManager::real("Visualization", "construction_deflection");
431           aAis->Attributes()->SetDeviationCoefficient(aDeflection);
432           myHighlights.Append(aAis);
433           aContext->Display(aAis, false);
434           aContext->Deactivate(aAis);
435         }
436       }
437     }
438   }
439   else {
440     TopoDS_Shape aTShape = myResult->shape()->impl<TopoDS_Shape>();
441     Handle(AIS_Shape) aAis = new AIS_Shape(aTShape);
442     aAis->SetColor(HIGHLIGHT_COLOR);
443     aAis->SetZLayer(Graphic3d_ZLayerId_Top); //Graphic3d_ZLayerId_Topmost
444     aDeflection = Config_PropManager::real("Visualization", "body_deflection");
445     aAis->Attributes()->SetDeviationCoefficient(aDeflection);
446     myHighlights.Append(aAis);
447     aContext->Display(aAis, false);
448     aContext->Deactivate(aAis);
449   }
450 }
451
452 void XGUI_ViewerProxy::eraseHighlight()
453 {
454   Handle(AIS_InteractiveContext) aContext = AISContext();
455   Handle(AIS_InteractiveObject) anAISIO;
456   AIS_ListIteratorOfListOfInteractive aLIt;
457   for (aLIt.Initialize(myHighlights); aLIt.More(); aLIt.Next()) {
458     anAISIO = aLIt.Value();
459     aContext->Remove(anAISIO, false);
460   }
461   myHighlights.Clear();
462 }
463
464 void XGUI_ViewerProxy::updateHighlight()
465 {
466   Handle(AIS_InteractiveContext) aContext = AISContext();
467   if (!aContext.IsNull()) {
468     Handle(StdSelect_BRepOwner) aOwner;
469     Handle(AIS_InteractiveObject) anIO;
470     bool isDisplayed = false;
471     TopoDS_Shape aShape, aShp;
472     ResultPtr aRes;
473     XGUI_Displayer* aDisplayer = myWorkshop->displayer();
474     for (aContext->InitDetected(); aContext->MoreDetected(); aContext->NextDetected()) {
475       aOwner = Handle(StdSelect_BRepOwner)::DownCast(aContext->DetectedOwner());
476       if ((!aOwner.IsNull()) && aOwner->HasShape()) {
477         aShape = aOwner->Shape();
478         anIO = Handle(AIS_InteractiveObject)::DownCast(aOwner->Selectable());
479         aRes = std::dynamic_pointer_cast<ModelAPI_Result>(aDisplayer->getObject(anIO));
480         if (aRes.get() && (aRes != myResult)) {
481           eraseHighlight();
482           FeaturePtr aFeature = ModelAPI_Feature::feature(aRes);
483           aShp = aRes->shape()->impl<TopoDS_Shape>();
484           if ((aFeature->results().size() > 1) || (!aShp.IsSame(aShape))) {
485             myResult = aRes;
486             displayHighlight(aFeature, aShape);
487           }
488           else {
489             myResult = ResultPtr();
490           }
491           aContext->UpdateCurrentViewer();
492         }
493         isDisplayed = aRes.get();
494       }
495     }
496     if (!isDisplayed) {
497       eraseHighlight();
498       aContext->UpdateCurrentViewer();
499       myResult = ResultPtr();
500     }
501   }
502 }
503
504 #ifdef HAVE_SALOME
505 void XGUI_ViewerProxy::onMouseMove(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent)
506 {
507   updateHighlight();
508   emit mouseMove(theWnd, theEvent);
509 }
510 #endif
511
512 //***************************************
513 //void XGUI_ViewerProxy::Zfitall()
514 //{
515 //#ifdef HAVE_SALOME
516 //  myWorkshop->salomeConnector()->viewer()->Zfitall();
517 //#else
518 //  AppElements_Viewer* aViewer = myWorkshop->mainWindow()->viewer();
519 //  AppElements_ViewWindow* aView = aViewer->activeViewWindow();
520 //  if (aView) {
521 //    Handle(V3d_View) aView3d = aView->v3dView();
522 //    aView3d->ZFitAll();
523 //    if (aView3d->Depth() < 0.1)
524 //      aView3d->DepthFitAll();
525 //  }
526 //#endif
527 //}