Merge branch 'BR_v14_rc' into BR_quadtree

[modules/hydro.git] / src / HYDROGUI / HYDROGUI_CalculationOp.cxx

diff --git a/src/HYDROGUI/HYDROGUI_CalculationOp.cxx b/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
index 8ace99e3218493a22b31b33848d9e366e47780c7..8ca26d1c48ff4c0df1caa6a6d667cfb72cb40d43 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
@@ -55,7 +55,9 @@ HYDROGUI_CalculationOp::HYDROGUI_CalculationOp( HYDROGUI_Module* theModule, bool
 : HYDROGUI_Operation( theModule ),
   myIsEdit( theIsEdit ),
   myActiveViewManager( NULL ),
-  myPreviewViewManager( NULL )
+  myPreviewViewManager( NULL ),
+  myShowGeomObjects( true ),
+  myShowZones( false )
 {
   setName( myIsEdit ? tr( "EDIT_CALCULATION" ) : tr( "CREATE_CALCULATION" ) );
 }
@@ -149,25 +151,7 @@ void HYDROGUI_CalculationOp::startOperation()
       aPanel->includeGeomObjects( aList );
 
       // set rules
-      HYDROData_ListOfRules aRules;
-      Handle(HYDROData_Object) anObject1, anObject2;
-      HYDROData_PriorityType aPriority;
-      HYDROData_Zone::MergeAltitudesType aMergeType;
-      for ( int anIndex = 0; ; anIndex++ )
-      {
-        if ( myEditedObject->GetRule( anIndex, anObject1, aPriority, anObject2, aMergeType ) ) {
-          HYDROData_CustomRule aRule;
-          aRule.Object1 = anObject1;
-          aRule.Object2 = anObject2;
-          aRule.Priority = aPriority;
-          aRule.MergeType = aMergeType;
-
-          aRules << aRule;
-        }
-        else
-          break;
-      }
-      aPanel->setRules( aRules );
+      setRules( HYDROData_CalculationCase::DataTag_CustomRules );      
     }
   }
   else
@@ -238,7 +222,7 @@ HYDROGUI_InputPanel* HYDROGUI_CalculationOp::createInputPanel() const
   connect( aPanel, SIGNAL( orderLandCoverChanged( bool& ) ), SLOT( onOrderLandCoverChanged( bool& ) ) );
     
   connect( aPanel, SIGNAL( Next( const int ) ), SLOT( onNext( const int ) ) );
