]> git.proxmox.com Git - mirror_edk2.git/blobdiff - Tools/Java/Source/PcdTools/org/tianocore/pcd/entity/MemoryDatabaseManager.java
Restructuring for better separation of Tool packages.
[mirror_edk2.git] / Tools / Java / Source / PcdTools / org / tianocore / pcd / entity / MemoryDatabaseManager.java
diff --git a/Tools/Java/Source/PcdTools/org/tianocore/pcd/entity/MemoryDatabaseManager.java b/Tools/Java/Source/PcdTools/org/tianocore/pcd/entity/MemoryDatabaseManager.java
new file mode 100644 (file)
index 0000000..f7c08f5
--- /dev/null
@@ -0,0 +1,313 @@
+/** @file\r
+  MemoryDatabaseManager class.\r
+\r
+  Database hold all PCD information comes from SPD, MSA, FPD file in memory.\r
+\r
+Copyright (c) 2006, Intel Corporation\r
+All rights reserved. This program and the accompanying materials\r
+are licensed and made available under the terms and conditions of the BSD License\r
+which accompanies this distribution.  The full text of the license may be found at\r
+http://opensource.org/licenses/bsd-license.php\r
+\r
+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+\r
+**/\r
+package org.tianocore.pcd.entity;\r
+\r
+import java.util.ArrayList;\r
+import java.util.HashMap;\r
+import java.util.List;\r
+import java.util.Map;\r
+\r
+import org.tianocore.pcd.entity.UsageIdentification;\r
+import org.tianocore.pcd.exception.EntityException;\r
+\r
+/**\r
+  Database hold all PCD information comes from SPD, MSA, FPD file in memory.\r
+**/\r
+public class MemoryDatabaseManager {\r
+    ///\r
+    ///  Memory database. The string "cName + SpaceNameGuid" is primary key.\r
+    ///  memory database is in global scope, and it will be used for others PCD tools.\r
+    ///\r
+    private static Map<String, Token>  memoryDatabase       = null;\r
+\r
+    ///\r
+    /// Before build a module, the used libary will be build firstly, the PCD of these\r
+    /// library is inheritted by the module, so stored module's PCD information as PCD\r
+    /// context of building libary.\r
+    ///\r
+    public static List<UsageInstance> UsageInstanceContext = null;\r
+\r
+    ///\r
+    /// Current module name, if now is buiding library, this value indicate this library\r
+    /// is for building what module.\r
+    ///\r
+    public static String CurrentModuleName                 = null;\r
+\r
+    ///\r
+    /// String for PCD PEIM and DXE autogen file\r
+    ///\r
+    public static String PcdPeimHString                    = "";\r
+    public static String PcdPeimCString                    = "";\r
+    public static String PcdDxeHString                     = "";\r
+    public static String PcdDxeCString                     = "";\r
+\r
+    /**\r
+      Constructure function\r
+    **/\r
+    public MemoryDatabaseManager() {\r
+        //\r
+        // Allocate memory for new database in global scope.\r
+        //\r
+        if (memoryDatabase == null) {\r
+            memoryDatabase = new HashMap<String, Token>();\r
+        }\r
+    }\r
+\r
+    /**\r
+      Judege whether token exists in memory database\r
+\r
+      @param primaryKey    the primaryKey for searching token\r
+\r
+      @retval  TRUE  - token already exist in database.\r
+      @retval  FALSE - token does not exist in database.\r
+    **/\r
+    public boolean isTokenInDatabase(String primaryKey) {\r
+        return (memoryDatabase.get(primaryKey) != null);\r
+    }\r
+\r
+    /**\r
+      Add a pcd token into memory database.\r
+\r
+      @param primaryKey   the primary key for searching token\r
+      @param token        token instance\r
+    **/\r
+    public void addTokenToDatabase(String primaryKey, Token token) {\r
+        memoryDatabase.put(primaryKey, token);\r
+    }\r
+\r
+    /**\r
+      Get a token instance from memory database with primary key.\r
+\r
+      @param primaryKey   the primary key for searching token\r
+\r
+      @return token instance.\r
+    **/\r
+    public Token getTokenByKey(String primaryKey) {\r
+        return memoryDatabase.get(primaryKey);\r
+    }\r
+\r
+    /**\r
+      Get the number of PCD token record in memory database.\r
+\r
+      @return the number of PCD token record in memory database.\r
+    **/\r
+    public int getDBSize() {\r
+        return memoryDatabase.size();\r
+    }\r
+\r
+    /**\r
+      Get the token record array contained all PCD token in memory database.\r
+\r
+      @return the token record array contained all PCD token in memory database.\r
+    **/\r
+    public Token[] getRecordArray() {\r
+        Token[]     tokenArray  = null;\r
+        Object[]    dataArray   = null;\r
+        Map.Entry   entry       = null;\r
+        int         index       = 0;\r
+        int         size        = 0;\r
+\r
+        if (memoryDatabase == null) {\r
+            return null;\r
+        }\r
+        dataArray  = memoryDatabase.entrySet().toArray();\r
+        size       = memoryDatabase.size();\r
+        tokenArray = new Token[memoryDatabase.size()];\r
+        for (index = 0; index < size; index++) {\r
+            entry =(Map.Entry) dataArray [index];\r
+            tokenArray[index] =(Token) entry.getValue();\r
+        }\r
+        return tokenArray;\r
+    }\r
+\r
+    /**\r
+       Get record array only contains DYNAMIC or DYNAMIC_EX type PCD.\r
+\r
+       @return ArrayList the array list contains all dynamic type PCD.\r
+    **/\r
+    private ArrayList getDynamicRecordArray() {\r
+        Token[]     tokenArray  =   getRecordArray();\r
+        int         index       =   0;\r
+        ArrayList<Token>   al   =   new ArrayList<Token>();\r
+\r
+        for (index = 0; index < tokenArray.length; index++) {\r
+            if (tokenArray[index].isDynamicPCD) {\r
+                al.add(tokenArray[index]);\r
+            }\r
+        }\r
+\r
+        return al;\r
+    }\r
+\r
+\r
+    /**\r
+      Get the token record array contained all PCD token referenced by PEI phase.\r
+      The output array is sorted based on descending order of the size of alignment for each feilds.\r
+\r
+      @return the token record array contained all PCD token referenced in PEI phase.\r
+      @throws EntityException\r
+    **/\r
+    public void getTwoPhaseDynamicRecordArray(ArrayList<Token> pei, ArrayList<Token> dxe)\r
+        throws EntityException {\r
+        int                     usageInstanceIndex  =   0;\r
+        int                     index               =   0;\r
+        ArrayList               tokenArrayList      =   getDynamicRecordArray();\r
+        Object[]                usageInstanceArray  =   null;\r
+        UsageInstance           usageInstance       =   null;\r
+        int                     size                =   0;\r
+        int                     consumerSize        =   0;\r
+\r
+        size = tokenArrayList.size();\r
+        for (index = 0; index < size; index++) {\r
+            boolean found   =   false;\r
+            Token       token = (Token) tokenArrayList.get(index);\r
+            if (token.consumers != null) {\r
+                usageInstanceArray = token.consumers.entrySet().toArray();\r
+                consumerSize       = token.consumers.size();\r
+                for (usageInstanceIndex = 0; usageInstanceIndex < consumerSize; usageInstanceIndex++) {\r
+                    usageInstance =(UsageInstance) (((Map.Entry)usageInstanceArray[usageInstanceIndex]).getValue());\r
+                    if (usageInstance.isPeiPhaseComponent()) {\r
+                        pei.add(token);\r
+                        found = true;\r
+                        break;\r
+                    }\r
+                }\r
+            }\r
+\r
+            //\r
+            // If no PEI components reference the PCD entry,\r
+            // we check if it is referenced in DXE driver.\r
+            //\r
+            if (!found) {\r
+                if (token.consumers != null) {\r
+                    usageInstanceArray = token.consumers.entrySet().toArray();\r
+                    consumerSize       = token.consumers.size();\r
+                    for (usageInstanceIndex = 0; usageInstanceIndex < consumerSize; usageInstanceIndex++) {\r
+                        usageInstance =(UsageInstance) (((Map.Entry)usageInstanceArray[usageInstanceIndex]).getValue());\r
+                        if (usageInstance.isDxePhaseComponent()) {\r
+                            dxe.add(token);\r
+                            found = true;\r
+                            break;\r
+                        }\r
+                    }\r
+                }\r
+\r
+                if (!found) {\r
+                    if (token.isDynamicPCD && token.consumers.size() == 0) {\r
+                        dxe.add(token);\r
+                    } else {\r
+                        //\r
+                        // We only support Dynamice(EX) type for PEI and DXE phase.\r
+                        // If it is not referenced in either PEI or DXE, throw exception now.\r
+                        //\r
+                        throw new EntityException("[PCD Tool Internal Error] Dynamic(EX) PCD Entries are referenced in a module that is not used in either PEI or DXE phases.");\r
+                    }\r
+                }\r
+            }\r
+        }\r
+    }\r
+\r
+    public void clearDatabase() {\r
+        memoryDatabase.clear();\r
+    }\r
+\r
+    /**\r
+       Get all PCD token for a usage instance according to primary key.\r
+\r
+       @param primaryKey    the primary key of usage instance.\r
+\r
+       @return List<UsageInstance>  the list contains all usage instances.\r
+    **/\r
+    public List<UsageInstance> getUsageInstanceArrayById(UsageIdentification usageId) {\r
+        Token[]               tokenArray          = null;\r
+        int                   recordIndex         = 0;\r
+        UsageInstance         usageInstance       = null;\r
+        List<UsageInstance>   returnArray         = new ArrayList<UsageInstance>();\r
+        String                primaryKey          = usageId.toString();\r
+\r
+        tokenArray = getRecordArray();\r
+        \r
+        //\r
+        // Loop to find all PCD record related to current module\r
+        //\r
+        for (recordIndex = 0; recordIndex < getDBSize(); recordIndex++) {\r
+            if (tokenArray[recordIndex].consumers.size() != 0) {\r
+                usageInstance = tokenArray[recordIndex].consumers.get(primaryKey);\r
+                if (usageInstance != null) {\r
+                    returnArray.add(usageInstance);\r
+                }\r
+            }\r
+        }\r
+\r
+        return returnArray;\r
+    }\r
+\r
+    public List<Token> getPcdTokenListForModule(UsageIdentification usageId) {\r
+        List<UsageInstance> usageList = getUsageInstanceArrayById(usageId);\r
+        List<Token>         tokenList = new ArrayList<Token>();\r
+\r
+        if (usageList == null) {\r
+            return null;\r
+        }\r
+\r
+        for (int usageIndex = 0; usageIndex < usageList.size(); usageIndex++) {\r
+            tokenList.add(usageList.get(usageIndex).parentToken);\r
+        }\r
+\r
+        return tokenList;\r
+    }\r
+\r
+    /**\r
+      Get all modules name who contains PCD information\r
+\r
+      @return Array for usage's identification\r
+    **/\r
+    public List<UsageIdentification> getAllModuleArray()\r
+    {\r
+        int                       tokenIndex    = 0;\r
+        int                       usageIndex    = 0;\r
+        int                       moduleIndex   = 0;\r
+        Token[]                   tokenArray    = null;\r
+        Object[]                  usageInstanceArray = null;\r
+        List<UsageIdentification> usageArray    = new ArrayList<UsageIdentification>();\r
+        UsageInstance             usageInstance = null;\r
+        boolean                   bFound        = false;\r
+        String                    primaryKey    = null;\r
+\r
+        tokenArray = getRecordArray();\r
+        //\r
+        // Find all consumer usage instance for retrieving module's name\r
+        //\r
+        for (tokenIndex = 0; tokenIndex < getDBSize(); tokenIndex++) {\r
+            usageInstanceArray = tokenArray[tokenIndex].consumers.entrySet().toArray();\r
+            for (usageIndex = 0; usageIndex < tokenArray[tokenIndex].consumers.size(); usageIndex++) {\r
+                usageInstance = (UsageInstance)((Map.Entry)usageInstanceArray[usageIndex]).getValue();\r
+                primaryKey    = usageInstance.getPrimaryKey();\r
+                bFound        = false;\r
+                for (moduleIndex = 0; moduleIndex < usageArray.size(); moduleIndex++) {\r
+                    if (usageArray.get(moduleIndex).toString().equalsIgnoreCase(primaryKey)) {\r
+                        bFound = true;\r
+                        break;\r
+                    }\r
+                }\r
+                if (!bFound) {\r
+                    usageArray.add(usageInstance.usageId);\r
+                }\r
+            }\r
+        }\r
+        return usageArray;\r
+    }\r
+}\r