Salome HOME
Merge remote-tracking branch 'origin/Custom_Fitter' into CEA_2019
[modules/shaper.git] / src / ModelHighAPI / ModelHighAPI_Dumper.h
index ee432d271dad58dee3cf395d979d814d9324e456..ccd80c626bd8da98bc5c46553cd73418c83284bf 100644 (file)
@@ -1,8 +1,21 @@
-// Copyright (C) 2016-20xx CEA/DEN, EDF R&D -->
-
-// File:        ModelHighAPI_Dumper.h
-// Created:     1 August 2016
-// Author:      Artem ZHIDKOV
+// Copyright (C) 2014-2019  CEA/DEN, 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, 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
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
 
 #ifndef ModelHighAPI_Dumper_H_
 #define ModelHighAPI_Dumper_H_
@@ -40,20 +53,89 @@ class ModelAPI_CompositeFeature;
 class ModelAPI_Document;
 class ModelAPI_Entity;
 class ModelAPI_Feature;
+class ModelAPI_Folder;
 class ModelAPI_Object;
 class ModelAPI_Result;
 
 typedef std::shared_ptr<ModelAPI_Document> DocumentPtr;
 typedef std::shared_ptr<ModelAPI_Entity>   EntityPtr;
 typedef std::shared_ptr<ModelAPI_Feature>  FeaturePtr;
+typedef std::shared_ptr<ModelAPI_Folder>   FolderPtr;
 typedef std::shared_ptr<ModelAPI_Result>   ResultPtr;
 
