FWLiteWithSelectorUtils.cc

Go to the documentation of this file.

#include <TH1F.h>
#include <TROOT.h>
#include <TFile.h>
#include <TSystem.h>

#include "DataFormats/FWLite/interface/Event.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/FWLite/interface/FWLiteEnabler.h"

#include "DataFormats/FWLite/interface/InputSource.h"
#include "DataFormats/FWLite/interface/OutputFiles.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSetReader/interface/ParameterSetReader.h"

#include "TStopwatch.h"
#include "PhysicsTools/FWLite/interface/WSelector.h"
#include "PhysicsTools/FWLite/interface/TFileService.h"
//#include "PhysicsTools/FWLite/interface/WSelectorFast.h"

int main(int argc, char* argv[]) {
  // define what muon you are using; this is necessary as FWLite is not
  // capable of reading edm::Views
  using reco::Muon;

  // ----------------------------------------------------------------------
  // First Part:
  //
  //  * enable FWLite
  //  * book the histograms of interest
  //  * open the input file
  // ----------------------------------------------------------------------

  // load framework libraries
  gSystem->Load("libFWCoreFWLite");
  FWLiteEnabler::enable();

  if (argc < 2) {
    std::cout << "Usage : " << argv[0] << " [parameters.py]" << std::endl;
    return 0;
  }

  if (!edm::readPSetsFrom(argv[1])->existsAs<edm::ParameterSet>("process")) {
    std::cout << " ERROR: ParametersSet 'process' is missing in your configuration file" << std::endl;
    exit(0);
  }
  // get the python configuration
  const edm::ParameterSet& process = edm::readPSetsFrom(argv[1])->getParameter<edm::ParameterSet>("process");
  fwlite::InputSource inputHandler_(process);
  fwlite::OutputFiles outputHandler_(process);

  // initialize the W selector
  edm::ParameterSet selection = process.getParameter<edm::ParameterSet>("selection");
  WSelector wSelector(selection);
  pat::strbitset wSelectorReturns = wSelector.getBitTemplate();

  // book a set of histograms
  fwlite::TFileService fs = fwlite::TFileService(outputHandler_.file());
  TFileDirectory theDir = fs.mkdir("analyzeBasicPat");
  TH1F* muonPt_ = theDir.make<TH1F>("muonPt", "pt", 100, 0., 300.);
  TH1F* muonEta_ = theDir.make<TH1F>("muonEta", "eta", 100, -3., 3.);
  TH1F* muonPhi_ = theDir.make<TH1F>("muonPhi", "phi", 100, -5., 5.);

  // start a CPU timer
  TStopwatch timer;
  timer.Start();

  // loop the events
  int ievt = 0;
  unsigned int nEventsAnalyzed = 0;
  int maxEvents_(inputHandler_.maxEvents());
  for (unsigned int iFile = 0; iFile < inputHandler_.files().size(); ++iFile) {
    // open input file (can be located on castor)
    TFile* inFile = TFile::Open(inputHandler_.files()[iFile].c_str());
    if (inFile) {
      // ----------------------------------------------------------------------
      // Second Part:
      //
      //  * loop the events in the input file
      //  * receive the collections of interest via fwlite::Handle
      //  * fill the histograms
      //  * after the loop close the input file
      // ----------------------------------------------------------------------
      fwlite::Event ev(inFile);
      for (ev.toBegin(); !ev.atEnd(); ++ev, ++ievt) {
        edm::EventBase const& event = ev;
        // break loop if maximal number of events is reached
        if (maxEvents_ > 0 ? ievt + 1 > maxEvents_ : false)
          break;
        // simple event counter
        if (inputHandler_.reportAfter() != 0 ? (ievt > 0 && ievt % inputHandler_.reportAfter() == 0) : false)
          std::cout << "  processing event: " << ievt << std::endl;

        if (wSelector(event, wSelectorReturns)) {
          pat::Muon const& wMuon = wSelector.wMuon();
          muonPt_->Fill(wMuon.pt());
          muonEta_->Fill(wMuon.eta());
          muonPhi_->Fill(wMuon.phi());
        }
        ++nEventsAnalyzed;
      }
      // close input file
      inFile->Close();
    }
    // break loop if maximal number of events is reached:
    // this has to be done twice to stop the file loop as well
    if (maxEvents_ > 0 ? ievt + 1 > maxEvents_ : false)
      break;
  }
  // stop CPU timer
  timer.Stop();

  // print selector
  wSelector.print(std::cout);

  // print some timing statistics
  double rtime = timer.RealTime();
  double ctime = timer.CpuTime();
  // timing printouts
  printf("Analyzed events: %d \n", nEventsAnalyzed);
  printf("RealTime=%f seconds, CpuTime=%f seconds\n", rtime, ctime);
  printf("%4.2f events / RealTime second .\n", (double)nEventsAnalyzed / rtime);
  printf("%4.2f events / CpuTime second .\n", (double)nEventsAnalyzed / ctime);
  return 0;
}