1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 #include "NewGeom_SalomeViewer.h"
4 #include "NewGeom_OCCSelector.h"
6 #include <OCCViewer_ViewPort3d.h>
7 #include <OCCViewer_ViewFrame.h>
9 #include <SUIT_ViewManager.h>
11 #include <QMouseEvent>
12 #include <QContextMenuEvent>
14 NewGeom_SalomeView::NewGeom_SalomeView(OCCViewer_Viewer* theViewer)
15 : ModuleBase_IViewWindow(), myCurrentView(0)
21 Handle(V3d_View) NewGeom_SalomeView::v3dView() const
23 Handle(V3d_View) aView;
25 OCCViewer_ViewWindow* aWnd = static_cast<OCCViewer_ViewWindow*>(myCurrentView);
26 aView = aWnd->getViewPort()->getView();
31 QWidget* NewGeom_SalomeView::viewPort() const
33 QWidget* aViewPort = 0;
35 OCCViewer_ViewWindow* aWnd = static_cast<OCCViewer_ViewWindow*>(myCurrentView);
36 aViewPort = aWnd->getViewPort();
41 //**********************************************
42 //**********************************************
43 //**********************************************
47 NewGeom_SalomeViewer::NewGeom_SalomeViewer(QObject* theParent)
48 : ModuleBase_IViewer(theParent),
49 mySelector(0), myView(0), myIsSelectionChanged(false)
53 NewGeom_SalomeViewer::~NewGeom_SalomeViewer()
60 //**********************************************
61 Handle(AIS_InteractiveContext) NewGeom_SalomeViewer::AISContext() const
63 if (mySelector && mySelector->viewer())
64 return mySelector->viewer()->getAISContext();
65 Handle(AIS_InteractiveContext) aNull;
69 //**********************************************
70 Handle(V3d_Viewer) NewGeom_SalomeViewer::v3dViewer() const
73 return mySelector->viewer()->getViewer3d();
74 return Handle(V3d_Viewer)();
77 //**********************************************
78 Handle(V3d_View) NewGeom_SalomeViewer::activeView() const
81 OCCViewer_Viewer* aViewer = mySelector->viewer();
82 SUIT_ViewManager* aMgr = aViewer->getViewManager();
83 OCCViewer_ViewWindow* aWnd = static_cast<OCCViewer_ViewWindow*>(aMgr->getActiveView());
84 return aWnd->getViewPort()->getView();
86 return Handle(V3d_View)();
89 //**********************************************
90 void NewGeom_SalomeViewer::setSelector(NewGeom_OCCSelector* theSel)
93 if (mySelector == theSel)
96 mySelector->viewer()->getViewManager()->disconnect(this);
97 OCCViewer_Viewer* aViewer = mySelector->viewer();
99 aViewer->disconnect(this);
105 OCCViewer_Viewer* aViewer = mySelector->viewer();
106 SUIT_ViewManager* aMgr = aViewer->getViewManager();
108 myView = new NewGeom_SalomeView(mySelector->viewer());
110 // TODO: Provide ModuleBase_IViewWindow interface
111 connect(aMgr, SIGNAL(lastViewClosed(SUIT_ViewManager*)), this, SIGNAL(lastViewClosed()));
113 connect(aMgr, SIGNAL(tryCloseView(SUIT_ViewWindow*)),
114 this, SLOT(onTryCloseView(SUIT_ViewWindow*)));
115 connect(aMgr, SIGNAL(deleteView(SUIT_ViewWindow*)),
116 this, SLOT(onDeleteView(SUIT_ViewWindow*)));
117 connect(aMgr, SIGNAL(viewCreated(SUIT_ViewWindow*)),
118 this, SLOT(onViewCreated(SUIT_ViewWindow*)));
119 connect(aMgr, SIGNAL(activated(SUIT_ViewWindow*)),
120 this, SLOT(onActivated(SUIT_ViewWindow*)));
122 connect(aMgr, SIGNAL(mousePress(SUIT_ViewWindow*, QMouseEvent*)), this,
123 SLOT(onMousePress(SUIT_ViewWindow*, QMouseEvent*)));
124 connect(aMgr, SIGNAL(mouseRelease(SUIT_ViewWindow*, QMouseEvent*)), this,
125 SLOT(onMouseRelease(SUIT_ViewWindow*, QMouseEvent*)));
126 connect(aMgr, SIGNAL(mouseDoubleClick(SUIT_ViewWindow*, QMouseEvent*)), this,
127 SLOT(onMouseDoubleClick(SUIT_ViewWindow*, QMouseEvent*)));
128 connect(aMgr, SIGNAL(mouseMove(SUIT_ViewWindow*, QMouseEvent*)), this,
129 SLOT(onMouseMove(SUIT_ViewWindow*, QMouseEvent*)));
131 connect(aMgr, SIGNAL(keyPress(SUIT_ViewWindow*, QKeyEvent*)), this,
132 SLOT(onKeyPress(SUIT_ViewWindow*, QKeyEvent*)));
133 connect(aMgr, SIGNAL(keyRelease(SUIT_ViewWindow*, QKeyEvent*)), this,
134 SLOT(onKeyRelease(SUIT_ViewWindow*, QKeyEvent*)));
136 connect(aViewer, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
139 //**********************************************
140 void NewGeom_SalomeViewer::onSelectionChanged()
142 // Selection event must be sent only after mouse release
143 myIsSelectionChanged = true;
146 //**********************************************
147 void NewGeom_SalomeViewer::onMousePress(SUIT_ViewWindow* theView, QMouseEvent* theEvent)
149 myView->setCurrentView(theView);
150 emit mousePress(myView, theEvent);
153 //**********************************************
154 void NewGeom_SalomeViewer::onMouseRelease(SUIT_ViewWindow* theView, QMouseEvent* theEvent)
156 myView->setCurrentView(theView);
157 emit mouseRelease(myView, theEvent);
158 if (myIsSelectionChanged) {
159 emit selectionChanged();
160 myIsSelectionChanged = false;
164 //**********************************************
165 void NewGeom_SalomeViewer::onMouseDoubleClick(SUIT_ViewWindow* theView, QMouseEvent* theEvent)
167 myView->setCurrentView(theView);
168 emit mouseDoubleClick(myView, theEvent);
171 //**********************************************
172 void NewGeom_SalomeViewer::onMouseMove(SUIT_ViewWindow* theView, QMouseEvent* theEvent)
174 myView->setCurrentView(theView);
175 OCCViewer_ViewWindow* aViewWnd = dynamic_cast<OCCViewer_ViewWindow*>(theView);
176 Handle(AIS_InteractiveContext) aContext = AISContext();
177 if (aContext->HasDetected()) // Set focus to provide key events in the view
178 aViewWnd->getViewPort()->setFocus(Qt::MouseFocusReason);
179 emit mouseMove(myView, theEvent);
182 //**********************************************
183 bool NewGeom_SalomeViewer::canDragByMouse() const
185 OCCViewer_Viewer* aViewer = mySelector->viewer();
186 SUIT_ViewWindow* aWnd = aViewer->getViewManager()->getActiveView();
187 OCCViewer_ViewWindow* aViewWnd = dynamic_cast<OCCViewer_ViewWindow*>(aWnd);
189 return (aViewWnd->interactionStyle() == 0);
195 //**********************************************
196 void NewGeom_SalomeViewer::onKeyPress(SUIT_ViewWindow* theView, QKeyEvent* theEvent)
198 OCCViewer_ViewWindow* aViewWnd = dynamic_cast<OCCViewer_ViewWindow*>(theView);
199 Handle(AIS_InteractiveContext) aContext = AISContext();
200 Handle(V3d_View) aView = aViewWnd->getViewPort()->getView();
202 bool noModifiers = (theEvent->modifiers() == Qt::NoModifier);
203 if ((theEvent->key() == Qt::Key_N) && noModifiers) {
204 aContext->HilightNextDetected(aView);
205 } else if ((theEvent->key() == Qt::Key_P) && noModifiers) {
206 aContext->HilightPreviousDetected(aView);
208 emit keyPress(myView, theEvent);
211 //**********************************************
212 void NewGeom_SalomeViewer::onKeyRelease(SUIT_ViewWindow* theView, QKeyEvent* theEvent)
214 emit keyRelease(myView, theEvent);
217 //**********************************************
218 void NewGeom_SalomeViewer::onTryCloseView(SUIT_ViewWindow*)
220 emit tryCloseView(myView);
223 //**********************************************
224 void NewGeom_SalomeViewer::onDeleteView(SUIT_ViewWindow*)
226 if(myWindowScale.contains(myView->v3dView()))
227 myWindowScale.remove(myView->v3dView());
228 emit deleteView(myView);
231 //**********************************************
232 void NewGeom_SalomeViewer::onViewCreated(SUIT_ViewWindow* theView)
234 myView->setCurrentView(theView);
236 OCCViewer_ViewFrame* aView = dynamic_cast<OCCViewer_ViewFrame*>(theView);
238 OCCViewer_ViewWindow* aWnd = aView->getView(OCCViewer_ViewFrame::MAIN_VIEW);
240 connect(aWnd, SIGNAL(vpTransformationFinished(OCCViewer_ViewWindow::OperationType)),
241 this, SLOT(onViewTransformed(OCCViewer_ViewWindow::OperationType)));
243 myWindowScale.insert (aView->getViewPort()->getView(), aView->getViewPort()->getView()->Camera()->Scale());
245 emit viewCreated(myView);
248 //**********************************************
249 void NewGeom_SalomeViewer::onActivated(SUIT_ViewWindow* theView)
251 myView->setCurrentView(theView);
252 emit activated(myView);
255 //**********************************************
256 void NewGeom_SalomeViewer::enableSelection(bool isEnabled)
259 mySelector->viewer()->enableSelection(isEnabled);
260 // The enableSelection() in SALOME 7.5 cause of forced Viewer update(we have blinking)
261 // After this is corrected, the first row should be recommented, the last - removed
262 //mySelector->viewer()->setInteractionStyle(isEnabled ? SUIT_ViewModel::STANDARD
263 // : SUIT_ViewModel::KEY_FREE);
266 //**********************************************
267 bool NewGeom_SalomeViewer::isSelectionEnabled() const
270 return mySelector->viewer()->isSelectionEnabled();
273 //**********************************************
274 void NewGeom_SalomeViewer::enableMultiselection(bool isEnable)
277 mySelector->viewer()->enableMultiselection(isEnable);
280 //**********************************************
281 bool NewGeom_SalomeViewer::isMultiSelectionEnabled() const
284 return mySelector->viewer()->isMultiSelectionEnabled();
288 //**********************************************
289 void NewGeom_SalomeViewer::fitAll()
292 SUIT_ViewManager* aMgr = mySelector->viewer()->getViewManager();
293 OCCViewer_ViewFrame* aVFrame = dynamic_cast<OCCViewer_ViewFrame*>(aMgr->getActiveView());
300 //**********************************************
301 void NewGeom_SalomeViewer::setViewProjection(double theX, double theY, double theZ)
306 SUIT_ViewManager* aMgr = mySelector->viewer()->getViewManager();
307 OCCViewer_ViewFrame* aVFrame = dynamic_cast<OCCViewer_ViewFrame*>(aMgr->getActiveView());
309 Handle(V3d_View) aView3d = aVFrame->getViewPort()->getView();
310 if (!aView3d.IsNull()) {
311 aView3d->SetProj(theX, theY, theZ);
312 aView3d->FitAll(0.01, true);
313 aView3d->SetZSize(0.);
318 //***************************************
319 void NewGeom_SalomeViewer::addSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
321 Handle(AIS_InteractiveContext) aContext = AISContext();
322 if (!aContext.IsNull()) {
323 aContext->AddFilter(theFilter);
327 //***************************************
328 void NewGeom_SalomeViewer::removeSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
330 Handle(AIS_InteractiveContext) aContext = AISContext();
331 if (!aContext.IsNull()) {
332 aContext->RemoveFilter(theFilter);
336 //***************************************
337 void NewGeom_SalomeViewer::clearSelectionFilters()
339 Handle(AIS_InteractiveContext) aContext = AISContext();
340 if (!aContext.IsNull()) {
341 aContext->RemoveFilters();
345 //***************************************
346 void NewGeom_SalomeViewer::update()
348 Handle(AIS_InteractiveContext) aContext = AISContext();
349 if (!aContext.IsNull()) {
350 aContext->UpdateCurrentViewer();
354 //***************************************
355 void NewGeom_SalomeViewer::onViewTransformed(OCCViewer_ViewWindow::OperationType theType)
357 emit viewTransformed((int) theType);
360 //***************************************
361 void NewGeom_SalomeViewer::activateViewer(bool toActivate)
363 if (!mySelector || !mySelector->viewer())
365 SUIT_ViewManager* aMgr = mySelector->viewer()->getViewManager();
366 QVector<SUIT_ViewWindow*> aViews = aMgr->getViews();
368 foreach (SUIT_ViewWindow* aView, aViews) {
369 OCCViewer_ViewFrame* aOCCView = dynamic_cast<OCCViewer_ViewFrame*>(aView);
370 OCCViewer_ViewWindow* aWnd = aOCCView->getView(OCCViewer_ViewFrame::MAIN_VIEW);
371 connect(aWnd, SIGNAL(vpTransformationFinished(OCCViewer_ViewWindow::OperationType)),
372 this, SLOT(onViewTransformed(OCCViewer_ViewWindow::OperationType)));
375 foreach (SUIT_ViewWindow* aView, aViews) {
376 OCCViewer_ViewFrame* aOCCView = dynamic_cast<OCCViewer_ViewFrame*>(aView);
377 OCCViewer_ViewWindow* aWnd = aOCCView->getView(OCCViewer_ViewFrame::MAIN_VIEW);
378 disconnect((OCCViewer_ViewWindow*)aWnd, SIGNAL(vpTransformationFinished(OCCViewer_ViewWindow::OperationType)),
379 this, SLOT(onViewTransformed(OCCViewer_ViewWindow::OperationType)));
384 void NewGeom_SalomeViewer::Zfitall()
386 if (!mySelector || !mySelector->viewer())
388 SUIT_ViewManager* aMgr = mySelector->viewer()->getViewManager();
389 OCCViewer_ViewFrame* aView = dynamic_cast<OCCViewer_ViewFrame*>(aMgr->getActiveView());
391 OCCViewer_ViewWindow* aWnd = aView->getView(OCCViewer_ViewFrame::MAIN_VIEW);
392 aWnd->getViewPort()->getView()->ZFitAll();