X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FSketchPlugin%2FSketchPlugin_ConstraintMirror.cpp;h=e17a704c5b5238d08f313cc0766a8eb9e39ec147;hb=bc13ffbc8c0dbdce3805bf9362ee8a3c05adf9b4;hp=f0da5515fa7a63ee32484a78dcfb5d5f2437f3bd;hpb=a731f82dccbfdb67cbf8e8d617222a4d3e32018a;p=modules%2Fshaper.git

diff --git a/src/SketchPlugin/SketchPlugin_ConstraintMirror.cpp b/src/SketchPlugin/SketchPlugin_ConstraintMirror.cpp
old mode 100644
new mode 100755
index f0da5515f..e17a704c5
--- a/src/SketchPlugin/SketchPlugin_ConstraintMirror.cpp
+++ b/src/SketchPlugin/SketchPlugin_ConstraintMirror.cpp
@@ -10,7 +10,9 @@
 #include <ModelAPI_Data.h>
 #include <ModelAPI_ResultConstruction.h>
 #include <ModelAPI_AttributeRefList.h>
+#include <ModelAPI_Events.h>
 #include <ModelAPI_Session.h>
+#include <ModelAPI_Validator.h>
 
 #include <SketchPlugin_Line.h>
 #include <SketchPlugin_Sketch.h>
@@ -28,33 +30,97 @@ void SketchPlugin_ConstraintMirror::initAttributes()
   data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::typeId());
   data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefList::typeId());
   data()->addAttribute(SketchPlugin_Constraint::ENTITY_C(), ModelAPI_AttributeRefList::typeId());
