[mirror_edk2.git] / Tools / Source / GenBuild / org / tianocore / build / pcd / entity / MemoryDatabaseManager.java

/** @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.build.pcd.entity;\r
\r
import java.io.BufferedWriter;\r
import java.io.File;\r
import java.io.FileWriter;\r
import java.io.IOException;\r
import java.util.ArrayList;\r
import java.util.Collections;\r
import java.util.Comparator;\r
import java.util.HashMap;\r
import java.util.Hashtable;\r
import java.util.List;\r
import java.util.Map;\r
import java.util.UUID;\r
\r
import org.tianocore.build.pcd.action.ActionMessage;\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
    /// libarry 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
    /// \r
    public static String CurrentModuleName                 = null;\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
\r
        if (memoryDatabase == null) {\r
            return null;\r
        }\r
\r
        dataArray  = memoryDatabase.entrySet().toArray();\r
        tokenArray = new Token[memoryDatabase.size()];\r
        for (index = 0; index < memoryDatabase.size(); index ++) {\r
            entry =(Map.Entry) dataArray [index];\r
            tokenArray[index] =(Token) entry.getValue();\r
        }\r
\r
        return tokenArray;\r
    }\r
\r
    /**\r
       Get record array only contains DYNAMIC or DYNAMIC_EX type PCD.\r
       \r
       @return ArrayList\r
     */\r
    private ArrayList getDynamicRecordArray() {\r
        Token[]     tokenArray  =   getRecordArray();\r
        int         index       =   0;\r
        int         count       =   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
    **/\r
    public void getTwoPhaseDynamicRecordArray(ArrayList<Token> pei, ArrayList<Token> dxe) {\r
        int                     usageInstanceIndex  =   0;\r
        int                     index               =   0;\r
        ArrayList               tokenArrayList      =   getDynamicRecordArray();\r
        Object[]                usageInstanceArray  =   null;\r
        UsageInstance           usageInstance       =   null;\r
\r
        //pei = new ArrayList<Token>();\r
        //dxe = new ArrayList<Token>();\r
\r
        for (index = 0; index < tokenArrayList.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
                for (usageInstanceIndex = 0; usageInstanceIndex < token.consumers.size(); 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
            // If no PEI components reference the PCD entry, we insert it to DXE list\r
            //\r
            if (!found) {\r
                dxe.add(token);\r
            }\r
        }\r
\r
        return;\r
    }\r
\r
    /**\r
      Get all PCD record for a module according to module's name, module's GUID,\r
      package name, package GUID, arch, version information.\r
     \r
      @param moduleName  the name of module.\r
      \r
      @return  all usage instance for this module in memory database.\r
    **/\r
    public List<UsageInstance> getUsageInstanceArrayByModuleName(String moduleName,\r
                                                                 UUID   moduleGuid,\r
                                                                 String packageName,\r
                                                                 UUID   packageGuid,\r
                                                                 String arch,\r
                                                                 String version) {\r
\r
        String primaryKey = UsageInstance.getPrimaryKey(moduleName, \r
                                                        moduleGuid,\r
                                                        packageName,\r
                                                        packageGuid,\r
                                                        arch,\r
                                                        version);\r
\r
        return getUsageInstanceArrayByKeyString(primaryKey);\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>\r
     */\r
    public List<UsageInstance> getUsageInstanceArrayByKeyString(String primaryKey) {\r
        Token[]               tokenArray          = null;\r
        int                   recordIndex         = 0; \r
        UsageInstance         usageInstance       = null;\r
        List<UsageInstance>   returnArray         = new ArrayList<UsageInstance>();\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
    /**\r
      Get all modules name who contains PCD information\r
     \r
      @return Array for module name\r
    **/\r
    public List<String> getAllModuleArray()\r
    {\r
        int                       indexToken    = 0;\r
        int                       usageIndex    = 0;\r
        int                       moduleIndex   = 0;\r
        Token[]                   tokenArray    = null;\r
        Object[]                  usageInstanceArray = null;\r
        List<String>              moduleNames   = new ArrayList<String>();\r
        UsageInstance             usageInstance = null;\r
        boolean                   bFound        = false;\r
\r
        tokenArray = getRecordArray();\r
        //\r
        // Find all consumer usage instance for retrieving module's name\r
        //\r
        for (indexToken = 0; indexToken < getDBSize(); indexToken ++) {\r
            usageInstanceArray = tokenArray[indexToken].consumers.entrySet().toArray();\r
            for (usageIndex = 0; usageIndex < tokenArray[indexToken].consumers.size(); usageIndex ++) {\r
                usageInstance = (UsageInstance)((Map.Entry)usageInstanceArray[usageIndex]).getValue();\r
                bFound        = false;\r
                for (moduleIndex = 0; moduleIndex < moduleNames.size(); moduleIndex ++) {\r
                    if (moduleNames.get(moduleIndex).equalsIgnoreCase(usageInstance.getPrimaryKey())) {\r
                        bFound = true;\r
                        break;\r
                    }\r
                }\r
                if (!bFound) {\r
                    moduleNames.add(usageInstance.getPrimaryKey());\r
                }\r
            }\r
        }\r
        return moduleNames;\r
    }\r
}\r
Commit	Line	Data
878ddf1f	1	/** @file\r
	2	MemoryDatabaseManager class.\r
	3	\r
	4	Database hold all PCD information comes from SPD, MSA, FPD file in memory.\r
	5	\r
	6	Copyright (c) 2006, Intel Corporation\r
	7	All rights reserved. This program and the accompanying materials\r
	8	are licensed and made available under the terms and conditions of the BSD License\r
	9	which accompanies this distribution. The full text of the license may be found at\r
	10	http://opensource.org/licenses/bsd-license.php\r
	11	\r
	12	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	13	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	14	\r
	15	**/\r
	16	package org.tianocore.build.pcd.entity;\r
	17	\r
	18	import java.io.BufferedWriter;\r
	19	import java.io.File;\r
	20	import java.io.FileWriter;\r
	21	import java.io.IOException;\r
	22	import java.util.ArrayList;\r
99d2c3c4	23	import java.util.Collections;\r
99d2c3c4	24	import java.util.Comparator;\r
878ddf1f	25	import java.util.HashMap;\r
99d2c3c4	26	import java.util.Hashtable;\r
878ddf1f	27	import java.util.List;\r
878ddf1f	28	import java.util.Map;\r
8840ad58	29	import java.util.UUID;\r
878ddf1f	30	\r
878ddf1f	31	import org.tianocore.build.pcd.action.ActionMessage;\r
	32	\r
	33	/** Database hold all PCD information comes from SPD, MSA, FPD file in memory.\r
	34	**/\r
	35	public class MemoryDatabaseManager {\r
	36	///\r
	37	/// Memory database. The string "cName + SpaceNameGuid" is primary key.\r
	38	/// memory database is in global scope, and it will be used for others PCD tools.\r
	39	///\r
8840ad58	40	private static Map<String, Token> memoryDatabase = null;\r
8840ad58	41	\r
878ddf1f	42	///\r
8840ad58	43	/// Before build a module, the used libary will be build firstly, the PCD of these\r
	44	/// libarry is inheritted by the module, so stored module's PCD information as PCD\r
	45	/// context of building libary.\r
	46	/// \r
	47	public static List<UsageInstance> UsageInstanceContext = null;\r
878ddf1f	48	///\r
8840ad58	49	/// \r
	50	/// \r
	51	public static String CurrentModuleName = null;\r
99d2c3c4	52	public static String PcdPeimHString = "";\r
32648c62	53	public static String PcdPeimCString = "";\r
	54	public static String PcdDxeHString = "";\r
	55	public static String PcdDxeCString = "";\r
99d2c3c4	56	\r
878ddf1f	57	/**\r
	58	Constructure function\r
	59	**/\r
	60	public MemoryDatabaseManager() {\r
	61	//\r
	62	// Allocate memory for new database in global scope.\r
	63	//\r
	64	if (memoryDatabase == null) {\r
	65	memoryDatabase = new HashMap<String, Token>();\r
	66	}\r
	67	}\r
	68	\r
878ddf1f	69	/**\r
	70	Judege whether token exists in memory database\r
	71	\r
	72	@param primaryKey the primaryKey for searching token\r
	73	\r
	74	@retval TRUE - token already exist in database.\r
	75	@retval FALSE - token does not exist in database.\r
	76	**/\r
	77	public boolean isTokenInDatabase(String primaryKey) {\r
	78	return (memoryDatabase.get(primaryKey) != null);\r
	79	}\r
	80	\r
	81	/**\r
	82	Add a pcd token into memory database.\r
	83	\r
	84	@param primaryKey the primary key for searching token\r
	85	@param token token instance\r
	86	**/\r
	87	public void addTokenToDatabase(String primaryKey, Token token) {\r
	88	memoryDatabase.put(primaryKey, token);\r
	89	}\r
	90	\r
	91	/**\r
	92	Get a token instance from memory database with primary key.\r
	93	\r
	94	@param primaryKey the primary key for searching token\r
	95	\r
	96	@return token instance.\r
	97	**/\r
	98	public Token getTokenByKey(String primaryKey) {\r
	99	return memoryDatabase.get(primaryKey);\r
	100	}\r
	101	\r
	102	/**\r
	103	Get the number of PCD token record in memory database.\r
	104	\r
	105	@return the number of PCD token record in memory database.\r
	106	**/\r
	107	public int getDBSize() {\r
	108	return memoryDatabase.size();\r
	109	}\r
	110	\r
	111	/**\r
	112	Get the token record array contained all PCD token in memory database.\r
	113	\r
	114	@return the token record array contained all PCD token in memory database.\r
	115	**/\r
	116	public Token[] getRecordArray() {\r
	117	Token[] tokenArray = null;\r
	118	Object[] dataArray = null;\r
	119	Map.Entry entry = null;\r
	120	int index = 0;\r
	121	\r
	122	if (memoryDatabase == null) {\r
	123	return null;\r
	124	}\r
	125	\r
	126	dataArray = memoryDatabase.entrySet().toArray();\r
	127	tokenArray = new Token[memoryDatabase.size()];\r
	128	for (index = 0; index < memoryDatabase.size(); index ++) {\r
	129	entry =(Map.Entry) dataArray [index];\r
	130	tokenArray[index] =(Token) entry.getValue();\r
	131	}\r
	132	\r
133	return tokenArray;\r
134	}\r
135	\r
8840ad58	136	/**\r
	137	Get record array only contains DYNAMIC or DYNAMIC_EX type PCD.\r
	138	\r
	139	@return ArrayList\r
	140	*/\r
	141	private ArrayList getDynamicRecordArray() {\r
99d2c3c4	142	Token[] tokenArray = getRecordArray();\r
	143	int index = 0;\r
	144	int count = 0;\r
6a4cae58	145	ArrayList<Token> al = new ArrayList<Token>();\r
99d2c3c4	146	\r
99d2c3c4	147	for (index = 0; index < tokenArray.length; index++) {\r
6ff7a41c	148	if (tokenArray[index].isDynamicPCD) {\r
99d2c3c4	149	al.add(tokenArray[index]);\r
	150	}\r
	151	}\r
	152	\r
	153	return al;\r
	154	}\r
	155	\r
	156	\r
878ddf1f	157	/**\r
99d2c3c4	158	Get the token record array contained all PCD token referenced by PEI phase.\r
	159	The output array is sorted based on descending order of the size of alignment for each feilds.\r
	160	\r
	161	@return the token record array contained all PCD token referenced in PEI phase.\r
	162	**/\r
	163	public void getTwoPhaseDynamicRecordArray(ArrayList<Token> pei, ArrayList<Token> dxe) {\r
	164	int usageInstanceIndex = 0;\r
	165	int index = 0;\r
8840ad58	166	ArrayList tokenArrayList = getDynamicRecordArray();\r
8840ad58	167	Object[] usageInstanceArray = null;\r
99d2c3c4	168	UsageInstance usageInstance = null;\r
99d2c3c4	169	\r
8840ad58	170	//pei = new ArrayList<Token>();\r
	171	//dxe = new ArrayList<Token>();\r
	172	\r
99d2c3c4	173	for (index = 0; index < tokenArrayList.size(); index++) {\r
	174	boolean found = false;\r
	175	Token token = (Token) tokenArrayList.get(index);\r
8840ad58	176	if (token.consumers != null) {\r
	177	usageInstanceArray = token.consumers.entrySet().toArray();\r
	178	for (usageInstanceIndex = 0; usageInstanceIndex < token.consumers.size(); usageInstanceIndex ++) {\r
	179	usageInstance =(UsageInstance) (((Map.Entry)usageInstanceArray[usageInstanceIndex]).getValue());\r
	180	if (usageInstance.isPeiPhaseComponent()) {\r
99d2c3c4	181	pei.add(token);\r
	182	found = true;\r
	183	break;\r
	184	}\r
	185	}\r
99d2c3c4	186	}\r
99d2c3c4	187	\r
32648c62	188	// If no PEI components reference the PCD entry, we insert it to DXE list\r
	189	//\r
	190	if (!found) {\r
	191	dxe.add(token);\r
	192	}\r
99d2c3c4	193	}\r
99d2c3c4	194	\r
32648c62	195	return;\r
99d2c3c4	196	}\r
99d2c3c4	197	\r
32648c62	198	/**\r
8840ad58	199	Get all PCD record for a module according to module's name, module's GUID,\r
8840ad58	200	package name, package GUID, arch, version information.\r
878ddf1f	201	\r
	202	@param moduleName the name of module.\r
	203	\r
	204	@return all usage instance for this module in memory database.\r
	205	**/\r
8840ad58	206	public List<UsageInstance> getUsageInstanceArrayByModuleName(String moduleName,\r
	207	UUID moduleGuid,\r
	208	String packageName,\r
	209	UUID packageGuid,\r
	210	String arch,\r
	211	String version) {\r
	212	\r
	213	String primaryKey = UsageInstance.getPrimaryKey(moduleName, \r
	214	moduleGuid,\r
	215	packageName,\r
	216	packageGuid,\r
	217	arch,\r
	218	version);\r
	219	\r
	220	return getUsageInstanceArrayByKeyString(primaryKey);\r
	221	}\r
	222	\r
	223	/**\r
	224	Get all PCD token for a usage instance according to primary key.\r
	225	\r
	226	@param primaryKey the primary key of usage instance.\r
	227	\r
	228	@return List<UsageInstance>\r
	229	*/\r
	230	public List<UsageInstance> getUsageInstanceArrayByKeyString(String primaryKey) {\r
878ddf1f	231	Token[] tokenArray = null;\r
878ddf1f	232	int recordIndex = 0; \r
878ddf1f	233	UsageInstance usageInstance = null;\r
	234	List<UsageInstance> returnArray = new ArrayList<UsageInstance>();\r
	235	\r
	236	tokenArray = getRecordArray();\r
	237	\r
	238	//\r
	239	// Loop to find all PCD record related to current module\r
	240	//\r
	241	for (recordIndex = 0; recordIndex < getDBSize(); recordIndex ++) {\r
8840ad58	242	if (tokenArray[recordIndex].consumers.size() != 0) {\r
	243	usageInstance = tokenArray[recordIndex].consumers.get(primaryKey);\r
	244	if (usageInstance != null) {\r
	245	returnArray.add(usageInstance);\r
878ddf1f	246	}\r
	247	}\r
	248	}\r
	249	\r
878ddf1f	250	return returnArray;\r
	251	}\r
	252	\r
	253	/**\r
	254	Get all modules name who contains PCD information\r
	255	\r
	256	@return Array for module name\r
	257	**/\r
	258	public List<String> getAllModuleArray()\r
	259	{\r
	260	int indexToken = 0;\r
	261	int usageIndex = 0;\r
	262	int moduleIndex = 0;\r
	263	Token[] tokenArray = null;\r
8840ad58	264	Object[] usageInstanceArray = null;\r
878ddf1f	265	List<String> moduleNames = new ArrayList<String>();\r
	266	UsageInstance usageInstance = null;\r
	267	boolean bFound = false;\r
	268	\r
8840ad58	269	tokenArray = getRecordArray();\r
878ddf1f	270	//\r
	271	// Find all consumer usage instance for retrieving module's name\r
	272	//\r
	273	for (indexToken = 0; indexToken < getDBSize(); indexToken ++) {\r
8840ad58	274	usageInstanceArray = tokenArray[indexToken].consumers.entrySet().toArray();\r
878ddf1f	275	for (usageIndex = 0; usageIndex < tokenArray[indexToken].consumers.size(); usageIndex ++) {\r
8840ad58	276	usageInstance = (UsageInstance)((Map.Entry)usageInstanceArray[usageIndex]).getValue();\r
878ddf1f	277	bFound = false;\r
878ddf1f	278	for (moduleIndex = 0; moduleIndex < moduleNames.size(); moduleIndex ++) {\r
8840ad58	279	if (moduleNames.get(moduleIndex).equalsIgnoreCase(usageInstance.getPrimaryKey())) {\r
878ddf1f	280	bFound = true;\r
	281	break;\r
	282	}\r
	283	}\r
	284	if (!bFound) {\r
8840ad58	285	moduleNames.add(usageInstance.getPrimaryKey());\r
878ddf1f	286	}\r
	287	}\r
	288	}\r
	289	return moduleNames;\r
	290	}\r
878ddf1f	291	}\r