EvtPlaneProducer Class Reference

#include <RecoHI/EvtPlaneProducer/src/EvtPlaneProducer.cc>

Inheritance diagram for EvtPlaneProducer:

Public Member Functions
	EvtPlaneProducer (const edm::ParameterSet &)
	~EvtPlaneProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

Private Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
double	caloCentRef_
double	caloCentRefWidth_
edm::Handle< CaloTowerCollection >	caloCollection_
edm::InputTag	caloTag_
edm::EDGetTokenT< CaloTowerCollection >	caloToken
edm::Handle< std::vector< reco::CastorTower > >	castorCollection_
edm::InputTag	castorTag_
edm::EDGetTokenT< std::vector< reco::CastorTower > >	castorToken
int	CentBinCompression_
edm::InputTag	centralityBinTag_
edm::EDGetTokenT< int >	centralityBinToken
std::string	centralityLabel_
std::string	centralityMC_
std::string	centralityVariable_
double	chi2_
double	dzerr_
HiEvtPlaneFlatten *	flat [NumEPNames]
int	FlatOrder_
edm::ESWatcher< HeavyIonRPRcd >	hirpWatcher
edm::ESWatcher< HeavyIonRcd >	hiWatcher
bool	loadDB_
double	maxet_
double	maxpt_
double	maxvtx_
double	minet_
double	minpt_
double	minvtx_
double	nCentBins_
int	NumFlatBins_
GenPlane *	rp [NumEPNames]
edm::Handle< reco::TrackCollection >	trackCollection_
edm::InputTag	trackTag_
edm::EDGetTokenT< reco::TrackCollection >	trackToken
edm::Handle< std::vector< reco::Vertex > >	vertex_
edm::InputTag	vertexTag_
edm::EDGetTokenT< std::vector< reco::Vertex > >	vertexToken

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Description: <one line="" class="" summary>="">

Implementation: <Notes on="" implementation>="">

Definition at line 158 of file EvtPlaneProducer.cc.

Constructor & Destructor Documentation

EvtPlaneProducer::EvtPlaneProducer ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 215 of file EvtPlaneProducer.cc.

References caloTag_, caloToken, castorTag_, castorToken, centralityBinTag_, centralityBinToken, centralityLabel_, centralityMC_, centralityVariable_, hi::EPEtaMax1, hi::EPEtaMax2, hi::EPEtaMin1, hi::EPEtaMin2, hi::EPNames, hi::EPOrder, edm::ParameterSet::exists(), flat, FlatOrder_, edm::ParameterSet::getParameter(), mps_fire::i, HiEvtPlaneFlatten::init(), nCentBins_, hi::NumEPNames, NumFlatBins_, rp, AlCaHLTBitMon_QueryRunRegistry::string, trackTag_, trackToken, vertexTag_, and vertexToken.

                                                                :
  centralityVariable_ ( iConfig.getParameter<std::string>("centralityVariable") ),
  centralityBinTag_ ( iConfig.getParameter<edm::InputTag>("centralityBinTag") ),
  vertexTag_  ( iConfig.getParameter<edm::InputTag>("vertexTag") ),
  caloTag_ ( iConfig.getParameter<edm::InputTag>("caloTag") ),
  castorTag_ ( iConfig.getParameter<edm::InputTag>("castorTag") ),
  trackTag_ ( iConfig.getParameter<edm::InputTag>("trackTag") ),
  loadDB_ ( iConfig.getParameter<bool>("loadDB") ),
  minet_ ( iConfig.getParameter<double>("minet") ),
  maxet_ ( iConfig.getParameter<double>("maxet") ),
  minpt_ ( iConfig.getParameter<double>("minpt") ),
  maxpt_ ( iConfig.getParameter<double>("maxpt") ),
  minvtx_ ( iConfig.getParameter<double>("minvtx") ),
  maxvtx_ ( iConfig.getParameter<double>("maxvtx") ),
  dzerr_ ( iConfig.getParameter<double>("dzerr") ),
  chi2_  ( iConfig.getParameter<double>("chi2") ),
  FlatOrder_ ( iConfig.getParameter<int>("FlatOrder") ),
  NumFlatBins_ ( iConfig.getParameter<int>("NumFlatBins") ),
  caloCentRef_ ( iConfig.getParameter<double>("caloCentRef") ),
  caloCentRefWidth_ ( iConfig.getParameter<double>("caloCentRefWidth") ),
  CentBinCompression_ ( iConfig.getParameter<int>("CentBinCompression") )
{

  nCentBins_ = 200.;

  if(iConfig.exists("nonDefaultGlauberModel")){
    centralityMC_ = iConfig.getParameter<std::string>("nonDefaultGlauberModel");
  }
  centralityLabel_ = centralityVariable_+centralityMC_;

  centralityBinToken = consumes<int>(centralityBinTag_);

  vertexToken = consumes<std::vector<reco::Vertex>>(vertexTag_);

  caloToken = consumes<CaloTowerCollection>(caloTag_);

  castorToken = consumes<std::vector<reco::CastorTower>>(castorTag_);

  trackToken = consumes<reco::TrackCollection>(trackTag_);

  produces<reco::EvtPlaneCollection>();
  for(int i = 0; i<NumEPNames; i++ ) {
    rp[i] = new GenPlane(EPNames[i],EPEtaMin1[i],EPEtaMax1[i],EPEtaMin2[i],EPEtaMax2[i],EPOrder[i]);
  }
  for(int i = 0; i<NumEPNames; i++) {
    flat[i] = new HiEvtPlaneFlatten();
    flat[i]->init(FlatOrder_,NumFlatBins_,EPNames[i],EPOrder[i]);
  }

}

