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
  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
  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
  34 DEFINE PRIME1 = 7
  35 DEFINE PRIME2 = 19
  36 DEFINE PRIME3 = 83
  37 DEFINE PRIME4 = 223
  38 DEFINE PRIME5 = 491
  39 DEFINE PRIME6 = 733
  40 DEFINE PRIME7 = 1009
  41 DEFINE PRIME8 = 4093
  42 DEFINE PRIME9 = 65599 (* Magic Constant used for hashing *)
  43
  44 DEFINE SMALL_H_SIZE  = PRIME2
  45 DEFINE MED_H_SIZE    = PRIME5
  46 DEFINE BIG_H_SIZE    = PRIME8
  47
  48
  49
  50 let read_procmem () =
  51   let pid = Unix.getpid() in
  52   let cin = open_in (Printf.sprintf "/proc/%i/status" pid) in
  53   let pattern = "VmRSS" in
  54   let matchline s = let l = String.length pattern in
  55   if (String.length s) < l then false
  56   else let s' = String.sub s 0 l in
  57   (s' = pattern)
  58   in
  59   let rec loop () =
  60     let s = input_line cin in
  61     if matchline s then s
  62     else loop ()
  63   in
  64   let s = try loop() with _ -> "Could not read mem!"
  65   in
  66   close_in cin;
  67   s
  68 ;;
  69
  70 let l = ref [] ;;
  71 let init_timer() = l := [];;
  72 let time f x =
  73   let s1  = read_procmem() in
  74   let t1 = Unix.gettimeofday () in
  75   let r = f x in
  76   let s2  = read_procmem() in
  77   let t2 = Unix.gettimeofday () in
  78   let t = (1000. *. (t2 -. t1)) in
  79     l:= t::!l;
  80     Printf.eprintf "  %fms\n%!" t ;
  81     Printf.eprintf "Mem use before: %s\n%!" s1;
  82     Printf.eprintf "Mem use after: %s\n\n\n%!" s2;
  83     r
  84 ;;
  85 let total_time () =  List.fold_left (+.) 0. !l;;
  86
  87 END (* IFNDEF UTILS__ML__ *)