]> git.proxmox.com Git - mirror_edk2.git/blobdiff - Tools/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/workspace/WorkspaceTools.java
Fix include .h file error.
[mirror_edk2.git] / Tools / Source / FrameworkWizard / src / org / tianocore / frameworkwizard / workspace / WorkspaceTools.java
index 73fba9a86140a8082b2c2758a1f95119d8592a3e..dda8844495e07b054947554a0aac52a34e713343 100644 (file)
  **/\r
 package org.tianocore.frameworkwizard.workspace;\r
 \r
+import java.io.File;\r
 import java.io.IOException;\r
+import java.util.List;\r
 import java.util.Vector;\r
 \r
+import org.apache.xmlbeans.XmlCursor;\r
 import org.apache.xmlbeans.XmlException;\r
 import org.tianocore.DbPathAndFilename;\r
-import org.tianocore.FrameworkDatabaseDocument;\r
+import org.tianocore.FrameworkDatabaseDocument.FrameworkDatabase;\r
+import org.tianocore.IndustryStdIncludesDocument.IndustryStdIncludes;\r
 import org.tianocore.ModuleSurfaceAreaDocument.ModuleSurfaceArea;\r
 import org.tianocore.MsaFilesDocument.MsaFiles;\r
 import org.tianocore.MsaHeaderDocument.MsaHeader;\r
@@ -34,6 +38,8 @@ import org.tianocore.frameworkwizard.common.SaveFile;
 import org.tianocore.frameworkwizard.common.Tools;\r
 import org.tianocore.frameworkwizard.common.Identifications.Identification;\r
 import org.tianocore.frameworkwizard.common.Identifications.OpenFile;\r
+import org.tianocore.frameworkwizard.far.FarHeader;\r
+import org.tianocore.frameworkwizard.far.FarIdentification;\r
 import org.tianocore.frameworkwizard.module.Identifications.ModuleIdentification;\r
 import org.tianocore.frameworkwizard.packaging.PackageIdentification;\r
 import org.tianocore.frameworkwizard.platform.PlatformIdentification;\r