EvtPlaneProducer::~EvtPlaneProducer ( )

override

Definition at line 267 of file EvtPlaneProducer.cc.

References flat, mps_fire::i, and hi::NumEPNames.

{

  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
  for(int i = 0; i<NumEPNames; i++) {
    delete flat[i];
  }

}

Member Function Documentation

void EvtPlaneProducer::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 285 of file EvtPlaneProducer.cc.

References hi::GenPlane::addParticle(), stringResolutionProvider_cfi::bin, caloCentRef_, caloCentRefWidth_, caloCollection_, caloToken, castorCollection_, castorToken, CentBinCompression_, centralityBinToken, centralityLabel_, edm::ESWatcher< T >::check(), chi2_, funct::cos(), cuy::cv, allConversions_cfi::d0, DEFINE_FWK_MODULE, PVValHelper::dz, dzerr_, hi::EPDet, hi::EPOrder, flat, edm::EventSetup::get(), hi::GenPlane::getAngle(), edm::Event::getByToken(), HiEvtPlaneFlatten::getEtScale(), HiEvtPlaneFlatten::getW(), CaloTPGTranscoder_cfi::HF, hirpWatcher, hiWatcher, mps_fire::i, fastTrackerRecHitType::isPixel(), LoadEPDB::IsSuccess(), edm::HandleBase::isValid(), loadDB_, CentralityTable::m_table, HiEvtPlane_cfi::maxet, maxet_, HiEvtPlane_cfi::maxpt, maxpt_, hi::maxTransverse, maxvtx_, HiEvtPlane_cfi::minet, minet_, HiEvtPlane_cfi::minpt, minpt_, hi::minTransverse, hi::MomConsWeight, eostools::move(), nCentBins_, hi::NumEPNames, edm::Handle< T >::product(), edm::Event::put(), reset(), rp, HiEvtPlaneFlatten::setCaloCentRefBins(), funct::sin(), mathSSE::sqrt(), pfDeepBoostedJetPreprocessParams_cfi::sv, trackCollection_, align::Tracker, trackToken, parallelization::uint(), vertex_, vertexToken, and w.

