Fix for the bug IPAL22828 TC6.4.0: Displayed entities are wrong

[modules/smesh.git] / src / OBJECT / SMESH_ObjectDef.h

diff --git a/src/OBJECT/SMESH_ObjectDef.h b/src/OBJECT/SMESH_ObjectDef.h
index 5bcbb3aedd3af700e3f8d6ee8776500fce56efde..edb905d54fd78482590f91a2d5915b66280e677b 100644 (file)
--- a/src/OBJECT/SMESH_ObjectDef.h
+++ b/src/OBJECT/SMESH_ObjectDef.h
@@ -32,6 +32,7 @@
 #include "SMESH_Controls.hxx"
 #include "SMESH_Object.h"
 #include "SMESH_Client.hxx"
+#include "SMESH_Actor.h"
 
 // IDL Headers
 #include <SALOMEconfig.h>
@@ -44,7 +45,6 @@
 class vtkPoints;
 class SALOME_ExtractUnstructuredGrid;
 
-class SMESH_Actor;
 class SMDS_MeshNode;
 class SMDS_MeshElement;
 
@@ -85,13 +85,17 @@ public:
   virtual vtkIdType         GetElemObjId( int theVTKID );
   virtual vtkIdType         GetElemVTKId( int theObjID );
   
+  virtual void              ClearEntitiesFlags();
+  virtual bool              GetEntitiesFlag();
+  virtual unsigned int      GetEntitiesState();
+  
 protected:
 
   void                      createPoints( vtkPoints* );
   void                      buildPrs(bool buildGrid = false);
   void                      buildNodePrs();
   void                      buildElemPrs();
-  
+  void                      updateEntitiesFlags();
 //private:
 
   TMapOfIds                 mySMDS2VTKNodes;
@@ -100,7 +104,11 @@ protected:
   TMapOfIds                 myVTK2SMDSElems;
   bool                      myLocalGrid;
 
+  bool                      myEntitiesFlag;
+  unsigned int              myEntitiesState;
+
   vtkUnstructuredGrid*      myGrid;
+  std::map<SMDSAbs_ElementType,int> myEntitiesCache;
 };