refs #514: add 'Cursor for specific operations' section into preferences of HYDRO...

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

diff --git a/src/HYDROGUI/HYDROGUI_CalculationOp.cxx b/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
index 898a3c1e64232536e08b267a9ff9379c20ea8d1f..015a6a8714ddd7a698c673ec1879eb6281612ab0 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
@@ -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>
 
@@ -716,6 +717,8 @@ void HYDROGUI_CalculationOp::onNext( const int theIndex )
       myShowZones = true;
       myEditedObject->Update();
       
+      AssignDefaultZonesColors();
+
       //aPanel->setEditedObject( myEditedObject );
       aPanel->refreshZonesBrowser();
     
@@ -783,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();