+  data()->addAttribute(SketchPlugin_ConstraintMirror::MIRROR_LIST_ID(), ModelAPI_AttributeRefList::typeId());
+  ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), SketchPlugin_Constraint::ENTITY_B());
+  ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), SketchPlugin_Constraint::ENTITY_C());
 }
 
 void SketchPlugin_ConstraintMirror::execute()
 {
-  // Objects to be mirrored will be created here
+  AttributeRefListPtr aMirrorObjectRefs = reflist(SketchPlugin_ConstraintMirror::MIRROR_LIST_ID());
+
+  // 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);
+
   std::shared_ptr<ModelAPI_Data> aData = data();
   AttributeRefListPtr aRefListOfShapes = std::dynamic_pointer_cast<ModelAPI_AttributeRefList>(
       aData->attribute(SketchPlugin_Constraint::ENTITY_B()));
-  if (!aRefListOfShapes->isInitialized())
-    return ;
-
   AttributeRefListPtr aRefListOfMirrored = std::dynamic_pointer_cast<ModelAPI_AttributeRefList>(
       aData->attribute(SketchPlugin_Constraint::ENTITY_C()));
-  // Check consistency of initial list and mirrored list
   std::list<ObjectPtr> anInitialList =  aRefListOfShapes->list();
   std::list<ObjectPtr> aMirroredList =  aRefListOfMirrored->list();
+  std::vector<bool> isUsed(anInitialList.size(), false);
+  // add new items to the list
+  for(int anInd = 0; anInd < aMirrorObjectRefs->size(); anInd++) {
+    ObjectPtr anObject = aMirrorObjectRefs->object(anInd);
+    std::list<ObjectPtr>::const_iterator anIt = anInitialList.begin();
+    std::vector<bool>::iterator aUsedIt = isUsed.begin();
+    for (; anIt != anInitialList.end(); anIt++, aUsedIt++)
+      if (*anIt == anObject) {
+        *aUsedIt = true;
+        break;
+      }
+    if (anIt == anInitialList.end())
+      aRefListOfShapes->append(anObject);
+  }
+  // remove unused items
   std::list<ObjectPtr>::iterator anInitIter = anInitialList.begin();
   std::list<ObjectPtr>::iterator aMirrorIter = aMirroredList.begin();
+  std::vector<bool>::iterator aUsedIter = isUsed.begin();
+  for (; aUsedIter != isUsed.end(); aUsedIter++) {
+    if (!(*aUsedIter)) {
+      aRefListOfShapes->remove(*anInitIter);
+      if (aMirrorIter != aMirroredList.end()) {
+        aRefListOfMirrored->remove(*aMirrorIter);
+        // remove the corresponding feature from the sketch
+        ResultConstructionPtr aRC =
+            std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(*aMirrorIter);
+        DocumentPtr aDoc = aRC ? aRC->document() : DocumentPtr();
+        FeaturePtr aFeature =  aDoc ? aDoc->feature(aRC) : FeaturePtr();
+        if (aFeature)
+          aDoc->removeFeature(aFeature);
+      }
+    }
+    if (anInitIter != anInitialList.end())
+      anInitIter++;
+    if (aMirrorIter != aMirroredList.end())
+      aMirrorIter++;
+  }
+
+  static Events_ID aRedisplayEvent = Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY);
+
+  // Check consistency of initial list and mirrored list
+  anInitialList =  aRefListOfShapes->list();
+  anInitIter = anInitialList.begin();
+  aMirrorIter = aMirroredList.begin();
   int indFirstWrong = 0; // index of element starts difference in the lists
   std::set<int> anInvalidInd; // list of indices of removed features
   std::shared_ptr<SketchPlugin_Feature> aFeatureIn, aFeatureOut;
   for ( ; anInitIter != anInitialList.end(); anInitIter++, indFirstWrong++) {
     // Add features and store indices of objects to remove
     aFeatureIn = std::dynamic_pointer_cast<SketchPlugin_Feature>(*anInitIter);
-    aFeatureOut = aMirrorIter != aMirroredList.end() ? 
-        std::dynamic_pointer_cast<SketchPlugin_Feature>(*aMirrorIter) :
-        std::shared_ptr<SketchPlugin_Feature>();
+    ResultConstructionPtr aRCIn;
+    if (!aFeatureIn) {
+      aRCIn = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(*anInitIter);
+      if (aRCIn)
+        aFeatureIn = std::dynamic_pointer_cast<SketchPlugin_Feature>(
+            aRCIn->document()->feature(aRCIn));
+    }
+    if (aMirrorIter == aMirroredList.end())
+      aFeatureOut = std::shared_ptr<SketchPlugin_Feature>();
+    else {
+      aFeatureOut = std::dynamic_pointer_cast<SketchPlugin_Feature>(*aMirrorIter);
+      if (!aFeatureOut) {
+        ResultConstructionPtr aRC =
+            std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(*anInitIter);
+        if (aRC)
+          aFeatureOut = std::dynamic_pointer_cast<SketchPlugin_Feature>(
+              aRC->document()->feature(aRC));
+      }
+    }
     if (!aFeatureIn) {
       if (aFeatureOut)
         break; // the lists are inconsistent
@@ -64,9 +130,22 @@ void SketchPlugin_ConstraintMirror::execute()
       if (aMirrorIter != aMirroredList.end())
         break; // the lists are inconsistent
       // There is no mirrored object yet, create it
-      FeaturePtr aNewFeature = sketch()->addFeature(aFeatureIn->getKind());
-      aFeatureIn->data()->copyTo(aNewFeature->data());
-      aRefListOfMirrored->append(aNewFeature);
+      FeaturePtr aNewFeature = SketchPlugin_Sketch::addUniqueNamedCopiedFeature(aFeatureIn, sketch(), true);
+      aNewFeature->execute();
+      ModelAPI_EventCreator::get()->sendUpdated(aNewFeature, aRedisplayEvent);
+
+      std::shared_ptr<GeomAPI_Shape> aShapeIn = aRCIn->shape();
+      const std::list<ResultPtr>& aResults = aNewFeature->results();
+      std::list<ResultPtr>::const_iterator anIt = aResults.begin();
+      for (; anIt != aResults.end(); anIt++) {
+        ResultConstructionPtr aRC = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(*anIt);
+        if (!aRC) continue;
+        std::shared_ptr<GeomAPI_Shape> aShapeOut = aRC->shape();
+        if ((aShapeIn->isVertex() && aShapeOut->isVertex()) ||
+            (aShapeIn->isEdge() && aShapeOut->isEdge()) ||
+            (aShapeIn->isFace() && aShapeOut->isFace()))
+          aRefListOfMirrored->append(aRC);
+      }
       continue;
     }
     if (aFeatureIn->getKind() != aFeatureOut->getKind())
@@ -98,6 +177,10 @@ void SketchPlugin_ConstraintMirror::execute()
       aRefListOfMirrored->append(aNewFeature);
     }
   }
