Salome HOME
Merge branch 'Dev_1.4.0' of newgeom:newgeom into Dev_1.4.0
[modules/shaper.git] / src / Model / Model_ResultPart.cpp
index 5023340b1b5f8f6a08a712b665966a4e20921e9d..bb12b10756fffa3c95096e4eda37a3d8ca10f63d 100644 (file)
@@ -13,6 +13,7 @@
 #include <ModelAPI_ResultBody.h>
 #include <ModelAPI_AttributeIntArray.h>
 #include <ModelAPI_AttributeSelectionList.h>
+#include <ModelAPI_AttributeReference.h>
 #include <Model_Document.h>
 #include <Events_Loop.h>
 #include <ModelAPI_Events.h>
 #include <BRep_Builder.hxx>
 #include <TopExp_Explorer.hxx>
 
+#define baseRef() std::dynamic_pointer_cast<Model_ResultPart>(data()->reference(BASE_REF_ID())->value())
+
 void Model_ResultPart::initAttributes()
 {
   // append the color attribute. It is empty, the attribute will be filled by a request
-  DataPtr aData = data();
-  aData->addAttribute(COLOR_ID(), ModelAPI_AttributeIntArray::typeId());
+  data()->addAttribute(DOC_REF(), ModelAPI_AttributeDocRef::typeId());
+  data()->addAttribute(COLOR_ID(), ModelAPI_AttributeIntArray::typeId());
+  data()->addAttribute(BASE_REF_ID(), ModelAPI_AttributeReference::typeId());
 }
 
 std::shared_ptr<ModelAPI_Document> Model_ResultPart::partDoc()
 {
-  return data()->document("PartDocument")->value();
-}
-
-std::shared_ptr<ModelAPI_Feature> Model_ResultPart::owner()
-{
-  return std::shared_ptr<ModelAPI_Feature>();  // return empty pointer
+  if (myTrsf.get()) {
+    return baseRef()->partDoc();
+  }
+  DocumentPtr aRes = data()->document(DOC_REF())->value();
+  if (!aRes.get() && myIsInLoad) { // trying to get this document from the session
+    aRes = document()->subDocument(data()->name());
+  }
+  return aRes;
 }
 
 Model_ResultPart::Model_ResultPart()
 {
-  myIsDisabled = true; // by default it is not initialized and false to be after created
-  setIsConcealed(false);
+  myIsInLoad = false;
 }
 
-void Model_ResultPart::setData(std::shared_ptr<ModelAPI_Data> theData)
+void Model_ResultPart::activate()
 {
-  ModelAPI_Result::setData(theData);
-  if (theData) {
-    data()->addAttribute(DOC_REF(), ModelAPI_AttributeDocRef::typeId());
+  if (myTrsf.get()) {
+    baseRef()->activate();
+    return;
   }
-}
 
-void Model_ResultPart::activate()
-{
   std::shared_ptr<ModelAPI_AttributeDocRef> aDocRef = data()->document(DOC_REF());
   
   if (!aDocRef->value().get()) {  // create (or open) a document if it is not yet created
+    myIsInLoad = true;
     std::shared_ptr<ModelAPI_Document> aDoc = document()->subDocument(data()->name());
+    myIsInLoad = false;
     if (aDoc) {
+      aDoc->synchronizeTransactions();
       aDocRef->setValue(aDoc);
     }
   }
@@ -80,27 +85,41 @@ void Model_ResultPart::activate()
   }
 }
 
