DroolsTreeAdaptor.java example

Explorer
droolsjbpm-master
package org.drools.lang;

import org.antlr.runtime.CommonToken;
import org.antlr.runtime.Token;
import org.antlr.runtime.tree.CommonTreeAdaptor;
import org.antlr.runtime.tree.Tree;

/**
 * This tree adaptor is a factory for DroolsTree.
 * 
 * DRLParser must use this tree adaptor.
 * 
 * @author porcelli
 */
public class DroolsTreeAdaptor extends CommonTreeAdaptor {

	/**
	 * Based on token parameter it returns a new DroolsTree. Also set the char
	 * offset info and editor type based on token type.
	 * 
	 * @param token
	 *            token
	 * @return DroolsTree object with char offset and editor type info
	 */
	public Object create(Token token) {
		DroolsTree tree = new DroolsTree(token);
		if (null != token
				&& (token.getClass().equals(CommonToken.class) || token
						.getClass().equals(DroolsToken.class))) {
			tree.setStartCharOffset(((CommonToken) token).getStartIndex());
			tree.setEndCharOffset(((CommonToken) token).getStopIndex());

			if (token.getType() == DRLLexer.VK_END
					|| token.getType() == DRLLexer.VK_EVAL
					|| token.getType() == DRLLexer.THEN
					|| token.getType() == DRLLexer.WHEN
					|| token.getType() == DRLLexer.ACCUMULATE
					|| token.getType() == DRLLexer.VK_ACTION
					|| token.getType() == DRLLexer.VK_ACTIVATION_GROUP
					|| token.getType() == DRLLexer.VK_AGENDA_GROUP
					|| token.getType() == DRLLexer.VK_AND
					|| token.getType() == DRLLexer.VK_ATTRIBUTES
					|| token.getType() == DRLLexer.VK_AUTO_FOCUS
					|| token.getType() == DRLLexer.COLLECT
					|| token.getType() == DRLLexer.VK_DATE_EFFECTIVE
					|| token.getType() == DRLLexer.VK_DATE_EXPIRES
					|| token.getType() == DRLLexer.VK_DECLARE
					|| token.getType() == DRLLexer.VK_DIALECT
					|| token.getType() == DRLLexer.VK_TIMER
					|| token.getType() == DRLLexer.VK_ENABLED
					|| token.getType() == DRLLexer.VK_ENTRY_POINT
					|| token.getType() == DRLLexer.VK_EXISTS
					|| token.getType() == DRLLexer.VK_FORALL
					|| token.getType() == DRLLexer.FROM
					|| token.getType() == DRLLexer.VK_FUNCTION
					|| token.getType() == DRLLexer.VK_GLOBAL
					|| token.getType() == DRLLexer.VK_IMPORT
					|| token.getType() == DRLLexer.VK_IN
					|| token.getType() == DRLLexer.VK_INIT
					|| token.getType() == DRLLexer.VK_LOCK_ON_ACTIVE
					|| token.getType() == DRLLexer.VK_NO_LOOP
					|| token.getType() == DRLLexer.VK_NOT
					|| token.getType() == DRLLexer.VK_OR
					|| token.getType() == DRLLexer.VK_PACKAGE
					|| token.getType() == DRLLexer.VK_QUERY
					|| token.getType() == DRLLexer.VK_RESULT
					|| token.getType() == DRLLexer.VK_REVERSE
					|| token.getType() == DRLLexer.VK_RULE
					|| token.getType() == DRLLexer.VK_RULEFLOW_GROUP
					|| token.getType() == DRLLexer.VK_SALIENCE) {
				tree.setEditorElementType(DroolsEditorType.KEYWORD);
			} else if (token.getType() == DRLLexer.FLOAT
					|| token.getType() == DRLLexer.DECIMAL) {
				tree.setEditorElementType(DroolsEditorType.NUMERIC_CONST);
			} else if (token.getType() == DRLLexer.STRING) {
				tree.setEditorElementType(DroolsEditorType.STRING_CONST);
			} else if (token.getType() == DRLLexer.BOOL) {
				tree.setEditorElementType(DroolsEditorType.BOOLEAN_CONST);
			} else if (token.getType() == DRLLexer.NULL) {
				tree.setEditorElementType(DroolsEditorType.NULL_CONST);
			} else if (token.getType() == DRLLexer.VT_SQUARE_CHUNK
					|| token.getType() == DRLLexer.VT_PAREN_CHUNK
					|| token.getType() == DRLLexer.VT_CURLY_CHUNK
					|| token.getType() == DRLLexer.VT_RHS_CHUNK) {
				tree.setEditorElementType(DroolsEditorType.CODE_CHUNK);
			} else if (token.getType() == DRLLexer.MISC
					|| token.getType() == DRLLexer.DOUBLE_PIPE
					|| token.getType() == DRLLexer.DOUBLE_AMPER
					|| token.getType() == DRLLexer.DOT
					|| token.getType() == DRLLexer.COMMA
					|| token.getType() == DRLLexer.RIGHT_CURLY
					|| token.getType() == DRLLexer.LEFT_CURLY
					|| token.getType() == DRLLexer.RIGHT_SQUARE
					|| token.getType() == DRLLexer.LEFT_SQUARE
					|| token.getType() == DRLLexer.RIGHT_PAREN
					|| token.getType() == DRLLexer.LEFT_PAREN
					|| token.getType() == DRLLexer.ARROW
					|| token.getType() == DRLLexer.LESS_EQUALS
					|| token.getType() == DRLLexer.LESS
					|| token.getType() == DRLLexer.GREATER_EQUALS
					|| token.getType() == DRLLexer.GREATER
					|| token.getType() == DRLLexer.NOT_EQUALS
					|| token.getType() == DRLLexer.EQUALS
					|| token.getType() == DRLLexer.COLON
					|| token.getType() == DRLLexer.SEMICOLON) {
				tree.setEditorElementType(DroolsEditorType.SYMBOL);
			} else if (token.getType() == DRLLexer.ID
					|| token.getType() == DRLLexer.DOT_STAR
					|| token.getType() == DRLLexer.VT_GLOBAL_ID
					|| token.getType() == DRLLexer.VT_FUNCTION_ID
					|| token.getType() == DRLLexer.VT_QUERY_ID
					|| token.getType() == DRLLexer.VT_RULE_ID
					|| token.getType() == DRLLexer.VT_ENTRYPOINT_ID) {
				tree.setEditorElementType(DroolsEditorType.IDENTIFIER);
			} else if (token.getType() == DRLLexer.VT_DATA_TYPE) {
				tree.setEditorElementType(DroolsEditorType.IDENTIFIER_TYPE);
			} else if (token.getType() == DRLLexer.VT_PATTERN_TYPE) {
				tree.setEditorElementType(DroolsEditorType.IDENTIFIER_PATTERN);
			}

		}
		return tree;
	}

