dostarbucks.ml

(* compute starbucks *)
   
open Common
open Getopt

let prefix' = ref "sbucks"
let outdir' = ref (Some !prefix')
let mark'   = ref ""
let specs =
[
  (noshort,"prefix",None,Some (fun x -> prefix' := x));
  (noshort,"outdir",None,Some (fun x -> outdir' := Some x));
  (noshort,"nodir", (set outdir' None), None);
  ('k',"mark",None,Some (fun x -> mark' := x))
]

let () =
  let args = getOptArgs specs in
  
  let mark = !mark' in

  let prefix, outdir =
      !prefix', !outdir' in
  
  let bucksName,starsName,outdir =
  match args with
    | bucksName::starsName::outdir::restArgs -> bucksName,starsName,Some outdir
    | bucksName::starsName::restArgs         -> bucksName,starsName,outdir
    | _ -> failwith "usage: dostarbucks bucksName starsName [outdir]"      
  in  

  let baseName = cutPathZ starsName in
  let sbucksName = sprintf "%s-%s%s" prefix mark baseName |> mayPrependDir outdir in
  leprintfln "reading bucks from %s, starrank from %s, storing starbucks in %s" bucksName starsName sbucksName;

  let bucks: day_buckets = loadData bucksName in
  let stars: starrank    = loadData starsName in
  
  let srankh                = Starrank.starrank_hash stars in
  let sbucks: day_starbucks = Starrank.starbucks bucks srankh in
  
  mayMkDir outdir;
  saveData sbucks sbucksName