challenge-codinggame/coders-strike-back/code6.ml

let open Printf in
let debug = fprintf stderr in

let checkpoint_ray = 600 in

let module Vector = struct
  type t = {
    x : int;
    y : int
  }

  let to_string dst : string =
    (Printf.sprintf "{ x = %d ; y = %d }" dst.x dst.y)

  let create x y = 
    { x ; y }

  let equal dst src = 
    (dst.x = src.x) && (dst.y = src.y)

  let diff dst src =
    { x = (dst.x - src.x) ;
      y = (dst.y - src.y)
    }

  let mult src a =
    { x = src.x * a ;
      y = src.y * a 
    }

  let add dst src = 
    { x = (src.x + dst.x) ;
      y = (src.y + dst.y) 
    }

  let distance (dst:t) (src:t) = 
    (src.x - dst.x) * (src.x - dst.x) + (src.y - dst.y) * (src.y - dst.y)
    |> float_of_int
    |> sqrt
    |> int_of_float

  let is_touching dst src = 
    (distance dst src) < (checkpoint_ray / 2)

  let is_near dst src = 
    (distance dst src) < (checkpoint_ray * 2)

  let is_far dst src = 
    ((distance dst src) >= checkpoint_ray * 6)
end in   

let module Pod = struct
  type t = {
    mutable id : int ;
    mutable pos : Vector.t ;
    mutable dir : Vector.t ;
    mutable has_boost : bool ;
    mutable checkpoint_pos : Vector.t ;
    mutable checkpoint_dist : int ;
    mutable checkpoint_angle : int ;
    mutable output_str : string 
  }

  let create () = 
    { 
      id = 0 ;
      pos = Vector.create 0 0 ;
      dir = Vector.create 0 0 ;
      has_boost = true ;
      checkpoint_pos = Vector.create 0 0 ;
      checkpoint_dist = 0 ;
      checkpoint_angle = 0 ;
      output_str = "" 
    }  

  let use_boost pod =
    pod.has_boost <- false

  let update pod (pos, checkpoint_pos, checkpoint_dist, checkpoint_angle) = 
    pod.pos <- pos ;
    pod.checkpoint_pos <- checkpoint_pos ;
    pod.checkpoint_dist <- checkpoint_dist ;
    pod.checkpoint_angle <- checkpoint_angle ;
    pod


end in

let module CheckpointList = struct
  type t = {
    mem: Vector.t list ;
    cur: Vector.t option
  }

  let add ck_li elem = 
    match ck_li.cur with
    | None -> 
      { 
        mem = [] ;
        cur = Some elem 
      }
    | Some cur_elem -> 
      if (Vector.equal cur_elem elem) then
        ck_li
      else
        { 
          mem = cur_elem :: ck_li.mem ;
          cur = Some elem 
        }

  let create () = 
    { mem = [] ;
      cur = None
    }

  let inc (ck_li:t) elem = 
    (ck_li.mem)
    |> List.map (fun cur -> (Vector.equal elem cur))
    |> List.fold_left (fun acc elem -> acc || elem) false

end in

let module Game = struct
  type t = {
    pods : Pod.t array ;
    opponents : Pod.t array ;
    ticks : int ;
  }
end in

let module Strategy = struct
  let apply game pod = 
    if !first_turn then begin
      old_pod := pod ;
      first_turn := false
    end ;
    let movement = Vector.diff pod !old_pod in

    if (CheckpointList.inc !checkpoint_list checkpoint) then begin
      prerr_endline "List already includes this checkpoint" ;
    end else begin 
      checkpoint_list := (CheckpointList.add !checkpoint_list checkpoint) ;
      fprintf stderr "New checkpoint : %s\n%!" (Vector.to_string checkpoint);
    end ;

    (* output "x y thrust" : the target position + the power *)
    let is_aligned = checkpoint_angle > -2 && checkpoint_angle < 2 in
    let is_near_aligned = 
      (checkpoint_angle > -15 && checkpoint_angle <= -2) ||
      (checkpoint_angle >= 2 && checkpoint_angle < 15)
    in
    let is_wide_aligned =  
      (checkpoint_angle > -45 && checkpoint_angle <= -15) ||
      (checkpoint_angle >= 15 && checkpoint_angle < 45)
    in
    let is_aside = 
      (checkpoint_angle > -85 && checkpoint_angle <= -45) ||
      (checkpoint_angle >= 45 && checkpoint_angle < 85)
    in
    let is_behind = checkpoint_angle >= 85 || checkpoint_angle <= -85 in
    let use_boost = 
      (!tick > 100) && is_aligned && (Vector.is_far pod checkpoint) 
    in

    debug "angle = %d\n%!" checkpoint_angle ;

    let thrust = 
      (* droit devant *)
      if is_behind && (Vector.is_touching pod opponent) then 20
      else if is_aside && (Vector.is_touching pod opponent) then 20
      else if is_behind then 6
      else if is_aside && (Vector.is_near checkpoint pod) then 10
      else if is_wide_aligned && (Vector.is_near checkpoint pod) then 20
      else if is_near_aligned && (Vector.is_near checkpoint pod) then 40
      else 100
    in

    let power_str = match use_boost with
      | false -> string_of_int thrust
      | true -> "BOOST"
    in

    let target = Vector.add checkpoint (Vector.mult movement (-3)) in

    printf "%d %d %s\n%!" target.x target.y power_str ;


end in

let parse_pod_line () =
  let line = input_line stdin in
  let build_results_fn = (
    fun x y checkpointx checkpointy checkpointdist checkpointangle -> 
      (Vector.create x y), 
      (Vector.create checkpointx checkpointy), 
      checkpointdist, 
      checkpointangle
  ) in
  Scanf.sscanf line "%d %d %d %d %d %d" build_results_fn
in

let map_center = Vector.create (16000 / 2) (9000 / 2) in 
let first_turn = ref true in
let old_pod = ref (Vector.create 0 0) in

let tick = ref 0 in
let checkpoint_list = ref (CheckpointList.create ()) in

(* laps : le nombre de tours à effectuer pour finir la course. *)
let parse_init () =
  let laps =
    let line = input_line stdin in
    Scanf.sscanf line "%d" (fun x -> x) 
  in
  let checkpoint_count =
    let line = input_line stdin in
    Scanf.sscanf line "%d" (fun x -> x)
  in
  let parse_checkpoint () = 
    let line = input_line stdin in
    Scanf.sscanf line "%d %d" (fun x y -> 
        Vector.create x y
      )
  in
  for i = 1 to checkpoint_count do
    let checkpoint = parse_checkpoint () in
    checkpoint_list := CheckpointList.add !checkpoint_list checkpoint ;
    ()
  done
in

let pods = Array.make 2 (Pod.create ()) in
let game = Game.create () in

(* game loop *)
while true do
  (* read pod1 line & update game data *)
  parse_pod_line () |> Pod.update game.pods.(0) ;
  parse_pod_line () |> Pod.update game.pods.(1) ;
  parse_pod_line () |> Pod.update game.opponents.(0) ;
  parse_pod_line () |> Pod.update game.opponents.(1) ;
  
  for i = 0 to 1 do
    game.pods.(i)
    |> Pod.update pod1_data
    |> Strategy.apply
    |> fun pod -> print_endline pod.output_str ; 
  done

  tick := !tick + 1 ;
  old_pod := pod ;
done;