29#include "absl/flags/flag.h"
30#include "absl/flags/parse.h"
31#include "absl/flags/usage.h"
32#include "absl/log/check.h"
33#include "absl/log/flags.h"
34#include "absl/log/initialize.h"
35#include "absl/log/log.h"
36#include "absl/strings/match.h"
37#include "absl/strings/str_split.h"
38#include "absl/strings/string_view.h"
39#include "absl/time/time.h"
51ABSL_FLAG(
bool, search_all_solutions,
false,
"Search for all solutions.");
53 "Display all improving solutions.");
55 "If false, the solver must follow the defined search."
56 "If true, other search are allowed.");
57ABSL_FLAG(
int, threads, 0,
"Number of threads the solver will use.");
58ABSL_FLAG(
bool, statistics,
false,
"Print solver statistics after search.");
60 "Read the FlatZinc from stdin, not from a file.");
63 "Define problem name when reading from stdin.");
64ABSL_FLAG(std::string, params,
"",
"SatParameters as a text proto.");
66 "Print logging information from the flatzinc interpreter.");
68 "Display solutions in the flatzinc format");
74 char all_param[] =
"--search_all_solutions";
75 char print_solutions[] =
"--display_all_solutions";
76 char free_param[] =
"--free_search";
77 char threads_param[] =
"--threads";
78 char logging_param[] =
"--fz_logging";
79 char statistics_param[] =
"--statistics";
80 char seed_param[] =
"--fz_seed";
81 char time_param[] =
"--time_limit";
82 bool use_time_param =
false;
83 bool set_free_search =
false;
84 for (
int i = 1; i < *argc; ++i) {
85 if (strcmp((*argv)[i],
"-a") == 0) {
86 (*argv)[i] = all_param;
88 if (strcmp((*argv)[i],
"-i") == 0) {
89 (*argv)[i] = print_solutions;
91 if (strcmp((*argv)[i],
"-f") == 0) {
92 (*argv)[i] = free_param;
93 set_free_search =
true;
95 if (strcmp((*argv)[i],
"-p") == 0) {
96 (*argv)[i] = threads_param;
98 if (strcmp((*argv)[i],
"-l") == 0) {
99 (*argv)[i] = logging_param;
101 if (strcmp((*argv)[i],
"-s") == 0) {
102 (*argv)[i] = statistics_param;
104 if (strcmp((*argv)[i],
"-r") == 0) {
105 (*argv)[i] = seed_param;
107 if (strcmp((*argv)[i],
"-t") == 0) {
108 (*argv)[i] = time_param;
109 use_time_param =
true;
112 if (strcmp((*argv)[i],
"-v") == 0) {
113 (*argv)[i] = logging_param;
118 "Usage: see flags.\nThis program parses and solve a flatzinc problem.";
120 absl::SetProgramUsageMessage(
kUsage);
121 const std::vector<char*> residual_flags =
122 absl::ParseCommandLine(*argc, *argv);
123 absl::InitializeLog();
126 if (use_time_param) {
127 absl::SetFlag(&FLAGS_time_limit, absl::GetFlag(FLAGS_time_limit) / 1000.0);
131 if (set_free_search && absl::GetFlag(FLAGS_threads) == 0) {
132 absl::SetFlag(&FLAGS_threads, 1);
135 return residual_flags;
144 if (input_is_filename && !absl::EndsWith(
input,
".fzn")) {
145 LOG(FATAL) <<
"Unrecognized flatzinc file: `" <<
input <<
"'";
149 const std::string problem_name = input_is_filename
151 : absl::GetFlag(FLAGS_fz_model_name);
152 Model model(problem_name);
153 if (input_is_filename) {
161 " parsed in ", absl::ToInt64Milliseconds(*parse_duration),
" ms");
172 if (multi_line_input.empty()) {
173 std::cout << std::endl;
176 const absl::string_view flatzinc_prefix =
177 absl::GetFlag(FLAGS_ortools_mode) ?
"%% " :
"";
178 const std::vector<absl::string_view> lines =
179 absl::StrSplit(multi_line_input,
'\n');
180 for (
const absl::string_view& line : lines) {
181 std::cout << flatzinc_prefix << line << std::endl;
188int main(
int argc,
char** argv) {
191 const std::vector<char*> residual_flags =
195 if (absl::GetFlag(FLAGS_read_from_stdin)) {
196 std::string currentLine;
197 while (std::getline(std::cin, currentLine)) {
198 input.append(currentLine);
202 if (residual_flags.empty()) {
203 LOG(ERROR) <<
"Usage: " << argv[0] <<
" <file>";
206 input = residual_flags.back();
210 if (absl::GetFlag(FLAGS_ortools_mode)) {
219 absl::Duration parse_duration;
222 input, !absl::GetFlag(FLAGS_read_from_stdin), &logger,
231 absl::GetFlag(FLAGS_fz_logging) || !absl::GetFlag(FLAGS_ortools_mode);
232 parameters.
random_seed = absl::GetFlag(FLAGS_fz_seed);
236 absl::GetFlag(FLAGS_time_limit) - absl::ToInt64Seconds(parse_duration);
237 parameters.
ortools_mode = absl::GetFlag(FLAGS_ortools_mode);
243 if (absl::GetFlag(FLAGS_time_limit) > 0 &&
244 parse_duration > absl::Seconds(absl::GetFlag(FLAGS_time_limit))) {
249 SOLVER_LOG(&logger,
"CpSolverResponse summary:");
256 absl::GetFlag(FLAGS_params),
257 &logger, &solution_logger);
absl::Duration GetDuration() const
void Start()
When Start() is called multiple times, only the most recent is used.
void EnableLogging(bool enable)
void SetLogToStdOut(bool enable)
Should all messages be displayed on stdout ?
void AddInfoLoggingCallback(std::function< void(const std::string &message)> callback)
void PrintStatistics() const
--— Model statistics --—
ABSL_FLAG(double, time_limit, 0, "time limit in seconds.")
int main(int argc, char **argv)
constexpr bool kOrToolsMode
absl::string_view Stem(absl::string_view path)
void LogInFlatzincFormat(const std::string &multi_line_input)
Model ParseFlatzincModel(const std::string &input, bool input_is_filename, SolverLogger *logger, absl::Duration *parse_duration)
bool ParseFlatzincString(const std::string &input, Model *model)
bool ParseFlatzincFile(const std::string &filename, Model *model)
--— public parsing API --—
std::vector< char * > FixAndParseParameters(int *argc, char ***argv)
void SolveFzWithCpModelProto(const fz::Model &fz_model, const fz::FlatzincSatParameters &p, const std::string &sat_params, SolverLogger *logger, SolverLogger *solution_logger)
Solves the given flatzinc model using the CP-SAT solver.
void ProcessFloatingPointOVariablesAndObjective(fz::Model *fz_model)
In SWIG mode, we don't want anything besides these top-level includes.
static int input(yyscan_t yyscanner)
static const char kUsage[]
bool search_all_solutions
double max_time_in_seconds
bool display_all_solutions
#define SOLVER_LOG(logger,...)