postProcessIDPVMHistos.cxx File Reference

#include "InDetPhysValMonitoring/ResolutionHelper.h"
#include "TFile.h"
#include "TSystem.h"
#include "TH1.h"
#include "TH2.h"
#include "TObject.h"
#include <iostream>
#include <memory>
#include <string>

Include dependency graph for postProcessIDPVMHistos.cxx:

Go to the source code of this file.

Functions
bool	file_exists (const string &p_name)
bool	isResolutionHelper (TObject *entry)
std::vector< std::string >	getObservableAndResoAndSuffix (const TObject *resHelper)
std::string	getResoType (const TObject *resHelper)
TH1 *	cloneExisting (const std::string &name)
std::pair< std::string, std::string >	getPullAndResoNames (const std::string &type)
int	postProcessHistos (TObject *resHelper, IDPVM::ResolutionHelper &theHelper)
int	postProcessDir (TDirectory *dir, IDPVM::ResolutionHelper &theHelper)
int	pproc_file (const std::string &p_infile)
int	main (int argc, char *argv[])

Detailed Description

---

Author

: Max Goblirsch Based on code by Liza Mijovic, Soeren Prell

Goal: Update resolutions extracted from 2D histograms after merging several output files (typically after grid running)

---

Definition in file postProcessIDPVMHistos.cxx.

Function Documentation

cloneExisting()

TH1* cloneExisting

(

const std::string &

name

)

Definition at line 72 of file postProcessIDPVMHistos.cxx.

                                          {
    auto *h = gDirectory->Get(name.c_str()); 
    if (!h){ 
        std::cerr << "Could not find existing histogram "<<name<<" - will not postprocess "<<std::endl; 
        return nullptr;
    }
    auto *ret = dynamic_cast<TH1*>(h->Clone(name.c_str()));
    if (!ret){ 
        std::cerr << "Found an existing object "<<name<<", but it is not a histogram ("<<h->IsA()->GetName()<<") - will not postprocess "<<std::endl; 
    }
    return ret; // will also catch ret == nullptr
}

file_exists()

bool file_exists

(

const string &

p_name

)

Definition at line 31 of file postProcessIDPVMHistos.cxx.

                                        {
  return !gSystem->AccessPathName(p_name.c_str(), kFileExists);    
}

getObservableAndResoAndSuffix()

std::vector<std::string> getObservableAndResoAndSuffix

(

const TObject *

resHelper

)

Definition at line 43 of file postProcessIDPVMHistos.cxx.

                                                                            {
    const std::string name{resHelper->GetName()};
    const std::string keyWord {"Helper_"}; 
    const size_t offset = keyWord.size();
    auto start = name.find(keyWord)+offset;
    auto sep = name.find('_',start);
    auto sep2 = name.find('_',sep+1);
 
    std::string first  = name.substr(start, sep - start);
    std::string second;
    std::string third;
    if (sep2 == std::string::npos) {
        // Only one underscore after Helper_
        second = name.substr(sep + 1);
    } else {
        second = name.substr(sep + 1, sep2 - sep - 1);
        third  = name.substr(sep2 + 1);
    }
    // coverity[copy_constructor_call]
    return {first, second, third};
}

getPullAndResoNames()

std::pair<std::string, std::string> getPullAndResoNames

(

const std::string &

type

)

Definition at line 86 of file postProcessIDPVMHistos.cxx.

                                                                         {
  if (type == "res"){
    return {"resolution","resmean"}; 
  }
  else if (type == "pull"){
    return {"pullwidth","pullmean"}; 
  }
  else {
    std::cerr << " Not able to identify the histogram names for a resolution type "<<type<<" - supported are 'res' and 'pull'. "<<std::endl;
  }
  return {"",""}; 
}

getResoType()

std::string getResoType

(

const TObject *

resHelper

)

Definition at line 66 of file postProcessIDPVMHistos.cxx.

                                               {
    std::string aux{resHelper->GetName()};
    return aux.substr(0, aux.find("Helper")); 
}

isResolutionHelper()

bool isResolutionHelper

(

TObject *

entry

)

Definition at line 36 of file postProcessIDPVMHistos.cxx.

                                       {
  const std::string objName{entry->GetName()}; 
  return ((objName.find("resHelper") == 0 || objName.find("pullHelper") == 0) && dynamic_cast<TH1*>(entry));
}

main()

int main	(	int	argc,
		char *	argv[]
	)

Standard usage. The user passes a file they wish to update.

