Files
mercury/tests/hard_coded/checked_nondet_tailcall.m
2020-10-04 13:20:42 +11:00

74 lines
1.7 KiB
Mathematica

%---------------------------------------------------------------------------%
% vim: ts=4 sw=4 et ft=mercury
%---------------------------------------------------------------------------%
:- module checked_nondet_tailcall.
:- interface.
:- import_module io.
:- pred main(io::di, io::uo) is det.
:- implementation.
:- import_module list.
:- import_module solutions.
main(!IO) :-
join(1, Ones),
write_ints(Ones, !IO),
join(2, Twos),
write_ints(Twos, !IO),
join(3, Threes),
write_ints(Threes, !IO),
join(4, Fours),
write_ints(Fours, !IO),
join(5, Fives),
write_ints(Fives, !IO).
:- pred edge1(int::in, int::out) is nondet.
edge1(1, 11).
edge1(1, 12).
:- pred edge2(int::in, int::out) is nondet.
edge2(11, 18).
edge2(12, 19).
edge2(2, 21).
edge2(2, 22).
edge2(3, 31).
edge2(4, 41).
:- pred edge12(int::in, int::out) is nondet.
edge12(A, B) :-
( if edge1(A, C) then
% When we come here after the last success of edge1,
% the call to edge2 will be a checked nondet tailcall.
% When we come here after a non-last success of edge1,
% the check will fail, and we will have to do a
% non-tail call.
D = C
else
% When this arm of the switch is taken, the call to edge2
% will be a checked nondet tailcall.
D = A
),
edge2(D, B).
:- pred join(int::in, list(int)::out) is det.
join(A, Bs) :-
solutions((pred(B::out) is nondet :- edge12(A, B)), Bs).
:- pred write_ints(list(int)::in, io::di, io::uo) is det.
write_ints([], !IO) :-
io.write_string("\n", !IO).
write_ints([X | Xs], !IO) :-
io.write_int(X, !IO),
io.write_string(" ", !IO),
write_ints(Xs, !IO).