Fixed EDKT400.

[mirror_edk2.git] / Tools / Java / Source / FrameworkWizard / src / org / tianocore / frameworkwizard / workspace / WorkspaceTools.java

diff --git a/Tools/Java/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/workspace/WorkspaceTools.java b/Tools/Java/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/workspace/WorkspaceTools.java
index c27b81eea0c029dd5a1d402f2f48396150dcb400..9db0fe587e15bae3d21e20489d08cde889c48f26 100644 (file)
--- a/Tools/Java/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/workspace/WorkspaceTools.java
+++ b/Tools/Java/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/workspace/WorkspaceTools.java
@@ -39,6 +39,7 @@ import org.tianocore.frameworkwizard.common.SaveFile;
 import org.tianocore.frameworkwizard.common.Sort;\r
 import org.tianocore.frameworkwizard.common.Tools;\r
 import org.tianocore.frameworkwizard.common.Identifications.Identification;\r
+import org.tianocore.frameworkwizard.common.Identifications.OpeningPackageType;\r
 import org.tianocore.frameworkwizard.far.FarHeader;\r
 import org.tianocore.frameworkwizard.far.FarIdentification;\r
 import org.tianocore.frameworkwizard.module.Identifications.ModuleIdentification;\r
@@ -219,6 +220,32 @@ public class WorkspaceTools {
         return GlobalData.vPackageList;\r
     }\r
 \r
+    /**\r
+     Get all package which match parameter isRepackagable\r
+     \r
+     @param isRepackagable\r
+     @return\r
+     \r
+     **/\r
+    public Vector<PackageIdentification> getAllRepackagablePackages() {\r
+        Vector<PackageIdentification> v = new Vector<PackageIdentification>();\r
+        for (int index = 0; index < GlobalData.openingPackageList.size(); index++) {\r
+            OpeningPackageType opt = GlobalData.openingPackageList.getOpeningPackageByIndex(index);\r
+            if (opt.getXmlSpd() != null) {\r
+                if (opt.getXmlSpd().getPackageDefinitions() != null) {\r
+                    if (opt.getXmlSpd().getPackageDefinitions().getRePackage()) {\r
+                        v.addElement(opt.getId());\r
+                    }\r
+                } else {\r
+                    v.addElement(opt.getId());\r
+                }\r
+            } else {\r
+                v.addElement(opt.getId());\r
+            }\r
+        }\r
+        return v;\r
+    }\r
+\r
     public Vector<FarIdentification> getAllFars() {\r
 \r
         Vector<FarIdentification> v = new Vector<FarIdentification>();\r
@@ -471,7 +498,7 @@ public class WorkspaceTools {
                         if (type.elementAt(indexOfType).equals(DataType.PCD_ITEM_TYPE_DYNAMIC)) {\r
                             hasDYNAMIC = true;\r
                         }\r
-                        if(type.elementAt(indexOfType).equals(DataType.PCD_ITEM_TYPE_FEATURE_FLAG)) {\r
+                        if (type.elementAt(indexOfType).equals(DataType.PCD_ITEM_TYPE_FEATURE_FLAG)) {\r
                             hasFEATURE_FLAG = true;\r
                         }\r
                     }\r
@@ -666,10 +693,11 @@ public class WorkspaceTools {
                 //\r
                 // Check id\r
                 //\r
-                if (mid.equals(id)) {\r
-                    return GlobalData.vPackageList.elementAt(indexI);\r
+                if (mid != null) {\r
+                    if (mid.equals(id)) {\r
+                        return GlobalData.vPackageList.elementAt(indexI);\r
+                    }\r
                 }\r
-\r
             }\r
         }\r
 \r
@@ -809,19 +837,10 @@ public class WorkspaceTools {
                     //\r
                     PackageIdentification pid = GlobalData.openingModuleList.getIdByPath(path).getPackageId();\r
                     PackageSurfaceArea spd = GlobalData.openingPackageList.getPackageSurfaceAreaFromId(pid);\r
-                    if (spd != null) {\r
-                        if (spd.getLibraryClassDeclarations() != null) {\r
-                            LibraryClassDeclarations lcdl = spd.getLibraryClassDeclarations();\r
-                            for (int indexOfLibOfSpd = 0; indexOfLibOfSpd < lcdl.sizeOfLibraryClassArray(); indexOfLibOfSpd++) {\r
-                                if (lcdl.getLibraryClassList().get(indexOfLibOfSpd).getName().equals(name)) {\r
-                                    v.addElement(Tools.convertPathToCurrentOsType(Tools.getFilePathOnly(pid.getPath())\r
-                                                                                  + DataType.FILE_SEPARATOR\r
-                                                                                  + lcdl.getLibraryClassList()\r
-                                                                                        .get(indexOfLibOfSpd)\r
-                                                                                        .getIncludeHeader()));\r
-                                }\r
-                            }\r
-                        }\r
+                    String headerFile = getHeaderFileFromPackageByLibraryClassName(spd, name);\r
+                    if (!Tools.isEmpty(headerFile)) {\r
+                        v.addElement(Tools.convertPathToCurrentOsType(Tools.getFilePathOnly(pid.getPath())\r
+                                                                      + DataType.FILE_SEPARATOR + headerFile));\r
                     }\r
                 }\r
             }\r
@@ -920,4 +939,20 @@ public class WorkspaceTools {
         }\r
         return v;\r
     }\r
+\r
+    public String getHeaderFileFromPackageByLibraryClassName(PackageSurfaceArea spd, String name) {\r
+        String headerFile = "";\r
+        if (spd != null) {\r
+            if (spd.getLibraryClassDeclarations() != null) {\r
+                LibraryClassDeclarations lcdl = spd.getLibraryClassDeclarations();\r
+                for (int indexOfLibOfSpd = 0; indexOfLibOfSpd < lcdl.sizeOfLibraryClassArray(); indexOfLibOfSpd++) {\r
+                    if (lcdl.getLibraryClassList().get(indexOfLibOfSpd).getName().equals(name)) {\r
+                        return lcdl.getLibraryClassList().get(indexOfLibOfSpd).getIncludeHeader();\r
+                    }\r
+                }\r
+            }\r
+        }\r
+\r
+        return headerFile;\r
+    }\r
 }\r