refs #430: incorrect coordinates in dump polyline

[modules/hydro.git] / src / HYDROGUI / HYDROGUI_SetColorOp.h

diff --git a/src/HYDROGUI/HYDROGUI_SetColorOp.h b/src/HYDROGUI/HYDROGUI_SetColorOp.h
index 10d02b9911320e673c2b883cdacb0627c763b696..08729bd39551bc46c1dbdee83083bf08adedfbc1 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_SetColorOp.h
+++ b/src/HYDROGUI/HYDROGUI_SetColorOp.h
@@ -25,7 +25,7 @@
 
 #include "HYDROGUI_Operation.h"
 
-#include <HYDROData_Object.h>
+#include <HYDROData_Entity.h>
 
 class HYDROGUI_ColorDlg;
 
@@ -37,15 +37,19 @@ public:
   HYDROGUI_SetColorOp( HYDROGUI_Module* theModule );
   virtual ~HYDROGUI_SetColorOp();
 
+
+  static bool CanObjectBeColored( const Handle(HYDROData_Entity)& theObject );
+
 protected:
   virtual void startOperation();
 
-  virtual bool processApply( int& theUpdateFlags, QString& theErrorMsg );
+  virtual bool processApply( int& theUpdateFlags, QString& theErrorMsg,
+                             QStringList& theBrowseObjectsEntries );
 
 private:
   HYDROGUI_ColorDlg* myColorDlg;
 
-  Handle(HYDROData_Object) myEditedObject;
+  Handle(HYDROData_Entity) myEditedObject;
 };
 
 #endif
\ No newline at end of file