(*  Title:      HOL/Tools/Nitpick/nitpick_isar.ML

    Author:     Jasmin Blanchette, TU Muenchen

    Copyright   2008, 2009, 2010

Adds the "nitpick" and "nitpick_params" commands to Isabelle/Isar's outer

syntax.

*)

signature NITPICK_ISAR =

sig

  type params = Nitpick.params

  val auto: bool Unsynchronized.ref

  val default_params : theory -> (string * string) list -> params

  val setup : theory -> theory

end;

structure Nitpick_Isar : NITPICK_ISAR =

struct

open Nitpick_Util

open Nitpick_HOL

open Nitpick_Rep

open Nitpick_Nut

open Nitpick

val auto = Unsynchronized.ref false;

val _ =

  ProofGeneralPgip.add_preference Preferences.category_tracing

      (Preferences.bool_pref auto "auto-nitpick"

           "Whether to run Nitpick automatically.")

type raw_param = string * string list

val default_default_params =

  [("card", "1\<midarrow>8"),

   ("iter", "0,1,2,4,8,12,16,24"),

   ("bits", "1,2,3,4,6,8,10,12"),

   ("bisim_depth", "7"),

   ("box", "smart"),

   ("finitize", "smart"),

   ("mono", "smart"),

   ("std", "true"),

   ("wf", "smart"),

   ("sat_solver", "smart"),

   ("batch_size", "smart"),

   ("blocking", "true"),

   ("falsify", "true"),

   ("user_axioms", "smart"),

   ("assms", "true"),

   ("merge_type_vars", "false"),

   ("binary_ints", "smart"),

   ("destroy_constrs", "true"),

   ("specialize", "true"),

   ("star_linear_preds", "true"),

   ("fast_descrs", "true"),

   ("peephole_optim", "true"),

   ("timeout", "30 s"),

   ("tac_timeout", "500 ms"),

   ("max_threads", "0"),

   ("debug", "false"),

   ("verbose", "false"),

   ("overlord", "false"),

   ("show_all", "false"),

   ("show_datatypes", "false"),

   ("show_consts", "false"),

   ("format", "1"),

   ("max_potential", "1"),

   ("max_genuine", "1"),

   ("check_potential", "false"),

   ("check_genuine", "false")]

val negated_params =

  [("dont_box", "box"),

   ("dont_finitize", "finitize"),

   ("non_mono", "mono"),

   ("non_std", "std"),

   ("non_wf", "wf"),

   ("non_blocking", "blocking"),

   ("satisfy", "falsify"),

   ("no_user_axioms", "user_axioms"),

   ("no_assms", "assms"),

   ("dont_merge_type_vars", "merge_type_vars"),

   ("unary_ints", "binary_ints"),

   ("dont_destroy_constrs", "destroy_constrs"),

   ("dont_specialize", "specialize"),

   ("dont_star_linear_preds", "star_linear_preds"),

   ("full_descrs", "fast_descrs"),

   ("no_peephole_optim", "peephole_optim"),

   ("no_debug", "debug"),

   ("quiet", "verbose"),

   ("no_overlord", "overlord"),

   ("dont_show_all", "show_all"),

   ("hide_datatypes", "show_datatypes"),

   ("hide_consts", "show_consts"),

   ("trust_potential", "check_potential"),

   ("trust_genuine", "check_genuine")]

fun is_known_raw_param s =

  AList.defined (op =) default_default_params s orelse

  AList.defined (op =) negated_params s orelse

  s = "max" orelse s = "eval" orelse s = "expect" orelse

  exists (fn p => String.isPrefix (p ^ " ") s)

         ["card", "max", "iter", "box", "dont_box", "finitize", "dont_finitize",

          "mono", "non_mono", "std", "non_std", "wf", "non_wf", "format"]

fun check_raw_param (s, _) =

  if is_known_raw_param s then ()

  else error ("Unknown parameter: " ^ quote s ^ ".")  

fun unnegate_param_name name =

  case AList.lookup (op =) negated_params name of

    NONE => if String.isPrefix "dont_" name then SOME (unprefix "dont_" name)

            else if String.isPrefix "non_" name then SOME (unprefix "non_" name)

            else NONE

  | some_name => some_name

fun unnegate_raw_param (name, value) =

  case unnegate_param_name name of

    SOME name' => (name', case value of

                            ["false"] => ["true"]

                          | ["true"] => ["false"]

                          | [] => ["false"]

                          | _ => value)

  | NONE => (name, value)

