This repository has been archived on 2023-08-20. You can view files and clone it, but cannot push or open issues or pull requests.
polynomialmani.pl/polimani.pl

167 lines
3.8 KiB
Prolog

%% Follows 'Coding guidelines for Prolog' - Theory and Practice of Logic Programming
%% https://doi.org/10.1017/S1471068411000391
polynomial_variables([x, y, z]).
polynomial_variable_p(X) :-
polynomial_variables(V), member(X, V).
power_p(X) :-
polynomial_variable_p(X), !.
power_p(X^N) :-
polynomial_variable_p(X), integer(N), N > 1, !.
%% coefficient_p(K) :-
%% number(K).
%% term_p(X) :-
%% polynomial_variable_p(X), !.
term_p(N) :-
number(N).
term_p(X) :-
power_p(X).
term_p(L * R) :-
term_p(L),
term_p(R), !.
polynomial_p(M) :-
term_p(M).
polynomial_p(L + R) :- % Left greedy
polynomial_p(L), !, % Why?
term_p(R), !.
%% ?- polynomial_p(3*x^2+y*z).
%@ true.
%% ?- polynomial_p(x^100*y*z).
%@ true.
%% ?- polynomial_p(x+y+z).
%@ true.
%@ false.
%@ false.
%% ?- polynomial_p(3*x^2+y*z+x^100*y*z).
%@ true.
%@ true .
%@ false.
%@ false.
%@ false.
%@ false.
simplify_term(1 * P, P) :-
term_p(P), !.
simplify_term(0 * _, 0) :-
!.
simplify_term(T, S) :-
term_to_list(T, L),
sort(L, L2),
join_like_terms(L2, S).
%% ?- simplify_term(2*y*z*x^3*x, X).
%@ X = [2, x^1, x^3, y^1, z^1].
join_like_terms([T1, T2 | L], [P^N | L]) :-
P^N1 is T,
P^N2 is T2,
N is N1 + N2,
!.
join_like_terms([N1, N2 | L], [N | L]) :-
N is N1 * N2,
!.
join_like_terms(T, T).
%% ?- join_like_terms([2, 3, x^1, x^2], T).
%@ ERROR: Arguments are not sufficiently instantiated
%@ ERROR: In:
%@ ERROR: [9] _3138^_3140 is _3134
%@ ERROR: [8] join_like_terms([2,3|...],[_3188^_3190,...|...]) at /tmp/ediprologODLYxF:62
%@ ERROR: [7] <user>
%@ Exception: (8) join_like_terms([2, 3, x^1, x^2], _2428) ? abort
%@ % Execution Aborted
%@ T = [6, x^1, x^2].
%@ T = [6, x^1].
%@ T = [6].
%@ T = [2].
term_to_list(L * N, [N | TS]) :-
number(N),
term_to_list(L, TS), !.
term_to_list(L * T, [T2 | TS]) :-
power_p(T),
power_to_canon(T, T2),
term_to_list(L, TS), !.
term_to_list(T, [T]).
power_to_canon(T, T^1) :-
polynomial_variable_p(T).
power_to_canon(T^N, T^N) :-
polynomial_variable_p(T).
simplify_polynomial(M, M2) :-
term_p(M), simplify_term(M, M2), !.
simplify_polynomial(P + 0, P) :-
term_p(P), !.
simplify_polynomial(0 + P, P) :-
term_p(P), !.
simplify_polynomial(P + M, P2 + M2) :-
simplify_polynomial(P, P2), simplify_term(M, M2).
simplify_polynomial(P + M, P2 + M3) :-
monomial_parts(M, _, XExp),
delete_monomial(P, XExp, M2, P2), !,
add_monomial(M, M2, M3).
simplify_polynomial(P + M, P2 + M2) :-
simplify_polynomial(P, P2), simplify_term(M, M2).
%% ?- simplify_polynomial(1*x+(-1)*x, P).
monomial_parts(X, 1, X) :-
power_p(X), !.
monomial_parts(X^N, 1, X^N) :-
power_p(X^N), !.
monomial_parts(K * M, K, M) :-
coeficient_p(K), !.
monomial_parts(K, K, indep) :-
coeficient_p(K), !.
delete_monomial(M, X, M, 0) :-
term_p(M),
monomial_parts(M, _, X), !.
delete_monomial(M + M2, X, M, M2) :-
term_p(M2), term_p(M),
monomial_parts(M, _, X), !.
delete_monomial(P + M, X, M, P) :-
term_p(M), monomial_parts(M, _, X), !.
delete_monomial(P + M2, X, M, P2 + M2) :-
delete_monomial(P, X, M, P2).
add_monomial(K1, K2, K3) :-
number(K1), number(K2), !,
K3 is K1 + K2.
add_monomial(M1, M2, M3) :-
monomial_parts(M1, K1, XExp),
monomial_parts(M2, K2, XExp),
K3 is K1 + K2,
p_aux_add_monomial(K3, XExp, M3).
p_aux_add_monomial(K, indep, K) :-
!.
p_aux_add_monomial(0, _, 0) :-
!.
p_aux_add_monomial(1, XExp, XExp) :-
!.
p_aux_add_monomial(K, XExp, K * XExp).
closure_simplify_polynomial(P, P) :-
simplify_polynomial(P, P2),
P==P2, !.
closure_simplify_polynomial(P, P3) :-
simplify_polynomial(P, P2),
closure_simplify_polynomial(P2, P3), !.
%% ?- simplify_polynomial(1*x+(-1)*x, P).
%@ P = x+ -1*x .
%@ P = x+ -1*x
%@ Unknown action: q (h for help)
%@ Action?
%@ Unknown action: q (h for help)
%@ Action? .