{
  using namespace edm;
  using namespace std;
  using namespace reco;

  if( loadDB_ && (hiWatcher.check(iSetup) || hirpWatcher.check(iSetup)) ) {
    //
    //Get Size of Centrality Table
    //
    edm::ESHandle<CentralityTable> centDB_;
    iSetup.get<HeavyIonRcd>().get(centralityLabel_,centDB_);
    nCentBins_ = centDB_->m_table.size();
    for(int i = 0; i<NumEPNames; i++) {
      if(caloCentRef_>0) {
    int minbin = (caloCentRef_-caloCentRefWidth_/2.)*nCentBins_/100.;
    int maxbin = (caloCentRef_+caloCentRefWidth_/2.)*nCentBins_/100.;
    minbin/=CentBinCompression_;
    maxbin/=CentBinCompression_;
    if(minbin>0 && maxbin>=minbin) {
      if(EPDet[i]==HF || EPDet[i]==Castor) flat[i]->setCaloCentRefBins(minbin,maxbin);
    }
      }
    }

    //
    //Get flattening parameter file.  
    //
    if ( loadDB_ ) {
    edm::ESHandle<RPFlatParams> flatparmsDB_;
    iSetup.get<HeavyIonRPRcd>().get(flatparmsDB_);
    LoadEPDB db(flatparmsDB_,flat);
    if(!db.IsSuccess()) {
      loadDB_ = kFALSE;
    }
    }

  } //rp record change

  //
  //Get Centrality
  //
  int bin = 0;
  if(loadDB_) {
    edm::Handle<int> cbin_;
    iEvent.getByToken(centralityBinToken, cbin_);
    int cbin = *cbin_;
    bin = cbin/CentBinCompression_;
  }
  //
  //Get Vertex
  //
  int vs_sell = 0.;
  float vzr_sell;
  iEvent.getByToken(vertexToken,vertex_);
  const reco::VertexCollection * vertices3 = nullptr;
  if ( vertex_.isValid() ) {
    vertices3 = vertex_.product();
    vs_sell = vertices3->size();
  }
  if(vs_sell>0) {
    vzr_sell = vertices3->begin()->z();
  } else
    vzr_sell = -999.9;
  //
  for(int i = 0; i<NumEPNames; i++) rp[i]->reset();
  if(vzr_sell<minvtx_ or vzr_sell>maxvtx_) return;

    //calorimetry part

    double tower_eta, tower_phi;
    double tower_energyet, tower_energyet_e, tower_energyet_h;

    iEvent.getByToken(caloToken,caloCollection_);

    if(caloCollection_.isValid()){
      for (CaloTowerCollection::const_iterator j = caloCollection_->begin();j !=caloCollection_->end(); j++) {
    tower_eta        = j->eta();
    tower_phi        = j->phi();
    tower_energyet_e   = j->emEt();
    tower_energyet_h   = j->hadEt();
    tower_energyet     = tower_energyet_e + tower_energyet_h;
    double minet = minet_;
    double maxet = maxet_;
    for(int i = 0; i<NumEPNames; i++) {
      if(minet_<0) minet = minTransverse[i];
      if(maxet_<0) maxet = maxTransverse[i];
      if(tower_energyet<minet) continue;
      if(tower_energyet>maxet) continue;
      if(EPDet[i]==HF) {
        double w = tower_energyet;
        if(loadDB_) w = tower_energyet*flat[i]->getEtScale(vzr_sell,bin);
        if(EPOrder[i]==1 ) {
          if(MomConsWeight[i][0]=='y' && loadDB_ ) {
        w = flat[i]->getW(tower_energyet, vzr_sell, bin);
          }
          if(tower_eta<0 ) w=-w;
        }
        rp[i]->addParticle(w,tower_energyet,sin(EPOrder[i]*tower_phi),cos(EPOrder[i]*tower_phi),tower_eta);
      }
    }
      }
    }

    //Castor part


    iEvent.getByToken(castorToken,castorCollection_);

    if(castorCollection_.isValid()){
      for (std::vector<reco::CastorTower>::const_iterator j = castorCollection_->begin();j !=castorCollection_->end(); j++) {
        tower_eta        = j->eta();
        tower_phi        = j->phi();
        tower_energyet     = j->et();
    double minet = minet_;
    double maxet = maxet_;
        for(int i = 0; i<NumEPNames; i++) {
          if(EPDet[i]==Castor) {
        if(minet_<0) minet = minTransverse[i];
        if(maxet_<0) maxet = maxTransverse[i];
        if(tower_energyet<minet) continue;
        if(tower_energyet>maxet) continue;
            double w = tower_energyet;
            if(EPOrder[i]==1 ) {
              if(MomConsWeight[i][0]=='y' && loadDB_ ) {
            w = flat[i]->getW(tower_energyet, vzr_sell, bin);
              }
              if(tower_eta<0 ) w=-w;
            }
            rp[i]->addParticle(w,tower_energyet,sin(EPOrder[i]*tower_phi),cos(EPOrder[i]*tower_phi),tower_eta);
          }
        }
      }
    }

    //Tracking part

    double track_eta;
    double track_phi;
    double track_pt;

    double vzErr2 =0.0, vxyErr=0.0;
    math::XYZPoint vtxPoint(0.0,0.0,0.0);
    if(vertex_.isValid() && !vertex_->empty()) {
        vtxPoint=vertex_->begin()->position();
        vzErr2= (vertex_->begin()->zError())*(vertex_->begin()->zError());
        vxyErr=vertex_->begin()->xError() * vertex_->begin()->yError();
    }

    iEvent.getByToken(trackToken, trackCollection_);
    if(trackCollection_.isValid()){
      for(reco::TrackCollection::const_iterator j = trackCollection_->begin(); j != trackCollection_->end(); j++){
    bool accepted = true;
    bool isPixel = false;
    // determine if the track is a pixel track
    if ( j->numberOfValidHits() < 7 ) isPixel = true;

    // determine the vertex significance 
    double d0=0.0, dz=0.0, d0sigma=0.0, dzsigma=0.0;
    d0 = -1.*j->dxy(vtxPoint);
    dz = j->dz(vtxPoint);
    d0sigma = sqrt(j->d0Error()*j->d0Error()+vxyErr);
    dzsigma = sqrt(j->dzError()*j->dzError()+vzErr2);

    // cuts for pixel tracks
    if( isPixel ){
      // dz significance cut 
      if ( fabs(dz/dzsigma) > dzerr_ ) accepted = false;
      // chi2/ndof cut 
      if ( j->normalizedChi2() > chi2_ ) accepted = false;
    }
    // cuts for full tracks
    if ( ! isPixel) {
      // dz and d0 significance cuts 
      if ( fabs(dz/dzsigma) > 3 ) accepted = false;
      if ( fabs(d0/d0sigma) > 3 ) accepted = false;
      // pt resolution cut
      if ( j->ptError()/j->pt() > 0.1 ) accepted = false;
      // number of valid hits cut
      if ( j->numberOfValidHits() < 12 ) accepted = false;
    }
    if( accepted ) {
      track_eta = j->eta();
      track_phi = j->phi();
      track_pt = j->pt();
      double minpt = minpt_;
      double maxpt = maxpt_;
      for(int i = 0; i<NumEPNames; i++) {
        if(minpt_<0) minpt = minTransverse[i];
        if(maxpt_<0) maxpt = maxTransverse[i];
        if(track_pt<minpt) continue;
        if(track_pt>maxpt) continue;
        if(EPDet[i]==Tracker) {
          double w = track_pt;
          if(w>2.5) w=2.0;   //v2 starts decreasing above ~2.5 GeV/c
          if(EPOrder[i]==1) {
        if(MomConsWeight[i][0]=='y' && loadDB_) {
          w = flat[i]->getW(track_pt, vzr_sell, bin);
        }
        if(track_eta<0) w=-w;
          }
          rp[i]->addParticle(w,track_pt,sin(EPOrder[i]*track_phi),cos(EPOrder[i]*track_phi),track_eta);
        }
      }
    }
      } //end for
    }

    auto evtplaneOutput = std::make_unique<EvtPlaneCollection>();

    double ang=-10;
    double sv = 0;
    double cv = 0;
    double svNoWgt = 0;
    double cvNoWgt = 0;

    double wv = 0;
    double wv2 = 0;
    double pe = 0;
    double pe2 = 0;
    uint epmult = 0;

    for(int i = 0; i<NumEPNames; i++) {
      rp[i]->getAngle(ang,sv,cv,svNoWgt, cvNoWgt, wv,wv2,pe,pe2,epmult);
      evtplaneOutput->push_back( EvtPlane(i,0,ang,sv,cv,wv,wv2,pe,pe2,epmult) );
      evtplaneOutput->back().addLevel(3, 0., svNoWgt, cvNoWgt);
    }

    iEvent.put(std::move(evtplaneOutput));
}

