ShallowTracksProducer.cc

Go to the documentation of this file.

#include "CalibTracker/SiStripCommon/interface/ShallowTracksProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "boost/foreach.hpp"

ShallowTracksProducer::ShallowTracksProducer(const edm::ParameterSet& iConfig)
  :  tracks_token_( consumes<edm::View<reco::Track> >( iConfig.getParameter<edm::InputTag>("Tracks") )),
     Prefix       ( iConfig.getParameter<std::string>("Prefix")    ),
     Suffix       ( iConfig.getParameter<std::string>("Suffix")    )
{
  produces <unsigned int>               ( Prefix + "number"    + Suffix );
  produces <std::vector<double> >       ( Prefix + "chi2"      + Suffix );
  produces <std::vector<double> >       ( Prefix + "ndof"      + Suffix );
  produces <std::vector<double> >       ( Prefix + "chi2ndof"  + Suffix );
  produces <std::vector<float> >        ( Prefix + "charge"    + Suffix );
  produces <std::vector<float> >        ( Prefix + "momentum"  + Suffix );
  produces <std::vector<float> >        ( Prefix + "pt"        + Suffix );
  produces <std::vector<float> >        ( Prefix + "pterr"     + Suffix );
  produces <std::vector<unsigned int> > ( Prefix + "hitsvalid" + Suffix );
  produces <std::vector<unsigned int> > ( Prefix + "hitslost"  + Suffix );
  produces <std::vector<double> >       ( Prefix + "theta"     + Suffix );
  produces <std::vector<double> >       ( Prefix + "thetaerr"  + Suffix );
  produces <std::vector<double> >       ( Prefix + "phi"       + Suffix );
  produces <std::vector<double> >       ( Prefix + "phierr"    + Suffix );
  produces <std::vector<double> >       ( Prefix + "eta"       + Suffix );
  produces <std::vector<double> >       ( Prefix + "etaerr"    + Suffix );
  produces <std::vector<double> >       ( Prefix + "dxy"       + Suffix );
  produces <std::vector<double> >       ( Prefix + "dxyerr"    + Suffix );
  produces <std::vector<double> >       ( Prefix + "dsz"       + Suffix );
  produces <std::vector<double> >       ( Prefix + "dszerr"    + Suffix );
  produces <std::vector<double> >       ( Prefix + "qoverp"    + Suffix );
  produces <std::vector<double> >       ( Prefix + "qoverperr" + Suffix );
  produces <std::vector<double> >       ( Prefix + "vx"        + Suffix );
  produces <std::vector<double> >       ( Prefix + "vy"        + Suffix );
  produces <std::vector<double> >       ( Prefix + "vz"        + Suffix );
  produces <std::vector<int> >          ( Prefix + "algo"        + Suffix );
}

