Salome HOME
Boost has been removed from code
[modules/shaper.git] / src / SketchSolver / SketchSolver_ConstraintManager.cpp
index d9e10c85e2a214d98e4fc3ac276aae62a0448476..6cad183e4525abf4cba2cfb13a08b91d88fac7db 100644 (file)
@@ -10,6 +10,7 @@
 #include <ModelAPI_Data.h>
 #include <ModelAPI_Events.h>
 #include <ModelAPI_Object.h>
+#include <ModelAPI_ResultConstruction.h>
 
 #include <SketchPlugin_Constraint.h>
 
 
 #include <list>
 
-
 // Initialization of constraint manager self pointer
 SketchSolver_ConstraintManager* SketchSolver_ConstraintManager::_self = 0;
 
 /// Global constraint manager object
 SketchSolver_ConstraintManager* myManager = SketchSolver_ConstraintManager::Instance();
 
-
 // ========================================================
 // ========= SketchSolver_ConstraintManager ===============
 // ========================================================
@@ -42,6 +41,7 @@ SketchSolver_ConstraintManager* SketchSolver_ConstraintManager::Instance()
 SketchSolver_ConstraintManager::SketchSolver_ConstraintManager()
 {
   myGroups.clear();
+  myIsComputed = false;
 
   // Register in event loop
   Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_CREATED));
@@ -57,84 +57,81 @@ SketchSolver_ConstraintManager::~SketchSolver_ConstraintManager()
 
 // ============================================================================
 //  Function: processEvent
-//  Class:    SketchSolver_PluginManager
+//  Class:    SketchSolver_Session
 //  Purpose:  listen the event loop and process the message
 // ============================================================================
