Inheritance diagram for MixEvtVtxGenerator:

Public Member Functions
virtual HepMC::FourVector *	getRecVertex (edm::Event &)

virtual HepMC::FourVector *	getVertex (edm::Event &)

	MixEvtVtxGenerator (const edm::ParameterSet &)

virtual void	produce (edm::Event &, const edm::EventSetup &) override

virtual	~MixEvtVtxGenerator ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Protected Attributes
TMatrixD *	boost_

HepMC::FourVector *	fVertex

Private Attributes
edm::EDGetTokenT < CrossingFrame< HepMCProduct > >	cfLabel

edm::EDGetTokenT < reco::VertexCollection >	hiLabel

edm::EDGetTokenT< HepMCProduct >	signalLabel

bool	useCF_

bool	useRecVertex

std::vector< double >	vtxOffset

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 34 of file MixEvtVtxGenerator.cc.

Constructor & Destructor Documentation

MixEvtVtxGenerator::MixEvtVtxGenerator ( const edm::ParameterSet & pset )

explicit

Definition at line 64 of file MixEvtVtxGenerator.cc.

References cfLabel, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), hiLabel, signalLabel, useCF_, useRecVertex, and vtxOffset.

     : fVertex(0), boost_(0),
       useRecVertex(pset.exists("useRecVertex")?pset.getParameter<bool>("useRecVertex"):false)
       
 {   
    produces<bool>("matchedVertex"); 
    vtxOffset.resize(3);
    if(pset.exists("vtxOffset")) vtxOffset=pset.getParameter< std::vector<double> >("vtxOffset");
 
    if(useRecVertex) useCF_ = 0;
    else{
      useCF_ = pset.getUntrackedParameter<bool>("useCF",false);
      cfLabel = consumes<CrossingFrame<HepMCProduct> >(pset.getUntrackedParameter<edm::InputTag>("mixLabel",edm::InputTag("mixGen","generator")));
    }
    signalLabel = consumes<HepMCProduct>(pset.getParameter<edm::InputTag>("signalLabel"));
    hiLabel = consumes<reco::VertexCollection>(pset.getParameter<edm::InputTag>("heavyIonLabel"));
 
 
 }

MixEvtVtxGenerator::~MixEvtVtxGenerator ( )

virtual

Definition at line 84 of file MixEvtVtxGenerator.cc.

References boost_, and fVertex.

 {
    delete fVertex ;
    if (boost_ != 0 ) delete boost_;
    // no need since now it's done in HepMCProduct
    // delete fEvt ;
 }

Member Function Documentation

HepMC::FourVector * MixEvtVtxGenerator::getRecVertex ( edm::Event & evt )

virtual

Definition at line 151 of file MixEvtVtxGenerator.cc.

References fVertex, edm::Event::getByToken(), hiLabel, input, and vtxOffset.

Referenced by produce().

                                                             {
 
   Handle<reco::VertexCollection> input;
   evt.getByToken(hiLabel,input);
 
   double aX,aY,aZ;
 
   aX = input->begin()->position().x() + vtxOffset[0];
   aY = input->begin()->position().y() + vtxOffset[1];
   aZ = input->begin()->position().z() + vtxOffset[2];
 
   /*
   std::cout << "reco::Vertex = " << input->begin()->position().x()
         << ", " << input->begin()->position().y()
         << ", " << input->begin()->position().z()
         << std::endl;
 
   std::cout << "offset = " << vtxOffset[0]
         << ", " << vtxOffset[1]
         << ", " << vtxOffset[2]
         << std::endl;
 
   std::cout << "embedded GEN vertex = " << aX
         << ", " << aY << ", " << aZ << std::endl;
 
   */
   if(!fVertex) fVertex = new HepMC::FourVector();
   fVertex->set(10.0*aX,10.0*aY,10.0*aZ,0.0); // HepMC positions in mm (RECO in cm)
   
   return fVertex;
 
 }

