Parts results and activation/deactivation management (show/hide also)

[modules/shaper.git] / src / Model / Model_ResultPart.cpp

diff --git a/src/Model/Model_ResultPart.cpp b/src/Model/Model_ResultPart.cpp
index 10e884ff3c4ed4ff939d1b650b2fef54c0dbf481..e49604794c6e2e4149106645dc762c744b582832 100644 (file)
--- a/src/Model/Model_ResultPart.cpp
+++ b/src/Model/Model_ResultPart.cpp
@@ -74,12 +74,15 @@ bool Model_ResultPart::setDisabled(std::shared_ptr<ModelAPI_Result> theThis,
   if (ModelAPI_ResultPart::setDisabled(theThis, theFlag)) {
     DocumentPtr aDoc = Model_ResultPart::partDoc();
     if (aDoc.get()) {
+      // 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);
-      } else { // enabled, so make the current feature the last inside of this document
-        FeaturePtr aLastFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aDoc->object(
-          ModelAPI_Feature::group(), aDoc->size(ModelAPI_Feature::group()) - 1));
-        aDoc->setCurrentFeature(aLastFeature, false);
       }
     }
     return true;
@@ -89,29 +92,79 @@ bool Model_ResultPart::setDisabled(std::shared_ptr<ModelAPI_Result> theThis,
 
 std::shared_ptr<GeomAPI_Shape> Model_ResultPart::shape()
 {
-  DocumentPtr aDoc = Model_ResultPart::partDoc();
-  if (aDoc.get()) {
-    const std::string& aBodyGroup = ModelAPI_ResultBody::group();
-    TopoDS_Compound aResultComp;
-    BRep_Builder aBuilder;
-    aBuilder.MakeCompound(aResultComp);
-    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()) {
-        TopoDS_Shape aShape = *(aBody->shape()->implPtr<TopoDS_Shape>());
-        if (!aShape.IsNull()) {
-          aBuilder.Add(aResultComp, aShape);
-          aNumSubs++;
+  if (myShape.IsNull()) {
+    DocumentPtr aDoc = Model_ResultPart::partDoc();
+    if (aDoc.get()) {
+      const std::string& aBodyGroup = ModelAPI_ResultBody::group();
+      TopoDS_Compound aResultComp;
+      BRep_Builder aBuilder;
+      aBuilder.MakeCompound(aResultComp);
+      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()) {
+          TopoDS_Shape aShape = *(aBody->shape()->implPtr<TopoDS_Shape>());
+          if (!aShape.IsNull()) {
+            aBuilder.Add(aResultComp, aShape);
+            aNumSubs++;
+          }
         }
       }
+      if (aNumSubs) {
+        myShape = aResultComp;
+      }
     }
-    if (aNumSubs) {
-      std::shared_ptr<GeomAPI_Shape> aResult(new GeomAPI_Shape);
-      aResult->setImpl(new TopoDS_Shape(aResultComp));
-      return aResult;
+  }
+  if (myShape.IsNull())
+    return std::shared_ptr<GeomAPI_Shape>();
+  std::shared_ptr<GeomAPI_Shape> aResult(new GeomAPI_Shape);
+  aResult->setImpl(new TopoDS_Shape(myShape));
+  return aResult;
+}
+
+#include <Model_Document.h>
+#include <TNaming_Tool.hxx>
+#include <TNaming_NamedShape.hxx>
+#include <TNaming_Iterator.hxx>
+#include <TDataStd_Name.hxx>
+
+std::string Model_ResultPart::nameInPart(const std::shared_ptr<GeomAPI_Shape>& theShape)
+{
+  TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
+  // 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
+    return "";
+  TDF_Label anAccessLabel = aDoc->generalLabel();
+
+  std::string aName;
+  // check if the subShape is already in DF
+  Handle(TNaming_NamedShape) aNS = TNaming_Tool::NamedShape(aShape, anAccessLabel);
+  Handle(TDataStd_Name) anAttr;
+  if(!aNS.IsNull() && !aNS->IsEmpty()) { // in the document    
+    if(aNS->Label().FindAttribute(TDataStd_Name::GetID(), anAttr)) {
+      aName = TCollection_AsciiString(anAttr->Get()).ToCString();
+      if(!aName.empty()) {         
+        const TDF_Label& aLabel = aDoc->findNamingName(aName);
+
+        static const std::string aPostFix("_");
+        TNaming_Iterator anItL(aNS);
+        for(int i = 1; anItL.More(); anItL.Next(), i++) {
+          if(anItL.NewShape() == aShape) {
+            aName += aPostFix;
+            aName += TCollection_AsciiString (i).ToCString();
+            break;
+          }
+        }
+      }        
     }
   }
+  return aName;
+}
+
+std::shared_ptr<GeomAPI_Shape> Model_ResultPart::shapeInPart(const std::string& theName)
+{
+  /// TODO: not implemented yet
   return std::shared_ptr<GeomAPI_Shape>();
 }