Salome HOME
refs #567: add "POLYLINES" partition and modified icon for Land Cover object.
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_DataModel.h
index 9b08ac8d6deb7469ec39ce8ca3b4d0fb6b5d5c72..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
@@ -80,6 +82,14 @@ public:
    */
   virtual bool close();
 
+  /**
+   * Dump study data to Python script. Reimplemented.
+   */
+  virtual bool                        dumpPython( const QString& theURL,
+                                                  CAM_Study*     theStudy,
+                                                  bool           isMultiFile,
+                                                  QStringList&   theListOfFiles );
+
   /**
    * Returns modification status. Reimplemented.
    * \return boolean value of modification status
@@ -101,14 +111,14 @@ public:
   /**
    * Returns data object corresponding to the model object.
    * \param the data model object
-   * \returns the only one object referenced to the given data model object, or null if not found
+   * \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.
    * \param the data object
-   * \returns the object referenced to the given object, or null if not found
+   * \return the object referenced to the given object, or null if not found
    */
   virtual HYDROGUI_DataObject* getReferencedDataObject( HYDROGUI_DataObject* theObject );
 
@@ -131,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
    */
@@ -139,8 +156,8 @@ public:
   /**
    * Find a data object by the specified entry and kind
    */
-  Handle(HYDROData_Object) objectByEntry( const QString& theEntry,
-                                          const ObjectKind theObjectKind );
+  Handle(HYDROData_Entity) objectByEntry( const QString& theEntry,
+                                          const ObjectKind theObjectKind = KIND_UNKNOWN );
 
   /**
    * Check if it is possible to perform 'undo' operation
@@ -182,6 +199,59 @@ public:
    */
   bool redo();
 
+  /**
+   * Check if it is possible to perform 'copy' operation
+   */
+  bool canCopy();
+
+  /**
+   * Check if it is possible to perform 'paste' operation
+   */
+  bool canPaste();
+
+  /**
+   * Perform the 'copy' operation
+   */
+  bool copy();
+
+  /**
+   * Perform the 'paste' operation
+   */
+  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
+   */
+  static void changeCopyingObjects( const HYDROData_SequenceOfObjects& );
+
+  /**
+   * Returns name of the partition containing the objects of the specified kind
+   * \param theObjectKind kind of objects
+   * \return partition name
+   */
+  static QString partitionName( const ObjectKind theObjectKind );
+
 protected:
   /**
    * Returns the document for the current study
@@ -191,18 +261,96 @@ protected:
   /**
    * 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 theObject model object
+   * \param theModelObject model object
+   * \param theParentEntry entry of parent object
    */
-  LightApp_DataObject* createObject( SUIT_DataObject* theParent,
-                                     Handle(HYDROData_Object) theModelObject );
+  LightApp_DataObject* createObject( SUIT_DataObject*         theParent,
+                                     Handle(HYDROData_Entity) theModelObject,
+                                     const QString&           theParentEntry = QString(),
+                                     const bool               theIsBuildTree = true );
 
   /**
    * Creates the GUI data object without corresponding model object: just by name
    * \param theParent a created object will be appended as a child of this object
    * \param theName name of this object
+   * \param theParentEntry entry of parent object
    */
   LightApp_DataObject* createObject( SUIT_DataObject* theParent,
-                                     const QString& theName );
+                                     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
+   * \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 bool       theIsInOperation = false );
 
   /**
    * Removes data object from the tree.
@@ -221,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