+std::shared_ptr<ModelAPI_ResultPart> Model_ResultPart::original()
+{
+  if (myTrsf.get()) {
+    return baseRef()->original();
+  }
+  return std::dynamic_pointer_cast<ModelAPI_ResultPart>(data()->owner());
+}
+
 bool Model_ResultPart::isActivated() 
 {
+  if (myTrsf.get()) {
+    return baseRef()->isActivated();
+  }
+
   std::shared_ptr<ModelAPI_AttributeDocRef> aDocRef = data()->document(DOC_REF());
-  return aDocRef->value().get();
+  return aDocRef->value().get() != NULL;
 }
 
 bool Model_ResultPart::setDisabled(std::shared_ptr<ModelAPI_Result> theThis,
     const bool theFlag)
 {
   if (ModelAPI_ResultPart::setDisabled(theThis, theFlag)) {
-    DocumentPtr aDoc = Model_ResultPart::partDoc();
-    if (aDoc.get() && aDoc->isOpened()) {
-      // make the current feature the last in any case: to update shapes defore deactivation too
-      FeaturePtr aLastFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aDoc->object(
-        ModelAPI_Feature::group(), aDoc->size(ModelAPI_Feature::group()) - 1));
-      aDoc->setCurrentFeature(aLastFeature, false);
-      if (theFlag) { // disable, so make all features disabled too
-        // update the shape just before the deactivation: it will be used outside of part
-        myShape.Nullify();
-        shape();
-        aDoc->setCurrentFeature(FeaturePtr(), false);
+    if (!myTrsf.get()) { // disable of base result part
+      DocumentPtr aDoc = Model_ResultPart::partDoc();
+      if (aDoc.get() && aDoc->isOpened()) {
+        std::shared_ptr<Model_Document> anIntDoc = std::dynamic_pointer_cast<Model_Document>(aDoc);
+        // make the current feature the last in any case: to update shapes defore deactivation too
+        FeaturePtr aLastFeature = anIntDoc->lastFeature();
+        aDoc->setCurrentFeature(aLastFeature, false);
+        if (theFlag) { // disable, so make all features disabled too
+          // update the shape just before the deactivation: it will be used outside of part
+          updateShape();
+          shape();
+          aDoc->setCurrentFeature(FeaturePtr(), false);
+        }
       }
     }
     return true;
@@ -110,9 +129,22 @@ bool Model_ResultPart::setDisabled(std::shared_ptr<ModelAPI_Result> theThis,
 
 std::shared_ptr<GeomAPI_Shape> Model_ResultPart::shape()
 {
-  if (myShape.IsNull()) {
+  std::shared_ptr<GeomAPI_Shape> aResult(new GeomAPI_Shape);
+  if (myTrsf.get()) { // get shape of the base result and apply the transformation
+    ResultPtr anOrigResult = baseRef();
+    std::shared_ptr<GeomAPI_Shape> anOrigShape = anOrigResult->shape();
+    if (anOrigShape.get()) {
+      TopoDS_Shape aShape = anOrigShape->impl<TopoDS_Shape>();
+      if (!aShape.IsNull()) {
+        aShape.Move(*(myTrsf.get()));
+        aResult->setImpl(new TopoDS_Shape(aShape));
+      }
+    }
+    return aResult;
+  }
+  if (myShape.IsNull()) { // shape is not produced yet, create it
     DocumentPtr aDoc = Model_ResultPart::partDoc();
-    if (aDoc.get()) {
+    if (aDoc.get() && aDoc->isOpened()) {
       const std::string& aBodyGroup = ModelAPI_ResultBody::group();
       TopoDS_Compound aResultComp;
       BRep_Builder aBuilder;
@@ -120,7 +152,8 @@ std::shared_ptr<GeomAPI_Shape> Model_ResultPart::shape()
       int aNumSubs = 0;
       for(int a = aDoc->size(aBodyGroup) - 1; a >= 0; a--) {
         ResultPtr aBody = std::dynamic_pointer_cast<ModelAPI_Result>(aDoc->object(aBodyGroup, a));
-        if (aBody.get() && aBody->shape().get() && !aBody->isDisabled()) {
+        // "object" method filters out disabled and concealed anyway, so don't check
+        if (aBody.get() && aBody->shape().get()) {
           TopoDS_Shape aShape = *(aBody->shape()->implPtr<TopoDS_Shape>());
           if (!aShape.IsNull()) {
             aBuilder.Add(aResultComp, aShape);
@@ -133,41 +166,56 @@ std::shared_ptr<GeomAPI_Shape> Model_ResultPart::shape()
       }
     }
   }
-  if (myShape.IsNull())
-    return std::shared_ptr<GeomAPI_Shape>();
-  std::shared_ptr<GeomAPI_Shape> aResult(new GeomAPI_Shape);
-  aResult->setImpl(new TopoDS_Shape(myShape));
+  if (!myShape.IsNull())
+    aResult->setImpl(new TopoDS_Shape(myShape));
   return aResult;
 }
 
+// Returns true is transformation matrices are equal
+static bool IsEqualTrsf(gp_Trsf& theT1, gp_Trsf theT2) {
+  for(int aRow = 1; aRow < 4; aRow++) {
+    for(int aCol = 1; aCol < 5; aCol++) {
+      double aDiff = theT1.Value(aRow, aCol) - theT2.Value(aRow, aCol);
+      if (Abs(aDiff) > 1.e-9)
+        return false;
+    }
+  }
+  return true;
+}
+
 std::string Model_ResultPart::nameInPart(const std::shared_ptr<GeomAPI_Shape>& theShape,
   int& theIndex)
 {
   theIndex = 0; // not initialized
-  TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
-  if (aShape.IsNull())
-    return "";
-  if (data()->isOwner(this)) { // if this is moved copy of part => return the name of original shape
-    FeaturePtr anOrigFeature = 
-      std::dynamic_pointer_cast<ModelAPI_Feature>(data()->attribute(COLOR_ID())->owner());
-    if (anOrigFeature.get()) {
-      if (anOrigFeature->firstResult().get() && anOrigFeature->firstResult()->shape().get()) {
-        TopoDS_Shape anOrigShape = anOrigFeature->firstResult()->shape()->impl<TopoDS_Shape>();
-        if (!anOrigShape.IsNull()) {
-          TopExp_Explorer anExp(anOrigShape, aShape.ShapeType());
-          for(; anExp.More(); anExp.Next()) {
-            if (aShape.IsPartner(anExp.Current())) {
-              std::shared_ptr<GeomAPI_Shape> anOrigGeomShape(new GeomAPI_Shape);
-              anOrigGeomShape->setImpl(new TopoDS_Shape(anExp.Current()));
 
-              return std::dynamic_pointer_cast<Model_ResultPart>(anOrigFeature->firstResult())->
-                nameInPart(theShape, theIndex);
-            }
+  if (myTrsf.get()) { // if this is moved copy of part => return the name of original shape
+    ResultPartPtr anOrigResult = baseRef();
+    // searching in the origin the shape equal to the given but with myTrsf
+    TopoDS_Shape aSelection = theShape->impl<TopoDS_Shape>();
+    gp_Trsf aSelTrsf = aSelection.Location().Transformation();
+    TopoDS_Shape anOrigMain = anOrigResult->shape()->impl<TopoDS_Shape>();
+    if (!aSelection.IsNull() && !anOrigMain.IsNull()) {
+      TopExp_Explorer anExp(anOrigMain, aSelection.ShapeType());
+      for(; anExp.More(); anExp.Next()) {
+        if (anExp.Current().IsPartner(aSelection)) {
+          TopoDS_Shape anOrigMoved = anExp.Current().Moved(*(myTrsf.get()));
+          //if (anOrigMoved.IsSame(aSelection)) {
+          if (IsEqualTrsf(aSelTrsf, anOrigMoved.Location().Transformation())) {
+            std::shared_ptr<GeomAPI_Shape> anOrigSel(new GeomAPI_Shape);
+            anOrigSel->setImpl(new TopoDS_Shape(anExp.Current()));
+            return anOrigResult->nameInPart(anOrigSel, theIndex);
           }
         }
       }
     }
+    // something is not right
+    return "";
   }
+
+  TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
+  if (aShape.IsNull())
+    return "";
+
   // getting an access to the document of part
   std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(partDoc());
   if (!aDoc.get()) // the part document is not presented for the moment
@@ -255,12 +303,16 @@ void Model_ResultPart::colorConfigInfo(std::string& theSection, std::string& the
 void Model_ResultPart::updateShape()
 {
   myShape.Nullify();
+  myTrsf.reset();
 }
 
-void Model_ResultPart::setShape(std::shared_ptr<ModelAPI_Result> theThis, 
-    const std::shared_ptr<GeomAPI_Shape>& theTransformed)
+void Model_ResultPart::setTrsf(std::shared_ptr<ModelAPI_Result> theThis, 
+    const std::shared_ptr<GeomAPI_Trsf>& theTransformation)
 {
-  myShape = theTransformed->impl<TopoDS_Shape>();
+  updateShape();
+  if (theTransformation.get()) {
+    myTrsf = std::shared_ptr<gp_Trsf>(new gp_Trsf(theTransformation->impl<gp_Trsf>()));
+  }
   // the result must be explicitly updated
   static Events_Loop* aLoop = Events_Loop::loop();
   static Events_ID EVENT_DISP = aLoop->eventByName(EVENT_OBJECT_TO_REDISPLAY);