-  connect( aPanel, SIGNAL( Back( const int ) ), SLOT( onHideZones() ) );
+  connect( aPanel, SIGNAL( Back( const int ) ), SLOT( onBack( const int ) ) );
   //connect( aPanel, SIGNAL( clicked( SUIT_DataObject* ) ), SLOT( onSelected( SUIT_DataObject* ) ) );
   connect( aPanel, SIGNAL( setMergeType( int, QString& ) ), SLOT( onSetMergeType( int, QString& ) ) );
   connect( aPanel, SIGNAL( moveZones( SUIT_DataObject*, const QList<SUIT_DataObject*>& ) ),
@@ -783,10 +767,10 @@ bool HYDROGUI_CalculationOp::processApply( int&     theUpdateFlags,
 
   // For manual mode priority rules are redundant
   if ( aPanel->getMode() == HYDROData_CalculationCase::MANUAL ) {
-    myEditedObject->ClearRules( false );
+    myEditedObject->ClearRules( HYDROData_CalculationCase::DataTag_CustomRules, false );
   }
   if ( aPanel->getLandCoverMode() == HYDROData_CalculationCase::MANUAL ) {
-    myEditedObject->ClearLandCoverRules( false );
+    myEditedObject->ClearRules( HYDROData_CalculationCase::DataTag_CustomLandCoverRules, false );
   }
  
   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init;
@@ -869,13 +853,15 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
 
     QStringList aList;
     QStringList anEntryList;
+    HYDROData_SequenceOfObjects aSeq;
+
+    Handle(HYDROData_StricklerTable) aStricklerTableObj;
+    QString aStricklerTableName;
 
     // Get all Strickler table objects to fill in combo-box
     aList.clear();
     anEntryList.clear();
-    HYDROData_Iterator anIter( doc(), KIND_STRICKLER_TABLE );
-    Handle(HYDROData_StricklerTable) aStricklerTableObj;
-    QString aStricklerTableName;
+    HYDROData_Iterator anIter( doc(), KIND_STRICKLER_TABLE );      
     for ( ; anIter.More(); anIter.Next() )
     {
       aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( anIter.Current() );
@@ -891,11 +877,18 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
       }
     }
     aPanel->setStricklerTableNames( aList, anEntryList );
+    if ( !aList.isEmpty() )
+      aPanel->setStricklerTable( aList.at( 0 ), false );
 
     // Fill in list widget with all available land covers
-    HYDROData_SequenceOfObjects aSeq = HYDROGUI_Tool::GetLandCovers( module() );
+    aSeq = HYDROGUI_Tool::GetLandCovers( module() );
     getNamesAndEntries( aSeq, aList, anEntryList );
     aPanel->setAllLandCovers( aList, anEntryList );
+
+    // Set list of included land covers
+    aSeq = myEditedObject->GetLandCovers();
+    getNamesAndEntries( aSeq, aList, anEntryList );
+    aPanel->includeLandCovers( aList, true );
     
     if ( !myEditedObject.IsNull() )
     {
@@ -915,6 +908,9 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
 
         // Set mode (Auto or Manual) to defined priority of land covers
         aPanel->setLandCoverMode( myEditedObject->GetAssignmentLandCoverMode() );
+
+        // Set rules defined on land covers
+        setRules( HYDROData_CalculationCase::DataTag_CustomLandCoverRules );
       }
       else
       {
@@ -930,6 +926,8 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
       return;
 
     QApplication::setOverrideCursor( Qt::WaitCursor );
+
+    setGeomObjectsVisible( false );
   
     QString aNewCaseName = aPanel->getObjectName();
     QString anOldCaseName = myEditedObject->GetName();
@@ -962,12 +960,13 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
 
         // Clear priority rules
         //@ASL if ( myEditedObject->GetRulesCount() > 0 ) {
-          myEditedObject->ClearRules( true );
+          myEditedObject->ClearRules( HYDROData_CalculationCase::DataTag_CustomRules, true );
         //@ASL }
         // Set priority rules
         foreach ( const HYDROData_CustomRule& aRule, aPanel->getRules() ) {
           myEditedObject->AddRule( aRule.Object1, aRule.Priority,
-                                  aRule.Object2, aRule.MergeType );
+                                  aRule.Object2, aRule.MergeType,
+                                  HYDROData_CalculationCase::DataTag_CustomRules );
         }
       }
     }
@@ -987,6 +986,8 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
       createPreview();
 
       anIsToUpdateOb = true;
+
+      myEditedObject->SetToUpdate( true );
     }
     else
     {
@@ -997,6 +998,7 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
       }
     }
 
+    // Move this code to processing od the last panel (see below)
     if ( anIsToUpdateOb ) {
       SUIT_DataBrowser* anObjBrowser = ((LightApp_Application*)module()->application())->objectBrowser();
       if ( anObjBrowser ) {
@@ -1016,8 +1018,6 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
 
     QApplication::setOverrideCursor( Qt::WaitCursor );
 
-    //...
-
     // Set parameters for automatic mode
     int aMode = aPanel->getLandCoverMode();
     if ( aMode == HYDROData_CalculationCase::AUTOMATIC )
@@ -1038,11 +1038,13 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
         }
 
         // Clear priority rules
-        myEditedObject->ClearLandCoverRules( true );
-        // TODO: Set priority rules
-        //foreach ( const HYDROData_CustomRule& aRule, aPanel->getLandCoverRules() ) {
-        //  myEditedObject->AddLandCoverRule( aRule.Object1, aRule.Priority,
-        //                                    aRule.Object2, aRule.MergeType );                
+        myEditedObject->ClearRules( HYDROData_CalculationCase::DataTag_CustomLandCoverRules, true );
+        // Set priority rules
+        foreach ( const HYDROData_CustomRule& aRule, aPanel->getLandCoverRules() ) {
+          myEditedObject->AddRule( aRule.Object1, aRule.Priority,
+                                   aRule.Object2, HYDROData_Zone::Merge_Object,
+                                   HYDROData_CalculationCase::DataTag_CustomLandCoverRules );
+        }
       }
     }
     aPanel->setEditLandCoversEnabled( aMode == HYDROData_CalculationCase::MANUAL );
