[EDF] (2023-T1) Sketch middle point constrain should create point if missing

[modules/shaper.git] / src / SketchPlugin / SketchPlugin_ConstraintMiddle.cpp

diff --git a/src/SketchPlugin/SketchPlugin_ConstraintMiddle.cpp b/src/SketchPlugin/SketchPlugin_ConstraintMiddle.cpp
index ad9ab375593c4ed2476952e8ca3c5d1cd2d80fac..9d478833b9d44ff94445cb1572fbb09c2d960587 100644 (file)
--- a/src/SketchPlugin/SketchPlugin_ConstraintMiddle.cpp
+++ b/src/SketchPlugin/SketchPlugin_ConstraintMiddle.cpp
@@ -21,28 +21,113 @@
 
 #include "SketcherPrs_Factory.h"
 
+#include <GeomDataAPI_Point2D.h>
+
+#include <ModelAPI_Validator.h>
+#include <ModelAPI_AttributeString.h>
+
+#include <ModelAPI_Events.h>
+#include <SketchPlugin_Point.h>
+#include <SketchPlugin_Tools.h>
+
 SketchPlugin_ConstraintMiddle::SketchPlugin_ConstraintMiddle()
 {
 }
 
+// Create new point for Middle constraint
+void SketchPlugin_ConstraintMiddle::CreatePoint()
+{
+  // Wait all objects being created, then send update events
+  static Events_ID anUpdateEvent = Events_Loop::eventByName(EVENT_OBJECT_UPDATED);
+  bool isUpdateFlushed = Events_Loop::loop()->isFlushed(anUpdateEvent);
+  if (isUpdateFlushed)
+    Events_Loop::loop()->setFlushed(anUpdateEvent, false);
+
+  auto aTrPnt = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(data()->attribute(POINT_REF_ID()));
+
+  if (!myPoint)
+  {
+    // Get last subfeature (constraintMiddle) for set as parent
+    FeaturePtr aCurrentFeature = sketch()->subFeature(sketch()->numberOfSubs() - 1);
+    keepCurrentFeature();
+    myPoint = sketch()->addFeature(SketchPlugin_Point::ID());
+    myPoint->boolean(SketchPlugin_SketchEntity::AUXILIARY_ID())->setValue(true);
+    restoreCurrentFeature();
+
+    myPoint->reference(SketchPlugin_Point::PARENT_ID())->setValue(aCurrentFeature);
+  }
+
+  AttributePoint2DPtr aCoord = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
+    myPoint->attribute(SketchPlugin_Point::COORD_ID()));
+  aCoord->setValue(aTrPnt->pnt());
+
+  myPoint->execute();
+
+  // Init second attr for constraint
+  refattr(SketchPlugin_Constraint::ENTITY_B())->setObject(myPoint);
+
+
+  if (isUpdateFlushed)
+    Events_Loop::loop()->setFlushed(anUpdateEvent, true);
+}
+
 void SketchPlugin_ConstraintMiddle::initAttributes()
 {
+  data()->addAttribute(SketchPlugin_ConstraintMiddle::MIDDLE_TYPE(), ModelAPI_AttributeString::typeId());
+
   data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::typeId());
   data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefAttr::typeId());
+
+  data()->addAttribute(POINT_REF_ID(), GeomDataAPI_Point2D::typeId());
+
+  ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), POINT_REF_ID());
 }
 
 void SketchPlugin_ConstraintMiddle::execute()
 {
+  if (string(MIDDLE_TYPE())->value() == MIDDLE_TYPE_BY_LINE())
+  {
+    std::dynamic_pointer_cast<GeomDataAPI_Point2D>(data()->attribute(POINT_REF_ID()))->setValue(1., 1.);
+    AttributeRefAttrPtr aPointRes = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
+      data()->attribute(SketchPlugin_Constraint::ENTITY_B()));
+
+    auto aRefAttr = data()->refattr(SketchPlugin_Constraint::ENTITY_A())->object();
+    if (!aRefAttr.get())
+      return;
+
+
+    if (!attribute(ENTITY_B())->isInitialized())
+      CreatePoint(); // Create new point
+  }
+}
+
+void SketchPlugin_ConstraintMiddle::attributeChanged(const std::string& theID)
+{
+  if (theID == MIDDLE_TYPE())
+  {
+    SketchPlugin_Tools::resetAttribute(this, ENTITY_A());
+    SketchPlugin_Tools::resetAttribute(this, ENTITY_B());
+  }
+  else if (theID == POINT_REF_ID())
+  {
+    if (!myPoint)
+      return;
+
+    auto aTrPnt = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(data()->attribute(POINT_REF_ID()));
+    AttributePoint2DPtr aCoord = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
+      myPoint->attribute(SketchPlugin_Point::COORD_ID()));
+    aCoord->setValue(aTrPnt->pnt());
+
+    myPoint->execute();
+  }
 }
 
 AISObjectPtr SketchPlugin_ConstraintMiddle::getAISObject(AISObjectPtr thePrevious)
 {
   if (!sketch())
     return thePrevious;
-
   AISObjectPtr anAIS = SketcherPrs_Factory::middleConstraint(this, sketch(),
                                                              thePrevious);
+
   return anAIS;
 }
-
-