updated copyright message

[modules/shaper.git] / src / SketchSolver / SketchSolver_ConstraintEqual.cpp

diff --git a/src/SketchSolver/SketchSolver_ConstraintEqual.cpp b/src/SketchSolver/SketchSolver_ConstraintEqual.cpp
index e91ca2cf0f2a8d0f25db12649303f8e081ac013e..e0e97f9563184d853c7252bd97d46b76e2ad6454 100644 (file)
--- a/src/SketchSolver/SketchSolver_ConstraintEqual.cpp
+++ b/src/SketchSolver/SketchSolver_ConstraintEqual.cpp
@@ -1,70 +1,110 @@
+// Copyright (C) 2014-2023  CEA, EDF
+//
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Lesser General Public
+// License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+
 #include <SketchSolver_ConstraintEqual.h>
-#include <SketchSolver_Group.h>
 #include <SketchSolver_Error.h>
+#include <PlaneGCSSolver_AttributeBuilder.h>
 
+#include <ModelAPI_AttributeRefAttr.h>
+#include <SketchPlugin_Line.h>
 
-void SketchSolver_ConstraintEqual::process()
+void SketchSolver_ConstraintEqual::getAttributes(
+    EntityWrapperPtr& theValue,
+    std::vector<EntityWrapperPtr>& theAttributes)
 {
-  cleanErrorMsg();
-  if (!myBaseConstraint || !myStorage || myGroup == 0) {
-    /// TODO: Put error message here
+  SketchSolver_Constraint::getAttributes(theValue, theAttributes);
+  if (!myErrorMsg.empty() || !theAttributes[2] || !theAttributes[3]) {
+    theAttributes.clear();
     return;
   }
-  if (!mySlvsConstraints.empty()) // some data is changed, update constraint
-    update(myBaseConstraint);
-
-  double aValue;
-  std::vector<Slvs_hEntity> anEntities;
-  getAttributes(aValue, anEntities);
-  if (!myErrorMsg.empty())
-    return;
 
   // Check the quantity of entities of each type
   int aNbLines = 0;
   int aNbArcs = 0;
   int aNbCircs = 0;
+  int aNbEllipses = 0;
   bool isArcFirst = false; // in line-arc equivalence, the line should be first
-  std::vector<Slvs_hEntity>::iterator anEntIter = anEntities.begin();
-  for (; anEntIter != anEntities.end(); anEntIter++) {
-    Slvs_Entity anEnt = myStorage->getEntity(*anEntIter);
-    if (anEnt.type == SLVS_E_LINE_SEGMENT)
-      aNbLines++;
-    else if (anEnt.type == SLVS_E_CIRCLE)
-      aNbCircs++;
-    else if (anEnt.type == SLVS_E_ARC_OF_CIRCLE) {
-      aNbArcs++;
+  std::vector<EntityWrapperPtr>::iterator anAttrIt = theAttributes.begin() + 2;
+  for (; anAttrIt != theAttributes.end(); ++anAttrIt) {
+    SketchSolver_EntityType aType = (*anAttrIt)->type();
+    if (aType == ENTITY_LINE)
+      ++aNbLines;
+    else if (aType == ENTITY_CIRCLE)
+      ++aNbCircs;
+    else if (aType == ENTITY_ARC) {
+      ++aNbArcs;
       isArcFirst = (aNbLines == 0);
     }
+    else if (aType == ENTITY_ELLIPSE || aType == ENTITY_ELLIPTIC_ARC)
+      ++aNbEllipses;
   }
 
-  if (aNbLines + aNbArcs + aNbCircs != 2 ||
-      (aNbLines == aNbCircs && aNbArcs == 0)) {
+  if (aNbLines + aNbArcs + aNbCircs + aNbEllipses != 2 ||
+     (aNbArcs == 1 && aNbEllipses != 0) || aNbEllipses == 1) {
     myErrorMsg = SketchSolver_Error::INCORRECT_ATTRIBUTE();
     return;
   }
 
   switch (aNbLines) {
   case 0:
-    myType = SLVS_C_EQUAL_RADIUS;
+    if (aNbEllipses == 2) {
+      myType = CONSTRAINT_EQUAL_ELLIPSES;
+      std::shared_ptr<PlaneGCSSolver_Storage> aStorage =
+          std::dynamic_pointer_cast<PlaneGCSSolver_Storage>(myStorage);
+      myAuxValue = ScalarWrapperPtr(new PlaneGCSSolver_ScalarWrapper(aStorage->createParameter()));
+      theValue = myAuxValue;
+    }
+    else
+      myType = CONSTRAINT_EQUAL_RADIUS;
     break;
   case 1:
-    myType = SLVS_C_EQUAL_LINE_ARC_LEN;
+    myType = CONSTRAINT_EQUAL_LINE_ARC;
     if (isArcFirst) { // change the order of arc and line
-      Slvs_hEntity aTmp = anEntities[2];
-      anEntities[2] = anEntities[3];
-      anEntities[3] = aTmp;
+      EntityWrapperPtr aTmp = theAttributes[2];
+      theAttributes[2] = theAttributes[3];
+      theAttributes[3] = aTmp;
     }
     break;
   default:
-    myType = SLVS_C_EQUAL_LENGTH_LINES;
+    myType = CONSTRAINT_EQUAL_LINES;
+
+    AttributeRefAttrPtr aRefLine1 =
+        myBaseConstraint->refattr(SketchPlugin_Constraint::ENTITY_A());
+    FeaturePtr aLine1 = ModelAPI_Feature::feature(aRefLine1->object());
+    if (aLine1) {
+      PlaneGCSSolver_AttributeBuilder aBuilder(myStorage);
+      // store length of first line as a value for constraint
+      // (will be used to make equal lengths of lines)
+      theValue = aBuilder.createAttribute(aLine1->attribute(SketchPlugin_Line::LENGTH_ID()));
+    }
     break;
   }
-
-  Slvs_Constraint aConstraint = Slvs_MakeConstraint(SLVS_C_UNKNOWN, myGroup->getId(),
-      getType(), myGroup->getWorkplaneId(), aValue,
-      anEntities[0], anEntities[1], anEntities[2], anEntities[3]);
-  aConstraint.h = myStorage->addConstraint(aConstraint);
-  mySlvsConstraints.push_back(aConstraint.h);
-  adjustConstraint();
 }
 
+bool SketchSolver_ConstraintEqual::remove()
+{
+  if (myAuxValue) {
+    std::shared_ptr<PlaneGCSSolver_Storage> aStorage =
+        std::dynamic_pointer_cast<PlaneGCSSolver_Storage>(myStorage);
+    GCS::SET_pD aParams;
+    aParams.insert(myAuxValue->scalar());
+    aStorage->removeParameters(aParams);
+  }
+  return SketchSolver_Constraint::remove();
+}