@@ -1051,8 +1053,9 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
   }
 }
 
-void HYDROGUI_CalculationOp::onHideZones()
+void HYDROGUI_CalculationOp::onBack( const int theIndex )
 {
+  setGeomObjectsVisible( theIndex != 2 );
   setZonesVisible( false );
 }
 
@@ -1090,6 +1093,18 @@ void HYDROGUI_CalculationOp::setZonesVisible( bool theIsVisible )
   }
 }
 
+void HYDROGUI_CalculationOp::setGeomObjectsVisible( bool theIsVisible )
+{
+  myShowGeomObjects = theIsVisible;
+
+  HYDROData_SequenceOfObjects aSeq = myEditedObject->GetGeometryObjects();
+
+  HYDROData_SequenceOfObjects::Iterator anIter( aSeq );
+  for ( ; anIter.More(); anIter.Next() ) {
+    setObjectVisibility( anIter.Value(), theIsVisible );
+  }
+}
+
 void HYDROGUI_CalculationOp::AssignDefaultZonesColors()
 {
   HYDROData_SequenceOfObjects aRegions = myEditedObject->GetRegions();
@@ -1161,10 +1176,48 @@ QColor HYDROGUI_CalculationOp::GenerateDefaultZonesColor( int theIndex,
   return ( aColor.isValid() ? aColor : HYDROData_ImmersibleZone::DefaultFillingColor() );
 }
 
+void HYDROGUI_CalculationOp::setRules( HYDROData_CalculationCase::DataTag theDataTag )
+{
+  HYDROGUI_CalculationDlg* aPanel = 
+    ::qobject_cast<HYDROGUI_CalculationDlg*>( inputPanel() );
+  if ( !aPanel )
+    return;
+
+  HYDROData_ListOfRules aRules;
+  Handle(HYDROData_Entity) anObject1, anObject2;
+  HYDROData_PriorityType aPriority;
+  HYDROData_Zone::MergeAltitudesType aMergeType;
+  HYDROData_CalculationCase::DataTag aDataTag = HYDROData_CalculationCase::DataTag_CustomRules;
+  for ( int anIndex = 0; ; anIndex++ )
+  {
+    if ( myEditedObject->GetRule( anIndex, anObject1, aPriority, anObject2, aMergeType, theDataTag ) ) {
+      HYDROData_CustomRule aRule;
+      aRule.Object1 = anObject1;
+      aRule.Object2 = anObject2;
+      aRule.Priority = aPriority;
+      aRule.MergeType = aMergeType;
+
+      aRules << aRule;
+    }
+    else
+      break;
+  }
+
+  if ( theDataTag == HYDROData_CalculationCase::DataTag_CustomRules )
+    aPanel->setRules( aRules );
+  else if ( theDataTag == HYDROData_CalculationCase::DataTag_CustomLandCoverRules )
+    aPanel->setLandCoverRules( aRules );
+}
+
 void HYDROGUI_CalculationOp::createPreview()
 {
   LightApp_Application* anApp = module()->getApp();
-  HYDROData_SequenceOfObjects aSeq = myEditedObject->GetGeometryObjects();
+  HYDROData_SequenceOfObjects aSeq;
+  if ( myShowGeomObjects )
+  {
+    HYDROData_SequenceOfObjects aGeomObj = myEditedObject->GetGeometryObjects();
+    aSeq.Append( aGeomObj );
+  }
   Handle(HYDROData_Entity) anEntity;
 
   if ( myShowZones )
@@ -1437,7 +1490,7 @@ void HYDROGUI_CalculationOp::onAddLandCovers()
 
   if ( !anAddedList.isEmpty() )
   {
-    aPanel->includeLandCovers( anAddedList );
+    aPanel->includeLandCovers( anAddedList, false );
     // TODO: create preview of included land covers
     //createPreview();
   }