]> SALOME platform Git repositories - modules/shaper.git/commitdiff
Salome HOME
Implementation of context menu AutoColor
authorJérôme <jerome.lucas@cesgenslab.fr>
Wed, 9 Dec 2020 12:00:24 +0000 (13:00 +0100)
committerNicolas Rechatin <nicolas.rechatin@cea.fr>
Mon, 30 Aug 2021 13:01:50 +0000 (15:01 +0200)
src/ModelAPI/ModelAPI_Feature.cpp
src/ModelAPI/ModelAPI_Tools.cpp
src/ModelAPI/ModelAPI_Tools.h
src/PartSet/PartSet_Module.cpp
src/XGUI/XGUI_ColorDialog.cpp
src/XGUI/XGUI_ColorDialog.h
src/XGUI/XGUI_Workshop.cpp
src/XGUI/XGUI_Workshop.h

index 14397b0fea7efed0e0c42ee1af67e89196f375c8..e71e99f293d2c021b1c860de1e47141671059459 100644 (file)
 #include <ModelAPI_Data.h>
 #include <ModelAPI_Document.h>
 #include <ModelAPI_Session.h>
+#include <ModelAPI_Tools.h>
+#include <ModelAPI_ResultGroup.h>
 #include <Events_Loop.h>
 #include <Config_Translator.h>
