1. Restructure module description on main UI

[mirror_edk2.git] / Tools / Source / FrameworkWizard / src / org / tianocore / frameworkwizard / common / ui / ITree.java

diff --git a/Tools/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/common/ui/ITree.java b/Tools/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/common/ui/ITree.java
index 25da85fa818f81466d808fc2fd394865191a7b95..5b7ea5900c1c13ec55db54f22d79827faf25be97 100644 (file)
--- a/Tools/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/common/ui/ITree.java
+++ b/Tools/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/common/ui/ITree.java
@@ -238,12 +238,19 @@ public class ITree extends JTree {
             if (iNode.getId().equals(id) && iNode.getCategory() == category) {\r
                 return iNode;\r
             }\r
+            IDefaultMutableTreeNode childNode = getNodeById(iNode, id, category);\r
+            if (childNode != null) {\r
+                return childNode;\r
+            }\r
         }\r
         return null;\r
     }\r
     \r
     public TreePath getPathOfNode(IDefaultMutableTreeNode node) {\r
-        TreePath treePath = new TreePath(treeModel.getPathToRoot(node));\r
-        return treePath;\r
+        if (node != null) {\r
+            TreePath treePath = new TreePath(treeModel.getPathToRoot(node));\r
+            return treePath;    \r
+        }\r
+        return null;\r
     }\r
 }\r