Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 30 additions & 3 deletions src/functions.jl
Original file line number Diff line number Diff line change
Expand Up @@ -1086,9 +1086,27 @@ end

_add_to_function(::ScalarAffineFunction, ::Any) = nothing

# This is a very rough-and-ready conversion function that only works for very
# basic expressions, such as those created by
# `convert(ScalarNonlinearFunction, f)`.
function _to_scalar_affine(::Type{ScalarAffineFunction{T}}, x::Real) where {T}
return ScalarAffineFunction{T}(ScalarAffineTerm{T}[], T(x))
end

function _to_scalar_affine(
::Type{ScalarAffineFunction{T}},
x::VariableIndex,
) where {T}
return ScalarAffineFunction{T}(x)
end

function _to_scalar_affine(
::Type{ScalarAffineFunction{T}},
f::ScalarNonlinearFunction,
) where {T}
return convert(ScalarAffineFunction{T}, f)
end

# This conversion function handles expressions created by
# `convert(ScalarNonlinearFunction, f)` and also expressions using `:+`, `:-`,
# and `:*` operators with affine structure.
function Base.convert(
::Type{ScalarAffineFunction{T}},
f::ScalarNonlinearFunction,
Expand All @@ -1097,6 +1115,15 @@ function Base.convert(
term = convert(ScalarAffineTerm{T}, f)
return ScalarAffineFunction{T}([term], zero(T))
end
if f.head == :- && length(f.args) == 2
lhs = _to_scalar_affine(ScalarAffineFunction{T}, f.args[1])
rhs = _to_scalar_affine(ScalarAffineFunction{T}, f.args[2])
return Utilities.operate(-, T, lhs, rhs)
end
if f.head == :- && length(f.args) == 1
inner = _to_scalar_affine(ScalarAffineFunction{T}, f.args[1])
return Utilities.operate(-, T, inner)
end
if f.head != :+
throw(InexactError(:convert, ScalarAffineFunction{T}, f))
end
Expand Down
11 changes: 11 additions & 0 deletions test/General/test_functions.jl
Original file line number Diff line number Diff line change
Expand Up @@ -371,6 +371,17 @@ function test_convert_ScalarNonlinearFunction_ScalarAffineFunction()
convert(MOI.ScalarAffineFunction{Float64}, f_error),
)
end
# Test :- support (binary subtraction and unary negation)
f_sub = MOI.ScalarNonlinearFunction(:-, Any[x, 1.0])
@test convert(MOI.ScalarAffineFunction{Float64}, f_sub) ≈ 1.0 * x - 1.0
f_sub2 = MOI.ScalarNonlinearFunction(
:-,
Any[MOI.ScalarNonlinearFunction(:+, Any[x, y]), 2.0],
)
@test convert(MOI.ScalarAffineFunction{Float64}, f_sub2) ≈
1.0 * x + 1.0 * y - 2.0
f_neg = MOI.ScalarNonlinearFunction(:-, Any[x])
@test convert(MOI.ScalarAffineFunction{Float64}, f_neg) ≈ -1.0 * x
return
end

Expand Down
21 changes: 21 additions & 0 deletions test/param.jl
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
import MathOptInterface as MOI

model = MOI.instantiate(
MOI.FileFormats.CBF.Model{Float64},
with_bridge_type = Float64,
)
x = MOI.add_variable(model)
p, _ = MOI.add_constrained_variable(model, MOI.Parameter(1.0))
MOI.add_constraint(model, 1.0x * p, MOI.LessThan(1.0))

model = Model(() -> DiffOpt.diff_optimizer(Ipopt.Optimizer))
set_silent(model)

p_val = 4.0
pc_val = 2.0
@variable(model, x)
@variable(model, p in Parameter(p_val))
@variable(model, pc in Parameter(pc_val))
@constraint(model, cons, pc * x >= 3 * p)
@objective(model, Min, x^4)
optimize!(model)
Loading