open Assert
open X86
open Ll
open Backend

(* Do NOT modify this file -- we will overwrite it with our *)
(* own version when we test your project.                   *)

(* These tests will be used to grade your assignment *)

let size_ty_tests =
  [ ("ty_size1", assert_eqf (fun () -> size_ty [] I1) 8)
  ; ("ty_size3", assert_eqf (fun () -> size_ty [] I64) 8)
  ; ("ty_size4", assert_eqf (fun () -> size_ty [] (Ptr Void)) 8)
  ; ("ty_size2", assert_eqf (fun () -> size_ty [] (Ptr I1)) 8)
  ; ("ty_size5", assert_eqf (fun () -> size_ty [] (Array (3, I64))) 24)
  ; ("ty_size6", assert_eqf
    (fun () -> size_ty [] (Struct [I64; I1; Ptr I8; Ptr I64; Array (10, I1) ])) 112)
  ; ("ty size7", assert_eqf
    (fun () -> size_ty [("O", I1);("S", I64)] (Struct [Namedt "O"; (Array (2, Namedt "S"))])) 24)
  ]

let arg_loc_tests =
  []

let exec_e2e_ast ll_ast args extra_files =
  let output_path = !Platform.output_path in
  let dot_s_file = Platform.gen_name output_path "test" ".s" in
  let exec_file = Platform.gen_name output_path "exec" "" in
  let asm_ast = Backend.compile_prog ll_ast in
  let asm_str = X86.string_of_prog asm_ast in
  let _ = Driver.write_file dot_s_file asm_str in
  let _ = Platform.link (dot_s_file::extra_files) exec_file in
  let result = Driver.run_executable args exec_file in
  let _ = Platform.sh (Printf.sprintf "rm -f %s %s" dot_s_file exec_file) Platform.ignore_error in
  let _ = Platform.verb @@ Printf.sprintf "** Executable exited with: %d\n" result in
  Int64.of_int result


let exec_e2e_file path args =
  let ast = Driver.parse_ll_file path in
  exec_e2e_ast ast args []

let io_test path args =
  let ll_ast = Driver.parse_ll_file path in
  let output_path = !Platform.output_path in
  let dot_s_file = Platform.gen_name output_path "test" ".s" in
  let exec_file = Platform.gen_name output_path "exec" "" in
  let tmp_file = Platform.gen_name output_path "tmp" ".txt" in
  let asm_ast = Backend.compile_prog ll_ast in
  let asm_str = X86.string_of_prog asm_ast in
  let _ = Driver.write_file dot_s_file asm_str in
  let _ = Platform.link (dot_s_file::["cinterop.c"]) exec_file in
  let args = String.concat " " args in
  let result = Driver.run_program args exec_file tmp_file in
  let _ = Platform.sh (Printf.sprintf "rm -f %s %s %s" dot_s_file exec_file tmp_file) Platform.ignore_error in
  let _ = Platform.verb @@ Printf.sprintf "** Executable output:\n%s\n" result in
  result

let c_link_test c_files path args =
  let ll_ast = Driver.parse_ll_file path in
  let output_path = !Platform.output_path in
  let dot_s_file = Platform.gen_name output_path "test" ".s" in
  let exec_file = Platform.gen_name output_path "exec" "" in
  let asm_ast = Backend.compile_prog ll_ast in
  let asm_str = X86.string_of_prog asm_ast in
  let _ = Driver.write_file dot_s_file asm_str in
  let _ = Platform.link (dot_s_file::c_files) exec_file in
  let args = String.concat " " args in
  let result = Driver.run_executable args exec_file in
  let _ = Platform.sh (Printf.sprintf "rm -f %s %s" dot_s_file exec_file) Platform.ignore_error in
    Int64.of_int result

let executed tests =
  List.map (fun (fn, ans) ->
      fn, assert_eqf (fun () -> exec_e2e_file fn "") ans)
    tests

