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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
(* SPDX-License-Identifier: Apache-2.0 *)
(* Copyright 2017 WebAssembly Community Group participants *)
(* This file is originally from the WebAssembly reference interpreter available at https://github.com/WebAssembly/spec/tree/main/interpreter *)
(* SPDX-License-Identifier: MIT *)
(* Copyright (c) 2016--2024 Jane Street Group, LLC <opensource-contacts@janestreet.com> *)
(* The code of the `popcnt` function is originally from the `base` library available at https://github.com/janestreet/base *)
(* SPDX-License-Identifier: AGPL-3.0-or-later *)
(* Copyright © 2021-2024 OCamlPro *)
(* Modified by the Owi programmers *)
include Prelude.Int64
let clz n = of_int (Ocaml_intrinsics.Int64.count_leading_zeros n)
let ctz n = of_int (Ocaml_intrinsics.Int64.count_trailing_zeros n)
(* Taken from Base *)
let popcnt =
let ( + ) = add in
let ( - ) = sub in
let ( * ) = mul in
let ( lsr ) = shift_right_logical in
let ( land ) = logand in
let m1 = 0x5555555555555555L in
(* 0b01010101... *)
let m2 = 0x3333333333333333L in
(* 0b00110011... *)
let m4 = 0x0f0f0f0f0f0f0f0fL in
(* 0b00001111... *)
let h01 = 0x0101010101010101L in
(* 1 bit set per byte *)
fun [@inline] x ->
(* gather the bit count for every pair of bits *)
let x = x - ((x lsr 1) land m1) in
(* gather the bit count for every 4 bits *)
let x = (x land m2) + ((x lsr 2) land m2) in
(* gather the bit count for every byte *)
let x = (x + (x lsr 4)) land m4 in
(* sum the bit counts in the top byte and shift it down *)
(x * h01) lsr 56
(*
* Unsigned comparison in terms of signed comparison.
*)
let cmp_u x op y = op (add x min_int) (add y min_int)
let eq (x : int64) y = equal x y
let ne (x : int64) y = not (equal x y)
let lt (x : int64) y = compare x y < 0
let gt (x : int64) y = compare x y > 0
let le (x : int64) y = compare x y <= 0
let ge (x : int64) y = compare x y >= 0
let lt_u x y = cmp_u x lt y
let le_u x y = cmp_u x le y
let gt_u x y = cmp_u x gt y
let ge_u x y = cmp_u x ge y
(*
* Unsigned division and remainder in terms of signed division; algorithm from
* Hacker's Delight, Second Edition, by Henry S. Warren, Jr., section 9-3
* "Unsigned Short Division from Signed Division".
*)
let divrem_u n d =
if equal d zero then raise Division_by_zero
else
let t = shift_right d 63 in
let n' = logand n (lognot t) in
let q = shift_left (div (shift_right_logical n' 1) d) 1 in
let r = sub n (mul q d) in
if cmp_u r lt d then (q, r) else (add q one, sub r d)
(* We don't override min_int and max_int since those are used
* by other functions (like parsing), and rely on it being
* min/max for int32 *)
(* The smallest signed |bitwidth|-bits int. *)
let low_int = shift_left minus_one 63
(* The largest signed |bitwidth|-bits int. *)
let high_int = logxor low_int minus_one
(* WebAssembly's shifts mask the shift count according to the bitwidth. *)
let shift f x y = f x (to_int (logand y (of_int 63)))
let shl x y = shift shift_left x y
let shr_s x y = shift shift_right x y
let shr_u x y = shift shift_right_logical x y
let rotl x y =
let n = logand y 63L in
logor (shl x n) (shr_u x (sub 64L n))
let rotr x y =
let n = logand y 63L in
logor (shr_u x n) (shl x (sub 64L n))
let extend_s n x =
let shift = 64 - n in
shift_right (shift_left x shift) shift
(* String conversion that allows leading signs and unsigned values *)
let require b = if not b then Log.err "of_string (int64)"
let dec_digit = function
| '0' .. '9' as c -> Char.code c - Char.code '0'
| _ -> Log.err "of_string"
let hex_digit = function
| '0' .. '9' as c -> Char.code c - Char.code '0'
| 'a' .. 'f' as c -> 0xa + Char.code c - Char.code 'a'
| 'A' .. 'F' as c -> 0xa + Char.code c - Char.code 'A'
| _ -> Log.err "of_string"
let max_upper, max_lower = divrem_u minus_one 10L
let of_string_exn s =
let len = String.length s in
let rec parse_hex i num =
if i = len then num
else if Char.equal s.[i] '_' then parse_hex (i + 1) num
else
let digit = of_int (hex_digit s.[i]) in
require (le_u num (shr_u minus_one (of_int 4)));
parse_hex (i + 1) (logor (shift_left num 4) digit)
in
let rec parse_dec i num =
if i = len then num
else if Char.equal s.[i] '_' then parse_dec (i + 1) num
else
let digit = of_int (dec_digit s.[i]) in
require (lt_u num max_upper || (eq num max_upper && le_u digit max_lower));
parse_dec (i + 1) (add (mul num 10L) digit)
in
let parse_int i =
require (len - i > 0);
if i + 2 <= len && Char.equal s.[i] '0' && Char.equal s.[i + 1] 'x' then
parse_hex (i + 2) zero
else parse_dec i zero
in
require (len > 0);
let parsed =
match s.[0] with
| '+' -> parse_int 1
| '-' ->
let n = parse_int 1 in
require (ge (sub n one) minus_one);
neg n
| _ -> parse_int 0
in
require (le low_int parsed && le parsed high_int);
parsed
let of_string s = try Some (of_string_exn s) with _ -> None