#571 - Land Cover: calculation of Strickler coefficient

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

diff --git a/src/HYDROGUI/HYDROGUI_CalculationOp.cxx b/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
index f8311d5a4f90c45bb0ef1dfa8626dd9995559e35..82a21857c12505f3c47ecfe19f151277d1d9f98c 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
@@ -163,7 +163,7 @@ void HYDROGUI_CalculationOp::startOperation()
   aPanel->setObjectName( anObjectName );
   aPanel->setEditedObject( myEditedObject );
 
-  createPreview();
+  createPreview( false );
 }
 
 void HYDROGUI_CalculationOp::getNamesAndEntries( const HYDROData_SequenceOfObjects& theSeq, 
@@ -220,11 +220,12 @@ 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( onHideZones( 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*>& ) ),
-    SLOT( onMoveZones( SUIT_DataObject*, const QList<SUIT_DataObject*>& ) ) );
+  connect( aPanel, SIGNAL( setMergeStricklerType( int, QString& ) ), SLOT( onSetMergeStricklerType( int, QString& ) ) );
+  connect( aPanel, SIGNAL( moveZones( SUIT_DataObject*, const QList<SUIT_DataObject*>&, bool ) ),
+    SLOT( onMoveZones( SUIT_DataObject*, const QList<SUIT_DataObject*>&, bool ) ) );
   connect( aPanel, SIGNAL( createRegion( const QList<SUIT_DataObject*>& ) ),
     SLOT( onCreateRegion( const QList<SUIT_DataObject*>& ) ) );
   connect( aPanel, SIGNAL( clickedInZonesBrowser( SUIT_DataObject* ) ),
@@ -233,6 +234,8 @@ HYDROGUI_InputPanel* HYDROGUI_CalculationOp::createInputPanel() const
            SLOT( onObjectsSelected() ) );
   connect( aPanel, SIGNAL( landCoversSelected() ), 
            SLOT( onLandCoversSelected() ) );
+  connect( aPanel, SIGNAL( createLandCoverRegion( const QList<SUIT_DataObject*>& ) ),
+    SLOT( onCreateLandCoverRegion( const QList<SUIT_DataObject*>& ) ) );
   connect( aPanel, SIGNAL( boundarySelected( const QString & ) ), 
     SLOT( onBoundarySelected( const QString & ) ) );
   connect( aPanel, SIGNAL( StricklerTableSelected( const QString & ) ), 
@@ -458,7 +461,9 @@ void HYDROGUI_CalculationOp::onClickedInZonesBrowser( SUIT_DataObject* theItem )
   }
 }
 
-void HYDROGUI_CalculationOp::onMoveZones( SUIT_DataObject* theRegionItem, const QList<SUIT_DataObject*>& theZonesList )
+void HYDROGUI_CalculationOp::onMoveZones( SUIT_DataObject* theRegionItem,
+                                          const QList<SUIT_DataObject*>& theZonesList,
+                                          bool theLandCover )
 {
   HYDROGUI_Region* aRegion = dynamic_cast<HYDROGUI_Region*>(theRegionItem);
   if ( aRegion )
@@ -481,40 +486,42 @@ void HYDROGUI_CalculationOp::onMoveZones( SUIT_DataObject* theRegionItem, const
         ::qobject_cast<HYDROGUI_CalculationDlg*>( inputPanel() );
       if ( aPanel )
       {
-        aPanel->refreshZonesBrowser();
+        theLandCover ? aPanel->refreshLandCoverZonesBrowser(): aPanel->refreshZonesBrowser();
       }
-      createPreview();
+      createPreview( theLandCover );
     }
   }
 }
 
 void HYDROGUI_CalculationOp::onCreateRegion( const QList<SUIT_DataObject*>& theZonesList )
 {
-  QList<HYDROGUI_Zone*> aZonesList;
-  HYDROGUI_Zone* aZone;
-  // Get a list of dropped zones
-  for ( int i = 0; i < theZonesList.length(); i++ )
+  if ( createRegion( theZonesList, false ) )
   {
-    aZone = dynamic_cast<HYDROGUI_Zone*>( theZonesList.at( i ) );
-    if ( aZone )
+    HYDROGUI_CalculationDlg* aPanel = 
+      ::qobject_cast<HYDROGUI_CalculationDlg*>( inputPanel() );
+    if ( aPanel )
     {
-      aZonesList.append( aZone );
+      aPanel->refreshZonesBrowser();
     }
+    createPreview( false );
   }
-  if ( aZonesList.length() > 0 )
+}
+
+void HYDROGUI_CalculationOp::onCreateLandCoverRegion( const QList<SUIT_DataObject*>& theZonesList )
+{
+  if ( createRegion( theZonesList, true ) )
   {
-    module()->getDataModel()->createNewRegion( myEditedObject, aZonesList );
     HYDROGUI_CalculationDlg* aPanel = 
       ::qobject_cast<HYDROGUI_CalculationDlg*>( inputPanel() );
     if ( aPanel )
     {
-      aPanel->refreshZonesBrowser();
+      aPanel->refreshLandCoverZonesBrowser();
     }
-    createPreview();
+    createPreview( true );
   }
 }
 