+
+  // send events to update the sub-features by the solver
+  if (isUpdateFlushed)
+    Events_Loop::loop()->setFlushed(anUpdateEvent, true);
 }
 
 AISObjectPtr SketchPlugin_ConstraintMirror::getAISObject(AISObjectPtr thePrevious)
@@ -105,9 +188,68 @@ AISObjectPtr SketchPlugin_ConstraintMirror::getAISObject(AISObjectPtr thePreviou
   if (!sketch())
     return thePrevious;
 
-  AISObjectPtr anAIS = thePrevious;
-  /// TODO: Equal constraint presentation should be put here
+  AISObjectPtr anAIS = SketcherPrs_Factory::mirrorConstraint(this, sketch()->coordinatePlane(),
+                                                             thePrevious);
   return anAIS;
 }
 
+void SketchPlugin_ConstraintMirror::erase()
+{
+  // Set copy attribute to false on all copied features.
+  AttributeRefListPtr aRefListOfMirrored = std::dynamic_pointer_cast<ModelAPI_AttributeRefList>(
+      data()->attribute(SketchPlugin_Constraint::ENTITY_C()));
+
+  if(aRefListOfMirrored.get()) {
+    static Events_Loop* aLoop = Events_Loop::loop();
+    static Events_ID aRedispEvent = aLoop->eventByName(EVENT_OBJECT_TO_REDISPLAY);
+
+    std::list<ObjectPtr> aTargetList = aRefListOfMirrored->list();
+    for(std::list<ObjectPtr>::const_iterator aTargetIt = aTargetList.cbegin(); aTargetIt != aTargetList.cend(); aTargetIt++) {
+      if((*aTargetIt).get()) {
+        ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(*aTargetIt);
+        if(aRes.get()) {
+          FeaturePtr aFeature = aRes->document()->feature(aRes);
+          if(aFeature.get()) {
+            AttributeBooleanPtr aBooleanAttr = aFeature->boolean(SketchPlugin_SketchEntity::COPY_ID());
+            if(aBooleanAttr.get()) {
+              if (ModelAPI_Session::get()->isOperation()) // if this is not undo or redo
+                aBooleanAttr->setValue(false);
+              // Redisplay object as it is not copy anymore.
+              ModelAPI_EventCreator::get()->sendUpdated(aRes, aRedispEvent);
+            }
+          }
+        }
+      }
+    }
+  }
+
+  SketchPlugin_ConstraintBase::erase();
+}
+
+void SketchPlugin_ConstraintMirror::attributeChanged(const std::string& theID)
+{
+  if (theID == MIRROR_LIST_ID()) {
+    AttributeRefListPtr aMirrorObjectRefs = reflist(MIRROR_LIST_ID());
+    if (aMirrorObjectRefs->size() == 0) {
+      // Clear list of objects
+      AttributeRefListPtr aRefListOfMirrored = std::dynamic_pointer_cast<ModelAPI_AttributeRefList>(
+          data()->attribute(SketchPlugin_Constraint::ENTITY_C()));
+      std::list<ObjectPtr> aTargetList = aRefListOfMirrored->list();
+      std::list<ObjectPtr>::iterator aTargetIter = aTargetList.begin();
+      for (; aTargetIter != aTargetList.end(); aTargetIter++) {
+        aRefListOfMirrored->remove(*aTargetIter);
+        // remove the corresponding feature from the sketch
+        ResultConstructionPtr aRC =
+            std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(*aTargetIter);
+        DocumentPtr aDoc = aRC ? aRC->document() : DocumentPtr();
+        FeaturePtr aFeature =  aDoc ? aDoc->feature(aRC) : FeaturePtr();
+        if (aFeature)
+          aDoc->removeFeature(aFeature);
+      }
+      aRefListOfMirrored->clear();
+      std::dynamic_pointer_cast<ModelAPI_AttributeRefList>(
+        data()->attribute(SketchPlugin_Constraint::ENTITY_B()))->clear();
+    }
+  }
+}