Chris@42: (*
Chris@42:  * Copyright (c) 1997-1999 Massachusetts Institute of Technology
Chris@42:  * Copyright (c) 2003, 2007-14 Matteo Frigo
Chris@42:  * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology
Chris@42:  *
Chris@42:  * This program is free software; you can redistribute it and/or modify
Chris@42:  * it under the terms of the GNU General Public License as published by
Chris@42:  * the Free Software Foundation; either version 2 of the License, or
Chris@42:  * (at your option) any later version.
Chris@42:  *
Chris@42:  * This program is distributed in the hope that it will be useful,
Chris@42:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@42:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@42:  * GNU General Public License for more details.
Chris@42:  *
Chris@42:  * You should have received a copy of the GNU General Public License
Chris@42:  * along with this program; if not, write to the Free Software
Chris@42:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
Chris@42:  *
Chris@42:  *)
Chris@42: 
Chris@42: type variable = 
Chris@42:       (* temporary variables generated automatically *)
Chris@42:   | Temporary of int
Chris@42:       (* memory locations, e.g., array elements *)
Chris@42:   | Locative of (Unique.unique * Unique.unique *
Chris@42: 		   (int -> string) * int * string)
Chris@42:       (* constant values, e.g., twiddle factors *)
Chris@42:   | Constant of (Unique.unique * string)
Chris@42: 
Chris@42: let hash v = Hashtbl.hash v
Chris@42: 
Chris@42: let same a b = (a == b)
Chris@42: 
Chris@42: let is_constant = function
Chris@42:   | Constant _ -> true
Chris@42:   | _ -> false
Chris@42: 
Chris@42: let is_temporary = function
Chris@42:   | Temporary _ -> true
Chris@42:   | _ -> false
Chris@42: 
Chris@42: let is_locative = function
Chris@42:   | Locative _ -> true
Chris@42:   | _ -> false
Chris@42: 
Chris@42: let same_location a b = 
Chris@42:   match (a, b) with
Chris@42:   | (Locative (location_a, _, _, _, _), Locative (location_b, _, _, _, _)) ->
Chris@42:       Unique.same location_a location_b
Chris@42:   | _ -> false
Chris@42: 
Chris@42: let same_class a b = 
Chris@42:   match (a, b) with
Chris@42:   | (Locative (_, class_a, _, _, _), Locative (_, class_b, _, _, _)) ->
Chris@42:       Unique.same class_a class_b
Chris@42:   | (Constant (class_a, _), Constant (class_b, _)) ->
Chris@42:       Unique.same class_a class_b
Chris@42:   | _ -> false
Chris@42: 
Chris@42: let make_temporary =
Chris@42:   let tmp_count = ref 0
Chris@42:   in fun () -> begin
Chris@42:     tmp_count := !tmp_count + 1;
Chris@42:     Temporary !tmp_count
Chris@42:   end
Chris@42: 
Chris@42: let make_constant class_token name = 
Chris@42:   Constant (class_token, name)
Chris@42: 
Chris@42: let make_locative location_token class_token name i vs =
Chris@42:   Locative (location_token, class_token, name, i, vs)
Chris@42: 
Chris@42: let vstride_of_locative = function
Chris@42:   | Locative (_, _, _, _, vs) -> vs
Chris@42:   | _ -> failwith "vstride_of_locative"
Chris@42: 
Chris@42: (* special naming conventions for variables *)
Chris@42: let rec base62_of_int k = 
Chris@42:   let x = k mod 62 
Chris@42:   and y = k / 62 in
Chris@42:   let c = 
Chris@42:     if x < 10 then 
Chris@42:       Char.chr (x + Char.code '0')
Chris@42:     else if x < 36 then
Chris@42:       Char.chr (x + Char.code 'a' - 10)
Chris@42:     else 
Chris@42:       Char.chr (x + Char.code 'A' - 36)
Chris@42:   in
Chris@42:   let s = String.make 1 c in
Chris@42:   let r = if y == 0 then "" else base62_of_int y in
Chris@42:   r ^ s
Chris@42: 
Chris@42: let varname_of_int k =
Chris@42:   if !Magic.compact then
Chris@42:     base62_of_int k
Chris@42:   else
Chris@42:     string_of_int k
Chris@42: 
Chris@42: let unparse = function
Chris@42:   | Temporary k -> "T" ^ (varname_of_int k)
Chris@42:   | Constant (_, name) -> name
Chris@42:   | Locative (_, _, name, i, _) -> name i
Chris@42: 
Chris@42: let unparse_for_alignment m = function
Chris@42:   | Locative (_, _, name, i, _) -> name (i mod m)
Chris@42:   | _ -> failwith "unparse_for_alignment"
Chris@42: