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

/** @file\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.migration;\r
\r
import java.io.*;\r
import java.util.*;\r
\r
import org.tianocore.*;\r
import org.tianocore.SupportedArchitectures.Enum;\r
import org.apache.xmlbeans.*;\r
\r
public class MsaWriter {\r
	MsaWriter(String path, ModuleInfo moduleinfo, Database database, UI u) {\r
		modulepath = path;\r
		mi = moduleinfo;\r
		db = database;\r
		ui = u;\r
	}\r
\r
	private String modulepath;\r
	private ModuleInfo mi;\r
	private Database db;\r
	private UI ui;\r
	\r
	private ModuleSurfaceAreaDocument msadoc = ModuleSurfaceAreaDocument.Factory.newInstance();\r
	\r
	private ModuleSurfaceAreaDocument.ModuleSurfaceArea msa = msadoc.addNewModuleSurfaceArea();\r
	private MsaHeaderDocument.MsaHeader msaheader = msa.addNewMsaHeader();\r
	private ModuleDefinitionsDocument.ModuleDefinitions md = msa.addNewModuleDefinitions();\r
	private SourceFilesDocument.SourceFiles sourcefiles = msa.addNewSourceFiles();	//found local .h files are not written\r
	private GuidsDocument.Guids guids;\r
	private ProtocolsDocument.Protocols protocols;\r
	private PPIsDocument.PPIs ppis;\r
	private PackageDependenciesDocument.PackageDependencies pd = msa.addNewPackageDependencies();\r
	private LibraryClassDefinitionsDocument.LibraryClassDefinitions libclassdefs = msa.addNewLibraryClassDefinitions();\r
	private ExternsDocument.Externs externs = msa.addNewExterns();\r
	\r
	private String Query (String requirement) throws Exception {\r
		String answer;\r
		BufferedReader rd = new BufferedReader(new InputStreamReader(System.in));\r
		System.out.println(requirement);\r
		while ((answer = rd.readLine()).length() == 0) ;\r
		return answer;\r
	}\r
	\r
	private void addSourceFiles (String name) {						// furthur modification needed\r
		List<Enum> arch = new ArrayList<Enum>();\r
		FilenameDocument.Filename filename;\r
		filename = sourcefiles.addNewFilename();\r
		filename.setStringValue(name);\r
		\r
		if (name.contains("x64" + File.separator)) {				// filename ???\r
			arch.add(SupportedArchitectures.X_64);\r
			System.out.println("x64" + File.separator);\r
			filename.setSupArchList(arch);\r
		} else if (name.contains("Ia32" + File.separator)) {		// filename ???\r
			arch.add(SupportedArchitectures.IA_32);\r
			System.out.println("Ia32" + File.separator);\r
			filename.setSupArchList(arch);\r
		} else if (name.contains("Ipf" + File.separator)) {			// filename ???\r
			arch.add(SupportedArchitectures.IPF);\r
			System.out.println("Ipf" + File.separator);\r
			filename.setSupArchList(arch);\r
		} else if (name.contains("Ebc" + File.separator)) {			// filename ???\r
			arch.add(SupportedArchitectures.EBC);\r
			System.out.println("Ebc" + File.separator);\r
			filename.setSupArchList(arch);\r
		}\r
	}\r
	\r
	private ModuleSurfaceAreaDocument fulfillMsadoc() throws Exception {\r
		Iterator<String> it;\r
		String temp;\r
		\r
		if (mi.modulename != null) {\r
			msaheader.setModuleName(mi.modulename);\r
		} else {\r
			msaheader.setModuleName(mi.modulename = Query("Module Name Not Found!  Please Input ModuleName"));\r
		}\r
		if (mi.guidvalue != null) {\r
			msaheader.setGuidValue(mi.guidvalue);\r
		} else {\r
			msaheader.setGuidValue(mi.guidvalue = Query("Guid Value Not Found!  Please Input Guid Value"));\r
		}\r
		if (mi.moduletype != null) {\r
			if (mi.moduletype.contains("PEI")) {\r
				msaheader.setModuleType(ModuleTypeDef.Enum.forString("PEIM"));\r
			} else {\r
				msaheader.setModuleType(ModuleTypeDef.Enum.forString("DXE_DRIVER"));\r
			}\r
		} else {\r
			msaheader.setModuleType(ModuleTypeDef.Enum.forString(mi.moduletype = Query("Guid Value Not Found!  Please Input Guid Value")));\r
		}\r
\r
		msaheader.setCopyright("Copyright (c) 2006, Intel Corporation");\r
		msaheader.setVersion("1.0");\r
		msaheader.setAbstract("Component name for module " + mi.modulename);\r
		msaheader.setDescription("FIX ME!");\r
		msaheader.addNewLicense().setStringValue("All rights reserved.\n" +\r
				"      This software and associated documentation (if any) is furnished\n" +\r
				"      under a license and may only be used or copied in accordance\n" +\r
				"      with the terms of the license. Except as permitted by such\n" +\r
				"      license, no part of this software or documentation may be\n" +\r
				"      reproduced, stored in a retrieval system, or transmitted in any\n" +\r
				"      form or by any means without the express written consent of\n" +\r
				"      Intel Corporation.");\r
		msaheader.setSpecification("FRAMEWORK_BUILD_PACKAGING_SPECIFICATION   0x00000052");\r
		\r
		List<Enum> arch = new ArrayList<Enum>();\r
		arch.add(SupportedArchitectures.IA_32);\r
		arch.add(SupportedArchitectures.X_64);\r
		arch.add(SupportedArchitectures.IPF);\r
		arch.add(SupportedArchitectures.EBC);\r
		md.setSupportedArchitectures(arch);\r
		md.setBinaryModule(false);\r
		md.setOutputFileBasename(mi.modulename);\r
		\r
		pd.addNewPackage().setPackageGuid("5e0e9358-46b6-4ae2-8218-4ab8b9bbdcec");\r
		externs.addNewSpecification().setStringValue("EFI_SPECIFICATION_VERSION 0x00020000");\r
		externs.addNewSpecification().setStringValue("EDK_RELEASE_VERSION 0x00020000");\r
		externs.addNewExtern().setModuleEntryPoint(mi.entrypoint);\r
		\r
		it = mi.localmodulesources.iterator();\r
		while (it.hasNext()) {\r
			addSourceFiles(it.next());\r
		}\r
		if (!mi.protocol.isEmpty()) {\r
			protocols = msa.addNewProtocols();\r
			it = mi.protocol.iterator();\r
			while (it.hasNext()) {\r
				if ((temp = it.next()) != null) {\r
					ProtocolsDocument.Protocols.Protocol pr = protocols.addNewProtocol();\r
					pr.setProtocolCName(temp);\r
					pr.setUsage(UsageTypes.ALWAYS_CONSUMED);\r
				}\r
			}\r
		}\r
		if (!mi.ppi.isEmpty()) {\r
			ppis = msa.addNewPPIs();\r
			it = mi.ppi.iterator();\r
			while (it.hasNext()) {\r
				if ((temp = it.next()) != null) {\r
					PPIsDocument.PPIs.Ppi pp = ppis.addNewPpi();\r
					pp.setPpiCName(temp);\r
					pp.setUsage(UsageTypes.ALWAYS_CONSUMED);\r
				}\r
			}\r
		}\r
		if (!mi.guid.isEmpty()) {\r
			guids = msa.addNewGuids();\r
			it = mi.guid.iterator();\r
			while (it.hasNext()) {\r
				if ((temp = it.next()) != null) {\r
					GuidsDocument.Guids.GuidCNames gcn = guids.addNewGuidCNames();\r
					gcn.setGuidCName(temp);\r
					gcn.setUsage(UsageTypes.ALWAYS_CONSUMED);\r
				}\r
			}\r
		}\r
		it = mi.hashrequiredr9libs.iterator();\r
		while (it.hasNext()) {\r
			if ((temp = it.next()) != null && !temp.matches("%")) {\r
				LibraryClassDocument.LibraryClass lc = libclassdefs.addNewLibraryClass();\r
				lc.setKeyword(temp);\r
				lc.setUsage(UsageTypes.ALWAYS_CONSUMED);\r
			}\r
		}\r
		\r
		return msadoc;\r
	}\r
	\r
	public void flush() throws Exception {\r
        XmlOptions options = new XmlOptions();\r
\r
        options.setCharacterEncoding("UTF-8");\r
        options.setSavePrettyPrint();\r
        options.setSavePrettyPrintIndent(2);\r
        options.setUseDefaultNamespace();\r
        \r
		BufferedWriter bw = new BufferedWriter(new FileWriter(modulepath + File.separator + "result" + File.separator + mi.modulename + ".msa"));\r
		fulfillMsadoc().save(bw, options);\r
		//MsaTreeEditor.init(mi, ui, msadoc);\r
		bw.flush();\r
		bw.close();\r
	}\r
}\r
Commit	Line	Data
b0282412	1	/** @file\r
	2	\r
	3	Copyright (c) 2006, Intel Corporation\r
	4	All rights reserved. This program and the accompanying materials\r
	5	are licensed and made available under the terms and conditions of the BSD License\r
	6	which accompanies this distribution. The full text of the license may be found at\r
	7	http://opensource.org/licenses/bsd-license.php\r
	8	\r
	9	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	10	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	11	\r
	12	**/\r