Member Data Documentation

double EvtPlaneProducer::caloCentRef_

private

Definition at line 209 of file EvtPlaneProducer.cc.

Referenced by produce().

double EvtPlaneProducer::caloCentRefWidth_

private

Definition at line 210 of file EvtPlaneProducer.cc.

Referenced by produce().

edm::Handle<CaloTowerCollection> EvtPlaneProducer::caloCollection_

private

Definition at line 184 of file EvtPlaneProducer.cc.

Referenced by produce().

edm::InputTag EvtPlaneProducer::caloTag_

private

Definition at line 182 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer().

edm::EDGetTokenT<CaloTowerCollection> EvtPlaneProducer::caloToken

private

Definition at line 183 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer(), and produce().

edm::Handle<std::vector<reco::CastorTower> > EvtPlaneProducer::castorCollection_

private

Definition at line 188 of file EvtPlaneProducer.cc.

Referenced by produce().

edm::InputTag EvtPlaneProducer::castorTag_

private

Definition at line 186 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer().

edm::EDGetTokenT<std::vector<reco::CastorTower> > EvtPlaneProducer::castorToken

private

Definition at line 187 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer(), and produce().

int EvtPlaneProducer::CentBinCompression_

private

Definition at line 211 of file EvtPlaneProducer.cc.

