Feature #86: The hierarchy in the Object Browser (T 19).

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

diff --git a/src/HYDROGUI/HYDROGUI_DataModel.h b/src/HYDROGUI/HYDROGUI_DataModel.h
index a4216902d1e1ee45dac3a2e123fd92f43bf13e13..060a3db56936e155a39dc47c94da2856ac9c7488 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DataModel.h
+++ b/src/HYDROGUI/HYDROGUI_DataModel.h
@@ -24,9 +24,13 @@
 #define HYDROGUI_DATAMODEL_H
 
 #include <HYDROData_Document.h>
-#include <HYDROData_Object.h>
+#include <HYDROData_Entity.h>
+#include <HYDROData_Zone.h>
+#include <HYDROData_Region.h>
+#include <HYDROData_CalculationCase.h>
 
 #include <QMap>
+#include <QList>
 #include <LightApp_DataModel.h>
 
 #include <SUIT_TreeModel.h>
@@ -34,6 +38,7 @@
 class CAM_DataObject;
 class SUIT_DataObject;
 class HYDROGUI_DataObject;
+class HYDROGUI_Zone;
 
 /**
  * \class HYDROGUI_DataModel
@@ -111,7 +116,7 @@ public:
    * \param the data model object
    * \return the only one object referenced to the given data model object, or null if not found
    */
-  virtual HYDROGUI_DataObject* getDataObject( const Handle(HYDROData_Object)& theModelObject );
+  virtual HYDROGUI_DataObject* getDataObject( const Handle(HYDROData_Entity)& theModelObject );
 
   /**
    * Returns a data object referenced to the given data object.
@@ -139,6 +144,11 @@ public:
    */
   CAM_DataObject* createRootModuleObject( SUIT_DataObject* theParent );
 
+  /**
+   * Create a new region in the given calculation case containing given zones. 
+   */
+  bool createNewRegion( Handle(HYDROData_CalculationCase) theCase, const QList<HYDROGUI_Zone*>& theZonesList );
+
   /**
    * Correct an internal model object according to the current document mode
    */
@@ -147,7 +157,7 @@ public:
   /**
    * Find a data object by the specified entry and kind
    */
-  Handle(HYDROData_Object) objectByEntry( const QString& theEntry,
+  Handle(HYDROData_Entity) objectByEntry( const QString& theEntry,
                                           const ObjectKind theObjectKind = KIND_UNKNOWN );
 
   /**
@@ -210,6 +220,15 @@ public:
    */
   bool paste();
 
+  /**
+   * Creates the GUI data object according to the model object.
+   * \param theParent a created object will be appended as a child of this object
+   * \param theModelObject model object
+   * \param theParentEntry entry of parent object
+   */
+  void buildCaseTree( SUIT_DataObject*                  theParent,
+                      Handle(HYDROData_CalculationCase) theCase );
+
   /**
    * Update the sequence of the objects to be copied
    */
@@ -235,8 +254,9 @@ protected:
    * \param theParentEntry entry of parent object
    */
   LightApp_DataObject* createObject( SUIT_DataObject*         theParent,
-                                     Handle(HYDROData_Object) theModelObject,
-                                     const QString&           theParentEntry = QString() );
+                                     Handle(HYDROData_Entity) theModelObject,
+                                     const QString&           theParentEntry = QString(),
+                                     const bool               theIsBuildTree = true );
 
   /**
    * Creates the GUI data object without corresponding model object: just by name
@@ -248,6 +268,28 @@ protected:
                                      const QString&   theName,
                                      const QString&   theParentEntry = QString() );
 
+  /**
+   * Build object tree if the flag theIsBuildTree is true. 
+   * This is a conditional wrapper for buildObjectTree method.
+   * \param theParent a created object will be appended as a child of this object
+   * \param theObject the GUI object
+   * \param theParentEntry the entry of parent object
+   * \param theIsBuildTree if true then build the subtree of the GUI object
+   */
+  LightApp_DataObject* buildObject( SUIT_DataObject*     theParent,
+                                    HYDROGUI_DataObject* theObject,
+                                    const QString&       theParentEntry,
+                                    const bool           theIsBuildTree );
+
+  LightApp_DataObject* createZone( SUIT_DataObject*       theParent,
+                                   Handle(HYDROData_Zone) theModelObject,
+                                   const QString&         theParentEntry,
+                                   const bool             theIsBuildTree );
+
+  LightApp_DataObject* createRegion( SUIT_DataObject*         theParent,
+                                     Handle(HYDROData_Region) theModelObject,
+                                     const QString&           theParentEntry,
+                                     const bool               theIsBuildTree );
   /**
    * Build tree of model object.
    * \param theParent a created object will be appended as a child of this object