Check ancestor in ANTLR

AntrlExt.java

import org.antlr.v4.runtime.ParserRuleContext;

public class AntlrExt {
  public static boolean underContext(ParserRuleContext curr, ParserRuleContext target, ParserRuleContext limit) {
    do {
      curr = curr.getParent();
    } while (curr.getRuleIndex() != target.getRuleIndex() && curr.getRuleIndex() != limit.getRuleIndex());
    return curr.getRuleIndex() == target.getRuleIndex();
  }
}