+typedef std::set<std::string>                                                ModulesSet;
+typedef std::map<DocumentPtr, std::map<std::string, std::pair<int, int> > >  NbFeaturesMap;
+
 /**\class ModelHighAPI_Dumper
  * \ingroup CPPHighAPI
  * \brief Dump engine for the model
+ *
+ * The dumper can be customized by the set of storages (DumpStorage), for example to identify
+ * dumped selected objects by their geometrical properties. By default, the dump is executed to
+ * store original names of the selected shapes.
  */
 class ModelHighAPI_Dumper
 {
+public:
+  /** \class DumpStorage
+   *  \ingroup CPPHighAPI
+   *  \brief Storage for the dumped data. Process selection attributes to be dumped by naming.
+   */
+  class DumpStorage
+  {
+  public:
+    DumpStorage() {}
+    MODELHIGHAPI_EXPORT DumpStorage(const DumpStorage& theOther);
+    MODELHIGHAPI_EXPORT const DumpStorage& operator=(const DumpStorage& theOther);
+
+    void setFilenameSuffix(const std::string& theSuffix) { myFilenameSuffix = theSuffix; }
+
+  protected:
+    std::ostringstream& buffer() { return myDumpBuffer; }
+    std::ostringstream& fullDump() { return myFullDump; }
+
+    MODELHIGHAPI_EXPORT virtual void reserveBuffer();
+    MODELHIGHAPI_EXPORT virtual void restoreReservedBuffer();
+
+    MODELHIGHAPI_EXPORT
+    virtual void write(const std::shared_ptr<ModelAPI_AttributeSelection>& theAttrSelect);
+
+    MODELHIGHAPI_EXPORT
+    virtual bool exportTo(const std::string& theFilename, const ModulesSet& theUsedModules);
+
+  private:
+    std::string myFilenameSuffix;
+    std::ostringstream myDumpBuffer;
+    std::ostringstream myFullDump;
+
+    std::stack<std::string> myDumpBufferHideout;
+
+    friend class ModelHighAPI_Dumper;
+  };
+  typedef std::shared_ptr<DumpStorage> DumpStoragePtr;
+
+  /** \class DumpStorageGeom
+   *  \ingroup CPPHighAPI
+   *  \brief Process selection attributes to be dumped by geometric properties.
+   */
+  class DumpStorageGeom : public DumpStorage
+  {
+  protected:
+    MODELHIGHAPI_EXPORT
+    virtual void write(const std::shared_ptr<ModelAPI_AttributeSelection>& theAttrSelect);
+  };
+
+  /** \class DumpStorageWeak
+   *  \ingroup CPPHighAPI
+   *  \brief Process selection attributes to be dumped by weak naming.
+   */
+  class DumpStorageWeak : public DumpStorage
+  {
+  protected:
+    MODELHIGHAPI_EXPORT
+    virtual void write(const std::shared_ptr<ModelAPI_AttributeSelection>& theAttrSelect);
+  };
+
 public:
   /// Default constructor
   MODELHIGHAPI_EXPORT
@@ -68,7 +150,15 @@ public:
   static ModelHighAPI_Dumper* getInstance();
 
   /// Destructor
-  virtual ~ModelHighAPI_Dumper() {}
+  MODELHIGHAPI_EXPORT
+  virtual ~ModelHighAPI_Dumper();
+
+  /// Add custom storage to collect corresponding dump
+  MODELHIGHAPI_EXPORT
+  void addCustomStorage(const DumpStoragePtr& theStorage);
+  /// Clear custom storages list
+  MODELHIGHAPI_EXPORT
+  void clearCustomStorage();
 
   /// Dump given document into the file
   /// \return \c true, if succeed
@@ -77,11 +167,8 @@ public:
 
   /// Add module to list of imported modules
   /// \param theModuleName  name of the module to be imported
-  /// \param theObject      name of the entity to be imported
-  ///                       from the module (if empty, while module will be imported)
   MODELHIGHAPI_EXPORT
-  void importModule(const std::string& theModuleName,
-                    const std::string& theObject = std::string());
+  void importModule(const std::string& theModuleName);
 
   /// Returns name of specified entity
   /// \param theEntity        [in] named entity
@@ -102,6 +189,18 @@ public:
   virtual void dumpParameter(const FeaturePtr& theFeature) = 0;
   /// Dump given feature
   virtual void dumpFeature(const FeaturePtr& theFeature, const bool theForce = false) = 0;
+  /// Dump folder
+  virtual void dumpFolder(const FolderPtr& theFolder) = 0;
+
+  /// Set feature postponed until all its dependencies are not dumped.
+  /// The name of the feature is stored anyway.
+  MODELHIGHAPI_EXPORT
+  void postpone(const EntityPtr& theEntity);
+
+  /// Set a feature that should not be dumped anyway
+  MODELHIGHAPI_EXPORT
+  void doNotDumpFeature(const FeaturePtr& theFeature)
+  { myFeaturesToSkip.insert(theFeature); }
 
   /// Dump sub-feature name and color, without dumping feature creation.
   /// Used for features which creates sub-features in their execute method.
@@ -118,10 +217,6 @@ public:
   /// Return name of wrapper feature
   virtual std::string featureWrapper(const FeaturePtr& theFeature) const = 0;
 
-  /// Save all dumps into specified file
-  MODELHIGHAPI_EXPORT
-  bool exportTo(const std::string& theFileName);
-
   /// Dump character
   MODELHIGHAPI_EXPORT
   ModelHighAPI_Dumper& operator<<(const char theChar);
@@ -184,10 +279,18 @@ public:
   MODELHIGHAPI_EXPORT
   ModelHighAPI_Dumper& operator<<(const FeaturePtr& theEntity);
 
+  /// Dump folder
+  MODELHIGHAPI_EXPORT
+  ModelHighAPI_Dumper& operator<<(const FolderPtr& theFolder);
+
   /// Dump result
   MODELHIGHAPI_EXPORT
   ModelHighAPI_Dumper& operator<<(const ResultPtr& theResult);
 
+  /// Dump a list of results
+  MODELHIGHAPI_EXPORT
+  ModelHighAPI_Dumper& operator<<(const std::list<ResultPtr>& theResults);
+
   /// Dump Attribute
   MODELHIGHAPI_EXPORT ModelHighAPI_Dumper&
     operator<<(const std::shared_ptr<ModelAPI_Attribute>& theAttr);
@@ -219,12 +322,19 @@ public:
   MODELHIGHAPI_EXPORT ModelHighAPI_Dumper&
     operator<<(const std::shared_ptr<ModelAPI_AttributeStringArray>& theArray);
 
-  /// Clear dump buffer
-  MODELHIGHAPI_EXPORT
-  void clear(bool bufferOnly = false);
   /// clear list of not dumped entities
   MODELHIGHAPI_EXPORT void clearNotDumped();
 
+  /// Check the entity is already dumped
+  MODELHIGHAPI_EXPORT
+  bool isDumped(const EntityPtr& theEntity) const;
+  /// Check theRefAttr is already dumped
+  MODELHIGHAPI_EXPORT
+  bool isDumped(const std::shared_ptr<ModelAPI_AttributeRefAttr>& theRefAttr) const;
+  /// Check all objects in theRefList are already dumped
+  MODELHIGHAPI_EXPORT
+  bool isDumped(const std::shared_ptr<ModelAPI_AttributeRefList>& theRefList) const;
+
 protected:
   /// Dump "setName" command if last entity had user-defined name
   MODELHIGHAPI_EXPORT void dumpEntitySetName();
@@ -247,9 +357,6 @@ private:
   bool processSubs(const std::shared_ptr<ModelAPI_CompositeFeature>& theComposite,
                    bool theDumpModelDo = false);
 
-  /// Check the entity is already dumped
-  bool isDumped(const EntityPtr& theEntity) const;
-
   /// Stores names of results for the given feature
   void saveResultNames(const FeaturePtr& theFeature);
 
@@ -259,22 +366,29 @@ private:
   /// Check the result feature has default deflection
   bool isDefaultDeflection(const ResultPtr& theResult) const;
 
+  /// Check the result feature has default transparency
+  bool isDefaultTransparency(const ResultPtr& theResult) const;
+
+  /// Dump postponed entities
+  void dumpPostponed(bool theDumpFolders = false);
+
 private:
   struct EntityName {
     std::string myCurrentName; ///< default name of current feature
     std::string myUserName;    ///< user-defined name
     bool        myIsDefault;   ///< \c true if the name is default
+    bool        myIsDumped;    ///< shows that the names of the feature are already stored
 
     EntityName() {}
 
     EntityName(const std::string& theCurName, const std::string& theUserName, bool theDefault)
-      : myCurrentName(theCurName), myUserName(theUserName), myIsDefault(theDefault)
+      : myCurrentName(theCurName),
+        myUserName(theUserName),
+        myIsDefault(theDefault),
+        myIsDumped(false)
     {}
   };
-
-  typedef std::map<EntityPtr, EntityName>                     EntityNameMap;
-  typedef std::map<std::string, std::set<std::string> >       ModulesMap;
-  typedef std::map<DocumentPtr, std::map<std::string, int> >  NbFeaturesMap;
+  typedef std::map<EntityPtr, EntityName> EntityNameMap;
 
   struct LastDumpedEntity {
     EntityPtr            myEntity;   ///< last dumped entity
@@ -283,28 +397,35 @@ private:
     std::list<ResultPtr> myResults;
 
     LastDumpedEntity(EntityPtr theEntity, bool theUserName,
-      const std::list<ResultPtr>& theResults)
+      const std::list<ResultPtr>& theResults = std::list<ResultPtr>())
       : myEntity(theEntity), myUserName(theUserName), myResults(theResults)
     {}
   };
