make origin work correct with multiple parts

src/Mod/PartDesign/Gui/TaskTransformedParameters.cpp

@@ -24,7 +24,7 @@
 
 #ifndef _PreComp_
 # include <QMessageBox>
-# include <QListWidgetItem>
+# include <QListWidget>
 # include <TopoDS_Shape.hxx>
 # include <TopoDS_Face.hxx>
 # include <TopoDS.hxx>
@@ -36,6 +36,10 @@
 #include "Workbench.h"
 #include <App/Application.h>
 #include <App/Document.h>
+#include <App/Part.h>
+#include <App/Origin.h>
+#include <App/Plane.h>
+#include <App/Line.h>
 #include <Gui/Application.h>
 #include <Gui/Document.h>
 #include <Gui/BitmapFactory.h>
@@ -174,6 +178,49 @@ PartDesign::Transformed *TaskTransformedParameters::getObject() const
         return static_cast<PartDesign::Transformed*>(TransformedView->getObject());
 }
 
+App::DocumentObject* TaskTransformedParameters::getPartPlanes(const char* str) const {
+
+    //TODO: Adjust to GRAPH handling when available
+    App::DocumentObject* obj = getObject();
+    App::Part* part = getPartFor(obj, true);
+    
+    std::vector<App::DocumentObject*> origs = part->getObjectsOfType(App::Origin::getClassTypeId());
+    if(origs.size()<1)
+        return nullptr;
+    
+    App::Origin* orig = static_cast<App::Origin*>(origs[0]);
+    auto planes = orig->getObjectsOfType(App::Plane::getClassTypeId());
+    for(App::DocumentObject* plane : planes) {
+        
+        if( strcmp(static_cast<App::Plane*>(plane)->PlaneType.getValue(), str) == 0)
+            return plane;
+    }
+    
+    return nullptr;
+}
+
+
+App::DocumentObject* TaskTransformedParameters::getPartLines(const char* str) const {
+
+    //TODO: Adjust to GRAPH handling when available
+    App::DocumentObject* obj = getObject();
+    App::Part* part = getPartFor(obj, true);
+    
+    std::vector<App::DocumentObject*> origs = part->getObjectsOfType(App::Origin::getClassTypeId());
+    if(origs.size()<1)
+        return nullptr;
+    
+    App::Origin* orig = static_cast<App::Origin*>(origs[0]);
+    auto lines = orig->getObjectsOfType(App::Line::getClassTypeId());
+    for(App::DocumentObject* line : lines) {
+        
+        if( strcmp(static_cast<App::Line*>(line)->LineType.getValue(), str) == 0)
+            return line;
+    }
+    
+    return nullptr;
+}
+
 void TaskTransformedParameters::recomputeFeature()
 {
     if (insideMultiTransform) {