- clean programming code

[modules/geom.git] / src / DependencyTree / DependencyTree_Object.h

diff --git a/src/DependencyTree/DependencyTree_Object.h b/src/DependencyTree/DependencyTree_Object.h
index 64be40adc8123bbed41e9a02924093d345a0e025..fae775edcff3765a2c4e820cc195ec6863ef60c9 100644 (file)
--- a/src/DependencyTree/DependencyTree_Object.h
+++ b/src/DependencyTree/DependencyTree_Object.h
@@ -23,8 +23,8 @@
 #include <GraphicsView_Object.h>
 
 // GEOM includes
-#include <GeometryGUI.h>
-#include <GEOM_BaseObject.hxx>
+#include <SALOMEconfig.h>
+#include CORBA_CLIENT_HEADER(GEOM_Gen)
 
 #include <QPen>
 
@@ -50,9 +50,10 @@ public:
 
   void                        updateName();
 
-  void                        setColor(const QColor& );
-  void                        setSelectColor(const QColor& );
-  void                        setMainObjectColor(const QColor& );
+  void                        setColor( const QColor& );
+  void                        setSelectColor( const QColor& );
+  void                        setMainObjectColor( const QColor& );
+  void                        setUnpublishObjectColor( const QColor& );
 
   void                        setIsMainObject( bool );
 
@@ -63,6 +64,7 @@ private:
   QColor                      myColor;
   QColor                      mySelectColor;
   QColor                      myMainObjectColor;
+  QColor                      myUnpublishObjectColor;
 
   QGraphicsPolygonItem*       myPolygonItem;
   QGraphicsSimpleTextItem*    myTextItem;