Salome HOME
bug #63: join of sections
[modules/hydro.git] / src / HYDROCurveCreator / CurveCreator_Widget.cxx
index 581ce1491ac4d3d3fa7480689b854584f88cc151..6961f01b44933e1707915be691da17cc9dcdd09f 100644 (file)
@@ -85,7 +85,8 @@ CurveCreator_Widget::CurveCreator_Widget(QWidget* parent,
   myDragStarted( false ), myDragInteractionStyle( SUIT_ViewModel::STANDARD ),
   myOCCViewer( 0 ), myLocalPointRowLimit( theLocalPointRowLimit )
 {
-  myNewSectionEditor = new CurveCreator_NewSectionDlg( this );
+  bool isToEnableClosed = !( theActionFlags & DisableClosedSection );
+  myNewSectionEditor = new CurveCreator_NewSectionDlg( this, isToEnableClosed );
   myNewSectionEditor->hide();
   connect( myNewSectionEditor, SIGNAL(addSection()), this, SLOT(onAddNewSection()) );
   connect( myNewSectionEditor, SIGNAL(modifySection()), this, SLOT(onModifySection()) );
@@ -94,6 +95,7 @@ CurveCreator_Widget::CurveCreator_Widget(QWidget* parent,
   QGroupBox* aSectionGroup = new QGroupBox(tr("Sections"),this);
 
   mySectionView = new CurveCreator_TreeView(myCurve, aSectionGroup);
+  mySectionView->setSelectionMode( QTreeView::ExtendedSelection );
   connect( mySectionView, SIGNAL(selectionChanged()), this, SLOT( onSelectionChanged() ) );
   connect( mySectionView, SIGNAL(sectionEntered(int)), this, SLOT(onEditSection(int)) );
   connect( mySectionView, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(onContextMenu(QPoint)) );
@@ -184,22 +186,20 @@ CurveCreator_Widget::CurveCreator_Widget(QWidget* parent,
   connect(anAct, SIGNAL(triggered()), this, SLOT(onRemove()) );
   aTB->addAction(anAct);
   
-  // TODO join
-  //aTB->addSeparator();
+  aTB->addSeparator();
 
-  //anAct = createAction( JOIN_ID, tr("JOIN"), aJoinPixmap, tr("JOIN_TLT"), 
-  //                      QKeySequence(Qt::ControlModifier|Qt::Key_Plus ) );
-  //connect( anAct, SIGNAL(triggered()), this, SLOT(onJoin()) );
-  //aTB->addAction(anAct);
+  anAct = createAction( JOIN_ID, tr("JOIN"), aJoinPixmap, tr("JOIN_TLT"), 
+                        QKeySequence(Qt::ControlModifier|Qt::Key_Plus ) );
+  connect( anAct, SIGNAL(triggered()), this, SLOT(onJoin()) );
+  aTB->addAction(anAct);
 
   anAct = createAction( CLEAR_ALL_ID, tr("CLEAR_ALL"), QPixmap(), tr("CLEAR_ALL_TLT"), 
                         QKeySequence(Qt::ControlModifier | Qt::ShiftModifier | Qt::Key_Delete ) );
   connect( anAct, SIGNAL(triggered()), this, SLOT( onClearAll()) );
 
-  // TODO join
-  //anAct = createAction( JOIN_ALL_ID, tr("JOIN_ALL"), QPixmap(), tr("JOIN_ALL_TLT"), 
-  //                      QKeySequence(Qt::ControlModifier | Qt::ShiftModifier | Qt::Key_Plus ) );
-  //connect( anAct, SIGNAL(triggered()), this, SLOT(onJoinAll()) );
+  anAct = createAction( JOIN_ALL_ID, tr("JOIN_ALL"), QPixmap(), tr("JOIN_ALL_TLT"), 
+                        QKeySequence(Qt::ControlModifier | Qt::ShiftModifier | Qt::Key_Plus ) );
+  connect( anAct, SIGNAL(triggered()), this, SLOT(onJoinAll()) );
 
   QVBoxLayout* aSectLayout = new QVBoxLayout();
   aSectLayout->setMargin( 5 );
@@ -372,18 +372,25 @@ void CurveCreator_Widget::updateActionsStates()
           int aSectCnt = myCurve->getNbSections();
           if( aSectCnt > 0 )
             anEnabledAct << CLEAR_ALL_ID;
-          // TODO
-          //if( aSectCnt > 1 )
-          //  anEnabledAct << JOIN_ALL_ID;
-          //if( aSelSections.size() > 1 ){
-          //  anEnabledAct << JOIN_ID;
-          //}
+          if( aSectCnt > 1 )
+            anEnabledAct << JOIN_ALL_ID;
+          if( aSelSections.size() > 1 ){
+            anEnabledAct << JOIN_ID;
+          }
         }
         break;
         case DetectionMode: {
         }
         break;
         case NoneMode:
+          {
+            int aSectCnt = myCurve->getNbSections();
+            if( aSectCnt > 1 )
+              anEnabledAct << JOIN_ALL_ID;
+            if( aSelSections.size() > 1 )
+              anEnabledAct << JOIN_ID;
+          }
+          break;
         default:
         break;
       }
@@ -416,10 +423,9 @@ void CurveCreator_Widget::updateActionsStates()
     if( (myCurve->getNbSections() + myCurve->getNbPoints()) > 0 ){
       anEnabledAct << REMOVE_ALL_ID;
     }*/
-    // TODO
-    //if( myCurve->getNbSections() > 1 ){
-    //  anEnabledAct << JOIN_ALL_ID;
-    //}
+    if( myCurve->getNbSections() > 1 ){
+      anEnabledAct << JOIN_ALL_ID;
+    }
   }
   QList<ActionId> anIds = myActionMap.keys();
   for( int i = 0 ; i < anIds.size() ; i++ ){
@@ -512,7 +518,7 @@ void CurveCreator_Widget::onNewSection()
   myNewSectionEditor->setEditMode(false);
   QString aSectName = QString( CurveCreator_UtilsICurve::getUniqSectionName( myCurve ).c_str() );
   myNewSectionEditor->setSectionParameters(aSectName, true, CurveCreator::Polyline );
-  emit subOperationStarted( myNewSectionEditor );
+  emit subOperationStarted( myNewSectionEditor, false );
 }
 
 void CurveCreator_Widget::onAddNewSection()
@@ -634,7 +640,7 @@ void CurveCreator_Widget::onEditSection( int theSection )
   myNewSectionEditor->setEditMode(true);
   myNewSectionEditor->setSectionParameters( aSectName, isClosed, aType );
 
-  emit subOperationStarted( myNewSectionEditor );
+  emit subOperationStarted( myNewSectionEditor, true );
 }
 
 void CurveCreator_Widget::onModifySection()
