Salome HOME
Update conditions to add arc into solver (issue #1165)
[modules/shaper.git] / src / SketchSolver / SketchSolver_Storage.h
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
2
3 // File:    SketchSolver_Storage.h
4 // Created: 30 Nov 2015
5 // Author:  Artem ZHIDKOV
6
7 #ifndef SketchSolver_Storage_H_
8 #define SketchSolver_Storage_H_
9
10 #include <SketchSolver.h>
11 #include <SketchSolver_IConstraintWrapper.h>
12 #include <SketchSolver_IEntityWrapper.h>
13 #include <SketchSolver_IParameterWrapper.h>
14 #include <SketchSolver_ISolver.h>
15
16 #include <ModelAPI_Attribute.h>
17 #include <ModelAPI_AttributeDouble.h>
18 #include <ModelAPI_Feature.h>
19 #include <SketchPlugin_Constraint.h>
20
21
22 typedef std::map<EntityWrapperPtr, std::set<EntityWrapperPtr> > CoincidentPointsMap;
23
24
25 /** \class   SketchSolver_Storage
26  *  \ingroup Plugins
27  *  \brief   Interface to map SketchPlugin features to the entities of corresponding solver.
28  */
29 class SketchSolver_Storage
30 {
31 private:
32   SketchSolver_Storage();
33   SketchSolver_Storage(const SketchSolver_Storage&);
34   SketchSolver_Storage& operator=(const SketchSolver_Storage&);
35
36 public:
37   SketchSolver_Storage(const GroupID& theGroup)
38     : myGroupID(theGroup),
39       myNeedToResolve(false),
40       myEventsBlocked(false),
41       myExistArc(false)
42   {}
43
44   /// \brief Change mapping between constraint from SketchPlugin and
45   ///        a constraint applicable for corresponding solver.
46   /// \param theConstraint       [in]   original SketchPlugin constraint
47   /// \param theSolverConstraint [in]   solver's constraints
48   SKETCHSOLVER_EXPORT void addConstraint(ConstraintPtr        theConstraint,
49                                          ConstraintWrapperPtr theSolverConstraints);
50   /// \brief Change mapping between constraint from SketchPlugin and
51   ///        the list of constraints applicable for corresponding solver.
52   /// \param theConstraint        [in]   original SketchPlugin constraint
53   /// \param theSolverConstraints [in]   list of solver's constraints
54   SKETCHSOLVER_EXPORT virtual
55     void addConstraint(ConstraintPtr                   theConstraint,
56                        std::list<ConstraintWrapperPtr> theSolverConstraints);
57
58   /// \brief Convert feature to the form applicable for specific solver and map it
59   /// \param theFeature [in]  feature to convert
60   /// \param theGroup   [in]  id of the group where the feature should be placed
61   /// \return \c true if the feature has been created or updated
62   SKETCHSOLVER_EXPORT bool update(FeaturePtr theFeature, const GroupID& theGroup = GID_UNKNOWN);
63   /// \brief Convert attribute to the form applicable for specific solver and map it
64   /// \param theFeature [in]  feature to convert
65   /// \return \c true if the attribute has been created or updated
66   SKETCHSOLVER_EXPORT bool update(AttributePtr theAttribute, const GroupID& theGroup = GID_UNKNOWN);
67
68   /// \brief Returns constraint related to corresponding constraint
69   SKETCHSOLVER_EXPORT
70     const std::list<ConstraintWrapperPtr>& constraint(const ConstraintPtr& theConstraint) const;
71
72   /// \brief Returns entity related to corresponding feature
73   SKETCHSOLVER_EXPORT const EntityWrapperPtr& entity(const FeaturePtr& theFeature) const;
74   /// \brief Returns entity related to corresponding attribute
75   SKETCHSOLVER_EXPORT const EntityWrapperPtr& entity(const AttributePtr& theAttribute) const;
76
77   /// \brief Return parsed sketch entity
78   const EntityWrapperPtr& sketch() const;
79   /// \brief Set parsed sketch entity.
80   /// Be careful, this method does not update fields of the storage specific for the solver.
81   /// Does not update if the sketch already exists.
82   void setSketch(const EntityWrapperPtr& theSketch);
83
84   /// \brief Mark two points as coincident
85   virtual void addCoincidentPoints(EntityWrapperPtr theMaster, EntityWrapperPtr theSlave) = 0;
86
87   /// \brief Shows the storage has any constraint twice
88   virtual bool hasDuplicatedConstraint() const = 0;
89
90   /// \brief Removes constraint from the storage
91   /// \return \c true if the constraint and all its parameters are removed successfully
92   SKETCHSOLVER_EXPORT bool removeConstraint(ConstraintPtr theConstraint);
93   /// \brief Removes feature from the storage
94   /// \return \c true if the feature and its attributes are removed successfully;
95   ///         \c false if the feature or any it attribute is used by remaining constraints.
96   SKETCHSOLVER_EXPORT bool removeEntity(FeaturePtr theFeature);
97   /// \brief Removes attribute from the storage
98   /// \return \c true if the attribute is not used by remaining features and constraints
99   SKETCHSOLVER_EXPORT bool removeEntity(AttributePtr theAttribute);
100
101   /// \brief Remove all features became invalid
102   SKETCHSOLVER_EXPORT void removeInvalidEntities();
103
104 ////  /// \brief Mark specified constraint as temporary
105 ////  virtual void setTemporary(ConstraintPtr theConstraint) = 0;
106 ////  /// \brief Returns number of temporary constraints
107 ////  virtual size_t nbTemporary() const = 0;
108 ////  /// \brief Remove temporary constraints
109 ////  /// \param theNbConstraints [in]  number of temporary constraints to be deleted
110 ////  /// \return number of remaining temporary constraints
111 ////  virtual size_t removeTemporary(size_t theNbConstraints = 1) = 0;
112
113   /// \brief Check whether the feature or its attributes are used by this storage
114   /// \param theFeature [in]  feature to be checked
115   /// \return \c true if the feature interacts with the storage
116   bool isInteract(const FeaturePtr& theFeature) const;
117   /// \brief Check whether the attribute is used by this storage
118   /// \param theAttribute [in]  attribute to be checked
119   /// \return \c true if the attribute interacts with the storage
120   bool isInteract(const AttributePtr& theAttribute) const;
121
122   /// \brief Check the features is not removed
123   bool isConsistent() const;
124
125   /// \brief Check the entity is fixed.
126   ///        If the point is under verification, all coincident points are checked too.
127   SKETCHSOLVER_EXPORT bool isFixed(EntityWrapperPtr theEntity) const;
128
129   /// \brief Shows the sketch should be resolved
130   virtual bool isNeedToResolve()
131   { return myNeedToResolve; }
132   /// \brief Changes the flag of group to be resolved
133   void setNeedToResolve(bool theFlag)
134   { myNeedToResolve = theFlag; }
135
136   /// \brief Initialize solver by constraints, entities and parameters
137   virtual void initializeSolver(SolverPtr theSolver) = 0;
138
139   /// \brief Update SketchPlugin features after resolving constraints
140   /// \param theFixedOnly [in]  if \c true the fixed points will be updated only
141   virtual void refresh(bool theFixedOnly = false) const = 0;
142
143   /// \brief Check if some parameters or entities are returned
144   ///        to the current group after removing temporary constraints
145   virtual void verifyFixed() = 0;
146
147   /// \brief Calculate point on theBase entity. Value theCoeff is in [0.0 .. 1.0] and
148   ///        shows the distance from the start point.
149   virtual EntityWrapperPtr calculateMiddlePoint(EntityWrapperPtr theBase,
150                                                 double theCoeff) = 0;
151
152   /// \brief Block or unblock events when refreshing features
153   SKETCHSOLVER_EXPORT void blockEvents(bool isBlocked);
154   /// \brief Shows the events are blocked for the features in the storage
155   bool isEventsBlocked() const
156   { return myEventsBlocked; }
157
158 protected:
159   /// \brief Change mapping feature from SketchPlugin and
160   ///        the entity applicable for corresponding solver.
161   /// \param theFeature      [in]  original SketchPlugin feature
162   /// \param theSolverEntity [in]  solver's entity, created outside
163   SKETCHSOLVER_EXPORT
164     void addEntity(FeaturePtr       theFeature,
165                    EntityWrapperPtr theSolverEntity);
166
167   /// \brief Change mapping attribute of a feature and the entity applicable for corresponding solver.
168   /// \param theAttribute    [in]  original attribute
169   /// \param theSolverEntity [in]  solver's entity, created outside
170   SKETCHSOLVER_EXPORT
171     void addEntity(AttributePtr     theAttribute,
172                    EntityWrapperPtr theSolverEntity);
173
174   /// \brief Update constraint's data
175   /// \return \c true if any value is updated
176   virtual bool update(ConstraintWrapperPtr theConstraint) = 0;
177   /// \brief Update entity's data
178   /// \return \c true if any value is updated
179   virtual bool update(EntityWrapperPtr theEntity) = 0;
180   /// \brief Update parameter's data
181   /// \return \c true if the value of parameter is updated
182   virtual bool update(ParameterWrapperPtr theParameter) = 0;
183
184   /// \brief Remove constraint
185   /// \return \c true if the constraint and all its parameters are removed successfully
186   SKETCHSOLVER_EXPORT virtual bool remove(ConstraintWrapperPtr theConstraint);
187   /// \brief Remove entity
188   /// \return \c true if the entity and all its parameters are removed successfully
189   SKETCHSOLVER_EXPORT virtual bool remove(EntityWrapperPtr theEntity);
190   /// \brief Remove parameter
191   /// \return \c true if the parameter has been removed
192   virtual bool remove(ParameterWrapperPtr theParameter) = 0;
193
194   /// \brief Update the group for the given entity, its sub-entities and parameters
195   virtual void changeGroup(EntityWrapperPtr theEntity, const GroupID& theGroup) = 0;
196   /// \brief Update the group for the given parameter
197   virtual void changeGroup(ParameterWrapperPtr theParam, const GroupID& theGroup) = 0;
198
199   /// \brief Verify the feature or any its attribute is used by constraint
200   SKETCHSOLVER_EXPORT bool isUsed(FeaturePtr theFeature) const;
201   /// \brief Verify the attribute is used by constraint
202   SKETCHSOLVER_EXPORT bool isUsed(AttributePtr theAttirubute) const;
203
204   /// \brief Find arcs without corresponding entity applicable for the solver and build them
205   SKETCHSOLVER_EXPORT void processArcs();
206
207 private:
208   /// \brief Find the normal of the sketch
209   EntityWrapperPtr getNormal() const;
210
211 protected:
212   GroupID myGroupID;       ///< identifier of the group, this storage belongs to
213   bool    myNeedToResolve; ///< parameters are changed and group needs to be resolved
214   bool    myEventsBlocked; ///< indicates that features do not send events
215   bool    myExistArc;      ///< the storage has any point of arc but not full arc, need to add it
216
217   /// map SketchPlugin constraint to a list of solver's constraints
218   std::map<ConstraintPtr, std::list<ConstraintWrapperPtr> > myConstraintMap;
219   /// map SketchPlugin feature to solver's entity
220   std::map<FeaturePtr, EntityWrapperPtr>                    myFeatureMap;
221   /// map attribute to solver's entity
222   std::map<AttributePtr, EntityWrapperPtr>                  myAttributeMap;
223
224   CoincidentPointsMap myCoincidentPoints; ///< lists of coincident points (first is a master point, second is a set of slaves)
225 };
226
227 typedef std::shared_ptr<SketchSolver_Storage> StoragePtr;
228
229 #endif