1 // Copyright (C) 2013 CEA/DEN, EDF R&D, OPEN CASCADE
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #include "CurveCreator_Widget.h"
21 #include "CurveCreator_TreeView.h"
22 #include "CurveCreator_ICurve.hxx"
23 #include "CurveCreator.hxx"
24 #include "CurveCreator_NewSectionDlg.h"
25 #include "CurveCreator_Utils.h"
26 #include "CurveCreator_UtilsICurve.hxx"
27 #include "CurveCreator_TableView.h"
29 #include <SUIT_Session.h>
30 #include <SUIT_Desktop.h>
31 #include <SUIT_ResourceMgr.h>
32 #include <SUIT_ViewManager.h>
34 #include <OCCViewer_ViewWindow.h>
35 #include <OCCViewer_ViewManager.h>
36 #include <OCCViewer_ViewPort3d.h>
37 #include "OCCViewer_Utilities.h"
39 #include <QHBoxLayout>
40 #include <QVBoxLayout>
44 #include <QToolButton>
48 #include <QMouseEvent>
49 #include <QApplication>
50 #include <QTableWidget>
53 //#define MEASURE_TIME
57 #define START_MEASURE_TIME \
61 #define END_MEASURE_TIME( theMsg ) \
62 double aTime = aTimer.elapsed() * 0.001; \
63 FILE* aFile = fopen( "performance", "a" ); \
64 fprintf( aFile, "%s = %.3lf sec\n", theMsg, aTime ); \
69 #define START_MEASURE_TIME
70 #define END_MEASURE_TIME( theMsg )
79 CurveCreator_Widget::CurveCreator_Widget(QWidget* parent,
80 CurveCreator_ICurve *theCurve,
81 const int theActionFlags,
83 int theLocalPointRowLimit )
84 : QWidget(parent), myNewSectionEditor(NULL), myCurve(theCurve), mySection(0),
85 myDragStarted( false ), myDragInteractionStyle( SUIT_ViewModel::STANDARD ),
86 myOCCViewer( 0 ), myLocalPointRowLimit( theLocalPointRowLimit )
88 bool isToEnableClosed = !( theActionFlags & DisableClosedSection );
89 myNewSectionEditor = new CurveCreator_NewSectionDlg( this, isToEnableClosed );
90 myNewSectionEditor->hide();
91 connect( myNewSectionEditor, SIGNAL(addSection()), this, SLOT(onAddNewSection()) );
92 connect( myNewSectionEditor, SIGNAL(modifySection()), this, SLOT(onModifySection()) );
93 connect( myNewSectionEditor, SIGNAL(cancelSection()), this, SLOT(onCancelSection()) );
95 QGroupBox* aSectionGroup = new QGroupBox(tr("Sections"),this);
97 mySectionView = new CurveCreator_TreeView(myCurve, aSectionGroup);
98 mySectionView->setSelectionMode( QTreeView::ExtendedSelection );
99 connect( mySectionView, SIGNAL(selectionChanged()), this, SLOT( onSelectionChanged() ) );
100 connect( mySectionView, SIGNAL(sectionEntered(int)), this, SLOT(onEditSection(int)) );
101 connect( mySectionView, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(onContextMenu(QPoint)) );
103 myLocalPointView = new CurveCreator_TableView( myCurve, this );
104 connect( myLocalPointView, SIGNAL( cellChanged( int, int ) ),
105 this, SLOT( onCellChanged( int, int ) ) );
107 QToolBar* aTB = new QToolBar(tr("TOOL_BAR_TLT"), aSectionGroup);
108 // QToolButton* anUndoBtn = new QToolButton(aTB);
110 SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
111 QPixmap anUndoPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_UNDO")));
112 QPixmap aRedoPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_REDO")));
113 QPixmap aNewSectionPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_NEW_SECTION")));
114 QPixmap aNewPointPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_NEW_POINT")));
115 QPixmap anEditPointsPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_EDIT_POINTS")));
116 QPixmap aDetectPointsPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_EDIT_POINTS")));
117 QPixmap aPolylinePixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_POLYLINE")));
118 QPixmap aSplinePixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_SPLINE")));
119 QPixmap aRemovePixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_DELETE")));
120 QPixmap aJoinPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_JOIN")));
121 QPixmap aStepUpPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_ARROW_UP")));
122 QPixmap aStepDownPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_ARROW_DOWN")));
124 QAction* anAct = createAction( UNDO_ID, tr("UNDO"), anUndoPixmap, tr("UNDO_TLT"),
125 QKeySequence(Qt::ControlModifier|Qt::Key_Z) );
126 connect(anAct, SIGNAL(triggered()), this, SLOT(onUndo()) );
127 aTB->addAction(anAct);
129 anAct = createAction( REDO_ID, tr("REDO"), aRedoPixmap, tr("REDO_TLT"),
130 QKeySequence(Qt::ControlModifier|Qt::Key_Y) );
131 connect(anAct, SIGNAL(triggered()), this, SLOT(onRedo()) );
132 aTB->addAction(anAct);
136 anAct = createAction( NEW_SECTION_ID, tr("NEW_SECTION"), aNewSectionPixmap, tr("NEW_SECTION_TLT"),
137 QKeySequence(Qt::ControlModifier|Qt::Key_N) );
138 connect(anAct, SIGNAL(triggered()), this, SLOT(onNewSection()) );
139 if ( !(theActionFlags & DisableNewSection) ) {
140 aTB->addAction(anAct);
144 anAct = createAction( ADDITION_MODE_ID, tr("ADDITION_MODE"), aNewPointPixmap, tr("ADDITION_MODE_TLT"),
146 anAct->setCheckable(true);
147 connect(anAct, SIGNAL(triggered(bool)), this, SLOT(onAdditionMode(bool)) );
148 connect(anAct, SIGNAL(toggled(bool)), this, SLOT(onModeChanged(bool)) );
149 aTB->addAction(anAct);
151 anAct = createAction( MODIFICATION_MODE_ID, tr("MODIFICATION_MODE"), anEditPointsPixmap, tr("MODIFICATION_MODE_TLT"),
153 anAct->setCheckable(true);
154 connect(anAct, SIGNAL(triggered(bool)), this, SLOT(onModificationMode(bool)) );
155 connect(anAct, SIGNAL(toggled(bool)), this, SLOT(onModeChanged(bool)) );
156 aTB->addAction(anAct);
158 anAct = createAction( DETECTION_MODE_ID, tr("DETECTION_MODE"), aDetectPointsPixmap, tr("DETECTION_MODE_TLT"),
160 anAct->setCheckable(true);
161 connect(anAct, SIGNAL(triggered(bool)), this, SLOT(onDetectionMode(bool)) );
162 connect(anAct, SIGNAL(toggled(bool)), this, SLOT(onModeChanged(bool)) );
163 if ( !(theActionFlags & DisableDetectionMode) ) {
164 aTB->addAction(anAct);
167 anAct = createAction( CLOSE_SECTIONS_ID, tr("CLOSE_SECTIONS"), QPixmap(), tr("CLOSE_SECTIONS_TLT"),
168 QKeySequence(Qt::ControlModifier|Qt::Key_W) );
169 connect(anAct, SIGNAL(triggered()), this, SLOT(onCloseSections()) );
171 anAct = createAction( UNCLOSE_SECTIONS_ID, tr("UNCLOSE_SECTIONS"), QPixmap(),
172 tr("UNCLOSE_SECTIONS_TLT"), QKeySequence(Qt::ControlModifier|Qt::Key_S) );
173 connect(anAct, SIGNAL(triggered()), this, SLOT(onUncloseSections()) );
175 anAct = createAction( SET_SECTIONS_POLYLINE_ID, tr("SET_SECTIONS_POLYLINE"),
176 aPolylinePixmap, tr("SET_POLYLINE_TLT"),
177 QKeySequence(Qt::ControlModifier|Qt::Key_E) );
178 connect(anAct, SIGNAL(triggered()), this, SLOT(onSetPolyline()) );
180 anAct = createAction( SET_SECTIONS_SPLINE_ID, tr("SET_SECTIONS_SPLINE"), aSplinePixmap,
181 tr("SET_SPLINE_TLT"), QKeySequence(Qt::ControlModifier|Qt::Key_R) );
182 connect(anAct, SIGNAL(triggered()), this, SLOT(onSetSpline()) );
184 anAct = createAction( REMOVE_ID, tr("REMOVE"), aRemovePixmap, tr("REMOVE_TLT"),
185 QKeySequence(Qt::ControlModifier|Qt::Key_Delete ) );
186 connect(anAct, SIGNAL(triggered()), this, SLOT(onRemove()) );
187 aTB->addAction(anAct);
191 anAct = createAction( JOIN_ID, tr("JOIN"), aJoinPixmap, tr("JOIN_TLT"),
192 QKeySequence(Qt::ControlModifier|Qt::Key_Plus ) );
193 connect( anAct, SIGNAL(triggered()), this, SLOT(onJoin()) );
194 aTB->addAction(anAct);
196 anAct = createAction( CLEAR_ALL_ID, tr("CLEAR_ALL"), QPixmap(), tr("CLEAR_ALL_TLT"),
197 QKeySequence(Qt::ControlModifier | Qt::ShiftModifier | Qt::Key_Delete ) );
198 connect( anAct, SIGNAL(triggered()), this, SLOT( onClearAll()) );
200 anAct = createAction( JOIN_ALL_ID, tr("JOIN_ALL"), QPixmap(), tr("JOIN_ALL_TLT"),
201 QKeySequence(Qt::ControlModifier | Qt::ShiftModifier | Qt::Key_Plus ) );
202 connect( anAct, SIGNAL(triggered()), this, SLOT(onJoinAll()) );
204 QVBoxLayout* aSectLayout = new QVBoxLayout();
205 aSectLayout->setMargin( 5 );
206 aSectLayout->setSpacing( 5 );
207 aSectLayout->addWidget(aTB);
208 aSectLayout->addWidget(mySectionView);
209 aSectLayout->addWidget( myLocalPointView );
210 aSectionGroup->setLayout(aSectLayout);
211 QVBoxLayout* aLay = new QVBoxLayout();
212 aLay->setMargin( 0 );
213 aLay->setSpacing( 5 );
214 // aLay->addLayout(aNameLayout);
215 aLay->addWidget(aSectionGroup);
218 updateActionsStates();
225 void CurveCreator_Widget::setOCCViewer( OCCViewer_Viewer* theViewer )
227 if ( myOCCViewer == theViewer )
231 OCCViewer_ViewManager* aViewManager = dynamic_cast<OCCViewer_ViewManager*>
232 ( myOCCViewer->getViewManager() );
233 disconnect( aViewManager, SIGNAL( mousePress( SUIT_ViewWindow*, QMouseEvent* ) ),
234 this, SLOT( onMousePress( SUIT_ViewWindow*, QMouseEvent* ) ) );
235 disconnect( aViewManager, SIGNAL( mouseRelease( SUIT_ViewWindow*, QMouseEvent* ) ),
236 this, SLOT( onMouseRelease( SUIT_ViewWindow*, QMouseEvent* ) ) );
237 disconnect( aViewManager, SIGNAL( mouseMove( SUIT_ViewWindow*, QMouseEvent* ) ),
238 this, SLOT( onMouseMove( SUIT_ViewWindow*, QMouseEvent* ) ) );
239 disconnect( aViewManager, SIGNAL( lastViewClosed( SUIT_ViewManager* ) ),
240 this, SLOT( onLastViewClosed( SUIT_ViewManager* ) ) );
241 // restore normal mode in the viewer
242 OCCViewer_Utilities::setViewer2DMode( myOCCViewer, OCCViewer_ViewWindow::No2dMode );
243 // all local contexts should be closed if the viewer is not more used
244 setLocalPointContext( false, true );
247 myOCCViewer = theViewer;
249 OCCViewer_ViewManager* aViewManager = dynamic_cast<OCCViewer_ViewManager*>
250 ( myOCCViewer->getViewManager() );
251 connect( aViewManager, SIGNAL( mousePress( SUIT_ViewWindow*, QMouseEvent* ) ),
252 this, SLOT( onMousePress( SUIT_ViewWindow*, QMouseEvent* ) ) );
253 connect( aViewManager, SIGNAL( mouseRelease( SUIT_ViewWindow*, QMouseEvent* ) ),
254 this, SLOT( onMouseRelease( SUIT_ViewWindow*, QMouseEvent* ) ) );
255 connect( aViewManager, SIGNAL( mouseMove( SUIT_ViewWindow*, QMouseEvent* ) ),
256 this, SLOT( onMouseMove( SUIT_ViewWindow*, QMouseEvent* ) ) );
257 connect( aViewManager, SIGNAL( lastViewClosed( SUIT_ViewManager* ) ),
258 this, SLOT( onLastViewClosed( SUIT_ViewManager* ) ) );
259 OCCViewer_Utilities::setViewer2DMode( myOCCViewer, OCCViewer_ViewWindow::XYPlane );
264 * Returns current OCC viewer
266 OCCViewer_Viewer* CurveCreator_Widget::getOCCViewer()
272 * Returns OCC viewer context
274 Handle(AIS_InteractiveContext) CurveCreator_Widget::getAISContext()
276 Handle(AIS_InteractiveContext) aContext;
277 OCCViewer_Viewer* aViewer = getOCCViewer();
279 aContext = aViewer->getAISContext();
285 * Returns OCC viewer view port
287 OCCViewer_ViewPort3d* CurveCreator_Widget::getViewPort()
289 OCCViewer_ViewPort3d* aViewPort = 0;
290 OCCViewer_Viewer* aViewer = getOCCViewer();
292 aViewPort = ((OCCViewer_ViewWindow*)aViewer->getViewManager()->getActiveView())->getViewPort();
298 * Set interaction style in the OCC viewer
299 * \param theStyle a new style
300 * \return the previous style
302 int CurveCreator_Widget::changeInteractionStyle( int theStyle )
304 OCCViewer_Viewer* aViewer = getOCCViewer();
308 int aPrevStyle = aViewer->interactionStyle();
309 aViewer->setInteractionStyle( theStyle );
314 //=======================================================================
316 // purpose: reset the widget viewer, close local context, clear selection
317 //=======================================================================
318 void CurveCreator_Widget::reset()
322 void CurveCreator_Widget::setCurve( CurveCreator_ICurve* theCurve )
325 mySectionView->setCurve( myCurve );
326 myLocalPointView->setCurve( myCurve );
327 updateActionsStates();
331 void CurveCreator_Widget::onSelectionChanged()
333 updateActionsStates();
335 emit selectionChanged();
338 void CurveCreator_Widget::updateActionsStates()
340 QList<ActionId> anEnabledAct;
342 anEnabledAct << NEW_SECTION_ID << MODIFICATION_MODE_ID;
343 if ( removeEnabled() )
344 anEnabledAct << REMOVE_ID;
345 QList<int> aSelSections = mySectionView->getSelectedSections();
346 CurveCreator_TreeView::SelectionType aSelType = mySectionView->getSelectionType();
348 case CurveCreator_TreeView::ST_NOSEL:{
351 case CurveCreator_TreeView::ST_SECTIONS:{
352 /*if( aSelSections[0] > 0 ){
353 anEnabledAct << UP_ID;
355 if( aSelSections.size() == 1 ){
356 anEnabledAct << ADDITION_MODE_ID << DETECTION_MODE_ID;
358 switch ( getActionMode() ) {
362 QList<int> aSelSection = mySectionView->getSelectedSections();
363 if( aSelSection.size() > 0 ){
364 mySection = aSelSection[0];
365 myPointNum = myCurve->getNbPoints(mySection);
369 case ModificationMode: {
370 anEnabledAct << CLOSE_SECTIONS_ID << UNCLOSE_SECTIONS_ID << SET_SECTIONS_POLYLINE_ID << SET_SECTIONS_SPLINE_ID;
371 int aSectCnt = myCurve->getNbSections();
373 anEnabledAct << CLEAR_ALL_ID;
375 anEnabledAct << JOIN_ALL_ID;
376 if( aSelSections.size() > 1 ){
377 anEnabledAct << JOIN_ID;
381 case DetectionMode: {
386 int aSectCnt = myCurve->getNbSections();
388 anEnabledAct << JOIN_ALL_ID;
389 if( aSelSections.size() > 1 )
390 anEnabledAct << JOIN_ID;
396 /*if( aSelSections[ aSelSections.size() - 1 ] < ( myCurve->getNbSections() - 1 ) ){
397 anEnabledAct << DOWN_ID;
401 /*case CurveCreator_TreeView::ST_POINTS_ONE_SECTION:{
402 if( aSelPoints[0].second > 0 ){
403 anEnabledAct << UP_ID;
405 int aLastIndex = aSelPoints.size()-1;
406 int aSect = aSelPoints[0].first;
407 if( aSelPoints[aLastIndex].second < (myCurve->getNbPoints(aSect) - 1)){
408 anEnabledAct << DOWN_ID;
410 if( aSelPoints.size() == 1){
411 anEnabledAct << INSERT_POINT_BEFORE_ID << INSERT_POINT_AFTER_ID;
418 /*int aSelObjsCnt = aSelPoints.size() + aSelSections.size();
419 if( aSelObjsCnt > 0 ){
420 anEnabledAct << REMOVE_ID;
422 if( (myCurve->getNbSections() + myCurve->getNbPoints()) > 0 ){
423 anEnabledAct << REMOVE_ALL_ID;
425 if( myCurve->getNbSections() > 1 ){
426 anEnabledAct << JOIN_ALL_ID;
429 QList<ActionId> anIds = myActionMap.keys();
430 for( int i = 0 ; i < anIds.size() ; i++ ){
431 if( myActionMap.contains(anIds[i]) ){
432 if( anEnabledAct.contains(anIds[i]) ){
433 myActionMap[anIds[i]]->setEnabled(true);
436 myActionMap[anIds[i]]->setEnabled(false);
442 void CurveCreator_Widget::onAdditionMode(bool checked)
447 Handle(AIS_InteractiveContext) aContext = getAISContext();
448 if( !myCurve || aContext.IsNull() )
453 QList<int> aSelSection = mySectionView->getSelectedSections();
454 if( aSelSection.size() > 0 ){
455 mySection = aSelSection[0];
457 // emit subOperationStarted( myNewPointEditor );
460 void CurveCreator_Widget::onModificationMode(bool checked)
462 myLocalPointView->setVisible( checked );
465 void CurveCreator_Widget::onDetectionMode(bool checked)
469 void CurveCreator_Widget::onModeChanged(bool checked)
471 ActionMode aMode = NoneMode;
473 QAction* anAction = (QAction*)sender();
474 switch(myActionMap.key(anAction)) {
475 case ADDITION_MODE_ID:
476 aMode = AdditionMode;
477 if (myActionMap[MODIFICATION_MODE_ID]->isChecked())
478 myActionMap[MODIFICATION_MODE_ID]->trigger();
479 else if (myActionMap[DETECTION_MODE_ID]->isChecked())
480 myActionMap[DETECTION_MODE_ID]->trigger();
482 case MODIFICATION_MODE_ID:
483 aMode = ModificationMode;
484 if (myActionMap[ADDITION_MODE_ID]->isChecked())
485 myActionMap[ADDITION_MODE_ID]->trigger();
486 else if (myActionMap[DETECTION_MODE_ID]->isChecked())
487 myActionMap[DETECTION_MODE_ID]->trigger();
489 case DETECTION_MODE_ID:
490 aMode = DetectionMode;
491 if (myActionMap[ADDITION_MODE_ID]->isChecked())
492 myActionMap[ADDITION_MODE_ID]->trigger();
493 else if (myActionMap[MODIFICATION_MODE_ID]->isChecked())
494 myActionMap[MODIFICATION_MODE_ID]->trigger();
498 updateActionsStates();
500 setLocalPointContext( aMode == ModificationMode, true );
503 void CurveCreator_Widget::onNewSection()
509 myNewSectionEditor->clear();
510 myNewSectionEditor->setEditMode(false);
511 QString aSectName = QString( CurveCreator_UtilsICurve::getUniqSectionName( myCurve ).c_str() );
512 myNewSectionEditor->setSectionParameters(aSectName, true, CurveCreator::Polyline );
513 emit subOperationStarted( myNewSectionEditor, false );
516 void CurveCreator_Widget::onAddNewSection()
520 myCurve->addSection( myNewSectionEditor->getName().toStdString(),
521 myNewSectionEditor->getSectionType(),
522 myNewSectionEditor->isClosed() );
523 mySectionView->sectionAdded( -1 ); // add a new section to the end of list
524 QString aNewName = QString( CurveCreator_UtilsICurve::getUniqSectionName( myCurve ).c_str() );
525 myNewSectionEditor->setSectionName(aNewName);
526 updateActionsStates();
531 void CurveCreator_Widget::onCancelSection()
533 emit subOperationFinished( myNewSectionEditor );
536 QAction* CurveCreator_Widget::createAction( ActionId theId, const QString& theName, const QPixmap& theImage,
537 const QString& theToolTip, const QKeySequence& theShortcut )
539 QAction* anAct = new QAction(theName,this);
540 if( !theImage.isNull() ){
541 anAct->setIcon(theImage);
543 anAct->setShortcut(theShortcut);
544 anAct->setToolTip(theToolTip);
545 myActionMap[theId] = anAct;
549 QAction* CurveCreator_Widget::getAction( ActionId theId )
551 if( myActionMap.contains(theId) )
552 return myActionMap[theId];
556 QAction* CurveCreator_Widget::getAction( ActionMode theMode )
558 ActionId anActionId = NONE_ID;
561 anActionId = ADDITION_MODE_ID;
563 case ModificationMode:
564 anActionId = MODIFICATION_MODE_ID;
567 anActionId = DETECTION_MODE_ID;
572 QAction* anAction = 0;
573 if ( anActionId != NONE_ID && myActionMap.contains( anActionId ) )
574 anAction = myActionMap[anActionId];
578 void CurveCreator_Widget::setActionMode( const ActionMode& theMode )
580 ActionMode aPrevMode = getActionMode();
581 QAction* aPrevAction = getAction( aPrevMode );
585 if ( aPrevAction->isChecked() )
586 aPrevAction->setChecked( false );
588 if ( aPrevMode == ModificationMode )
589 onModificationMode( false );
590 if ( aPrevMode == AdditionMode )
591 onAdditionMode( false );
598 case ModificationMode:
608 CurveCreator_Widget::ActionMode CurveCreator_Widget::getActionMode() const
610 ActionMode aMode = NoneMode;
612 if ( myActionMap[ADDITION_MODE_ID]->isChecked() )
613 aMode = AdditionMode;
614 else if ( myActionMap[MODIFICATION_MODE_ID]->isChecked() )
615 aMode = ModificationMode;
616 else if ( myActionMap[DETECTION_MODE_ID]->isChecked() )
617 aMode = DetectionMode;
622 void CurveCreator_Widget::onEditSection( int theSection )
628 mySection = theSection;
629 QString aSectName = QString::fromStdString( myCurve->getSectionName(theSection));
630 bool isClosed = myCurve->isClosed(theSection);
631 CurveCreator::SectionType aType = myCurve->getSectionType(theSection);
632 myNewSectionEditor->setEditMode(true);
633 myNewSectionEditor->setSectionParameters( aSectName, isClosed, aType );
635 emit subOperationStarted( myNewSectionEditor, true );
638 void CurveCreator_Widget::onModifySection()
642 QString aName = myNewSectionEditor->getName();
643 bool isClosed = myNewSectionEditor->isClosed();
644 CurveCreator::SectionType aSectType = myNewSectionEditor->getSectionType();
645 if( myCurve->getSectionName(mySection) != aName.toStdString() )
646 myCurve->setSectionName( mySection , aName.toStdString() );
648 if( myCurve->getSectionType(mySection) != aSectType )
649 myCurve->setSectionType( mySection, aSectType );
651 if( myCurve->isClosed(mySection) != isClosed )
652 myCurve->setClosed( mySection, isClosed );
653 mySectionView->sectionChanged(mySection);
658 void CurveCreator_Widget::onJoin()
662 QList<int> aSections = mySectionView->getSelectedSections();
663 if( aSections.size() == 0 ){
668 std::list<int> aSectionsToJoin;
669 for( int i = 0; i < aSections.size() ; i++ ){
670 aSectionsToJoin.push_back( aSections[i] );
672 //int aMainSect = aSectionsToJoin.front();
673 //int aMainSectSize = myCurve->getNbPoints(aMainSect);
674 if ( myCurve->join( aSectionsToJoin ) )
676 std::list<int>::const_iterator anIt = aSectionsToJoin.begin(),
677 aLast = aSectionsToJoin.end();
678 // the first section should be skipped. It is not removed, but is modified
680 for ( ; anIt != aLast; anIt++ )
681 mySectionView->sectionsRemoved( *anIt );
684 /* The update for the points of the main section
685 int aNewSectSize = myCurve->getNbPoints(aMainSect);
686 if( aNewSectSize != aMainSectSize )
687 mySectionView->pointsAdded( aMainSect, aMainSectSize, aNewSectSize-aMainSectSize );*/
691 void CurveCreator_Widget::onRemove()
696 switch( getActionMode() ) {
700 case ModificationMode:
708 void CurveCreator_Widget::onClearAll()
714 mySectionView->reset();
715 updateActionsStates();
719 void CurveCreator_Widget::onJoinAll()
725 std::list<int> aSectionsToJoin;
726 for( int i = 0, aNb = myCurve->getNbSections(); i < aNb ; i++ ){
727 aSectionsToJoin.push_back( i );
729 bool aRes = myCurve->join( aSectionsToJoin );
731 mySectionView->reset();
732 updateActionsStates();
736 void CurveCreator_Widget::onUndoSettings()
741 void CurveCreator_Widget::onSetSpline()
746 QList<int> aSelSections = mySectionView->getSelectedSections();
747 for( int i = 0 ; i < aSelSections.size() ; i++ ){
748 myCurve->setSectionType(aSelSections[i], CurveCreator::Spline );
749 mySectionView->sectionChanged(aSelSections[i]);
754 void CurveCreator_Widget::onSetPolyline()
759 QList<int> aSelSections = mySectionView->getSelectedSections();
760 for( int i = 0 ; i < aSelSections.size() ; i++ ){
761 myCurve->setSectionType( aSelSections[i], CurveCreator::Polyline );
762 mySectionView->sectionChanged( aSelSections[i] );
767 void CurveCreator_Widget::onCloseSections()
772 QList<int> aSelSections = mySectionView->getSelectedSections();
773 for( int i = 0 ; i < aSelSections.size() ; i++ ){
774 myCurve->setClosed(aSelSections[i], true);
775 mySectionView->sectionChanged(aSelSections[i]);
780 void CurveCreator_Widget::onUncloseSections()
785 QList<int> aSelSections = mySectionView->getSelectedSections();
786 for( int i = 0 ; i < aSelSections.size() ; i++ ){
787 myCurve->setClosed(aSelSections[i], false);
788 mySectionView->sectionChanged(aSelSections[i]);
793 void CurveCreator_Widget::onUndo()
798 CurveCreator_ICurve::SectionToPointList aPoints;
799 startCurveModification( aPoints, false );
801 finishCurveModification();
802 mySectionView->reset();
805 void CurveCreator_Widget::onRedo()
809 CurveCreator_ICurve::SectionToPointList aPoints;
810 startCurveModification( aPoints, false );
812 finishCurveModification();
813 mySectionView->reset();
816 void CurveCreator_Widget::updateUndoRedo()
820 QAction* anAct = myActionMap[UNDO_ID];
822 if( myCurve->getNbUndo() != 0 ){
823 anAct->setEnabled(true);
826 anAct->setDisabled(true);
829 anAct = myActionMap[REDO_ID];
831 if( myCurve->getNbRedo() != 0 ){
832 anAct->setEnabled(true);
835 anAct->setDisabled(true);
840 void CurveCreator_Widget::onContextMenu( QPoint thePoint )
842 QList<ActionId> aContextActions;
843 aContextActions << CLEAR_ALL_ID << JOIN_ID << JOIN_ALL_ID << SEPARATOR_ID <<
844 CLOSE_SECTIONS_ID << UNCLOSE_SECTIONS_ID << SET_SECTIONS_POLYLINE_ID <<
845 SET_SECTIONS_SPLINE_ID;
846 QPoint aGlPoint = mySectionView->mapToGlobal(thePoint);
848 QList<ActionId> aResAct;
849 for( int i = 0 ; i < aContextActions.size() ; i++ ){
850 if( aContextActions[i] != SEPARATOR_ID ){
851 if( myActionMap.contains(aContextActions[i]) ){
852 QAction* anAct = myActionMap[aContextActions[i]];
853 if( anAct->isEnabled() ){
854 aResAct << aContextActions[i];
860 aResAct << SEPARATOR_ID;
866 QMenu* aMenu = new QMenu(this);
867 for( int i = 0 ; i < aResAct.size() ; i++ ){
868 if( aResAct[i] == SEPARATOR_ID ){
869 aMenu->addSeparator();
872 QAction* anAct = myActionMap[aResAct[i]];
873 aMenu->insertAction(NULL, anAct);
876 aMenu->exec(aGlPoint);
879 QList<int> CurveCreator_Widget::getSelectedSections()
881 return mySectionView->getSelectedSections();
885 * According to the widget state, performs the remove action
887 void CurveCreator_Widget::removeSelected()
893 * Checks whether there are some selection to be removed
895 bool CurveCreator_Widget::removeEnabled()
897 bool isEnabled = getActionMode() == ModificationMode;
899 QList<int> aSelSections = mySectionView->getSelectedSections();
900 CurveCreator_TreeView::SelectionType aSelType = mySectionView->getSelectionType();
901 isEnabled = aSelType == CurveCreator_TreeView::ST_SECTIONS &&
902 aSelSections.size() == 1;
908 //=================================================================================
909 // function : GeometryGUI::addCoordsByClick()
910 // purpose : Manage mouse press events in Additon mode
911 //=================================================================================
912 void CurveCreator_Widget::addCoordsByClick( QMouseEvent* pe )
914 if (pe->button() != Qt::LeftButton)
917 if ( pe->modifiers() != Qt::ControlModifier ) {
918 Handle(AIS_InteractiveContext) ic = getAISContext();
925 if ( pe->modifiers() == Qt::ShiftModifier )
926 ic->ShiftSelect(); // Append selection
928 ic->Select(); // New selection
931 OCCViewer_ViewPort3d* vp = getViewPort();
932 aPnt = CurveCreator_Utils::ConvertClickToPoint( pe->x(), pe->y(), vp->getView() );
934 // set the coordinates into dialog
935 CurveCreator::Coordinates aCoords;
936 aCoords.push_back( aPnt.X() );
937 aCoords.push_back( aPnt.Y() );
938 if ( myCurve->getDimension() == 3 ) {
939 aCoords.push_back( aPnt.Z() );
941 addNewPoint(aCoords);
946 * Manage mouse press events
947 * \param theWindow an owner of the signal
948 * \param theEvent a mouse event
950 void CurveCreator_Widget::onMousePress( SUIT_ViewWindow*, QMouseEvent* theEvent )
952 if ( theEvent->button() != Qt::LeftButton )
955 switch( getActionMode() ) {
956 case ModificationMode: {
957 //store initial cursor position for Drag&Drop
958 setDragStarted( true, theEvent->pos() );
962 addCoordsByClick( theEvent );
971 * Manage mouse release events in Modification mode
972 * \param theWindow an owner of the signal
973 * \param theEvent a mouse event
975 void CurveCreator_Widget::onMouseRelease( SUIT_ViewWindow*, QMouseEvent* theEvent )
977 if ( getActionMode() != ModificationMode )
980 if ( myDragStarted ) {
981 bool isDragged = myDragged;
982 CurveCreator_ICurve::SectionToPointList aDraggedPoints;
983 QMap<CurveCreator_ICurve::SectionToPoint, std::deque< float > > anInitialDragPointsCoords;
985 aDraggedPoints = myDragPoints;
986 anInitialDragPointsCoords = myInitialDragPointsCoords;
989 setDragStarted( false );
991 if ( aDraggedPoints.size() > 0 ) {
992 // Collect old coordinates of the dragged points
993 CurveCreator_ICurve::SectionToPointCoordsList anOldPoints;
994 foreach ( const CurveCreator_ICurve::SectionToPoint aSectionToPoint, anInitialDragPointsCoords.keys() ) {
995 CurveCreator::Coordinates aCoords = anInitialDragPointsCoords.value( aSectionToPoint );
996 anOldPoints.push_back( std::make_pair( aSectionToPoint, aCoords ) );
999 if ( myCurve->canPointsBeSorted() ) {
1000 // Add old coordinates of the curve points (except the dragged points) to the list
1001 for( int aSectionId = 0 ; aSectionId < myCurve->getNbSections() ; aSectionId++ ) {
1002 CurveCreator::Coordinates aCoords;
1003 for ( int aPointId = 0, aNb = myCurve->getNbPoints( aSectionId ); aPointId < aNb; aPointId++ ) {
1004 aCoords = myCurve->getPoint( aSectionId, aPointId );
1005 if ( aCoords.size() < 2 ) {
1009 CurveCreator_ICurve::SectionToPoint aSectionToPoint = std::make_pair( aSectionId, aPointId );
1011 if ( !anInitialDragPointsCoords.contains( aSectionToPoint ) ) {
1012 anOldPoints.push_back( std::make_pair( aSectionToPoint, aCoords ) );
1017 // Apply points sorting
1018 CurveCreator_ICurve::SectionToPointList aPoints;
1019 startCurveModification( aPoints, false );
1021 myCurve->setSkipSorting( false );
1023 CurveCreator_ICurve::SectionToPointCoordsList aCoordList;
1024 CurveCreator_ICurve::SectionToPointList::const_iterator anIt = aDraggedPoints.begin(),
1025 aLast = aDraggedPoints.end();
1026 for ( ; anIt != aLast; anIt++ ) {
1027 int aSectionId = anIt->first;
1028 int aPointId = anIt->second;
1029 std::deque<float> aPos = myCurve->getPoint( aSectionId, aPointId );
1031 aCoordList.push_back(
1032 std::make_pair( std::make_pair( aSectionId, aPointId ), aPos ) );
1035 myCurve->setSeveralPoints( aCoordList, false );
1037 finishCurveModification( aDraggedPoints );
1039 // if the drag of some points has happened, restore the drag selection
1041 setSelectedPoints( aDraggedPoints );
1042 END_MEASURE_TIME( "drop" );
1045 // Save drag difference
1046 myCurve->saveCoordDiff( anOldPoints );
1049 else // check whether the segment is clicked an a new point should be added to the segment
1050 insertPointToSelectedSegment( theEvent->pos().x(), theEvent->pos().y() );
1052 // updates the input panel table to show the selected point coordinates
1053 updateLocalPointView();
1057 * Manage mouse move events in Modification mode
1058 * \param theWindow an owner of the signal
1059 * \param theEvent a mouse event
1061 void CurveCreator_Widget::onMouseMove( SUIT_ViewWindow*, QMouseEvent* theEvent )
1063 if ( getActionMode() != ModificationMode || !myDragStarted )
1066 QPoint aPos = theEvent->pos();
1067 if ( (aPos - myDragStartPosition).manhattanLength() < QApplication::startDragDistance() )
1072 moveSelectedPoints( aPos.x(), aPos.y() );
1073 myDragStartPosition = aPos;
1075 END_MEASURE_TIME( "drag" );
1079 * Set zero viewer by the last view closed in
1080 * \param theManager a viewer manager
1082 void CurveCreator_Widget::onLastViewClosed( SUIT_ViewManager* theManager )
1087 void CurveCreator_Widget::onMousePress( QMouseEvent* theEvent )
1089 onMousePress( 0, theEvent );
1092 void CurveCreator_Widget::onMouseRelease( QMouseEvent* theEvent )
1094 onMouseRelease( 0, theEvent );
1097 void CurveCreator_Widget::onMouseMove( QMouseEvent* theEvent )
1099 onMouseMove( 0, theEvent );
1102 void CurveCreator_Widget::onCellChanged( int theRow, int theColumn )
1104 int aCurrSect = myLocalPointView->getSectionId( theRow );
1105 int aPntIndex = myLocalPointView->getPointId( theRow );
1107 if ( aPntIndex < 0 )
1110 CurveCreator_ICurve::SectionToPointList aSelPoints;
1111 startCurveModification( aSelPoints );
1113 double aX = myLocalPointView->item( theRow, 2 )->data( Qt::UserRole ).toDouble();
1114 double anY = myLocalPointView->item( theRow, 3 )->data( Qt::UserRole ).toDouble();
1115 std::deque<float> aChangedPos;
1116 aChangedPos.push_back( aX );
1117 aChangedPos.push_back( anY );
1118 myCurve->setPoint( aCurrSect, aPntIndex, aChangedPos );
1120 finishCurveModification( aSelPoints );
1124 * Removes a selected section from the curve. Updates undo/redo status
1126 void CurveCreator_Widget::removeSection()
1130 QList<int> aSections = mySectionView->getSelectedSections();
1131 for( int i = 0 ; i < aSections.size() ; i++ ){
1132 int aSectNum = aSections[i] - (i);
1133 myCurve->removeSection( aSectNum );
1134 mySectionView->sectionsRemoved( aSectNum );
1136 mySectionView->clearSelection();
1141 * Removes a selected points from the curve. Updates undo/redo status
1143 void CurveCreator_Widget::removePoint()
1145 CurveCreator_ICurve::SectionToPointList aPoints;
1146 getSelectedPoints( aPoints );
1147 if ( aPoints.size() == 0 )
1150 CurveCreator_ICurve::SectionToPointList aSelPoints;
1151 startCurveModification( aSelPoints, false );
1153 myCurve->removeSeveralPoints( aPoints );
1154 finishCurveModification( CurveCreator_ICurve::SectionToPointList() );
1157 void CurveCreator_Widget::addNewPoint(const CurveCreator::Coordinates& theCoords)
1161 QList<int> aSections = mySectionView->getSelectedSections();
1162 if( aSections.size() == 0 ){
1165 int aSection = aSections[0];
1166 myCurve->addPoints(theCoords, aSection); // add to the end of section
1167 mySectionView->pointsAdded( aSection, myCurve->getNbPoints( aSection ) );
1168 updateActionsStates();
1172 void CurveCreator_Widget::insertPointToSelectedSegment( const int theX,
1175 Handle(AIS_InteractiveContext) aContext = getAISContext();
1177 OCCViewer_ViewPort3d* aViewPort = getViewPort();
1178 Handle(V3d_View) aView;
1180 aView = aViewPort->getView();
1182 if ( aContext.IsNull() || aView.IsNull() )
1185 gp_Pnt aPoint1, aPoint2;
1186 Handle(AIS_InteractiveObject) anAISObject = myCurve->getAISObject();
1187 bool isFoundPoint = CurveCreator_Utils::pointOnObject( aView, anAISObject, theX, theY,
1188 aPoint, aPoint1, aPoint2 );
1189 if ( !isFoundPoint )
1192 // insert the point to the model curve
1193 CurveCreator_ICurve::SectionToPointList aSelPoints;
1194 startCurveModification( aSelPoints );
1196 CurveCreator::Coordinates aCoords;
1197 aCoords.push_back( aPoint.X() );
1198 aCoords.push_back( aPoint.Y() );
1200 CurveCreator_ICurve::SectionToPointList aPoints1, aPoints2;
1201 findSectionsToPoints( aPoint1.X(), aPoint1.Y(), aPoints1 );
1202 findSectionsToPoints( aPoint2.X(), aPoint2.Y(), aPoints2 );
1203 CurveCreator_ICurve::SectionToPointList::const_iterator anIt = aPoints1.begin(),
1204 aLast = aPoints1.end();
1205 int aSectionId = -1;
1206 // there can be a case when a new point is added into two sections
1207 int aPoint1Id = -1, aPoint2Id = -1;
1208 for ( ; anIt != aLast && aSectionId < 0; anIt++ ) {
1209 int aSectionCur = anIt->first;
1210 CurveCreator_ICurve::SectionToPointList::const_iterator anIt2 = aPoints2.begin(),
1211 aLast2 = aPoints2.end();
1212 for ( ; anIt2 != aLast2 && aSectionId < 0; anIt2++ ) {
1213 if ( anIt2->first == aSectionCur ) {
1214 aSectionId = aSectionCur;
1215 aPoint1Id = anIt->second;
1216 aPoint2Id = anIt2->second;
1221 int anInsertPos = -1;
1222 int aLastPoint = myCurve->getNbPoints( aSectionId )-1;
1223 if ( ( aPoint1Id == aLastPoint && aPoint2Id == 0 ) ||
1224 ( aPoint2Id == aLastPoint && aPoint1Id == 0 ) )
1225 anInsertPos = -1; // if the section happens between first and last points
1227 anInsertPos = aPoint1Id < aPoint2Id ? aPoint1Id + 1 : aPoint2Id + 1;
1229 myCurve->addPoints( aCoords, aSectionId, anInsertPos );
1230 mySectionView->pointsAdded( aSectionId, myCurve->getNbPoints( aSectionId ) );
1232 finishCurveModification( aSelPoints );
1234 setSelectedPoints();
1237 void CurveCreator_Widget::moveSelectedPoints( const int theXPosition,
1238 const int theYPosition )
1240 OCCViewer_ViewPort3d* aViewPort = getViewPort();
1244 CurveCreator_ICurve::SectionToPointList aPoints;
1245 startCurveModification( aPoints, false );
1247 gp_Pnt aStartPnt = CurveCreator_Utils::ConvertClickToPoint( myDragStartPosition.x(),
1248 myDragStartPosition.y(),
1249 aViewPort->getView() );
1250 gp_Pnt anEndPnt = CurveCreator_Utils::ConvertClickToPoint( theXPosition, theYPosition,
1251 aViewPort->getView() );
1252 double aXDelta = aStartPnt.X() - anEndPnt.X();
1253 double anYDelta = aStartPnt.Y() - anEndPnt.Y();
1255 CurveCreator_ICurve::SectionToPointCoordsList aCoordList;
1256 std::deque<float> aChangedPos;
1257 CurveCreator_ICurve::SectionToPointList::const_iterator anIt = myDragPoints.begin(),
1258 aLast = myDragPoints.end();
1259 for ( ; anIt != aLast; anIt++ ) {
1260 int aSectionId = anIt->first;
1261 int aPointId = anIt->second;
1262 aChangedPos = myCurve->getPoint( aSectionId, aPointId );
1263 if ( aChangedPos.size() < 2 )
1266 // Remember drag points coordinates
1268 myInitialDragPointsCoords.insert( std::make_pair( aSectionId, aPointId ), aChangedPos );
1271 aChangedPos[0] = aChangedPos[0] - aXDelta;
1272 aChangedPos[1] = aChangedPos[1] - anYDelta;
1274 aCoordList.push_back(
1275 std::make_pair(std::make_pair( aSectionId, aPointId ),
1278 myCurve->setSeveralPoints( aCoordList, false );
1281 finishCurveModification( myDragPoints );
1284 void CurveCreator_Widget::updateLocalPointView()
1286 if ( myDragStarted )
1288 Handle(AIS_InteractiveContext) aContext = getAISContext();
1289 if ( aContext.IsNull() )
1292 CurveCreator_Utils::getSelectedPoints( aContext, myCurve, myLocalPoints );
1293 int aNbPoints = myLocalPoints.size();
1295 bool isRowLimit = aNbPoints > myLocalPointRowLimit;
1296 myLocalPointView->setVisible( getActionMode() == ModificationMode && !isRowLimit );
1298 if ( !isRowLimit ) {
1299 bool isBlocked = myLocalPointView->blockSignals(true);
1301 myLocalPointView->setLocalPointsToTable( myLocalPoints );
1303 myLocalPointView->blockSignals( isBlocked );
1310 void CurveCreator_Widget::setLocalPointContext( const bool theOpen, const bool isUpdateTable )
1312 CurveCreator_Utils::setLocalPointContext( myCurve, getAISContext(), theOpen );
1313 if ( !theOpen && isUpdateTable )
1314 updateLocalPointView();
1318 * Set drag operation started. Save the position and a list of dragged points
1319 * \param theState the drag operation state: started/finished
1320 * \param thePoint the start drag position
1322 void CurveCreator_Widget::setDragStarted( const bool theState, const QPoint& thePoint )
1325 getSelectedPoints( myDragPoints );
1327 myDragStarted = myDragPoints.size();
1328 myDragStartPosition = thePoint;
1329 if ( myDragStarted ) {
1330 // change a viewer interaction style in order to avoid a select rectangle build
1331 myDragInteractionStyle = changeInteractionStyle( SUIT_ViewModel::KEY_FREE );
1332 myCurve->setSkipSorting( true );
1336 if ( myDragStarted )
1337 changeInteractionStyle( myDragInteractionStyle );
1338 myDragStarted = false;
1339 myDragPoints.clear();
1340 myInitialDragPointsCoords.clear();
1345 void CurveCreator_Widget::getSelectedPoints( CurveCreator_ICurve::SectionToPointList& thePoints )
1348 thePoints = myLocalPoints;
1351 void CurveCreator_Widget::setSelectedPoints( const CurveCreator_ICurve::SectionToPointList& thePoints )
1353 if ( myDragStarted )
1355 Handle(AIS_InteractiveContext) aContext = getAISContext();
1356 if ( aContext.IsNull() || !aContext->HasOpenedContext() )
1359 CurveCreator_Utils::setSelectedPoints( aContext, myCurve, thePoints );
1361 updateLocalPointView();
1364 void CurveCreator_Widget::stopActionMode()
1366 setActionMode( NoneMode );
1370 * Get viewer information before perform the curve modification.
1371 * Take a list of selected cuve points an close local context.
1372 * The context should be closed because the curve presentation is
1373 * redisplayed and if it is not closed, when we close the local context
1374 * later, the presentation shown in the local context is disappeared.
1375 * \param thePoints an output list of curve selected points
1376 * \param theFillPoints a flag whether the selection list should be filled
1378 void CurveCreator_Widget::startCurveModification(
1379 CurveCreator_ICurve::SectionToPointList& thePoints,
1380 const bool theFillPoints )
1382 if ( theFillPoints ) {
1384 getSelectedPoints( thePoints );
1386 setLocalPointContext( false );
1390 * Restore the viewer state after the curve modification is done.
1391 * Open local context and select given points inside it.
1392 * \param thePoints a list of curve selected points
1394 void CurveCreator_Widget::finishCurveModification(
1395 const CurveCreator_ICurve::SectionToPointList& thePoints )
1397 if ( getActionMode() == ModificationMode )
1398 setLocalPointContext( true );
1399 setSelectedPoints( thePoints );
1404 * Returns a point index in the model curve by the point coordinates in the viewer
1405 * \param theX the X coordinate of the point
1406 * \param theY the Y coordinate of the point
1408 int CurveCreator_Widget::findLocalPointIndex( int theSectionId, float theX, float theY )
1410 return CurveCreator_UtilsICurve::findLocalPointIndex( myCurve, theSectionId, theX, theY );
1413 void CurveCreator_Widget::findSectionsToPoints( const double theX, const double theY,
1414 CurveCreator_ICurve::SectionToPointList& thePoints )
1416 return CurveCreator_UtilsICurve::findSectionsToPoints( myCurve, theX, theY, thePoints );
1419 void CurveCreator_Widget::convert( const CurveCreator_ICurve::SectionToPointList& thePoints,
1420 QMap<int, QList<int> >& theConvPoints )
1422 return CurveCreator_UtilsICurve::convert( thePoints, theConvPoints );
1426 * Returns whethe the container has the value
1427 * \param theList a container of values
1428 * \param theValue a value
1430 bool CurveCreator_Widget::contains( const CurveCreator_ICurve::SectionToPointList& theList,
1431 const CurveCreator_ICurve::SectionToPoint& theValue ) const
1433 return CurveCreator_UtilsICurve::contains( theList, theValue );