Can I get ParserRuleContext from ASTNode?

[oovm]

ANTLRPsiNode created via createElement seems unable to obtain ParserRuleContext.

ParserDefinition.createElement(node: ASTNode): PsiElement =  ANTLRPsiNode(node)

I want to get child nodes by judging ParserRuleContext's type.

But the child node is also a ParserRuleContext, I don’t know how to convert it to a PsiElement

[ris58h]

You can obtain PsiElement from ASTNode via getPsi method. In Kotlin it's just psi property. So node.psi will do the trick.

[bjansen]

Right, normally you would walk a PSI tree using psiElement.getChildren(), then call child.getNode().getElementType() to know whick kind of node it is. You can't access an instance of ParserRuleContext, but you can do something like this:

var rules  = PSIElementTypeFactory.getRuleIElementTypes(myLanguage);

if (child.getNode().getElementType().equals(rules.get(MyParser.RULE_myRule))) {
    // `child` was created from a parser rule named `myRule`
}