src/NewGeom/NewGeom_SalomeViewer.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 #include "NewGeom_SalomeViewer.h"
   4 #include "NewGeom_OCCSelector.h"
   5
   6 #include <OCCViewer_ViewPort3d.h>
   7 #include <OCCViewer_ViewFrame.h>
   8
   9 #include <SUIT_ViewManager.h>
  10
  11 #include <QMouseEvent>
  12 #include <QContextMenuEvent>
  13
  14 NewGeom_SalomeView::NewGeom_SalomeView(OCCViewer_Viewer* theViewer)
  15 : ModuleBase_IViewWindow(), myCurrentView(0)
  16 {
  17   myViewer = theViewer;
  18 }
  19
  20
  21 Handle(V3d_View) NewGeom_SalomeView::v3dView() const
  22 {
  23   Handle(V3d_View) aView;
  24   if (myCurrentView) {
  25     OCCViewer_ViewWindow* aWnd = static_cast<OCCViewer_ViewWindow*>(myCurrentView);
  26     aView = aWnd->getViewPort()->getView();
  27   }
  28   return aView;
  29 }
  30
  31 QWidget* NewGeom_SalomeView::viewPort() const
  32 {
  33   QWidget* aViewPort = 0;
  34   if (myCurrentView) {
  35     OCCViewer_ViewWindow* aWnd = static_cast<OCCViewer_ViewWindow*>(myCurrentView);
  36     aViewPort = aWnd->getViewPort();
  37   }
  38   return aViewPort;
  39 }
  40
  41 //**********************************************
  42 //**********************************************
  43 //**********************************************
  44
  45
  46
  47 NewGeom_SalomeViewer::NewGeom_SalomeViewer(QObject* theParent)
  48     : ModuleBase_IViewer(theParent),
  49       mySelector(0), myView(0), myIsSelectionChanged(false)
  50 {
  51 }
  52
  53 NewGeom_SalomeViewer::~NewGeom_SalomeViewer()
  54 {
  55   if (myView)
  56     delete myView;
  57 }
  58
  59
  60 //**********************************************
  61 Handle(AIS_InteractiveContext) NewGeom_SalomeViewer::AISContext() const
  62 {
  63   if (mySelector && mySelector->viewer())
  64     return mySelector->viewer()->getAISContext();
  65   Handle(AIS_InteractiveContext) aNull;
  66   return aNull;
  67 }
  68
  69 //**********************************************
  70 Handle(V3d_Viewer) NewGeom_SalomeViewer::v3dViewer() const
  71 {
  72   if (mySelector)
  73     return mySelector->viewer()->getViewer3d();
  74   return Handle(V3d_Viewer)();
  75 }
  76
  77 //**********************************************
  78 Handle(V3d_View) NewGeom_SalomeViewer::activeView() const
  79 {
  80   if (mySelector) {
  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();
  85   }
  86   return Handle(V3d_View)();
  87 }
  88
  89 //**********************************************
  90 void NewGeom_SalomeViewer::setSelector(NewGeom_OCCSelector* theSel)
  91 {
  92   if (mySelector) {
  93     if (mySelector == theSel)
  94       return;
  95     else {
  96       mySelector->viewer()->getViewManager()->disconnect(this);
  97       OCCViewer_Viewer* aViewer = mySelector->viewer();
  98       if (aViewer)
  99         aViewer->disconnect(this);
 100     }
 101   }
 102   mySelector = theSel;
 103   if (!mySelector)
 104     return;
 105   OCCViewer_Viewer* aViewer = mySelector->viewer();
 106   SUIT_ViewManager* aMgr = aViewer->getViewManager();
 107
 108   myView = new NewGeom_SalomeView(mySelector->viewer());
 109
 110   // TODO: Provide ModuleBase_IViewWindow interface
 111   connect(aMgr, SIGNAL(lastViewClosed(SUIT_ViewManager*)), this, SIGNAL(lastViewClosed()));
 112
 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*)));
 121
 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*)));
 130
 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*)));
 135
 136   connect(aViewer, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
 137 }
 138
 139 //**********************************************
 140 void NewGeom_SalomeViewer::onSelectionChanged()
 141 {
 142   // Selection event must be sent only after mouse release
 143   myIsSelectionChanged = true;
 144 }
 145
 146 //**********************************************
 147 void NewGeom_SalomeViewer::onMousePress(SUIT_ViewWindow* theView, QMouseEvent* theEvent)
 148 {
 149   myView->setCurrentView(theView);
 150   emit mousePress(myView, theEvent);
 151 }
 152
 153 //**********************************************
 154 void NewGeom_SalomeViewer::onMouseRelease(SUIT_ViewWindow* theView, QMouseEvent* theEvent)
 155 {
 156   myView->setCurrentView(theView);
 157   emit mouseRelease(myView, theEvent);
 158   if (myIsSelectionChanged) {
 159     emit selectionChanged();
 160     myIsSelectionChanged = false;
 161   }
 162 }
 163
 164 //**********************************************
 165 void NewGeom_SalomeViewer::onMouseDoubleClick(SUIT_ViewWindow* theView, QMouseEvent* theEvent)
 166 {
 167   myView->setCurrentView(theView);
 168   emit mouseDoubleClick(myView, theEvent);
 169 }
 170
 171 //**********************************************
 172 void NewGeom_SalomeViewer::onMouseMove(SUIT_ViewWindow* theView, QMouseEvent* theEvent)
 173 {
 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);
 180 }
 181
 182 //**********************************************
 183 bool NewGeom_SalomeViewer::canDragByMouse() const
 184 {
 185   OCCViewer_Viewer* aViewer = mySelector->viewer();
 186   SUIT_ViewWindow* aWnd = aViewer->getViewManager()->getActiveView();
 187   OCCViewer_ViewWindow* aViewWnd = dynamic_cast<OCCViewer_ViewWindow*>(aWnd);
 188   if (aViewWnd) {
 189     return (aViewWnd->interactionStyle() == 0);
 190   }
 191   return true;
 192 }
 193
 194
 195 //**********************************************
 196 void NewGeom_SalomeViewer::onKeyPress(SUIT_ViewWindow* theView, QKeyEvent* theEvent)
 197 {
 198   OCCViewer_ViewWindow* aViewWnd = dynamic_cast<OCCViewer_ViewWindow*>(theView);
 199   Handle(AIS_InteractiveContext) aContext = AISContext();
 200   Handle(V3d_View) aView = aViewWnd->getViewPort()->getView();
 201
 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);
 207   }
 208   emit keyPress(myView, theEvent);
 209 }
 210
 211 //**********************************************
 212 void NewGeom_SalomeViewer::onKeyRelease(SUIT_ViewWindow* theView, QKeyEvent* theEvent)
 213 {
 214   emit keyRelease(myView, theEvent);
 215 }
 216
 217 //**********************************************
 218 void NewGeom_SalomeViewer::onTryCloseView(SUIT_ViewWindow*)
 219 {
 220   emit tryCloseView(myView);
 221 }
 222
 223 //**********************************************
 224 void NewGeom_SalomeViewer::onDeleteView(SUIT_ViewWindow*)
 225 {
 226   if(myWindowScale.contains(myView->v3dView()))
 227     myWindowScale.remove(myView->v3dView());
 228   emit deleteView(myView);
 229 }
 230
 231 //**********************************************
 232 void NewGeom_SalomeViewer::onViewCreated(SUIT_ViewWindow* theView)
 233 {
 234   myView->setCurrentView(theView);
 235
 236   OCCViewer_ViewFrame* aView = dynamic_cast<OCCViewer_ViewFrame*>(theView);
 237
 238   OCCViewer_ViewWindow* aWnd = aView->getView(OCCViewer_ViewFrame::MAIN_VIEW);
 239   if (aWnd)
 240     connect(aWnd, SIGNAL(vpTransformationFinished(OCCViewer_ViewWindow::OperationType)),
 241       this, SLOT(onViewTransformed(OCCViewer_ViewWindow::OperationType)));
 242
 243   myWindowScale.insert (aView->getViewPort()->getView(), aView->getViewPort()->getView()->Camera()->Scale());
 244
 245   emit viewCreated(myView);
 246 }
 247
 248 //**********************************************
 249 void NewGeom_SalomeViewer::onActivated(SUIT_ViewWindow* theView)
 250 {
 251   myView->setCurrentView(theView);
 252   emit activated(myView);
 253 }
 254
 255 //**********************************************
 256 void NewGeom_SalomeViewer::enableSelection(bool isEnabled)
 257 {
 258   if (mySelector)
 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);
 264 }
 265
 266 //**********************************************
 267 bool NewGeom_SalomeViewer::isSelectionEnabled() const
 268 {
 269   if (mySelector)
 270     return mySelector->viewer()->isSelectionEnabled();
 271 }
 272
 273 //**********************************************
 274 void NewGeom_SalomeViewer::enableMultiselection(bool isEnable)
 275 {
 276   if (mySelector)
 277     mySelector->viewer()->enableMultiselection(isEnable);
 278 }
 279
 280 //**********************************************
 281 bool NewGeom_SalomeViewer::isMultiSelectionEnabled() const
 282 {
 283   if (mySelector)
 284     return mySelector->viewer()->isMultiSelectionEnabled();
 285   return false;
 286 }
 287
 288 //**********************************************
 289 void NewGeom_SalomeViewer::fitAll()
 290 {
 291   if (mySelector) {
 292     SUIT_ViewManager* aMgr = mySelector->viewer()->getViewManager();
 293     OCCViewer_ViewFrame* aVFrame = dynamic_cast<OCCViewer_ViewFrame*>(aMgr->getActiveView());
 294     if (aVFrame) {
 295       aVFrame->onFitAll();
 296     }
 297   }
 298 }
 299
 300 //**********************************************
 301 void NewGeom_SalomeViewer::setViewProjection(double theX, double theY, double theZ)
 302 {
 303   if (!mySelector)
 304     return;
 305
 306   SUIT_ViewManager* aMgr = mySelector->viewer()->getViewManager();
 307   OCCViewer_ViewFrame* aVFrame = dynamic_cast<OCCViewer_ViewFrame*>(aMgr->getActiveView());
 308   if (aVFrame) {
 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.);
 314     }
 315   }
 316 }
 317
 318 //***************************************
 319 void NewGeom_SalomeViewer::addSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
 320 {
 321   Handle(AIS_InteractiveContext) aContext = AISContext();
 322   if (!aContext.IsNull()) {
 323     aContext->AddFilter(theFilter);
 324   }
 325 }
 326
 327 //***************************************
 328 void NewGeom_SalomeViewer::removeSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
 329 {
 330   Handle(AIS_InteractiveContext) aContext = AISContext();
 331   if (!aContext.IsNull()) {
 332     aContext->RemoveFilter(theFilter);
 333   }
 334 }
 335
 336 //***************************************
 337 void NewGeom_SalomeViewer::clearSelectionFilters()
 338 {
 339   Handle(AIS_InteractiveContext) aContext = AISContext();
 340   if (!aContext.IsNull()) {
 341     aContext->RemoveFilters();
 342   }
 343 }
 344
 345 //***************************************
 346 void NewGeom_SalomeViewer::update()
 347 {
 348   Handle(AIS_InteractiveContext) aContext = AISContext();
 349   if (!aContext.IsNull()) {
 350     aContext->UpdateCurrentViewer();
 351   }
 352 }
 353
 354 //***************************************
 355 void NewGeom_SalomeViewer::onViewTransformed(OCCViewer_ViewWindow::OperationType theType)
 356 {
 357   emit viewTransformed((int) theType);
 358 }