This commit is contained in:
rhiannon morris 2022-12-01 19:23:53 +01:00
parent 835eeefc0a
commit 7797aa4002
4 changed files with 70 additions and 59 deletions

17
aoc.m
View file

@ -6,31 +6,28 @@
:- implementation. :- implementation.
:- import_module basics. :- import_module basics.
:- import_module die.
:- import_module string. :- import_module string.
:- import_module list. :- import_module list.
:- import_module exception. :- import_module exception.
main(!IO) :- wrap_main(main2, !IO). main(!IO) :- wrap_main(main2, !IO).
:- pred main2(io::di, io::uo) is cc_multi. :- pred main2(io::di, io::uo) is det.
:- pragma no_determinism_warning(main2/2).
main2(!IO) :- main2(!IO) :-
command_line_arguments(Args, !IO), command_line_arguments(Args, !IO),
(if (Args = [DayS, PartS | Rest], (if Args = [DayS, PartS | Rest],
to_int(DayS, Day), to_int(DayS, Day),
part(PartS, Part)) part(PartS, Part) then
then
run(Day, Part, Rest, !IO) run(Day, Part, Rest, !IO)
else if Args = [D, P | _] then else if Args = [D, P | _] then
die("expected a day and a part, got ""%s"" ""%s""", [s(D), s(P)]) die("expected a day and a part, got ""%s"" ""%s""", [s(D), s(P)])
else else
die("expected at least two args")). die("expected at least two args")).
:- type solution == (pred(part, list(string), io, io)). :- type sol == (pred(part, list(string), io, io)).
:- inst solution == (pred(in, in, di, uo) is det). :- inst sol == (pred(in, in, di, uo) is det).
:- pred solution(int::in, solution::out(solution)) is semidet. :- pred solution(int::in, sol::out(sol)) is semidet.
:- import_module day1. :- import_module day1.
solution(1, day1.run). solution(1, day1.run).

View file

@ -1,13 +1,62 @@
:- module basics. :- module basics.
:- interface. :- interface.
:- type part ---> one; two. :- type part ---> one; two.
:- pred part(string, part). :- pred part(string, part).
:- mode part(in, out) is semidet. :- mode part(in, out) is semidet.
:- mode part(out, in) is semidet. :- mode part(out, in) is det.
:- import_module io.
:- import_module list.
:- import_module string.
:- pred die(string::in) is erroneous.
:- pred die(string::in, list(poly_type)::in) is erroneous.
:- pred need(io.res(T)::in, T::out) is det.
:- pred read_lines_need(string::in, list(string)::out, io::di, io::uo) is det.
:- type main == (pred(io, io)).
:- inst main == (pred(di, uo) is det).
:- pred wrap_main(main::in(main), io::di, io::uo) is cc_multi.
:- import_module maybe.
:- func to_int_may(string) = maybe(int).
:- implementation. :- implementation.
part("1", one). part("1", one).
part("2", two). part("2", two).
:- import_module exception.
:- type die ---> death(string).
die(Str) :- throw(death(Str)).
die(Fmt, Args) :- die(format(Fmt, Args)).
wrap_main(Main, !IO) :-
try [io(!IO)] Main(!IO)
then true
catch death(Err) -> (
format("%s\n", [s(Err)], !IO),
set_exit_status(1, !IO)
).
need(ok(X), X).
need(error(E), _) :- die(error_message(E)).
read_lines_need(File, Lines, !IO) :-
read_named_file_as_lines(File, Res, !IO),
need(Res, Lines).
to_int_may(Str) = Out :-
if to_int(Str, N) then Out = yes(N) else Out = no.

30
day1.m
View file

@ -7,15 +7,14 @@
:- pred run(part::in, list(string)::in, io::di, io::uo) is det. :- pred run(part::in, list(string)::in, io::di, io::uo) is det.
:- implementation. :- implementation.
:- import_module die.
:- import_module std_util. :- import_module std_util.
:- import_module maybe.
:- import_module string. :- import_module string.
:- import_module int. :- import_module int.
run(Part, Args, !IO) :- run(Part, Args, !IO) :-
if Args = [File] then ( if Args = [File] then (
read_named_file_as_lines(File, LinesM, !IO), read_lines_need(File, Lines, !IO),
need(LinesM, Lines),
Totals = go(Lines), Totals = go(Lines),
( Part = one, ( Part = one,
format("%d\n", [i(det_head(Totals))], !IO) format("%d\n", [i(det_head(Totals))], !IO)
@ -26,21 +25,16 @@ run(Part, Args, !IO) :-
) else die("expected one readable file"). ) else die("expected one readable file").
:- func go(list(string)) = list(int). :- func go(list(string)) = list(int).
go(Lines) = sort(converse(ordering), map(int_sum, gather(Lines))). go(Lines) =
sort(converse(ordering), map(sum, gather(map(to_int_may, Lines)))).
:- func gather(list(string)::in) = (list(list(string))::out(non_empty_list)). :- func gather(list(maybe(T))) = list(list(T)).
gather([]) = [[]]. gather(Xs) = Ys :- gather(Xs, Ys).
gather([X | Xs]) = Groups :-
[Ys | Yss] = gather(Xs), :- pred gather(list(maybe(T))::in, list(list(T))::out(non_empty_list)) is det.
(if X = "" then Groups = [[], Ys | Yss] gather([], [[]]).
else Groups = [[X | Ys] | Yss]). gather([no | Xs], [[] | Yss]) :- gather(Xs, Yss).
gather([yes(Y) | Xs], [[Y | Ys] | Yss]) :- gather(Xs, [Ys | Yss]).
:- func sum(list(int)) = int. :- func sum(list(int)) = int.
sum(Xs) = foldl(func(X, Y) = X + Y, Xs, 0). sum(Xs) = foldl(plus, Xs, 0).
:- func int_sum(list(string)) = int.
int_sum(Xs) = sum(map(det_to_int, Xs)).
:- pred need(io.res(T)::in, T::out) is det.
need(ok(X), X).
need(error(E), _) :- die(error_message(E)).

29
die.m
View file

@ -1,29 +0,0 @@
:- module die.
:- interface.
:- import_module io.
:- import_module list.
:- import_module string.
:- pred die(string::in) is erroneous.
:- pred die(string::in, list(poly_type)::in) is erroneous.
:- type main == (pred(io, io)).
:- inst main == (pred(di, uo) is cc_multi).
:- pred wrap_main(main::in(main), io::di, io::uo) is cc_multi.
:- implementation.
:- import_module exception.
:- type die ---> death(string).
die(Str) :- throw(death(Str)).
die(Fmt, Args) :- die(format(Fmt, Args)).
wrap_main(Main, !IO) :-
try [io(!IO)] Main(!IO)
then true
catch death(Err) -> (
format("%s\n", [s(Err)], !IO),
set_exit_status(1, !IO)
).