From b79d717f1acf6478cae234cdce004636255b6f9d Mon Sep 17 00:00:00 2001
From: Danila Fedorin <danila.fedorin@gmail.com>
Date: Mon, 6 Aug 2018 23:43:46 -0700
Subject: [PATCH] Generate sprite data.

---
 src/chalk/compiler.cr | 19 ++++++++++++++++---
 src/chalk/table.cr    | 12 +++++++++---
 2 files changed, 25 insertions(+), 6 deletions(-)

diff --git a/src/chalk/compiler.cr b/src/chalk/compiler.cr
index 4d4e18b..a3d3044 100644
--- a/src/chalk/compiler.cr
+++ b/src/chalk/compiler.cr
@@ -140,9 +140,9 @@ module Chalk
         binary = instructions.map_with_index { |it, i| it.to_bin(table, instructions.size, i).to_u16 }
         binary.each do |inst|
           first = (inst >> 8).to_u8
-          dest.write_byte(first)
+          dest << first
           second = (inst & 0xff).to_u8
-          dest.write_byte(second)
+          dest << second
         end
       end
 
@@ -180,6 +180,15 @@ module Chalk
         names = collect_calls(table)
         names.delete "main"
 
+        sprite_bytes = [] of UInt8
+        offset = 0
+        table.sprites.each do |k, v|
+          data = v.sprite.encode
+          v.offset = offset
+          offset += data.size
+          sprite_bytes.concat data
+        end
+
         main_entry = table.get_function?("main").not_nil!
         all_instructions.concat create_code(main_entry.function.as(Trees::TreeFunction),
           table, Ir::JumpRelativeInstruction.new 0)
@@ -194,8 +203,12 @@ module Chalk
           all_instructions << Ir::ReturnInstruction.new
         end
 
+        binary = [] of UInt8
         file = File.open(@config.output, "w")
-        generate_binary(table, all_instructions, file)
+        generate_binary(table, all_instructions, binary)
+        binary.each do |byte|
+          file.write_byte byte
+        end
         file.close
       end
 
diff --git a/src/chalk/table.cr b/src/chalk/table.cr
index 7d4da62..3549fe6 100644
--- a/src/chalk/table.cr
+++ b/src/chalk/table.cr
@@ -35,10 +35,10 @@ module Chalk
     end
 
     class SpriteEntry
-      getter sprite : Sprite
-      getter addr : Int32
+      property sprite : Sprite
+      property offset : Int32
 
-      def initialize(@sprite, @addr = -1)
+      def initialize(@sprite, @offset = -1)
       end
     end
 
@@ -46,6 +46,12 @@ module Chalk
     class Table
       # Gets the parent of this table.
       getter parent
+      # Gets the functions hash.
+      getter functions
+      # Gets the variables hash.
+      getter vars
+      # Gets the sprites hash.
+      getter sprites
 
       def initialize(@parent : Table? = nil)
         @functions = {} of String => FunctionEntry