let executed_io tests =
  List.map (fun (fn, args, ans) ->
      (fn ^ ":" ^ (String.concat " " args)), assert_eqf (fun () -> io_test fn args) ans)
    tests

let executed_c_link tests =
  List.map (fun (c_file, fn, args, ans) ->
      (fn ^ ":" ^ (String.concat " " args)), assert_eqf (fun () -> c_link_test c_file fn args) ans)
    tests

let binop_tests =
  [ "llprograms/add.ll", 14L
  ; "llprograms/sub.ll", 1L
  ; "llprograms/mul.ll", 45L
  ; "llprograms/and.ll", 0L
  ; "llprograms/or.ll",  1L
  ; "llprograms/xor.ll", 0L
  ; "llprograms/shl.ll", 168L
  ; "llprograms/lshr.ll", 10L
  ; "llprograms/ashr.ll", 5L ]

let calling_convention_tests =
  [ "llprograms/call.ll", 42L
  ; "llprograms/call1.ll", 17L
  ; "llprograms/call2.ll", 19L
  ; "llprograms/call3.ll", 34L
  ; "llprograms/call4.ll", 34L
  ; "llprograms/call5.ll", 24L
  ; "llprograms/call6.ll", 26L
  ]

let memory_tests =
  [ "llprograms/alloca1.ll", 17L
  ; "llprograms/alloca2.ll", 17L
  ; "llprograms/global1.ll", 12L
  ]

let terminator_tests =
  [ "llprograms/return.ll", 0L
  ; "llprograms/return42.ll", 42L
  ; "llprograms/br1.ll", 9L
  ; "llprograms/br2.ll", 17L
  ; "llprograms/cbr1.ll", 7L
  ; "llprograms/cbr2.ll", 9L
  ; "llprograms/duplicate_lbl.ll", 1L
  ]

let bitcast_tests =
  [ "llprograms/bitcast1.ll", 3L
  ]

let gep_tests =
  [ "llprograms/gep1.ll", 6L
  ; "llprograms/gep2.ll", 4L
  ; "llprograms/gep3.ll", 1L
  ; "llprograms/gep4.ll", 2L
  ; "llprograms/gep5.ll", 4L
  ; "llprograms/gep6.ll", 7L
  ; "llprograms/gep7.ll", 7L
  ; "llprograms/gep8.ll", 2L
  ]

let io_tests =
  [ "llprograms/helloworld.ll", [], "hello, world!"
  ; "llprograms/string1.ll", [], "hello, world!hello, world!"
  ; "llprograms/callback1.ll", [], "38"
  ; "llprograms/args1.ll", ["hello"], "argc < 3"
  ; "llprograms/args1.ll", ["hello"; "compilerdesign"], "hellocompilerdesign"
  ; "llprograms/args1.ll", ["hello"; "compilerdesign"; "foo"], "argc > 3"
  ]




(* Hidden *)
let hidden_large_tests =
  []

let large_tests = [ "llprograms/list1.ll", 3L
                  ; "llprograms/cbr.ll", 42L
                  ; "llprograms/factorial.ll", 120L
                  ; "llprograms/factrect.ll", 120L
                  ; "llprograms/duplicate_factorial.ll", 240L
                  ]



let tests : suite =
  [ GradedTest("size_ty tests", 5, size_ty_tests)
  ; GradedTest("arg_loc tests", 5, arg_loc_tests)
  ; GradedTest("executed binop tests", 5, executed binop_tests)
  ; GradedTest("terminator tests", 10, executed terminator_tests)
  ; GradedTest("memory tests", 10, executed memory_tests)
  ; GradedTest("calling convention tests", 15, executed calling_convention_tests)
  ; GradedTest("bitcast tests", 2, executed bitcast_tests)
  ; GradedTest("gep tests", 10, executed gep_tests)
  ; GradedTest("large tests", 10, executed large_tests)
  ; GradedTest("hidden large tests", 18, hidden_large_tests)
  ; GradedTest("io tests", 10, executed_io io_tests)
  ]


let graded_tests : suite =
  tests