-void HYDROGUI_CalculationOp::onSetMergeType( int theMergeType, QString& theAltitudeName )
+void HYDROGUI_CalculationOp::onSetMergeType( int theMergeType, QString& theMergeObjectName )
 {
   HYDROGUI_CalculationDlg* aPanel = 
     ::qobject_cast<HYDROGUI_CalculationDlg*>( inputPanel() );
@@ -523,7 +530,7 @@ void HYDROGUI_CalculationOp::onSetMergeType( int theMergeType, QString& theAltit
     HYDROGUI_Zone* aZone = aPanel->getCurrentZone();
     if ( aZone )
     {
-      aZone->setMergeType( theMergeType, theAltitudeName );
+      aZone->setMergeType( theMergeType, theMergeObjectName );
       HYDROGUI_Shape* aShape = module()->getObjectShape( HYDROGUI_Module::VMR_PreviewCaseZones, aZone->modelObject() );
       if ( aShape )
       {
@@ -534,6 +541,26 @@ void HYDROGUI_CalculationOp::onSetMergeType( int theMergeType, QString& theAltit
   }
 }
 
+void HYDROGUI_CalculationOp::onSetMergeStricklerType( int theMergeType, QString& theStricklerTypeName )
+{
+  HYDROGUI_CalculationDlg* aPanel = 
+    ::qobject_cast<HYDROGUI_CalculationDlg*>( inputPanel() );
+  if ( aPanel )
+  {
+    HYDROGUI_Zone* aZone = aPanel->getCurrentZone();
+    if ( aZone )
+    {
+      aZone->setMergeType( theMergeType, theStricklerTypeName );
+      HYDROGUI_Shape* aShape = module()->getObjectShape( HYDROGUI_Module::VMR_PreviewCaseZones, aZone->modelObject() );
+      if ( aShape )
+      {
+        aShape->update( true, false );
+      }
+    }
+    aPanel->refreshLandCoverZonesBrowser();
+  }
+}
+
 void HYDROGUI_CalculationOp::onAddObjects()
 {
   HYDROGUI_CalculationDlg* aPanel = 
@@ -561,7 +588,7 @@ void HYDROGUI_CalculationOp::onAddObjects()
   if ( !anAddedList.isEmpty() )
   {
     aPanel->includeGeomObjects( anAddedList );
-    createPreview();
+    createPreview( false );
   }
 }
 
@@ -599,7 +626,7 @@ bool HYDROGUI_CalculationOp::confirmRegionsChange() const
   if ( !isConfirmed )
   {
     // If not modified check if the case has already defined regions with zones
-    HYDROData_SequenceOfObjects aSeq = myEditedObject->GetRegions();
+    HYDROData_SequenceOfObjects aSeq = myEditedObject->GetRegions( false );
     if ( aSeq.Length() > 0 )
     {
       // If there are already defined zones then ask a user to confirm zones recalculation
@@ -624,7 +651,7 @@ bool HYDROGUI_CalculationOp::confirmOrderChange() const
   if ( !isConfirmed )
   {
     // If not modified check if the case has already defined regions with zones
-    HYDROData_SequenceOfObjects aSeq = myEditedObject->GetRegions();
+    HYDROData_SequenceOfObjects aSeq = myEditedObject->GetRegions( false );
     if ( aSeq.Length() > 0 )
     {
       // If there are already defined zones then ask a user to confirm zones recalculation
@@ -649,7 +676,7 @@ bool HYDROGUI_CalculationOp::confirmModeChange() const
   if ( !isConfirmed )
   {
     // If not modified check if the case has already defined regions with zones
-    HYDROData_SequenceOfObjects aSeq = myEditedObject->GetRegions();
+    HYDROData_SequenceOfObjects aSeq = myEditedObject->GetRegions( false );
     if ( aSeq.Length() > 0 )
     {
       // If there are already defined zones then ask a user to confirm zones recalculation
@@ -915,6 +942,9 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
         myEditedObject->SetAssignmentLandCoverMode( (HYDROData_CalculationCase::AssignmentMode)aPanel->getLandCoverMode() );
       }
     }
+
+    closePreview( false );
+    createPreview( true );
   }
   else if( theIndex==3 )
   {
@@ -973,13 +1003,13 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
       myShowZones = true;
       myEditedObject->Update();
       
-      AssignDefaultZonesColors();
+      AssignDefaultZonesColors( false );
 
       //aPanel->setEditedObject( myEditedObject );
       aPanel->refreshZonesBrowser();
     
-      closePreview();
-      createPreview();
+      closePreview( false );
+      createPreview( false );
 
       anIsToUpdateOb = true;
 
@@ -987,21 +1017,14 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
     }
     else
     {
-      setZonesVisible( true );
+      // Show zones
+      setZonesVisible( true, false );
 
       if ( isNameChanged ) {
         module()->getDataModel()->updateObjectTree( myEditedObject );
       }
     }
 
-    // Move this code to processing od the last panel (see below)
-    if ( anIsToUpdateOb ) {
-      SUIT_DataBrowser* anObjBrowser = ((LightApp_Application*)module()->application())->objectBrowser();
-      if ( anObjBrowser ) {
-        anObjBrowser->updateTree( module()->getDataModel()->getDataObject( myEditedObject ), false );
-      }
-    }
-
     QApplication::restoreOverrideCursor();
   }
   else if( theIndex==4 )
@@ -1043,21 +1066,79 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
         }
       }
     }
-    aPanel->setEditLandCoversEnabled( aMode == HYDROData_CalculationCase::MANUAL );
+    aPanel->setEditLandCoverZonesEnabled( aMode == HYDROData_CalculationCase::MANUAL );
+
+    bool anIsToUpdateOb = false;
+    bool anIsToUpdate = myEditedObject->IsMustBeUpdated();
+    if ( anIsToUpdate )
+    {
+      myShowZones = true;
+      myEditedObject->Update();
+      
+      AssignDefaultZonesColors( true );
+
+      aPanel->refreshLandCoverZonesBrowser();
+
+      anIsToUpdateOb = true;
+    }
+
+    closePreview( false );
+    createPreview( true );
+    
+    if ( !anIsToUpdate )
+    {
+      // Hide zones
+      setZonesVisible( false, false );
+      // Show land cover zones
+      setZonesVisible( true, true );
+    }
+
+    if ( anIsToUpdateOb ) {
+      SUIT_DataBrowser* anObjBrowser = ((LightApp_Application*)module()->application())->objectBrowser();
+      if ( anObjBrowser ) {
+        anObjBrowser->updateTree( module()->getDataModel()->getDataObject( myEditedObject ), false );
+      }
+    }
 
     QApplication::restoreOverrideCursor();
   }
 }
 
