XSecAdder.cc

Go to the documentation of this file.

/**
  @file   XSecAdder.cc
  Describes the XSecAdder class.
  @author Daniel Abercrombie <dabercro@mit.edu>
*/

#include <iostream>
#include <map>
#include "TFile.h"
#include "TTree.h"
#include "TBranch.h"
#include "XSecAdder.h"

ClassImp(XSecAdder)

//--------------------------------------------------------------------
XSecAdder::XSecAdder(TString name) :
  fBranchName(name)
{
  SetKeepAllFiles(true);
  SetMultiplyLumi(false);
}

//--------------------------------------------------------------------
XSecAdder::~XSecAdder()
{ }

//--------------------------------------------------------------------
void XSecAdder::AddXSecs()
{
  RunThreads();
}

void XSecAdder::RunFile(FileInfo& info) {

  TFile *file = new TFile(info.fFileName,"UPDATE");
  if (!file)
    exit(0);

  TTree *tree = (TTree*) file->Get(fInTreeName);
  TTree *outTree;

  if (fOutTreeName != fInTreeName)
    outTree = new TTree(fOutTreeName, fOutTreeName);
  else
    outTree = tree;

  Int_t nentries = tree->GetEntries();
  Float_t weight = info.fXSecWeight;

  std::map<TString, Float_t> mergeBranches;

  for (UInt_t iMerge = 0; iMerge != fMergeBranches.size(); ++iMerge) {
    mergeBranches[fMergeBranches[iMerge]] = 1.0;
    tree->SetBranchAddress(fMergeBranches[iMerge], &(mergeBranches[fMergeBranches[iMerge]]));
  }

  if (fOutTreeName != fInTreeName || (!tree->GetBranch(fBranchName))) {

    Float_t XSecWeight = 1.;
    TBranch *XSecWeightBr = outTree->Branch(fBranchName, &XSecWeight,fBranchName + "/F");

    for (int entry = 0; entry < nentries; entry++) {
      if (entry % 500000 == 0) {  /// @todo Make a ReportFrequency class
        output_lock.Lock();
        std::cout << "Processing " << info.fFileName << " ... " << float(entry)/float(nentries)*100 << "%" << std::endl;
        output_lock.UnLock();
      }
      XSecWeight = weight;

      if (fMergeBranches.size() != 0)
        tree->GetEntry(entry);

      for (UInt_t iMerge = 0; iMerge != fMergeBranches.size(); ++iMerge)
        XSecWeight *= mergeBranches[fMergeBranches[iMerge]];

      if (fOutTreeName == fInTreeName)
        XSecWeightBr->Fill();
      else
        outTree->Fill();
    }
    if (fOutTreeName == fInTreeName)
      outTree->Write();
    else
      outTree->Write(fOutTreeName, TObject::kOverwrite);
  }

  file->Close();

}