Import: fix STEP export crash on vertex only shape

realthunder · Nov 12, 2022 · c64e4a4 · c64e4a4
1 parent 0e8deb6
commit c64e4a4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/Mod/Import/App/ImportOCAF2.cpp b/src/Mod/Import/App/ImportOCAF2.cpp
@@ -1246,7 +1246,8 @@ TDF_Label ExportOCAF2::exportObject(App::DocumentObject* parentObj,
         const char *sub, TDF_Label parent, const char *name) 
 {
     App::DocumentObject *obj;
-    auto shape = Part::Feature::getTopoShape(parentObj,sub,false,0,&obj,false,!sub);
+    bool needSubElement = sub && Data::ComplexGeoData::findElementName(sub) == sub;
+    auto shape = Part::Feature::getTopoShape(parentObj,sub,needSubElement,0,&obj,false,!sub);
     if(!obj || shape.isNull()) {
         if (obj)
             FC_WARN(obj->getFullName() << " has null shape");
@@ -1314,7 +1315,9 @@ TDF_Label ExportOCAF2::exportObject(App::DocumentObject* parentObj,
         links.push_back(linked);
     }
 
-    auto subs = obj->getSubObjects();
+    std::vector<std::string> subs;
+    if (!needSubElement)
+        subs = obj->getSubObjects();
     // subs empty means obj is not a container.
     if(subs.empty()) {