#include <stdio.h>
#include <cstdlib>
#include <string>
#include <vector>
#include "absl/base/log_severity.h"
#include "absl/flags/flag.h"
#include "absl/log/check.h"
#include "absl/log/globals.h"
#include "absl/log/log.h"
#include "absl/status/status.h"
#include "absl/strings/match.h"
#include "absl/strings/str_format.h"
#include "google/protobuf/text_format.h"
#include "ortools/base/helpers.h"
#include "ortools/base/init_google.h"
#include "ortools/base/options.h"
#include "ortools/base/timer.h"
#include "ortools/glop/lp_solver.h"
#include "ortools/glop/parameters.pb.h"
#include "ortools/lp_data/lp_data.h"
#include "ortools/lp_data/lp_types.h"
#include "ortools/lp_data/mps_reader.h"
#include "ortools/lp_data/proto_utils.h"
#include "ortools/util/file_util.h"
#include "ortools/util/proto_tools.h"
ABSL_FLAG(bool, mps_dump_problem, false, "Dumps problem in readable form.");
ABSL_FLAG(bool, mps_solve, true, "Solves problem.");
ABSL_FLAG(bool, mps_terse_result, false,
"Displays the result in form of a single CSV line.");
ABSL_FLAG(bool, mps_verbose_result, true,
"Displays the result in verbose form.");
ABSL_FLAG(bool, mps_display_full_path, true,
"Displays the full path of the input file in the result line.");
ABSL_FLAG(std::string, input, "", "File pattern for problems to be optimized.");
ABSL_FLAG(std::string, params_file, "",
"Path to a GlopParameters file in text format.");
ABSL_FLAG(std::string, params, "",
"GlopParameters in text format. If --params_file was "
"also specified, the --params will be merged onto "
"them (i.e. in case of conflicts, --params wins)");
using google::protobuf::TextFormat;
using operations_research::FullProtocolMessageAsString;
using operations_research::ReadFileToProto;
using operations_research::glop::GetProblemStatusString;
using operations_research::glop::GlopParameters;
using operations_research::glop::LinearProgram;
using operations_research::glop::LPSolver;
using operations_research::glop::MPModelProtoToLinearProgram;
using operations_research::glop::MPSReader;
using operations_research::glop::ProblemStatus;
using operations_research::glop::ToDouble;
void ReadGlopParameters(GlopParameters* parameters) {
if (!absl::GetFlag(FLAGS_params_file).empty()) {
std::string params;
CHECK_OK(file::GetContents(absl::GetFlag(FLAGS_params_file), ¶ms,
file::Defaults()));
CHECK(TextFormat::ParseFromString(params, parameters)) << params;
}
if (!absl::GetFlag(FLAGS_params).empty()) {
CHECK(TextFormat::MergeFromString(absl::GetFlag(FLAGS_params), parameters))
<< absl::GetFlag(FLAGS_params);
}
if (absl::GetFlag(FLAGS_mps_verbose_result)) {
absl::PrintF("GlopParameters {\n%s}\n",
FullProtocolMessageAsString(*parameters, 1));
}
}
int main(int argc, char* argv[]) {
InitGoogle(
"Runs Glop on a given pattern of files given by --input. "
"The files must be in Mps or linear_solver.proto format and can be "
"compressed with gzip.",
&argc, &argv, true);
absl::SetStderrThreshold(absl::LogSeverityAtLeast::kInfo);
GlopParameters parameters;
ReadGlopParameters(¶meters);
LinearProgram linear_program;
std::vector<std::string> file_list;
file_list.push_back(absl::GetFlag(FLAGS_input));
for (int i = 0; i < file_list.size(); ++i) {
const std::string& file_name = file_list[i];
MPSReader mps_reader;
operations_research::MPModelProto model_proto;
if (absl::EndsWith(file_name, ".mps") ||
absl::EndsWith(file_name, ".mps.gz")) {
const absl::Status parse_status =
mps_reader.ParseFile(file_name, &linear_program);
if (!parse_status.ok()) {
LOG(INFO) << "Parse error for " << file_name << ": " << parse_status;
continue;
}
} else {
const absl::Status status = ReadFileToProto(file_name, &model_proto);
if (!status.ok()) {
LOG(INFO) << status;
continue;
}
MPModelProtoToLinearProgram(model_proto, &linear_program);
}
if (absl::GetFlag(FLAGS_mps_dump_problem)) {
absl::PrintF("%s", linear_program.Dump());
}
LPSolver solver;
solver.SetParameters(parameters);
ProblemStatus solve_status = ProblemStatus::INIT;
std::string status_string;
double objective_value;
double solving_time_in_sec = 0;
if (absl::GetFlag(FLAGS_mps_solve)) {
ScopedWallTime timer(&solving_time_in_sec);
solve_status = solver.Solve(linear_program);
status_string = GetProblemStatusString(solve_status);
objective_value = ToDouble(solver.GetObjectiveValue());
}
if (absl::GetFlag(FLAGS_mps_terse_result)) {
if (absl::GetFlag(FLAGS_mps_display_full_path)) {
absl::PrintF("%s,", file_name);
}
absl::PrintF("%s,", linear_program.name());
if (absl::GetFlag(FLAGS_mps_solve)) {
absl::PrintF("%15.15e,%s,%-6.4g,", objective_value, status_string,
solving_time_in_sec);
}
absl::PrintF("%s,%s\n", linear_program.GetProblemStats(),
linear_program.GetNonZeroStats());
}
if (absl::GetFlag(FLAGS_mps_verbose_result)) {
if (absl::GetFlag(FLAGS_mps_display_full_path)) {
absl::PrintF("%-45s: %s\n", "File path", file_name);
}
absl::PrintF("%-45s: %s\n", "Problem name", linear_program.name());
if (absl::GetFlag(FLAGS_mps_solve)) {
absl::PrintF("%-45s: %15.15e\n", "Objective value", objective_value);
absl::PrintF("%-45s: %s\n", "Problem status", status_string);
absl::PrintF("%-45s: %-6.4g\n", "Solving time", solving_time_in_sec);
}
absl::PrintF("%s%s", linear_program.GetPrettyProblemStats(),
linear_program.GetPrettyNonZeroStats());
}
}
return EXIT_SUCCESS;
}