Похоже, что обычный посетитель AST должен делать, не так ли? Предполагая, что у вас есть классы, интерфейсы и атрибуты на вашем языке, это должно быть легко для обработки.
Правильная схема диаграммы - это другая проблема, которая может быть решена впоследствии.