import org.antlr.v4.runtime.*; import org.antlr.v4.runtime.tree.*; public class MainRunner { /** * Extract an enums collection from given file. *
* This function is the most commonly used function for extracting enums. *
* 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. *
* 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. *
* 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("dest/CKERROR.hpp", ckerror); // PythonWriter.writeEnum("dest/CKERROR.py", ckerror); // CSharpWriter.writeEnum("dest/CKERROR.cs", ckerror); // CppWriter.writeCkErrorAccVal("dest/CKERROR.AccVal.hpp", ckerror); // PythonWriter.writeAccVal("dest/CKERROR.AccVal.py", ckerror); // CSharpWriter.writeAccVal("dest/CKERROR.AccVal.cs", ckerror); // =========== CK_CLASSID =========== EnumsHelper.BEnum classid = organiseClassid("CK_CLASSID.txt"); JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_CLASSID.json"), classid); // CppWriter.writeEnum("dest/CK_CLASSID.hpp", classid); // PythonWriter.writeEnum("dest/CK_CLASSID.py", classid); // CSharpWriter.writeEnum("dest/CK_CLASSID.cs", classid); // CppWriter.writeCkClassidAccVal("dest/CK_CLASSID.AccVal.hpp", classid); // PythonWriter.writeAccVal("dest/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("dest/Defines2.hpp", def2); // PythonWriter.writeEnums("dest/Defines2.py", def2); // CSharpWriter.writeEnums("dest/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("dest/CKEnums.hpp", ck2Enums); // PythonWriter.writeEnums("dest/CKEnums.py", ck2Enums); // CSharpWriter.writeEnums("dest/CKEnums.cs", ck2Enums); // CppWriter.writeAccVals("dest/CKEnums.AccVal.hpp", ck2Enums, CommonHelper.CKParts.CK2); // PythonWriter.writeAccVals("dest/CKEnums.AccVal.py", ck2Enums); // CSharpWriter.writeAccVals("dest/CKEnums.AccVal.cs", ck2Enums); // CppWriter.writeEnums("dest/VxEnums.hpp", vxEnums); // PythonWriter.writeEnums("dest/VxEnums.py", vxEnums); // CSharpWriter.writeEnums("dest/VxEnums.cs", vxEnums); // CppWriter.writeAccVals("dest/VxEnums.AccVal.hpp", vxEnums, CommonHelper.CKParts.VxMath); // PythonWriter.writeAccVals("dest/VxEnums.AccVal.py", vxEnums); // CSharpWriter.writeAccVals("dest/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("dest/CK_STATECHUNK_CHUNKVERSION.hpp", single); // PythonWriter.writeEnum("dest/CK_STATECHUNK_CHUNKVERSION.py", single); // CSharpWriter.writeEnum("dest/CK_STATECHUNK_CHUNKVERSION.cs", single); // CppWriter.writeAccVal("dest/CK_STATECHUNK_CHUNKVERSION.AccVal.hpp", single, CommonHelper.CKParts.CK2); // PythonWriter.writeAccVal("dest/CK_STATECHUNK_CHUNKVERSION.AccVal.py", single); // CSharpWriter.writeAccVal("dest/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("dest/CK_STATECHUNK_DATAVERSION.hpp", single); // PythonWriter.writeEnum("dest/CK_STATECHUNK_DATAVERSION.py", single); // CSharpWriter.writeEnum("dest/CK_STATECHUNK_DATAVERSION.cs", single); // CppWriter.writeAccVal("dest/CK_STATECHUNK_DATAVERSION.AccVal.hpp", single, CommonHelper.CKParts.CK2); // PythonWriter.writeAccVal("dest/CK_STATECHUNK_DATAVERSION.AccVal.py", single); // CSharpWriter.writeAccVal("dest/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("dest/CK_BITMAPDATA_FLAGS.hpp", single); // PythonWriter.writeEnum("dest/CK_BITMAPDATA_FLAGS.py", single); // CSharpWriter.writeEnum("dest/CK_BITMAPDATA_FLAGS.cs", single); // CppWriter.writeAccVal("dest/CK_BITMAPDATA_FLAGS.AccVal.hpp", single, CommonHelper.CKParts.CK2); // PythonWriter.writeAccVal("dest/CK_BITMAPDATA_FLAGS.AccVal.py", single); // CSharpWriter.writeAccVal("dest/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("dest/CK_CAMERA_PROJECTION.hpp", single); // PythonWriter.writeEnum("dest/CK_CAMERA_PROJECTION.py", single); // CSharpWriter.writeEnum("dest/CK_CAMERA_PROJECTION.cs", single); // CppWriter.writeAccVal("dest/CK_CAMERA_PROJECTION.AccVal.hpp", single, CommonHelper.CKParts.CK2); // PythonWriter.writeAccVal("dest/CK_CAMERA_PROJECTION.AccVal.py", single); // CSharpWriter.writeAccVal("dest/CK_CAMERA_PROJECTION.AccVal.cs", single); // print message. System.out.println("Done"); } }