@@ -42,7 +48,7 @@ public class WorkspaceTools {
     //\r
     // Define class members\r
     //\r
-    private FrameworkDatabaseDocument.FrameworkDatabase fdb = null;\r
+    private FrameworkDatabase fdb = null;\r
 \r
     private Vector<ModuleIdentification> vModuleList = new Vector<ModuleIdentification>();\r
 \r
@@ -55,20 +61,147 @@ public class WorkspaceTools {
      Open Framework Database file\r
      \r
      */\r
-    private void openFrameworkDb() {\r
+    private FrameworkDatabase openFrameworkDb() {\r
         String strFrameworkDbFilePath = Workspace.getCurrentWorkspace() + Workspace.getStrWorkspaceDatabaseFile();\r
         strFrameworkDbFilePath = Tools.convertPathToCurrentOsType(strFrameworkDbFilePath);\r
         try {\r
             fdb = OpenFile.openFrameworkDb(strFrameworkDbFilePath);\r
         } catch (XmlException e) {\r
             Log.err("Open Framework Database " + strFrameworkDbFilePath, e.getMessage());\r
-            return;\r
+            return null;\r
         } catch (Exception e) {\r
             Log.err("Open Framework Database " + strFrameworkDbFilePath, "Invalid file type");\r
-            return;\r
+            return null;\r
         }\r
+        return fdb;\r
     }\r
 \r
+    public void addFarToDb(List<String> packageList, List<String> platformList, FarHeader far) {\r
+      FrameworkDatabase fdb = openFrameworkDb();\r
+      \r
+      for (int i = 0; i < packageList.size(); i++) {\r
+        DbPathAndFilename item = DbPathAndFilename.Factory.newInstance();\r
+        item.setFarGuid(far.getGuidValue());\r
+        item.setStringValue(packageList.get(i));\r
+        fdb.getPackageList().getFilenameList().add(item);\r
+      }\r
+      \r
+      for (int i = 0; i < platformList.size(); i++) {\r
+        DbPathAndFilename item = DbPathAndFilename.Factory.newInstance();\r
+        item.setFarGuid(far.getGuidValue());\r
+        item.setStringValue(platformList.get(i));\r
+        fdb.getPlatformList().getFilenameList().add(item);\r
+      }\r
+      \r
+      DbPathAndFilename farItem = DbPathAndFilename.Factory.newInstance();\r
+      farItem.setFarGuid(far.getGuidValue());\r
+      farItem.setStringValue(far.getFarName());\r
+      fdb.getFarList().getFilenameList().add(farItem);\r
+      \r
+      String strFrameworkDbFilePath = Workspace.getCurrentWorkspace() + Workspace.getStrWorkspaceDatabaseFile();\r
+      strFrameworkDbFilePath = Tools.convertPathToCurrentOsType(strFrameworkDbFilePath);\r
+      \r
+      try {\r
+        SaveFile.saveDbFile(strFrameworkDbFilePath, fdb);\r
+      }\r
+      catch (Exception e) {\r
+        e.printStackTrace();\r
+      }\r
+    }\r
+    \r
+    \r
+    public void removeFarFromDb(FarIdentification far) {\r
+        FrameworkDatabase fdb = openFrameworkDb();\r
+        //\r
+        // Remove Packages\r
+        //\r
+        XmlCursor cursor = fdb.getPackageList().newCursor();\r
+        cursor.toFirstChild();\r
+        do {\r
+          DbPathAndFilename item = (DbPathAndFilename)cursor.getObject();\r
+          \r
+          if (item.getFarGuid() != null && item.getFarGuid().equalsIgnoreCase(far.getGuid())) {\r
+            cursor.removeXml();\r
+          }\r
+        } while (cursor.toNextSibling());\r
+        cursor.dispose();\r
+        \r
+        //\r
+        // Remove Platforms\r
+        //\r
+        cursor = fdb.getPlatformList().newCursor();\r
+        cursor.toFirstChild();\r
+        do {\r
+          DbPathAndFilename item = (DbPathAndFilename)cursor.getObject();\r
+          if (item.getFarGuid() != null && item.getFarGuid().equalsIgnoreCase(far.getGuid())) {\r
+            cursor.removeXml();\r
+          }\r
+        } while (cursor.toNextSibling());\r
+        \r
+        //\r
+        // Remove Far\r
+        //\r
+        cursor = fdb.getFarList().newCursor();\r
+        cursor.toFirstChild();\r
+        do {\r
+          DbPathAndFilename item = (DbPathAndFilename)cursor.getObject();\r
+          if (item.getFarGuid() != null && item.getFarGuid().equalsIgnoreCase(far.getGuid())) {\r
+            cursor.removeXml();\r
+          }\r
+        } while (cursor.toNextSibling());\r
+        cursor.dispose();\r
+        \r
+        String strFrameworkDbFilePath = Workspace.getCurrentWorkspace() + Workspace.getStrWorkspaceDatabaseFile();\r
+        strFrameworkDbFilePath = Tools.convertPathToCurrentOsType(strFrameworkDbFilePath);\r
+        try {\r
+          SaveFile.saveDbFile(strFrameworkDbFilePath, fdb);\r
+        }\r
+        catch (Exception e) {\r
+          e.printStackTrace();\r
+        }\r
+    }\r
+    \r
+    public String getPackageFarGuid(PackageIdentification packageId) {\r
+      openFrameworkDb();\r
+      \r
+      for (int index = 0; index < fdb.getPackageList().getFilenameList().size(); index++) {\r
+        DbPathAndFilename item = fdb.getPackageList().getFilenameArray(index);\r
+        String path = Workspace.getCurrentWorkspace() + DataType.FILE_SEPARATOR\r
+                      + item.getStringValue();\r
+        File tempFile = new File(path);\r
+        if (tempFile.getPath().equalsIgnoreCase(packageId.getSpdFile().getPath())) {\r
+          return fdb.getPackageList().getFilenameArray(index).getFarGuid();\r
+        }\r
+      }\r
+      \r
+      return null;\r
+    }\r
+    \r
+    public String getPlatformFarGuid(PlatformIdentification platformId) {\r
+      openFrameworkDb();\r
+      \r
+      for (int index = 0; index < fdb.getPlatformList().getFilenameList().size(); index++) {\r
+        DbPathAndFilename item = fdb.getPlatformList().getFilenameArray(index);\r
+        String path = Workspace.getCurrentWorkspace() + DataType.FILE_SEPARATOR\r
+                      + item.getStringValue();\r
+        File tempFile = new File(path);\r
+        if (tempFile.getPath().equalsIgnoreCase(platformId.getFpdFile().getPath())) {\r
+          return fdb.getPlatformList().getFilenameArray(index).getFarGuid();\r
+        }\r
+      }\r
+      \r
+      return null;\r
+    }\r
+    \r
+    public String getModuleFarGuid(ModuleIdentification moduleId) {\r
+      PackageIdentification packageId = getPackageIdByModuleId(moduleId);\r
+      if (packageId != null) {\r
+          return getPackageFarGuid(packageId);\r
+      }\r
+      else {\r
+        return null;\r
+      }\r
+    }\r
     /**\r
      Get all modules' paths from one package\r
      \r
@@ -88,14 +221,42 @@ public class WorkspaceTools {
                 }\r
             }\r
         } catch (IOException e) {\r
-            e.printStackTrace();\r
+\r
         } catch (XmlException e) {\r
-            e.printStackTrace();\r
+\r
         } catch (Exception e) {\r
-            e.printStackTrace();\r
+\r
         }\r
         return modulePath;\r
     }\r
+    \r
+    /**\r
+    Get all Industry Std Includes' paths from one package\r
+    \r
+    @return a Vector with all paths\r
+    \r
+    **/\r
+   public Vector<String> getAllIndustryStdIncludesOfPackage(String path) {\r
+       Vector<String> includePath = new Vector<String>();\r
+       try {\r
+           IndustryStdIncludes files = OpenFile.openSpdFile(path).getIndustryStdIncludes();\r
+           if (files != null) {\r
+               for (int index = 0; index < files.getIndustryStdHeaderList().size(); index++) {\r
+                   String temp = files.getIndustryStdHeaderList().get(index).getName();\r
+                   temp = Tools.addFileSeparator(Tools.getFilePathOnly(path)) + temp;\r
+                   temp = Tools.convertPathToCurrentOsType(temp);\r
+                   includePath.addElement(temp);\r
+               }\r
+           }\r
+       } catch (IOException e) {\r
+\r
+       } catch (XmlException e) {\r
+\r
+       } catch (Exception e) {\r
+\r
+       }\r
+       return includePath;\r
+   }\r
 \r
     /**\r
      Get all package basic information form the FrameworkDatabase.db file\r
@@ -118,22 +279,126 @@ public class WorkspaceTools {
                 vPackageList.addElement(new PackageIdentification(id));\r
             } catch (IOException e) {\r
                 Log.err("Open Package Surface Area " + path, e.getMessage());\r
-                e.printStackTrace();\r
+\r
             } catch (XmlException e) {\r
                 Log.err("Open Package Surface Area " + path, e.getMessage());\r
-                e.printStackTrace();\r
+\r
             } catch (Exception e) {\r
                 Log.err("Open Package Surface Area " + path, "Invalid file type");\r
-                e.printStackTrace();\r
+\r
             }\r
         }\r
+        Tools.sortPackages(vPackageList, DataType.SORT_TYPE_ASCENDING);\r
         return vPackageList;\r
     }\r
 \r
+    public Vector<FarIdentification> getAllFars() {\r
+      openFrameworkDb();\r
+      Vector<FarIdentification> v = new Vector<FarIdentification>();\r
+      for (int index = 0; index < fdb.getFarList().getFilenameList().size(); index++) {\r
+        DbPathAndFilename item = fdb.getFarList().getFilenameList().get(index);\r
+        FarIdentification far = new FarIdentification(item.getFarGuid(), item.getMd5Sum(), item.getStringValue());\r
+        v.addElement(far);\r
+      }\r
+      return v;\r
+    }\r
+    \r
+    public Vector<PackageIdentification> getPackagesByFar(FarIdentification far) {\r
+      Identification id = null;\r
+      openFrameworkDb();\r
+      Vector<PackageIdentification> v = new Vector<PackageIdentification>();\r
+      \r
+      for (int index = 0; index < fdb.getPackageList().getFilenameList().size(); index++) {\r
+        DbPathAndFilename item = fdb.getPackageList().getFilenameArray(index);\r
+        String path = Workspace.getCurrentWorkspace() + DataType.FILE_SEPARATOR\r
+                      + item.getStringValue();\r
+        path = Tools.convertPathToCurrentOsType(path);\r
+        \r
+        if (item.getFarGuid() != null && item.getFarGuid().equalsIgnoreCase(far.getGuid())) {\r
+        \r
+          try {\r
+            id = getId(path, OpenFile.openSpdFile(path));\r
+            v.addElement(new PackageIdentification(id));\r
+          } catch (IOException e) {\r
+            Log.err("Open Package Surface Area " + path, e.getMessage());\r
+            e.printStackTrace();\r
+          } catch (XmlException e) {\r
+            Log.err("Open Package Surface Area " + path, e.getMessage());\r
+            e.printStackTrace();\r
+          } catch (Exception e) {\r
+            Log.err("Open Package Surface Area " + path, "Invalid file type");\r
+            e.printStackTrace();\r
+          }\r
+        }\r
+      }\r
+      return v;\r
+    }\r
+    \r
+    public Vector<PlatformIdentification> getPlatformsByFar(FarIdentification far) {\r
+      Identification id = null;\r
+      openFrameworkDb();\r
+      Vector<PlatformIdentification> v = new Vector<PlatformIdentification>();\r
+      \r
+      for (int index = 0; index < fdb.getPlatformList().getFilenameList().size(); index++) {\r
+        DbPathAndFilename item = fdb.getPlatformList().getFilenameArray(index);\r
+        String path = Workspace.getCurrentWorkspace() + DataType.FILE_SEPARATOR\r
+                      + item.getStringValue();\r
+        path = Tools.convertPathToCurrentOsType(path);\r
+        \r
+        if (item.getFarGuid() != null && item.getFarGuid().equalsIgnoreCase(far.getGuid())) {\r
+          try {\r
+            id = getId(path, OpenFile.openFpdFile(path));\r
+            v.addElement(new PlatformIdentification(id));\r
+          } catch (IOException e) {\r
+            Log.err("Open Platform Surface Area " + path, e.getMessage());\r
+            e.printStackTrace();\r
+          } catch (XmlException e) {\r
+            Log.err("Open Platform Surface Area " + path, e.getMessage());\r
+            e.printStackTrace();\r
+          } catch (Exception e) {\r
+            Log.err("Open Platform Surface Area " + path, "Invalid file type");\r
+            e.printStackTrace();\r
+          }\r
+        }\r
+    }\r
+      return v;\r
+    }\r
+\r
     /**\r
-     Get all package basic information form the FrameworkDatabase.db file\r
+     Get all module basic information from a package\r
+    \r
+     @param id Package id\r
+     @return A vector includes all modules' basic information\r
+    \r
+    **/\r
+    public Vector<ModuleIdentification> getAllModules(PackageIdentification pid) {\r
+        Vector<ModuleIdentification> v = new Vector<ModuleIdentification>();\r
+        Vector<String> modulePaths = this.getAllModulesOfPackage(pid.getPath());\r
+        Identification id = null;\r
+        String modulePath = null;\r
+        \r
+        for (int index = 0; index < modulePaths.size(); index++) {\r
+            try {\r
+                modulePath = modulePaths.get(index);\r
+                id = getId(modulePath, OpenFile.openMsaFile(modulePath));\r
+            } catch (IOException e) {\r
+                Log.log("Error when Open Module Surface Area " + modulePath, e.getMessage());\r
+            } catch (XmlException e) {\r
+                Log.log("Error when Open Module Surface Area " + modulePath, e.getMessage());\r
+            } catch (Exception e) {\r
+                Log.log("Error when Open Module Surface Area " + modulePath, "Invalid file type " + e.getMessage());\r
+            }\r
+            v.addElement(new ModuleIdentification(id, pid));\r
+        }\r
+        Tools.sortModules(v, DataType.SORT_TYPE_ASCENDING);\r
+        return v;\r
+        \r
+    }\r
+\r
+    /**\r
+     Get all module basic information form the FrameworkDatabase.db file\r
      \r
-     @return vPackageList A vector includes all packages' basic information\r
+     @return vModuleList A vector includes all modules' basic information\r
      \r
      */\r
     public Vector<ModuleIdentification> getAllModules() {\r
@@ -161,17 +426,14 @@ public class WorkspaceTools {
                     vModuleList.addElement(new ModuleIdentification(id, vPackageList.elementAt(indexI)));\r
                 } catch (IOException e) {\r
                     Log.err("Open Module Surface Area " + modulePath, e.getMessage());\r
-                    e.printStackTrace();\r
                 } catch (XmlException e) {\r
                     Log.err("Open Module Surface Area " + modulePath, e.getMessage());\r
-                    e.printStackTrace();\r
                 } catch (Exception e) {\r
                     Log.err("Open Module Surface Area " + modulePath, "Invalid file type");\r
-                    e.printStackTrace();\r
                 }\r
             }\r
         }\r
-\r
+        Tools.sortModules(vModuleList, DataType.SORT_TYPE_ASCENDING);\r
         return vModuleList;\r
     }\r
 \r
@@ -196,15 +458,13 @@ public class WorkspaceTools {
                 vPlatformList.addElement(new PlatformIdentification(id));\r
             } catch (IOException e) {\r
                 Log.err("Open Platform Surface Area " + path, e.getMessage());\r
-                e.printStackTrace();\r
             } catch (XmlException e) {\r
                 Log.err("Open Platform Surface Area " + path, e.getMessage());\r
-                e.printStackTrace();\r
             } catch (Exception e) {\r
                 Log.err("Open Platform Surface Area " + path, "Invalid file type");\r
-                e.printStackTrace();\r
             }\r
         }\r
