You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.3 KiB
Julia
60 lines
1.3 KiB
Julia
# External dependencies
|
|
using TypedPolynomials
|
|
|
|
# Local dependencies
|
|
include("homotopy.jl")
|
|
include("plot.jl")
|
|
include("euler-newton.jl")
|
|
include("adapt-step.jl")
|
|
include("start-system.jl")
|
|
include("homogenize.jl")
|
|
using .Homotopy
|
|
using .Plot
|
|
using .EulerNewton
|
|
using .AdaptStep
|
|
using .StartSystem
|
|
using .Homogenize
|
|
|
|
# Main homotopy continuation loop
|
|
function solve(F, (G, roots) = start_system(F), maxsteps=10000)
|
|
# F=homogenize(F)
|
|
H=homotopy(F,G)
|
|
solutions = []
|
|
|
|
for r in roots
|
|
t = 1.0
|
|
step_size = 0.01
|
|
x0 = r
|
|
m = 0
|
|
|
|
while t > 0 && maxsteps > 0
|
|
x = en_step(H, x0, t, step_size)
|
|
(m, step_size) = adapt_step(x, x0, step_size, m)
|
|
x0 = x
|
|
t -= step_size
|
|
maxsteps -= 1
|
|
end
|
|
push!(solutions, x0)
|
|
end
|
|
|
|
return solutions
|
|
end
|
|
|
|
# Input polynomial system
|
|
@polyvar x y
|
|
F = [x*y - 1, x^2 + y^2 - 4]
|
|
T = [x*y - 1, x^2 + y^2 - 2]
|
|
C = [x^3 - y + 5x^2 - 10, 2x^2 - y - 10]
|
|
P = [x*y - 1, x*y]
|
|
|
|
sF = filter(u -> imag(u[1]) < 0.1 && imag(u[2]) < 0.1, solve(F))
|
|
sT = filter(u -> imag(u[1]) < 0.1 && imag(u[2]) < 0.1, solve(T))
|
|
sC = filter(u -> imag(u[1]) < 0.1 && imag(u[2]) < 0.1, solve(C))
|
|
# sP = filter(u -> imag(u[1]) < 0.1 && imag(u[2]) < 0.1, solve(P))
|
|
|
|
# Plotting the system and the real solutions
|
|
plot_real(sF, F, 4, 4, "1")
|
|
plot_real(sT, T, 4, 4, "2")
|
|
plot_real(sC, C, 6, 12, "3")
|
|
# plot_real(sP, P, 5, 5, "4")
|