# Modify a basic optimization problem.
#
# Construct a problem using addVariable and addConstraint, then use
# the Xpress API routines to amend the problem with rows and quadratic
# terms.
#
# (C) 1983-2025 Fair Isaac Corporation

import xpress as xp

p = xp.problem()

N = 5
S = range(N)

x = [p.addVariable(vartype=xp.binary) for i in S]

# Vectors can be used whole or addressed with their index.

c0 = xp.Sum(x) <= 10
cc = [x[i]/1.1 <= x[i+1]*2 for i in range(N-1)]

p.addConstraint(c0, cc)

p.setObjective(3 - x[0])

mysol = [0, 0, 1, 1, 1, 1.4]

# Add a binary variable with its coefficients.
p.addCols([4], [0, 3], [c0, 4, 2], [-3, 2.4, 1.4], [0], [2])
p.addNames(xp.Namespaces.COLUMN, ['YY'], 5, 5)
p.chgColType([5], ['B'])

p.writeProb("problem1", "lp")

# Load a MIP solution.
p.loadMipSol([0, 0, 1, 1, 1, 1.4])

# Add quadratic term x[0]^2 - 2 x[0] x[1] + x[1]^2 to the second
# constraint. Note that the -2 coefficient for an off-diagonal element
# must be passed divided by two.
#
# The same effect would be obtained with p.addqmatrix (cc[0],
# [x[0],x[3],x[3]], [x[0],x[0],x[3]], [1,-1,1]).
#
# As constraint vector 'cc' was added after c0.

p.addQMatrix(1, [x[0], x[3], x[3]], [x[0], x[0], x[3]], [1, -1, 1])

# Add seventh and eighth constraints:
#
#   x[0] + 2 x[1] + 3 x[2] >= 4
# 4 x[0] + 5 x[1] + 6 x[2] + 7 x[3] + 8 x[4] - 3 YY <= 4.4
#
# Note the new column named 'YY' added with its index 5 (variables'
# indices begin at 0). The same would happen if 5 were substituted by
# YY.

p.addRows(rowtype=['G', 'L'],
          rhs=[4, 4.4],
          start=[0, 3, 9],
          colind=[x[0], x[1], x[2], x[0], x[1], x[2], x[3], x[4], 5],
          rowcoef=[1, 2, 3, 4, 5, 6, 7, 8, -3])

p.optimize()
p.writeProb("amended", "lp")

slacks = p.calcSlacks(mysol)

print("Slacks:", slacks)

# The five commands below are equivalent to the following:
#
# p.addCols([4, 4, 4, 4, 4],
#            [0, 3, 6, 9, 12, 15],
#            [c0, 4, 2, c0, 4, 2, c0, 4, 2, c0, 4, 2, c0, 4, 2],
#            [3,  -2,  1,  -3,  2.4,  1.4, 3, 2, 1, -3, 2.4, 4, 3, 2, 1],
#            [0,0,0,0,0],
#            [2,10,1,2,2])

p.addCols([4], [0, 3], [c0, 4, 2], [-3, -2, 1], [0], [2])
p.addCols([4], [0, 3], [c0, 4, 2], [-3, 2.4, 1.4], [0], [10])
p.addCols([4], [0, 3], [c0, 4, 2], [-3, 2, 1], [0], [1])
p.addCols([4], [0, 3], [c0, 4, 2], [-3, 2.4, 4], [0], [2])
p.addCols([4], [0, 3], [c0, 4, 2], [-3, 2, 1], [0], [2])

p.chgColType(range(6, 11), ['I','C','S','P','R'])

p.optimize()

if p.attributes.solstatus in [xp.SolStatus.OPTIMAL, xp.SolStatus.FEASIBLE]:
    print("New solution:", p.getSolution())
else:
    print("Solve status: ", p.attributes.solvestatus.name)
    print("Solution status: ", p.attributes.solstatus.name)