Referenced by produce().

edm::InputTag EvtPlaneProducer::centralityBinTag_

private

Definition at line 174 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer().

edm::EDGetTokenT<int> EvtPlaneProducer::centralityBinToken

private

Definition at line 175 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer(), and produce().

std::string EvtPlaneProducer::centralityLabel_

private

Definition at line 171 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer(), and produce().

std::string EvtPlaneProducer::centralityMC_

private

Definition at line 172 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer().

std::string EvtPlaneProducer::centralityVariable_

private

Definition at line 170 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer().

double EvtPlaneProducer::chi2_

private

Definition at line 205 of file EvtPlaneProducer.cc.

Referenced by produce().

double EvtPlaneProducer::dzerr_

private

Definition at line 204 of file EvtPlaneProducer.cc.

Referenced by produce().

HiEvtPlaneFlatten* EvtPlaneProducer::flat[NumEPNames]

private

Definition at line 212 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer(), produce(), and ~EvtPlaneProducer().

int EvtPlaneProducer::FlatOrder_

private

Definition at line 206 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer().

edm::ESWatcher<HeavyIonRPRcd> EvtPlaneProducer::hirpWatcher

private

Definition at line 195 of file EvtPlaneProducer.cc.

Referenced by produce().

edm::ESWatcher<HeavyIonRcd> EvtPlaneProducer::hiWatcher

private

Definition at line 194 of file EvtPlaneProducer.cc.

Referenced by produce().

bool EvtPlaneProducer::loadDB_

private

Definition at line 197 of file EvtPlaneProducer.cc.

Referenced by produce().

double EvtPlaneProducer::maxet_

private

Definition at line 199 of file EvtPlaneProducer.cc.

Referenced by produce().

double EvtPlaneProducer::maxpt_

private

Definition at line 201 of file EvtPlaneProducer.cc.

Referenced by produce().

double EvtPlaneProducer::maxvtx_

private

Definition at line 203 of file EvtPlaneProducer.cc.

Referenced by produce().

double EvtPlaneProducer::minet_

private

Definition at line 198 of file EvtPlaneProducer.cc.

Referenced by produce().

double EvtPlaneProducer::minpt_

private

Definition at line 200 of file EvtPlaneProducer.cc.

Referenced by produce().

double EvtPlaneProducer::minvtx_

private

Definition at line 202 of file EvtPlaneProducer.cc.

double EvtPlaneProducer::nCentBins_

private

Definition at line 208 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer(), and produce().

int EvtPlaneProducer::NumFlatBins_

private

Definition at line 207 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer().

GenPlane* EvtPlaneProducer::rp[NumEPNames]

private

Definition at line 164 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer(), and produce().

edm::Handle<reco::TrackCollection> EvtPlaneProducer::trackCollection_

private

Definition at line 192 of file EvtPlaneProducer.cc.

Referenced by produce().

edm::InputTag EvtPlaneProducer::trackTag_

private

Definition at line 190 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer().

edm::EDGetTokenT<reco::TrackCollection> EvtPlaneProducer::trackToken

private

Definition at line 191 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer(), and produce().

edm::Handle<std::vector<reco::Vertex> > EvtPlaneProducer::vertex_

private

Definition at line 180 of file EvtPlaneProducer.cc.

Referenced by produce().

edm::InputTag EvtPlaneProducer::vertexTag_

private

Definition at line 178 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer().

edm::EDGetTokenT<std::vector<reco::Vertex> > EvtPlaneProducer::vertexToken

private

Definition at line 179 of file EvtPlaneProducer.cc.

Referenced by EvtPlaneProducer(), and produce().