structure Data = Theory_Data(

  type T = raw_param list

  val empty = map (apsnd single) default_default_params

  val extend = I

  fun merge (x, y) = AList.merge (op =) (K true) (x, y))

val set_default_raw_param = Data.map o AList.update (op =) o unnegate_raw_param

val default_raw_params = Data.get

fun is_punctuation s = (s = "," orelse s = "-" orelse s = "\<midarrow>")

fun stringify_raw_param_value [] = ""

  | stringify_raw_param_value [s] = s

  | stringify_raw_param_value (s1 :: s2 :: ss) =

    s1 ^ (if is_punctuation s1 orelse is_punctuation s2 then "" else " ") ^

    stringify_raw_param_value (s2 :: ss)

fun maxed_int_from_string min_int s = Int.max (min_int, the (Int.fromString s))

fun extract_params ctxt auto default_params override_params =

  let

    val override_params = map unnegate_raw_param override_params

    val raw_params = rev override_params @ rev default_params

    val lookup =

      Option.map stringify_raw_param_value o AList.lookup (op =) raw_params

    val lookup_string = the_default "" o lookup

    fun general_lookup_bool option default_value name =

      case lookup name of

        SOME s => parse_bool_option option name s

      | NONE => default_value

    val lookup_bool = the o general_lookup_bool false (SOME false)

    val lookup_bool_option = general_lookup_bool true NONE

    fun do_int name value =

      case value of

        SOME s => (case Int.fromString s of

                     SOME i => i

                   | NONE => error ("Parameter " ^ quote name ^

                                    " must be assigned an integer value."))

      | NONE => 0

    fun lookup_int name = do_int name (lookup name)

    fun lookup_int_option name =

      case lookup name of

        SOME "smart" => NONE

      | value => SOME (do_int name value)

    fun int_range_from_string name min_int s =

      let

        val (k1, k2) =

          (case space_explode "-" s of

             [s] => the_default (s, s) (first_field "\<midarrow>" s)

           | ["", s2] => ("-" ^ s2, "-" ^ s2)

           | [s1, s2] => (s1, s2)

           | _ => raise Option)

          |> pairself (maxed_int_from_string min_int)

      in if k1 <= k2 then k1 upto k2 else k1 downto k2 end

      handle Option.Option =>

             error ("Parameter " ^ quote name ^

                    " must be assigned a sequence of integers.")

    fun int_seq_from_string name min_int s =

      maps (int_range_from_string name min_int) (space_explode "," s)

    fun lookup_int_seq name min_int =

      case lookup name of

        SOME s => (case int_seq_from_string name min_int s of

                     [] => [min_int]

                   | value => value)

      | NONE => [min_int]

    fun lookup_ints_assigns read prefix min_int =

      (NONE, lookup_int_seq prefix min_int)

      :: map (fn (name, value) =>

                 (SOME (read (String.extract (name, size prefix + 1, NONE))),

                  value |> stringify_raw_param_value

                        |> int_seq_from_string name min_int))

             (filter (String.isPrefix (prefix ^ " ") o fst) raw_params)

    fun lookup_bool_assigns read prefix =

      (NONE, lookup_bool prefix)

      :: map (fn (name, value) =>

                 (SOME (read (String.extract (name, size prefix + 1, NONE))),

                  value |> stringify_raw_param_value

                        |> parse_bool_option false name |> the))

             (filter (String.isPrefix (prefix ^ " ") o fst) raw_params)

    fun lookup_bool_option_assigns read prefix =

      (NONE, lookup_bool_option prefix)

      :: map (fn (name, value) =>

                 (SOME (read (String.extract (name, size prefix + 1, NONE))),

                  value |> stringify_raw_param_value

                        |> parse_bool_option true name))

             (filter (String.isPrefix (prefix ^ " ") o fst) raw_params)

    fun lookup_time name =

      case lookup name of

        NONE => NONE

      | SOME s => parse_time_option name s

    val lookup_term_list =

      AList.lookup (op =) raw_params #> these #> Syntax.read_terms ctxt

    val read_type_polymorphic =

      Syntax.read_typ ctxt #> Logic.mk_type

      #> singleton (Variable.polymorphic ctxt) #> Logic.dest_type

    val read_term_polymorphic =

      Syntax.read_term ctxt #> singleton (Variable.polymorphic ctxt)

    val read_const_polymorphic = read_term_polymorphic #> dest_Const

    val cards_assigns = lookup_ints_assigns read_type_polymorphic "card" 1

    val maxes_assigns = lookup_ints_assigns read_const_polymorphic "max" ~1

    val iters_assigns = lookup_ints_assigns read_const_polymorphic "iter" 0

    val bitss = lookup_int_seq "bits" 1

    val bisim_depths = lookup_int_seq "bisim_depth" ~1

    val boxes = lookup_bool_option_assigns read_type_polymorphic "box"

    val finitizes = lookup_bool_option_assigns read_type_polymorphic "finitize"

    val monos = lookup_bool_option_assigns read_type_polymorphic "mono"

    val stds = lookup_bool_assigns read_type_polymorphic "std"

    val wfs = lookup_bool_option_assigns read_const_polymorphic "wf"

    val sat_solver = lookup_string "sat_solver"

    val blocking = not auto andalso lookup_bool "blocking"

    val falsify = lookup_bool "falsify"

    val debug = not auto andalso lookup_bool "debug"

    val verbose = debug orelse (not auto andalso lookup_bool "verbose")

    val overlord = lookup_bool "overlord"

    val user_axioms = lookup_bool_option "user_axioms"

    val assms = lookup_bool "assms"

    val merge_type_vars = lookup_bool "merge_type_vars"

    val binary_ints = lookup_bool_option "binary_ints"

    val destroy_constrs = lookup_bool "destroy_constrs"

    val specialize = lookup_bool "specialize"

    val star_linear_preds = lookup_bool "star_linear_preds"

    val fast_descrs = lookup_bool "fast_descrs"

    val peephole_optim = lookup_bool "peephole_optim"

    val timeout = if auto then NONE else lookup_time "timeout"

    val tac_timeout = lookup_time "tac_timeout"

    val max_threads = Int.max (0, lookup_int "max_threads")

    val show_all = debug orelse lookup_bool "show_all"

    val show_datatypes = show_all orelse lookup_bool "show_datatypes"

    val show_consts = show_all orelse lookup_bool "show_consts"

    val formats = lookup_ints_assigns read_term_polymorphic "format" 0

    val evals = lookup_term_list "eval"

    val max_potential =

      if auto then 0 else Int.max (0, lookup_int "max_potential")

    val max_genuine = Int.max (0, lookup_int "max_genuine")

    val check_potential = lookup_bool "check_potential"

    val check_genuine = lookup_bool "check_genuine"

    val batch_size =

      case lookup_int_option "batch_size" of

        SOME n => Int.max (1, n)

      | NONE => if debug then 1 else 64

    val expect = lookup_string "expect"

  in

    {cards_assigns = cards_assigns, maxes_assigns = maxes_assigns,

     iters_assigns = iters_assigns, bitss = bitss, bisim_depths = bisim_depths,

     boxes = boxes, finitizes = finitizes, monos = monos, stds = stds,

     wfs = wfs, sat_solver = sat_solver, blocking = blocking, falsify = falsify,

     debug = debug, verbose = verbose, overlord = overlord,

     user_axioms = user_axioms, assms = assms,

     merge_type_vars = merge_type_vars, binary_ints = binary_ints,

     destroy_constrs = destroy_constrs, specialize = specialize,

     star_linear_preds = star_linear_preds, fast_descrs = fast_descrs,

     peephole_optim = peephole_optim, timeout = timeout,

     tac_timeout = tac_timeout, max_threads = max_threads,

     show_datatypes = show_datatypes, show_consts = show_consts,

     formats = formats, evals = evals, max_potential = max_potential,

     max_genuine = max_genuine, check_potential = check_potential,

     check_genuine = check_genuine, batch_size = batch_size, expect = expect}

  end