+#include <Config_PropManager.h>
 
 void ModelAPI_Feature::setError(const std::string& theError,
                                 bool isSend,
@@ -70,6 +73,14 @@ void ModelAPI_Feature::setResult(const std::shared_ptr<ModelAPI_Result>& theResu
   // in any case result becomes enabled
   if (!isDisabled()) // disabled feature may be executed when it is added as not enabled (#2078)
     theResult->setDisabled(theResult, false);
+
+  bool anIsAutoColor = Config_PropManager::boolean("Visualization", "result_group_Auto_color");
+
+  if (anIsAutoColor && theResult->groupName() == ModelAPI_ResultGroup::group()) {
+    std::vector<int> aColor;
+    ModelAPI_Tools::findRandomColor(aColor);
+    ModelAPI_Tools::setColor(theResult,  aColor);
+  }
 }
 
 void ModelAPI_Feature::setResult(const std::shared_ptr<ModelAPI_Result>& theResult,
index e15b280afe3e0a8303c5a71dbbe701388de8e254..a669033db22d44ff7dbfb6b5eb03310d0452ccab 100644 (file)
@@ -1116,6 +1116,103 @@ std::list<FeaturePtr> referencedFeatures(
   return aResList;
 }
 
+// contains global cash for integer index of the color -> RGB of this color
+static std::map<int, std::vector<int> > myColorMap;
+
+void appendValues(std::vector<int>& theRGB, const int theRed, const int theGreen, const int theBlue)
+{
+  theRGB.push_back(theRed);
+  theRGB.push_back(theGreen);
+  theRGB.push_back(theBlue);
+}
+
+bool containsValues(std::map<int, std::vector<int> >& theColorMap, std::vector<int>& theValues)
+{
+  std::map<int, std::vector<int> >::const_iterator anIt = theColorMap.begin(),
+                                                   aLast = theColorMap.end();
+  bool isFound = false;
+  for (; anIt != aLast && !isFound; anIt++) {
+    std::vector<int> aValues = anIt->second;
+    isFound = aValues[0] == theValues[0] &&
+              aValues[1] == theValues[1] &&
+              aValues[2] == theValues[2];
+  }
+  return isFound;
+}
+
+std::vector<int> HSVtoRGB(int theH, int theS, int theV)
+{
+  std::vector<int> aRGB;
+  if (theH < 0 || theH > 360 ||
+      theS < 0 || theS > 100 ||
+      theV < 0 || theV > 100)
+    return aRGB;
+
+  int aHi = (int)theH/60;
+
+  double aV = theV;
+  double aVmin = (100 - theS)*theV/100;
+
+  double anA = (theV - aVmin)* (theH % 60) / 60;
+
+  double aVinc = aVmin + anA;
+  double aVdec = theV - anA;
+
+  double aPercentToValue = 255./100;
+  int aV_int    = (int)(aV*aPercentToValue);
+  int aVinc_int = (int)(aVinc*aPercentToValue);
+  int aVmin_int = (int)(aVmin*aPercentToValue);
+  int aVdec_int = (int)(aVdec*aPercentToValue);
+
+  switch(aHi) {
+    case 0: appendValues(aRGB, aV_int,    aVinc_int, aVmin_int); break;
+    case 1: appendValues(aRGB, aVdec_int, aV_int,    aVmin_int); break;
+    case 2: appendValues(aRGB, aVmin_int, aV_int,    aVinc_int); break;
+    case 3: appendValues(aRGB, aVmin_int, aVdec_int, aV_int); break;
+    case 4: appendValues(aRGB, aVinc_int, aVmin_int, aV_int); break;
+    case 5: appendValues(aRGB, aV_int,    aVmin_int, aVdec_int); break;
+    default: break;
+  }
+  return aRGB;
+}
+
+
+void fillColorMap()
+{
+  if (!myColorMap.empty())
+    return;
+
+  int i = 0;
+  for (int s = 100; s > 0; s = s - 50)
+  {
+    for (int v = 100; v >= 40; v = v - 20)
+    {
+      for (int h = 0; h < 359 ; h = h + 60)
+      {
+        std::vector<int> aColor = HSVtoRGB(h, s, v);
+        if (containsValues(myColorMap, aColor))
+          continue;
+        myColorMap[i] = aColor;
+        i++;
+      }
+    }
+  }
+}
+
+void findRandomColor(std::vector<int>& theValues)
+{
+  theValues.clear();
+  if (myColorMap.empty()) {
+    fillColorMap();
+  }
+
+  size_t aSize = myColorMap.size();
+  int anIndex = rand() % aSize;
+  if (myColorMap.find(anIndex) != myColorMap.end()) {
+    theValues = myColorMap.at(anIndex);
+  }
+}
+
 // LCOV_EXCL_STOP
 
 } // namespace ModelAPI_Tools
index e740cd1c95c59801de766a54963eb8fc1f01fcf5..2e2bbaeb4a930aab2686e22ccf255cb370e4f370 100644 (file)
@@ -302,6 +302,15 @@ MODELAPI_EXPORT void copyVisualizationAttrs(std::shared_ptr<ModelAPI_Result> the
 MODELAPI_EXPORT std::list<std::shared_ptr<ModelAPI_Feature> > referencedFeatures(
   std::shared_ptr<ModelAPI_Result> theTarget, const std::string& theFeatureKind,
   const bool theSortResults);
+
+
+/*! Returns a container with the current color value.
+*   These are tree int values for RGB definition.
+*   It returns the next random color.
+* \param theValues vector of values
+*/
+MODELAPI_EXPORT void findRandomColor(std::vector<int>& theValues);
+
 }
 
 #endif
index 78130efcafbc6d3a45d2cfae54db9276b6fa7d15..0b5915d4f3362643073d57e59b8710e31f0dd332 100644 (file)
@@ -202,9 +202,13 @@ PartSet_Module::PartSet_Module(ModuleBase_IWorkshop* theWshop)
   Config_PropManager::registerProp("Visualization", "result_group_color", "Group color",
     Config_Prop::Color, ModelAPI_ResultGroup::DEFAULT_COLOR());
 
-  Config_PropManager::registerProp("Visualization",
-    ModelAPI_ResultConstruction::RESULT_COLOR_NAME(),
-    "Construction color", Config_Prop::Color, ModelAPI_ResultConstruction::DEFAULT_COLOR());
+  Config_PropManager::registerProp("Visualization", "result_group_Auto_color", "Auto color",
+    Config_Prop::Boolean, "false");
+
+  Config_PropManager::registerProp("Visualization", "result_construction_color",
+    "Construction color",
+    Config_Prop::Color,
+    ModelAPI_ResultConstruction::DEFAULT_COLOR());
 
   Config_PropManager::registerProp("Visualization", "result_part_color", "Part color",
     Config_Prop::Color, ModelAPI_ResultPart::DEFAULT_COLOR());
@@ -1700,8 +1704,7 @@ void PartSet_Module::processEvent(const std::shared_ptr<Events_Message>& theMess
     CompositeFeaturePtr aSketch = mySketchMgr->activeSketch();
     if (aSketch.get()) {
       ModuleBase_Operation* anOperation = myWorkshop->currentOperation();
-      if (PartSet_SketcherMgr::isSketchOperation(anOperation) &&
-        mySketchMgr->previewSketchPlane()->isDisplayed())
+      if (PartSet_SketcherMgr::isSketchOperation(anOperation))
         mySketchMgr->previewSketchPlane()->createSketchPlane(aSketch, myWorkshop);
     }
   }
index 3cbcca53977dcc52b4d3b00463b0ab861188adaf..a37df57fbfd3d83498a18d3bd739365657219974 100644 (file)
@@ -84,115 +84,13 @@ std::vector<int> XGUI_ColorDialog::getColor() const
 
   return aValues;
 }
