extra.orsl-glsl-parser.0.4.5-alpha3.source-code.ASTPrinter.kt Maven / Gradle / Ivy

Go to download
import org.antlr.v4.runtime.ParserRuleContext
import org.antlr.v4.runtime.RuleContext
import org.openrndr.orsl.glslparser.antlr.GLSLParser

class AstPrinter {
    var ignoringWrappers = true

    fun print(ctx: RuleContext) {
        explore(ctx, 0)
    }

    private fun explore(ctx: RuleContext, indentation: Int) {
        val toBeIgnored = ignoringWrappers && ctx.childCount == 1 && ctx.getChild(0) is ParserRuleContext
        if (!toBeIgnored) {
            val ruleName: String = GLSLParser.ruleNames.get(ctx.ruleIndex)
            for (i in 0 until indentation) {
                print("  ")
            }
            println("$ruleName ${ctx.text}")
        }
        for (i in 0 until ctx.childCount) {
            val element = ctx.getChild(i)
            if (element is RuleContext) {
                explore(element, indentation + if (toBeIgnored) 0 else 1)
            }
        }
    }
}