Working with ANTLR4 contexts and map