always-cautious/ortools-clone
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement official output for opb problems

Browse Source
This commit is contained in:
Laurent Perron
2025-04-07 15:52:14 +02:00
parent 886634f3c4
commit 0f20448ade
3 changed files with 102 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ortools/sat/BUILD.bazel
View File

@@ -3903,7 +3903,7 @@ cc_binary(
"//ortools/base:file",
"//ortools/base:path",
"//ortools/util:file_util",
"//ortools/util:filelineiter",
"//ortools/util:logging",
"//ortools/util:sorted_interval_list",
"@abseil-cpp//absl/flags:flag",
"@abseil-cpp//absl/log",

1
ortools/sat/cp_model_solver_test.cc
View File

@@ -133,6 +133,7 @@ TEST(RelativeGapLimitTest, BooleanLinearOptimizationProblem) {
Model model;
SatParameters params;
params.set_num_workers(1);
params.set_relative_gap_limit(1e10); // Should stop at the first solution!
int num_solutions = 0;

103
ortools/sat/sat_runner.cc
View File

@@ -11,8 +11,12 @@
// See the License for the specific language governing permissions and
// limitations under the License.
#include <cstdint>
#include <cstdlib>
#include <functional>
#include <iostream>
#include <string>
#include <vector>
#include "absl/flags/flag.h"
#include "absl/flags/parse.h"
@@ -23,6 +27,7 @@
#include "absl/log/log.h"
#include "absl/strings/match.h"
#include "absl/strings/str_format.h"
#include "absl/strings/str_split.h"
#include "absl/strings/string_view.h"
#include "google/protobuf/arena.h"
#include "google/protobuf/text_format.h"
@@ -37,6 +42,7 @@
#include "ortools/sat/sat_cnf_reader.h"
#include "ortools/sat/sat_parameters.pb.h"
#include "ortools/util/file_util.h"
#include "ortools/util/logging.h"
#include "ortools/util/sorted_interval_list.h"
ABSL_FLAG(
@@ -67,11 +73,20 @@ ABSL_FLAG(bool, wcnf_use_strong_slack, true,
"enforce the fact that when it is true, the clause must be false.");
ABSL_FLAG(bool, fingerprint_intermediate_solutions, false,
"Attach the fingerprint of intermediate solutions to the output.");
ABSL_FLAG(bool, competition_mode, false,
"If true, output the log in a competition format.");
namespace operations_research {
namespace sat {
namespace {
struct CompetitionLog {
std::function<void(const std::string&)> log_callback = nullptr;
std::function<void(const CpSolverResponse&)> response_callback = nullptr;
std::function<void(const CpSolverResponse&)> final_response_callback =
nullptr;
};
void TryToRemoveSuffix(absl::string_view suffix, std::string* str) {
if (file::Extension(*str) == suffix) *str = file::Stem(*str);
}
@@ -91,13 +106,79 @@ std::string ExtractName(absl::string_view full_filename) {
}
bool LoadProblem(const std::string& filename, absl::string_view hint_file,
absl::string_view domain_file, CpModelProto* cp_model) {
absl::string_view domain_file, CpModelProto* cp_model,
CompetitionLog* competition_log) {
if (absl::EndsWith(filename, ".opb") ||
absl::EndsWith(filename, ".opb.bz2") ||
absl::EndsWith(filename, ".opb.gz")) {
OpbReader reader;
if (!reader.LoadAndValidate(filename, cp_model)) {
LOG(FATAL) << "Cannot load file '" << filename << "'.";
if (absl::GetFlag(FLAGS_competition_mode)) {
std::cout << "s UNSUPPORTED" << std::endl;
} else {
LOG(FATAL) << "Cannot load file '" << filename << "'.";
}
}
if (absl::GetFlag(FLAGS_competition_mode)) {
competition_log->log_callback = [](const std::string& multi_line_input) {
if (multi_line_input.empty()) {
std::cout << "c" << std::endl;
return;
}
const std::vector<absl::string_view> lines =
absl::StrSplit(multi_line_input, '\n');
for (const absl::string_view& line : lines) {
std::cout << "c " << line << std::endl;
}
};
competition_log->response_callback = [](const CpSolverResponse& r) {
std::cout << "o " << static_cast<int64_t>(r.objective_value())
<< std::endl;
};
const int num_variables = reader.num_variables();
const bool has_objective = cp_model->has_objective();
competition_log->final_response_callback =
[num_variables, has_objective](const CpSolverResponse& r) {
switch (r.status()) {
case CpSolverStatus::OPTIMAL:
if (has_objective) {
std::cout << "s OPTIMUM FOUND " << std::endl;
} else {
std::cout << "s SATISFIABLE" << std::endl;
}
break;
case CpSolverStatus::FEASIBLE:
std::cout << "s SATISFIABLE" << std::endl;
break;
case CpSolverStatus::INFEASIBLE:
std::cout << "s UNSATISFIABLE" << std::endl;
break;
case CpSolverStatus::MODEL_INVALID:
std::cout << "s UNSUPPORTED" << std::endl;
break;
case CpSolverStatus::UNKNOWN:
std::cout << "s UNKNOWN" << std::endl;
break;
default:
break;
}
std::string line;
for (int i = 0; i < num_variables; ++i) {
if (r.solution(i)) {
absl::StrAppend(&line, "x", i + 1, " ");
} else {
absl::StrAppend(&line, "-x", i + 1, " ");
}
if (line.size() >= 75) {
std::cout << "v " << line << std::endl;
line.clear();
}
}
if (!line.empty()) {
std::cout << "v " << line << std::endl;
}
};
}
} else if (absl::EndsWith(filename, ".cnf") ||
absl::EndsWith(filename, ".cnf.xz") ||
@@ -181,15 +262,28 @@ int Run() {
google::protobuf::Arena arena;
CpModelProto* cp_model =
google::protobuf::Arena::Create<CpModelProto>(&arena);
CompetitionLog competition_log;
if (!LoadProblem(absl::GetFlag(FLAGS_input), absl::GetFlag(FLAGS_hint_file),
absl::GetFlag(FLAGS_domain_file), cp_model)) {
absl::GetFlag(FLAGS_domain_file), cp_model,
&competition_log)) {
CpSolverResponse response;
response.set_status(CpSolverStatus::MODEL_INVALID);
if (competition_log.final_response_callback != nullptr) {
competition_log.final_response_callback(response);
}
return EXIT_SUCCESS;
}
Model model;
model.Add(NewSatParameters(parameters));
if (competition_log.log_callback != nullptr) {
model.GetOrCreate<SolverLogger>()->AddInfoLoggingCallback(
competition_log.log_callback);
model.GetOrCreate<SatParameters>()->set_log_to_stdout(false);
}
if (competition_log.response_callback != nullptr) {
model.Add(NewFeasibleSolutionObserver(competition_log.response_callback));
}
if (absl::GetFlag(FLAGS_fingerprint_intermediate_solutions)) {
// Let's add a solution callback that will display the fingerprint of all
// solutions.
@@ -200,6 +294,9 @@ int Run() {
}));
}
const CpSolverResponse response = SolveCpModel(*cp_model, &model);
if (competition_log.final_response_callback != nullptr) {
competition_log.final_response_callback(response);
}
if (!absl::GetFlag(FLAGS_output).empty()) {
if (absl::EndsWith(absl::GetFlag(FLAGS_output), "txt")) {