Salome HOME
refs #514: add 'Cursor for specific operations' section into preferences of HYDRO...
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_CalculationOp.cxx
index 79a9611092b81d4b3a1fe2d5c9ebce17470ffc74..015a6a8714ddd7a698c673ec1879eb6281612ab0 100644 (file)
@@ -1,4 +1,4 @@
-// Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
+// Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
 //
 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
@@ -6,7 +6,7 @@
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 // License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
 //
 // This library is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -33,6 +33,7 @@
 #include <HYDROData_PolylineXY.h>
 #include <HYDROData_ShapesGroup.h>
 #include <HYDROData_Iterator.h>
+#include <HYDROData_ImmersibleZone.h>
 #include <HYDROData_Object.h>
 #include <HYDROData_Tool.h>
 
@@ -155,7 +156,8 @@ void HYDROGUI_CalculationOp::startOperation()
       Handle(HYDROData_Object) anObject1, anObject2;
       HYDROData_PriorityType aPriority;
       HYDROData_Zone::MergeAltitudesType aMergeType;
-      for ( int anIndex = 0; anIndex < myEditedObject->GetRulesCount(); anIndex++ ) {
+      for ( int anIndex = 0; ; anIndex++ )
+      {
         if ( myEditedObject->GetRule( anIndex, anObject1, aPriority, anObject2, aMergeType ) ) {
           HYDROData_CustomRule aRule;
           aRule.Object1 = anObject1;
@@ -165,6 +167,8 @@ void HYDROGUI_CalculationOp::startOperation()
 
           aRules << aRule;
         }
+        else
+          break;
       }
       aPanel->setRules( aRules );
     }
@@ -543,6 +547,35 @@ bool HYDROGUI_CalculationOp::confirmModeChange() const
   return isConfirmed;
 }
 
+bool HYDROGUI_CalculationOp::confirmContinueWithWarning( const HYDROData_Warning& theWarning ) const
+{
+  HYDROData_WarningType aType = theWarning.Type;
+  if ( aType == WARN_OK ) {
+    return true;
+  }
+
+  QString aTitle;
+  QString aMsg;
+  switch ( aType )
+  {
+    case WARN_EMPTY_REGIONS:
+      aTitle = tr( "EMPTY_REGIONS" );
+      aMsg = tr( "CONFIRM_CONTINUE_WITH_OBJECTS_NOT_INCLUDED_TO_REGION" ).arg( theWarning.Data );
+      break;
+    default:
+      aTitle = tr( "WARNING" );
+      aMsg = theWarning.Data;
+  }
+
+
+  int anAnswer = SUIT_MessageBox::warning( module()->getApp()->desktop(),
+                                           aTitle, aMsg,
+                                           QMessageBox::Yes | QMessageBox::No,
+                                           QMessageBox::No );
+
+  return ( anAnswer == QMessageBox::Yes );
+}
+
 bool HYDROGUI_CalculationOp::processApply( int&     theUpdateFlags,
                                            QString& theErrorMsg,
                                            QStringList& theBrowseObjectsEntries )
@@ -558,6 +591,11 @@ bool HYDROGUI_CalculationOp::processApply( int&     theUpdateFlags,
     theBrowseObjectsEntries.append( anEntry );
   }
 
+  // For manual mode priority rules are redundant
+  if ( aPanel->getMode() == HYDROData_CalculationCase::MANUAL ) {
+    myEditedObject->ClearRules( false );
+  }
   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init;
 
   return true;
@@ -565,6 +603,20 @@ bool HYDROGUI_CalculationOp::processApply( int&     theUpdateFlags,
 
 void HYDROGUI_CalculationOp::onApply()
 {
+  // Check warnings
+  HYDROData_Warning aWarning = myEditedObject->GetLastWarning();
+  if ( aWarning.Type != WARN_OK ) {
+    if ( !confirmContinueWithWarning( aWarning ) ) {
+      // Go back to the first page
+      HYDROGUI_CalculationDlg* aPanel = 
+        ::qobject_cast<HYDROGUI_CalculationDlg*>( inputPanel() );
+      if ( aPanel ) {
+        aPanel->onFirstPage();
+      }
+      return;
+    }
+  }
+
   QApplication::setOverrideCursor( Qt::WaitCursor );
 
   int anUpdateFlags = 0;
@@ -625,13 +677,14 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
   
     QString aNewCaseName = aPanel->getObjectName();
     QString anOldCaseName = myEditedObject->GetName();
+    bool isNameChanged = anOldCaseName != aNewCaseName;
   
-    bool anIsToUpdateOb = myIsEdit && anOldCaseName != aNewCaseName;
-  
+    bool anIsToUpdateOb = isNameChanged;
+
     // At first we must to update the case name because of 
     // automatic names generation for regions and zones
     myEditedObject->SetName( aNewCaseName );
-
+    
     // Set parameters for automatic mode
     int aMode = aPanel->getMode();
     if ( aMode == HYDROData_CalculationCase::AUTOMATIC ) {
@@ -647,32 +700,48 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
         myEditedObject->AddGeometryObject( anObject );
       }
 
+      // Clear priority rules
+      //@ASL if ( myEditedObject->GetRulesCount() > 0 ) {
+        myEditedObject->ClearRules( true );
+      //@ASL }
       // Set priority rules
-      myEditedObject->ClearRules();
       foreach ( const HYDROData_CustomRule& aRule, aPanel->getRules() ) {
         myEditedObject->AddRule( aRule.Object1, aRule.Priority,
                                  aRule.Object2, aRule.MergeType );
       }
     }
-    aPanel->setMoveZonesEnabled( aMode == HYDROData_CalculationCase::MANUAL );
+    aPanel->setEditZonesEnabled( aMode == HYDROData_CalculationCase::MANUAL );
          
     if ( myEditedObject->IsMustBeUpdated() )
     {
       myShowZones = true;
       myEditedObject->Update();
+      
+      AssignDefaultZonesColors();
 
       //aPanel->setEditedObject( myEditedObject );
       aPanel->refreshZonesBrowser();
-
+    
+      closePreview();
       createPreview();
+
+      anIsToUpdateOb = true;
     }
     else
     {
       setZonesVisible( true );
+
+      if ( isNameChanged ) {
+        module()->getDataModel()->updateObjectTree( myEditedObject );
+      }
     }
 
-    if ( anIsToUpdateOb )
-      module()->getApp()->updateObjectBrowser( false );
+    if ( anIsToUpdateOb ) {
+      SUIT_DataBrowser* anObjBrowser = ((LightApp_Application*)module()->application())->objectBrowser();
+      if ( anObjBrowser ) {
+        anObjBrowser->updateTree( module()->getDataModel()->getDataObject( myEditedObject ), false );
+      }
+    }
 
     QApplication::restoreOverrideCursor();
   }