0dc8c589	13	package org.tianocore.migration;\r
	14	\r
	15	import java.io.*;\r
	16	import java.util.*;\r
	17	\r
	18	import org.tianocore.*;\r
	19	import org.tianocore.SupportedArchitectures.Enum;\r
	20	import org.apache.xmlbeans.*;\r
	21	\r
	22	public class MsaWriter {\r
fed802b1	23	MsaWriter(String path, ModuleInfo moduleinfo, Database database, UI u) {\r
0dc8c589	24	modulepath = path;\r
	25	mi = moduleinfo;\r
	26	db = database;\r
fed802b1	27	ui = u;\r
0dc8c589	28	}\r
	29	\r
	30	private String modulepath;\r
	31	private ModuleInfo mi;\r
	32	private Database db;\r
fed802b1	33	private UI ui;\r
0dc8c589	34	\r
	35	private ModuleSurfaceAreaDocument msadoc = ModuleSurfaceAreaDocument.Factory.newInstance();\r
	36	\r
	37	private ModuleSurfaceAreaDocument.ModuleSurfaceArea msa = msadoc.addNewModuleSurfaceArea();\r
	38	private MsaHeaderDocument.MsaHeader msaheader = msa.addNewMsaHeader();\r
	39	private ModuleDefinitionsDocument.ModuleDefinitions md = msa.addNewModuleDefinitions();\r
	40	private SourceFilesDocument.SourceFiles sourcefiles = msa.addNewSourceFiles(); //found local .h files are not written\r
fb840aa8	41	private GuidsDocument.Guids guids;\r
	42	private ProtocolsDocument.Protocols protocols;\r
	43	private PPIsDocument.PPIs ppis;\r
0dc8c589	44	private PackageDependenciesDocument.PackageDependencies pd = msa.addNewPackageDependencies();\r
	45	private LibraryClassDefinitionsDocument.LibraryClassDefinitions libclassdefs = msa.addNewLibraryClassDefinitions();\r
	46	private ExternsDocument.Externs externs = msa.addNewExterns();\r
	47	\r
	48	private String Query (String requirement) throws Exception {\r
	49	String answer;\r
	50	BufferedReader rd = new BufferedReader(new InputStreamReader(System.in));\r
	51	System.out.println(requirement);\r
	52	while ((answer = rd.readLine()).length() == 0) ;\r
	53	return answer;\r
	54	}\r
	55	\r
2be54d60	56	private void addSourceFiles (String name) { // furthur modification needed\r
	57	List<Enum> arch = new ArrayList<Enum>();\r
	58	FilenameDocument.Filename filename;\r
	59	filename = sourcefiles.addNewFilename();\r
	60	filename.setStringValue(name);\r
	61	\r
	62	if (name.contains("x64" + File.separator)) { // filename ???\r
	63	arch.add(SupportedArchitectures.X_64);\r
	64	System.out.println("x64" + File.separator);\r
	65	filename.setSupArchList(arch);\r
	66	} else if (name.contains("Ia32" + File.separator)) { // filename ???\r
	67	arch.add(SupportedArchitectures.IA_32);\r
	68	System.out.println("Ia32" + File.separator);\r
	69	filename.setSupArchList(arch);\r
	70	} else if (name.contains("Ipf" + File.separator)) { // filename ???\r
	71	arch.add(SupportedArchitectures.IPF);\r
	72	System.out.println("Ipf" + File.separator);\r
	73	filename.setSupArchList(arch);\r
	74	} else if (name.contains("Ebc" + File.separator)) { // filename ???\r
	75	arch.add(SupportedArchitectures.EBC);\r
	76	System.out.println("Ebc" + File.separator);\r
	77	filename.setSupArchList(arch);\r
	78	}\r
	79	}\r
	80	\r
0dc8c589	81	private ModuleSurfaceAreaDocument fulfillMsadoc() throws Exception {\r
	82	Iterator<String> it;\r
	83	String temp;\r
	84	\r
	85	if (mi.modulename != null) {\r
	86	msaheader.setModuleName(mi.modulename);\r
	87	} else {\r
719cebfe	88	msaheader.setModuleName(mi.modulename = Query("Module Name Not Found! Please Input ModuleName"));\r
0dc8c589	89	}\r
	90	if (mi.guidvalue != null) {\r
	91	msaheader.setGuidValue(mi.guidvalue);\r
	92	} else {\r
719cebfe	93	msaheader.setGuidValue(mi.guidvalue = Query("Guid Value Not Found! Please Input Guid Value"));\r
0dc8c589	94	}\r
	95	if (mi.moduletype != null) {\r
	96	if (mi.moduletype.contains("PEI")) {\r
	97	msaheader.setModuleType(ModuleTypeDef.Enum.forString("PEIM"));\r
	98	} else {\r
	99	msaheader.setModuleType(ModuleTypeDef.Enum.forString("DXE_DRIVER"));\r
	100	}\r
	101	} else {\r
719cebfe	102	msaheader.setModuleType(ModuleTypeDef.Enum.forString(mi.moduletype = Query("Guid Value Not Found! Please Input Guid Value")));\r
0dc8c589	103	}\r
fb840aa8	104	\r
0dc8c589	105	msaheader.setCopyright("Copyright (c) 2006, Intel Corporation");\r
	106	msaheader.setVersion("1.0");\r
	107	msaheader.setAbstract("Component name for module " + mi.modulename);\r
90503bad	108	msaheader.setDescription("FIX ME!");\r
0dc8c589	109	msaheader.addNewLicense().setStringValue("All rights reserved.\n" +\r
	110	" This software and associated documentation (if any) is furnished\n" +\r
	111	" under a license and may only be used or copied in accordance\n" +\r
	112	" with the terms of the license. Except as permitted by such\n" +\r
	113	" license, no part of this software or documentation may be\n" +\r
	114	" reproduced, stored in a retrieval system, or transmitted in any\n" +\r
	115	" form or by any means without the express written consent of\n" +\r
	116	" Intel Corporation.");\r
fb840aa8	117	msaheader.setSpecification("FRAMEWORK_BUILD_PACKAGING_SPECIFICATION 0x00000052");\r
0dc8c589	118	\r
	119	List<Enum> arch = new ArrayList<Enum>();\r
	120	arch.add(SupportedArchitectures.IA_32);\r
	121	arch.add(SupportedArchitectures.X_64);\r
	122	arch.add(SupportedArchitectures.IPF);\r
	123	arch.add(SupportedArchitectures.EBC);\r
	124	md.setSupportedArchitectures(arch);\r
	125	md.setBinaryModule(false);\r
	126	md.setOutputFileBasename(mi.modulename);\r
	127	\r
	128	pd.addNewPackage().setPackageGuid("5e0e9358-46b6-4ae2-8218-4ab8b9bbdcec");\r
	129	externs.addNewSpecification().setStringValue("EFI_SPECIFICATION_VERSION 0x00020000");\r
	130	externs.addNewSpecification().setStringValue("EDK_RELEASE_VERSION 0x00020000");\r
	131	externs.addNewExtern().setModuleEntryPoint(mi.entrypoint);\r
	132	\r
	133	it = mi.localmodulesources.iterator();\r
	134	while (it.hasNext()) {\r
2be54d60	135	addSourceFiles(it.next());\r
0dc8c589	136	}\r
fb840aa8	137	if (!mi.protocol.isEmpty()) {\r
	138	protocols = msa.addNewProtocols();\r
	139	it = mi.protocol.iterator();\r
	140	while (it.hasNext()) {\r
	141	if ((temp = it.next()) != null) {\r
	142	ProtocolsDocument.Protocols.Protocol pr = protocols.addNewProtocol();\r
	143	pr.setProtocolCName(temp);\r
	144	pr.setUsage(UsageTypes.ALWAYS_CONSUMED);\r
	145	}\r
0dc8c589	146	}\r
0dc8c589	147	}\r
fb840aa8	148	if (!mi.ppi.isEmpty()) {\r
	149	ppis = msa.addNewPPIs();\r
	150	it = mi.ppi.iterator();\r
	151	while (it.hasNext()) {\r
	152	if ((temp = it.next()) != null) {\r
	153	PPIsDocument.PPIs.Ppi pp = ppis.addNewPpi();\r
	154	pp.setPpiCName(temp);\r
	155	pp.setUsage(UsageTypes.ALWAYS_CONSUMED);\r
	156	}\r
0dc8c589	157	}\r
0dc8c589	158	}\r
fb840aa8	159	if (!mi.guid.isEmpty()) {\r
	160	guids = msa.addNewGuids();\r
	161	it = mi.guid.iterator();\r
	162	while (it.hasNext()) {\r
	163	if ((temp = it.next()) != null) {\r
	164	GuidsDocument.Guids.GuidCNames gcn = guids.addNewGuidCNames();\r
	165	gcn.setGuidCName(temp);\r
	166	gcn.setUsage(UsageTypes.ALWAYS_CONSUMED);\r
	167	}\r
0dc8c589	168	}\r
	169	}\r
	170	it = mi.hashrequiredr9libs.iterator();\r
	171	while (it.hasNext()) {\r
	172	if ((temp = it.next()) != null && !temp.matches("%")) {\r
	173	LibraryClassDocument.LibraryClass lc = libclassdefs.addNewLibraryClass();\r
	174	lc.setKeyword(temp);\r
	175	lc.setUsage(UsageTypes.ALWAYS_CONSUMED);\r
	176	}\r
	177	}\r
	178	\r
	179	return msadoc;\r
	180	}\r
	181	\r
	182	public void flush() throws Exception {\r
	183	XmlOptions options = new XmlOptions();\r
	184	\r
	185	options.setCharacterEncoding("UTF-8");\r
	186	options.setSavePrettyPrint();\r
	187	options.setSavePrettyPrintIndent(2);\r
	188	options.setUseDefaultNamespace();\r
	189	\r
	190	BufferedWriter bw = new BufferedWriter(new FileWriter(modulepath + File.separator + "result" + File.separator + mi.modulename + ".msa"));\r
	191	fulfillMsadoc().save(bw, options);\r
fed802b1	192	//MsaTreeEditor.init(mi, ui, msadoc);\r
0dc8c589	193	bw.flush();\r
	194	bw.close();\r
	195	}\r
	196	}\r