From d05bd3592050a9496dd87bcd8a49f8fdc8b6b58d Mon Sep 17 00:00:00 2001
From: Nathanael Sensfelder <SpamShield0@MultiAgentSystems.org>
Date: Fri, 21 Jul 2017 16:50:25 +0200
Subject: Adds output support.

---
 ast-to-instr/src/Functions.java | 21 ++++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

(limited to 'ast-to-instr/src/Functions.java')

diff --git a/ast-to-instr/src/Functions.java b/ast-to-instr/src/Functions.java
index b500f21..d4d0593 100644
--- a/ast-to-instr/src/Functions.java
+++ b/ast-to-instr/src/Functions.java
@@ -6,15 +6,26 @@ public class Functions
       final IDs... params
    )
    {
-      System.out.print("[FUN] (");
-      System.out.print(function_name);
+      add_entry(Main.get_main_output(), function_name, params);
+   }
+
+   public static void add_entry
+   (
+      final OutputFile output,
+      final String function_name,
+      final IDs... params
+   )
+   {
+      output.write("(set_function ");
+
+      output.write(function_name);
 
       for (final IDs param: params)
       {
-         System.out.print(" " + param.get_value());
+         output.write(" " + param.get_value());
       }
 
-      System.out.println(")");
-      /* TODO */
+      output.write(")");
+      output.insert_newline();
    }
 }
-- 
cgit v1.2.3-70-g09d2