src/SketchSolver/SketchSolver_Group.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:    SketchSolver_Group.cpp
   4 // Created: 27 May 2014
   5 // Author:  Artem ZHIDKOV
   6
   7 #include "SketchSolver_Group.h"
   8
   9 #include <SketchSolver_Constraint.h>
  10 #include <SketchSolver_ConstraintCoincidence.h>
  11 #include <SketchSolver_ConstraintMulti.h>
  12 #include <SketchSolver_Error.h>
  13 #include <SketchSolver_Manager.h>
  14
  15 #include <Events_Error.h>
  16 #include <Events_Loop.h>
  17 #include <ModelAPI_AttributeString.h>
  18 #include <ModelAPI_Events.h>
  19 #include <ModelAPI_Session.h>
  20 #include <ModelAPI_Validator.h>
  21
  22 #include <SketchPlugin_Arc.h>
  23 #include <SketchPlugin_ConstraintAngle.h>
  24 #include <SketchPlugin_ConstraintCoincidence.h>
  25 #include <SketchPlugin_ConstraintDistance.h>
  26 #include <SketchPlugin_ConstraintEqual.h>
  27 #include <SketchPlugin_ConstraintHorizontal.h>
  28 #include <SketchPlugin_ConstraintLength.h>
  29 #include <SketchPlugin_ConstraintFillet.h>
  30 #include <SketchPlugin_ConstraintMirror.h>
  31 #include <SketchPlugin_ConstraintParallel.h>
  32 #include <SketchPlugin_ConstraintPerpendicular.h>
  33 #include <SketchPlugin_ConstraintRadius.h>
  34 #include <SketchPlugin_ConstraintRigid.h>
  35 #include <SketchPlugin_ConstraintTangent.h>
  36 #include <SketchPlugin_ConstraintVertical.h>
  37 #include <SketchPlugin_MultiRotation.h>
  38 #include <SketchPlugin_MultiTranslation.h>
  39
  40 #include <math.h>
  41 #include <assert.h>
  42
  43
  44 /// \brief This class is used to give unique index to the groups
  45 class GroupIndexer
  46 {
  47 public:
  48   /// \brief Return vacant index
  49   static GroupID NEW_GROUP() { return ++myGroupIndex; }
  50   /// \brief Removes the index
  51   static void REMOVE_GROUP(const GroupID& theIndex) {
  52     if (myGroupIndex == theIndex)
  53       myGroupIndex--;
  54   }
  55
  56 private:
  57   GroupIndexer() {};
  58
  59   static GroupID myGroupIndex; ///< index of the group
  60 };
  61
  62 GroupID GroupIndexer::myGroupIndex = GID_OUTOFGROUP;
  63
  64
  65 static void sendMessage(const char* theMessageName)
  66 {
  67   std::shared_ptr<Events_Message> aMessage = std::shared_ptr<Events_Message>(
  68       new Events_Message(Events_Loop::eventByName(theMessageName)));
  69   Events_Loop::loop()->send(aMessage);
  70 }
  71
  72 static void sendMessage(const char* theMessageName, const std::set<ObjectPtr>& theConflicting)
  73 {
  74   std::shared_ptr<ModelAPI_SolverFailedMessage> aMessage =
  75       std::shared_ptr<ModelAPI_SolverFailedMessage>(
  76       new ModelAPI_SolverFailedMessage(Events_Loop::eventByName(theMessageName)));
  77   aMessage->setObjects(theConflicting);
  78   Events_Loop::loop()->send(aMessage);
  79 }
  80
  81
  82
  83 // ========================================================
  84 // =========  SketchSolver_Group  ===============
  85 // ========================================================
  86
  87 SketchSolver_Group::SketchSolver_Group(
  88     std::shared_ptr<ModelAPI_CompositeFeature> theWorkplane)
  89     : myID(GroupIndexer::NEW_GROUP()),
  90       myPrevResult(STATUS_UNKNOWN)
  91 {
  92   // Initialize workplane
  93   myWorkplaneID = EID_UNKNOWN;
  94   addWorkplane(theWorkplane);
  95 }
  96
  97 SketchSolver_Group::~SketchSolver_Group()
  98 {
  99   myConstraints.clear();
 100   GroupIndexer::REMOVE_GROUP(myID);
 101   // send the message that there is no more conflicting constraints
 102   if (!myConflictingConstraints.empty()) {
 103     sendMessage(EVENT_SOLVER_REPAIRED, myConflictingConstraints);
 104     myConflictingConstraints.clear();
 105   }
 106 }
 107
 108 // ============================================================================
 109 //  Function: isBaseWorkplane
 110 //  Class:    SketchSolver_Group
 111 //  Purpose:  verify the group is based on the given workplane
 112 // ============================================================================
 113 bool SketchSolver_Group::isBaseWorkplane(CompositeFeaturePtr theWorkplane) const
 114 {
 115   return theWorkplane == mySketch;
 116 }
 117
 118 // ============================================================================
 119 //  Function: isInteract
 120 //  Class:    SketchSolver_Group
 121 //  Purpose:  verify are there any entities in the group used by given constraint
 122 // ============================================================================
 123 bool SketchSolver_Group::isInteract(FeaturePtr theFeature) const
 124 {
 125   // Empty group interacts with everything
 126   if (isEmpty())
 127     return true;
 128   // Check interaction with the storage
 129   bool isInteracted = myStorage->isInteract(theFeature);
 130   ConstraintConstraintMap::const_iterator anIt = myConstraints.begin();
 131   for (; !isInteracted && anIt != myConstraints.end(); ++anIt)
 132     if (anIt->first->getKind() == SketchPlugin_MultiRotation::ID() ||
 133         anIt->first->getKind() == SketchPlugin_MultiTranslation::ID())
 134       isInteracted = anIt->second->isUsed(theFeature);
 135   return isInteracted;
 136 }
 137
 138 // ============================================================================
 139 //  Function: changeConstraint
 140 //  Class:    SketchSolver_Group
 141 //  Purpose:  create/update the constraint in the group
 142 // ============================================================================
 143 bool SketchSolver_Group::changeConstraint(
 144     std::shared_ptr<SketchPlugin_Constraint> theConstraint)
 145 {
 146   // There is no workplane yet, something wrong
 147   if (myWorkplaneID == EID_UNKNOWN)
 148     return false;
 149
 150   if (!theConstraint || !theConstraint->data())
 151     return false;
 152
 153   if (!checkFeatureValidity(theConstraint))
 154     return false;
 155
 156   BuilderPtr aBuilder = SketchSolver_Manager::instance()->builder();
 157   myStorage->blockEvents(true);
 158
 159   bool isNewConstraint = myConstraints.find(theConstraint) == myConstraints.end();
 160   if (isNewConstraint) {
 161     // Add constraint to the current group
 162     SolverConstraintPtr aConstraint = aBuilder->createConstraint(theConstraint);
 163     if (!aConstraint)
 164       return false;
 165     aConstraint->process(myStorage, getId(), getWorkplaneId());
 166     if (!aConstraint->error().empty()) {
 167       if (aConstraint->error() == SketchSolver_Error::NOT_INITIALIZED())
 168         return false; // some attribute are not initialized yet, don't show message
 169       Events_Error::send(aConstraint->error(), this);
 170     }
 171     myConstraints[theConstraint] = aConstraint;
 172
 173     if (theConstraint->getKind() == SketchPlugin_ConstraintCoincidence::ID())
 174       notifyCoincidenceChanged(myConstraints[theConstraint]);
 175   }
 176   else
 177     myConstraints[theConstraint]->update();
 178
 179   // Fix mirror line
 180   if (theConstraint->getKind() == SketchPlugin_ConstraintMirror::ID()) {
 181     AttributeRefAttrPtr aRefAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
 182         theConstraint->attribute(SketchPlugin_ConstraintMirror::ENTITY_A()));
 183     if (aRefAttr && aRefAttr->isObject()) {
 184       std::shared_ptr<SketchPlugin_Feature> aFeature =
 185           std::dynamic_pointer_cast<SketchPlugin_Feature>(
 186           ModelAPI_Feature::feature(aRefAttr->object()));
 187       if (aFeature) {
 188         SolverConstraintPtr aConstraint = aBuilder->createFixedConstraint(aFeature);
 189         if (aConstraint) {
 190           aConstraint->process(myStorage, getId(), getWorkplaneId());
 191           setTemporary(aConstraint);
 192         }
 193       }
 194     }
 195   }
 196   return true;
 197 }
 198
 199 // Update constraints if they contain specific feature
 200 static void updateMultiConstraints(ConstraintConstraintMap& theConstraints, FeaturePtr theFeature)
 201 {
 202   ConstraintConstraintMap::iterator aCIt = theConstraints.begin();
 203   for (; aCIt != theConstraints.end(); ++aCIt) {
 204     SketchSolver_ConstraintType aType = aCIt->second->getType();
 205     if ((aType == CONSTRAINT_MULTI_ROTATION ||
 206          aType == CONSTRAINT_MULTI_TRANSLATION)
 207         && aCIt->second->isUsed(theFeature))
 208       std::dynamic_pointer_cast<SketchSolver_ConstraintMulti>(aCIt->second)->update(true);
 209     else if ((aType == CONSTRAINT_TANGENT_CIRCLE_LINE ||
 210               aType == CONSTRAINT_SYMMETRIC || aType == CONSTRAINT_ANGLE)
 211              && aCIt->second->isUsed(theFeature))
 212       aCIt->second->update();
 213   }
 214 }
 215
 216 // Recalculate slave features of the Multi constraints
 217 static void updateMultiConstraints(ConstraintConstraintMap& theConstraints)
 218 {
 219   ConstraintConstraintMap::iterator aCIt = theConstraints.begin();
 220   for (; aCIt != theConstraints.end(); ++aCIt) {
 221     SketchSolver_ConstraintType aType = aCIt->second->getType();
 222     if ((aType == CONSTRAINT_MULTI_ROTATION ||
 223          aType == CONSTRAINT_MULTI_TRANSLATION))
 224       std::dynamic_pointer_cast<SketchSolver_ConstraintMulti>(aCIt->second)->update(true);
 225   }
 226 }
 227
 228 bool SketchSolver_Group::updateFeature(FeaturePtr theFeature)
 229 {
 230   if (!checkFeatureValidity(theFeature))
 231     return false;
 232
 233   bool isBlocked = myStorage->isEventsBlocked();
 234   if (!isBlocked)
 235     myStorage->blockEvents(true);
 236
 237   myStorage->refresh(true);
 238   bool isUpdated = myStorage->update(theFeature);
 239
 240   updateMultiConstraints(myConstraints, theFeature);
 241
 242   // events were not blocked before, the feature has not been updated,
 243   // so it is necessary to revert blocking
 244   if (!isUpdated && !isBlocked)
 245     myStorage->blockEvents(false);
 246   return isUpdated;
 247 }
 248
 249 void SketchSolver_Group::moveFeature(FeaturePtr theFeature)
 250 {
 251   BuilderPtr aBuilder = SketchSolver_Manager::instance()->builder();
 252
 253   // Firstly, revert changes in the fixed entities
 254   myStorage->blockEvents(true);
 255   myStorage->refresh(true);
 256
 257   // Then, create temporary Fixed constraint
 258   SolverConstraintPtr aConstraint = aBuilder->createMovementConstraint(theFeature);
 259   if (!aConstraint)
 260     return;
 261   aConstraint->process(myStorage, getId(), getWorkplaneId());
 262   if (aConstraint->error().empty())
 263     setTemporary(aConstraint);
 264
 265   // Secondly, search attributes of the feature in the list of the Multi constraints and update them
 266   updateMultiConstraints(myConstraints, theFeature);
 267
 268   // Workaround to process arcs.
 269   // When move unconstrained arc, add temporary constraint to fix radius.
 270   if (theFeature->getKind() == SketchPlugin_Arc::ID()) {
 271     bool hasDup = myStorage->hasDuplicatedConstraint();
 272     SolverConstraintPtr aFixedRadius = aBuilder->createFixedArcRadiusConstraint(theFeature);
 273     if (aFixedRadius) {
 274       aFixedRadius->process(myStorage, getId(), getWorkplaneId());
 275       hasDup = myStorage->hasDuplicatedConstraint() && !hasDup;
 276       if (aFixedRadius->error().empty() && !hasDup)
 277         setTemporary(aFixedRadius);
 278       else
 279         aFixedRadius->remove();
 280     }
 281   }
 282 }
 283
 284 // ============================================================================
 285 //  Function: addWorkplane
 286 //  Class:    SketchSolver_Group
 287 //  Purpose:  create workplane for the group
 288 // ============================================================================
 289 bool SketchSolver_Group::addWorkplane(CompositeFeaturePtr theSketch)
 290 {
 291   if (myWorkplaneID != EID_UNKNOWN || theSketch->getKind() != SketchPlugin_Sketch::ID())
 292     return false;  // the workplane already exists or the function parameter is not Sketch
 293
 294   mySketch = theSketch;
 295   if (!updateWorkplane()) {
 296     mySketch = CompositeFeaturePtr();
 297     return false;
 298   }
 299   return true;
 300 }
 301
 302 // ============================================================================
 303 //  Function: updateWorkplane
 304 //  Class:    SketchSolver_Group
 305 //  Purpose:  update parameters of workplane
 306 // ============================================================================
 307 bool SketchSolver_Group::updateWorkplane()
 308 {
 309   BuilderPtr aBuilder = SketchSolver_Manager::instance()->builder();
 310   if (!myStorage) // Create storage if not exists
 311     myStorage = aBuilder->createStorage(getId());
 312
 313   // sketch should be unchanged, set it out of current group
 314   bool isUpdated = myStorage->update(FeaturePtr(mySketch), GID_OUTOFGROUP);
 315   if (isUpdated) {
 316     EntityWrapperPtr anEntity = myStorage->entity(FeaturePtr(mySketch));
 317     myWorkplaneID = anEntity->id();
 318   }
 319   return isUpdated;
 320 }
 321
 322 // ============================================================================
 323 //  Function: resolveConstraints
 324 //  Class:    SketchSolver_Group
 325 //  Purpose:  solve the set of constraints for the current group
 326 // ============================================================================
 327 bool SketchSolver_Group::resolveConstraints()
 328 {
 329   bool aResolved = false;
 330   bool isGroupEmpty = isEmpty() && myStorage->isEmpty();
 331   if (myStorage->isNeedToResolve() && !isGroupEmpty) {
 332     if (!mySketchSolver)
 333       mySketchSolver = SketchSolver_Manager::instance()->builder()->createSolver();
 334
 335     mySketchSolver->setGroup(myID);
 336     mySketchSolver->calculateFailedConstraints(false);
 337     myStorage->initializeSolver(mySketchSolver);
 338     mySketchSolver->prepare();
 339
 340     SketchSolver_SolveStatus aResult = STATUS_OK;
 341     try {
 342       if (myStorage->hasDuplicatedConstraint())
 343         aResult = STATUS_INCONSISTENT;
 344       else {
 345         // To avoid overconstraint situation, we will remove temporary constraints one-by-one
 346         // and try to find the case without overconstraint
 347         bool isLastChance = false;
 348         while (true) {
 349           aResult = mySketchSolver->solve();
 350           if (aResult == STATUS_OK || aResult == STATUS_EMPTYSET || isLastChance)
 351             break;
 352 ////          // try to update parameters and resolve once again
 353 ////          ConstraintConstraintMap::iterator aConstrIt = myConstraints.begin();
 354 ////          for (; aConstrIt != myConstraints.end(); ++aConstrIt)
 355 ////            aConstrIt->second->update();
 356           isLastChance = true;
 357
 358           removeTemporaryConstraints();
 359           mySketchSolver->calculateFailedConstraints(true); // something failed => need to find it
 360           myStorage->initializeSolver(mySketchSolver);
 361         }
 362       }
 363     } catch (...) {
 364 //      Events_Error::send(SketchSolver_Error::SOLVESPACE_CRASH(), this);
 365       getWorkplane()->string(SketchPlugin_Sketch::SOLVER_ERROR())->setValue(SketchSolver_Error::SOLVESPACE_CRASH());
 366       Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 367       if (myPrevResult == STATUS_OK || myPrevResult == STATUS_UNKNOWN) {
 368         // the error message should be changed before sending the message
 369         sendMessage(EVENT_SOLVER_FAILED);
 370         myPrevResult = STATUS_FAILED;
 371       }
 372       mySketchSolver->undo();
 373       return false;
 374     }
 375     if (aResult == STATUS_OK || aResult == STATUS_EMPTYSET) {  // solution succeeded, store results into correspondent attributes
 376       myStorage->refresh();
 377       updateMultiConstraints(myConstraints);
 378       if (myPrevResult != STATUS_OK || myPrevResult == STATUS_UNKNOWN) {
 379         getWorkplane()->string(SketchPlugin_Sketch::SOLVER_ERROR())->setValue("");
 380         Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 381         // the error message should be changed before sending the message
 382         sendMessage(EVENT_SOLVER_REPAIRED, myConflictingConstraints);
 383         myConflictingConstraints.clear();
 384         myPrevResult = STATUS_OK;
 385       }
 386     } else {
 387       mySketchSolver->undo();
 388       if (!myConstraints.empty()) {
 389         // the error message should be changed before sending the message
 390         getWorkplane()->string(SketchPlugin_Sketch::SOLVER_ERROR())->setValue(SketchSolver_Error::CONSTRAINTS());
 391         Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 392         if (myPrevResult != aResult || myPrevResult == STATUS_UNKNOWN) {
 393           // Obtain list of conflicting constraints
 394           std::set<ObjectPtr> aConflicting = myStorage->getConflictingConstraints(mySketchSolver);
 395
 396           if (myConflictingConstraints.empty())
 397             sendMessage(EVENT_SOLVER_FAILED, aConflicting);
 398           else {
 399             std::set<ObjectPtr>::iterator anIt = aConflicting.begin();
 400             for (; anIt != aConflicting.end(); ++anIt)
 401               myConflictingConstraints.erase(*anIt);
 402             if (!myConflictingConstraints.empty()) {
 403               // some constraints does not conflict, send corresponding message
 404               sendMessage(EVENT_SOLVER_REPAIRED, myConflictingConstraints);
 405             }
 406           }
 407           myConflictingConstraints = aConflicting;
 408           myPrevResult = aResult;
 409         }
 410       }
 411     }
 412
 413     aResolved = true;
 414   } else if (!isGroupEmpty) {
 415     // Check there are constraints Fixed. If they exist, update parameters by stored values
 416     ConstraintConstraintMap::iterator aCIt = myConstraints.begin();
 417     for (; aCIt != myConstraints.end(); ++aCIt)
 418       if (aCIt->first->getKind() == SketchPlugin_ConstraintRigid::ID()) {
 419         aResolved = true;
 420         break;
 421       }
 422     if (aCIt != myConstraints.end())
 423       myStorage->refresh();
 424   }
 425   removeTemporaryConstraints();
 426   myStorage->blockEvents(false);
 427   myStorage->setNeedToResolve(false);
 428   return aResolved;
 429 }
 430
 431 // ============================================================================
 432 //  Function: mergeGroups
 433 //  Class:    SketchSolver_Group
 434 //  Purpose:  append specified group to the current group
 435 // ============================================================================
 436 void SketchSolver_Group::mergeGroups(const SketchSolver_Group& theGroup)
 437 {
 438   // If specified group is empty, no need to merge
 439   if (theGroup.isEmpty())
 440     return;
 441
 442   std::set<ObjectPtr> aConstraints;
 443   ConstraintConstraintMap::const_iterator aConstrIter = theGroup.myConstraints.begin();
 444   for (; aConstrIter != theGroup.myConstraints.end(); aConstrIter++)
 445     aConstraints.insert(aConstrIter->first);
 446
 447   std::list<FeaturePtr> aSortedConstraints = selectApplicableFeatures(aConstraints);
 448   std::list<FeaturePtr>::iterator aSCIter = aSortedConstraints.begin();
 449   for (; aSCIter != aSortedConstraints.end(); ++aSCIter) {
 450     ConstraintPtr aConstr = std::dynamic_pointer_cast<SketchPlugin_Constraint>(*aSCIter);
 451     if (!aConstr)
 452       continue;
 453     changeConstraint(aConstr);
 454   }
 455 }
 456
 457 // ============================================================================
 458 //  Function: splitGroup
 459 //  Class:    SketchSolver_Group
 460 //  Purpose:  divide the group into several subgroups
 461 // ============================================================================
 462 void SketchSolver_Group::splitGroup(std::list<SketchSolver_Group*>& theCuts)
 463 {
 464   // New storage will be used in trimmed way to store the list of constraint interacted together.
 465   StoragePtr aNewStorage = SketchSolver_Manager::instance()->builder()->createStorage(getId());
 466   std::list<ConstraintWrapperPtr> aDummyVec; // empty vector to avoid creation of solver's constraints
 467
 468   // Obtain constraints, which should be separated
 469   std::list<ConstraintPtr> anUnusedConstraints;
 470   ConstraintConstraintMap::iterator aCIter = myConstraints.begin();
 471   for ( ; aCIter != myConstraints.end(); aCIter++) {
 472     if (aNewStorage->isInteract(FeaturePtr(aCIter->first)))
 473       aNewStorage->addConstraint(aCIter->first, aDummyVec);
 474     else
 475       anUnusedConstraints.push_back(aCIter->first);
 476   }
 477
 478   // Check the unused constraints once again, because they may become interacted with new storage since adding constraints
 479   std::list<ConstraintPtr>::iterator aUnuseIt = anUnusedConstraints.begin();
 480   while (aUnuseIt != anUnusedConstraints.end()) {
 481     if (aNewStorage->isInteract(FeaturePtr(*aUnuseIt))) {
 482       aNewStorage->addConstraint(*aUnuseIt, aDummyVec);
 483       anUnusedConstraints.erase(aUnuseIt);
 484       aUnuseIt = anUnusedConstraints.begin();
 485       continue;
 486     }
 487     aUnuseIt++;
 488   }
 489
 490   std::list<SketchSolver_Group*>::iterator aCutsIter;
 491   aUnuseIt = anUnusedConstraints.begin();
 492   for ( ; aUnuseIt != anUnusedConstraints.end(); ++aUnuseIt) {
 493     // Remove unused constraints
 494     removeConstraint(*aUnuseIt);
 495     // Try to append constraint to already existent group
 496     for (aCutsIter = theCuts.begin(); aCutsIter != theCuts.end(); ++aCutsIter)
 497       if ((*aCutsIter)->isInteract(*aUnuseIt)) {
 498         (*aCutsIter)->changeConstraint(*aUnuseIt);
 499         break;
 500       }
 501     if (aCutsIter == theCuts.end()) {
 502       // Add new group
 503       SketchSolver_Group* aGroup = new SketchSolver_Group(mySketch);
 504       aGroup->changeConstraint(*aUnuseIt);
 505       theCuts.push_back(aGroup);
 506     } else {
 507       // Find other groups interacting with constraint
 508       std::list<SketchSolver_Group*>::iterator aBaseGroupIt = aCutsIter;
 509       for (++aCutsIter; aCutsIter != theCuts.end(); ++aCutsIter)
 510         if ((*aCutsIter)->isInteract(*aUnuseIt)) {
 511           (*aBaseGroupIt)->mergeGroups(**aCutsIter);
 512           std::list<SketchSolver_Group*>::iterator aRemoveIt = aCutsIter--;
 513           theCuts.erase(aRemoveIt);
 514         }
 515     }
 516   }
 517 }
 518
 519 // ============================================================================
 520 //  Function: isConsistent
 521 //  Class:    SketchSolver_Group
 522 //  Purpose:  search removed entities and constraints
 523 // ============================================================================
 524 bool SketchSolver_Group::isConsistent()
 525 {
 526   if (isEmpty()) // no one constraint is initialized yet
 527     return true;
 528
 529   // Check the features and constraint is the storage are valid
 530   bool aResult = myStorage->isConsistent();
 531   if (aResult) {
 532     // additional check of consistency of the Fixed constraint,
 533     // because they are not added to the storage
 534     ConstraintConstraintMap::iterator aCIter = myConstraints.begin();
 535     for (; aCIter != myConstraints.end(); ++aCIter)
 536       if (aCIter->first->getKind() == SketchPlugin_ConstraintRigid::ID() &&
 537          (!aCIter->first->data() || !aCIter->first->data()->isValid())) {
 538         aResult = false;
 539         break;
 540       }
 541   }
 542   if (!aResult) {
 543     // remove invalid constraints
 544     std::set<ConstraintPtr> anInvalidConstraints;
 545     ConstraintConstraintMap::iterator aCIter = myConstraints.begin();
 546     for (; aCIter != myConstraints.end(); ++aCIter) {
 547       if (!aCIter->first->data() || !aCIter->first->data()->isValid())
 548         anInvalidConstraints.insert(aCIter->first);
 549     }
 550     std::set<ConstraintPtr>::const_iterator aRemoveIt = anInvalidConstraints.begin();
 551     for (; aRemoveIt != anInvalidConstraints.end(); ++aRemoveIt)
 552       removeConstraint(*aRemoveIt);
 553     // remove invalid features
 554     myStorage->removeInvalidEntities();
 555   }
 556   return aResult;
 557 }
 558
 559 // ============================================================================
 560 //  Function: removeTemporaryConstraints
 561 //  Class:    SketchSolver_Group
 562 //  Purpose:  remove all transient SLVS_C_WHERE_DRAGGED constraints after
 563 //            resolving the set of constraints
 564 // ============================================================================
 565 void SketchSolver_Group::removeTemporaryConstraints()
 566 {
 567   std::set<SolverConstraintPtr>::iterator aTmpIt = myTempConstraints.begin();
 568   for (; aTmpIt != myTempConstraints.end(); ++aTmpIt)
 569     (*aTmpIt)->remove();
 570
 571   if (!myTempConstraints.empty())
 572     myStorage->verifyFixed();
 573   myStorage->setNeedToResolve(false);
 574   myTempConstraints.clear();
 575 }
 576
 577 // ============================================================================
 578 //  Function: removeConstraint
 579 //  Class:    SketchSolver_Group
 580 //  Purpose:  remove constraint and all unused entities
 581 // ============================================================================
 582 void SketchSolver_Group::removeConstraint(ConstraintPtr theConstraint)
 583 {
 584   ConstraintConstraintMap::iterator aCIter = myConstraints.begin();
 585   for (; aCIter != myConstraints.end(); aCIter++)
 586     if (aCIter->first == theConstraint) {
 587       aCIter->second->remove(); // the constraint is not fully removed
 588       if (aCIter->first->getKind() == SketchPlugin_ConstraintCoincidence::ID())
 589         notifyCoincidenceChanged(aCIter->second);
 590       break;
 591     }
 592   if (aCIter != myConstraints.end())
 593     myConstraints.erase(aCIter);
 594 }
 595
 596 // ============================================================================
 597 //  Function: setTemporary
 598 //  Class:    SketchSolver_Group
 599 //  Purpose:  append given constraint to the group of temporary constraints
 600 // ============================================================================
 601 void SketchSolver_Group::setTemporary(SolverConstraintPtr theConstraint)
 602 {
 603   myTempConstraints.insert(theConstraint);
 604 }
 605
 606
 607 // ============================================================================
 608 //  Function: checkFeatureValidity
 609 //  Class:    SketchSolver_Group
 610 //  Purpose:  verifies is the feature valid
 611 // ============================================================================
 612 bool SketchSolver_Group::checkFeatureValidity(FeaturePtr theFeature)
 613 {
 614   if (!theFeature || !theFeature->data()->isValid())
 615     return true;
 616
 617   SessionPtr aMgr = ModelAPI_Session::get();
 618   ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
 619   return aFactory->validate(theFeature);
 620 }
 621
 622
 623
 624
 625 // ===========   Auxiliary functions   ========================================
 626 static double featureToVal(FeaturePtr theFeature)
 627 {
 628   if (theFeature->getKind() == SketchPlugin_Sketch::ID())
 629     return 0.0; // sketch
 630   ConstraintPtr aConstraint = std::dynamic_pointer_cast<SketchPlugin_Constraint>(theFeature);
 631   if (!aConstraint)
 632     return 1.0; // features (arc, circle, line, point)
 633
 634   const std::string& anID = aConstraint->getKind();
 635   if (anID == SketchPlugin_ConstraintCoincidence::ID()) {
 636     AttributeRefAttrPtr anAttrA = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
 637         aConstraint->attribute(SketchPlugin_Constraint::ENTITY_A()));
 638     AttributeRefAttrPtr anAttrB = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
 639         aConstraint->attribute(SketchPlugin_Constraint::ENTITY_B()));
 640     if (anAttrA && anAttrB && (anAttrA->isObject() || anAttrB->isObject()))
 641       return 2.0; // point-on-line and point-on-circle should go before points coincidence constraint
 642     return 2.5;
 643   }
 644   if (anID == SketchPlugin_ConstraintDistance::ID() ||
 645       anID == SketchPlugin_ConstraintLength::ID() ||
 646       anID == SketchPlugin_ConstraintRadius::ID())
 647     return 3.0;
 648   if (anID == SketchPlugin_ConstraintAngle::ID())
 649     return 3.5;
 650   if (anID == SketchPlugin_ConstraintHorizontal::ID() ||
 651       anID == SketchPlugin_ConstraintVertical::ID() ||
 652       anID == SketchPlugin_ConstraintParallel::ID() ||
 653       anID == SketchPlugin_ConstraintPerpendicular::ID())
 654     return 4.0;
 655   if (anID == SketchPlugin_ConstraintEqual::ID())
 656     return 5.0;
 657   if (anID == SketchPlugin_ConstraintTangent::ID() ||
 658       anID == SketchPlugin_ConstraintMirror::ID())
 659     return 6.0;
 660   if (anID == SketchPlugin_ConstraintRigid::ID())
 661     return 7.0;
 662   if (anID == SketchPlugin_MultiRotation::ID() ||
 663       anID == SketchPlugin_MultiTranslation::ID())
 664     return 8.0;
 665
 666   // all other constraints are placed between Equal and Tangent constraints
 667   return 5.5;
 668 }
 669
 670 static bool isLess(FeaturePtr theFeature1, FeaturePtr theFeature2)
 671 {
 672   return featureToVal(theFeature1) < featureToVal(theFeature2);
 673 }
 674
 675 std::list<FeaturePtr> SketchSolver_Group::selectApplicableFeatures(const std::set<ObjectPtr>& theObjects)
 676 {
 677   std::list<FeaturePtr> aResult;
 678   std::list<FeaturePtr>::iterator aResIt;
 679
 680   std::set<ObjectPtr>::const_iterator anObjIter = theObjects.begin();
 681   for (; anObjIter != theObjects.end(); ++anObjIter) {
 682     // Operate sketch itself and SketchPlugin features only.
 683     // Also, the Fillet need to be skipped, because there are several separated constraints composing it.
 684     FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(*anObjIter);
 685     if (!aFeature)
 686       continue;
 687     std::shared_ptr<SketchPlugin_Feature> aSketchFeature =
 688         std::dynamic_pointer_cast<SketchPlugin_Feature>(aFeature);
 689     if ((aFeature->getKind() != SketchPlugin_Sketch::ID() && !aSketchFeature) ||
 690         aFeature->getKind() == SketchPlugin_ConstraintFillet::ID())
 691       continue;
 692
 693     // Find the place where to insert a feature
 694     for (aResIt = aResult.begin(); aResIt != aResult.end(); ++aResIt)
 695       if (isLess(aFeature, *aResIt))
 696         break;
 697     aResult.insert(aResIt, aFeature);
 698   }
 699
 700   return aResult;
 701 }
 702
 703 void SketchSolver_Group::notifyCoincidenceChanged(SolverConstraintPtr theCoincidence)
 704 {
 705   const std::list<EntityWrapperPtr>& aCoincident = theCoincidence->attributes();
 706   EntityWrapperPtr anAttr1 = aCoincident.front();
 707   EntityWrapperPtr anAttr2 = aCoincident.back();
 708
 709   ConstraintConstraintMap::iterator anIt = myConstraints.begin();
 710   for (; anIt != myConstraints.end(); ++anIt)
 711     anIt->second->notifyCoincidenceChanged(anAttr1, anAttr2);
 712 }