Add MigrationTools

[mirror_edk2.git] / Tools / Source / MigrationTools / org / tianocore / migration / Database.java

package org.tianocore.migration;\r
\r
import java.io.*;\r
import java.util.*;\r
\r
public class Database {\r
	Database() throws Exception {\r
		if (System.getenv("WORKSPACE") == null) {\r
			DatabasePath = "C:" + File.separator + "tianocore" + File.separator + "edk2" + File.separator + "Tools" + File.separator + "Conf" + File.separator + "Migration";\r
		} else {\r
			DatabasePath = System.getenv("WORKSPACE") + File.separator + "Tools" + File.separator + "Conf" + File.separator + "Migration";\r
		}\r
		\r
		importDBLib("Library.csv");\r
		importDBGuid("Guid.csv", "Guid");\r
		importDBGuid("Ppi.csv", "Ppi");\r
		importDBGuid("Protocol.csv", "Protocol");\r
		importDBMacro("Macro.csv");\r
	}\r
	\r
	public static String defaultpath = "C:" + File.separator + "tianocore" + File.separator + "edk2" + File.separator + "Tools" + File.separator + "Conf" + File.separator + "Migration";\r
	\r
	public String DatabasePath;\r
	public Set<String> error = new HashSet<String>();\r
	\r
	private Map<String,Guid> hashguid = new HashMap<String,Guid>();\r
	private Map<String,Func> hashfunc = new HashMap<String,Func>();\r
	private Map<String,Macro> hashmacro = new HashMap<String,Macro>();\r
	\r
	private void importDBLib(String filename) throws Exception {\r
		BufferedReader rd = new BufferedReader(new FileReader(DatabasePath + File.separator + filename));\r
		String line;\r
		String[] linecontext;\r
		Func lf;\r
		\r
		if (rd.ready()) {\r
			System.out.println("Found " + filename + " , Importing Library Database");\r
			while ((line = rd.readLine()) != null) {\r
				if (line.length() != 0) {\r
					linecontext = line.split(",");\r
		        	lf = new Func(linecontext);\r
		        	hashfunc.put(lf.r8funcname,lf);\r
				}\r
			}\r
		}\r
	}\r
	\r
	private void importDBGuid(String filename, String type) throws Exception {\r
		BufferedReader rd = new BufferedReader(new FileReader(DatabasePath + File.separator + filename));\r
		String line;\r
		String[] linecontext;\r
		Guid gu;\r
		\r
		if (rd.ready()) {\r
			System.out.println("Found " + filename + " , Importing " + type + " Database");\r
			while ((line = rd.readLine()) != null) {\r
				if (line.length() != 0) {\r
					linecontext = line.split(",");\r
		        	gu = new Guid(linecontext, type);\r
		            hashguid.put(gu.r8name,gu);\r
				}\r
			}\r
		}\r
	}\r
	\r
	private void importDBMacro(String filename) throws Exception {\r
		BufferedReader rd = new BufferedReader(new FileReader(DatabasePath + File.separator + filename));\r
		String line;\r
		String[] linecontext;\r
		Macro mc;\r
		\r
		if (rd.ready()) {\r
			System.out.println("Found " + filename + " , Importing Macro Database");\r
			while ((line = rd.readLine()) != null) {\r
				if (line.length() != 0) {\r
					linecontext = line.split(",");\r
		        	mc = new Macro(linecontext);\r
		        	hashmacro.put(mc.r8name,mc);\r
				}\r
			}\r
		}\r
	}\r
	\r
	public String getR9Lib(String r8funcname) {\r
		String temp = null;\r
		if (hashfunc.containsKey(r8funcname)) {\r
			temp = hashfunc.get(r8funcname).r9libname;\r
		}\r
		return temp;\r
	}\r
	\r
	public String getR9Func(String r8funcname) {\r
		String temp = null;\r
		if (hashfunc.containsKey(r8funcname)) {\r
			temp = hashfunc.get(r8funcname).r9funcname;\r
		}\r
		return temp;\r
	}\r
	\r
	public boolean hasFunc(String r8lib) {\r
		return hashfunc.containsKey(r8lib);\r
	}\r
\r
	public boolean hasGuid(String r8guid) {\r
		return hashguid.containsKey(r8guid);\r
	}\r
\r
	public boolean hasMacro(String r8macro) {\r
		return hashmacro.containsKey(r8macro);\r
	}\r
	\r
	public String getR9Macro(String r8macro) {\r
		return hashmacro.get(r8macro).r9name;			// the verification job of if the macro exists in the database is done when registering it\r
	}\r
	\r
	public String getR9Guidname(String r8Guid) {\r
		String temp = null;\r
		try {\r
			temp = hashguid.get(r8Guid).r9name;\r
		} catch (NullPointerException e) {\r
			error.add("getR9Guidname :" + r8Guid);\r
		}\r
		return temp;\r
	}\r
	\r
	public String getGuidType(String r8Guid) {\r
		String temp = null;\r
		try {\r
			temp =  hashguid.get(r8Guid).type;\r
		} catch (NullPointerException e) {\r
			error.add("getR9Guidname :" + r8Guid);\r
		}\r
		return temp;\r
	}\r
}\r