fun default_params thy =

  extract_params (ProofContext.init_global thy) false (default_raw_params thy)

  o map (apsnd single)

val parse_key = Scan.repeat1 Parse.typ_group >> space_implode " "

val parse_value =

  Scan.repeat1 (Parse.minus >> single

                || Scan.repeat1 (Scan.unless Parse.minus Parse.name)

                || Parse.$$$ "," |-- Parse.number >> prefix "," >> single) >> flat

val parse_param = parse_key -- Scan.optional (Parse.$$$ "=" |-- parse_value) []

val parse_params =

  Scan.optional (Parse.$$$ "[" |-- Parse.list parse_param --| Parse.$$$ "]") []

fun handle_exceptions ctxt f x =

  f x

  handle ARG (loc, details) =>

         error ("Bad argument(s) to " ^ quote loc ^ ": " ^ details ^ ".")

       | BAD (loc, details) =>

         error ("Internal error (" ^ quote loc ^ "): " ^ details ^ ".")

       | NOT_SUPPORTED details =>

         (warning ("Unsupported case: " ^ details ^ "."); x)

       | NUT (loc, us) =>

         error ("Invalid intermediate term" ^ plural_s_for_list us ^

                " (" ^ quote loc ^ "): " ^

                commas (map (string_for_nut ctxt) us) ^ ".")

       | REP (loc, Rs) =>

         error ("Invalid representation" ^ plural_s_for_list Rs ^

                " (" ^ quote loc ^ "): " ^ commas (map string_for_rep Rs) ^ ".")

       | TERM (loc, ts) =>

         error ("Invalid term" ^ plural_s_for_list ts ^

                " (" ^ quote loc ^ "): " ^

                commas (map (Syntax.string_of_term ctxt) ts) ^ ".")

       | TOO_LARGE (_, details) =>

         (warning ("Limit reached: " ^ details ^ "."); x)

       | TOO_SMALL (_, details) =>

         (warning ("Limit reached: " ^ details ^ "."); x)

       | TYPE (loc, Ts, ts) =>

         error ("Invalid type" ^ plural_s_for_list Ts ^

                (if null ts then

                   ""

                 else

                   " for term" ^ plural_s_for_list ts ^ " " ^

                   commas (map (quote o Syntax.string_of_term ctxt) ts)) ^

                " (" ^ quote loc ^ "): " ^

                commas (map (Syntax.string_of_typ ctxt) Ts) ^ ".")

       | Kodkod.SYNTAX (_, details) =>

         (warning ("Ill-formed Kodkodi output: " ^ details ^ "."); x)

       | Refute.REFUTE (loc, details) =>

         error ("Unhandled Refute error (" ^ quote loc ^ "): " ^ details ^ ".")

