src/HOL/Tools/Nunchaku/nunchaku_tool.ML
author blanchet
Fri, 08 Sep 2017 00:02:52 +0200
changeset 66632 6950d3da13f8
parent 66627 4145169ae609
child 66637 809d40cfa4de
permissions -rw-r--r--
rephrased error
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
     1
(*  Title:      HOL/Nunchaku/Tools/nunchaku_tool.ML
66614
1f1c5d85d232 moved Nunchaku to Main; the goal is to move Nitpick out in the next 1-2 years
blanchet
parents: 66163
diff changeset
     2
    Author:     Jasmin Blanchette, VU Amsterdam
1f1c5d85d232 moved Nunchaku to Main; the goal is to move Nitpick out in the next 1-2 years
blanchet
parents: 66163
diff changeset
     3
    Copyright   2015, 2016, 2017
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
     4
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
     5
Interface to the external "nunchaku" tool.
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
     6
*)
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
     7
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
     8
signature NUNCHAKU_TOOL =
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
     9
sig
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    10
  type ty = Nunchaku_Problem.ty
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    11
  type tm = Nunchaku_Problem.tm
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    12
  type nun_problem = Nunchaku_Problem.nun_problem
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    13
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    14
  type tool_params =
66163
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
    15
    {solvers: string list,
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
    16
     overlord: bool,
66627
4145169ae609 extended and renamed Nunchaku's Kodkod bounds
blanchet
parents: 66626
diff changeset
    17
     min_bound: int,
4145169ae609 extended and renamed Nunchaku's Kodkod bounds
blanchet
parents: 66626
diff changeset
    18
     max_bound: int option,
66625
2cd22f070929 added Kodkod-specific options to Nunchaku
blanchet
parents: 66621
diff changeset
    19
     bound_increment: int,
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    20
     debug: bool,
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    21
     specialize: bool,
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    22
     timeout: Time.time}
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    23
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    24
  type nun_solution =
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    25
    {tys: (ty * tm list) list,
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    26
     tms: (tm * tm) list}
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    27
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    28
  datatype nun_outcome =
66620
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
    29
    Unsat of string
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
    30
  | Sat of string * string * nun_solution
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
    31
  | Unknown of (string * string * nun_solution) option
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    32
  | Timeout
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    33
  | Nunchaku_Var_Not_Set
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    34
  | Nunchaku_Cannot_Execute
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    35
  | Nunchaku_Not_Found
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    36
  | CVC4_Cannot_Execute
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    37
  | CVC4_Not_Found
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    38
  | Unknown_Error of int * string
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    39
64469
488d4e627238 added Nunchaku component and tuned Nunchaku integration accordingly
blanchet
parents: 64407
diff changeset
    40
  val nunchaku_home_env_var: string
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    41
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    42
  val solve_nun_problem: tool_params -> nun_problem -> nun_outcome
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    43
end;
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    44
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    45
structure Nunchaku_Tool : NUNCHAKU_TOOL =
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    46
struct
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    47
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    48
open Nunchaku_Util;
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    49
open Nunchaku_Problem;
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    50
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    51
type tool_params =
66163
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
    52
  {solvers: string list,
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
    53
   overlord: bool,
66627
4145169ae609 extended and renamed Nunchaku's Kodkod bounds
blanchet
parents: 66626
diff changeset
    54
   min_bound: int,
4145169ae609 extended and renamed Nunchaku's Kodkod bounds
blanchet
parents: 66626
diff changeset
    55
   max_bound: int option,
66625
2cd22f070929 added Kodkod-specific options to Nunchaku
blanchet
parents: 66621
diff changeset
    56
   bound_increment: int,
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    57
   debug: bool,
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    58
   specialize: bool,
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    59
   timeout: Time.time};
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    60
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    61
type nun_solution =
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    62
  {tys: (ty * tm list) list,
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    63
   tms: (tm * tm) list};
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    64
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    65
datatype nun_outcome =
66620
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
    66
  Unsat of string
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
    67
| Sat of string * string * nun_solution
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
    68
| Unknown of (string * string * nun_solution) option
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    69
| Timeout
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    70
| Nunchaku_Var_Not_Set
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    71
| Nunchaku_Cannot_Execute
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    72
| Nunchaku_Not_Found
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    73
| CVC4_Cannot_Execute
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    74
| CVC4_Not_Found
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    75
| Unknown_Error of int * string;
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    76
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    77
fun bash_output_error s =
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    78
  let val {out, err, rc, ...} = Bash.process s in
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    79
    ((out, err), rc)
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    80
  end;
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    81
64469
488d4e627238 added Nunchaku component and tuned Nunchaku integration accordingly
blanchet
parents: 64407
diff changeset
    82
val nunchaku_home_env_var = "NUNCHAKU_HOME";
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    83
66620
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
    84
val unknown_solver = "unknown";
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
    85
66163
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
    86
val cached_outcome = Synchronized.var "Nunchaku_Tool.cached_outcome"
66626
e3dccf7725a3 repaired Nunchaku cache handing
blanchet
parents: 66625
diff changeset
    87
  (NONE : ((tool_params * nun_problem) * nun_outcome) option);
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    88
66627
4145169ae609 extended and renamed Nunchaku's Kodkod bounds
blanchet
parents: 66626
diff changeset
    89
