mirror of
https://github.com/kalmarek/PropertyT.jl.git
synced 2024-09-07 07:11:42 +02:00
86 lines
2.6 KiB
Julia
86 lines
2.6 KiB
Julia
|
using Dates
|
||
|
using Serialization
|
||
|
using Logging
|
||
|
|
||
|
import JuMP
|
||
|
|
||
|
function get_solution(model)
|
||
|
λ = JuMP.value(model[:λ])
|
||
|
Q = real.(sqrt(JuMP.value.(model[:P])))
|
||
|
solution = Dict(:λ => λ, :Q => Q)
|
||
|
return solution
|
||
|
end
|
||
|
|
||
|
function get_solution(model, wd, varP; logdir)
|
||
|
λ = JuMP.value(model[:λ])
|
||
|
|
||
|
Qs = [real.(sqrt(JuMP.value.(P))) for P in varP]
|
||
|
Q = PropertyT.reconstruct(Qs, wd)
|
||
|
solution = Dict(:λ => λ, :Q => Q)
|
||
|
return solution
|
||
|
end
|
||
|
|
||
|
function solve_in_loop(model::JuMP.Model, args...; logdir, optimizer, data)
|
||
|
@info "logging to $logdir"
|
||
|
status = JuMP.UNKNOWN_RESULT_STATUS
|
||
|
warm = try
|
||
|
solution = deserialize(joinpath(logdir, "solution.sjl"))
|
||
|
warm = solution[:warm]
|
||
|
@info "trying to warm-start model with λ=$(solution[:λ])..."
|
||
|
warm
|
||
|
catch
|
||
|
nothing
|
||
|
end
|
||
|
old_lambda = 0.0
|
||
|
while status != JuMP.OPTIMAL
|
||
|
date = now()
|
||
|
log_file = joinpath(logdir, "solver_$date.log")
|
||
|
@info "Current logfile is $log_file."
|
||
|
isdir(dirname(log_file)) || mkpath(dirname(log_file))
|
||
|
|
||
|
λ, flag, certified_λ = let
|
||
|
# logstream = current_logger().logger.stream
|
||
|
# v = @ccall setvbuf(logstream.handle::Ptr{Cvoid}, C_NULL::Ptr{Cvoid}, 1::Cint, 0::Cint)::Cint
|
||
|
# @warn v
|
||
|
status, warm = @time PropertyT.solve(log_file, model, optimizer, warm)
|
||
|
|
||
|
solution = get_solution(model, args...; logdir=logdir)
|
||
|
solution[:warm] = warm
|
||
|
|
||
|
serialize(joinpath(logdir, "solution_$date.sjl"), solution)
|
||
|
serialize(joinpath(logdir, "solution.sjl"), solution)
|
||
|
|
||
|
flag, λ_cert = open(log_file, append=true) do io
|
||
|
with_logger(SimpleLogger(io)) do
|
||
|
PropertyT.certify_solution(
|
||
|
data.elt,
|
||
|
data.unit,
|
||
|
solution[:λ],
|
||
|
solution[:Q],
|
||
|
halfradius=data.halfradius,
|
||
|
)
|
||
|
end
|
||
|
end
|
||
|
|
||
|
solution[:λ], flag, λ_cert
|
||
|
end
|
||
|
|
||
|
if flag == true && certified_λ ≥ 0
|
||
|
@info "Certification done with λ = $certified_λ"
|
||
|
return certified_λ
|
||
|
else
|
||
|
rel_change = abs(certified_λ - old_lambda) / (abs(certified_λ) + abs(old_lambda))
|
||
|
@info "Certification failed with λ = $λ" certified_λ rel_change
|
||
|
end
|
||
|
|
||
|
old_lambda = certified_λ
|
||
|
|
||
|
if rel_change < 1e-9
|
||
|
@info "No progress detected, breaking"
|
||
|
break
|
||
|
end
|
||
|
end
|
||
|
|
||
|
return status == JuMP.OPTIMAL ? old_lambda : NaN
|
||
|
end
|