Correctly set the save contents type when opening a class.

OpenModelica · Nov 25, 2015 · 1978c95 · 1978c95
1 parent 734e3df
commit 1978c95
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/OMEdit/OMEditGUI/Modeling/LibraryTreeWidget.cpp b/OMEdit/OMEditGUI/Modeling/LibraryTreeWidget.cpp
@@ -341,11 +341,19 @@ LibraryTreeItem::LibraryTreeItem(LibraryType type, QString text, QString nameStr
   setIsDocumentationClass(false);
   if (isFilePathValid()) {
     QFileInfo fileInfo(getFileName());
-    // if item has file name as package.mo then its save folder structure
-    if (fileInfo.fileName().compare("package.mo") == 0) {
+    // if item has file name as package.mo and is top level then its save folder structure
+    if (isTopLevel() && (fileInfo.fileName().compare("package.mo") == 0)) {
       setSaveContentsType(LibraryTreeItem::SaveFolderStructure);
-    } else {
+    } else if (isTopLevel()) {
       setSaveContentsType(LibraryTreeItem::SaveInOneFile);
+    } else {
+      if (mpParentLibraryTreeItem->getFileName().compare(getFileName()) == 0) {
+        setSaveContentsType(LibraryTreeItem::SaveInOneFile);
+      } else if (fileInfo.fileName().compare("package.mo") == 0) {
+        setSaveContentsType(LibraryTreeItem::SaveFolderStructure);
+      } else {
+        setSaveContentsType(LibraryTreeItem::SaveInOneFile);
+      }
     }
   }
   setClassText("");