src/HOL/Tools/Quickcheck/random_generators.ML
author huffman
Mon Feb 20 12:37:17 2012 +0100 (2012-02-20)
changeset 46547 d1dcb91a512e
parent 46331 f5598b604a54
child 50046 0051dc4f301f
permissions -rw-r--r--
use qualified constant names instead of suffixes (from Florian Haftmann)
bulwahn@41923
     1
(*  Title:      HOL/Tools/Quickcheck/random_generators.ML
haftmann@37744
     2
    Author:     Florian Haftmann, TU Muenchen
haftmann@31260
     3
bulwahn@41923
     4
Random generators for various types.
haftmann@31260
     5
*)
haftmann@31260
     6
bulwahn@41923
     7
signature RANDOM_GENERATORS =
haftmann@31260
     8
sig
haftmann@31260
     9
  type seed = Random_Engine.seed
haftmann@31260
    10
  val random_fun: typ -> typ -> ('a -> 'a -> bool) -> ('a -> term)
haftmann@31260
    11
    -> (seed -> ('b * (unit -> term)) * seed) -> (seed -> seed * seed)
wenzelm@33243
    12
    -> seed -> (('a -> 'b) * (unit -> term)) * seed
bulwahn@35378
    13
  val compile_generator_expr:
bulwahn@45754
    14
    Proof.context -> (term * term list) list -> bool -> int list -> (bool * term list) option * Quickcheck.report option
bulwahn@45754
    15
  val put_counterexample: (unit -> int -> bool -> int -> seed -> (bool * term list) option * seed)
haftmann@39388
    16
    -> Proof.context -> Proof.context
bulwahn@45763
    17
  val put_counterexample_report: (unit -> int -> bool -> int -> seed -> ((bool * term list) option * (bool list * bool)) * seed)
haftmann@39388
    18
    -> Proof.context -> Proof.context
bulwahn@45940
    19
  val instantiate_random_datatype : Datatype_Aux.config -> Datatype_Aux.descr ->
bulwahn@45940
    20
    (string * sort) list -> string list -> string -> string list * string list -> typ list * typ list -> theory -> theory
haftmann@31260
    21
  val setup: theory -> theory
haftmann@31260
    22
end;
haftmann@31260
    23
bulwahn@41923
    24
structure Random_Generators : RANDOM_GENERATORS =
haftmann@31260
    25
struct
haftmann@31260
    26
haftmann@31950
    27
(** abstract syntax **)
haftmann@31260
    28
haftmann@31950
    29
fun termifyT T = HOLogic.mk_prodT (T, @{typ "unit => term"})
haftmann@31950
    30
val size = @{term "i::code_numeral"};
haftmann@31984
    31
val size_pred = @{term "(i::code_numeral) - 1"};
haftmann@31950
    32
val size' = @{term "j::code_numeral"};
haftmann@31950
    33
val seed = @{term "s::Random.seed"};
haftmann@31260
    34
bulwahn@45723
    35
val resultT =  @{typ "(bool * term list) option"};
bulwahn@45721
    36
haftmann@31260
    37
(** typ "'a => 'b" **)
haftmann@31260
    38
haftmann@31260
    39
type seed = Random_Engine.seed;
haftmann@31260
    40
haftmann@31603
    41
fun random_fun T1 T2 eq term_of random random_split seed =
haftmann@31260
    42
  let
haftmann@31260
    43
    val fun_upd = Const (@{const_name fun_upd},
haftmann@31260
    44
      (T1 --> T2) --> T1 --> T2 --> T1 --> T2);
haftmann@32344
    45
    val ((y, t2), seed') = random seed;
haftmann@32344
    46
    val (seed'', seed''') = random_split seed';
haftmann@31933
    47
wenzelm@32740
    48
    val state = Unsynchronized.ref (seed'', [], fn () => Abs ("x", T1, t2 ()));
haftmann@31260
    49
    fun random_fun' x =
haftmann@31260
    50
      let
haftmann@31260
    51
        val (seed, fun_map, f_t) = ! state;
haftmann@31260
    52
      in case AList.lookup (uncurry eq) fun_map x
haftmann@31260
    53
       of SOME y => y
haftmann@31260
    54
        | NONE => let
haftmann@31260
    55
              val t1 = term_of x;
haftmann@31260
    56
              val ((y, t2), seed') = random seed;
haftmann@31260
    57
              val fun_map' = (x, y) :: fun_map;
haftmann@31933
    58
              val f_t' = fn () => fun_upd $ f_t () $ t1 $ t2 ();
haftmann@31260
    59
              val _ = state := (seed', fun_map', f_t');
haftmann@31260
    60
            in y end
haftmann@31260
    61
      end;
haftmann@31933
    62
    fun term_fun' () = #3 (! state) ();
haftmann@32344
    63
  in ((random_fun', term_fun'), seed''') end;
haftmann@31260
    64
bulwahn@45762
    65
fun mk_if (b, t, e) =
bulwahn@45762
    66
  let
bulwahn@45762
    67
    val T = fastype_of t
bulwahn@45762
    68
  in Const (@{const_name "HOL.If"}, @{typ bool} --> T --> T --> T) $ b $ t $ e end
bulwahn@45762
    69
  
haftmann@31260
    70
(** datatypes **)
haftmann@31260
    71
haftmann@31485
    72
(* definitional scheme for random instances on datatypes *)
haftmann@31485
    73
haftmann@31611
    74
local
haftmann@31611
    75
haftmann@31485
    76
fun dest_ctyp_nth k cT = nth (Thm.dest_ctyp cT) k;
haftmann@31611
    77
val eq = Thm.cprop_of @{thm random_aux_rec} |> Thm.dest_arg |> Thm.dest_arg |> Thm.dest_arg;
haftmann@31611
    78
val lhs = eq |> Thm.dest_arg1;
haftmann@31611
    79
val pt_random_aux = lhs |> Thm.dest_fun;
haftmann@31611
    80
val ct_k = lhs |> Thm.dest_arg;
haftmann@31611
    81
val pt_rhs = eq |> Thm.dest_arg |> Thm.dest_fun;
haftmann@31611
    82
val aT = pt_random_aux |> Thm.ctyp_of_term |> dest_ctyp_nth 1;
haftmann@31611
    83
haftmann@31611
    84
val rew_thms = map mk_meta_eq [@{thm code_numeral_zero_minus_one},
haftmann@31611
    85
  @{thm Suc_code_numeral_minus_one}, @{thm select_weight_cons_zero}, @{thm beyond_zero}];
haftmann@31611
    86
val rew_ts = map (Logic.dest_equals o Thm.prop_of) rew_thms;
haftmann@31611
    87
val rew_ss = HOL_ss addsimps rew_thms;
haftmann@31611
    88
haftmann@31611
    89
in
haftmann@31485
    90
haftmann@31485
    91
fun random_aux_primrec eq lthy =
haftmann@31485
    92
  let
wenzelm@42361
    93
    val thy = Proof_Context.theory_of lthy;
haftmann@31611
    94
    val ((t_random_aux as Free (random_aux, T)) $ (t_k as Free (v, _)), proto_t_rhs) =
haftmann@31611
    95
      (HOLogic.dest_eq o HOLogic.dest_Trueprop) eq;
haftmann@31485
    96
    val Type (_, [_, iT]) = T;
haftmann@31485
    97
    val icT = Thm.ctyp_of thy iT;
haftmann@31611
    98
    val cert = Thm.cterm_of thy;
haftmann@31611
    99
    val inst = Thm.instantiate_cterm ([(aT, icT)], []);
haftmann@31485
   100
    fun subst_v t' = map_aterms (fn t as Free (w, _) => if v = w then t' else t | t => t);
haftmann@31611
   101
    val t_rhs = lambda t_k proto_t_rhs;
haftmann@31785
   102
    val eqs0 = [subst_v @{term "0::code_numeral"} eq,
huffman@46547
   103
      subst_v (@{const Code_Numeral.Suc} $ t_k) eq];
haftmann@31611
   104
    val eqs1 = map (Pattern.rewrite_term thy rew_ts []) eqs0;
haftmann@35166
   105
    val ((_, (_, eqs2)), lthy') = Primrec.add_primrec_simple
haftmann@33205
   106
      [((Binding.conceal (Binding.name random_aux), T), NoSyn)] eqs1 lthy;
haftmann@31611
   107
    val cT_random_aux = inst pt_random_aux;
haftmann@31611
   108
    val cT_rhs = inst pt_rhs;
haftmann@31485
   109
    val rule = @{thm random_aux_rec}
wenzelm@43333
   110
      |> Drule.instantiate_normalize ([(aT, icT)],
haftmann@31785
   111
           [(cT_random_aux, cert t_random_aux), (cT_rhs, cert t_rhs)]);
haftmann@31785
   112
    val tac = ALLGOALS (rtac rule)
haftmann@31785
   113
      THEN ALLGOALS (simp_tac rew_ss)
wenzelm@42361
   114
      THEN (ALLGOALS (Proof_Context.fact_tac eqs2))
wenzelm@32970
   115
    val simp = Skip_Proof.prove lthy' [v] [] eq (K tac);
haftmann@31485
   116
  in (simp, lthy') end;
haftmann@31485
   117
haftmann@31611
   118
end;
haftmann@31611
   119
haftmann@31868
   120
fun random_aux_primrec_multi auxname [eq] lthy =
haftmann@31485
   121
      lthy
haftmann@31485
   122
      |> random_aux_primrec eq
haftmann@31485
   123
      |>> (fn simp => [simp])
haftmann@31868
   124
  | random_aux_primrec_multi auxname (eqs as _ :: _ :: _) lthy =
haftmann@31485
   125
      let
wenzelm@42361
   126
        val thy = Proof_Context.theory_of lthy;
haftmann@31485
   127
        val (lhss, rhss) = map_split (HOLogic.dest_eq o HOLogic.dest_Trueprop) eqs;
haftmann@31485
   128
        val (vs, (arg as Free (v, _)) :: _) = map_split (fn (t1 $ t2) => (t1, t2)) lhss;
haftmann@31485
   129
        val Ts = map fastype_of lhss;
haftmann@31485
   130
        val tupleT = foldr1 HOLogic.mk_prodT Ts;
haftmann@31868
   131
        val aux_lhs = Free ("mutual_" ^ auxname, fastype_of arg --> tupleT) $ arg;
haftmann@31485
   132
        val aux_eq = (HOLogic.mk_Trueprop o HOLogic.mk_eq)
haftmann@31485
   133
          (aux_lhs, foldr1 HOLogic.mk_prod rhss);
haftmann@31485
   134
        fun mk_proj t [T] = [t]
haftmann@31485
   135
          | mk_proj t (Ts as T :: (Ts' as _ :: _)) =
haftmann@31485
   136
              Const (@{const_name fst}, foldr1 HOLogic.mk_prodT Ts --> T) $ t
haftmann@31485
   137
                :: mk_proj (Const (@{const_name snd},
haftmann@31485
   138
                  foldr1 HOLogic.mk_prodT Ts --> foldr1 HOLogic.mk_prodT Ts') $ t) Ts';
haftmann@31485
   139
        val projs = mk_proj (aux_lhs) Ts;
haftmann@31485
   140
        val proj_eqs = map2 (fn v => fn proj => (v, lambda arg proj)) vs projs;
haftmann@31485
   141
        val proj_defs = map2 (fn Free (name, _) => fn (_, rhs) =>
haftmann@33205
   142
          ((Binding.conceal (Binding.name name), NoSyn),
wenzelm@33280
   143
            (apfst Binding.conceal Attrib.empty_binding, rhs))) vs proj_eqs;
haftmann@31485
   144
        val aux_eq' = Pattern.rewrite_term thy proj_eqs [] aux_eq;
haftmann@31485
   145
        fun prove_eqs aux_simp proj_defs lthy = 
haftmann@31485
   146
          let
haftmann@31485
   147
            val proj_simps = map (snd o snd) proj_defs;
wenzelm@31645
   148
            fun tac { context = ctxt, prems = _ } =
haftmann@31625
   149
              ALLGOALS (simp_tac (HOL_ss addsimps proj_simps))
haftmann@31485
   150
              THEN ALLGOALS (EqSubst.eqsubst_tac ctxt [0] [aux_simp])
haftmann@37136
   151
              THEN ALLGOALS (simp_tac (HOL_ss addsimps [@{thm fst_conv}, @{thm snd_conv}]));
wenzelm@32970
   152
          in (map (fn prop => Skip_Proof.prove lthy [v] [] prop tac) eqs, lthy) end;
haftmann@31485
   153
      in
haftmann@31485
   154
        lthy
haftmann@31485
   155
        |> random_aux_primrec aux_eq'
wenzelm@33766
   156
        ||>> fold_map Local_Theory.define proj_defs
haftmann@31485
   157
        |-> (fn (aux_simp, proj_defs) => prove_eqs aux_simp proj_defs)
haftmann@31485
   158
      end;
haftmann@31485
   159
haftmann@31868
   160
fun random_aux_specification prfx name eqs lthy =
haftmann@31485
   161
  let
haftmann@31595
   162
    val vs = fold Term.add_free_names ((snd o strip_comb o fst o HOLogic.dest_eq
haftmann@31595
   163
      o HOLogic.dest_Trueprop o hd) eqs) [];
haftmann@31485
   164
    fun mk_proto_eq eq =
haftmann@31485
   165
      let
haftmann@31595
   166
        val (head $ t $ u, rhs) = (HOLogic.dest_eq o HOLogic.dest_Trueprop) eq;
haftmann@31595
   167
      in ((HOLogic.mk_Trueprop o HOLogic.mk_eq) (head, lambda t (lambda u rhs))) end;
haftmann@31485
   168
    val proto_eqs = map mk_proto_eq eqs;
haftmann@31485
   169
    fun prove_simps proto_simps lthy =
haftmann@31485
   170
      let
haftmann@31625
   171
        val ext_simps = map (fn thm => fun_cong OF [fun_cong OF [thm]]) proto_simps;
wenzelm@42361
   172
        val tac = ALLGOALS (Proof_Context.fact_tac ext_simps);
wenzelm@32970
   173
      in (map (fn prop => Skip_Proof.prove lthy vs [] prop (K tac)) eqs, lthy) end;
haftmann@33205
   174
    val b = Binding.conceal (Binding.qualify true prfx
haftmann@33205
   175
      (Binding.qualify true name (Binding.name "simps")));
haftmann@31485
   176
  in
haftmann@31485
   177
    lthy
haftmann@31868
   178
    |> random_aux_primrec_multi (name ^ prfx) proto_eqs
haftmann@31485
   179
    |-> (fn proto_simps => prove_simps proto_simps)
wenzelm@33671
   180
    |-> (fn simps => Local_Theory.note
wenzelm@45592
   181
      ((b, Code.add_default_eqn_attrib :: @{attributes [simp, nitpick_simp]}), simps))
haftmann@31485
   182
    |> snd
haftmann@31485
   183
  end
haftmann@31485
   184
haftmann@31485
   185
haftmann@31485
   186
(* constructing random instances on datatypes *)
haftmann@31485
   187
haftmann@31868
   188
val random_auxN = "random_aux";
haftmann@31868
   189
haftmann@31603
   190
fun mk_random_aux_eqs thy descr vs tycos (names, auxnames) (Ts, Us) =
haftmann@31595
   191
  let
haftmann@31595
   192
    val mk_const = curry (Sign.mk_const thy);
haftmann@31603
   193
    val random_auxsN = map (prefix (random_auxN ^ "_")) (names @ auxnames);
haftmann@31603
   194
    val rTs = Ts @ Us;
haftmann@31595
   195
    fun random_resultT T = @{typ Random.seed}
haftmann@31595
   196
      --> HOLogic.mk_prodT (termifyT T,@{typ Random.seed});
haftmann@31595
   197
    val pTs = map random_resultT rTs;
haftmann@31595
   198
    fun sizeT T = @{typ code_numeral} --> @{typ code_numeral} --> T;
haftmann@31595
   199
    val random_auxT = sizeT o random_resultT;
haftmann@31595
   200
    val random_auxs = map2 (fn s => fn rT => Free (s, random_auxT rT))
haftmann@31595
   201
      random_auxsN rTs;
haftmann@31950
   202
    fun mk_random_call T = (NONE, (HOLogic.mk_random T size', T));
haftmann@31603
   203
    fun mk_random_aux_call fTs (k, _) (tyco, Ts) =
haftmann@31595
   204
      let
haftmann@31623
   205
        val T = Type (tyco, Ts);
haftmann@31623
   206
        fun mk_random_fun_lift [] t = t
haftmann@31623
   207
          | mk_random_fun_lift (fT :: fTs) t =
haftmann@31623
   208
              mk_const @{const_name random_fun_lift} [fTs ---> T, fT] $
haftmann@31623
   209
                mk_random_fun_lift fTs t;
haftmann@31984
   210
        val t = mk_random_fun_lift fTs (nth random_auxs k $ size_pred $ size');
haftmann@33968
   211
        val size = Option.map snd (Datatype_Aux.find_shortest_path descr k)
haftmann@31595
   212
          |> the_default 0;
haftmann@31623
   213
      in (SOME size, (t, fTs ---> T)) end;
haftmann@33968
   214
    val tss = Datatype_Aux.interpret_construction descr vs
haftmann@31603
   215
      { atyp = mk_random_call, dtyp = mk_random_aux_call };
haftmann@31595
   216
    fun mk_consexpr simpleT (c, xs) =
haftmann@31595
   217
      let
haftmann@31595
   218
        val (ks, simple_tTs) = split_list xs;
haftmann@31595
   219
        val T = termifyT simpleT;
haftmann@31595
   220
        val tTs = (map o apsnd) termifyT simple_tTs;
haftmann@31595
   221
        val is_rec = exists is_some ks;
wenzelm@33029
   222
        val k = fold (fn NONE => I | SOME k => Integer.max k) ks 0;
wenzelm@43329
   223
        val vs = Name.invent_names Name.context "x" (map snd simple_tTs);
haftmann@31595
   224
        val tc = HOLogic.mk_return T @{typ Random.seed}
haftmann@31595
   225
          (HOLogic.mk_valtermify_app c vs simpleT);
haftmann@38543
   226
        val t = HOLogic.mk_ST
haftmann@38543
   227
          (map2 (fn (t, _) => fn (v, T') => ((t, @{typ Random.seed}), SOME ((v, termifyT T')))) tTs vs)
haftmann@38543
   228
            tc @{typ Random.seed} (SOME T, @{typ Random.seed});
haftmann@31595
   229
        val tk = if is_rec
haftmann@31950
   230
          then if k = 0 then size
haftmann@31595
   231
            else @{term "Quickcheck.beyond :: code_numeral \<Rightarrow> code_numeral \<Rightarrow> code_numeral"}
haftmann@31950
   232
             $ HOLogic.mk_number @{typ code_numeral} k $ size
haftmann@31595
   233
          else @{term "1::code_numeral"}
haftmann@31595
   234
      in (is_rec, HOLogic.mk_prod (tk, t)) end;
haftmann@31595
   235
    fun sort_rec xs =
haftmann@31595
   236
      map_filter (fn (true, t) => SOME t | _ =>  NONE) xs
haftmann@31595
   237
      @ map_filter (fn (false, t) => SOME t | _ =>  NONE) xs;
haftmann@31603
   238
    val gen_exprss = tss
haftmann@31603
   239
      |> (map o apfst) Type
haftmann@31595
   240
      |> map (fn (T, cs) => (T, (sort_rec o map (mk_consexpr T)) cs));
haftmann@31595
   241
    fun mk_select (rT, xs) =
haftmann@31595
   242
      mk_const @{const_name Quickcheck.collapse} [@{typ "Random.seed"}, termifyT rT]
haftmann@31595
   243
      $ (mk_const @{const_name Random.select_weight} [random_resultT rT]
haftmann@31595
   244
        $ HOLogic.mk_list (HOLogic.mk_prodT (@{typ code_numeral}, random_resultT rT)) xs)
haftmann@31595
   245
          $ seed;
haftmann@31950
   246
    val auxs_lhss = map (fn t => t $ size $ size' $ seed) random_auxs;
haftmann@31595
   247
    val auxs_rhss = map mk_select gen_exprss;
haftmann@31868
   248
  in (random_auxs, auxs_lhss ~~ auxs_rhss) end;
haftmann@31595
   249
haftmann@38543
   250
fun instantiate_random_datatype config descr vs tycos prfx (names, auxnames) (Ts, Us) thy =
haftmann@31595
   251
  let
haftmann@33968
   252
    val _ = Datatype_Aux.message config "Creating quickcheck generators ...";
haftmann@31595
   253
    val mk_prop_eq = HOLogic.mk_Trueprop o HOLogic.mk_eq;
haftmann@33968
   254
    fun mk_size_arg k = case Datatype_Aux.find_shortest_path descr k
haftmann@31950
   255
     of SOME (_, l) => if l = 0 then size
haftmann@31595
   256
          else @{term "max :: code_numeral \<Rightarrow> code_numeral \<Rightarrow> code_numeral"}
haftmann@31950
   257
            $ HOLogic.mk_number @{typ code_numeral} l $ size
haftmann@31950
   258
      | NONE => size;
haftmann@31868
   259
    val (random_auxs, auxs_eqs) = (apsnd o map) mk_prop_eq
haftmann@31603
   260
      (mk_random_aux_eqs thy descr vs tycos (names, auxnames) (Ts, Us));
haftmann@31595
   261
    val random_defs = map_index (fn (k, T) => mk_prop_eq
haftmann@31950
   262
      (HOLogic.mk_random T size, nth random_auxs k $ mk_size_arg k $ size)) Ts;
haftmann@31595
   263
  in
haftmann@31595
   264
    thy
haftmann@38348
   265
    |> Class.instantiation (tycos, vs, @{sort random})
haftmann@31868
   266
    |> random_aux_specification prfx random_auxN auxs_eqs
haftmann@31595
   267
    |> `(fn lthy => map (Syntax.check_term lthy) random_defs)
haftmann@31595
   268
    |-> (fn random_defs' => fold_map (fn random_def =>
wenzelm@33280
   269
          Specification.definition (NONE, (apfst Binding.conceal
haftmann@33205
   270
            Attrib.empty_binding, random_def))) random_defs')
haftmann@31595
   271
    |> snd
haftmann@31595
   272
    |> Class.prove_instantiation_exit (K (Class.intro_classes_tac []))
haftmann@31595
   273
  end;
haftmann@31595
   274
haftmann@31950
   275
(** building and compiling generator expressions **)
haftmann@31950
   276
wenzelm@41472
   277
(* FIXME just one data slot (record) per program unit *)
wenzelm@41472
   278
wenzelm@41472
   279
structure Counterexample = Proof_Data
wenzelm@41472
   280
(
bulwahn@45754
   281
  type T = unit -> int -> bool -> int -> int * int -> (bool * term list) option * (int * int)
wenzelm@41472
   282
  (* FIXME avoid user error with non-user text *)
haftmann@39388
   283
  fun init _ () = error "Counterexample"
haftmann@39388
   284
);
haftmann@39388
   285
val put_counterexample = Counterexample.put;
haftmann@31950
   286
wenzelm@41472
   287
structure Counterexample_Report = Proof_Data
wenzelm@41472
   288
(
bulwahn@45763
   289
  type T = unit -> int -> bool -> int -> seed -> ((bool * term list) option * (bool list * bool)) * seed
wenzelm@41472
   290
  (* FIXME avoid user error with non-user text *)
haftmann@39388
   291
  fun init _ () = error "Counterexample_Report"
haftmann@39388
   292
);
haftmann@39388
   293
val put_counterexample_report = Counterexample_Report.put;
bulwahn@35378
   294
haftmann@31950
   295
val target = "Quickcheck";
haftmann@31950
   296
bulwahn@42159
   297
fun mk_generator_expr ctxt (t, eval_terms) =
bulwahn@42159
   298
  let  
wenzelm@42361
   299
    val thy = Proof_Context.theory_of ctxt
wenzelm@44241
   300
    val prop = fold_rev absfree (Term.add_frees t []) t
bulwahn@42159
   301
    val Ts = (map snd o fst o strip_abs) prop
haftmann@31950
   302
    val bound_max = length Ts - 1;
haftmann@31950
   303
    val bounds = map_index (fn (i, ty) =>
haftmann@31950
   304
      (2 * (bound_max - i) + 1, 2 * (bound_max - i), 2 * i, ty)) Ts;
haftmann@31950
   305
    val result = list_comb (prop, map (fn (i, _, _, _) => Bound i) bounds);
haftmann@31950
   306
    val terms = HOLogic.mk_list @{typ term} (map (fn (_, i, _, _) => Bound i $ @{term "()"}) bounds);
bulwahn@45754
   307
    val ([genuine_only_name], ctxt') = Variable.variant_fixes ["genuine_only"] ctxt
bulwahn@45754
   308
    val genuine_only = Free (genuine_only_name, @{typ bool})
bulwahn@45763
   309
    val none_t = Const (@{const_name "None"}, resultT)
bulwahn@45763
   310
    val check = Quickcheck_Common.mk_safe_if genuine_only none_t (result, none_t,
bulwahn@45721
   311
      fn genuine => @{term "Some :: bool * term list => (bool * term list) option"} $
bulwahn@45721
   312
        HOLogic.mk_prod (Quickcheck_Common.reflect_bool genuine, terms))
bulwahn@45723
   313
    val return = HOLogic.pair_const resultT @{typ Random.seed};
haftmann@31950
   314
    fun liftT T sT = sT --> HOLogic.mk_prodT (T, sT);
haftmann@31950
   315
    fun mk_termtyp T = HOLogic.mk_prodT (T, @{typ "unit => term"});
haftmann@31950
   316
    fun mk_scomp T1 T2 sT f g = Const (@{const_name scomp},
haftmann@31950
   317
      liftT T1 sT --> (T1 --> liftT T2 sT) --> liftT T2 sT) $ f $ g;
haftmann@31950
   318
    fun mk_split T = Sign.mk_const thy
bulwahn@45723
   319
      (@{const_name prod_case}, [T, @{typ "unit => term"}, liftT resultT @{typ Random.seed}]);
haftmann@31950
   320
    fun mk_scomp_split T t t' =
bulwahn@45723
   321
      mk_scomp (mk_termtyp T) resultT @{typ Random.seed} t
haftmann@31950
   322
        (mk_split T $ Abs ("", T, Abs ("", @{typ "unit => term"}, t')));
haftmann@31950
   323
    fun mk_bindclause (_, _, i, T) = mk_scomp_split T
haftmann@31950
   324
      (Sign.mk_const thy (@{const_name Quickcheck.random}, [T]) $ Bound i);
bulwahn@45754
   325
  in
bulwahn@45754
   326
    lambda genuine_only
bulwahn@45761
   327
      (Abs ("n", @{typ code_numeral}, fold_rev mk_bindclause bounds (return $ check true)))
bulwahn@45754
   328
  end;
haftmann@31950
   329
bulwahn@42159
   330
fun mk_reporting_generator_expr ctxt (t, eval_terms) =
bulwahn@35378
   331
  let
wenzelm@42361
   332
    val thy = Proof_Context.theory_of ctxt
bulwahn@45762
   333
    val resultT = @{typ "(bool * term list) option * (bool list * bool)"}
wenzelm@44241
   334
    val prop = fold_rev absfree (Term.add_frees t []) t
bulwahn@42159
   335
    val Ts = (map snd o fst o strip_abs) prop
bulwahn@42159
   336
    val bound_max = length Ts - 1
bulwahn@35378
   337
    val bounds = map_index (fn (i, ty) =>
bulwahn@35378
   338
      (2 * (bound_max - i) + 1, 2 * (bound_max - i), 2 * i, ty)) Ts;
bulwahn@35378
   339
    val prop' = betapplys (prop, map (fn (i, _, _, _) => Bound i) bounds);
bulwahn@35378
   340
    val terms = HOLogic.mk_list @{typ term} (map (fn (_, i, _, _) => Bound i $ @{term "()"}) bounds)
bulwahn@42195
   341
    val (assms, concl) = Quickcheck_Common.strip_imp prop'
bulwahn@45762
   342
    val return = HOLogic.pair_const resultT @{typ "Random.seed"};
bulwahn@35378
   343
    fun mk_assms_report i =
bulwahn@45762
   344
      HOLogic.mk_prod (@{term "None :: (bool * term list) option"},
haftmann@38553
   345
        HOLogic.mk_prod (HOLogic.mk_list HOLogic.boolT
haftmann@38553
   346
          (replicate i @{term True} @ replicate (length assms - i) @{term False}),
haftmann@38553
   347
        @{term False}))
bulwahn@35378
   348
    fun mk_concl_report b =
haftmann@38553
   349
      HOLogic.mk_prod (HOLogic.mk_list HOLogic.boolT (replicate (length assms) @{term True}),
bulwahn@45763
   350
        Quickcheck_Common.reflect_bool b)
bulwahn@45763
   351
    val ([genuine_only_name], ctxt') = Variable.variant_fixes ["genuine_only"] ctxt
bulwahn@45763
   352
    val genuine_only = Free (genuine_only_name, @{typ bool})
bulwahn@45763
   353
    val none_t = HOLogic.mk_prod (@{term "None :: (bool * term list) option"}, mk_concl_report true)
bulwahn@45763
   354
    val concl_check = Quickcheck_Common.mk_safe_if genuine_only none_t (concl, none_t,
bulwahn@45763
   355
      fn genuine => HOLogic.mk_prod (@{term "Some :: bool * term list => (bool * term list) option"} $
bulwahn@45763
   356
        HOLogic.mk_prod (Quickcheck_Common.reflect_bool genuine, terms), mk_concl_report false))
bulwahn@45763
   357
    val check = fold_rev (fn (i, assm) => fn t => Quickcheck_Common.mk_safe_if genuine_only
bulwahn@45763
   358
      (mk_assms_report i) (HOLogic.mk_not assm, mk_assms_report i, t))
bulwahn@35378
   359
      (map_index I assms) concl_check
bulwahn@35378
   360
    fun liftT T sT = sT --> HOLogic.mk_prodT (T, sT);
bulwahn@35378
   361
    fun mk_termtyp T = HOLogic.mk_prodT (T, @{typ "unit => term"});
bulwahn@35378
   362
    fun mk_scomp T1 T2 sT f g = Const (@{const_name scomp},
bulwahn@35378
   363
      liftT T1 sT --> (T1 --> liftT T2 sT) --> liftT T2 sT) $ f $ g;
bulwahn@35378
   364
    fun mk_split T = Sign.mk_const thy
bulwahn@45762
   365
      (@{const_name prod_case}, [T, @{typ "unit => term"}, liftT resultT @{typ Random.seed}]);
bulwahn@35378
   366
    fun mk_scomp_split T t t' =
bulwahn@45762
   367
      mk_scomp (mk_termtyp T) resultT @{typ Random.seed} t
bulwahn@35378
   368
        (mk_split T $ Abs ("", T, Abs ("", @{typ "unit => term"}, t')));
bulwahn@35378
   369
    fun mk_bindclause (_, _, i, T) = mk_scomp_split T
bulwahn@35378
   370
      (Sign.mk_const thy (@{const_name Quickcheck.random}, [T]) $ Bound i);
bulwahn@35378
   371
  in
bulwahn@45763
   372
    lambda genuine_only
bulwahn@45763
   373
      (Abs ("n", @{typ code_numeral}, fold_rev mk_bindclause bounds (return $ check true)))
bulwahn@35378
   374
  end
bulwahn@35378
   375
bulwahn@42159
   376
val mk_parametric_generator_expr = Quickcheck_Common.gen_mk_parametric_generator_expr 
bulwahn@42159
   377
  ((mk_generator_expr, 
bulwahn@45759
   378
    absdummy @{typ bool} (absdummy @{typ code_numeral}
bulwahn@45759
   379
      @{term "Pair None :: Random.seed => (bool * term list) option * Random.seed"})),
bulwahn@45759
   380
    @{typ "bool => code_numeral => Random.seed => (bool * term list) option * Random.seed"})
bulwahn@42159
   381
bulwahn@42159
   382
val mk_parametric_reporting_generator_expr = Quickcheck_Common.gen_mk_parametric_generator_expr 
bulwahn@42159
   383
  ((mk_reporting_generator_expr,
bulwahn@45763
   384
    absdummy @{typ bool} (absdummy @{typ code_numeral}
bulwahn@45763
   385
      @{term "Pair (None, ([], False)) :: Random.seed =>
bulwahn@45763
   386
        ((bool * term list) option * (bool list * bool)) * Random.seed"})),
bulwahn@45763
   387
    @{typ "bool => code_numeral => Random.seed => ((bool * term list) option * (bool list * bool)) * Random.seed"})
bulwahn@42159
   388
    
bulwahn@42159
   389
    
bulwahn@40911
   390
(* single quickcheck report *)
bulwahn@40911
   391
bulwahn@40911
   392
datatype single_report = Run of bool list * bool | MatchExc
bulwahn@40911
   393
bulwahn@40911
   394
fun collect_single_report single_report
bulwahn@40911
   395
    (Quickcheck.Report {iterations = iterations, raised_match_errors = raised_match_errors,
bulwahn@40911
   396
    satisfied_assms = satisfied_assms, positive_concl_tests = positive_concl_tests}) =
bulwahn@40911
   397
  case single_report
bulwahn@40911
   398
  of MatchExc =>
bulwahn@40911
   399
    Quickcheck.Report {iterations = iterations + 1, raised_match_errors = raised_match_errors + 1,
bulwahn@40911
   400
      satisfied_assms = satisfied_assms, positive_concl_tests = positive_concl_tests}
bulwahn@40911
   401
   | Run (assms, concl) =>
bulwahn@40911
   402
    Quickcheck.Report {iterations = iterations + 1, raised_match_errors = raised_match_errors,
bulwahn@40911
   403
      satisfied_assms =
bulwahn@40911
   404
        map2 (fn b => fn s => if b then s + 1 else s) assms
bulwahn@40911
   405
         (if null satisfied_assms then replicate (length assms) 0 else satisfied_assms),
bulwahn@40911
   406
      positive_concl_tests = if concl then positive_concl_tests + 1 else positive_concl_tests}
bulwahn@40911
   407
bulwahn@40911
   408
val empty_report = Quickcheck.Report { iterations = 0, raised_match_errors = 0,
bulwahn@40911
   409
  satisfied_assms = [], positive_concl_tests = 0 }
bulwahn@40911
   410
    
bulwahn@42159
   411
fun compile_generator_expr ctxt ts =
haftmann@31950
   412
  let
wenzelm@42361
   413
    val thy = Proof_Context.theory_of ctxt
bulwahn@40911
   414
    val iterations = Config.get ctxt Quickcheck.iterations
bulwahn@40911
   415
  in
bulwahn@40911
   416
    if Config.get ctxt Quickcheck.report then
bulwahn@40911
   417
      let
bulwahn@42159
   418
        val t' = mk_parametric_reporting_generator_expr ctxt ts;
bulwahn@40911
   419
        val compile = Code_Runtime.dynamic_value_strict
bulwahn@41923
   420
          (Counterexample_Report.get, put_counterexample_report, "Random_Generators.put_counterexample_report")
bulwahn@45762
   421
          thy (SOME target)
bulwahn@45763
   422
          (fn proc => fn g => fn c => fn b => fn s => g c b s
bulwahn@45763
   423
            #>> (apfst o Option.map o apsnd o map) proc) t' [];
bulwahn@45763
   424
        fun single_tester c b s = compile c b s |> Random_Engine.run
bulwahn@45763
   425
        fun iterate_and_collect _ (card, size) 0 report = (NONE, report)
bulwahn@45763
   426
          | iterate_and_collect genuine_only (card, size) j report =
bulwahn@40911
   427
            let
bulwahn@45763
   428
              val (test_result, single_report) = apsnd Run (single_tester card genuine_only size)
bulwahn@40911
   429
              val report = collect_single_report single_report report
bulwahn@40911
   430
            in
bulwahn@45763
   431
              case test_result of NONE => iterate_and_collect genuine_only (card, size) (j - 1) report
bulwahn@45762
   432
                | SOME q => (SOME q, report)
bulwahn@40911
   433
            end
bulwahn@40911
   434
      in
bulwahn@45763
   435
        fn genuine_only => fn [card, size] =>
bulwahn@45763
   436
          apsnd SOME (iterate_and_collect genuine_only (card, size) iterations empty_report)
bulwahn@40911
   437
      end
bulwahn@40911
   438
    else
bulwahn@40911
   439
      let
bulwahn@42159
   440
        val t' = mk_parametric_generator_expr ctxt ts;
bulwahn@40911
   441
        val compile = Code_Runtime.dynamic_value_strict
bulwahn@41923
   442
          (Counterexample.get, put_counterexample, "Random_Generators.put_counterexample")
bulwahn@45723
   443
          thy (SOME target)
bulwahn@45754
   444
          (fn proc => fn g => fn c => fn b => fn s => g c b s
bulwahn@45754
   445
            #>> (Option.map o apsnd o map) proc) t' [];
bulwahn@45754
   446
        fun single_tester c b s = compile c b s |> Random_Engine.run
bulwahn@45754
   447
        fun iterate _ (card, size) 0 = NONE
bulwahn@45754
   448
          | iterate genuine_only (card, size) j =
bulwahn@45754
   449
            case single_tester card genuine_only size of
bulwahn@45754
   450
              NONE => iterate genuine_only (card, size) (j - 1)
bulwahn@45754
   451
            | SOME q => SOME q
bulwahn@40911
   452
      in
bulwahn@45754
   453
        fn genuine_only => fn [card, size] =>
bulwahn@45754
   454
          (rpair NONE (iterate genuine_only (card, size) iterations))
bulwahn@40911
   455
      end
bulwahn@35378
   456
  end;
haftmann@31950
   457
bulwahn@46331
   458
fun size_matters_for _ Ts = not (forall
bulwahn@46331
   459
  (fn Type (tyco, []) => is_some (try (unprefix "Enum.finite") tyco) | _ => false) Ts)
bulwahn@46331
   460
bulwahn@46331
   461
val test_goals =
bulwahn@46331
   462
  Quickcheck_Common.generator_test_goal_terms ("random", (size_matters_for, compile_generator_expr));
bulwahn@43875
   463
  
haftmann@31260
   464
(** setup **)
haftmann@31260
   465
bulwahn@43878
   466
val active = Attrib.setup_config_bool @{binding quickcheck_random_active} (K false);
bulwahn@43878
   467
haftmann@38393
   468
val setup =
bulwahn@45923
   469
  Quickcheck_Common.datatype_interpretation (@{sort random}, instantiate_random_datatype)
bulwahn@43878
   470
  #> Context.theory_map (Quickcheck.add_tester ("random", (active, test_goals)));
haftmann@31260
   471
haftmann@31260
   472
end;