File: args.ml

package info (click to toggle)
herdtools7 7.58-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 19,732 kB
  • sloc: ml: 128,583; ansic: 3,827; makefile: 670; python: 407; sh: 212; awk: 14
file content (61 lines) | stat: -rw-r--r-- 2,293 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
(****************************************************************************)
(*                           the diy toolsuite                              *)
(*                                                                          *)
(* Jade Alglave, University College London, UK.                             *)
(* Luc Maranget, INRIA Paris-Rocquencourt, France.                          *)
(*                                                                          *)
(* Copyright 2013-present Institut National de Recherche en Informatique et *)
(* en Automatique, ARM Ltd and the authors. All rights reserved.            *)
(*                                                                          *)
(* This software is governed by the CeCILL-B license under French law and   *)
(* abiding by the rules of distribution of free software. You can use,      *)
(* modify and/ or redistribute the software under the terms of the CeCILL-B *)
(* license as circulated by CEA, CNRS and INRIA at the following URL        *)
(* "http://www.cecill.info". We also give a copy in LICENSE.txt.            *)
(****************************************************************************)

(** Utilities for using the built-in Arg module. *)

type spec = Arg.key * Arg.spec * Arg.doc

(** Specs. *)

let append_string r =
  Arg.String (fun v -> r := !r @ [v])

let set_string_option r =
  Arg.String (fun v -> r := Some v)

(** Common options *)

let npar j =
  "-j",Arg.Int (fun i -> j := Some (max i 1)),
  "<n> concurrent run with at most <n> instances"

let nohash b =
  "-nohash",Arg.Unit (fun () -> b := true),
  "do not check hashes"

(** Validators. *)

let validate check msg (key, spec, doc) =
  let check_value v =
    if not (check v) then
      raise (Arg.Bad (Printf.sprintf "Invalid %s: %s" key msg))
  in
  let spec =
    match spec with
    | Arg.String f ->
        Arg.String (fun v -> check_value v ; f v)
    | Arg.Set_string r ->
        Arg.String (fun v -> check_value v ; r := v)
    | _ ->
        failwith "Args.validate only accepts Arg.String or Arg.Set_string"
  in
  key, spec, doc

let is_file =
  validate (fun v -> Sys.file_exists v && not (Sys.is_directory v)) "Must be a path to a file"

let is_dir =
  validate Sys.is_directory "Must be a path to a directory"