-void SketchSolver_ConstraintManager::processEvent(const Events_Message* theMessage)
+void SketchSolver_ConstraintManager::processEvent(
+  const std::shared_ptr<Events_Message>& theMessage)
 {
-  if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_CREATED) ||
-      theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED) || 
-      theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_MOVED))
-  {
-    const ModelAPI_ObjectUpdatedMessage* anUpdateMsg = 
-      dynamic_cast<const ModelAPI_ObjectUpdatedMessage*>(theMessage);
-    std::set< ObjectPtr > aFeatures = anUpdateMsg->objects();
-
-    bool isModifiedEvt = 
-      theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_MOVED);
-    if (!isModifiedEvt)
-    {
-      std::set< ObjectPtr >::iterator aFeatIter;
-      for (aFeatIter = aFeatures.begin(); aFeatIter != aFeatures.end(); aFeatIter++)
-      {
-        FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(*aFeatIter);
-        if (!aFeature) continue;
-        // Only sketches and constraints can be added by Create event
-        const std::string& aFeatureKind = aFeature->getKind();
-        if (aFeatureKind.compare(SketchPlugin_Sketch::ID()) == 0)
-        {
-          boost::shared_ptr<SketchPlugin_Feature> aSketch =
-            boost::dynamic_pointer_cast<SketchPlugin_Feature>(aFeature);
-          if (aSketch)
-            changeWorkplane(aSketch);
+  if (myIsComputed)
+    return;
+  if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_CREATED)
+      || theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED)
+      || theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_MOVED)) {
+    std::shared_ptr<ModelAPI_ObjectUpdatedMessage> anUpdateMsg =
+        std::dynamic_pointer_cast<ModelAPI_ObjectUpdatedMessage>(theMessage);
+    std::set<ObjectPtr> aFeatures = anUpdateMsg->objects();
+
+    bool isMovedEvt = theMessage->eventID()
+        == Events_Loop::loop()->eventByName(EVENT_OBJECT_MOVED);
+    if (isMovedEvt) {
+      std::set<ObjectPtr>::iterator aFeatIter;
+      for (aFeatIter = aFeatures.begin(); aFeatIter != aFeatures.end(); aFeatIter++) {
+        std::shared_ptr<SketchPlugin_Feature> aSFeature = 
+            std::dynamic_pointer_cast<SketchPlugin_Feature>(*aFeatIter);
+        if (aSFeature)
+          updateEntity(aSFeature);
+      }
+    } else {
+      std::set<ObjectPtr>::iterator aFeatIter;
+      // iterate sketchers fisrt to create all sketches before (on load may exist several sketches)
+      for (aFeatIter = aFeatures.begin(); aFeatIter != aFeatures.end(); aFeatIter++) {
+        FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(*aFeatIter);
+        if (!aFeature)
           continue;
+        const std::string& aFeatureKind = aFeature->getKind();
+        if (aFeatureKind.compare(SketchPlugin_Sketch::ID()) == 0) {
+          std::shared_ptr<ModelAPI_CompositeFeature> aSketch = std::dynamic_pointer_cast<
+              ModelAPI_CompositeFeature>(aFeature);
+          changeWorkplane(aSketch);
         }
-        boost::shared_ptr<SketchPlugin_Constraint> aConstraint =
-          boost::dynamic_pointer_cast<SketchPlugin_Constraint>(aFeature);
-        if (aConstraint)
-          changeConstraint(aConstraint);
-        else
-        {
-          // Sketch plugin features can be only updated
-          boost::shared_ptr<SketchPlugin_Feature> aSFeature =
-            boost::dynamic_pointer_cast<SketchPlugin_Feature>(aFeature);
-          if (aSFeature)
-            updateEntity(aSFeature);
-        }
+      }
+      // then get anything but not the sketch
+      for (aFeatIter = aFeatures.begin(); aFeatIter != aFeatures.end(); aFeatIter++) {
+        std::shared_ptr<SketchPlugin_Feature> aFeature = 
+          std::dynamic_pointer_cast<SketchPlugin_Feature>(*aFeatIter);
+        if (!aFeature)
+          continue;
+          changeConstraintOrEntity(aFeature);
       }
     }
 
     // Solve the set of constraints
     resolveConstraints();
-  }
-  else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED))
-  {
-    const ModelAPI_ObjectDeletedMessage* aDeleteMsg = 
-      dynamic_cast<const ModelAPI_ObjectDeletedMessage*>(theMessage);
+  } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED)) {
+    std::shared_ptr<ModelAPI_ObjectDeletedMessage> aDeleteMsg =
+        std::dynamic_pointer_cast<ModelAPI_ObjectDeletedMessage>(theMessage);
     const std::set<std::string>& aFeatureGroups = aDeleteMsg->groups();
 
     // Find SketchPlugin_Sketch::ID() in groups. The constraint groups should be updated when an object removed from Sketch
     std::set<std::string>::const_iterator aFGrIter;
     for (aFGrIter = aFeatureGroups.begin(); aFGrIter != aFeatureGroups.end(); aFGrIter++)
-      if (aFGrIter->compare(SketchPlugin_Sketch::ID()) == 0)
+      if (aFGrIter->compare(ModelAPI_ResultConstruction::group()) == 0 ||
+          aFGrIter->compare(ModelAPI_Feature::group()) == 0)
         break;
-    
-    if (aFGrIter != aFeatureGroups.end())
-    {
+
+    if (aFGrIter != aFeatureGroups.end()) {
       std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter = myGroups.begin();
       std::vector<SketchSolver_ConstraintGroup*> aSeparatedGroups;
-      while (aGroupIter != myGroups.end())
-      {
-        if (!(*aGroupIter)->isWorkplaneValid())
-        { // the group should be removed
+      while (aGroupIter != myGroups.end()) {
+        if (!(*aGroupIter)->isWorkplaneValid()) {  // the group should be removed
           delete *aGroupIter;
           int aShift = aGroupIter - myGroups.begin();
           myGroups.erase(aGroupIter);
           aGroupIter = myGroups.begin() + aShift;
           continue;
         }
-        if ((*aGroupIter)->updateGroup())
-        { // some constraints were removed, try to split the group
+        if ((*aGroupIter)->updateGroup()) {  // some constraints were removed, try to split the group
           (*aGroupIter)->splitGroup(aSeparatedGroups);
         }
         aGroupIter++;
@@ -147,29 +144,27 @@ void SketchSolver_ConstraintManager::processEvent(const Events_Message* theMessa
 
 // ============================================================================
 //  Function: changeWorkplane
-//  Class:    SketchSolver_PluginManager
+//  Class:    SketchSolver_Session
 //  Purpose:  update workplane by given parameters of the sketch
 // ============================================================================
-bool SketchSolver_ConstraintManager::changeWorkplane(boost::shared_ptr<SketchPlugin_Feature> theSketch)
+bool SketchSolver_ConstraintManager::changeWorkplane(
+    std::shared_ptr<ModelAPI_CompositeFeature> theSketch)
 {
-  bool aResult = true; // changed when a workplane wrongly updated
+  bool aResult = true;  // changed when a workplane wrongly updated
   bool isUpdated = false;
   // Try to update specified workplane in all groups
   std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter;
   for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
-    if ((*aGroupIter)->isBaseWorkplane(theSketch))
-    {
+    if ((*aGroupIter)->isBaseWorkplane(theSketch)) {
       isUpdated = true;
       if (!(*aGroupIter)->updateWorkplane())
         aResult = false;
     }
   // If the workplane is not updated, so this is a new workplane
-  if (!isUpdated)
-  {
+  if (!isUpdated) {
     SketchSolver_ConstraintGroup* aNewGroup = new SketchSolver_ConstraintGroup(theSketch);
     // Verify that the group is created successfully
-    if (!aNewGroup->isBaseWorkplane(theSketch))
-    {
+    if (!aNewGroup->isBaseWorkplane(theSketch)) {
       delete aNewGroup;
       return false;
     }
@@ -179,41 +174,46 @@ bool SketchSolver_ConstraintManager::changeWorkplane(boost::shared_ptr<SketchPlu
 }
 
 // ============================================================================
-//  Function: changeConstraint
-//  Class:    SketchSolver_PluginManager
-//  Purpose:  create/update the constraint and place it into appropriate group
+//  Function: changeConstraintOrEntity
+//  Class:    SketchSolver_Session
+//  Purpose:  create/update the constraint or the feature and place it into appropriate group
 // ============================================================================
-bool SketchSolver_ConstraintManager::changeConstraint(
-              boost::shared_ptr<SketchPlugin_Constraint> theConstraint)
+bool SketchSolver_ConstraintManager::changeConstraintOrEntity(
+    std::shared_ptr<SketchPlugin_Feature> theFeature)
 {
-  // Search the groups which this constraint touches
+  // Search the groups which this feature touches
   std::set<Slvs_hGroup> aGroups;
-  findGroups(theConstraint, aGroups);
+  findGroups(theFeature, aGroups);
+
+  std::shared_ptr<SketchPlugin_Constraint> aConstraint = 
+      std::dynamic_pointer_cast<SketchPlugin_Constraint>(theFeature);
 
   // Process the groups list
-  if (aGroups.size() == 0)
-  { // There are no groups applicable for this constraint => create new one
-    boost::shared_ptr<SketchPlugin_Feature> aWP = findWorkplaneForConstraint(theConstraint);
-    if (!aWP) return false;
+  if (aGroups.size() == 0) {
+    // There are no groups applicable for this constraint => create new one
+    // The group will be created only for constraints, not for features
+    if (!aConstraint) return false;
+    std::shared_ptr<ModelAPI_CompositeFeature> aWP = findWorkplane(aConstraint);
+    if (!aWP)
+      return false;
     SketchSolver_ConstraintGroup* aGroup = new SketchSolver_ConstraintGroup(aWP);
-    if (!aGroup->changeConstraint(theConstraint))
-    {
+    if (!aGroup->changeConstraint(aConstraint)) {
       delete aGroup;
       return false;
     }
     myGroups.push_back(aGroup);
     return true;
-  }
-  else if (aGroups.size() == 1)
-  { // Only one group => add constraint into it
+  } else if (aGroups.size() == 1) {  // Only one group => add feature into it
     Slvs_hGroup aGroupId = *(aGroups.begin());
     std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter;
     for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
-      if ((*aGroupIter)->getId() == aGroupId)
-        return (*aGroupIter)->changeConstraint(theConstraint);
-  }
-  else if (aGroups.size() > 1)
-  { // Several groups applicable for this constraint => need to merge them
+      if ((*aGroupIter)->getId() == aGroupId) {
+        // If the group is empty, the feature is not added (the constraint only)
+        if (!aConstraint && !(*aGroupIter)->isEmpty())
+          return (*aGroupIter)->changeEntity(theFeature) != SLVS_E_UNKNOWN;
+        return (*aGroupIter)->changeConstraint(aConstraint);
+      }
+  } else if (aGroups.size() > 1) {  // Several groups applicable for this feature => need to merge them
     std::set<Slvs_hGroup>::const_iterator aGroupsIter = aGroups.begin();
 
     // Search first group
@@ -226,13 +226,11 @@ bool SketchSolver_ConstraintManager::changeConstraint(
 
     // Append other groups to the first one
     std::vector<SketchSolver_ConstraintGroup*>::iterator anOtherGroupIter = aFirstGroupIter + 1;
-    for (aGroupsIter++; aGroupsIter != aGroups.end(); aGroupsIter++)
-    {
-      for ( ; anOtherGroupIter != myGroups.end(); anOtherGroupIter++)
+    for (aGroupsIter++; aGroupsIter != aGroups.end(); aGroupsIter++) {
+      for (; anOtherGroupIter != myGroups.end(); anOtherGroupIter++)
         if ((*anOtherGroupIter)->getId() == *aGroupsIter)
           break;
-      if (anOtherGroupIter == myGroups.end())
-      { // Group disappears
+      if (anOtherGroupIter == myGroups.end()) {  // Group disappears
         anOtherGroupIter = aFirstGroupIter + 1;
         continue;
       }
@@ -242,11 +240,13 @@ bool SketchSolver_ConstraintManager::changeConstraint(
       int aShiftOther = anOtherGroupIter - myGroups.begin();
       delete *anOtherGroupIter;
       myGroups.erase(anOtherGroupIter);
-      aFirstGroupIter  = myGroups.begin() + aShiftFirst;
+      aFirstGroupIter = myGroups.begin() + aShiftFirst;
       anOtherGroupIter = myGroups.begin() + aShiftOther;
     }
 
-    return (*aFirstGroupIter)->changeConstraint(theConstraint);
+    if (aConstraint)
+      return (*aFirstGroupIter)->changeConstraint(aConstraint);
+    return (*aFirstGroupIter)->changeEntity(theFeature) != SLVS_E_UNKNOWN;
   }
 
   // Something goes wrong
@@ -255,10 +255,11 @@ bool SketchSolver_ConstraintManager::changeConstraint(
 
 // ============================================================================
 //  Function: updateEntity
-//  Class:    SketchSolver_PluginManager
+//  Class:    SketchSolver_Session
 //  Purpose:  update any element on the sketch, which is used by constraints
 // ============================================================================
-void SketchSolver_ConstraintManager::updateEntity(boost::shared_ptr<SketchPlugin_Feature> theFeature)
+void SketchSolver_ConstraintManager::updateEntity(
+    std::shared_ptr<SketchPlugin_Feature> theFeature)
 {
   // Create list of attributes depending on type of the feature
   std::vector<std::string> anAttrList;
@@ -267,20 +268,17 @@ void SketchSolver_ConstraintManager::updateEntity(boost::shared_ptr<SketchPlugin
   if (aFeatureKind.compare(SketchPlugin_Point::ID()) == 0)
     anAttrList.push_back(SketchPlugin_Point::COORD_ID());
   // Line
-  else if (aFeatureKind.compare(SketchPlugin_Line::ID()) == 0)
-  {
+  else if (aFeatureKind.compare(SketchPlugin_Line::ID()) == 0) {
     anAttrList.push_back(SketchPlugin_Line::START_ID());
     anAttrList.push_back(SketchPlugin_Line::END_ID());
   }
   // Circle
-  else if (aFeatureKind.compare(SketchPlugin_Circle::ID()) == 0)
-  {
+  else if (aFeatureKind.compare(SketchPlugin_Circle::ID()) == 0) {
     anAttrList.push_back(SketchPlugin_Circle::CENTER_ID());
     anAttrList.push_back(SketchPlugin_Circle::RADIUS_ID());
   }
   // Arc
-  else if (aFeatureKind.compare(SketchPlugin_Arc::ID()) == 0)
-  {
+  else if (aFeatureKind.compare(SketchPlugin_Arc::ID()) == 0) {
     anAttrList.push_back(SketchPlugin_Arc::CENTER_ID());
     anAttrList.push_back(SketchPlugin_Arc::START_ID());
     anAttrList.push_back(SketchPlugin_Arc::END_ID());
@@ -289,15 +287,13 @@ void SketchSolver_ConstraintManager::updateEntity(boost::shared_ptr<SketchPlugin
 
   // Check changing of feature's attributes (go through the groups and search usage of the attributes)
   std::vector<std::string>::const_iterator anAttrIter;
-  for (anAttrIter = anAttrList.begin(); anAttrIter != anAttrList.end(); anAttrIter++)
-  {
+  for (anAttrIter = anAttrList.begin(); anAttrIter != anAttrList.end(); anAttrIter++) {
     std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter;
-    for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
-    {
-      if ((*aGroupIter)->isEmpty()) 
+    for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++) {
+      if ((*aGroupIter)->isEmpty())
         continue;
-      boost::shared_ptr<ModelAPI_Attribute> anAttribute =
-        boost::dynamic_pointer_cast<ModelAPI_Attribute>(theFeature->data()->attribute(*anAttrIter));
+      std::shared_ptr<ModelAPI_Attribute> anAttribute = std::dynamic_pointer_cast<
+          ModelAPI_Attribute>(theFeature->data()->attribute(*anAttrIter));
       (*aGroupIter)->updateEntityIfPossible(anAttribute);
     }
   }
@@ -308,23 +304,21 @@ void SketchSolver_ConstraintManager::updateEntity(boost::shared_ptr<SketchPlugin
       (*aGroupIter)->updateRelatedConstraints(theFeature);
 }
 
-
 // ============================================================================
 //  Function: findGroups
-//  Class:    SketchSolver_PluginManager
-//  Purpose:  search groups of entities interacting with given constraint
+//  Class:    SketchSolver_Session
+//  Purpose:  search groups of entities interacting with given feature
 // ============================================================================
 void SketchSolver_ConstraintManager::findGroups(
-              boost::shared_ptr<SketchPlugin_Constraint> theConstraint,
-              std::set<Slvs_hGroup>&                     theGroupIDs) const
+    std::shared_ptr<SketchPlugin_Feature> theFeature,
+    std::set<Slvs_hGroup>& theGroupIDs) const
 {
-  boost::shared_ptr<SketchPlugin_Feature> aWP = findWorkplaneForConstraint(theConstraint);
+  std::shared_ptr<ModelAPI_CompositeFeature> aWP = findWorkplane(theFeature);
 
-  SketchSolver_ConstraintGroup* anEmptyGroup = 0; // appropriate empty group for specified constraint
+  SketchSolver_ConstraintGroup* anEmptyGroup = 0;  // appropriate empty group for specified constraint
   std::vector<SketchSolver_ConstraintGroup*>::const_iterator aGroupIter;
   for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
-    if (aWP == (*aGroupIter)->getWorkplane() && (*aGroupIter)->isInteract(theConstraint))
-    {
+    if (aWP == (*aGroupIter)->getWorkplane() && (*aGroupIter)->isInteract(theFeature)) {
       if (!(*aGroupIter)->isEmpty())
         theGroupIDs.insert((*aGroupIter)->getId());
       else if (!anEmptyGroup)
@@ -337,48 +331,52 @@ void SketchSolver_ConstraintManager::findGroups(
 }
 
 // ============================================================================
-//  Function: findWorkplaneForConstraint
-//  Class:    SketchSolver_PluginManager
-//  Purpose:  search workplane containing given constraint
+//  Function: findWorkplane
+//  Class:    SketchSolver_Session
+//  Purpose:  search workplane containing given feature
 // ============================================================================
-boost::shared_ptr<SketchPlugin_Feature> SketchSolver_ConstraintManager::findWorkplaneForConstraint(
-              boost::shared_ptr<SketchPlugin_Constraint> theConstraint) const
+std::shared_ptr<ModelAPI_CompositeFeature> SketchSolver_ConstraintManager::findWorkplane(
+    std::shared_ptr<SketchPlugin_Feature> theFeature) const
 {
   // Already verified workplanes
-  std::set< boost::shared_ptr<SketchPlugin_Feature> > aVerified;
+  std::set<std::shared_ptr<ModelAPI_CompositeFeature> > aVerified;
 
   std::vector<SketchSolver_ConstraintGroup*>::const_iterator aGroupIter;
-  for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
-  {
-    boost::shared_ptr<SketchPlugin_Feature> aWP = (*aGroupIter)->getWorkplane();
+  for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++) {
+    std::shared_ptr<ModelAPI_CompositeFeature> aWP = (*aGroupIter)->getWorkplane();
     if (aVerified.find(aWP) != aVerified.end())
       continue;
 
-    boost::shared_ptr<ModelAPI_AttributeRefList> aWPFeatures =
-      boost::dynamic_pointer_cast<ModelAPI_AttributeRefList>(aWP->data()->attribute(SketchPlugin_Sketch::FEATURES_ID()));
+    std::shared_ptr<ModelAPI_AttributeRefList> aWPFeatures = std::dynamic_pointer_cast<
+        ModelAPI_AttributeRefList>(aWP->data()->attribute(SketchPlugin_Sketch::FEATURES_ID()));
     std::list<ObjectPtr> aFeaturesList = aWPFeatures->list();
     std::list<ObjectPtr>::const_iterator anIter;
     for (anIter = aFeaturesList.begin(); anIter != aFeaturesList.end(); anIter++)
-      if (*anIter == theConstraint)
-        return aWP; // workplane is found
+      if (*anIter == theFeature)
+        return aWP;  // workplane is found
     aVerified.insert(aWP);
   }
 
-  return boost::shared_ptr<SketchPlugin_Feature>();
+  return std::shared_ptr<ModelAPI_CompositeFeature>();
 }
 
 // ============================================================================
 //  Function: resolveConstraints
-//  Class:    SketchSolver_PluginManager
+//  Class:    SketchSolver_Session
 //  Purpose:  change entities according to available constraints
 // ============================================================================
 void SketchSolver_ConstraintManager::resolveConstraints()
 {
+  myIsComputed = true;
+  bool needToUpdate = false;
   std::vector<SketchSolver_ConstraintGroup*>::iterator aGroupIter;
   for (aGroupIter = myGroups.begin(); aGroupIter != myGroups.end(); aGroupIter++)
-    (*aGroupIter)->resolveConstraints();
+    if ((*aGroupIter)->resolveConstraints())
+      needToUpdate = true;
 
   // Features may be updated => send events
-  Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
+  if (needToUpdate)
+    Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
+  myIsComputed = false;
 }