void ShallowTracksProducer::
produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  auto       number      = std::make_unique<unsigned int>(0);
  auto       chi2        = std::make_unique<std::vector<double>>();
  auto       ndof        = std::make_unique<std::vector<double>>();
  auto       chi2ndof    = std::make_unique<std::vector<double>>();
  auto       charge      = std::make_unique<std::vector<float>>();
  auto       momentum    = std::make_unique<std::vector<float>>();
  auto       pt          = std::make_unique<std::vector<float>>();
  auto       pterr       = std::make_unique<std::vector<float>>();
  auto       hitsvalid   = std::make_unique<std::vector<unsigned int>>();
  auto       hitslost    = std::make_unique<std::vector<unsigned int>>();
  auto       theta       = std::make_unique<std::vector<double>>();
  auto       thetaerr    = std::make_unique<std::vector<double>>();
  auto       phi         = std::make_unique<std::vector<double>>();
  auto       phierr      = std::make_unique<std::vector<double>>();
  auto       eta         = std::make_unique<std::vector<double>>();
  auto       etaerr      = std::make_unique<std::vector<double>>();
  auto       dxy         = std::make_unique<std::vector<double>>();
  auto       dxyerr      = std::make_unique<std::vector<double>>();
  auto       dsz         = std::make_unique<std::vector<double>>();
  auto       dszerr      = std::make_unique<std::vector<double>>();
  auto       qoverp      = std::make_unique<std::vector<double>>();
  auto       qoverperr   = std::make_unique<std::vector<double>>();
  auto       vx          = std::make_unique<std::vector<double>>();
  auto       vy          = std::make_unique<std::vector<double>>();
  auto       vz          = std::make_unique<std::vector<double>>();
  auto       algo        = std::make_unique<std::vector<int>>();

  edm::Handle<edm::View<reco::Track> > tracks;  iEvent.getByToken(tracks_token_, tracks);
  
  *number = tracks->size();
  BOOST_FOREACH( const reco::Track track, *tracks) {
    chi2->push_back(      track.chi2()              );
    ndof->push_back(      track.ndof()              );
    chi2ndof->push_back(  track.chi2()/track.ndof() );
    charge->push_back(    track.charge()            );
    momentum->push_back(  track.p()                 );
    pt->push_back(        track.pt()                );
    pterr->push_back(     track.ptError()           );
    hitsvalid->push_back( track.numberOfValidHits() );
    hitslost->push_back(  track.numberOfLostHits()  );
    theta->push_back(     track.theta()             );
    thetaerr->push_back(  track.thetaError()        );
    phi->push_back(       track.phi()               );
    phierr->push_back(    track.phiError()          );
    eta->push_back(       track.eta()               );
    etaerr->push_back(    track.etaError()          );
    dxy->push_back(       track.dxy()               );
    dxyerr->push_back(    track.dxyError()          );
    dsz->push_back(       track.dsz()               );
    dszerr->push_back(    track.dszError()          );
    qoverp->push_back(    track.qoverp()            );
    qoverperr->push_back( track.qoverpError()       );
    vx->push_back(        track.vx()                );
    vy->push_back(        track.vy()                );
    vz->push_back(        track.vz()                );
    algo->push_back(      (int) track.algo()              );
  }           
              
  iEvent.put(std::move(number),       Prefix + "number"     + Suffix );
  iEvent.put(std::move(chi2),         Prefix + "chi2"       + Suffix );
  iEvent.put(std::move(ndof),         Prefix + "ndof"       + Suffix );
  iEvent.put(std::move(chi2ndof),     Prefix + "chi2ndof"   + Suffix );
  iEvent.put(std::move(charge),       Prefix + "charge"     + Suffix );
  iEvent.put(std::move(momentum),     Prefix + "momentum"   + Suffix );
  iEvent.put(std::move(pt),           Prefix + "pt"         + Suffix );
  iEvent.put(std::move(pterr),        Prefix + "pterr"      + Suffix );
  iEvent.put(std::move(hitsvalid),    Prefix + "hitsvalid"  + Suffix );
  iEvent.put(std::move(hitslost),     Prefix + "hitslost"   + Suffix );
  iEvent.put(std::move(theta),        Prefix + "theta"      + Suffix );
  iEvent.put(std::move(thetaerr),     Prefix + "thetaerr"   + Suffix );
  iEvent.put(std::move(phi),          Prefix + "phi"        + Suffix );
  iEvent.put(std::move(phierr),       Prefix + "phierr"     + Suffix );
  iEvent.put(std::move(eta),          Prefix + "eta"        + Suffix );
  iEvent.put(std::move(etaerr),       Prefix + "etaerr"     + Suffix );
  iEvent.put(std::move(dxy),          Prefix + "dxy"        + Suffix );
  iEvent.put(std::move(dxyerr),       Prefix + "dxyerr"     + Suffix );
  iEvent.put(std::move(dsz),          Prefix + "dsz"        + Suffix );
  iEvent.put(std::move(dszerr),       Prefix + "dszerr"     + Suffix );
  iEvent.put(std::move(qoverp),       Prefix + "qoverp"     + Suffix );
  iEvent.put(std::move(qoverperr),    Prefix + "qoverperr"  + Suffix );
  iEvent.put(std::move(vx),           Prefix + "vx"         + Suffix );
  iEvent.put(std::move(vy),           Prefix + "vy"         + Suffix );
  iEvent.put(std::move(vz),           Prefix + "vz"         + Suffix );
  iEvent.put(std::move(algo),         Prefix + "algo"       + Suffix );

}