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