Вы можете легко изменить грамматику Java самостоятельно (при условии, что вы немного знаете ANTLR). Написание грамматики лексера и синтаксического анализатора - самая трудная часть, которая, в данном случае, уже сделана для вас.
Если вы хотите знать, как разрешить грамматике синтаксического анализатора выводить AST вместо простого дерева разбора, посмотрите на эти предыдущие вопросы и ответы: Как вывести AST, построенный с использованием ANTLR?