Add WriteModelToMpsFile to MPSolver in Java/Python/.NET

google · Oct 7, 2024 · 5912937 · 5912937
1 parent ae517c0
commit 5912937
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/ortools/linear_solver/csharp/linear_solver.i b/ortools/linear_solver/csharp/linear_solver.i
@@ -166,6 +166,8 @@ CONVERT_VECTOR(operations_research::MPVariable, MPVariable)
 // Extend code.
 %unignore operations_research::MPSolver::ExportModelAsLpFormat(bool);
 %unignore operations_research::MPSolver::ExportModelAsMpsFormat(bool, bool);
+%unignore operations_research::MPSolver::WriteModelToMpsFile(
+  const std::string& filename, bool, bool);
 %unignore operations_research::MPSolver::SetHint(
     const std::vector<operations_research::MPVariable*>&,
     const std::vector<double>&);
@@ -187,6 +189,15 @@ CONVERT_VECTOR(operations_research::MPVariable, MPVariable)
     return ExportModelAsMpsFormat(model, options).value_or("");
   }
 
+  bool WriteModelToMpsFile(const std::string& filename, bool fixed_format,
+                           bool obfuscated) {
+    operations_research::MPModelExportOptions options;
+    options.obfuscate = obfuscated;
+    operations_research::MPModelProto model;
+    $self->ExportModelToProto(&model);
+    return WriteModelToMpsFile(filename, model, options).ok();
+  }
+
   void SetHint(const std::vector<operations_research::MPVariable*>& variables,
                const std::vector<double>& values) {
     if (variables.size() != values.size()) {

diff --git a/ortools/linear_solver/java/linear_solver.i b/ortools/linear_solver/java/linear_solver.i
@@ -207,6 +207,18 @@ PROTO2_RETURN(
     return ExportModelAsMpsFormat(model, options).value_or("");
   }
 
+  /**
+   * Write the model to file in MPS format.
+   */
+   bool writeModelToMpsFile(const std::string& filename, bool fixed_format,
+                            bool obfuscated) {
+    operations_research::MPModelExportOptions options;
+    options.obfuscate = obfuscated;
+    operations_research::MPModelProto model;
+    $self->ExportModelToProto(&model);
+    return WriteModelToMpsFile(filename, model, options).ok();
+  }
+
   /**
    * Sets a hint for solution.
    *

diff --git a/ortools/linear_solver/python/linear_solver.i b/ortools/linear_solver/python/linear_solver.i
@@ -143,6 +143,15 @@ from ortools.linear_solver.python.linear_solver_natural_api import VariableExpr
     return ExportModelAsMpsFormat(model, options).value_or("");
   }
 
+   bool WriteModelToMpsFile(const std::string& filename, bool fixed_format,
+                            bool obfuscated) {
+    operations_research::MPModelExportOptions options;
+    options.obfuscate = obfuscated;
+    operations_research::MPModelProto model;
+    $self->ExportModelToProto(&model);
+    return WriteModelToMpsFile(filename, model, options).ok();
+  }
+
   /// Set a hint for solution.
   ///
   /// If a feasible or almost-feasible solution to the problem is already known,