module Soml
  Compiler.class_eval do
#    if - attr_reader  :cond, :if_true, :if_false

    def on_if_statement statement
      branch_type , condition , if_true , if_false = *statement
      condition = condition.first
      # to execute the logic as the if states it, the blocks are the other way around
      # so we can the jump over the else if true ,
      # and the else joins unconditionally after the true_block
      merge_block = @method.source.new_block "if_merge"    # last one, created first
      true_block =  @method.source.new_block "if_true"     # second, linked in after current, before merge
      false_block = @method.source.new_block "if_false"    # directly next in order, ie if we don't jump we land here

      reset_regs
      is = process(condition)
      branch_class = Object.const_get "Register::Is#{branch_type.capitalize}"
      add_code branch_class.new( condition , true_block )

      # compile the true block (as we think of it first, even it is second in sequential order)
      @method.source.current true_block

      reset_regs
      last = process_all(if_true).last

      # compile the false block
      @method.source.current false_block
      reset_regs
      last = process_all(if_false).last if if_false
      add_code Register::Branch.new(statement, merge_block )

      #puts "compiled if: end"
      @method.source.current merge_block

      #TODO should return the union of the true and false types
      last
    end
  end
end