fun pick_nits override_params auto i step state =

  let

    val thy = Proof.theory_of state

    val ctxt = Proof.context_of state

    val _ = List.app check_raw_param override_params

    val params as {blocking, debug, ...} =

      extract_params ctxt auto (default_raw_params thy) override_params

    fun go () =

      (false, state)

      |> (if auto then perhaps o try

          else if debug then fn f => fn x => f x

          else handle_exceptions ctxt)

         (fn (_, state) => pick_nits_in_subgoal state params auto i step

                           |>> curry (op =) "genuine")

  in

    if auto orelse blocking then go ()

    else (Toplevel.thread true (fn () => (go (); ())); (false, state))

  end

fun nitpick_trans (params, i) =

  Toplevel.keep (fn st =>

      (pick_nits params false i (Toplevel.proof_position_of st)

                 (Toplevel.proof_of st); ()))

fun string_for_raw_param (name, value) =

  name ^ " = " ^ stringify_raw_param_value value

fun nitpick_params_trans params =

  Toplevel.theory

      (fold set_default_raw_param params

       #> tap (fn thy => 

                  writeln ("Default parameters for Nitpick:\n" ^

                           (case rev (default_raw_params thy) of

                              [] => "none"

                            | params =>

                              (map check_raw_param params;

                               params |> map string_for_raw_param

                                      |> sort_strings |> cat_lines)))))

val parse_nitpick_command =

  (parse_params -- Scan.optional Parse.nat 1) #>> nitpick_trans

val parse_nitpick_params_command = parse_params #>> nitpick_params_trans

val _ = Outer_Syntax.improper_command "nitpick"

            "try to find a counterexample for a given subgoal using Nitpick"

            Keyword.diag parse_nitpick_command

val _ = Outer_Syntax.command "nitpick_params"

            "set and display the default parameters for Nitpick"

            Keyword.thy_decl parse_nitpick_params_command

fun auto_nitpick state =

  if not (!auto) then (false, state) else pick_nits [] true 1 0 state

val setup = Auto_Counterexample.register_tool ("nitpick", auto_nitpick)

end;