Salome HOME
Fix for the issue #2753 : error when dump/load script
[modules/shaper.git] / src / SketcherPrs / SketcherPrs_PositionMgr.h
index 34529101a92af4f02fb9e139d5141826c863d206..7acc36e1150d9c56bc2031fc26df6711499f5190 100644 (file)
@@ -21,6 +21,7 @@
 #ifndef SketcherPrs_PositionMgr_H
 #define SketcherPrs_PositionMgr_H
 
+#include "SketcherPrs.h"
 #include "SketcherPrs_SymbolPrs.h"
 
 #include <GeomAPI_Shape.h>
 #include <ModelAPI_Object.h>
 
 #include <map>
+#include <array>
 
 /**
 * \ingroup GUI
 * A class Position Manager which manages position of constraints symbols along a source object line.
 * it expects that symbol icons have size 16x16 px
 */
-class SketcherPrs_PositionMgr
+class SKETCHERPRS_EXPORT SketcherPrs_PositionMgr
 {
 public:
   /// Returns current instance of position manager
@@ -52,6 +54,9 @@ public:
   /// \param thePrs a constraint presentation
   void deleteConstraint(const SketcherPrs_SymbolPrs* thePrs);
 
+  /// Cleares all stored positions for all constraints
+  void clearAll()  { myShapes.clear(); myPntShapes.clear(); }
+
 private:
   /// Constructor
   SketcherPrs_PositionMgr();
@@ -64,7 +69,8 @@ private:
   /// Returns position index of the given constraint around a point
   /// \param theLine constrained object
   /// \param thePrs a presentation of constraint
-  int getPositionIndex(GeomPointPtr thePos, const SketcherPrs_SymbolPrs* thePrs);
+  const std::array<int, 2>& getPositionIndex(GeomPointPtr thePos,
+                                      const SketcherPrs_SymbolPrs* thePrs);
 
   /// Returns position of a constraint around a point
   /// \param theLine a base object of the constraint
@@ -78,10 +84,7 @@ private:
 
 private:
   typedef std::map<const SketcherPrs_SymbolPrs*, int> PositionsMap;
-  typedef std::map<const ModelAPI_Feature*, int> FeaturesMap;
-
-  /// The map which contains position of presentation
-  PositionsMap myIndexes;
+  typedef std::map<const ModelAPI_Feature*, std::array<int, 2>> FeaturesMap;
 
   /// The map contains position index
   std::map<ObjectPtr, PositionsMap> myShapes;