]> git.proxmox.com Git - mirror_edk2.git/blobdiff - Tools/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/workspace/WorkspaceTools.java
1. Fix bug in get industry std headers
[mirror_edk2.git] / Tools / Source / FrameworkWizard / src / org / tianocore / frameworkwizard / workspace / WorkspaceTools.java
index e4800816cb6264b9e8ab66d188addc5a4fc6c1c4..46c325e3b3b8d004debbb006baaf2d18d9d40e93 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
@@ -35,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
@@ -43,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
@@ -56,17 +61,139 @@ 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
+        } catch (Exception e) {\r
+            e.printStackTrace();\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
+        } catch (Exception e) {\r
+            e.printStackTrace();\r
+        }\r
+    }\r
+\r
+    public String getPackageFarGuid(Identification 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 + item.getStringValue();\r
+            File tempFile = new File(path);\r
+            if (tempFile.getPath().equalsIgnoreCase(packageId.getPath())) {\r
+                return fdb.getPackageList().getFilenameArray(index).getFarGuid();\r
+            }\r
+        }\r
+\r
+        return null;\r
+    }\r
+\r
+    public String getPlatformFarGuid(Identification 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 + item.getStringValue();\r
+            File tempFile = new File(path);\r
+            if (tempFile.getPath().equalsIgnoreCase(platformId.getPath())) {\r
+                return fdb.getPlatformList().getFilenameArray(index).getFarGuid();\r
+            }\r
+        }\r
+\r
+        return null;\r
+    }\r
+\r
+    public String getModuleFarGuid(Identification moduleId) {\r
+        PackageIdentification packageId = getPackageIdByModuleId(moduleId);\r
+        if (packageId != null) {\r
+            return getPackageFarGuid(packageId);\r
+        } else {\r
+            return null;\r
         }\r
     }\r
 \r
@@ -97,34 +224,34 @@ public class WorkspaceTools {
         }\r
         return modulePath;\r
     }\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
+     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).getIncludeHeader();\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
@@ -159,20 +286,90 @@ public class WorkspaceTools {
         Tools.sortPackages(vPackageList, DataType.SORT_TYPE_ASCENDING);\r
         return vPackageList;\r
     }\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 + 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 + 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 module basic information from a package\r
-    \r
+     \r
      @param id Package id\r
      @return A vector includes all modules' basic information\r
-    \r
-    **/\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
+\r
         for (int index = 0; index < modulePaths.size(); index++) {\r
             try {\r
                 modulePath = modulePaths.get(index);\r
@@ -188,7 +385,7 @@ public class WorkspaceTools {
         }\r
         Tools.sortModules(v, DataType.SORT_TYPE_ASCENDING);\r
         return v;\r
-        \r
+\r
     }\r
 \r
     /**\r
@@ -214,7 +411,7 @@ public class WorkspaceTools {
         for (int indexI = 0; indexI < vPackageList.size(); indexI++) {\r
             packagePath = vPackageList.elementAt(indexI).getPath();\r
             modulePaths = this.getAllModulesOfPackage(packagePath);\r
-            \r
+\r
             for (int indexJ = 0; indexJ < modulePaths.size(); indexJ++) {\r
                 try {\r
                     modulePath = modulePaths.get(indexJ);\r
@@ -519,7 +716,7 @@ public class WorkspaceTools {
                 } catch (IOException e) {\r
 \r
                 } catch (XmlException e) {\r
-                \r
+\r
                 } catch (Exception e) {\r
 \r
                 }\r
@@ -673,7 +870,7 @@ public class WorkspaceTools {
         // First add package\r
         //\r
         v.addElement(path);\r
-        \r
+\r
         //\r
         // Add the package's industry std includes one by one\r
         //\r
@@ -682,7 +879,7 @@ public class WorkspaceTools {
         for (int index = 0; index < f1.size(); index++) {\r
             v.addElement(f1.get(index));\r
         }\r
-        \r
+\r
         //\r
         // Add module's files one by one\r
         //\r