blob: 1469358c9dd61761f15b953c4e0f36083b23c309 (
plain)
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
(************************************************************************)
(* * The Coq Proof Assistant / The Coq Development Team *)
(* v * INRIA, CNRS and contributors - Copyright 1999-2018 *)
(* <O___,, * (see CREDITS file for the list of authors) *)
(* \VV/ **************************************************************)
(* // * This file is distributed under the terms of the *)
(* * GNU Lesser General Public License Version 2.1 *)
(* * (see LICENSE file for the text of the license) *)
(************************************************************************)
(** This module implements an "untyped store", in this particular case
we see it as an extensible record whose fields are left
unspecified. ***)
(** We use a dynamic "name" allocator. But if we needed to serialise
stores, we might want something static to avoid troubles with
plugins order. *)
module type S =
sig
type t
type 'a field
val empty : t
val set : t -> 'a field -> 'a -> t
val get : t -> 'a field -> 'a option
val remove : t -> 'a field -> t
val merge : t -> t -> t
val field : unit -> 'a field
end
module Make () : S =
struct
let next =
let count = ref 0 in fun () ->
let n = !count in
incr count;
n
type t = Obj.t option array
(** Store are represented as arrays. For small values, which is typicial,
is slightly quicker than other implementations. *)
type 'a field = int
let allocate len : t = Array.make len None
let empty : t = [||]
let set (s : t) (i : 'a field) (v : 'a) : t =
let len = Array.length s in
let nlen = if i < len then len else succ i in
let () = assert (0 <= i) in
let ans = allocate nlen in
Array.blit s 0 ans 0 len;
Array.unsafe_set ans i (Some (Obj.repr v));
ans
let get (s : t) (i : 'a field) : 'a option =
let len = Array.length s in
if len <= i then None
else Obj.magic (Array.unsafe_get s i)
let remove (s : t) (i : 'a field) =
let len = Array.length s in
let () = assert (0 <= i) in
let ans = allocate len in
Array.blit s 0 ans 0 len;
if i < len then Array.unsafe_set ans i None;
ans
let merge (s1 : t) (s2 : t) : t =
let len1 = Array.length s1 in
let len2 = Array.length s2 in
let nlen = if len1 < len2 then len2 else len1 in
let ans = allocate nlen in
(** Important: No more allocation from here. *)
Array.blit s2 0 ans 0 len2;
for i = 0 to pred len1 do
let v = Array.unsafe_get s1 i in
match v with
| None -> ()
| Some _ -> Array.unsafe_set ans i v
done;
ans
let field () = next ()
end
|
