envvars.hh

Go to the documentation of this file.

// -*- mode: c++; c-basic-offset: 2; -*-
 
#pragma once
 
#include <string>
 
#define NVSL_DECL_ENV(name) \
  static const char *name##_ENV __attribute__((unused)) = (char *)(#name)
 
NVSL_DECL_ENV(NVSL_NO_STACKTRACE);
NVSL_DECL_ENV(NVSL_LOG_WILDCARD);
NVSL_DECL_ENV(NVSL_GEN_STATS);
 
static inline bool get_env_val(const std::string var) {
  bool result = false;
 
  const char *val = std::getenv(var.c_str());
  if (val != nullptr) {
    const std::string val_str = std::string(val);
 
    if (val_str == "1") {
      result = true;
    }
  }
 
  return result;
}
 
static inline std::string get_env_str(const std::string var,
                                      const std::string def = "") {
  std::string result = def;
 
  const char *val = std::getenv(var.c_str());
  if (val != nullptr) {
    result = std::string(val);
  }
 
  return result;
}