TGRSIUtilities.cxx

Go to the documentation of this file.

#include "TGRSIUtilities.h"
 
#include <cstdlib>
#include <sys/stat.h>
#include <iostream>
#include <fstream>
#include <algorithm>
 
#include "TObjArray.h"
#include "TObjString.h"
#include "TPRegexp.h"
#include "TString.h"
 
bool FileExists(const char* filename)
{
   /// This checks if the path exist, and if it is a file and not a directory!
   struct stat buffer{};
   int         state = stat(filename, &buffer);
   // state != 0 means we couldn't get file attributes. This doesn't necessary mean the file
   // does not exist, we might just be missing permission to access it. But for our purposes
   // this is the same as the file not existing.
   if(state != 0) { return false; }
   // we got the file attributes, so it exsist, we just need to check if it is a directory.
   return !S_ISDIR(buffer.st_mode);
}
 
bool DirectoryExists(const char* dirname)
{
   /// This checks if the directory exists
   struct stat buffer{};
   int         state = stat(dirname, &buffer);
   // state != 0 means we couldn't get file attributes. This doesn't necessary mean the file
   // does not exist, we might just be missing permission to access it. But for our purposes
   // this is the same as the file not existing.
   if(state != 0) { return false; }
   // we got the file attributes, so it exsist, we just need to check if it is a directory.
   return S_ISDIR(buffer.st_mode);
}
 
bool AllFilesExist(const std::vector<std::string>& filenames)
{
   return std::all_of(filenames.begin(), filenames.end(), [](auto filename) { return FileExists(filename.c_str()); });
}
 
void trim(std::string& line, const std::string& trimChars)
{
   /// Removes the string "trimCars" from  the string 'line'
   if(line.length() == 0) {
      return;
   }
   std::size_t found = line.find_first_not_of(trimChars);
   if(found != std::string::npos) {
      line = line.substr(found, line.length());
   }
   found = line.find_last_not_of(trimChars);
   if(found != std::string::npos) {
      line = line.substr(0, found + 1);
   }
}
 
void trimWS(std::string& line)
{
   /// Removes whitespace from the string 'line'
   line.erase(line.begin(), std::find_if(line.begin(), line.end(), [](int ch) { return std::isspace(ch) == 0; }));
   line.erase(std::find_if(line.rbegin(), line.rend(), [](int ch) { return std::isspace(ch) == 0; }).base(), line.end());
}
 
int GetRunNumber(const std::string& fileName)
{
   if(fileName.length() == 0) {
      return 0;
   }
   std::size_t found = fileName.rfind(".root");
   if(found == std::string::npos) {
      return 0;
   }
   std::size_t found2 = fileName.rfind('-');
 
   if(found2 == std::string::npos) {
      found2 = fileName.rfind('_');
   }
   std::string temp;
   if(found2 == std::string::npos || fileName.compare(found2 + 4, 5, ".root") != 0) {
      temp = fileName.substr(found - 5, 5);
   } else {
      temp = fileName.substr(found - 9, 5);
   }
   return atoi(temp.c_str());
}
 
int GetSubRunNumber(const std::string& fileName)
{
   if(fileName.length() == 0) {
      return -1;
   }
 
   std::size_t found = fileName.rfind('-');
   if(found != std::string::npos) {
      std::string temp = fileName.substr(found + 1, 3);
      return atoi(temp.c_str());
   }
   found = fileName.rfind('_');
   if(found != std::string::npos) {
      std::string temp = fileName.substr(found + 1, 3);
      return atoi(temp.c_str());
   }
   return -1;
}
 
const char* GetColorFromNumber(int number)
{
   switch(number) {
   case(0): return "B";
   case(1): return "G";
   case(2): return "R";
   case(3): return "W";
   };
   return "X";
}
 
const char* GetLongColorFromNumber(int number)
{
   switch(number) {
   case(0): return "Blue";
   case(1): return "Green";
   case(2): return "Red";
   case(3): return "White";
   };
   return "Unknown";
}