Avoid crash in case of non-initialized document

diff --git a/src/PartSet/PartSet_TreeNodes.cpp b/src/PartSet/PartSet_TreeNodes.cpp
index bf050f7237d4c625202e73a48216bed77b007db4..5870827053098a206022edf6b9dcb072494273c1 100644 (file)
--- a/src/PartSet/PartSet_TreeNodes.cpp
+++ b/src/PartSet/PartSet_TreeNodes.cpp
@@ -441,17 +441,19 @@ QVariant PartSet_FolderNode::data(int theColumn, int theRole) const
     }
   }
   if ((theColumn == 2) && (theRole == Qt::DecorationRole)) {
-    FeaturePtr aFeature = document()->currentFeature(true);
-    if (!aFeature.get()) { // There is no current feature
-      ModuleBase_ITreeNode* aLastFolder = 0;
-      foreach(ModuleBase_ITreeNode* aNode, parent()->children()) {
-        if (aNode->type() == PartSet_FolderNode::typeId())
-          aLastFolder = aNode;
-        else
-          break;
+    if (document().get()) {
+      FeaturePtr aFeature = document()->currentFeature(true);
+      if (!aFeature.get()) { // There is no current feature
+        ModuleBase_ITreeNode* aLastFolder = 0;
+        foreach(ModuleBase_ITreeNode* aNode, parent()->children()) {
+          if (aNode->type() == PartSet_FolderNode::typeId())
+            aLastFolder = aNode;
+          else
+            break;
+        }
+        if (aLastFolder == this)
+          return QIcon(":pictures/arrow.png");
       }
-      if (aLastFolder == this)
-        return QIcon(":pictures/arrow.png");
     }
   }
   return PartSet_TreeNode::data(theColumn, theRole);