kaze/orx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add OliveProgram scriptless replacement

Browse Source
This commit is contained in:
Edwin Jakobs
2020-05-01 15:04:43 +02:00
parent d53925efb1
commit 542ca90379
10 changed files with 113 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
orx-kotlin-parser/src/main/kotlin/ExtractProgram.kt
View File

@@ -3,6 +3,7 @@ package org.openrndr.extra.kotlinparser
import org.antlr.v4.runtime.CharStreams
import org.antlr.v4.runtime.CommonTokenStream
import org.antlr.v4.runtime.ParserRuleContext
import org.antlr.v4.runtime.RuleContext
import org.antlr.v4.runtime.misc.Interval
import org.antlr.v4.runtime.tree.ParseTreeWalker
import org.openrndr.extra.kotlin.antlr.KotlinLexer
@@ -10,14 +11,14 @@ import org.openrndr.extra.kotlin.antlr.KotlinParser
import org.openrndr.extra.kotlin.antlr.KotlinParserBaseListener
import java.io.File
fun ParserRuleContext.verbatimText(marginLeft:Int = 0, marginRight:Int = 0): String {
fun ParserRuleContext.verbatimText(marginLeft: Int = 0, marginRight: Int = 0): String {
val startIndex = start.startIndex + marginLeft
val stopIndex = stop.stopIndex - marginRight
val interval = Interval(startIndex, stopIndex)
return start.inputStream.getText(interval)
}
class ImportsExtractor : KotlinParserBaseListener() {
class ImportsExtractor(val ruleNames: List<String>) : KotlinParserBaseListener() {
var result: String? = null
override fun enterImportList(ctx: KotlinParser.ImportListContext) {
@@ -25,29 +26,34 @@ class ImportsExtractor : KotlinParserBaseListener() {
}
}
class LambdaExtractor(val lambdaName: String) : KotlinParserBaseListener() {
class LambdaExtractor(val ruleNames: List<String>, val lambdaName: String) : KotlinParserBaseListener() {
fun RuleContext.named(): String {
return ruleNames[this.ruleIndex]
}
var result: String? = null
override fun enterAnnotatedLambda(ctx: KotlinParser.AnnotatedLambdaContext?) {
val puec = ctx!!.parent!!.parent!!.parent!! as KotlinParser.PostfixUnaryExpressionContext
val identifier = puec.primaryExpression().simpleIdentifier().Identifier().text
if (identifier == lambdaName) {
result = ctx.verbatimText(1, 1)
if (result == null) {
result = ctx.verbatimText(1, 1)
}
}
}
}
class ProgramSource(val imports: String, val programLambda: String)
fun extractProgram(source: String) : ProgramSource {
fun extractProgram(source: String, programIdentifier: String = "program"): ProgramSource {
val parser = KotlinParser(CommonTokenStream(KotlinLexer(CharStreams.fromString(source))))
val root = parser.kotlinFile()
val importsExtractor = ImportsExtractor()
// val rules = parser.ruleNames.toList()
// val pt = TreeUtils.toPrettyTree(root, rules)
val ruleNames = parser.ruleNames.toList()
val importsExtractor = ImportsExtractor(ruleNames)
ParseTreeWalker.DEFAULT.walk(importsExtractor, root)
val lambdaExtractor = LambdaExtractor("program")
val lambdaExtractor = LambdaExtractor(ruleNames, programIdentifier)
ParseTreeWalker.DEFAULT.walk(lambdaExtractor, root)
return ProgramSource(importsExtractor.result!!, lambdaExtractor.result!!)
}

2
orx-kotlin-parser/src/main/kotlin/TreeUtils.kt
View File

@@ -1,3 +1,5 @@
package org.openrndr.extra.kotlinparser
import org.antlr.v4.runtime.misc.Utils
import org.antlr.v4.runtime.tree.Tree
import org.antlr.v4.runtime.tree.Trees