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