OutputConvention.cxx

#include "OutputConvention.h"





//---------------------------------------------------------------------------------------------------------
Acclaim::OutputConvention::OutputConvention(int argcIn, char* argvIn[]){
  dateTime = TDatime();
  argc = argcIn;
  argv = argvIn;
  outFileName = "";
  dateTimeSuffix = "";
  outputDir = "";
  // subDir = "";
}





//---------------------------------------------------------------------------------------------------------
TString Acclaim::OutputConvention::getOutputFileName(TString ext){

  if(outFileName==""){

    // Directory
    outFileName = getOutputDir();

    // Excutable name
    TString theArgv0 = TString::Format("%s", argv[0]);
    TObjArray* tkns = theArgv0.Tokenize("/");
    if(tkns->GetSize() > 0){
      TObjString* lastPartOfName = (TObjString*) tkns->At(tkns->GetLast());
      theArgv0 = lastPartOfName->GetString();
    }

    // outFileName += theArgv0 + "Plots";
    outFileName += theArgv0;

    // Excutable args
    if(argc > 1){
      for(int argInd=1; argInd < argc; argInd++){
    outFileName += TString::Format("_%s", argv[argInd]);
      }
    }

    // Date and time of running executable
    outFileName += getDateTimeSuffix();

    if(ext==""){
      // ROOT suffix
      outFileName += ".root";
    }
    else{
      if(strncmp(ext.Data(), ".", 1)!=0){
    ext = TString::Format(".%s", ext.Data());
      }
      outFileName += ext;
    }
  }


  return outFileName;


}



TFile* Acclaim::OutputConvention::makeFile(){
  TString outFileName = getOutputFileName();
  TFile* outFile = new TFile(outFileName, "recreate");
  if(outFile->IsZombie()){
    std::cerr << "Error! Unable to open output file " << outFileName.Data() << std::endl;
    outFile = NULL;
  }
  return outFile;
}





//---------------------------------------------------------------------------------------------------------
TString Acclaim::OutputConvention::getDateTimeSuffix(){
  if(dateTimeSuffix==""){
    // dateTimeSuffix = TString::Format("_%d-%d-%d_%d-%d-%d",
    //                   dateTime.GetYear(),
    //                   dateTime.GetMonth(),
    //                   dateTime.GetDay(),
    //                   dateTime.GetHour(),
    //                   dateTime.GetMinute(),
    //                   dateTime.GetSecond()
    //                   );
    dateTimeSuffix = TString::Format("_%d", dateTime.GetYear());

    Int_t m = dateTime.GetMonth();
    if(m < 10){
      dateTimeSuffix += TString::Format("-0%d", m);
    }
    else{
      dateTimeSuffix += TString::Format("-%d", m);
    }

    Int_t d = dateTime.GetDay();
    if(d < 10){
      dateTimeSuffix += TString::Format("-0%d", d);
    }
    else{
      dateTimeSuffix += TString::Format("-%d", d);
    }

    Int_t h = dateTime.GetHour();
    if(h < 10){
      dateTimeSuffix += TString::Format("_0%d", h);
    }
    else{
      dateTimeSuffix += TString::Format("_%d", h);
    }

    Int_t m2 = dateTime.GetMinute();
    if(m2 < 10){
      dateTimeSuffix += TString::Format("-0%d", m2);
    }
    else{
      dateTimeSuffix += TString::Format("-%d", m2);
    }

    Int_t s = dateTime.GetSecond();
    if(s < 10){
      dateTimeSuffix += TString::Format("-0%d", s);
    }
    else{
      dateTimeSuffix += TString::Format("-%d", s);
    }
  }
  return dateTimeSuffix;
}





//---------------------------------------------------------------------------------------------------------
TString Acclaim::OutputConvention::getOutputDir(){

  outputDir = "";
  const char* outputDirPoss = getenv("OUTPUT_DIR");
  if(outputDirPoss!=NULL){
    outputDir += TString::Format("%s/", outputDirPoss); // Add trailing forward slash...
  }

  return outputDir;
}




//---------------------------------------------------------------------------------------------------------
TFile* Acclaim::OutputConvention::getFile(TString fileNameWithWildcards){
  // This might be my favourite little bit of stand alone code.

  TFile* theFile = NULL;

  TChain* tempChain = new TChain("tempChain");
  tempChain->Add(fileNameWithWildcards);

  TObjArray* fileList = tempChain->GetListOfFiles();

  const int numFiles = fileList->GetEntries();
  if(numFiles > 0){
    // std::cerr << "numFiles = " << numFiles << std::endl;
    std::vector<TString> fileNames(numFiles, "");;

    for(int fileInd=0; fileInd < numFiles; fileInd++){
      fileNames.at(fileInd) = TString::Format("%s", fileList->At(fileInd)->GetTitle());
    }
    std::sort(fileNames.begin(), fileNames.end(), std::greater<TString>());
    for(int fileInd=0; fileInd < numFiles; fileInd++){
      // std::cerr << fileInd << "\t" << fileNames.at(fileInd) << std::endl;
    }
    theFile = TFile::Open(fileNames.at(0));
  }
  delete tempChain;

  return theFile;

}