libcmo21/Assets/CodeGen/EnumsMigration/EnumsAnalyzer/MainRunner.java

import org.antlr.v4.runtime.*;
import org.antlr.v4.runtime.tree.*;

public class MainRunner {

	/**
	 * Extract an enums collection from given file.
	 * <p>
	 * This function is the most commonly used function for extracting enums.
	 * <p>
	 * This function is used for a file which only contain enum declarations. This
	 * is not suit for extracting CKERROR and CK_CLASSID. For these declarations,
	 * please use their specialized extractor as described following.
	 * 
	 * @param filename The file name relative to input directory for reading.
	 * @return An {@linkplain EnumsHelper.BEnumCollection} instance.
	 * @throws Exception
	 */
	private static EnumsHelper.BEnumCollection getEnumsCollection(String filename) throws Exception {
		String infile = CommonHelper.getInputFilePath(filename);
		CommonHelper.InputFilePair pair = CommonHelper.openInputFile(infile);
		CKGeneralLexer lexer = new CKGeneralLexer(pair.mAntlrStream);
		CommonTokenStream tokens = new CommonTokenStream(lexer);
		CKEnumsParser parser = new CKEnumsParser(tokens);

		ParseTree tree = parser.prog();
		ParseTreeWalker walker = new ParseTreeWalker();
		EnumsWalker worker = new EnumsWalker(tokens);
		walker.walk(worker, tree);

		pair.mUnderlyingStream.close();
		return worker.getEnums();
	}

	/**
	 * Extract a series of "#define" syntax as an enum.
	 * <p>
	 * This function will assume that given file only contain C++ "#define" syntax.
	 * After reading it, it will re-organize it as an enum and return. This only is
	 * used by CKERROR now. But it suit for more scenarios if there are something
	 * like CKERROR in future.
	 *
	 * @param filename The file name relative to input directory for reading.
	 * @param assignedEnumName The desired name of organized enum instance.
	 *                         Contemporary this field should always be "CKERROR"
	 *                         because no one else is using it.
	 * @return An {@linkplain EnumsHelper.BEnum} instance.
	 * @throws Exception
	 */
	private static EnumsHelper.BEnum organiseDefines(String filename, String assignedEnumName) throws Exception {
		String infile = CommonHelper.getInputFilePath(filename);
		CommonHelper.InputFilePair pair = CommonHelper.openInputFile(infile);
		CKGeneralLexer lexer = new CKGeneralLexer(pair.mAntlrStream);
		CommonTokenStream tokens = new CommonTokenStream(lexer);
		CKDefinesParser parser = new CKDefinesParser(tokens);

		ParseTree tree = parser.prog();
		ParseTreeWalker walker = new ParseTreeWalker();
		DefinesWalker worker = new DefinesWalker(tokens);
		walker.walk(worker, tree);

		pair.mUnderlyingStream.close();

		EnumsHelper.BEnum result = worker.getEnum();
		result.mEnumName = assignedEnumName;
		return result;
	}

	/**
	 * Extract a series of macro define as an enum, considering its indent to build
	 * hierarchy.
	 * <p>
	 * This is specialized enum extractor of CK_CLASSID. The given file should use a
	 * series "#define" syntax to describe enum, and use Tab as the indent before
	 * each "#define" syntax to indicate its hierarchy.
	 * 
	 * @param filename The file name relative to input directory for reading.
	 * @return An {@linkplain EnumsHelper.BEnum} instance. Actually it is an
	 *         instance to {@linkplain EnumsHelper.BEnum} whose entries is
	 *         {@linkplain EnumsHelper.BHierarchyEnumEntry}, the child class of
	 *         {@linkplain EnumsHelper.BEnumEntry} (the entry type of common
	 *         {@linkplain EnumsHelper.BEnum}) with extra hierarchy infos.
	 * @throws Exception
	 */
	private static EnumsHelper.BEnum organiseClassid(String filename) throws Exception {
		String infile = CommonHelper.getInputFilePath(filename);
		CommonHelper.InputFilePair pair = CommonHelper.openInputFile(infile);
		CKGeneralLexer lexer = new CKGeneralLexer(pair.mAntlrStream);
		CommonTokenStream tokens = new CommonTokenStream(lexer);
		CKDefinesParser parser = new CKDefinesParser(tokens);

		ParseTree tree = parser.prog();
		ParseTreeWalker walker = new ParseTreeWalker();
		ClassidWalker worker = new ClassidWalker(tokens);
		walker.walk(worker, tree);

		pair.mUnderlyingStream.close();

		EnumsHelper.BEnum result = worker.getEnum();
		result.mEnumName = "CK_CLASSID";
		return result;
	}

