Initial part document shapes management implementation.

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

diff --git a/src/Model/Model_ResultPart.cpp b/src/Model/Model_ResultPart.cpp
index dfd8c85949c0756870bc22422489135182774dd2..bdae260d67d2a865c8496a275504b7530f138609 100644 (file)
--- a/src/Model/Model_ResultPart.cpp
+++ b/src/Model/Model_ResultPart.cpp
@@ -9,6 +9,10 @@
 #include <ModelAPI_AttributeDocRef.h>
 #include <ModelAPI_Session.h>
 #include <ModelAPI_Feature.h>
+#include <ModelAPI_ResultBody.h>
+
+#include <TopoDS_Compound.hxx>
+#include <BRep_Builder.hxx>
 
 std::shared_ptr<ModelAPI_Document> Model_ResultPart::partDoc()
 {
@@ -82,3 +86,31 @@ bool Model_ResultPart::setDisabled(std::shared_ptr<ModelAPI_Result> theThis,
   }
   return false;
 }
+
+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 (aNumSubs) {
+      std::shared_ptr<GeomAPI_Shape> aResult(new GeomAPI_Shape);
+      aResult->setImpl(new TopoDS_Shape(aResultComp));
+      return aResult;
+    }
+  }
+  return std::shared_ptr<GeomAPI_Shape>();
+}