+        Tools.sortPlatforms(vPlatformList, DataType.SORT_TYPE_ASCENDING);\r
         return vPlatformList;\r
     }\r
 \r
@@ -222,6 +482,7 @@ public class WorkspaceTools {
                 }\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -239,6 +500,7 @@ public class WorkspaceTools {
                 }\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -256,6 +518,7 @@ public class WorkspaceTools {
                 }\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -273,6 +536,7 @@ public class WorkspaceTools {
                 }\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -290,6 +554,7 @@ public class WorkspaceTools {
                 }\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -311,15 +576,13 @@ public class WorkspaceTools {
                 }\r
             } catch (IOException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (XmlException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (Exception e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -339,15 +602,13 @@ public class WorkspaceTools {
                 }\r
             } catch (IOException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (XmlException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (Exception e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -367,15 +628,13 @@ public class WorkspaceTools {
                 }\r
             } catch (IOException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (XmlException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (Exception e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -395,15 +654,13 @@ public class WorkspaceTools {
                 }\r
             } catch (IOException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (XmlException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (Exception e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -423,15 +680,13 @@ public class WorkspaceTools {
                 }\r
             } catch (IOException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (XmlException e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             } catch (Exception e) {\r
                 // TODO Auto-generated catch block\r
-                e.printStackTrace();\r
             }\r
         }\r
+        Tools.sortVectorString(vector, DataType.SORT_TYPE_ASCENDING);\r
         return vector;\r
     }\r
 \r
@@ -466,11 +721,11 @@ public class WorkspaceTools {
                         return vPackageList.elementAt(indexI);\r
                     }\r
                 } catch (IOException e) {\r
-                    e.printStackTrace();\r
+\r
                 } catch (XmlException e) {\r
-                    e.printStackTrace();\r
+                \r
                 } catch (Exception e) {\r
-                    e.printStackTrace();\r
+\r
                 }\r
             }\r
         }\r
@@ -583,7 +838,7 @@ public class WorkspaceTools {
      @throws Exception\r
      \r
      **/\r
-    public Vector<String> getAllModuleFiles(String path) throws IOException, XmlException, Exception {\r
+    public Vector<String> getAllModuleFilesPath(String path) throws IOException, XmlException, Exception {\r
         Vector<String> v = new Vector<String>();\r
         path = Tools.convertPathToCurrentOsType(path);\r
         v.addElement(path);\r
@@ -615,7 +870,7 @@ public class WorkspaceTools {
      @throws Exception\r
      \r
      **/\r
-    public Vector<String> getAllPakcageFiles(String path) throws IOException, XmlException, Exception {\r
+    public Vector<String> getAllPakcageFilesPath(String path) throws IOException, XmlException, Exception {\r
         Vector<String> v = new Vector<String>();\r
         path = Tools.convertPathToCurrentOsType(path);\r
         //\r
@@ -624,17 +879,25 @@ public class WorkspaceTools {
         v.addElement(path);\r
         \r
         //\r
-        // Add module's files one by one\r
+        // Add the package's industry std includes one by one\r
         //\r
         Vector<String> f1 = new Vector<String>();\r
+        f1 = getAllIndustryStdIncludesOfPackage(path);\r
+        for (int index = 0; index < f1.size(); index++) {\r
+            v.addElement(f1.get(index));\r
+        }\r
+        \r
+        //\r
+        // Add module's files one by one\r
+        //\r
+        f1 = new Vector<String>();\r
         f1 = getAllModulesOfPackage(path);\r
         for (int indexI = 0; indexI < f1.size(); indexI++) {\r
-            Vector<String> f2 = getAllModuleFiles(f1.get(indexI));\r
+            Vector<String> f2 = getAllModuleFilesPath(f1.get(indexI));\r
             for (int indexJ = 0; indexJ < f2.size(); indexJ++) {\r
                 v.addElement(f2.get(indexJ));\r
             }\r
         }\r
-        //v.add(0, path);\r
 \r
         return v;\r
     }\r