@@ -713,7 +719,7 @@ void CurveCreator_Widget::onJoinAll()
   if( !myCurve )
     return;
   stopActionMode();
-  myCurve->join();
+  myCurve->join( 0, myCurve->getNbSections() );
   mySectionView->reset();
   updateActionsStates();
   updateUndoRedo();
@@ -826,7 +832,7 @@ void CurveCreator_Widget::updateUndoRedo()
 void CurveCreator_Widget::onContextMenu( QPoint thePoint )
 {
   QList<ActionId> aContextActions;
-  aContextActions << CLEAR_ALL_ID << JOIN_ALL_ID << SEPARATOR_ID <<
+  aContextActions << CLEAR_ALL_ID << JOIN_ID << JOIN_ALL_ID << SEPARATOR_ID <<
                      CLOSE_SECTIONS_ID << UNCLOSE_SECTIONS_ID << SET_SECTIONS_POLYLINE_ID <<
                      SET_SECTIONS_SPLINE_ID;
   QPoint aGlPoint = mySectionView->mapToGlobal(thePoint);
@@ -971,16 +977,48 @@ void CurveCreator_Widget::onMouseRelease( SUIT_ViewWindow*, QMouseEvent* theEven
   if ( myDragStarted ) {
     bool isDragged = myDragged;
     CurveCreator_ICurve::SectionToPointList aDraggedPoints;
-    if ( myDragged )
+    CurveCreator_ICurve::SectionToPointCoordsList anInitialDragPointsCoords;
+    if ( myDragged ) {
       aDraggedPoints = myDragPoints;
+      anInitialDragPointsCoords = myInitialDragPointsCoords;
+    }
 
     setDragStarted( false );
-    // if the drag of some points has happened, restore the drag selection
-    if ( aDraggedPoints.size() > 0 )
-    {
-      START_MEASURE_TIME;
-      setSelectedPoints( aDraggedPoints );
-      END_MEASURE_TIME( "drop" );
+
+    if ( aDraggedPoints.size() > 0 ) {
+      if ( myCurve->canPointsBeSorted() ) {
+        // Apply points sorting
+        CurveCreator_ICurve::SectionToPointList aPoints;
+        startCurveModification( aPoints, false );
+
+        myCurve->setSkipSorting( false );
+
+        CurveCreator_ICurve::SectionToPointCoordsList aCoordList;
+        CurveCreator_ICurve::SectionToPointList::const_iterator anIt = aDraggedPoints.begin(),
+                                                                aLast = aDraggedPoints.end();
+        for ( ; anIt != aLast; anIt++ ) {
+          int aSectionId = anIt->first;
+          int aPointId = anIt->second;
+          std::deque<float> aPos = myCurve->getPoint( aSectionId, aPointId );
+
+          aCoordList.push_back(
+            std::make_pair( std::make_pair( aSectionId, aPointId ), aPos ) );
+        }
+
+        myCurve->setSeveralPoints( aCoordList, false );
+    
+        finishCurveModification( aDraggedPoints );
+      } else {
+        // if the drag of some points has happened, restore the drag selection
+        if ( aDraggedPoints.size() > 0 ) {
+          START_MEASURE_TIME;
+          setSelectedPoints( aDraggedPoints );
+          END_MEASURE_TIME( "drop" );
+        }
+      }
+
+      // Save drag difference
+      myCurve->saveCoordDiff( anInitialDragPointsCoords );
     }
   }
   else // check whether the segment is clicked an a new point should be added to the segment
@@ -1212,6 +1250,14 @@ void CurveCreator_Widget::moveSelectedPoints( const int theXPosition,
     aChangedPos = myCurve->getPoint( aSectionId, aPointId );
     if ( aChangedPos.size() < 2 )
       continue;
+
+    // Remember drag points coordinates
+    if ( !myDragged ) {
+      myInitialDragPointsCoords.push_back(
+        std::make_pair(std::make_pair( aSectionId, aPointId ), 
+                       aChangedPos ));
+    }
+
     aChangedPos[0] = aChangedPos[0] - aXDelta;
     aChangedPos[1] = aChangedPos[1] - anYDelta;
     
@@ -1219,7 +1265,7 @@ void CurveCreator_Widget::moveSelectedPoints( const int theXPosition,
       std::make_pair(std::make_pair( aSectionId, aPointId ), 
                      aChangedPos ));
   }
-  myCurve->setSeveralPoints( aCoordList );
+  myCurve->setSeveralPoints( aCoordList, false );
 
   myDragged = true;
   finishCurveModification( myDragPoints );
@@ -1235,6 +1281,7 @@ void CurveCreator_Widget::updateLocalPointView()
 
   CurveCreator_Utils::getSelectedPoints( aContext, myCurve, myLocalPoints );
   int aNbPoints = myLocalPoints.size();
+
   bool isRowLimit = aNbPoints > myLocalPointRowLimit;
   myLocalPointView->setVisible( getActionMode() == ModificationMode && !isRowLimit );
 
@@ -1269,11 +1316,13 @@ void CurveCreator_Widget::setDragStarted( const bool theState, const QPoint& the
 {
   if ( theState ) {
     getSelectedPoints( myDragPoints );
+
     myDragStarted = myDragPoints.size();
     myDragStartPosition = thePoint;
     if ( myDragStarted ) {
       // change a viewer interaction style in order to avoid a select rectangle build
       myDragInteractionStyle = changeInteractionStyle( SUIT_ViewModel::KEY_FREE );
+      myCurve->setSkipSorting( true );
     }
   }
   else {
@@ -1281,6 +1330,7 @@ void CurveCreator_Widget::setDragStarted( const bool theState, const QPoint& the
       changeInteractionStyle( myDragInteractionStyle );
     myDragStarted = false;
     myDragPoints.clear();
+    myInitialDragPointsCoords.clear();
   }
   myDragged = false;
 }