-void HYDROGUI_CalculationOp::onHideZones()
+void HYDROGUI_CalculationOp::onHideZones( const int theIndex )
 {
-  setZonesVisible( false );
+  if( theIndex==1 )
+  {
+    closePreview( false );
+    createPreview( false );
+  }
+  if( theIndex==2 )
+  {
+    closePreview( false );
+    createPreview( true );
+
+    // Hide zones
+    setZonesVisible( false, false );
+    // Hide land covers
+    setZonesVisible( false, true );
+  }
+  else if( theIndex==3 )
+  {
+    AssignDefaultZonesColors( false );
+
+    closePreview( false );
+    createPreview( false );
+
+    // Hide land cover zones
+    setZonesVisible( false, true );
+    // Show zones
+    setZonesVisible( true, false );
+  }
 }
 
-void HYDROGUI_CalculationOp::setZonesVisible( bool theIsVisible )
+void HYDROGUI_CalculationOp::setZonesVisible( bool theIsVisible, const bool theLandCover )
 {
   myShowZones = theIsVisible;
-  HYDROData_SequenceOfObjects aRegions = myEditedObject->GetRegions();
+  HYDROData_SequenceOfObjects aRegions = myEditedObject->GetRegions( theLandCover );
   HYDROData_SequenceOfObjects::Iterator aRegionsIter( aRegions );
   HYDROData_SequenceOfObjects aZones;
   Handle(HYDROData_Region) aRegion;
@@ -1088,9 +1169,9 @@ void HYDROGUI_CalculationOp::setZonesVisible( bool theIsVisible )
   }
 }
 
