(* RUN: rm -rf %t.builddir
 * RUN: mkdir -p %t.builddir
 * RUN: cp %s %t.builddir
 * RUN: %ocamlopt -g -warn-error A llvm.cmxa llvm_target.cmxa %t.builddir/target.ml -o %t
 * RUN: %t %t.bc
 * XFAIL: vg_leak
 *)

(* Note: It takes several seconds for ocamlopt to link an executable with
         libLLVMCore.a, so it's better to write a big test than a bunch of
         little ones. *)

open Llvm
open Llvm_target


let context = global_context ()
let i32_type = Llvm.i32_type context
let i64_type = Llvm.i64_type context

(* Tiny unit test framework - really just to help find which line is busted *)
let print_checkpoints = false

let _ =
  Printexc.record_backtrace true

let assert_equal a b =
  if a <> b then failwith "assert_equal"


(*===-- Fixture -----------------------------------------------------------===*)

let filename = Sys.argv.(1)
let m = create_module context filename


(*===-- Target Data -------------------------------------------------------===*)

let test_target_data () =
  let layout = "e-p:32:32:32-S32-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-" ^
               "f16:16:16-f32:32:32-f64:32:64-f128:128:128-v64:32:64-v128:32:128-" ^
               "a0:0:64-n32" in
  let td     = DataLayout.create layout in
  let sty    = struct_type context [| i32_type; i64_type |] in
  
  assert_equal (DataLayout.as_string td) layout;
  assert_equal (byte_order td) Endian.Little;
  assert_equal (pointer_size td) 4;
  assert_equal (intptr_type td) i32_type;
  assert_equal (size_in_bits td sty) (Int64.of_int 96);
  assert_equal (store_size td sty) (Int64.of_int 12);
  assert_equal (abi_size td sty) (Int64.of_int 12);
  assert_equal (stack_align td sty) 4;
  assert_equal (preferred_align td sty) 8;
  assert_equal (preferred_align_of_global td (declare_global sty "g" m)) 8;
  assert_equal (element_at_offset td sty (Int64.of_int 1)) 0;
  assert_equal (offset_of_element td sty 1) (Int64.of_int 4);
  DataLayout.dispose td


(*===-- Driver ------------------------------------------------------------===*)

let _ =
  test_target_data ();
  dispose_module m
