init.h

Go to the documentation of this file.

// Copyright 2010-2021 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.
 
#ifndef OR_TOOLS_OPEN_SOURCE_INIT_INIT_H_
#define OR_TOOLS_OPEN_SOURCE_INIT_INIT_H_
 
#include <string>
#include <vector>
 
#include "absl/flags/flag.h"
#include "ortools/base/logging.h"
#include "ortools/gurobi/environment.h"
#include "ortools/sat/cp_model_solver.h"
 
ABSL_DECLARE_FLAG(std::string, cp_model_dump_prefix);
ABSL_DECLARE_FLAG(bool, cp_model_dump_models);
ABSL_DECLARE_FLAG(bool, cp_model_dump_lns);
ABSL_DECLARE_FLAG(bool, cp_model_dump_response);
 
namespace operations_research {
 
struct CppFlags {
  bool logtostderr = false;
 
  bool log_prefix = false;
 
  std::string cp_model_dump_prefix;
  bool cp_model_dump_models = false;
 
  bool cp_model_dump_lns;
 
  bool cp_model_dump_response;
};
 
class CppBridge {
 public:
  static void InitLogging(const std::string& program_name) {
    google::InitGoogleLogging(program_name.c_str());
  }
 
  static void ShutdownLogging() { google::ShutdownGoogleLogging(); }
 
  static void SetFlags(const CppFlags& flags) {
    absl::SetFlag(&FLAGS_logtostderr, flags.logtostderr);
    absl::SetFlag(&FLAGS_log_prefix, flags.log_prefix);
    if (!flags.cp_model_dump_prefix.empty()) {
      absl::SetFlag(&FLAGS_cp_model_dump_prefix, flags.cp_model_dump_prefix);
    }
    absl::SetFlag(&FLAGS_cp_model_dump_models, flags.cp_model_dump_models);
    absl::SetFlag(&FLAGS_cp_model_dump_lns, flags.cp_model_dump_lns);
    absl::SetFlag(&FLAGS_cp_model_dump_response, flags.cp_model_dump_response);
  }
 
  static bool LoadGurobiSharedLibrary(const std::string& full_library_path) {
    return LoadGurobiDynamicLibrary({full_library_path}).ok();
  }
};
 
}  // namespace operations_research
 
#endif  // OR_TOOLS_OPEN_SOURCE_INIT_INIT_H_