require_relative "../helper"
require "register/interpreter"

module Register
  module Ticker
    include AST::Sexp

    def setup
      Register.machine.boot
      do_clean_compile
      Vm.compile_ast( @input )
      Collector.collect_space
      @interpreter = Interpreter.new
      @interpreter.start Register.machine.init
    end

    # must be after boot, but before main compile, to define method
    def do_clean_compile
    end

    def check_chain should
      should.each_with_index do |name , index|
        got = ticks(1)
        assert_equal got.class ,name , "Wrong class for #{index+1}, expect #{name} , got #{got}"
      end
    end

    def check_return val
      assert_equal Parfait::Message , @interpreter.get_register(:r0).class
      assert_equal val , @interpreter.get_register(:r0).return_value
    end

    def ticks num
      last = nil
      num.times do
        last = @interpreter.instruction
        @interpreter.tick
      end
      return last
    end

    def show_ticks
      classes = []
      tick = 1
      begin
        while true and (classes.length < 200)
          cl = ticks(1).class
          tick += 1
          classes << cl
          break if cl == NilClass
        end
      rescue => e
        puts "Error at tick #{tick}"
        puts e
        puts e.backtrace
      end
      str = classes.to_s.gsub("Register::","")
      str.split(",").each_slice(5).each do |line|
        puts "            " + line.join(",") + ","
      end
      puts "length = #{classes.length}"
      exit(1)
    end
  end
end