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);
		CKGenericLexer lexer = new CKGenericLexer(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 is suit for more scenarios if there are something
	 * like CKERROR in the 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);
		CKGenericLexer lexer = new CKGenericLexer(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);
		CKGenericLexer lexer = new CKGenericLexer(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);

		// =========== CK_CLASSID ===========
		EnumsHelper.BEnum classid = organiseClassid("CK_CLASSID.txt");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_CLASSID.json"), classid);

		// =========== Define2 ===========
		EnumsHelper.BEnumCollection def2 = getEnumsCollection("Defines2.txt");
		JsonWriter.writeEnums(CommonHelper.getOutputFilePath("Defines2.json"), 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);

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

		single = organiseDefines("CK_STATECHUNK_CHUNKVERSION.txt", "CK_STATECHUNK_CHUNKVERSION");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_STATECHUNK_CHUNKVERSION.json"), single);

		single = organiseDefines("CK_STATECHUNK_DATAVERSION.txt", "CK_STATECHUNK_DATAVERSION");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_STATECHUNK_DATAVERSION.json"), single);

		single = organiseDefines("CK_BITMAPDATA_FLAGS.txt", "CK_BITMAPDATA_FLAGS");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_BITMAPDATA_FLAGS.json"), single);

		single = organiseDefines("CK_CAMERA_PROJECTION.txt", "CK_CAMERA_PROJECTION");
		JsonWriter.writeEnum(CommonHelper.getOutputFilePath("CK_CAMERA_PROJECTION.json"), single);

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