-
Notifications
You must be signed in to change notification settings - Fork 76
/
int32.ml
120 lines (103 loc) · 4.99 KB
/
int32.ml
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
# 2 "int32.ml"
(**************************************************************************)
(* *)
(* OCaml *)
(* *)
(* Xavier Leroy, projet Cristal, INRIA Rocquencourt *)
(* *)
(* Copyright 1996 Institut National de Recherche en Informatique et *)
(* en Automatique. *)
(* *)
(* All rights reserved. This file is distributed under the terms of *)
(* the GNU Lesser General Public License version 2.1, with the *)
(* special exception on linking described in the file LICENSE. *)
(* *)
(**************************************************************************)
open! Stdlib
[@@@ocaml.flambda_o3]
(* Module [Int32]: 32-bit integers *)
external neg : (int32[@local_opt]) -> (int32[@local_opt]) = "%int32_neg"
external add : (int32[@local_opt]) -> (int32[@local_opt]) -> (int32[@local_opt]) = "%int32_add"
external sub : (int32[@local_opt]) -> (int32[@local_opt]) -> (int32[@local_opt]) = "%int32_sub"
external mul : (int32[@local_opt]) -> (int32[@local_opt]) -> (int32[@local_opt]) = "%int32_mul"
external div : (int32[@local_opt]) -> (int32[@local_opt]) -> (int32[@local_opt]) = "%int32_div"
external rem : (int32[@local_opt]) -> (int32[@local_opt]) -> (int32[@local_opt]) = "%int32_mod"
external logand : (int32[@local_opt]) -> (int32[@local_opt]) -> (int32[@local_opt]) = "%int32_and"
external logor : (int32[@local_opt]) -> (int32[@local_opt]) -> (int32[@local_opt]) = "%int32_or"
external logxor : (int32[@local_opt]) -> (int32[@local_opt]) -> (int32[@local_opt]) = "%int32_xor"
external shift_left : (int32[@local_opt]) -> int -> (int32[@local_opt]) = "%int32_lsl"
external shift_right : (int32[@local_opt]) -> int -> (int32[@local_opt]) = "%int32_asr"
external shift_right_logical : (int32[@local_opt]) -> int -> (int32[@local_opt]) = "%int32_lsr"
external of_int : int -> (int32[@local_opt]) = "%int32_of_int"
external to_int : (int32[@local_opt]) -> int = "%int32_to_int"
external of_float : float -> int32
= "caml_int32_of_float" "caml_int32_of_float_unboxed"
[@@unboxed] [@@noalloc]
external to_float : int32 -> float
= "caml_int32_to_float" "caml_int32_to_float_unboxed"
[@@unboxed] [@@noalloc]
external bits_of_float : float -> int32
= "caml_int32_bits_of_float" "caml_int32_bits_of_float_unboxed"
[@@unboxed] [@@noalloc]
external float_of_bits : int32 -> float
= "caml_int32_float_of_bits" "caml_int32_float_of_bits_unboxed"
[@@unboxed] [@@noalloc]
let zero = 0l
let one = 1l
let minus_one = -1l
let[@inline available] succ n = add n 1l
let[@inline available] pred n = sub n 1l
let[@inline available] abs n = if n >= 0l then n else neg n
let min_int = 0x80000000l
let max_int = 0x7FFFFFFFl
let[@inline available] lognot n = logxor n (-1l)
let unsigned_to_int =
match Sys.word_size with
| 32 ->
let max_int = of_int Stdlib.max_int in
fun[@inline available] n ->
if n >= 0l && n <= max_int then
Some (to_int n)
else
None
| 64 ->
(* So that it compiles in 32-bit *)
let mask = 0xFFFF lsl 16 lor 0xFFFF in
fun[@inline available] n -> Some (to_int n land mask)
| _ ->
assert false
external format : string -> int32 -> string = "caml_int32_format"
let[@inline available] to_string n = format "%d" n
external of_string : string -> (int32[@unboxed])
= "caml_int32_of_string" "caml_int32_of_string_unboxed"
let[@inline available] of_string_opt s =
try Some (of_string s)
with Failure _ -> None
type t = int32
let[@inline available] compare (x: t) (y: t) = Stdlib.compare x y
let[@inline available] equal (x: t) (y: t) = x = y
let[@inline available] unsigned_compare n m =
compare (sub n min_int) (sub m min_int)
let unsigned_lt n m =
sub n min_int < sub m min_int
let[@inline available] min x y : t = if x <= y then x else y
let[@inline available] max x y : t = if x >= y then x else y
(* Unsigned division from signed division of the same bitness.
See Warren Jr., Henry S. (2013). Hacker's Delight (2 ed.), Sec 9-3.
*)
let[@inline available] unsigned_div n d =
if d < zero then
if unsigned_lt n d then zero else one
else
let q = shift_left (div (shift_right_logical n 1) d) 1 in
let r = sub n (mul q d) in
if unsigned_lt r d then q else succ q
let[@inline available] unsigned_rem n d =
sub n (mul ((unsigned_div[@inlined]) n d) d)
(* [caml_hash_exn] doesn't raise on int32s, so it's safe for
it to be marked as [@@noalloc].
*)
external seeded_hash_param :
int -> int -> int -> int32 -> int = "caml_hash_exn" [@@noalloc]
let seeded_hash seed x = seeded_hash_param 10 100 seed x
let hash x = seeded_hash_param 10 100 0 x