118 lines
2.7 KiB
OCaml
118 lines
2.7 KiB
OCaml
open Cmm
|
|
|
|
type t =
|
|
{ mutable name: string;
|
|
stamp: int;
|
|
typ: Cmm.machtype_component;
|
|
mutable loc: location;
|
|
mutable spill: bool;
|
|
mutable interf: t list;
|
|
mutable prefer: (t * int) list;
|
|
mutable degree: int;
|
|
mutable spill_cost: int;
|
|
mutable visited: bool }
|
|
|
|
and location =
|
|
Unknown
|
|
| Reg of int
|
|
| Stack of stack_location
|
|
|
|
and stack_location =
|
|
Local of int
|
|
| Incoming of int
|
|
| Outgoing of int
|
|
|
|
type reg = t
|
|
|
|
let dummy =
|
|
{ name = ""; stamp = 0; typ = Int; loc = Unknown; spill = false;
|
|
interf = []; prefer = []; degree = 0; spill_cost = 0; visited = false }
|
|
|
|
let currstamp = ref 0
|
|
let reg_list = ref([] : t list)
|
|
|
|
let new ty =
|
|
let r = { name = ""; stamp = !currstamp; typ = ty; loc = Unknown;
|
|
spill = false; interf = []; prefer = []; degree = 0;
|
|
spill_cost = 0; visited = false } in
|
|
reg_list := r :: !reg_list;
|
|
incr currstamp;
|
|
r
|
|
|
|
let newv tyv =
|
|
let n = Array.length tyv in
|
|
let rv = Array.new n dummy in
|
|
for i = 0 to n-1 do rv.(i) <- new tyv.(i) done;
|
|
rv
|
|
|
|
let clone r =
|
|
let nr = new r.typ in
|
|
nr.name <- r.name;
|
|
nr
|
|
|
|
let at_location ty loc =
|
|
let r = { name = ""; stamp = !currstamp; typ = ty; loc = loc; spill = false;
|
|
interf = []; prefer = []; degree = 0; spill_cost = 0;
|
|
visited = false } in
|
|
incr currstamp;
|
|
r
|
|
|
|
let reset() = reg_list := []
|
|
let all_registers() = !reg_list
|
|
let num_registers() = !currstamp
|
|
|
|
let reinit_reg r =
|
|
r.loc <- Unknown;
|
|
r.interf <- [];
|
|
r.prefer <- [];
|
|
r.degree <- 0;
|
|
r.spill_cost <- 0
|
|
|
|
let reinit() =
|
|
List.iter reinit_reg !reg_list
|
|
|
|
module RegOrder =
|
|
struct
|
|
type t = reg
|
|
let compare r1 r2 = r1.stamp - r2.stamp
|
|
end
|
|
|
|
module Set = Set.Make(RegOrder)
|
|
module Map = Map.Make(RegOrder)
|
|
|
|
let add_set_array s v =
|
|
match Array.length v with
|
|
0 -> s
|
|
| 1 -> Set.add v.(0) s
|
|
| n -> let rec add_all i =
|
|
if i >= n then s else Set.add v.(i) (add_all(i+1))
|
|
in add_all 0
|
|
|
|
let diff_set_array s v =
|
|
match Array.length v with
|
|
0 -> s
|
|
| 1 -> Set.remove v.(0) s
|
|
| n -> let rec remove_all i =
|
|
if i >= n then s else Set.remove v.(i) (remove_all(i+1))
|
|
in remove_all 0
|
|
|
|
let inter_set_array s v =
|
|
match Array.length v with
|
|
0 -> Set.empty
|
|
| 1 -> if Set.mem v.(0) s
|
|
then Set.add v.(0) Set.empty
|
|
else Set.empty
|
|
| n -> let rec inter_all i =
|
|
if i >= n then Set.empty
|
|
else if Set.mem v.(i) s then Set.add v.(i) (inter_all(i+1))
|
|
else inter_all(i+1)
|
|
in inter_all 0
|
|
|
|
let set_of_array v =
|
|
match Array.length v with
|
|
0 -> Set.empty
|
|
| 1 -> Set.add v.(0) Set.empty
|
|
| n -> let rec add_all i =
|
|
if i >= n then Set.empty else Set.add v.(i) (add_all(i+1))
|
|
in add_all 0
|