init.h

Go to the documentation of this file.

// Copyright 2010-2022 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 <cstdint>
#include <string>
#include <vector>
 
#include "absl/flags/flag.h"
#include "absl/flags/usage.h"
#include "absl/log/globals.h"
#include "absl/log/initialize.h"
#include "ortools/base/logging.h"
#include "ortools/base/version.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);
ABSL_DECLARE_FLAG(int, stderrthreshold);
 
namespace operations_research {
 
struct CppFlags {
  int stderrthreshold = 2;
 
  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& usage) {
    absl::SetProgramUsageMessage(usage);
    absl::InitializeLog();
  }
 
  static void ShutdownLogging() {}
 
  static void SetFlags(const CppFlags& flags) {
    absl::SetFlag(&FLAGS_stderrthreshold, flags.stderrthreshold);
    absl::EnableLogPrefix(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();
  }
 
  static void DeleteByteArray(uint8_t* buffer) { delete[] buffer; }
};
 
class OrToolsVersion {
 public:
  static int MajorNumber() {
    return ::operations_research::OrToolsMajorVersion();
  }
 
  static int MinorNumber() {
    return ::operations_research::OrToolsMinorVersion();
  }
 
  static int PatchNumber() {
    return ::operations_research::OrToolsPatchVersion();
  }
 
  static std::string VersionString() {
    return ::operations_research::OrToolsVersionString();
  }
};
 
}  // namespace operations_research
 
#endif  // OR_TOOLS_OPEN_SOURCE_INIT_INIT_H_