libcmo21/CodeGen/EnumsMigration/EnumsWalker.java

import java.util.List;
import java.util.stream.Collectors;

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

/**
 * The generic walker for collecting multiple enums.
 */
public class EnumsWalker extends CKEnumsParserBaseListener {
	public EnumsWalker(BufferedTokenStream tokenStream) {
		mTokenStream = tokenStream;
		mCommentsFinder = new CommentsFinder(tokenStream);
		mResult = null;

		mCurrentProg = null;
		mCurrentEnum = null;
		mCurrentEntry = null;
	}

	public EnumsHelper.EnumCollection_t getEnums() {
		return mResult;
	}

	private String getEnumComment(Token enumHead) {
		return CommonHelper
				.cutComments(CommonHelper.getPreChannelTokens(mTokenStream, enumHead, CKGeneralLexer.COMMENTS));
	}

	private BufferedTokenStream mTokenStream;
	private CommentsFinder mCommentsFinder;
	private EnumsHelper.EnumCollection_t mResult;

	private EnumsHelper.EnumCollection_t mCurrentProg;
	private EnumsHelper.Enum_t mCurrentEnum;
	private EnumsHelper.EnumEntry_t mCurrentEntry;

	@Override
	public void enterProg(CKEnumsParser.ProgContext ctx) {
		mCurrentProg = new EnumsHelper.EnumCollection_t();
	}

	@Override
	public void exitProg(CKEnumsParser.ProgContext ctx) {
		mResult = mCurrentProg;
		mCurrentProg = null;
	}

	@Override
	public void enterEnumBody(CKEnumsParser.EnumBodyContext ctx) {
		mCurrentEnum = new EnumsHelper.Enum_t();
	}

	@Override
	public void exitEnumBody(CKEnumsParser.EnumBodyContext ctx) {
		// get enum comment
		mCurrentEnum.mEnumComment = getEnumComment(ctx.getStart());
		// get the last name (for typedef case)
		List<TerminalNode> allNames = ctx.CKGENERAL_ID();
		mCurrentEnum.mEnumName = allNames.get(allNames.size() - 1).getText();

		mCurrentProg.mEnums.add(mCurrentEnum);
		mCurrentEnum = null;
	}

	@Override
	public void enterEntryPair(CKEnumsParser.EntryPairContext ctx) {
		mCurrentEntry = new EnumsHelper.EnumEntry_t();
	}

	@Override
	public void exitEntryPair(CKEnumsParser.EntryPairContext ctx) {
		// get entry comment
		mCurrentEntry.mEntryComment = mCommentsFinder.getComment(ctx.getStart(), ctx.getStop());
		// get entry name
		mCurrentEntry.mEntryName = ctx.CKGENERAL_ID().getText();

		mCurrentEnum.mEntries.add(mCurrentEntry);
		mCurrentEntry = null;
	}

	@Override
	public void exitEntryDirectValue(CKEnumsParser.EntryDirectValueContext ctx) {
		// get all numbers
		List<TerminalNode> nums = ctx.CKGENERAL_NUM();

		switch (nums.size()) {
		case 1: {
			// set value
			TerminalNode node = nums.get(0);
			mCurrentEntry.mEntryValue = node.getText();

			// check whether this enum can be unsigned
			if (CommonHelper.isNegativeNumber(node.getText())) {
				mCurrentEnum.mCanUnsigned = false;
			}
			// if the number is in hex form, this enum MIGHT have flags feature
			if (CommonHelper.isHexNumber(node.getText())) {
				mCurrentEnum.mUseFlags = true;
			}

			break;
		}
		case 2: {
			// set value
			TerminalNode num = nums.get(0), offset = nums.get(1);
			mCurrentEntry.mEntryValue = String.format("%s << %s", num.getText(), offset.getText());

			// << operator appears. this enum must have flags feature
			mCurrentEnum.mUseFlags = true;

			break;
		}
		default:
			throw new IllegalArgumentException("Unexpected value: " + nums.size());
		}

	}

	@Override
	public void exitEntryRelativeValue(CKEnumsParser.EntryRelativeValueContext ctx) {
		// get all identifiers and join them
		mCurrentEntry.mEntryValue = ctx.CKGENERAL_ID().stream().map(value -> value.getText())
				.collect(Collectors.joining(" | "));

		// | operator appears. this enum must have flags feature
		mCurrentEnum.mUseFlags = true;

	}

}