-  typedef std::stack<LastDumpedEntity>                              DumpStack;
+  typedef std::stack<LastDumpedEntity> DumpStack;
 
   static ModelHighAPI_Dumper* mySelf;
 
-  std::ostringstream  myDumpBuffer;         ///< intermediate buffer to store dumping data
-  std::ostringstream  myFullDump;           ///< full buffer of dumped data
+  class DumpStorageBuffer;
+  DumpStorageBuffer*  myDumpStorage;        ///< storage of dumped data
 
-  ModulesMap          myModules;            ///< modules and entities to be imported
+  ModulesSet          myModules;            ///< modules and entities to be imported
   EntityNameMap       myNames;              ///< names of the entities
   DumpStack           myEntitiesStack;      ///< stack of dumped entities
 
   NbFeaturesMap       myFeatureCount;       ///< number of features of each kind
 
+  /// features which should not be dumped (like coincidence and tangency created by tangent arc)
+  std::set<FeaturePtr> myFeaturesToSkip;
+
+  std::list<EntityPtr> myPostponed; ///< list of postponed entities (sketch constraints or folders)
+  bool myDumpPostponedInProgress; ///< processing postponed is in progress
+
 protected:
-   /// list of entities, used by other features but not dumped yet
+  /// list of entities, used by other features but not dumped yet
   std::set<EntityPtr> myNotDumpedEntities;
 
   friend class SketchAPI_Sketch;
+  friend class ModelHighAPI_Folder;
 };
 
 #endif