HepMC::FourVector * MixEvtVtxGenerator::getVertex ( edm::Event & evt )

virtual

Definition at line 92 of file MixEvtVtxGenerator.cc.

References cfLabel, gather_cfg::cout, fVertex, edm::Event::getByToken(), edm::HepMCProduct::GetEvent(), input, edm::HepMCProduct::isVtxGenApplied(), reco_application_tbsim_DetSim-Digi_cfg::mix, edm::Handle< T >::product(), EnergyCorrector::pt, signalLabel, and useCF_.

Referenced by produce().

                                                          {
 
   HepMC::GenVertex* genvtx = 0;
   const HepMC::GenEvent* inev = 0;
 
   //cout<<" use CF "<<useCF_<<endl;
   
   if(useCF_){
     Handle<CrossingFrame<HepMCProduct> > cf;
     evt.getByToken(cfLabel,cf);
     MixCollection<HepMCProduct> mix(cf.product());
     if(mix.size() < 2){
       cout<<"Less than 2 sub-events, mixing seems to have failed!"<<endl;
     }
     const HepMCProduct& bkg = mix.getObject(1);
     if(!(bkg.isVtxGenApplied())){
       cout<<"Input does not have smeared vertex!"<<endl;
     }else{
       inev = bkg.GetEvent();
    } 
   }else{
     //cout<<" hiLabel "<<hiLabel<<endl;
     Handle<HepMCProduct> input;
     evt.getByToken(signalLabel,input);
     inev = input->GetEvent();
   }
 
   genvtx = inev->signal_process_vertex();
   if(!genvtx){
     //cout<<"No Signal Process Vertex!"<<endl;
     HepMC::GenEvent::particle_const_iterator pt=inev->particles_begin();
     HepMC::GenEvent::particle_const_iterator ptend=inev->particles_end();
     while(!genvtx || ( genvtx->particles_in_size() == 1 && pt != ptend ) ){
       //if(!genvtx) cout<<"No Gen Vertex!"<<endl;
       if(pt == ptend) cout<<"End reached, No Gen Vertex!"<<endl;
       genvtx = (*pt)->production_vertex();
       ++pt;
     }
   }
   double aX,aY,aZ,aT;
 
   aX = genvtx->position().x();
   aY = genvtx->position().y();
   aZ = genvtx->position().z();
   aT = genvtx->position().t();
   
   if(!fVertex){
     fVertex = new HepMC::FourVector();
     //cout<<" creating new vertex "<<endl;
   }
   //cout<<" setting vertex "<<" aX "<<aX<<" aY "<<aY<<" aZ "<<aZ<<" aT "<<aT<<endl;
   fVertex->set(aX,aY,aZ,aT);
   
 
   return fVertex;
 
 }

void MixEvtVtxGenerator::produce	(	edm::Event &	evt,
		const edm::EventSetup &
	)

overridevirtual

Implements edm::EDProducer.

Definition at line 184 of file MixEvtVtxGenerator.cc.

References edm::Event::getByToken(), getRecVertex(), getVertex(), edm::Event::put(), mathSSE::return(), signalLabel, and useRecVertex.

 {
    
    
    Handle<HepMCProduct> HepMCEvt ;
    
    evt.getByToken( signalLabel, HepMCEvt ) ;
    
    // generate new vertex & apply the shift 
    //
 
    HepMCEvt->applyVtxGen( useRecVertex ? getRecVertex(evt) : getVertex(evt) ) ;
 
    //   HepMCEvt->boostToLab( GetInvLorentzBoost(), "vertex" );
    //   HepMCEvt->boostToLab( GetInvLorentzBoost(), "momentum" );
    
    // OK, create a (pseudo)product and put in into edm::Event
    //
    auto_ptr<bool> NewProduct(new bool(true)) ;      
    evt.put( NewProduct ,"matchedVertex") ;
       
    return ;
 
 }