fun uncached_solve_nun_problem ({solvers, overlord, min_bound, max_bound, bound_increment,
4145169ae609 extended and renamed Nunchaku's Kodkod bounds
blanchet
parents: 66626
diff changeset
    90
      specialize, timeout, ...} : tool_params)
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    91
    (problem as {sound, complete, ...}) =
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    92
  with_tmp_or_overlord_file overlord "nunchaku" "nun" (fn prob_path =>
64469
488d4e627238 added Nunchaku component and tuned Nunchaku integration accordingly
blanchet
parents: 64407
diff changeset
    93
    if getenv nunchaku_home_env_var = "" then
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    94
      Nunchaku_Var_Not_Set
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    95
    else
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    96
      let
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
    97
        val bash_cmd =
64469
488d4e627238 added Nunchaku component and tuned Nunchaku integration accordingly
blanchet
parents: 64407
diff changeset
    98
          "PATH=\"$CVC4_HOME:$KODKODI/bin:$PATH\" \"$" ^
488d4e627238 added Nunchaku component and tuned Nunchaku integration accordingly
blanchet
parents: 64407
diff changeset
    99
          nunchaku_home_env_var ^ "\"/nunchaku --skolems-in-model --no-color " ^
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   100
          (if specialize then "" else "--no-specialize ") ^
66621
1eb8e87f7f8b proper Bash escaping
blanchet
parents: 66620
diff changeset
   101
          "--solvers \"" ^ space_implode "," (map Bash_Syntax.string solvers) ^ "\" " ^
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   102
          "--timeout " ^ string_of_int (Time.toSeconds timeout) ^ " " ^
66627
4145169ae609 extended and renamed Nunchaku's Kodkod bounds
blanchet
parents: 66626
diff changeset
   103
          "--kodkod-min-bound " ^ string_of_int min_bound ^ " " ^
4145169ae609 extended and renamed Nunchaku's Kodkod bounds
blanchet
parents: 66626
diff changeset
   104
          (case max_bound of NONE => "" | SOME n => "--kodkod-max-bound " ^ string_of_int n ^ " ") ^
4145169ae609 extended and renamed Nunchaku's Kodkod bounds
blanchet
parents: 66626
diff changeset
   105
          "--kodkod-bound-increment " ^ string_of_int bound_increment ^ " " ^
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   106
          File.bash_path prob_path;
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   107
        val comments =
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   108
          [bash_cmd, "This file was generated by Isabelle (most likely Nunchaku)", timestamp ()];
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   109
        val prob_str = cat_lines (map (prefix "# ") comments) ^ "\n\n" ^ str_of_nun_problem problem;
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   110
        val _ = File.write prob_path prob_str;
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   111
        val ((output, error), code) = bash_output_error bash_cmd;
66620
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   112
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   113
        val backend =
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   114
          (case filter_out (curry (op =) "") (split_lines output) of
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   115
            [] => unknown_solver
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   116
          | lines =>
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   117
            (case try (unprefix "{backend:") (List.last lines) of
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   118
              NONE => unknown_solver
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   119
            | SOME "" => unknown_solver
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   120
            | SOME s => hd (space_explode "," s)));
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   121
      in
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   122
        if String.isPrefix "SAT" output then
66620
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   123
          (if sound then Sat else Unknown o SOME) (backend, output, {tys = [], tms = []})
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   124
        else if String.isPrefix "UNSAT" output then
66620
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   125
          if complete then Unsat backend else Unknown NONE
64469
488d4e627238 added Nunchaku component and tuned Nunchaku integration accordingly
blanchet
parents: 64407
diff changeset
   126
        else if String.isSubstring "TIMEOUT" output
488d4e627238 added Nunchaku component and tuned Nunchaku integration accordingly
blanchet
parents: 64407
diff changeset
   127
            (* FIXME: temporary *)
488d4e627238 added Nunchaku component and tuned Nunchaku integration accordingly
blanchet
parents: 64407
diff changeset
   128
            orelse String.isSubstring "kodkod failed (errcode 152)" error then
488d4e627238 added Nunchaku component and tuned Nunchaku integration accordingly
blanchet
parents: 64407
diff changeset
   129
          Timeout
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   130
        else if String.isPrefix "UNKNOWN" output then
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   131
          Unknown NONE
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   132
        else if code = 126 then
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   133
          Nunchaku_Cannot_Execute
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   134
        else if code = 127 then
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   135
          Nunchaku_Not_Found
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   136
        else
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   137
          Unknown_Error (code,
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   138
            simplify_spaces (elide_string 1000 (if error <> "" then error else output)))
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   139
      end);
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   140
66163
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   141
fun solve_nun_problem (params as {solvers, overlord, debug, ...}) problem =
66626
e3dccf7725a3 repaired Nunchaku cache handing
blanchet
parents: 66625
diff changeset
   142
  let val key = (params, problem) in
66163
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   143
    (case (overlord orelse debug,
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   144
        AList.lookup (op =) (the_list (Synchronized.value cached_outcome)) key) of
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   145
      (false, SOME outcome) => outcome
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   146
    | _ =>
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   147
      let
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   148
        val outcome = uncached_solve_nun_problem params problem;
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   149
66163
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   150
        fun update_cache () =
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   151
          Synchronized.change cached_outcome (K (SOME (key, outcome)));
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   152
      in
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   153
        (case outcome of
66620
984c179a00d3 more precise output for Nunchaku
blanchet
parents: 66615
diff changeset
   154
          Unsat _ => update_cache ()
66163
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   155
        | Sat _ => update_cache ()
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   156
        | Unknown _ => update_cache ()
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   157
        | _ => ());
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   158
        outcome
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   159
      end)
45d3d43abee7 added 'solvers' option to Nunchaku
blanchet
parents: 64469
diff changeset
   160
  end;
64389
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   161
6273d4c8325b added Nunchaku integration
blanchet
parents:
diff changeset
   162
end;