Files
mercury/tests/hard_coded/rotate_uint32.m
Julien Fischer 9e58712489 Add bitwise rotations for uint32s.
library/uint32.m:
    As above.

tests/hard_coded/Mmakefile:
tests/hard_coded/rotate_uint32.{m,exp}:
    Add a test for the new operations.
2021-03-28 14:28:22 +11:00

125 lines
3.5 KiB
Mathematica

%---------------------------------------------------------------------------%
% vim: ft=mercury ts=4 sw=4 et
:- module rotate_uint32.
:- interface.
:- import_module io.
:- pred main(io::di, io::uo) is cc_multi.
:- implementation.
:- import_module exception.
:- import_module list.
:- import_module uint32.
:- import_module string.
%---------------------------------------------------------------------------%
main(!IO) :-
run_test(rotate_left, unchecked_rotate_left, "rotate_left", !IO),
run_test(rotate_right,unchecked_rotate_right, "rotate_right", !IO).
%---------------------------------------------------------------------------%
:- pred run_test((func(uint32, uint) = uint32)::in,
(func(uint32, uint) = uint32)::in,
string::in, io::di, io::uo) is cc_multi.
run_test(CheckedFunc, UncheckedFunc, Desc, !IO) :-
io.format("*** Test '%s' ***\n\n", [s(Desc)], !IO),
Ns = numbers,
Ds = distances,
list.foldl(run_test_2(CheckedFunc, UncheckedFunc, Desc, Ds), Ns, !IO).
:- pred run_test_2((func(uint32, uint) = uint32)::in,
(func(uint32, uint) = uint32)::in, string::in,
list(uint)::in, uint32::in, io::di, io::uo) is cc_multi.
run_test_2(CheckedFunc, UncheckedFunc, Desc, Ds, N, !IO) :-
list.foldl(run_test_3(CheckedFunc, UncheckedFunc, Desc, N), Ds, !IO).
:- pred run_test_3((func(uint32, uint) = uint32)::in,
(func(uint32, uint) = uint32)::in, string::in, uint32::in,
uint::in, io::di, io::uo) is cc_multi.
run_test_3(CheckedFunc, UncheckedFunc, Desc, N, D, !IO) :-
do_eval(CheckedFunc, N, D, CheckedResult),
io.format(" %s(%s, %u) = %s\n",
[s(Desc), s(to_binary_string_lz(N)), u(D), s(CheckedResult)], !IO),
do_eval(UncheckedFunc, N, D, UncheckedResult),
io.format("unchecked_%s(%s, %u) = %s\n",
[s(Desc), s(to_binary_string_lz(N)), u(D), s(UncheckedResult)], !IO),
io.nl(!IO).
:- pred do_eval((func(uint32, uint) = uint32)::in, uint32::in, uint::in,
string::out) is cc_multi.
do_eval(Func, N, D, ResultStr) :-
( try []
Result0 = Func(N, D)
then
ResultStr = to_binary_string_lz(Result0)
catch_any _ ->
ResultStr = "<<exception>>"
).
%---------------------------------------------------------------------------%
:- func numbers = list(uint32).
numbers = [
1_u32,
255_u32
].
:- func distances = list(uint).
distances = [
0_u,
1_u,
2_u,
31_u,
32_u,
63_u,
64_u
].
%---------------------------------------------------------------------------%
:- func to_binary_string_lz(uint32::in) = (string::uo) is det.
:- pragma foreign_proc("C",
to_binary_string_lz(U::in) = (S::uo),
[will_not_call_mercury, promise_pure, thread_safe, will_not_modify_trail],
"
int i;
MR_allocate_aligned_string_msg(S, 32, MR_ALLOC_ID);
S[32] = '\\0';
for (i = 31; i >= 0; i--) {
S[i] = (U & 1) ? '1' : '0';
U = U >> 1;
}
").
:- pragma foreign_proc("C#",
to_binary_string_lz(U::in) = (S::uo),
[will_not_call_mercury, promise_pure, thread_safe],
"
S = System.Convert.ToString(U, 2).PadLeft(32, '0');
").
:- pragma foreign_proc("Java",
to_binary_string_lz(U::in) = (S::uo),
[will_not_call_mercury, promise_pure, thread_safe],
"
S = java.lang.String.format(""%32s"",
java.lang.Integer.toBinaryString(U)).replace(' ', '0');
").
%---------------------------------------------------------------------------%
:- end_module rotate_uint32.
%---------------------------------------------------------------------------%