EnvironmentVariable.cc

#include "EnvironmentVariable.h"
#include <iostream>
#include <fstream>
#include <stdlib.h>
#include "TString.h"

const char* getEnv(const char* envName, const char* description, bool failHard=true){

  const char* envVar = std::getenv(envName);
  if(!envVar){
    const char* RED = "\x1b[31m";
    const char* RESET = "\x1b[0m";

    // print fatal error or warning
    if(failHard){
      std::cerr << RED << "Fatal error! " << RESET;
    }
    else{
      std::cerr << RED << "Warning! " << RESET;
    }

    // what we couldn't find
    std::cerr << "Could not find environment variable "
              << RED << envName << RESET << std::endl;

    // helpful message, if you gave me one.
    if(description){
      std::cerr << description << std::endl;
    }

    // maybe actually give up
    if(failHard){
      std::cerr << "Giving up." << std::endl;
      exit(1);
    }
  }
  return envVar;
}


const char* EnvironmentVariable::ICEMC_SRC_DIR(){

#ifdef ANITA_BUILD_TOOL  
  const char* icemc_src_dir = getEnv("ICEMC_SRC_DIR", "Without this environment variable I can't find input data or config files!", true);
  return icemc_src_dir;
#else
  const char* icemc_src_dir = getEnv("ICEMC_SRC_DIR", "Will guess icemc source directory is present working directory", false);
  if(!icemc_src_dir){
    icemc_src_dir = ".";
  }
  return icemc_src_dir;
  
#endif
}

const char* EnvironmentVariable::ICEMC_VERSION(TString outputdir){

  system(Form("git rev-parse HEAD > %s/gitversion.txt", outputdir.Data()));
  static std::string gitversion;
  std::ifstream gitversionfile (Form("%s/gitversion.txt", outputdir.Data()));
  if (gitversionfile.is_open())
    {
      getline (gitversionfile,gitversion);
      gitversionfile.close();
    }

  return gitversion.c_str();
  
}