@@ -717,6 +786,77 @@ void HYDROGUI_CalculationOp::setZonesVisible( bool theIsVisible )
   }
 }
 
+void HYDROGUI_CalculationOp::AssignDefaultZonesColors()
+{
+  HYDROData_SequenceOfObjects aRegions = myEditedObject->GetRegions();
+  HYDROData_SequenceOfObjects::Iterator aRegionsIter( aRegions );
+  HYDROData_SequenceOfObjects aZones;
+  Handle(HYDROData_Region) aRegion;
+  if ( myPreviewViewManager ) 
+  {
+    if ( OCCViewer_Viewer* aViewer = myPreviewViewManager->getOCCViewer() )
+    {
+      Handle(AIS_InteractiveContext) aCtx = aViewer->getAISContext();
+      if ( !aCtx.IsNull() )
+      {
+        int aCounter = 0;        
+        for ( ; aRegionsIter.More(); aRegionsIter.Next() )
+        {
+          aRegion = Handle(HYDROData_Region)::DownCast( aRegionsIter.Value() );
+          if ( !aRegion.IsNull() )
+          {
+            aZones = aRegion->GetZones();
+            HYDROData_SequenceOfObjects::Iterator aZonesIter( aZones );
+            for ( ; aZonesIter.More(); aZonesIter.Next() )
+            {
+              // Zone
+              Handle(HYDROData_Zone) aZone = Handle(HYDROData_Zone)::DownCast( aZonesIter.Value() );
+              if ( !aZone.IsNull() )
+              {
+                QColor aFillingColor = GenerateDefaultZonesColor(++aCounter);
+                while (aFillingColor == Qt::red)
+                  aFillingColor = GenerateDefaultZonesColor(++aCounter);
+                
+                aZone->SetColor(aFillingColor);
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+}
+
+QColor HYDROGUI_CalculationOp::GenerateDefaultZonesColor( int theIndex,
+                                                          float theSaturation/* = 0.5*/,
+                                                          float theValue/* = 0.95*/ ) const
+{
+  float aGoldenRatioConjugate = (float)(360./582.);
+  float aHue = (float)rand();
+  aHue += aGoldenRatioConjugate*theIndex;
+  aHue -= floor(aHue);
+
+  float aR = 0., aG = 0., aB = 0.;
+  int aHueInt = (int)(aHue*6.);
+  float aF = aHue*6. - aHueInt;
+  float aP = theValue * (1. - theSaturation);
+  float aQ = theValue * (1. - aF*theSaturation);
+  float aT = theValue * (1. - (1. - aF) * theSaturation);
+  switch (aHueInt)
+  {
+  case 0: { aR = theValue; aG = aT; aB = aP; break; }
+  case 1: { aR = aQ; aG = theValue; aB = aP; break; }
+  case 2: { aR = aP; aG = theValue; aB = aT; break; }
+  case 3: { aR = aP; aG = aQ; aB = theValue; break; }
+  case 4: { aR = aT; aG = aP; aB = theValue; break; }
+  case 5: { aR = theValue; aG = aP; aB = aQ; break; }
+  default: break;
+  }
+
+  QColor aColor = QColor( (int)(aR*256.), (int)(aG*256.), (int)(aB*256.) );
+  return ( aColor.isValid() ? aColor : HYDROData_ImmersibleZone::DefaultFillingColor() );
+}
+
 void HYDROGUI_CalculationOp::createPreview()
 {
   LightApp_Application* anApp = module()->getApp();