-void HYDROGUI_CalculationOp::AssignDefaultZonesColors()
+void HYDROGUI_CalculationOp::AssignDefaultZonesColors( const bool theLandCover )
 {
-  HYDROData_SequenceOfObjects aRegions = myEditedObject->GetRegions();
+  HYDROData_SequenceOfObjects aRegions = myEditedObject->GetRegions( theLandCover );
   HYDROData_SequenceOfObjects::Iterator aRegionsIter( aRegions );
   HYDROData_SequenceOfObjects aZones;
   Handle(HYDROData_Region) aRegion;
@@ -1169,7 +1250,7 @@ void HYDROGUI_CalculationOp::setRules( HYDROData_CalculationCase::DataTag theDat
   HYDROData_ListOfRules aRules;
   Handle(HYDROData_Entity) anObject1, anObject2;
   HYDROData_PriorityType aPriority;
-  HYDROData_Zone::MergeAltitudesType aMergeType;
+  HYDROData_Zone::MergeType aMergeType;
   HYDROData_CalculationCase::DataTag aDataTag = HYDROData_CalculationCase::DataTag_CustomRules;
   for ( int anIndex = 0; ; anIndex++ )
   {
@@ -1192,16 +1273,45 @@ void HYDROGUI_CalculationOp::setRules( HYDROData_CalculationCase::DataTag theDat
     aPanel->setLandCoverRules( aRules );
 }
 
-void HYDROGUI_CalculationOp::createPreview()
+bool HYDROGUI_CalculationOp::createRegion( const QList<SUIT_DataObject*>& theZonesList,
+                                           const bool theLandCover )
+{
+  bool aRetValue = false;
+
+  QList<HYDROGUI_Zone*> aZonesList;
+  HYDROGUI_Zone* aZone;
+  // Get a list of dropped zones
+  for ( int i = 0; i < theZonesList.length(); i++ )
+  {
+    aZone = dynamic_cast<HYDROGUI_Zone*>( theZonesList.at( i ) );
+    if ( aZone )
+    {
+      aZonesList.append( aZone );
+    }
+  }
+  if ( aZonesList.length() > 0 )
+  {
+    module()->getDataModel()->createNewRegion( myEditedObject, aZonesList, theLandCover );
+    aRetValue = true;
+  }
+   
+  return aRetValue;
+}
+
+void HYDROGUI_CalculationOp::createPreview( const bool theLandCover )
 {
   LightApp_Application* anApp = module()->getApp();
-  HYDROData_SequenceOfObjects aSeq = myEditedObject->GetGeometryObjects();
+  HYDROData_SequenceOfObjects aSeq;
+  if ( theLandCover )
+    aSeq = myEditedObject->GetLandCovers();
+  else
+    aSeq = myEditedObject->GetGeometryObjects();
   Handle(HYDROData_Entity) anEntity;
 
   if ( myShowZones )
   {
     // Gather zones for displaying
-    HYDROData_SequenceOfObjects aRegions = myEditedObject->GetRegions();
+    HYDROData_SequenceOfObjects aRegions = myEditedObject->GetRegions( theLandCover );
     HYDROData_SequenceOfObjects::Iterator aRegionsIter( aRegions );
     HYDROData_SequenceOfObjects aZones;
     Handle(HYDROData_Region) aRegion;
@@ -1288,7 +1398,7 @@ void HYDROGUI_CalculationOp::onLastViewClosed( SUIT_ViewManager* theViewManager
   closePreview();
 }
 
-void HYDROGUI_CalculationOp::closePreview()
+void HYDROGUI_CalculationOp::closePreview( bool theRemoveViewManager )
 {
   SUIT_DataBrowser* aOb = ((LightApp_Application*)module()->application())->objectBrowser();
   QList<QShortcut*> aShortcuts = aOb->findChildren<QShortcut*>();
@@ -1319,14 +1429,17 @@ void HYDROGUI_CalculationOp::closePreview()
       }
     }
 
-    disconnect( myPreviewViewManager, SIGNAL( lastViewClosed( SUIT_ViewManager* ) ),
-                this, SLOT( onLastViewClosed( SUIT_ViewManager* ) ) );
+    if ( theRemoveViewManager )
+    {
+      disconnect( myPreviewViewManager, SIGNAL( lastViewClosed( SUIT_ViewManager* ) ),
+                  this, SLOT( onLastViewClosed( SUIT_ViewManager* ) ) );
 
-    module()->getApp()->removeViewManager( myPreviewViewManager ); // myPreviewViewManager is deleted here
-    myPreviewViewManager = NULL;
+      module()->getApp()->removeViewManager( myPreviewViewManager ); // myPreviewViewManager is deleted here
+      myPreviewViewManager = NULL;
+    }
   }
 
-  if( myActiveViewManager )
+  if( myActiveViewManager && theRemoveViewManager )
   {
     HYDROGUI_Tool::SetActiveViewManager( module(), myActiveViewManager );
     myActiveViewManager = NULL;
@@ -1469,8 +1582,7 @@ void HYDROGUI_CalculationOp::onAddLandCovers()
   if ( !anAddedList.isEmpty() )
   {
     aPanel->includeLandCovers( anAddedList, false );
-    // TODO: create preview of included land covers
-    //createPreview();
+    createPreview( true );
   }
 }