Vertices.cc

Go to the documentation of this file.

//
// Original Author:  Roberval Walsh Bastos Rangel
//         Created:  Mon, 20 Oct 2014 14:24:08 GMT
//
//

// system include files
// 
// user include files
#include "DataFormats/Provenance/interface/Provenance.h"

#include "FWCore/Framework/interface/Event.h"
// 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "DataFormats/PatCandidates/interface/Jet.h"

#include "CommonTools/Utils/interface/PtComparator.h"



#include "Analysis/Ntuplizer/interface/Vertices.h"

#include "TTree.h"


//
// class declaration
//

using namespace analysis;
using namespace analysis::ntuple;

//
// constructors and destructor
//
Vertices::Vertices()
{
   // default constructor
}

Vertices::Vertices(const edm::InputTag& tag, TTree* tree)
{
   input_collection_ = tag;
   tree_ = tree;

   tree_->Branch("n",   &this->n_,    "n/I");
   tree_->Branch("x",    this->x_,    "x[n]/F");
   tree_->Branch("y",    this->y_,    "y[n]/F");
   tree_->Branch("z",    this->z_,    "z[n]/F");
   tree_->Branch("xe",   this->xe_,   "xe[n]/F");
   tree_->Branch("ye",   this->ye_,   "ye[n]/F");
   tree_->Branch("ze",   this->ze_,   "ze[n]/F");
   tree_->Branch("rho",  this->rho_,  "rho[n]/F");
   tree_->Branch("chi2", this->chi2_, "chi2[n]/F");
   tree_->Branch("ndof", this->ndof_, "ndof[n]/F");
   tree_->Branch("fake", this->fake_, "fake[n]/O");
      
}

Vertices::~Vertices()
{
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)
}


//
// member functions
//

// ------------ method called for each event  ------------
void Vertices::ReadFromEvent(const edm::Event& event)
{
   using namespace edm;
   
   edm::Handle<reco::VertexCollection> handler;
   event.getByLabel(this->input_collection_, handler);
   candidates_ = *(handler.product());
   
   int n = 0;
   for ( size_t i = 0 ; i < candidates_.size(); ++i )
   {
      if ( n >= maxPVs ) break;
      
      this->x_[n]    = candidates_[i].x();
      this->y_[n]    = candidates_[i].y();
      this->z_[n]    = candidates_[i].z();
      this->xe_[n]   = candidates_[i].xError();
      this->ye_[n]   = candidates_[i].yError();
      this->ze_[n]   = candidates_[i].zError();
      this->fake_[n] = candidates_[i].isFake();
      this->ndof_[n] = candidates_[i].ndof();
      this->chi2_[n] = candidates_[i].chi2();
      this->rho_[n]  = candidates_[i].position().Rho();
      
      ++n;
      
   }
   this->n_ = n;

   
}

void Vertices::Fill(const edm::Event& event)
{
   this->ReadFromEvent(event);
   tree_->Fill();
}

// ------------ method called once each job just before starting event loop  ------------