/**
 ******************************************************************************
 * Xenia : Xbox 360 Emulator Research Project                                 *
 ******************************************************************************
 * Copyright 2013 Ben Vanik. All rights reserved.                             *
 * Released under the BSD license - see LICENSE in the root for more details. *
 ******************************************************************************
 */

#include <alloy/backend/ivm/ivm_assembler.h>

#include <alloy/backend/backend.h>
#include <alloy/backend/ivm/ivm_intcode.h>
#include <alloy/backend/ivm/ivm_function.h>
#include <alloy/backend/ivm/tracing.h>
#include <alloy/hir/hir_builder.h>
#include <alloy/hir/label.h>
#include <alloy/runtime/runtime.h>

using namespace alloy;
using namespace alloy::backend;
using namespace alloy::backend::ivm;
using namespace alloy::hir;
using namespace alloy::runtime;


IVMAssembler::IVMAssembler(Backend* backend) :
    source_map_arena_(128 * 1024),
    Assembler(backend) {
}

IVMAssembler::~IVMAssembler() {
  alloy::tracing::WriteEvent(EventType::AssemblerDeinit({
  }));
}

int IVMAssembler::Initialize() {
  int result = Assembler::Initialize();
  if (result) {
    return result;
  }

  alloy::tracing::WriteEvent(EventType::AssemblerInit({
  }));

  return result;
}

void IVMAssembler::Reset() {
  intcode_arena_.Reset();
  source_map_arena_.Reset();
  scratch_arena_.Reset();
  Assembler::Reset();
}

int IVMAssembler::Assemble(
    FunctionInfo* symbol_info, HIRBuilder* builder,
    DebugInfo* debug_info, Function** out_function) {
  IVMFunction* fn = new IVMFunction(symbol_info);
  fn->set_debug_info(debug_info);

  TranslationContext ctx;
  ctx.access_callbacks = backend_->runtime()->access_callbacks();
  ctx.register_count = 0;
  ctx.intcode_count = 0;
  ctx.intcode_arena = &intcode_arena_;
  ctx.source_map_count = 0;
  ctx.source_map_arena = &source_map_arena_;
  ctx.scratch_arena = &scratch_arena_;
  ctx.label_ref_head = NULL;

  // Reset label tags as we use them.
  builder->ResetLabelTags();

  // Function prologue.

  auto block = builder->first_block();
  while (block) {
    Label* label = block->label_head;
    while (label) {
      label->tag = (void*)(0x80000000 | ctx.intcode_count);
      label = label->next;
    }

    Instr* i = block->instr_head;
    while (i) {
      int result = TranslateIntCodes(ctx, i);
      i = i->next;
    }
    block = block->next;
  }

  // Function epilogue.

  // Fixup label references.
  LabelRef* label_ref = ctx.label_ref_head;
  while (label_ref) {
    label_ref->instr->src1_reg = (uint32_t)label_ref->label->tag & ~0x80000000;
    label_ref = label_ref->next;
  }

  fn->Setup(ctx);

  *out_function = fn;
  return 0;
}