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_