1 // File: SketchSolver_ConstraintManager.cpp
2 // Created: 08 May 2014
3 // Author: Artem ZHIDKOV
5 #include "SketchSolver_ConstraintManager.h"
7 #include <Events_Loop.h>
8 #include <ModelAPI_AttributeDouble.h>
9 #include <ModelAPI_AttributeRefList.h>
10 #include <ModelAPI_Data.h>
11 #include <Model_Events.h>
13 #include <SketchPlugin_Constraint.h>
15 #include <SketchPlugin_Arc.h>
16 #include <SketchPlugin_Circle.h>
17 #include <SketchPlugin_Line.h>
18 #include <SketchPlugin_Point.h>
19 #include <SketchPlugin_Sketch.h>
22 // Initialization of constraint manager self pointer
23 SketchSolver_ConstraintManager* SketchSolver_ConstraintManager::_self = 0;
25 /// Global constraint manager object
26 SketchSolver_ConstraintManager* myManager = SketchSolver_ConstraintManager::Instance();
29 // ========================================================
30 // ========= SketchSolver_ConstraintManager ===============
31 // ========================================================
32 SketchSolver_ConstraintManager* SketchSolver_ConstraintManager::Instance()
35 _self = new SketchSolver_ConstraintManager();
39 SketchSolver_ConstraintManager::SketchSolver_ConstraintManager()
43 // Register in event loop
44 Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_CREATED));
45 Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_UPDATED));
46 Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_DELETED));
47 Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_MOVED));
50 SketchSolver_ConstraintManager::~SketchSolver_ConstraintManager()
55 // ============================================================================
56 // Function: processEvent
57 // Class: SketchSolver_PluginManager
58 // Purpose: listen the event loop and process the message
59 // ============================================================================
60 void SketchSolver_ConstraintManager::processEvent(const Events_Message* theMessage)
62 if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_CREATED) ||
63 theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_UPDATED) ||
64 theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_MOVED))
66 const Model_FeatureUpdatedMessage* anUpdateMsg = dynamic_cast<const Model_FeatureUpdatedMessage*>(theMessage);
67 std::set< boost::shared_ptr<ModelAPI_Feature> > aFeatures = anUpdateMsg->features();
69 bool isModifiedEvt = theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_MOVED);
72 std::set< boost::shared_ptr<ModelAPI_Feature> >::iterator aFeatIter;
73 for (aFeatIter = aFeatures.begin(); aFeatIter != aFeatures.end(); aFeatIter++)
75 // Only sketches and constraints can be added by Create event
76 const std::string& aFeatureKind = (*aFeatIter)->getKind();
77 if (aFeatureKind.compare("Sketch") == 0)
79 boost::shared_ptr<SketchPlugin_Feature> aSketch =
80 boost::dynamic_pointer_cast<SketchPlugin_Feature>(*aFeatIter);
82 changeWorkplane(aSketch);
85 boost::shared_ptr<SketchPlugin_Constraint> aConstraint =
86 boost::dynamic_pointer_cast<SketchPlugin_Constraint>(*aFeatIter);
88 changeConstraint(aConstraint);
91 // Sketch plugin features can be only updated
92 boost::shared_ptr<SketchPlugin_Feature> aFeature =
93 boost::dynamic_pointer_cast<SketchPlugin_Feature>(*aFeatIter);
95 updateEntity(aFeature);
100 // Solve the set of constraints
101 resolveConstraints();
103 else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_DELETED))
105 const Model_FeatureDeletedMessage* aDeleteMsg = dynamic_cast<const Model_FeatureDeletedMessage*>(theMessage);
106 const std::set<std::string>& aFeatureGroups = aDeleteMsg->groups();
108 // Find "Sketch" in groups. The constraint groups should be updated when an object removed from Sketch
109 std::set<std::string>::const_iterator aFGrIter;
110 for (aFGrIter = aFeatureGroups.begin(); aFGrIter != aFeatureGroups.end(); aFGrIter++)
111 if (aFGrIter->compare("Sketch") == 0)
114 if (aFGrIter != aFeatureGroups.end())
116 std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter = myGroups.begin();
117 while (aGroupIter != myGroups.end())
119 if ((*aGroupIter)->updateGroup())
120 { // the group should be removed
122 int aShift = aGroupIter - myGroups.begin();
123 myGroups.erase(aGroupIter);
124 aGroupIter = myGroups.begin() + aShift;
132 // ============================================================================
133 // Function: changeWorkplane
134 // Class: SketchSolver_PluginManager
135 // Purpose: update workplane by given parameters of the sketch
136 // ============================================================================
137 bool SketchSolver_ConstraintManager::changeWorkplane(boost::shared_ptr<SketchPlugin_Feature> theSketch)
139 bool aResult = true; // changed when a workplane wrongly updated
140 bool isUpdated = false;
141 // Try to update specified workplane in all groups
142 std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter;
143 for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
144 if ((*aGroupIter)->isBaseWorkplane(theSketch))
147 if (!(*aGroupIter)->updateWorkplane())
150 // If the workplane is not updated, so this is a new workplane
153 SketchSolver_ConstraintGroup* aNewGroup = new SketchSolver_ConstraintGroup(theSketch);
154 // Verify that the group is created successfully
155 if (!aNewGroup->isBaseWorkplane(theSketch))
160 myGroups.push_back(aNewGroup);
165 // ============================================================================
166 // Function: changeConstraint
167 // Class: SketchSolver_PluginManager
168 // Purpose: create/update the constraint and place it into appropriate group
169 // ============================================================================
170 bool SketchSolver_ConstraintManager::changeConstraint(
171 boost::shared_ptr<SketchPlugin_Constraint> theConstraint)
173 // Search the groups which this constraint touches
174 std::set<Slvs_hGroup> aGroups;
175 findGroups(theConstraint, aGroups);
177 // Process the groups list
178 if (aGroups.size() == 0)
179 { // There are no groups applicable for this constraint => create new one
180 boost::shared_ptr<SketchPlugin_Feature> aWP = findWorkplaneForConstraint(theConstraint);
181 if (!aWP) return false;
182 SketchSolver_ConstraintGroup* aGroup = new SketchSolver_ConstraintGroup(aWP);
183 if (!aGroup->changeConstraint(theConstraint))
188 myGroups.push_back(aGroup);
191 else if (aGroups.size() == 1)
192 { // Only one group => add constraint into it
193 Slvs_hGroup aGroupId = *(aGroups.begin());
194 std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter;
195 for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
196 if ((*aGroupIter)->getId() == aGroupId)
197 return (*aGroupIter)->changeConstraint(theConstraint);
199 else if (aGroups.size() > 1)
200 { // Several groups applicable for this constraint => need to merge them
201 std::set<Slvs_hGroup>::const_iterator aGroupsIter = aGroups.begin();
203 // Search first group
204 std::vector<SketchSolver_ConstraintGroup*>::iterator aFirstGroupIter;
205 for (aFirstGroupIter = myGroups.begin(); aFirstGroupIter != myGroups.end(); aFirstGroupIter++)
206 if ((*aFirstGroupIter)->getId() == *aGroupsIter)
208 if (aFirstGroupIter == myGroups.end())
211 // Append other groups to the first one
212 std::vector<SketchSolver_ConstraintGroup*>::iterator anOtherGroupIter = aFirstGroupIter + 1;
213 for (aGroupsIter++; aGroupsIter != aGroups.end(); aGroupsIter++)
215 for ( ; anOtherGroupIter != myGroups.end(); anOtherGroupIter++)
216 if ((*anOtherGroupIter)->getId() == *aGroupsIter)
218 if (anOtherGroupIter == myGroups.end())
219 { // Group disappears
220 anOtherGroupIter = aFirstGroupIter + 1;
224 (*aFirstGroupIter)->mergeGroups(**anOtherGroupIter);
225 int aShiftFirst = aFirstGroupIter - myGroups.begin();
226 int aShiftOther = anOtherGroupIter - myGroups.begin();
227 delete *anOtherGroupIter;
228 myGroups.erase(anOtherGroupIter);
229 aFirstGroupIter = myGroups.begin() + aShiftFirst;
230 anOtherGroupIter = myGroups.begin() + aShiftOther;
233 return (*aFirstGroupIter)->changeConstraint(theConstraint);
236 // Something goes wrong
240 // ============================================================================
241 // Function: updateEntity
242 // Class: SketchSolver_PluginManager
243 // Purpose: update any element on the sketch, which is used by constraints
244 // ============================================================================
245 void SketchSolver_ConstraintManager::updateEntity(boost::shared_ptr<SketchPlugin_Feature> theFeature)
247 // Create list of attributes depending on type of the feature
248 std::vector<std::string> anAttrList;
249 const std::string& aFeatureKind = theFeature->getKind();
251 if (aFeatureKind.compare("SketchPoint") == 0)
252 anAttrList.push_back(POINT_ATTR_COORD);
254 else if (aFeatureKind.compare("SketchLine") == 0)
256 anAttrList.push_back(LINE_ATTR_START);
257 anAttrList.push_back(LINE_ATTR_END);
260 else if (aFeatureKind.compare("SketchCircle") == 0)
262 anAttrList.push_back(CIRCLE_ATTR_CENTER);
263 anAttrList.push_back(CIRCLE_ATTR_RADIUS);
266 else if (aFeatureKind.compare("SketchArc") == 0)
268 anAttrList.push_back(ARC_ATTR_CENTER);
269 anAttrList.push_back(ARC_ATTR_START);
270 anAttrList.push_back(ARC_ATTR_END);
272 /// \todo Other types of features should be implemented
274 // Check changing of feature's attributes (go through the groups and search usage of the attributes)
275 std::vector<std::string>::const_iterator anAttrIter;
276 for (anAttrIter = anAttrList.begin(); anAttrIter != anAttrList.end(); anAttrIter++)
278 std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter;
279 for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
281 if ((*aGroupIter)->isEmpty())
283 boost::shared_ptr<ModelAPI_Attribute> anAttribute =
284 boost::dynamic_pointer_cast<ModelAPI_Attribute>(theFeature->data()->attribute(*anAttrIter));
285 (*aGroupIter)->updateEntityIfPossible(anAttribute);
291 // ============================================================================
292 // Function: findGroups
293 // Class: SketchSolver_PluginManager
294 // Purpose: search groups of entities interacting with given constraint
295 // ============================================================================
296 void SketchSolver_ConstraintManager::findGroups(
297 boost::shared_ptr<SketchPlugin_Constraint> theConstraint,
298 std::set<Slvs_hGroup>& theGroupIDs) const
300 boost::shared_ptr<SketchPlugin_Feature> aWP = findWorkplaneForConstraint(theConstraint);
302 SketchSolver_ConstraintGroup* anEmptyGroup = 0; // appropriate empty group for specified constraint
303 std::vector<SketchSolver_ConstraintGroup*>::const_iterator aGroupIter;
304 for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
305 if (aWP == (*aGroupIter)->getWorkplane() && (*aGroupIter)->isInteract(theConstraint))
307 if (!(*aGroupIter)->isEmpty())
308 theGroupIDs.insert((*aGroupIter)->getId());
309 else if (!anEmptyGroup)
310 anEmptyGroup = *aGroupIter;
313 // When only empty group is found, use it
314 if (anEmptyGroup && theGroupIDs.empty())
315 theGroupIDs.insert(anEmptyGroup->getId());
318 // ============================================================================
319 // Function: findWorkplaneForConstraint
320 // Class: SketchSolver_PluginManager
321 // Purpose: search workplane containing given constraint
322 // ============================================================================
323 boost::shared_ptr<SketchPlugin_Feature> SketchSolver_ConstraintManager::findWorkplaneForConstraint(
324 boost::shared_ptr<SketchPlugin_Constraint> theConstraint) const
326 // Already verified workplanes
327 std::set< boost::shared_ptr<SketchPlugin_Feature> > aVerified;
329 std::vector<SketchSolver_ConstraintGroup*>::const_iterator aGroupIter;
330 for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
332 boost::shared_ptr<SketchPlugin_Feature> aWP = (*aGroupIter)->getWorkplane();
333 if (aVerified.find(aWP) != aVerified.end())
336 boost::shared_ptr<ModelAPI_AttributeRefList> aWPFeatures =
337 boost::dynamic_pointer_cast<ModelAPI_AttributeRefList>(aWP->data()->attribute(SKETCH_ATTR_FEATURES));
338 std::list< boost::shared_ptr<ModelAPI_Feature> > aFeaturesList = aWPFeatures->list();
339 std::list< boost::shared_ptr<ModelAPI_Feature> >::const_iterator anIter;
340 for (anIter = aFeaturesList.begin(); anIter != aFeaturesList.end(); anIter++)
341 if (*anIter == theConstraint)
342 return aWP; // workplane is found
343 aVerified.insert(aWP);
346 return boost::shared_ptr<SketchPlugin_Feature>();
349 // ============================================================================
350 // Function: resolveConstraints
351 // Class: SketchSolver_PluginManager
352 // Purpose: change entities according to available constraints
353 // ============================================================================
354 void SketchSolver_ConstraintManager::resolveConstraints()
356 std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter;
357 for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
358 (*aGroupIter)->resolveConstraints();
360 // Features may be updated => send events
361 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_FEATURE_UPDATED));