c# - ANTLR4 Custom class for ParserRuleContexts -

antlr generates contexts derived parserrulecontext. have amount of logic , modified behaviour want applied each context. possible subclass parserrulecontext , somehow tell antlr generate contexts using custom base class instead? either all, or ideally, on context context basis. have far found documentation change class of parser itself.

at moment using way contexts declared partial , injecting code writing in partial class. needs done once per context there's lot of repetition , isn't nice.