Bug NPAL12872 - EDF189 GEOM, SMESH : Visualization of groups with many colors

[modules/smesh.git] / src / SMESH / SMESH_Group.hxx

diff --git a/src/SMESH/SMESH_Group.hxx b/src/SMESH/SMESH_Group.hxx
index 1b13bc57859985b2efbf44f4820e7f935b151b3d..64df569def5ab015e9d8c001ab5534579c56efc5 100644 (file)
--- a/src/SMESH/SMESH_Group.hxx
+++ b/src/SMESH/SMESH_Group.hxx
@@ -34,6 +34,9 @@
 #include <string>
 #include <TopoDS_Shape.hxx>
 
+#include "SALOMEconfig.h"
+#include CORBA_SERVER_HEADER(SALOMEDS)
+#include CORBA_SERVER_HEADER(SALOMEDS_Attributes)
 
 class SMESHDS_GroupBase;
 class SMESH_Mesh;
@@ -55,8 +58,8 @@ class SMESH_EXPORT  SMESH_Group
 
   SMESHDS_GroupBase * GetGroupDS () { return myGroupDS; }
 
-  void SetColorNumber (int theColorNumber) { myColorNumber = theColorNumber; }
-  int GetColorNumber() const { return myColorNumber; }
+  void SetColor (const SALOMEDS::Color& theColor) { myColor = theColor; }
+  SALOMEDS::Color GetColor() const { return myColor; }
 
  private:
   SMESH_Group (const SMESH_Group& theOther);
@@ -67,6 +70,7 @@ class SMESH_EXPORT  SMESH_Group
   SMESHDS_GroupBase * myGroupDS;
   std::string         myName;
   int                 myColorNumber;
+  SALOMEDS::Color     myColor;
 };
 
 #endif