-// contains global cash for integer index of the color -> RGB of this color
-static std::map<int, std::vector<int> > myColorMap;
-
-void appendValues(std::vector<int>& theRGB, const int theRed, const int theGreen, const int theBlue)
-{
-  theRGB.push_back(theRed);
-  theRGB.push_back(theGreen);
-  theRGB.push_back(theBlue);
-}
-
-bool containsValues(std::map<int, std::vector<int> >& theColorMap, std::vector<int>& theValues)
-{
-  std::map<int, std::vector<int> >::const_iterator anIt = theColorMap.begin(),
-                                                   aLast = theColorMap.end();
-  bool isFound = false;
-  for (; anIt != aLast && !isFound; anIt++) {
-    std::vector<int> aValues = anIt->second;
-    isFound = aValues[0] == theValues[0] &&
-              aValues[1] == theValues[1] &&
-              aValues[2] == theValues[2];
-  }
-  return isFound;
-}
-
-std::vector<int> HSVtoRGB(int theH, int theS, int theV)
-{
-  std::vector<int> aRGB;
-  if (theH < 0 || theH > 360 ||
-      theS < 0 || theS > 100 ||
-      theV < 0 || theV > 100)
-    return aRGB;
-
-  int aHi = (int)theH/60;
-
-  double aV = theV;
-  double aVmin = (100 - theS)*theV/100;
-
-  double anA = (theV - aVmin)* (theH % 60) / 60;
-
-  double aVinc = aVmin + anA;
-  double aVdec = theV - anA;
-
-  double aPercentToValue = 255./100;
-  int aV_int    = (int)(aV*aPercentToValue);
-  int aVinc_int = (int)(aVinc*aPercentToValue);
-  int aVmin_int = (int)(aVmin*aPercentToValue);
-  int aVdec_int = (int)(aVdec*aPercentToValue);
-
-  switch(aHi) {
-    case 0: appendValues(aRGB, aV_int,    aVinc_int, aVmin_int); break;
-    case 1: appendValues(aRGB, aVdec_int, aV_int,    aVmin_int); break;
-    case 2: appendValues(aRGB, aVmin_int, aV_int,    aVinc_int); break;
-    case 3: appendValues(aRGB, aVmin_int, aVdec_int, aV_int); break;
-    case 4: appendValues(aRGB, aVinc_int, aVmin_int, aV_int); break;
-    case 5: appendValues(aRGB, aV_int,    aVmin_int, aVdec_int); break;
-    default: break;
-  }
-  return aRGB;
-}
-
-
-void fillColorMap()
-{
-  if (!myColorMap.empty())
-    return;
-
-  int i = 0;
-  for (int s = 100; s > 0; s = s - 50)
-  {
-    for (int v = 100; v >= 40; v = v - 20)
-    {
-      for (int h = 0; h < 359 ; h = h + 60)
-      {
-        std::vector<int> aColor = HSVtoRGB(h, s, v);
-        if (containsValues(myColorMap, aColor))
-          continue;
-        myColorMap[i] = aColor;
-        i++;
-      }
-    }
-  }
-}
-
-void findRandomColor(std::vector<int>& theValues)
-{
-  theValues.clear();
-  if (myColorMap.empty()) {
-    fillColorMap();
-  }
-
-  size_t aSize = myColorMap.size();
-  int anIndex = rand() % aSize;
-  if (myColorMap.find(anIndex) != myColorMap.end()) {
-    theValues = myColorMap.at(anIndex);
-  }
-}
 
 std::vector<int> XGUI_ColorDialog::getRandomColor() const
 {
   std::vector<int> aValues;
   if (isRandomColor()) {
-    findRandomColor(aValues);
+    ModelAPI_Tools::findRandomColor(aValues);
   }
   return aValues;
 }
 
-std::vector<int> XGUI_ColorDialog::randomColor()
-{
-  std::vector<int> aValues;
-  findRandomColor(aValues);
-  return aValues;
-}
index 172e1840f406c4bbfde89afa5cdd4dbded75018b..fc4ea08991a0f04a6a323e62bfd4779fba5baa6b 100644 (file)
@@ -64,12 +64,6 @@ public:
   /// \return a vector of values
   std::vector<int> getRandomColor() const;
 
