utils.ml

   1 IFNDEF UTILS_ML__
   2 THEN
   3 DEFINE UTILS__ML__
   4
   5
   6 IFDEF WORDSIZE64
   7 THEN
   8     DEFINE WORDSIZE = 64
   9     DEFINE HALFWORDSIZE = 32
  10     DEFINE INTSIZE = 63
  11     DEFINE HALFINTSIZE = 31
  12     DEFINE HALF_MAX_INT = 2305843009213693951
  13     DEFINE HPARAM = 65599
  14     DEFINE HPARAM2 = 4303228801
  15     DEFINE HPARAM3 = 282287506116799
  16     DEFINE HPARAM4 = 71034040046345985
  17 ELSE
  18     DEFINE WORDSIZE = 32
  19     DEFINE HALFWORDSIZE = 16
  20     DEFINE INTSIZE = 31
  21     DEFINE HALFINTSIZE = 15
  22     DEFINE HALF_MAX_INT = 536870911
  23     DEFINE HPARAM = 65599
  24     DEFINE HPARAM2 = 8261505
  25     DEFINE HPARAM3 = 780587199
  26     DEFINE HPARAM4 = 549173308
  27 END
  28
  29
  30 DEFINE HASHINT2 (x,y) = ((x)+HPARAM*(y))
  31 DEFINE HASHINT3 (x,y,z) = ((x)  + (y) * HPARAM + (z) * HPARAM2)
  32 DEFINE HASHINT4 (x,y,z,t) =  ((x)  + (y) * HPARAM + (z)*HPARAM2 + (t)* HPARAM3)
  33 DEFINE HASHINT5 (x,y,z,t,u) =  ((x)  + (y) * HPARAM + (z)*HPARAM2 + (t)* HPARAM3 + (u)*HPARAM4)
  34
  35 DEFINE PRIME1 = 7
  36 DEFINE PRIME2 = 19
  37 DEFINE PRIME3 = 83
  38 DEFINE PRIME4 = 223
  39 DEFINE PRIME5 = 491
  40 DEFINE PRIME6 = 733
  41 DEFINE PRIME7 = 1009
  42 DEFINE PRIME8 = 4093
  43 DEFINE PRIME9 = 65599 (* Magic Constant used for hashing *)
  44
  45 DEFINE SMALL_H_SIZE  = PRIME2
  46 DEFINE MED_H_SIZE    = PRIME5
  47 DEFINE BIG_H_SIZE    = PRIME8
  48
  49
  50
  51 let read_procmem () =
  52   let pid = Unix.getpid() in
  53   let cin = open_in (Printf.sprintf "/proc/%i/status" pid) in
  54   let pattern = "VmRSS" in
  55   let matchline s = let l = String.length pattern in
  56   if (String.length s) < l then false
  57   else let s' = String.sub s 0 l in
  58   (s' = pattern)
  59   in
  60   let rec loop () =
  61     let s = input_line cin in
  62     if matchline s then s
  63     else loop ()
  64   in
  65   let s = try loop() with _ -> "Could not read mem!"
  66   in
  67   close_in cin;
  68   s
  69 ;;
  70
  71 let l = ref [] ;;
  72 let init_timer() = l := [];;
  73 let time_mem f x =
  74   let s1  = read_procmem() in
  75   let t1 = Unix.gettimeofday () in
  76   let r = f x in
  77   let s2  = read_procmem() in
  78   let t2 = Unix.gettimeofday () in
  79   let t = (1000. *. (t2 -. t1)) in
  80     l:= t::!l;
  81     Printf.eprintf "  %fms\n%!" t ;
  82     Printf.eprintf "Mem use before: %s\n%!" s1;
  83     Printf.eprintf "Final Mem: %s\n\n\n%!" s2;
  84     r
  85 ;;
  86 let time f x =
  87   let t1 = Unix.gettimeofday () in
  88   let r = f x in
  89   let t2 = Unix.gettimeofday () in
  90   let t = (1000. *. (t2 -. t1)) in
  91     l:= t::!l;
  92     Printf.eprintf "  %fms\n%!" t ;
  93     r
  94 ;;
  95 let total_time () =  List.fold_left (+.) 0. !l;;
  96
  97 END (* IFNDEF UTILS__ML__ *)