Issue #1648: Dump Python in the High Level Parameterized Geometry API

[modules/shaper.git] / src / FeaturesAPI / FeaturesAPI_Pipe.cpp

diff --git a/src/FeaturesAPI/FeaturesAPI_Pipe.cpp b/src/FeaturesAPI/FeaturesAPI_Pipe.cpp
index 3ca9a4e54e78263ea85fd030522843c8feeee35e..f99cccef2a3d2630b9e76c3b89649855420a33c8 100644 (file)
--- a/src/FeaturesAPI/FeaturesAPI_Pipe.cpp
+++ b/src/FeaturesAPI/FeaturesAPI_Pipe.cpp
@@ -6,6 +6,7 @@
 
 #include "FeaturesAPI_Pipe.h"
 
+#include <ModelHighAPI_Dumper.h>
 #include <ModelHighAPI_Tools.h>
 
 //==================================================================================================
@@ -110,6 +111,35 @@ void FeaturesAPI_Pipe::setByBasePathLocations(const std::list<ModelHighAPI_Selec
   execute();
 }
 
+//==================================================================================================
+void FeaturesAPI_Pipe::dump(ModelHighAPI_Dumper& theDumper) const
+{
+  FeaturePtr aBase = feature();
+  const std::string& aDocName = theDumper.name(aBase->document());
+
+  AttributeSelectionListPtr anAttrObjects = aBase->selectionList(FeaturesPlugin_Pipe::BASE_OBJECTS_ID());
+  AttributeSelectionPtr anAttrPath = aBase->selection(FeaturesPlugin_Pipe::PATH_OBJECT_ID());
+
+  theDumper << aBase << " = model.addPipe(" << aDocName << ", "
+            << anAttrObjects << ", " << anAttrPath;
+
+  std::string aCreationMethod = aBase->string(FeaturesPlugin_Pipe::CREATION_METHOD())->value();
+
+  if(aCreationMethod == FeaturesPlugin_Pipe::CREATION_METHOD_SIMPLE()) {
+    // Do nothing;
+  } else if(aCreationMethod == FeaturesPlugin_Pipe::CREATION_METHOD_BINORMAL()) {
+    AttributeSelectionPtr anAttrBiNormal = aBase->selection(FeaturesPlugin_Pipe::BINORMAL_ID());
+
+    theDumper << ", " << anAttrBiNormal;
+  } else if(aCreationMethod == FeaturesPlugin_Pipe::CREATION_METHOD_LOCATIONS()) {
+    AttributeSelectionListPtr anAttrLocations = aBase->selectionList(FeaturesPlugin_Pipe::LOCATIONS_ID());
+
+    theDumper << ", " << anAttrLocations;
+  }
+
+  theDumper << ")" << std::endl;
+}
+
 //==================================================================================================
 PipePtr addPipe(const std::shared_ptr<ModelAPI_Document>& thePart,
                 const std::list<ModelHighAPI_Selection>& theBaseObjects,