+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
#include <SketchSolver_ConstraintCoincidence.h>
#include <SketchSolver_Error.h>
-#include <SketchSolver_Group.h>
-
-#include <map>
+#include <SketchSolver_Manager.h>
void SketchSolver_ConstraintCoincidence::getAttributes(
double& theValue,
- std::vector<Slvs_hEntity>& theAttributes)
+ std::vector<EntityWrapperPtr>& theAttributes)
{
SketchSolver_Constraint::getAttributes(theValue, theAttributes);
- if (!myErrorMsg.empty() || theAttributes[0] == SLVS_E_UNKNOWN)
+ if (!myErrorMsg.empty() || !theAttributes[0]) {
+ theAttributes.clear();
return;
+ }
- if (theAttributes[1] != SLVS_E_UNKNOWN)
- myType = SLVS_C_POINTS_COINCIDENT;
- else if (theAttributes[2] != SLVS_E_UNKNOWN) {
+ if (theAttributes[1]) {
+ myType = CONSTRAINT_PT_PT_COINCIDENT;
+ if (myStorage->isFixed(theAttributes[1]) && !myStorage->isFixed(theAttributes[0])) {
+ // fixed point should go first
+ EntityWrapperPtr aTemp = theAttributes[0];
+ theAttributes[0] = theAttributes[1];
+ theAttributes[1] = aTemp;
+ }
+ // Set the slave (second) point the same as master (first) point.
+ // This will allow to skip adding point-point coincidence to the set of constraints
+ // and give us speed-up in solving the set of equations
+ myStorage->addCoincidentPoints(theAttributes[0], theAttributes[1]);
+ }
+ else if (theAttributes[2]) {
// check the type of entity (line or circle)
- Slvs_Entity anEnt = myStorage->getEntity(theAttributes[2]);
- if (anEnt.type == SLVS_E_LINE_SEGMENT)
- myType = SLVS_C_PT_ON_LINE;
- else if (anEnt.type == SLVS_E_CIRCLE || anEnt.type == SLVS_E_ARC_OF_CIRCLE)
- myType = SLVS_C_PT_ON_CIRCLE;
+ SketchSolver_EntityType anEntType = theAttributes[2]->type();
+ if (anEntType == ENTITY_LINE)
+ myType = CONSTRAINT_PT_ON_LINE;
+ else if (anEntType == ENTITY_CIRCLE || anEntType == ENTITY_ARC)
+ myType = CONSTRAINT_PT_ON_CIRCLE;
else
myErrorMsg = SketchSolver_Error::INCORRECT_ATTRIBUTE();
} else
myErrorMsg = SketchSolver_Error::INCORRECT_ATTRIBUTE();
}
-
-
-bool SketchSolver_ConstraintCoincidence::hasConstraint(ConstraintPtr theConstraint) const
-{
- if (myBaseConstraint == theConstraint)
- return true;
- return myExtraCoincidence.find(theConstraint) != myExtraCoincidence.end();
-}
-
-std::list<ConstraintPtr> SketchSolver_ConstraintCoincidence::constraints() const
-{
- std::list<ConstraintPtr> aConstraints;
- aConstraints.push_back(myBaseConstraint);
- std::map<ConstraintPtr, Slvs_hConstraint>::const_iterator anIt = myExtraCoincidence.begin();
- for (; anIt != myExtraCoincidence.end(); anIt++)
- aConstraints.push_back(anIt->first);
- return aConstraints;
-}
-
-bool SketchSolver_ConstraintCoincidence::isCoincide(
- std::shared_ptr<SketchSolver_ConstraintCoincidence> theConstraint) const
-{
- // Multi-coincidence allowed for two points only
- if (getType() != theConstraint->getType() || getType() != SLVS_C_POINTS_COINCIDENT)
- return false;
-
- std::set<AttributePtr>::const_iterator anAttrIter = theConstraint->myCoincidentPoints.begin();
- for (; anAttrIter != theConstraint->myCoincidentPoints.end(); anAttrIter++)
- if (myCoincidentPoints.find(*anAttrIter) != myCoincidentPoints.end())
- return true;
- return false;
-}
-
-void SketchSolver_ConstraintCoincidence::attach(
- std::shared_ptr<SketchSolver_ConstraintCoincidence> theConstraint)
-{
- cleanErrorMsg();
- // Remove constraints from theConstraint
- std::vector<Slvs_hConstraint>::iterator aCIter = theConstraint->mySlvsConstraints.begin();
- for (; aCIter != theConstraint->mySlvsConstraints.end(); aCIter++)
- theConstraint->myStorage->removeConstraint(*aCIter);
-
- if (myStorage == theConstraint->myStorage) {
- // Clean removed items
- std::set<Slvs_hParam> aRemParams;
- std::set<Slvs_hEntity> aRemEnts;
- std::set<Slvs_hConstraint> aRemConstr;
- theConstraint->myStorage->getRemoved(aRemParams, aRemEnts, aRemConstr);
-
- if (!aRemEnts.empty()) {
- std::map<FeaturePtr, Slvs_hEntity>::iterator aFeatIt = theConstraint->myFeatureMap.begin();
- while (aFeatIt != theConstraint->myFeatureMap.end()) {
- if (aRemEnts.find(aFeatIt->second) != aRemEnts.end()) {
- // remove feature
- std::map<FeaturePtr, Slvs_hEntity>::iterator aRemoveIt = aFeatIt++;
- theConstraint->myFeatureMap.erase(aRemoveIt);
- } else
- ++aFeatIt;
- }
- std::map<AttributePtr, Slvs_hEntity>::iterator anAttrIt = theConstraint->myAttributeMap.begin();
- while (anAttrIt != theConstraint->myAttributeMap.end()) {
- if (aRemEnts.find(anAttrIt->second) != aRemEnts.end()) {
- // remove attribute
- std::map<AttributePtr, Slvs_hEntity>::iterator aRemoveIt = anAttrIt++;
- theConstraint->myAttributeMap.erase(aRemoveIt);
- } else
- ++anAttrIt;
- }
- }
- }
-
- // Copy data.
- addConstraint(theConstraint->myBaseConstraint);
- std::map<ConstraintPtr, Slvs_hConstraint>::iterator aConstrIter =
- theConstraint->myExtraCoincidence.begin();
- for (; aConstrIter != theConstraint->myExtraCoincidence.end(); aConstrIter++)
- addConstraint(aConstrIter->first);
- // Clear the lists to not remove the entities on destruction
- theConstraint->mySlvsConstraints.clear();
- theConstraint->myFeatureMap.clear();
- theConstraint->myAttributeMap.clear();
-}
-
-Slvs_hConstraint SketchSolver_ConstraintCoincidence::addConstraint(
- Slvs_hEntity thePoint1, Slvs_hEntity thePoint2)
-{
- bool hasDuplicated = myStorage->hasDuplicatedConstraint();
- Slvs_Constraint aNewConstraint = Slvs_MakeConstraint(SLVS_E_UNKNOWN, myGroup->getId(),
- SLVS_C_POINTS_COINCIDENT, myGroup->getWorkplaneId(), 0.0, thePoint1, thePoint2,
- SLVS_E_UNKNOWN, SLVS_E_UNKNOWN);
- Slvs_hConstraint aNewID = myStorage->addConstraint(aNewConstraint);
- if (!hasDuplicated && myStorage->hasDuplicatedConstraint()) {
- // the duplicated constraint appears
- myStorage->removeConstraint(aNewID);
- return SLVS_E_UNKNOWN;
- }
- mySlvsConstraints.push_back(aNewID);
- return aNewID;
-}
-
-void SketchSolver_ConstraintCoincidence::addConstraint(ConstraintPtr theConstraint)
-{
- std::list<AttributePtr> anAttrList =
- theConstraint->data()->attributes(ModelAPI_AttributeRefAttr::typeId());
- std::list<AttributePtr>::iterator anIter = anAttrList.begin();
- std::vector<Slvs_hEntity> anEntities;
- for (; anIter != anAttrList.end(); anIter++) {
- AttributeRefAttrPtr aRefAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(*anIter);
- if (!aRefAttr || aRefAttr->isObject() ||
- myAttributeMap.find(aRefAttr->attr()) != myAttributeMap.end())
- continue;
- int aType;
- Slvs_hEntity anEntityID = myGroup->getAttributeId(aRefAttr->attr());
- if (anEntityID == SLVS_E_UNKNOWN)
- anEntityID = changeEntity(aRefAttr->attr(), aType);
- anEntities.push_back(anEntityID);
- myCoincidentPoints.insert(aRefAttr->attr());
- }
-
- Slvs_hConstraint aNewConstr = SLVS_E_UNKNOWN;
- std::vector<Slvs_hEntity>::iterator anEntIter = anEntities.begin();
- if (mySlvsConstraints.empty()) {
- aNewConstr = addConstraint(*anEntIter, *(anEntIter + 1));
- anEntIter += 2;
- }
- Slvs_Constraint aBaseCoincidence = myStorage->getConstraint(mySlvsConstraints.front());
- for (; anEntIter != anEntities.end(); anEntIter++)
- aNewConstr = addConstraint(aBaseCoincidence.ptA, *anEntIter);
- myExtraCoincidence[theConstraint] = aNewConstr;
-}
-
-void SketchSolver_ConstraintCoincidence::process()
-{
- SketchSolver_Constraint::process();
-
- // Fill the list of coincident points
- std::list<AttributePtr> anAttrList =
- myBaseConstraint->data()->attributes(ModelAPI_AttributeRefAttr::typeId());
- std::list<AttributePtr>::iterator anIt = anAttrList.begin();
- for (; anIt != anAttrList.end(); anIt++) {
- AttributeRefAttrPtr aRefAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(*anIt);
- if (!aRefAttr || aRefAttr->isObject())
- continue;
- myCoincidentPoints.insert(aRefAttr->attr());
- }
-}
-
-bool SketchSolver_ConstraintCoincidence::remove(ConstraintPtr theConstraint)
-{
- cleanErrorMsg();
- if (mySlvsConstraints.empty())
- return true;
- ConstraintPtr aConstraint = theConstraint ? theConstraint : myBaseConstraint;
- int aPos = -1; // position of constraint in the list (-1 for base constraint)
- std::map<ConstraintPtr, Slvs_hConstraint>::iterator anExtraIt;
- if (aConstraint != myBaseConstraint) {
- anExtraIt = myExtraCoincidence.find(aConstraint);
- if (anExtraIt == myExtraCoincidence.end())
- return false; // there is no constraint, which is specified to remove
- else {
- bool isEmpty = anExtraIt->second == SLVS_E_UNKNOWN;
- if (!isEmpty) {
- isEmpty = true;
- for (aPos = 0; aPos < (int)mySlvsConstraints.size(); aPos++)
- if (mySlvsConstraints[aPos] == anExtraIt->second) {
- isEmpty = false;
- break;
- }
- aPos -= 1;
- }
- myExtraCoincidence.erase(anExtraIt);
- if (isEmpty)
- return false;
- }
- }
-
- bool isFullyRemoved = myStorage->removeConstraint(mySlvsConstraints[aPos+1]);
- mySlvsConstraints.erase(mySlvsConstraints.begin() + (1+aPos));
- if (aPos < 0 && !myExtraCoincidence.empty()) {
- anExtraIt = myExtraCoincidence.begin();
- // Remove invalid constraints
- while (anExtraIt != myExtraCoincidence.end()) {
- if (!anExtraIt->first->data() || !anExtraIt->first->data()->isValid()) {
- std::map<ConstraintPtr, Slvs_hConstraint>::iterator aTempIt = anExtraIt++;
- if (aTempIt->first != SLVS_E_UNKNOWN) {
- myStorage->removeConstraint(aTempIt->second);
- std::vector<Slvs_hConstraint>::iterator anIt = mySlvsConstraints.begin();
- for (; anIt != mySlvsConstraints.end(); anIt++)
- if (*anIt == aTempIt->second) {
- mySlvsConstraints.erase(anIt);
- break;
- }
- }
- myExtraCoincidence.erase(aTempIt);
- continue;
- }
- anExtraIt++;
- }
- // Find first non-extra conststraint
- anExtraIt = myExtraCoincidence.begin();
- while (anExtraIt != myExtraCoincidence.end() && anExtraIt->second == SLVS_E_UNKNOWN)
- anExtraIt++;
- if (anExtraIt != myExtraCoincidence.end()) {
- // Need to specify another base coincidence constraint
- myBaseConstraint = anExtraIt->first;
- myExtraCoincidence.erase(anExtraIt);
- if (mySlvsConstraints.empty()) {
- std::vector<Slvs_hConstraint>::iterator aCIter = mySlvsConstraints.begin();
- Slvs_Constraint aBase = myStorage->getConstraint(*aCIter);
- for (++aCIter; aCIter != mySlvsConstraints.end(); aCIter++) {
- Slvs_Constraint aConstr = myStorage->getConstraint(*aCIter);
- aConstr.ptA = aBase.ptA;
- myStorage->updateConstraint(aConstr);
- }
- }
- }
- }
- // Clear removed attributes
- std::set<Slvs_hParam> aParamRemoved;
- std::set<Slvs_hEntity> anEntRemoved;
- std::set<Slvs_hConstraint> aConstrRemoved;
- myStorage->getRemoved(aParamRemoved, anEntRemoved, aConstrRemoved);
- std::map<AttributePtr, Slvs_hEntity>::iterator anAttrIter = myAttributeMap.begin();
- while (anAttrIter != myAttributeMap.end()) {
- if (anEntRemoved.find(anAttrIter->second) != anEntRemoved.end()) {
- std::map<AttributePtr, Slvs_hEntity>::iterator aTempIt = anAttrIter++;
- myCoincidentPoints.erase(aTempIt->first);
- myAttributeMap.erase(aTempIt);
- continue;
- }
- anAttrIter++;
- }
-
- // Go through remaining extra coincidence and try to add or remove them
- anExtraIt = myExtraCoincidence.begin();
- while (anExtraIt != myExtraCoincidence.end()) {
- if (anExtraIt->first == SLVS_E_UNKNOWN) {
- if (!anExtraIt->first->data() || !anExtraIt->first->data()->isValid()) {
- std::map<ConstraintPtr, Slvs_hConstraint>::iterator aTempIt = anExtraIt++;
- myExtraCoincidence.erase(aTempIt);
- continue;
- }
- if (mySlvsConstraints.empty()) {
- myBaseConstraint = anExtraIt->first;
- std::map<ConstraintPtr, Slvs_hConstraint>::iterator aTempIt = anExtraIt++;
- myExtraCoincidence.erase(aTempIt);
- process();
- continue;
- } else
- addConstraint(anExtraIt->first);
- }
- anExtraIt++;
- }
- return mySlvsConstraints.empty();
-}
-
