Salome HOME
refs #567: add "POLYLINES" partition and modified icon for Land Cover object.
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_DataModel.h
index a4216902d1e1ee45dac3a2e123fd92f43bf13e13..6707dd2edd8b0cdc4013f6413b689450520ae2eb 100644 (file)
@@ -1,12 +1,8 @@
-// Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
-//
-// Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
-//
+// Copyright (C) 2014-2015  EDF-R&D
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 // License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
 //
 // This library is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
 //
 
+
 #ifndef HYDROGUI_DATAMODEL_H
 #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 +35,7 @@
 class CAM_DataObject;
 class SUIT_DataObject;
 class HYDROGUI_DataObject;
+class HYDROGUI_Zone;
 
 /**
  * \class HYDROGUI_DataModel
@@ -111,7 +113,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 +141,13 @@ 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,
+                        const bool theLandCover );
+
   /**
    * Correct an internal model object according to the current document mode
    */
@@ -147,7 +156,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 +219,27 @@ public:
    */
   bool paste();
 
+  /**
+   * Rename the object
+   */
+  bool rename( Handle(HYDROData_Entity) theEntity, const QString& theName );
+
+  /**
+   * Creates the Calculation Case subtree for usage within an operation dialog.
+   * \param theParent a created object will be appended as a child of this GUI object
+   * \param theCase the calculation case model object
+   * \param theLandCover if true - land cover regions will be represented in the tree
+   */
+  void buildCaseTree( SUIT_DataObject*                  theParent,
+                      Handle(HYDROData_CalculationCase) theCase,
+                      const bool theLandCover );
+
+  /**
+   * Updates the object subtree.
+   * \param theObj the data model entity
+   */
+  void updateObjectTree( Handle(HYDROData_Entity)& theObj );
+
   /**
    * Update the sequence of the objects to be copied
    */
@@ -235,8 +265,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
@@ -249,14 +280,77 @@ protected:
                                      const QString&   theParentEntry = QString() );
 
   /**
-   * Build tree of model object.
+   * 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
+   * \param theIsInOperation if true then the tree is used for a browser within an operation, it is false by default
+   */
+  LightApp_DataObject* buildObject( SUIT_DataObject*     theParent,
+                                    HYDROGUI_DataObject* theObject,
+                                    const QString&       theParentEntry,
+                                    const bool           theIsBuildTree,
+                                    const bool           theIsInOperation = false );
+
+  /**
+   * Build object tree if the flag theIsBuildTree is true. 
+   * \param theParent a created object will be appended as a child of this GUI object
+   * \param theModelObject the data model zone object
+   * \param theParentEntry the entry of parent object
+   * \param theIsBuildTree if true then build the subtree of the GUI object
+   * \param theIsInOperation if true then the tree is used for a browser within an operation, it is false by default
+   */
+  LightApp_DataObject* createZone( SUIT_DataObject*       theParent,
+                                   Handle(HYDROData_Zone) theModelObject,
+                                   const QString&         theParentEntry,
+                                   const bool             theIsBuildTree ,
+                                   const bool             theIsInOperation = false  );
+
+  /**
+   * Build object tree if the flag theIsBuildTree is true. 
+   * \param theParent a created object will be appended as a child of this GUI object
+   * \param theModelObject the data model region object
+   * \param theParentEntry the entry of parent object
+   * \param theIsBuildTree if true then build the subtree of the GUI object
+   * \param theIsInOperation if true then the tree is used for a browser within an operation, it is false by default
+   */
+  LightApp_DataObject* createRegion( SUIT_DataObject*         theParent,
+                                     Handle(HYDROData_Region) theModelObject,
+                                     const QString&           theParentEntry,
+                                     const bool               theIsBuildTree ,
+                                     const bool               theIsInOperation = false );
+  /**
+   * Creates the default Strickler table object: both GUI data object and corresponding model object
+   * \param theDocument a document into which created object will be added
+   * \param theParent a created object will be appended as a child of this GUI object
+   */
+  void                 createDefaultStricklerTable( const Handle(HYDROData_Document)& theDocument,
+                                                    LightApp_DataObject*              theParent );
+  /**
+   * Build partition for object.
+   * \param theObject gui object for which the partition will be build
+   * \param theObjects sequence of builded objects
+   * \param thePartName name of created partition
+   * \param theIsCreateEmpty if true then partition will be created in any case
+   */
+  void                 buildObjectPartition( SUIT_DataObject*                   theObject,
+                                             const HYDROData_SequenceOfObjects& theObjects,
+                                             const QString&                     thePartName,
+                                             const bool                         theIsCreateEmpty );
+
+  /**
+   * Build tree of a model object.
    * \param theParent a created object will be appended as a child of this object
    * \param theObject gui object for which the tree will be build
    * \param theParentEntry entry of parent object
+   * \param theIsInOperation if true then the tree is used for a browser within an operation, it is false by default
    */
   void                 buildObjectTree( SUIT_DataObject* theParent,
                                         SUIT_DataObject* theObject,
-                                        const QString&   theParentEntry = QString() );
+                                        const QString&   theParentEntry = QString(),
+                                        const bool       theIsInOperation = false );
 
   /**
    * Removes data object from the tree.
@@ -275,8 +369,18 @@ protected:
   static SUIT_DataObject* findChildByName( const SUIT_DataObject* theFather,
                                            const QString& theName );
 
+  /**
+   * Set object visibility state.
+   * \param theModelObject the data model object
+   * \param theDataObject the GUI object
+   */
+  void setObjectVisibilityState( Handle(HYDROData_Entity) theModelObject,
+                                 HYDROGUI_DataObject* theObject );
+                                 
+
 protected:
   QString myStudyURL; ///< the saved/opened document URL
+  QByteArray myStates;
 };
 
 #endif