From: Alexandre Terrasa <alexandre@moz-code.org>
Date: Fri, 17 Jul 2015 17:31:48 +0000 (-0400)
Subject: nitj: introduce JavaCompilerVisitor to generate java code
X-Git-Tag: v0.7.7~17^2~38
X-Git-Url: http://nitlanguage.org

nitj: introduce JavaCompilerVisitor to generate java code

Signed-off-by: Alexandre Terrasa <alexandre@moz-code.org>
---

diff --git a/src/compiler/java_compiler.nit b/src/compiler/java_compiler.nit
index 3d1384d..be527f5 100644
--- a/src/compiler/java_compiler.nit
+++ b/src/compiler/java_compiler.nit
@@ -194,12 +194,44 @@ class JavaCompiler
 		return file
 	end
 
+	# Kind of visitor to use
+	type VISITOR: JavaCompilerVisitor
+
+	# Initialize a visitor specific for the compiler engine
+	fun new_visitor(filename: String): VISITOR do
+		return new JavaCompilerVisitor(self, new_file(filename))
+	end
+
 	# Compile Nit code to Java
 	fun do_compilation do
 		modelbuilder.toolcontext.info("NOT YET IMPLEMENTED", 0)
 	end
 end
 
+# The class visiting the AST
+#
+# A visitor is attached to one JavaCodeFile it writes into.
+class JavaCompilerVisitor
+	super Visitor
+
+	# JavaCompiler used with this visitor
+	type COMPILER: JavaCompiler
+
+	# The associated compiler
+	var compiler: JavaCompiler
+
+	# The file to write generated code into
+	var file: JavaCodeFile
+
+	# Code generation
+
+	# Add a line (will be suffixed by `\n`)
+	fun add(line: String) do file.lines.add("{line}\n")
+
+	# Add a new partial line (no `\n` suffix)
+	fun addn(line: String) do file.lines.add(line)
+end
+
 # A file containing Java code.
 class JavaCodeFile