	public static void main(String[] args) throws Exception {

		// =========== CKERROR ===========
		EnumsHelper.BEnum ckerror = organiseDefines("CKERROR.txt", "CKERROR");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CKERROR.json"), ckerror);
//		CppWriter.writeEnum("CKERROR.hpp", ckerror);
//		PythonWriter.writeEnum("CKERROR.py", ckerror);
//		CSharpWriter.writeEnum("CKERROR.cs", ckerror);
//		CppWriter.writeCkErrorAccVal("CKERROR.AccVal.hpp", ckerror);
//		PythonWriter.writeAccVal("CKERROR.AccVal.py", ckerror);
//		CSharpWriter.writeAccVal("CKERROR.AccVal.cs", ckerror);

		// =========== CK_CLASSID ===========
		EnumsHelper.BEnum classid = organiseClassid("CK_CLASSID.txt");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_CLASSID.json"), classid);
//		CppWriter.writeEnum("CK_CLASSID.hpp", classid);
//		PythonWriter.writeEnum("CK_CLASSID.py", classid);
//		CSharpWriter.writeEnum("CK_CLASSID.cs", classid);
//		CppWriter.writeCkClassidAccVal("CK_CLASSID.AccVal.hpp", classid);
//		PythonWriter.writeAccVal("CK_CLASSID.AccVal.py", classid);

		// =========== Define2 ===========
		// Define2 do not need annotation output.
		// Because they are CKStateChunk used value which are not exposed to outside.
		EnumsHelper.BEnumCollection def2 = getEnumsCollection("Defines2.txt");
		JsonWriter.writeEnums(CommonHelper.getOutputFilePath("Defines2.json"), def2);
//		CppWriter.writeEnums("Defines2.hpp", def2);
//		PythonWriter.writeEnums("Defines2.py", def2);
//		CSharpWriter.writeEnums("Defines2.cs", def2);

		// =========== Combined enums ===========
		EnumsHelper.BEnumCollection ck2Enums = getEnumsCollection("CKEnums.txt"),
				vxEnums = getEnumsCollection("VxEnums.txt");
		JsonWriter.writeEnums(CommonHelper.getOutputFilePath("CKEnums.json"), ck2Enums);
		JsonWriter.writeEnums(CommonHelper.getOutputFilePath("VxEnums.json"), vxEnums);
		
//		CppWriter.writeEnums("CKEnums.hpp", ck2Enums);
//		PythonWriter.writeEnums("CKEnums.py", ck2Enums);
//		CSharpWriter.writeEnums("CKEnums.cs", ck2Enums);
//		CppWriter.writeAccVals("CKEnums.AccVal.hpp", ck2Enums, CommonHelper.CKParts.CK2);
//		PythonWriter.writeAccVals("CKEnums.AccVal.py", ck2Enums);
//		CSharpWriter.writeAccVals("CKEnums.AccVal.cs", ck2Enums);
		
//		CppWriter.writeEnums("VxEnums.hpp", vxEnums);
//		PythonWriter.writeEnums("VxEnums.py", vxEnums);
//		CSharpWriter.writeEnums("VxEnums.cs", vxEnums);
//		CppWriter.writeAccVals("VxEnums.AccVal.hpp", vxEnums, CommonHelper.CKParts.VxMath);
//		PythonWriter.writeAccVals("VxEnums.AccVal.py", vxEnums);
//		CSharpWriter.writeAccVals("VxEnums.AccVal.cs", vxEnums);

		// =========== Single enums ===========
		EnumsHelper.BEnum single;

		single = organiseDefines("CK_STATECHUNK_CHUNKVERSION.txt", "CK_STATECHUNK_CHUNKVERSION");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_STATECHUNK_CHUNKVERSION.json"), single);
//		CppWriter.writeEnum("CK_STATECHUNK_CHUNKVERSION.hpp", single);
//		PythonWriter.writeEnum("CK_STATECHUNK_CHUNKVERSION.py", single);
//		CSharpWriter.writeEnum("CK_STATECHUNK_CHUNKVERSION.cs", single);
//		CppWriter.writeAccVal("CK_STATECHUNK_CHUNKVERSION.AccVal.hpp", single, CommonHelper.CKParts.CK2);
//		PythonWriter.writeAccVal("CK_STATECHUNK_CHUNKVERSION.AccVal.py", single);
//		CSharpWriter.writeAccVal("CK_STATECHUNK_CHUNKVERSION.AccVal.cs", single);

		single = organiseDefines("CK_STATECHUNK_DATAVERSION.txt", "CK_STATECHUNK_DATAVERSION");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_STATECHUNK_DATAVERSION.json"), single);
//		CppWriter.writeEnum("CK_STATECHUNK_DATAVERSION.hpp", single);
//		PythonWriter.writeEnum("CK_STATECHUNK_DATAVERSION.py", single);
//		CSharpWriter.writeEnum("CK_STATECHUNK_DATAVERSION.cs", single);
//		CppWriter.writeAccVal("CK_STATECHUNK_DATAVERSION.AccVal.hpp", single, CommonHelper.CKParts.CK2);
//		PythonWriter.writeAccVal("CK_STATECHUNK_DATAVERSION.AccVal.py", single);
//		CSharpWriter.writeAccVal("CK_STATECHUNK_DATAVERSION.AccVal.cs", single);

		single = organiseDefines("CK_BITMAPDATA_FLAGS.txt", "CK_BITMAPDATA_FLAGS");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_BITMAPDATA_FLAGS.json"), single);
//		CppWriter.writeEnum("CK_BITMAPDATA_FLAGS.hpp", single);
//		PythonWriter.writeEnum("CK_BITMAPDATA_FLAGS.py", single);
//		CSharpWriter.writeEnum("CK_BITMAPDATA_FLAGS.cs", single);
//		CppWriter.writeAccVal("CK_BITMAPDATA_FLAGS.AccVal.hpp", single, CommonHelper.CKParts.CK2);
//		PythonWriter.writeAccVal("CK_BITMAPDATA_FLAGS.AccVal.py", single);
//		CSharpWriter.writeAccVal("CK_BITMAPDATA_FLAGS.AccVal.cs", single);

		single = organiseDefines("CK_CAMERA_PROJECTION.txt", "CK_CAMERA_PROJECTION");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_CAMERA_PROJECTION.json"), single);
//		CppWriter.writeEnum("CK_CAMERA_PROJECTION.hpp", single);
//		PythonWriter.writeEnum("CK_CAMERA_PROJECTION.py", single);
//		CSharpWriter.writeEnum("CK_CAMERA_PROJECTION.cs", single);
//		CppWriter.writeAccVal("CK_CAMERA_PROJECTION.AccVal.hpp", single, CommonHelper.CKParts.CK2);
//		PythonWriter.writeAccVal("CK_CAMERA_PROJECTION.AccVal.py", single);
//		CSharpWriter.writeAccVal("CK_CAMERA_PROJECTION.AccVal.cs", single);

		// print message.
		System.out.println("Done");
	}
}