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_CurveEditor.hxx"
24 #include "CurveCreator.hxx"
25 //#include "CurveCreator_NewPointDlg.h"
26 #include "CurveCreator_NewSectionDlg.h"
27 #include "CurveCreator_Utils.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>
38 #include <BRep_Tool.hxx>
40 #include <TopoDS_Vertex.hxx>
41 #include <TopoDS_Wire.hxx>
42 #include <TopoDS_Edge.hxx>
45 #include <TopExp_Explorer.hxx>
47 #include <GeomAPI_ProjectPointOnCurve.hxx>
49 #include <AIS_ListOfInteractive.hxx>
50 #include <AIS_ListIteratorOfListOfInteractive.hxx>
51 #include <AIS_Shape.hxx>
52 #include <AIS_Point.hxx>
53 #include <AIS_Line.hxx>
54 #include <Geom_Point.hxx>
55 #include <Geom_BSplineCurve.hxx>
56 #include <Geom_Line.hxx>
57 #include <StdSelect_BRepOwner.hxx>
59 #include <QHBoxLayout>
60 #include <QVBoxLayout>
64 #include <QToolButton>
68 #include <QMouseEvent>
69 #include <QApplication>
70 #include <QTableWidget>
72 const double LOCAL_SELECTION_TOLERANCE = 0.0001;
73 const int SECTION_NAME_COLUMN_WIDTH = 75;
74 const int POINT_INDEX_COLUMN_WIDTH = 40;
76 const int SCENE_PIXEL_TOLERANCE = 10;
78 CurveCreator_Widget::CurveCreator_Widget(QWidget* parent,
79 CurveCreator_ICurve *theCurve,
81 : QWidget(parent), myNewSectionEditor(NULL), myCurve(theCurve), mySection(0),
82 myDragStarted( false ), myDragInteractionStyle( SUIT_ViewModel::STANDARD ),
85 myNewSectionEditor = new CurveCreator_NewSectionDlg( this );
86 myNewSectionEditor->hide();
87 connect( myNewSectionEditor, SIGNAL(addSection()), this, SLOT(onAddNewSection()) );
88 connect( myNewSectionEditor, SIGNAL(modifySection()), this, SLOT(onModifySection()) );
89 connect( myNewSectionEditor, SIGNAL(cancelSection()), this, SLOT(onCancelSection()) );
91 QGroupBox* aSectionGroup = new QGroupBox(tr("Sections"),this);
93 mySectionView = new CurveCreator_TreeView(myCurve, aSectionGroup);
94 connect( mySectionView, SIGNAL(selectionChanged()), this, SLOT( onSelectionChanged() ) );
95 connect( mySectionView, SIGNAL(sectionEntered(int)), this, SLOT(onEditSection(int)) );
96 connect( mySectionView, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(onContextMenu(QPoint)) );
98 myLocalPointView = new QTableWidget();
99 myLocalPointView->setVisible( false );
100 myLocalPointView->setColumnCount( 4 );
101 myLocalPointView->setColumnWidth( 0, SECTION_NAME_COLUMN_WIDTH );
102 myLocalPointView->setColumnWidth( 1, POINT_INDEX_COLUMN_WIDTH );
104 //aLabels << tr( "SECTION_LABEL" ) << tr( "IDENTIFIER_LABEL" ) << tr( "X_POSITION_LBL" ) << tr( "Y_POSITION_LBL" );
105 aLabels << tr( "Section" ) << "Index" << tr( "X" ) << tr( "Y" );
106 myLocalPointView->setHorizontalHeaderLabels( aLabels );
107 connect( myLocalPointView, SIGNAL( cellChanged( int, int ) ),
108 this, SLOT( onCellChanged( int, int ) ) );
110 QToolBar* aTB = new QToolBar(tr("TOOL_BAR_TLT"), aSectionGroup);
111 // QToolButton* anUndoBtn = new QToolButton(aTB);
113 SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
114 QPixmap anUndoPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_UNDO")));
115 QPixmap aRedoPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_REDO")));
116 QPixmap aNewSectionPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_NEW_SECTION")));
117 QPixmap aNewPointPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_NEW_POINT")));
118 QPixmap anEditPointsPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_EDIT_POINTS")));
119 QPixmap aDetectPointsPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_EDIT_POINTS")));
120 QPixmap aPolylinePixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_POLYLINE")));
121 QPixmap aSplinePixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_SPLINE")));
122 QPixmap aRemovePixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_DELETE")));
123 QPixmap aJoinPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_JOIN")));
124 QPixmap aStepUpPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_ARROW_UP")));
125 QPixmap aStepDownPixmap(aResMgr->loadPixmap("GEOM", tr("ICON_CC_ARROW_DOWN")));
127 QAction* anAct = createAction( UNDO_ID, tr("UNDO"), anUndoPixmap, tr("UNDO_TLT"),
128 QKeySequence(Qt::ControlModifier|Qt::Key_Z) );
129 connect(anAct, SIGNAL(triggered()), this, SLOT(onUndo()) );
130 aTB->addAction(anAct);
132 anAct = createAction( REDO_ID, tr("REDO"), aRedoPixmap, tr("REDO_TLT"),
133 QKeySequence(Qt::ControlModifier|Qt::Key_Y) );
134 connect(anAct, SIGNAL(triggered()), this, SLOT(onRedo()) );
135 aTB->addAction(anAct);
139 anAct = createAction( NEW_SECTION_ID, tr("NEW_SECTION"), aNewSectionPixmap, tr("NEW_SECTION_TLT"),
140 QKeySequence(Qt::ControlModifier|Qt::Key_N) );
141 connect(anAct, SIGNAL(triggered()), this, SLOT(onNewSection()) );
142 aTB->addAction(anAct);
145 anAct = createAction( ADDITION_MODE_ID, tr("ADDITION_MODE"), aNewPointPixmap, tr("ADDITION_MODE_TLT"),
147 anAct->setCheckable(true);
148 connect(anAct, SIGNAL(triggered(bool)), this, SLOT(onAdditionMode(bool)) );
149 connect(anAct, SIGNAL(toggled(bool)), this, SLOT(onModeChanged(bool)) );
150 aTB->addAction(anAct);
152 anAct = createAction( MODIFICATION_MODE_ID, tr("MODIFICATION_MODE"), anEditPointsPixmap, tr("MODIFICATION_MODE_TLT"),
154 anAct->setCheckable(true);
155 connect(anAct, SIGNAL(triggered(bool)), this, SLOT(onModificationMode(bool)) );
156 connect(anAct, SIGNAL(toggled(bool)), this, SLOT(onModeChanged(bool)) );
157 aTB->addAction(anAct);
159 anAct = createAction( DETECTION_MODE_ID, tr("DETECTION_MODE"), aDetectPointsPixmap, tr("DETECTION_MODE_TLT"),
161 anAct->setCheckable(true);
162 connect(anAct, SIGNAL(triggered(bool)), this, SLOT(onDetectionMode(bool)) );
163 connect(anAct, SIGNAL(toggled(bool)), this, SLOT(onModeChanged(bool)) );
164 aTB->addAction(anAct);
166 anAct = createAction( CLOSE_SECTIONS_ID, tr("CLOSE_SECTIONS"), QPixmap(), tr("CLOSE_SECTIONS_TLT"),
167 QKeySequence(Qt::ControlModifier|Qt::Key_W) );
168 connect(anAct, SIGNAL(triggered()), this, SLOT(onCloseSections()) );
170 anAct = createAction( UNCLOSE_SECTIONS_ID, tr("UNCLOSE_SECTIONS"), QPixmap(),
171 tr("UNCLOSE_SECTIONS_TLT"), QKeySequence(Qt::ControlModifier|Qt::Key_S) );
172 connect(anAct, SIGNAL(triggered()), this, SLOT(onUncloseSections()) );
174 anAct = createAction( SET_SECTIONS_POLYLINE_ID, tr("SET_SECTIONS_POLYLINE"),
175 aPolylinePixmap, tr("SET_POLYLINE_TLT"),
176 QKeySequence(Qt::ControlModifier|Qt::Key_E) );
177 connect(anAct, SIGNAL(triggered()), this, SLOT(onSetPolyline()) );
179 anAct = createAction( SET_SECTIONS_SPLINE_ID, tr("SET_SECTIONS_SPLINE"), aSplinePixmap,
180 tr("SET_SPLINE_TLT"), QKeySequence(Qt::ControlModifier|Qt::Key_R) );
181 connect(anAct, SIGNAL(triggered()), this, SLOT(onSetSpline()) );
183 anAct = createAction( REMOVE_ID, tr("REMOVE"), aRemovePixmap, tr("REMOVE_TLT"),
184 QKeySequence(Qt::ControlModifier|Qt::Key_Delete ) );
185 connect(anAct, SIGNAL(triggered()), this, SLOT(onRemove()) );
186 aTB->addAction(anAct);
189 anAct = createAction( JOIN_ID, tr("JOIN"), aJoinPixmap, tr("JOIN_TLT"),
190 QKeySequence(Qt::ControlModifier|Qt::Key_Plus ) );
191 connect( anAct, SIGNAL(triggered()), this, SLOT(onJoin()) );
192 aTB->addAction(anAct);
195 anAct = createAction( CLEAR_ALL_ID, tr("CLEAR_ALL"), QPixmap(), tr("CLEAR_ALL_TLT"),
196 QKeySequence(Qt::ControlModifier | Qt::ShiftModifier | Qt::Key_Delete ) );
197 connect( anAct, SIGNAL(triggered()), this, SLOT( onClearAll()) );
199 anAct = createAction( JOIN_ALL_ID, tr("JOIN_ALL"), QPixmap(), tr("JOIN_ALL_TLT"),
200 QKeySequence(Qt::ControlModifier | Qt::ShiftModifier | Qt::Key_Plus ) );
201 connect( anAct, SIGNAL(triggered()), this, SLOT(onJoinAll()) );
203 QVBoxLayout* aSectLayout = new QVBoxLayout();
204 aSectLayout->setMargin( 5 );
205 aSectLayout->setSpacing( 5 );
206 aSectLayout->addWidget(aTB);
207 aSectLayout->addWidget(mySectionView);
208 aSectLayout->addWidget( myLocalPointView );
209 aSectionGroup->setLayout(aSectLayout);
210 QVBoxLayout* aLay = new QVBoxLayout();
211 aLay->setMargin( 0 );
212 aLay->setSpacing( 5 );
213 // aLay->addLayout(aNameLayout);
214 aLay->addWidget(aSectionGroup);
216 onSelectionChanged();
222 void CurveCreator_Widget::setOCCViewer( OCCViewer_Viewer* theViewer )
224 if ( myOCCViewer == theViewer )
228 OCCViewer_ViewManager* aViewManager = dynamic_cast<OCCViewer_ViewManager*>
229 ( myOCCViewer->getViewManager() );
230 disconnect( aViewManager, SIGNAL( mousePress( SUIT_ViewWindow*, QMouseEvent* ) ),
231 this, SLOT( onMousePress( SUIT_ViewWindow*, QMouseEvent* ) ) );
232 disconnect( aViewManager, SIGNAL( mouseRelease( SUIT_ViewWindow*, QMouseEvent* ) ),
233 this, SLOT( onMouseRelease( SUIT_ViewWindow*, QMouseEvent* ) ) );
234 disconnect( aViewManager, SIGNAL( mouseMove( SUIT_ViewWindow*, QMouseEvent* ) ),
235 this, SLOT( onMouseMove( SUIT_ViewWindow*, QMouseEvent* ) ) );
236 // all local contexts should be closed if the viewer is not more used
237 setLocalPointContext( false, true );
240 myOCCViewer = theViewer;
242 OCCViewer_ViewManager* aViewManager = dynamic_cast<OCCViewer_ViewManager*>
243 ( myOCCViewer->getViewManager() );
244 connect( aViewManager, SIGNAL( mousePress( SUIT_ViewWindow*, QMouseEvent* ) ),
245 this, SLOT( onMousePress( SUIT_ViewWindow*, QMouseEvent* ) ) );
246 connect( aViewManager, SIGNAL( mouseRelease( SUIT_ViewWindow*, QMouseEvent* ) ),
247 this, SLOT( onMouseRelease( SUIT_ViewWindow*, QMouseEvent* ) ) );
248 connect( aViewManager, SIGNAL( mouseMove( SUIT_ViewWindow*, QMouseEvent* ) ),
249 this, SLOT( onMouseMove( SUIT_ViewWindow*, QMouseEvent* ) ) );
254 * Returns current OCC viewer
256 OCCViewer_Viewer* CurveCreator_Widget::getOCCViewer()
261 //=======================================================================
262 // function: getUniqSectionName
263 // purpose: return unique section name
264 //=======================================================================
265 std::string CurveCreator_Widget::getUniqSectionName( CurveCreator_ICurve* theCurve ) const
267 for( int i = 0 ; i < 1000000 ; i++ ){
269 sprintf( aBuffer, "Section_%d", i+1 );
270 std::string aName(aBuffer);
272 for( j = 0 ; j < theCurve->getNbSections() ; j++ ){
273 if( theCurve->getSectionName(j) == aName )
276 if( j == theCurve->getNbSections() )
282 void CurveCreator_Widget::setCurve( CurveCreator_ICurve* theCurve )
285 mySectionView->setCurve(myCurve);
286 onSelectionChanged();
290 void CurveCreator_Widget::onSelectionChanged()
292 QList<ActionId> anEnabledAct;
294 anEnabledAct << NEW_SECTION_ID << MODIFICATION_MODE_ID;
295 if ( getActionMode() == ModificationMode )
296 anEnabledAct << REMOVE_ID;
297 QList<int> aSelSections = mySectionView->getSelectedSections();
298 QList< QPair< int, int > > aSelPoints = mySectionView->getSelectedPoints();
299 CurveCreator_TreeView::SelectionType aSelType = mySectionView->getSelectionType();
301 case CurveCreator_TreeView::ST_NOSEL:{
304 case CurveCreator_TreeView::ST_SECTIONS:{
305 /*if( aSelSections[0] > 0 ){
306 anEnabledAct << UP_ID;
308 if( aSelSections.size() == 1 ){
309 anEnabledAct << ADDITION_MODE_ID << DETECTION_MODE_ID;
310 anEnabledAct << REMOVE_ID;
312 switch ( getActionMode() ) {
316 QList<int> aSelSection = mySectionView->getSelectedSections();
317 if( aSelSection.size() > 0 ){
318 mySection = aSelSection[0];
319 myPointNum = myCurve->getNbPoints(mySection);
323 case ModificationMode: {
324 anEnabledAct << CLOSE_SECTIONS_ID << UNCLOSE_SECTIONS_ID << SET_SECTIONS_POLYLINE_ID << SET_SECTIONS_SPLINE_ID;
325 int aSectCnt = myCurve->getNbSections();
327 anEnabledAct << CLEAR_ALL_ID;
329 anEnabledAct << JOIN_ALL_ID;
330 if( aSelSections.size() > 1 ){
331 anEnabledAct << JOIN_ID;
335 case DetectionMode: {
342 /*if( aSelSections[ aSelSections.size() - 1 ] < ( myCurve->getNbSections() - 1 ) ){
343 anEnabledAct << DOWN_ID;
347 /*case CurveCreator_TreeView::ST_POINTS_ONE_SECTION:{
348 if( aSelPoints[0].second > 0 ){
349 anEnabledAct << UP_ID;
351 int aLastIndex = aSelPoints.size()-1;
352 int aSect = aSelPoints[0].first;
353 if( aSelPoints[aLastIndex].second < (myCurve->getNbPoints(aSect) - 1)){
354 anEnabledAct << DOWN_ID;
356 if( aSelPoints.size() == 1){
357 anEnabledAct << INSERT_POINT_BEFORE_ID << INSERT_POINT_AFTER_ID;
364 /*int aSelObjsCnt = aSelPoints.size() + aSelSections.size();
365 if( aSelObjsCnt > 0 ){
366 anEnabledAct << REMOVE_ID;
368 if( (myCurve->getNbSections() + myCurve->getNbPoints()) > 0 ){
369 anEnabledAct << REMOVE_ALL_ID;
371 if( myCurve->getNbSections() > 1 ){
372 anEnabledAct << JOIN_ALL_ID;
375 QList<ActionId> anIds = myActionMap.keys();
376 for( int i = 0 ; i < anIds.size() ; i++ ){
377 if( myActionMap.contains(anIds[i]) ){
378 if( anEnabledAct.contains(anIds[i]) ){
379 myActionMap[anIds[i]]->setEnabled(true);
382 myActionMap[anIds[i]]->setEnabled(false);
387 emit selectionChanged();
390 void CurveCreator_Widget::onAdditionMode(bool checked)
392 OCCViewer_Viewer* aViewer = getOCCViewer();
393 if( !myCurve || !aViewer )
396 Handle(AIS_InteractiveContext) aContext = aViewer->getAISContext();
397 OCCViewer_ViewManager* aViewManager = dynamic_cast<OCCViewer_ViewManager*>
398 (aViewer->getViewManager());
399 // if ( aViewManager->getType() == OCCViewer_Viewer::Type() ) {
402 /* myGuiState = aViewWindow->saveState();
403 aViewer->enableMultiselection(false);
404 aViewer->enableSelection(false);*/
406 /* aViewer->enableMultiselection(true);
407 aViewer->enableSelection(true);
408 aViewWindow->restoreState( myGuiState );*/
415 QList<int> aSelSection = mySectionView->getSelectedSections();
416 if( aSelSection.size() > 0 ){
417 mySection = aSelSection[0];
420 QList< QPair<int,int> > aSelPoints = mySectionView->getSelectedPoints();
421 if( aSelPoints.size() > 0 ){
422 mySection = aSelPoints[0].first;
423 myPointNum = aSelPoints[0].second + 1;
426 // emit subOperationStarted( myNewPointEditor );
429 void CurveCreator_Widget::onModificationMode(bool checked)
431 myLocalPointView->setVisible( checked );
434 void CurveCreator_Widget::onDetectionMode(bool checked)
438 void CurveCreator_Widget::onModeChanged(bool checked)
440 ActionMode aMode = NoneMode;
442 QAction* anAction = (QAction*)sender();
443 switch(myActionMap.key(anAction)) {
444 case ADDITION_MODE_ID:
445 aMode = AdditionMode;
446 if (myActionMap[MODIFICATION_MODE_ID]->isChecked())
447 myActionMap[MODIFICATION_MODE_ID]->trigger();
448 else if (myActionMap[DETECTION_MODE_ID]->isChecked())
449 myActionMap[DETECTION_MODE_ID]->trigger();
451 case MODIFICATION_MODE_ID:
452 aMode = ModificationMode;
453 if (myActionMap[ADDITION_MODE_ID]->isChecked())
454 myActionMap[ADDITION_MODE_ID]->trigger();
455 else if (myActionMap[DETECTION_MODE_ID]->isChecked())
456 myActionMap[DETECTION_MODE_ID]->trigger();
458 case DETECTION_MODE_ID:
459 aMode = DetectionMode;
460 if (myActionMap[ADDITION_MODE_ID]->isChecked())
461 myActionMap[ADDITION_MODE_ID]->trigger();
462 else if (myActionMap[MODIFICATION_MODE_ID]->isChecked())
463 myActionMap[MODIFICATION_MODE_ID]->trigger();
467 onSelectionChanged();
468 setLocalPointContext( aMode == ModificationMode, true );
471 void CurveCreator_Widget::onAddNewPoint(const CurveCreator::Coordinates& theCoords)
475 //myCurve->addPoints(theCoords, mySection, myPointNum );
476 //mySectionView->pointsAdded( mySection, myPointNum );
478 QList<int> aSections = mySectionView->getSelectedSections();
479 if( aSections.size() == 0 ){
482 int aSection = aSections[0];
483 myCurve->addPoints(theCoords, aSection); // add to the end of section
484 mySectionView->pointsAdded( aSection, myCurve->getNbPoints( aSection ) );
485 onSelectionChanged();
489 void CurveCreator_Widget::onNewSection()
493 myNewSectionEditor->clear();
494 myNewSectionEditor->setEditMode(false);
495 QString aSectName = QString( getUniqSectionName(myCurve).c_str() );
496 myNewSectionEditor->setSectionParameters(aSectName, true, CurveCreator::Polyline );
497 emit subOperationStarted( myNewSectionEditor );
500 void CurveCreator_Widget::onAddNewSection()
504 myCurve->addSection( myNewSectionEditor->getName().toStdString(), myNewSectionEditor->getSectionType(),
505 myNewSectionEditor->isClosed() );
506 mySectionView->sectionAdded( -1 ); // add a new section to the end of list
507 QString aNewName = QString(getUniqSectionName(myCurve).c_str());
508 myNewSectionEditor->setSectionName(aNewName);
509 onSelectionChanged();
514 void CurveCreator_Widget::onCancelSection()
516 emit subOperationFinished( myNewSectionEditor );
519 QAction* CurveCreator_Widget::createAction( ActionId theId, const QString& theName, const QPixmap& theImage,
520 const QString& theToolTip, const QKeySequence& theShortcut )
522 QAction* anAct = new QAction(theName,this);
523 if( !theImage.isNull() ){
524 anAct->setIcon(theImage);
526 anAct->setShortcut(theShortcut);
527 anAct->setToolTip(theToolTip);
528 myActionMap[theId] = anAct;
532 QAction* CurveCreator_Widget::getAction(ActionId theId)
534 if( myActionMap.contains(theId) )
535 return myActionMap[theId];
539 CurveCreator_Widget::ActionMode CurveCreator_Widget::getActionMode() const
541 ActionMode aMode = NoneMode;
543 if ( myActionMap[ADDITION_MODE_ID]->isChecked() )
544 aMode = AdditionMode;
545 else if ( myActionMap[MODIFICATION_MODE_ID]->isChecked() )
546 aMode = ModificationMode;
547 else if ( myActionMap[DETECTION_MODE_ID]->isChecked() )
548 aMode = DetectionMode;
553 void CurveCreator_Widget::onEditSection( int theSection )
557 mySection = theSection;
558 QString aSectName = QString::fromStdString( myCurve->getSectionName(theSection));
559 bool isClosed = myCurve->isClosed(theSection);
560 CurveCreator::SectionType aType = myCurve->getSectionType(theSection);
561 myNewSectionEditor->setEditMode(true);
562 myNewSectionEditor->setSectionParameters( aSectName, isClosed, aType );
564 emit subOperationStarted( myNewSectionEditor );
567 void CurveCreator_Widget::onModifySection()
571 QString aName = myNewSectionEditor->getName();
572 bool isClosed = myNewSectionEditor->isClosed();
573 CurveCreator::SectionType aSectType = myNewSectionEditor->getSectionType();
574 // myCurve->startOperation();
575 myCurve->setClosed( isClosed, mySection );
576 myCurve->setSectionName( mySection , aName.toStdString() );
577 myCurve->setSectionType( mySection, aSectType );
578 // myCurve->finishOperation();
579 mySectionView->sectionChanged(mySection);
584 /*void CurveCreator_Widget::onEditPoint( int theSection, int thePoint )
586 if( !myNewPointEditor || !myEdit )
588 mySection = theSection;
589 myPointNum = thePoint;
590 QString aSectName = QString::fromStdString( myCurve->getSectionName(theSection));
591 myNewPointEditor->setEditMode(true);
592 myNewPointEditor->setSectionName(aSectName);
593 myNewPointEditor->setDimension( myCurve->getDimension() );
594 CurveCreator::Coordinates aCoords = myCurve->getCoordinates(theSection,thePoint);
595 myNewPointEditor->setCoordinates(aCoords);
596 emit subOperationStarted( myNewPointEditor );
599 void CurveCreator_Widget::onModifyPoint()
603 CurveCreator::Coordinates aCoords = myNewPointEditor->getCoordinates();
604 myEdit->setCoordinates( aCoords, mySection, myPointNum );
605 mySectionView->pointDataChanged( mySection, myPointNum );
610 void CurveCreator_Widget::onJoin()
614 QList<int> aSections = mySectionView->getSelectedSections();
615 if( aSections.size() == 0 ){
618 int aMainSect = aSections[0];
619 int aMainSectSize = myCurve->getNbPoints(aMainSect);
620 // myCurve->startOperation();
621 for( int i = 1 ; i < aSections.size() ; i++ ){
622 int aSectNum = aSections[i] - (i-1);
623 myCurve->join( aMainSect, aSectNum );
624 mySectionView->sectionsRemoved( aSectNum );
626 // myCurve->finishOperation();
627 int aNewSectSize = myCurve->getNbPoints(aMainSect);
628 if( aNewSectSize != aMainSectSize )
629 mySectionView->pointsAdded( aMainSect, aMainSectSize, aNewSectSize-aMainSectSize );
633 void CurveCreator_Widget::onRemove()
638 switch( getActionMode() ) {
642 case ModificationMode:
650 void CurveCreator_Widget::onClearAll()
655 mySectionView->reset();
656 onSelectionChanged();
660 void CurveCreator_Widget::onJoinAll()
665 mySectionView->reset();
666 onSelectionChanged();
670 void CurveCreator_Widget::onUndoSettings()
675 void CurveCreator_Widget::onSetSpline()
679 QList<int> aSelSections = mySectionView->getSelectedSections();
680 // myCurve->startOperation();
681 for( int i = 0 ; i < aSelSections.size() ; i++ ){
682 myCurve->setSectionType(aSelSections[i], CurveCreator::Spline );
683 mySectionView->sectionChanged(aSelSections[i]);
685 // myCurve->finishOperation();
689 void CurveCreator_Widget::onSetPolyline()
693 // myCurve->startOperation();
694 QList<int> aSelSections = mySectionView->getSelectedSections();
695 for( int i = 0 ; i < aSelSections.size() ; i++ ){
696 myCurve->setSectionType( aSelSections[i], CurveCreator::Polyline );
697 mySectionView->sectionChanged( aSelSections[i] );
699 // myCurve->finishOperation();
703 void CurveCreator_Widget::onCloseSections()
707 // myCurve->startOperation();
708 QList<int> aSelSections = mySectionView->getSelectedSections();
709 for( int i = 0 ; i < aSelSections.size() ; i++ ){
710 myCurve->setClosed(true, aSelSections[i]);
711 mySectionView->sectionChanged(aSelSections[i]);
713 // myCurve->finishOperation();
717 void CurveCreator_Widget::onUncloseSections()
721 // myCurve->startOperation();
722 QList<int> aSelSections = mySectionView->getSelectedSections();
723 for( int i = 0 ; i < aSelSections.size() ; i++ ){
724 myCurve->setClosed(false, aSelSections[i]);
725 mySectionView->sectionChanged(aSelSections[i]);
727 // myCurve->finishOperation();
731 void CurveCreator_Widget::onUndo()
736 CurveCreator_Widget::SectionToPointList aPoints;
737 startCurveModification( aPoints, false );
739 finishCurveModification();
740 mySectionView->reset();
744 void CurveCreator_Widget::onRedo()
748 CurveCreator_Widget::SectionToPointList aPoints;
749 startCurveModification( aPoints, false );
751 finishCurveModification();
752 mySectionView->reset();
756 void CurveCreator_Widget::updateUndoRedo()
760 QAction* anAct = myActionMap[UNDO_ID];
762 if( myCurve->getNbUndo() != 0 ){
763 anAct->setEnabled(true);
766 anAct->setDisabled(true);
769 anAct = myActionMap[REDO_ID];
771 if( myCurve->getNbRedo() != 0 ){
772 anAct->setEnabled(true);
775 anAct->setDisabled(true);
780 void CurveCreator_Widget::onContextMenu( QPoint thePoint )
782 QList<ActionId> aContextActions;
783 aContextActions << CLEAR_ALL_ID << JOIN_ALL_ID << SEPARATOR_ID <<
784 CLOSE_SECTIONS_ID << UNCLOSE_SECTIONS_ID << SET_SECTIONS_POLYLINE_ID <<
785 SET_SECTIONS_SPLINE_ID;
786 QPoint aGlPoint = mySectionView->mapToGlobal(thePoint);
788 QList<ActionId> aResAct;
789 for( int i = 0 ; i < aContextActions.size() ; i++ ){
790 if( aContextActions[i] != SEPARATOR_ID ){
791 if( myActionMap.contains(aContextActions[i]) ){
792 QAction* anAct = myActionMap[aContextActions[i]];
793 if( anAct->isEnabled() ){
794 aResAct << aContextActions[i];
800 aResAct << SEPARATOR_ID;
806 QMenu* aMenu = new QMenu(this);
807 for( int i = 0 ; i < aResAct.size() ; i++ ){
808 if( aResAct[i] == SEPARATOR_ID ){
809 aMenu->addSeparator();
812 QAction* anAct = myActionMap[aResAct[i]];
813 aMenu->insertAction(NULL, anAct);
816 aMenu->exec(aGlPoint);
819 QList<int> CurveCreator_Widget::getSelectedSections()
821 return mySectionView->getSelectedSections();
824 QList< QPair< int, int > > CurveCreator_Widget::getSelectedPoints()
826 return mySectionView->getSelectedPoints();
829 //=================================================================================
830 // function : GeometryGUI::onGetCoordsByClick()
831 // purpose : Manage mouse press events in Additon mode
832 //=================================================================================
833 void CurveCreator_Widget::onGetCoordsByClick( QMouseEvent* pe )
835 if (pe->button() != Qt::LeftButton)
838 if ( pe->modifiers() != Qt::ControlModifier ) {
839 OCCViewer_Viewer* aViewer = getOCCViewer();
842 Handle(AIS_InteractiveContext) ic = aViewer->getAISContext();
847 if ( pe->modifiers() == Qt::ShiftModifier )
848 ic->ShiftSelect(); // Append selection
850 ic->Select(); // New selection
852 /*TopoDS_Shape aShape;
855 if ( ic->MoreSelected() )
856 aShape = ic->SelectedShape();
858 if ( !aShape.IsNull() && aShape.ShapeType() == TopAbs_VERTEX )
859 aPnt = BRep_Tool::Pnt( TopoDS::Vertex( ic->SelectedShape() ) );
862 OCCViewer_ViewPort3d* vp =
863 ((OCCViewer_ViewWindow*)aViewer->getViewManager()->getActiveView())->getViewPort();
864 aPnt = CurveCreator_Utils::ConvertClickToPoint( pe->x(), pe->y(), vp->getView() );
866 // set the coordinates into dialog
867 CurveCreator::Coordinates aCoords;
868 aCoords.push_back( aPnt.X() );
869 aCoords.push_back( aPnt.Y() );
870 if ( myCurve->getDimension() == 3 ) {
871 aCoords.push_back( aPnt.Z() );
873 onAddNewPoint(aCoords);
874 // myNewPointEditor->setCoordinates( aCoords );
878 //=================================================================================
879 // function : CurveCreator_Widget::onMousePress()
880 // purpose : Manage mouse press events in Modification mode
881 //=================================================================================
882 void CurveCreator_Widget::onMousePress( SUIT_ViewWindow*, QMouseEvent* theEvent )
884 if ( theEvent->button() != Qt::LeftButton )
887 switch( getActionMode() ) {
888 case ModificationMode: {
889 //store initial cursor position for Drag&Drop
890 setDragStarted( true, theEvent->pos() );
894 onGetCoordsByClick( theEvent );
902 //=================================================================================
903 // function : HYDROGUI_PolylineOp::onMouseRelease()
904 // purpose : Manage mouse press events in Modification mode
905 //=================================================================================
906 void CurveCreator_Widget::onMouseRelease( SUIT_ViewWindow*, QMouseEvent* theEvent )
908 if ( getActionMode() != ModificationMode )
911 if ( myDragStarted ) {
912 if ( myDragged ) // if the drag of some points has happened, restore the drag selection
913 setSelectedPonts( myDragPoints );
914 setDragStarted( false );
916 else // check whether the segment is clicked an a new point should be added to the segment
917 insertPointToSelectedSegment( theEvent->pos().x(), theEvent->pos().y() );
919 // updates the input panel table to show the selected point coordinates
920 updateLocalPointView();
923 //=================================================================================
924 // function : GeometryGUI::onMouseMove()
925 // purpose : Manage mouse move events in Modification mode
926 //=================================================================================
927 void CurveCreator_Widget::onMouseMove( SUIT_ViewWindow*, QMouseEvent* theEvent )
929 if ( getActionMode() != ModificationMode || !myDragStarted )
932 QPoint aPos = theEvent->pos();
933 if ( (aPos - myDragStartPosition).manhattanLength() < QApplication::startDragDistance() )
936 moveSelectedPoints( aPos.x(), aPos.y() );
937 myDragStartPosition = aPos;
940 void CurveCreator_Widget::onCellChanged( int theRow, int theColumn )
942 int aCurrSect = getSectionId( theRow );
943 int aPntIndex = getPointId( theRow );
948 SectionToPointList aSelPoints;
949 startCurveModification( aSelPoints );
951 double aX = myLocalPointView->item( theRow, 2 )->text().toDouble();
952 double anY = myLocalPointView->item( theRow, 3 )->text().toDouble();
953 std::deque<float> aChangedPos;
954 aChangedPos.push_back( aX );
955 aChangedPos.push_back( anY );
956 myCurve->setPoint( aCurrSect, aPntIndex, aChangedPos );
958 finishCurveModification( aSelPoints );
962 * Removes a selected section from the curve. Updates undo/redo status
964 void CurveCreator_Widget::removeSection()
966 QList< QPair<int,int> > aSelPoints = mySectionView->getSelectedPoints();
969 // myCurve->startOperation();
970 for( int i = 0 ; i < aSelPoints.size() ; i++ ){
971 if( aCurrSect != aSelPoints[i].first ){
973 aCurrSect = aSelPoints[i].first;
975 int aPntIndx = aSelPoints[i].second - aRemoveCnt;
976 myCurve->removePoint( aCurrSect, aPntIndx );
977 mySectionView->pointsRemoved( aCurrSect, aPntIndx );
980 QList<int> aSections = mySectionView->getSelectedSections();
981 for( int i = 0 ; i < aSections.size() ; i++ ){
982 int aSectNum = aSections[i] - (i);
983 myCurve->removeSection( aSectNum );
984 mySectionView->sectionsRemoved( aSectNum );
986 // myCurve->finishOperation();
987 mySectionView->clearSelection();
992 * Removes a selected points from the curve. Updates undo/redo status
994 void CurveCreator_Widget::removePoint()
996 SectionToPointList aPoints;
997 getSelectedPonts( aPoints );
998 if ( aPoints.size() == 0 )
1001 SectionToPointList aSelPoints;
1002 startCurveModification( aSelPoints, false );
1004 // the points should be removed in a decreased order
1005 QMap<int, QList<int> > aConvPoints;
1006 convert( aPoints, aConvPoints );
1007 QMap<int, QList<int> >::const_iterator anIt = aConvPoints.begin(),
1008 aLast = aConvPoints.end();
1009 for ( ; anIt != aLast; anIt++ ) {
1010 int aSectionId = anIt.key();
1012 QList<int> aSectionPoints = anIt.value();
1013 qSort( aSectionPoints );
1014 for( int i = aSectionPoints.size()-1; i >= 0; i-- ){
1015 int aPntIndx = aSectionPoints[i];
1016 myCurve->removePoint( aSectionId, aPntIndx );
1017 mySectionView->pointsRemoved( aSectionId, aPntIndx );
1021 finishCurveModification( SectionToPointList() );
1024 void CurveCreator_Widget::insertPointToSelectedSegment( const int theX,
1027 OCCViewer_Viewer* aViewer = getOCCViewer();
1032 gp_Pnt aPoint1, aPoint2;
1033 bool isFoundPoint = false;
1034 Handle(AIS_InteractiveContext) aContext = aViewer->getAISContext();
1035 for ( aContext->InitSelected(); aContext->MoreSelected() && !isFoundPoint;
1036 aContext->NextSelected() ) {
1037 TopoDS_Shape aTShape = aContext->SelectedShape();
1038 if ( !aTShape.IsNull() && aTShape.ShapeType() == TopAbs_VERTEX )
1041 Handle(SelectMgr_EntityOwner) anOwner = aContext->SelectedOwner();
1042 if ( anOwner.IsNull() )
1044 const TopLoc_Location& aLocation = anOwner->Location();
1045 Handle(AIS_InteractiveObject) anAIS = Handle(AIS_InteractiveObject)::DownCast( anOwner->Selectable() );
1046 isFoundPoint = pointOnObject( anAIS, theX, theY, aPoint, aPoint1, aPoint2 );
1049 if ( !isFoundPoint )
1052 // insert the point to the model curve
1053 SectionToPointList aSelPoints;
1054 startCurveModification( aSelPoints );
1056 CurveCreator::Coordinates aCoords;
1057 aCoords.push_back( aPoint.X() );
1058 aCoords.push_back( aPoint.Y() );
1060 SectionToPointList aPoints1, aPoints2;
1061 findSectionsToPoints( aPoint1.X(), aPoint1.Y(), aPoints1 );
1062 findSectionsToPoints( aPoint2.X(), aPoint2.Y(), aPoints2 );
1063 SectionToPointList::const_iterator anIt = aPoints1.begin(), aLast = aPoints1.end();
1064 int aSectionId = -1;
1065 // there can be a case when a new point is added into two sections
1066 int aPoint1Id = -1, aPoint2Id = -1;
1067 for ( ; anIt != aLast && aSectionId < 0; anIt++ ) {
1068 int aSectionCur = anIt->first;
1069 SectionToPointList::const_iterator anIt2 = aPoints2.begin(), aLast2 = aPoints2.end();
1070 for ( ; anIt2 != aLast2 && aSectionId < 0; anIt2++ ) {
1071 if ( anIt2->first == aSectionCur ) {
1072 aSectionId = aSectionCur;
1073 aPoint1Id = anIt->second;
1074 aPoint2Id = anIt2->second;
1079 int anInsertPos = -1;
1080 int aLastPoint = myCurve->getNbPoints( aSectionId )-1;
1081 if ( ( aPoint1Id == aLastPoint && aPoint2Id == 0 ) ||
1082 ( aPoint2Id == aLastPoint && aPoint1Id == 0 ) )
1083 anInsertPos = -1; // if the section happens between first and last points
1085 anInsertPos = aPoint1Id < aPoint2Id ? aPoint1Id + 1 : aPoint2Id + 1;
1087 myCurve->addPoints( aCoords, aSectionId, anInsertPos );
1088 mySectionView->pointsAdded( aSectionId, myCurve->getNbPoints( aSectionId ) );
1090 finishCurveModification( aSelPoints );
1096 void CurveCreator_Widget::moveSelectedPoints( const int theXPosition,
1097 const int theYPosition )
1099 OCCViewer_Viewer* aViewer = getOCCViewer();
1103 SectionToPointList aPoints;
1104 startCurveModification( aPoints, false );
1106 OCCViewer_ViewWindow* aWindow =
1107 (OCCViewer_ViewWindow*)aViewer->getViewManager()->getActiveView();
1108 gp_Pnt aStartPnt = CurveCreator_Utils::ConvertClickToPoint( myDragStartPosition.x(),
1109 myDragStartPosition.y(),
1110 aWindow->getViewPort()->getView() );
1111 gp_Pnt anEndPnt = CurveCreator_Utils::ConvertClickToPoint( theXPosition, theYPosition,
1112 aWindow->getViewPort()->getView() );
1113 double aXDelta = aStartPnt.X() - anEndPnt.X();
1114 double anYDelta = aStartPnt.Y() - anEndPnt.Y();
1118 std::deque<float> aChangedPos;
1119 SectionToPointList::const_iterator anIt = myDragPoints.begin(), aLast = myDragPoints.end();
1120 for ( ; anIt != aLast; anIt++ ) {
1121 aSectionId = anIt->first;
1122 aPointId = anIt->second;
1123 aChangedPos = myCurve->getPoint( aSectionId, aPointId );
1124 if ( aChangedPos.size() < 2 )
1126 aChangedPos[0] = aChangedPos[0] - aXDelta;
1127 aChangedPos[1] = aChangedPos[1] - anYDelta;
1128 myCurve->setPoint( aSectionId, aPointId, aChangedPos );
1131 finishCurveModification( myDragPoints );
1134 void CurveCreator_Widget::updateLocalPointView()
1136 OCCViewer_Viewer* aViewer = getOCCViewer();
1139 Handle(AIS_InteractiveContext) aContext = aViewer->getAISContext();
1141 bool isBlocked = myLocalPointView->blockSignals(true);
1143 myLocalPointView->setRowCount( 0 );
1144 int anInitialCount = myLocalPointView->rowCount();
1145 for ( aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected() ) {
1146 TopoDS_Vertex aVertex;
1147 TopoDS_Shape aShape = aContext->SelectedShape();
1148 if ( !aShape.IsNull() && aShape.ShapeType() == TopAbs_VERTEX )
1149 aVertex = TopoDS::Vertex( aContext->SelectedShape() );
1151 Handle(SelectMgr_EntityOwner) anOwner = aContext->SelectedOwner();
1152 if ( !anOwner.IsNull() ) {
1153 Handle(AIS_InteractiveObject) anAIS = Handle(AIS_InteractiveObject)::DownCast( anOwner->Selectable() );
1154 if ( !anAIS.IsNull() ) {
1155 Handle(AIS_Point) aPoint = Handle(AIS_Point)::DownCast( anAIS);
1156 if ( !aPoint.IsNull() )
1157 aVertex = TopoDS::Vertex( aPoint->Vertex() );
1159 if ( aVertex.IsNull() ) {
1160 // the following happens if there are no points in the current curve, there is only a shape
1161 /*Handle(StdSelect_BRepOwner) aBrepOwner = Handle(StdSelect_BRepOwner)::DownCast(anOwner);
1162 if ( aBrepOwner.IsNull() )
1164 if ( aBrepOwner->HasShape() ) {
1165 const TopoDS_Shape& aShape = aBrepOwner->Shape();
1166 if ( !aShape.IsNull() && aShape.ShapeType() == TopAbs_VERTEX )
1168 aVertex = TopoDS::Vertex( aShape );
1173 if ( aVertex.IsNull() )
1175 aPnt = BRep_Tool::Pnt( aVertex );
1176 addLocalPointToTable( aPnt.X(), aPnt.Y() );
1179 int aFinalCount = myLocalPointView->rowCount();
1180 if ( anInitialCount == aFinalCount )
1182 myLocalPointView->blockSignals(isBlocked);
1188 void CurveCreator_Widget::setLocalPointContext( const bool theOpen, const bool isUpdateTable )
1190 OCCViewer_Viewer* aViewer = getOCCViewer();
1193 Handle(AIS_InteractiveContext) ic = aViewer->getAISContext();
1196 // Open local context if there is no one
1197 if ( !ic->HasOpenedContext() ) {
1198 ic->ClearCurrents( false );
1199 ic->OpenLocalContext( false/*use displayed objects*/, true/*allow shape decomposition*/ );
1201 AIS_ListOfInteractive aList;
1202 ic->DisplayedObjects( aList );
1204 for ( AIS_ListIteratorOfListOfInteractive it( aList ); it.More(); it.Next() )
1207 for ( AIS_ListIteratorOfListOfInteractive it( aList ); it.More(); it.Next() )
1209 Handle(AIS_InteractiveObject) anAIS = it.Value();
1210 if ( !anAIS.IsNull() )
1212 if ( anAIS->IsKind( STANDARD_TYPE( AIS_Shape ) ) )
1214 ic->Load( anAIS, -1/*selection mode*/, true/*allow decomposition*/ );
1215 ic->Activate( anAIS, AIS_Shape::SelectionMode( (TopAbs_ShapeEnum)TopAbs_WIRE ) );
1217 else if ( anAIS->DynamicType() != STANDARD_TYPE(AIS_Trihedron) )
1219 ic->Load( anAIS, -1/*selection mode*/, false/*allow decomposition*/ );
1220 ic->Activate( anAIS, TopAbs_VERTEX );
1227 if ( ic->HasOpenedContext() )
1228 ic->CloseAllContexts();
1229 if ( isUpdateTable )
1230 updateLocalPointView();
1234 void CurveCreator_Widget::addLocalPointToTable( const double theX, const double theY )
1236 SectionToPointList aPoints;
1237 findSectionsToPoints( theX, theY, aPoints );
1239 SectionToPointList aSkipList;
1240 // table could not contain two equal value rows
1241 int aRowId = myLocalPointView->rowCount();
1242 double aCurrentX, aCurrentY;
1243 int aSectionId, aPointId;
1244 SectionToPoint aPoint;
1245 for ( int i = 0; i < aRowId; i++ ) {
1246 aCurrentX = myLocalPointView->item( i, 2 )->data( Qt::UserRole ).toDouble();
1247 aCurrentY = myLocalPointView->item( i, 3 )->data( Qt::UserRole ).toDouble();
1248 if ( fabs( aCurrentX - theX ) < LOCAL_SELECTION_TOLERANCE &&
1249 fabs( aCurrentY - theY ) < LOCAL_SELECTION_TOLERANCE ) {
1250 aPoint = qMakePair( getSectionId( i ), getPointId( i ) );
1251 if ( !aSkipList.contains( aPoint ) )
1252 aSkipList.append( aPoint );
1255 if ( aSkipList.size() == aPoints.size() )
1258 QTableWidgetItem* anItem;
1259 SectionToPointList::const_iterator anIt = aPoints.begin(), aLast = aPoints.end();
1260 for ( ; anIt != aLast; anIt++ ) {
1262 if ( aSkipList.contains( aPoint ) )
1265 myLocalPointView->setRowCount( aRowId+1 );
1266 aSectionId = aPoint.first;
1267 aPointId = aPoint.second;
1269 anItem = new QTableWidgetItem( myCurve->getSectionName( aSectionId ).c_str() );
1270 anItem->setFlags( anItem->flags() & ~Qt::ItemIsEnabled );
1271 anItem->setData( Qt::UserRole, aSectionId );
1272 myLocalPointView->setItem( aRowId, 0, anItem );
1274 anItem = new QTableWidgetItem( QString::number( aPointId + 1 ) );
1275 anItem->setFlags( anItem->flags() & ~Qt::ItemIsEnabled );
1276 anItem->setData( Qt::UserRole, aPointId );
1277 myLocalPointView->setItem( aRowId, 1, anItem );
1279 anItem = new QTableWidgetItem( QString::number( theX ) );
1280 anItem->setData( Qt::UserRole, theX );
1281 myLocalPointView->setItem( aRowId, 2, anItem );
1283 anItem = new QTableWidgetItem( QString::number( theY ) );
1284 anItem->setData( Qt::UserRole, theY );
1285 myLocalPointView->setItem( aRowId, 3, anItem );
1290 * Set drag operation started. Save the position and a list of dragged points
1291 * \param theState the drag operation state: started/finished
1292 * \param thePoint the start drag position
1294 void CurveCreator_Widget::setDragStarted( const bool theState, const QPoint& thePoint )
1296 OCCViewer_Viewer* aViewer = getOCCViewer();
1298 getSelectedPonts( myDragPoints );
1299 myDragStarted = myDragPoints.size();
1300 myDragStartPosition = thePoint;
1301 if ( aViewer && myDragStarted ) {
1302 // change a viewer interaction style in order to avoid a select rectangle build
1303 myDragInteractionStyle = aViewer->interactionStyle();
1304 aViewer->setInteractionStyle(SUIT_ViewModel::KEY_FREE);
1308 if ( aViewer && myDragStarted )
1309 aViewer->setInteractionStyle( myDragInteractionStyle );
1310 myDragStarted = false;
1311 myDragPoints.clear();
1316 void CurveCreator_Widget::getSelectedPonts( CurveCreator_Widget::SectionToPointList& thePoints )
1319 for ( int i = 0, aNb = myLocalPointView->rowCount(); i < aNb; i++ )
1320 thePoints.append( qMakePair( getSectionId( i ), getPointId( i ) ) );
1323 void CurveCreator_Widget::setSelectedPonts( const CurveCreator_Widget::SectionToPointList& thePoints )
1325 OCCViewer_Viewer* aViewer = getOCCViewer();
1329 AIS_ListOfInteractive aListToSelect;
1331 Handle(AIS_InteractiveContext) ic = aViewer->getAISContext();
1332 if ( !ic->HasOpenedContext() )
1335 AIS_ListOfInteractive aDisplayedList;
1336 ic->DisplayedObjects( aDisplayedList );
1337 for ( AIS_ListIteratorOfListOfInteractive it( aDisplayedList ); it.More(); it.Next() )
1339 Handle(AIS_InteractiveObject) anAIS = it.Value();
1340 if ( anAIS.IsNull() )
1342 Handle(AIS_Point) anAISPoint = Handle(AIS_Point)::DownCast( anAIS );
1343 if ( anAISPoint.IsNull() )
1346 TopoDS_Vertex aVertex = TopoDS::Vertex( anAISPoint->Vertex() );
1348 if ( aVertex.IsNull() )
1351 gp_Pnt aPnt = BRep_Tool::Pnt( aVertex );
1353 SectionToPointList aPoints;
1354 findSectionsToPoints( aPnt.X(), aPnt.Y(), aPoints );
1356 SectionToPointList::const_iterator anIt = aPoints.begin(), aLast = aPoints.end();
1357 SectionToPoint aPoint;
1358 for ( ; anIt != aLast; anIt++ ) {
1360 if ( thePoints.contains( aPoint ) )
1361 aListToSelect.Append( anAIS );
1365 ic->ClearSelected( Standard_False );
1366 aViewer->setObjectsSelected( aListToSelect );
1367 updateLocalPointView();
1371 * Get viewer information before perform the curve modification.
1372 * Take a list of selected cuve points an close local context.
1373 * The context should be closed because the curve presentation is
1374 * redisplayed and if it is not closed, when we close the local context
1375 * later, the presentation shown in the local context is disappeared.
1376 * \param thePoints an output list of curve selected points
1377 * \param theFillPoints a flag whether the selection list should be filled
1379 void CurveCreator_Widget::startCurveModification(
1380 CurveCreator_Widget::SectionToPointList& thePoints,
1381 const bool theFillPoints )
1383 if ( theFillPoints ) {
1385 getSelectedPonts( thePoints );
1387 setLocalPointContext( false );
1391 * Restore the viewer state after the curve modification is done.
1392 * Open local context and select given points inside it.
1393 * \param thePoints a list of curve selected points
1395 void CurveCreator_Widget::finishCurveModification(
1396 const CurveCreator_Widget::SectionToPointList& thePoints )
1398 if ( getActionMode() == ModificationMode )
1399 setLocalPointContext( true );
1400 setSelectedPonts( 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 )
1412 CurveCreator::Coordinates aCoords;
1413 for ( int i = 0, aNb = myCurve->getNbPoints( theSectionId ); i < aNb && aPntIndex < 0; i++ ) {
1414 aCoords = myCurve->getPoint( theSectionId, i );
1415 if ( aCoords.size() < 2 )
1417 if ( fabs( aCoords[0] - theX ) < LOCAL_SELECTION_TOLERANCE &&
1418 fabs( aCoords[1] - theY ) < LOCAL_SELECTION_TOLERANCE )
1425 void CurveCreator_Widget::findSectionsToPoints( const double theX, const double theY,
1426 CurveCreator_Widget::SectionToPointList& thePoints )
1431 for ( int i = 0, aNb = myCurve->getNbSections(); i < aNb; i++ ) {
1432 aPointId = findLocalPointIndex( i, theX, theY );
1435 SectionToPoint aPoint = qMakePair( i, aPointId );
1436 if ( !thePoints.contains( aPoint ) )
1437 thePoints.append( aPoint );
1441 void CurveCreator_Widget::convert( const SectionToPointList& thePoints,
1442 QMap<int, QList<int> >& theConvPoints )
1444 theConvPoints.clear();
1446 SectionToPointList::const_iterator anIt = thePoints.begin(), aLast = thePoints.end();
1448 int aSectionId, aPointId;
1449 for ( ; anIt != aLast; anIt++ ) {
1450 aSectionId = anIt->first;
1451 aPointId = anIt->second;
1453 if ( theConvPoints.contains( aSectionId ) )
1454 aPoints = theConvPoints[aSectionId];
1455 if ( aPoints.contains( aPointId ) )
1457 aPoints.append( aPointId );
1458 theConvPoints[aSectionId] = aPoints;
1463 * Checks whether the point belongs to the OCC object
1464 * \param theObject a line or shape with a bspline inside
1465 * \param theX the X coordinate in the view.
1466 * \param theY the Y coordinate in the view.
1467 * \param thePoint the output point to be append to the model curve
1468 * \param thePoint1 the output point to bound the line where a new point should be inserted
1469 * \param thePoint2 the output point to bound the line where a new point should be inserted
1471 bool CurveCreator_Widget::pointOnObject( Handle(AIS_InteractiveObject) theObject,
1472 const int theX, const int theY,
1474 gp_Pnt& thePoint1, gp_Pnt& thePoint2 )
1476 bool isFound = false;
1478 OCCViewer_Viewer* aViewer = getOCCViewer();
1479 if ( theObject.IsNull() || !aViewer )
1483 Standard_Real aParameter;
1484 gp_Pnt aPnt1, aPnt2;
1485 Handle(AIS_Line) aLine = Handle(AIS_Line)::DownCast( theObject );
1486 if ( !aLine.IsNull() ) {
1487 const Handle(Geom_Line) aGLine = aLine->Line();
1488 isFound = hasProjectPointOnCurve( theX, theY, aGLine, aParameter );
1490 aPoint = aGLine->Value( aParameter );
1492 Handle(Geom_Point) aPStart;
1493 Handle(Geom_Point) aPEnd;
1494 aLine->Points( aPStart, aPEnd );
1495 aPnt1 = aPStart->Pnt();
1496 aPnt2 = aPEnd->Pnt();
1498 // in case of Geom line a projection is performed to the infinite line,
1499 // so it is necessary to bound it by the line size
1501 aLineBox.Set( aPnt1, gp_Vec( aPnt1, aPnt2 ) );
1502 isFound = !aLineBox.IsOut( aPoint );
1506 Handle(AIS_Shape) aShape = Handle(AIS_Shape)::DownCast( theObject );
1507 if ( !aShape.IsNull() ) {
1508 const TopoDS_Wire& aWire = TopoDS::Wire( aShape->Shape() );
1509 if ( !aWire.IsNull() ) {
1510 TopExp_Explorer anExp( aWire, TopAbs_EDGE );
1511 for ( ; anExp.More(); anExp.Next())
1513 const TopoDS_Edge& anEdge = TopoDS::Edge(anExp.Current());
1514 if ( !anEdge.IsNull() ) {
1515 Standard_Real aFirst, aLast;
1516 Handle(Geom_Curve) aCurve = BRep_Tool::Curve( anEdge, aFirst, aLast );
1518 if ( aCurve->IsKind( STANDARD_TYPE(Geom_BSplineCurve) ) ) {
1519 Handle(Geom_BSplineCurve) aBSplineCurve = Handle(Geom_BSplineCurve)::DownCast( aCurve );
1520 if ( !aBSplineCurve.IsNull() ) {
1521 isFound = hasProjectPointOnCurve( theX, theY, aCurve, aParameter );
1523 aPoint = aBSplineCurve->Value( aParameter );
1524 Standard_Integer anI1, anI2;
1525 aBSplineCurve->LocateU( aParameter, LOCAL_SELECTION_TOLERANCE, anI1, anI2 );
1527 aPnt1 = aBSplineCurve->Value( aBSplineCurve->Knot( anI1 ) );
1528 aPnt2 = aBSplineCurve->Value( aBSplineCurve->Knot( anI2 ) );
1546 * Returns whether the clicked point belong to the curve or has a very near projection
1547 * \param theX the X coordinate of a point clicked in the OCC viewer
1548 * \param theY the Y coordinate of a point clicked in the OCC viewer
1549 * \param theCurve a geometry curve
1550 * \param theOutPoint a found projected point on the curve
1552 bool CurveCreator_Widget::hasProjectPointOnCurve( const int theX, const int theY,
1553 const Handle(Geom_Curve)& theCurve,
1554 Standard_Real& theParameter )
1556 bool isFound = false;
1557 OCCViewer_Viewer* aViewer = getOCCViewer();
1561 OCCViewer_ViewWindow* aWindow =
1562 (OCCViewer_ViewWindow*)aViewer->getViewManager()->getActiveView();
1563 Handle(V3d_View) aView = aWindow->getViewPort()->getView();
1564 gp_Pnt aPoint = CurveCreator_Utils::ConvertClickToPoint( theX, theY, aView );
1566 GeomAPI_ProjectPointOnCurve aProj( aPoint, theCurve );
1567 Standard_Integer aNbPoint = aProj.NbPoints();
1569 for (Standard_Integer j = 1; j <= aNbPoint && !isFound; j++) {
1570 gp_Pnt aNewPoint = aProj.Point( j );
1571 theParameter = aProj.Parameter( j );
1574 CurveCreator_Utils::ConvertPointToClick( aNewPoint, aView, aX, anY );
1576 int aXDelta = abs( aX - theX );
1577 int anYDelta = abs( anY - theY );
1578 isFound = aXDelta < SCENE_PIXEL_TOLERANCE && anYDelta < SCENE_PIXEL_TOLERANCE;
1585 * Returns a section index from the table
1586 * \param theRowId a table row
1588 int CurveCreator_Widget::getSectionId( const int theRowId ) const
1590 return myLocalPointView->item( theRowId, 0 )->data( Qt::UserRole ).toInt();
1594 * Returns a point index from the table
1595 * \param theRowId a table row
1597 int CurveCreator_Widget::getPointId( const int theRowId ) const
1599 return myLocalPointView->item( theRowId, 1 )->data( Qt::UserRole ).toInt();