-  /// Returns a container with the current color value.
-  /// These are tree int values for RGB definition.
-  /// It returns the next random color.
-  /// \return a vector of values
-  static std::vector<int> randomColor();
-
 private:
   QButtonGroup* myButtonGroup; /// a group, contained random and certain color radio button choice
   QtxColorButton* myColorButton; /// a control to select a color
index fb85f74322ac45c3c8bb31be91def208852aa571..a36af5684a4585d82e9f558733d2893da60674c1 100644 (file)
@@ -2509,6 +2509,7 @@ void XGUI_Workshop::changeColor(const QObjectPtrList& theObjects)
   // 3. abort the previous operation and start a new one
   SessionPtr aMgr = ModelAPI_Session::get();
   QString aDescription = contextMenuMgr()->action("COLOR_CMD")->text();
+
   aMgr->startOperation(aDescription.toStdString());
 
   // 4. set the value to all results
@@ -2539,32 +2540,49 @@ void XGUI_Workshop::changeAutoColor(const QObjectPtrList& theObjects)
   if (!abortAllOperations())
   return;
 
-  std::vector<int> aColor = XGUI_ColorDialog::randomColor();
+  std::vector<int> aColor;
+  ModelAPI_Tools::findRandomColor(aColor);
 
   // abort the previous operation and start a new one
   SessionPtr aMgr = ModelAPI_Session::get();
   QString aDescription = contextMenuMgr()->action("AUTOCOLOR_CMD")->text();
   aMgr->startOperation(aDescription.toStdString());
 
-  Config_Prop* aProp = Config_PropManager::findProp("Visualization", "result_group_color");
-
-  if( aDescription == tr("Disable auto color"))
-  {
-     contextMenuMgr()->action("AUTOCOLOR_CMD")->setText(tr("Auto color") );
-
-    aProp->setValue(ModelAPI_ResultGroup::DEFAULT_COLOR());
-
-  }else{
-    std::stringstream streamColor;
-    streamColor<< aColor[0] <<","<< aColor[1] <<"," <<aColor[2];
-
-    aProp->setValue(streamColor.str());
+  Config_Prop* aProp = Config_PropManager::findProp("Visualization", "result_group_Auto_color");
+  bool anIsAutoColor = Config_PropManager::boolean("Visualization", "result_group_Auto_color");
 
+  if (anIsAutoColor) {
+     contextMenuMgr()->action("AUTOCOLOR_CMD")->setText(tr("Auto color"));
+     aProp->setValue("false");
+  } else {
+    // set the value to all results
+    foreach (ObjectPtr anObj, theObjects) {
+      DocumentPtr aDocument = anObj->document();
+      std::list<FeaturePtr> anAllFeatures = allFeatures(aDocument);
+      // find the object iterator
+      std::list<FeaturePtr>::iterator aObjectIt = anAllFeatures.begin();
+      for (; aObjectIt !=  anAllFeatures.end(); ++ aObjectIt) {
+        FeaturePtr aFeature = *aObjectIt;
+        if (aFeature.get()) {
+          std::list<ResultPtr> aResults;
+          ModelAPI_Tools::allResults(aFeature, aResults);
+          std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator aIt;
+          for (aIt = aResults.cbegin(); aIt != aResults.cend(); aIt++) {
+            ResultPtr aGroupResult = *aIt;
+            if (aGroupResult.get() && aGroupResult->groupName() == ModelAPI_ResultGroup::group()) {
+              ModelAPI_Tools::findRandomColor(aColor);
+              ModelAPI_Tools::setColor(aGroupResult, aColor);
+            }
+          }
+        }
+      }
+    }
     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
     aMgr->finishOperation();
     updateCommandStatus();
     myViewerProxy->update();
-    contextMenuMgr()->action("AUTOCOLOR_CMD")->setText(tr("Disable auto color") );
+    contextMenuMgr()->action("AUTOCOLOR_CMD")->setText(tr("Disable auto color"));
+    aProp->setValue("true");
   }
 }
 
index 0190b0c581e91cf2f47b8ea8afff3f73f887294b..3050590c0f7209caabade14539286425008036c7 100644 (file)
@@ -445,7 +445,7 @@ signals:
   /// Sets the granted operations for the parameter operation. Firstly, it finds the nested features
   /// and set them into the operation. Secondly, it asks the module about ids of granted operations.
   /// \param theOperation an operation
-   void setGrantedFeatures(ModuleBase_Operation* theOperation);
+  void setGrantedFeatures(ModuleBase_Operation* theOperation);
 
 private:
   /// Display results from document