Fix for the bug #45: check and warning when the same image is used in 2 arguments.

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

diff --git a/src/HYDROGUI/HYDROGUI_DataObject.h b/src/HYDROGUI/HYDROGUI_DataObject.h
index 69693349d7fd1e519cdd089f8824c86d17890966..ea9fb38c00edada61d75d32e37f9c25f085d0888 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DataObject.h
+++ b/src/HYDROGUI/HYDROGUI_DataObject.h
@@ -23,13 +23,14 @@
 #ifndef HYDROGUI_DATAOBJECT_H
 #define HYDROGUI_DATAOBJECT_H
 
-#include <HYDROData_Object.h>
+#include <HYDROData_Entity.h>
 
 #include <LightApp_DataObject.h>
 
 #include <QString>
 #include <QMap>
 #include <QPixmap>
+#include <QColor>
 
 /**
  * \class HYDROGUI_DataObject
@@ -43,6 +44,12 @@
 class HYDROGUI_DataObject : public LightApp_DataObject
 {
 public:
+  //! Column id
+  enum { 
+    RefObjectId = RefEntryId + 1,    //!< Ref.Object column
+    BathymetryId                     //!< Bathymetry column
+  };
+
   /**
    * Constructor.
    * \param theParent parent data object
@@ -50,7 +57,7 @@ public:
    * \param theParentEntry entry of the parent data object (for reference objects)
    */
   HYDROGUI_DataObject( SUIT_DataObject* theParent,
-                       Handle(HYDROData_Object) theData,
+                       Handle(HYDROData_Entity) theData,
                        const QString& theParentEntry );
     
   /**
@@ -73,15 +80,25 @@ public:
    */
   virtual QFont font( const int = SUIT_DataObject::NameId ) const;
 
+  /**
+   * Returns the object color.
+   */
+  virtual QColor color( const ColorRole, const int = NameId ) const;
+
+  /**
+   * Returns the object icon.
+   */
+  virtual QPixmap icon( const int = NameId ) const;
+
   /**
    * Returns the model data object.
    */
-  const Handle(HYDROData_Object)& modelObject() const { return myData; }
+  const Handle(HYDROData_Entity)& modelObject() const { return myData; }
 
   /**
    * Redefines the object.
    */
-  void setObject( Handle(HYDROData_Object) theObject ) { myData = theObject; }
+  void setObject( const Handle(HYDROData_Entity)& theObject ) { myData = theObject; }
 
   /**
    * Returns the entry prefix for all HYDRO data objects.
@@ -91,12 +108,25 @@ public:
   /**
    * Returns the full entry for the specified data object.
    */
-  static QString dataObjectEntry( const Handle(HYDROData_Object)& theObject,
+  static QString dataObjectEntry( const Handle(HYDROData_Entity)& theObject,
                                   const bool theWithPrefix = true );
+  /**
+   * Sets the validity flag: if object is valid or not.
+   * \param theIsValid is true for valid objects, false for invalid
+   */
+  void setIsValid( const bool theIsValid );
+
+  /**
+   * Returns the validity flag: is object valid or not
+   * \returns false if object is not valid
+   */
+  bool isValid() const;
 
 protected:
-  Handle(HYDROData_Object) myData; ///< object from data model
+  Handle(HYDROData_Entity) myData; ///< object from data model
   QString myParentEntry;
+  bool myIsValid; ///< indicates if the object is valid
+  QPixmap myIcon;
 };
 
 /**
@@ -130,9 +160,35 @@ public:
    */
   virtual QString name() const;
 
+  /**
+   * Returns the object icon.
+   */
+  virtual QPixmap icon( const int = NameId ) const;
+
 private:
   QString myName; ///< name in the OB
   QString myParentEntry;
+  QPixmap myIcon;
+};
+
+/**
+ * \brief Module data object, used for dropping items in the object browser.
+ *
+ * It inherits NamedObject with only difference - it accepts dropping.
+ */
+class HYDROGUI_DropTargetObject : public HYDROGUI_NamedObject
+{
+public:
+  /**
+   * Constructor.
+   * \param theParent parent data object
+   * \param theName displayed name
+   */
+  HYDROGUI_DropTargetObject( SUIT_DataObject* theParent,
+                             const QString& theName,
+                             const QString& theParentEntry  );
+    
+  bool isDropAccepted() const { return true; }
 };
 
 #endif