	/**
	 * Create a DroolsTree and keeps the char offset info.
	 */
	public Object create(int tokenType, Token fromToken, String text) {
		if (fromToken instanceof DroolsToken) {
			DroolsTree result = (DroolsTree) super.create(tokenType, fromToken,
					text);
			result
					.setStartCharOffset(((DroolsToken) fromToken)
							.getStartIndex());
			if (text == null) {
				result.setEndCharOffset(((DroolsToken) fromToken)
						.getStopIndex());
			} else {
				result.setEndCharOffset(result.getStartCharOffset()
						+ text.length());
			}
			return result;
		}
		return super.create(tokenType, fromToken, text);
	}

	/**
	 * Add a child to the tree t. If t does not have start offset info, it set t
	 * start offset info from child. And always set t end char offset from
	 * child.
	 * 
	 * The exception is RIGHT_PAREN, it is used just to keep the char set info
	 * but not added on t.
	 * 
	 * @param t
	 *            parent tree
	 * @param child
	 *            child tree
	 */
	public void addChild(Object t, Object child) {
		if (t != null && child != null) {
			if (t instanceof DroolsTree && child instanceof DroolsTree) {
				DroolsTree tParent = (DroolsTree) t;
				DroolsTree tChild = (DroolsTree) child;

				if (0 >= tParent.getStartCharOffset()) {
					tParent.setStartCharOffset(tChild.getStartCharOffset());
					tParent.setEndCharOffset(tChild.getEndCharOffset());
				}
				if (0 < tParent.getChildCount()) {
					tParent.setEndCharOffset(tChild.getEndCharOffset());
				}
			}
			if (DRLLexer.RIGHT_PAREN != ((Tree) child).getType()) {
				((Tree) t).addChild((Tree) child);
			}
		}
	}

	/**
	 * Overrides createToken, returning a DroolsToken instead of CommonToken
	 */
	public Token createToken(int tokenType, String text) {
		return new DroolsToken(tokenType, text);
	}

	/**
	 * Overrides createToken, returning a DroolsToken instead of CommonToken
	 */
	public Token createToken(Token fromToken) {
		return new DroolsToken(fromToken);
	}
}