Add 'scriptless' mode to orx-olive
This commit is contained in:
47
orx-kotlin-parser/src/main/kotlin/TreeUtils.kt
Normal file
47
orx-kotlin-parser/src/main/kotlin/TreeUtils.kt
Normal file
@@ -0,0 +1,47 @@
|
||||
import org.antlr.v4.runtime.misc.Utils
|
||||
import org.antlr.v4.runtime.tree.Tree
|
||||
import org.antlr.v4.runtime.tree.Trees
|
||||
|
||||
object TreeUtils {
|
||||
/** Platform dependent end-of-line marker */
|
||||
val Eol = System.lineSeparator()
|
||||
|
||||
/** The literal indent char(s) used for pretty-printing */
|
||||
const val Indents = " "
|
||||
private var level = 0
|
||||
|
||||
/**
|
||||
* Pretty print out a whole tree. [.getNodeText] is used on the node payloads to get the text
|
||||
* for the nodes. (Derived from Trees.toStringTree(....))
|
||||
*/
|
||||
fun toPrettyTree(t: Tree, ruleNames: List<String>): String {
|
||||
level = 0
|
||||
return process(t, ruleNames).replace("(?m)^\\s+$".toRegex(), "").replace("\\r?\\n\\r?\\n".toRegex(), Eol)
|
||||
}
|
||||
|
||||
private fun process(t: Tree, ruleNames: List<String>): String {
|
||||
if (t.getChildCount() === 0) return Utils.escapeWhitespace(Trees.getNodeText(t, ruleNames), false)
|
||||
val sb = StringBuilder()
|
||||
sb.append(lead(level))
|
||||
level++
|
||||
val s: String = Utils.escapeWhitespace(Trees.getNodeText(t, ruleNames), false)
|
||||
sb.append("$s ")
|
||||
for (i in 0 until t.getChildCount()) {
|
||||
sb.append(process(t.getChild(i), ruleNames))
|
||||
}
|
||||
level--
|
||||
sb.append(lead(level))
|
||||
return sb.toString()
|
||||
}
|
||||
|
||||
private fun lead(level: Int): String {
|
||||
val sb = StringBuilder()
|
||||
if (level > 0) {
|
||||
sb.append(Eol)
|
||||
for (cnt in 0 until level) {
|
||||
sb.append(Indents)
|
||||
}
|
||||
}
|
||||
return sb.toString()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user