hadd imitation. Ugly, but allows this to run in PhysVal merge transforms. We deliberately ignore the first arg ("-f") in the following. This is because the physval merge step requires passing a "-f" ...

copy the input file to the output file

and mark the output for postprocessing. Input stays invariant in this mode

check if the input exists

and post-process if it does

Definition at line 166 of file postProcessIDPVMHistos.cxx.

                                 {
  
  std::string infile{""}; 
  if (argc == 2){
     infile  = argv[1];
  }
  else if (argc == 4 && std::string(argv[1]) == std::string{"-f"}){
    gSystem->CopyFile(argv[3],argv[2],true); 
    infile = argv[2]; 
  }
  else {
    std::cerr<<" Usage: postProcessIDPVMHistos <File to post-process>"<<std::endl;
    std::cerr<< "    where the file is typically obtained by hadding" << std::endl;
    std::cerr<< "    outputs of several independent IDPVM runs." << std::endl;
    std::cerr<<" Alternative usage: postProcessIDPVMHistos -f <desired output file name> <File to post-process>"<<std::endl;
    std::cerr<< "    imitates a hadd-like signature for PhysVal merging." << std::endl;
    return 1; 
  }
  if (!file_exists(infile)) {
    std::cerr << "Error: invalid input file: " << infile << std::endl;
    return 1;
  }
  std::cout << " Post-processing file " << infile << "\n" << std::endl;
  return pproc_file(infile);
}

postProcessDir()

int postProcessDir	(	TDirectory *	dir,
		IDPVM::ResolutionHelper &	theHelper
	)

Definition at line 125 of file postProcessIDPVMHistos.cxx.

                                                                      {
 
  int outcome = 0; 
  auto theCWD = gDirectory; 
  // walk through all keys in this directory 
  dir->cd();
  auto *keys = dir->GetListOfKeys();
  for (auto *const key : *keys){
    // Check if it's a directory and handle separately
    TDirectory* theDir = dynamic_cast<TDirectory*>(dir->Get(key->GetName()));
    if (theDir){
      outcome |= postProcessDir(theDir, theHelper);
      continue;  // Do NOT delete directories, ROOT manages them
    }
 
    // If it's a histogram, wrap it in a unique_ptr
    std::unique_ptr<TObject> gotIt(dir->Get(key->GetName()));
    if (isResolutionHelper(gotIt.get())){
      outcome |= postProcessHistos(gotIt.get(), theHelper);
    }
  }
  theCWD->cd();
  return outcome;
}

postProcessHistos()

int postProcessHistos	(	TObject *	resHelper,
		IDPVM::ResolutionHelper &	theHelper
	)

Definition at line 99 of file postProcessIDPVMHistos.cxx.

                                                                            {
    // here we have to rely on the naming conventions of IDPVM to identify what we are looking at 
    auto vars = getObservableAndResoAndSuffix(resHelper);
    auto type = getResoType(resHelper); 
    // cast to TH2
    TH2* resHelper2D = dynamic_cast<TH2*>(resHelper); 
    if (!resHelper2D){
        std::cerr <<"Unable to reduce the histogram "<<resHelper->GetName()<<" to a TH2 - this histo can not yet be postprocessed! " <<std::endl; 
        return 1; 
    }
    const auto & oneDimNames = getPullAndResoNames(type);  
    // get the corresponding 1D histos by cloning the existing ones in the same folder
    std::string suffix = "_vs_"+vars[0]+"_"+vars[1];
    if(!vars[2].empty()) suffix += "_" + vars[2];
    TH1* h_width = cloneExisting(oneDimNames.first+suffix);
    TH1* h_mean = cloneExisting(oneDimNames.second+suffix);
    // then call the resolution helper as done in "online" IDPVM
    theHelper.makeResolutions(resHelper2D, h_width, h_mean); 
    // update our 1D histos 
    h_width->Write();
    h_mean->Write();
    // and we are done
    return 0;
}

pproc_file()

int pproc_file

(

const std::string &

p_infile

)

Definition at line 151 of file postProcessIDPVMHistos.cxx.

                                           {
 
    IDPVM::ResolutionHelper theHelper; 
 
    std::unique_ptr<TFile> infile(TFile::Open(p_infile.c_str(),"UPDATE"));
    if (!infile || infile->IsZombie()) {
      std::cerr << "could not open input file "<<p_infile<<" for updating "<< std::endl;
      return 1;
    }
 
    int res = postProcessDir(infile.get(), theHelper); // recursively post-process the directory tree, starting from the root.  
    return res;
}