Files
ortools-clone/ortools/linear_solver/csharp/LinearConstraint.cs

169 lines
4.6 KiB
C#
Raw Permalink Normal View History

2025-01-10 11:35:44 +01:00
// Copyright 2010-2025 Google LLC
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
2020-11-03 10:04:19 +01:00
namespace Google.OrTools.LinearSolver
{
2022-01-03 09:47:51 +01:00
using System;
using System.Collections.Generic;
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public class LinearConstraint
{
public virtual String ToString()
2020-11-03 10:04:19 +01:00
{
2022-01-03 09:47:51 +01:00
return "LinearConstraint";
2020-11-03 10:04:19 +01:00
}
2022-01-03 09:47:51 +01:00
public virtual Constraint Extract(Solver solver)
2020-11-03 10:04:19 +01:00
{
2022-01-03 09:47:51 +01:00
return null;
}
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public class RangeConstraint : LinearConstraint
{
public RangeConstraint(LinearExpr expr, double lb, double ub)
{
this.expr_ = expr;
this.lb_ = lb;
this.ub_ = ub;
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public override String ToString()
{
return "" + lb_ + " <= " + expr_.ToString() + " <= " + ub_;
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public override Constraint Extract(Solver solver)
{
Dictionary<Variable, double> coefficients = new Dictionary<Variable, double>();
double constant = expr_.Visit(coefficients);
Constraint ct = solver.MakeConstraint(lb_ - constant, ub_ - constant);
foreach (KeyValuePair<Variable, double> pair in coefficients)
2020-11-03 10:04:19 +01:00
{
2022-01-03 09:47:51 +01:00
ct.SetCoefficient(pair.Key, pair.Value);
2020-11-03 10:04:19 +01:00
}
2022-01-03 09:47:51 +01:00
return ct;
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public static implicit operator bool(RangeConstraint ct)
{
return false;
2020-11-03 10:04:19 +01:00
}
2022-01-03 09:47:51 +01:00
private LinearExpr expr_;
private double lb_;
private double ub_;
}
public class Equality : LinearConstraint
{
public Equality(LinearExpr left, LinearExpr right, bool equality)
2020-11-03 10:04:19 +01:00
{
2022-01-03 09:47:51 +01:00
this.left_ = left;
this.right_ = right;
this.equality_ = equality;
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public override String ToString()
{
return "" + left_.ToString() + " == " + right_.ToString();
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public override Constraint Extract(Solver solver)
{
if (!equality_)
2020-11-03 10:04:19 +01:00
{
2022-01-03 09:47:51 +01:00
throw new ArgumentException("Operator != not supported for LinearExpression");
2020-11-03 10:04:19 +01:00
}
2022-01-03 09:47:51 +01:00
Dictionary<Variable, double> coefficients = new Dictionary<Variable, double>();
double constant = left_.Visit(coefficients);
constant += right_.DoVisit(coefficients, -1);
Constraint ct = solver.MakeConstraint(-constant, -constant);
foreach (KeyValuePair<Variable, double> pair in coefficients)
2020-11-03 10:04:19 +01:00
{
2022-01-03 09:47:51 +01:00
ct.SetCoefficient(pair.Key, pair.Value);
2020-11-03 10:04:19 +01:00
}
2022-01-03 09:47:51 +01:00
return ct;
2020-11-03 10:04:19 +01:00
}
2022-01-03 09:47:51 +01:00
public static implicit operator bool(Equality ct)
2020-11-03 10:04:19 +01:00
{
2022-01-03 09:47:51 +01:00
return (object)ct.left_ == (object)ct.right_ ? ct.equality_ : !ct.equality_;
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
private LinearExpr left_;
private LinearExpr right_;
private bool equality_;
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public class VarEquality : LinearConstraint
{
public VarEquality(Variable left, Variable right, bool equality)
{
this.left_ = left;
this.right_ = right;
this.equality_ = equality;
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public override String ToString()
{
return "" + left_.Name() + " == " + right_.Name();
}
public override Constraint Extract(Solver solver)
{
Constraint ct = solver.MakeConstraint(0.0, 0.0);
ct.SetCoefficient(left_, 1.0);
ct.SetCoefficient(right_, -1.0);
return ct;
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
public static implicit operator bool(VarEquality ct)
{
return (object)ct.left_ == (object)ct.right_ ? ct.equality_ : !ct.equality_;
2020-11-03 10:04:19 +01:00
}
2022-01-03 09:47:51 +01:00
private Variable left_;
private Variable right_;
private bool equality_;
}
2022-01-21 17:51:38 +01:00
// TODO(user): Try to move this code back to the .i with @define macros.
2022-01-03 09:47:51 +01:00
public partial class MPConstraintVector : IDisposable,
System.Collections.IEnumerable
#if !SWIG_DOTNET_1
2022-01-03 09:47:51 +01:00
,
System.Collections.Generic.IList<Constraint>
#endif
2022-01-03 09:47:51 +01:00
{
// cast from C# MPConstraint array
public static implicit operator MPConstraintVector(Constraint[] inVal)
2020-11-03 10:04:19 +01:00
{
2022-01-03 09:47:51 +01:00
var outVal = new MPConstraintVector();
foreach (Constraint element in inVal)
2020-11-03 10:04:19 +01:00
{
2022-01-03 09:47:51 +01:00
outVal.Add(element);
2020-11-03 10:04:19 +01:00
}
2022-01-03 09:47:51 +01:00
return outVal;
}
2020-11-03 10:04:19 +01:00
2022-01-03 09:47:51 +01:00
// cast to C# MPConstraint array
public static implicit operator Constraint[](MPConstraintVector inVal)
{
var outVal = new Constraint[inVal.Count];
inVal.CopyTo(outVal);
return outVal;
2020-11-03 10:04:19 +01:00
}
2022-01-03 09:47:51 +01:00
}
2020-11-03 10:04:19 +01:00
} // namespace Google.OrTools.LinearSolver