#include <FamosProducer.h>

Inheritance diagram for FamosProducer:

Public Member Functions
virtual void	beginRun (edm::Run &run, const edm::EventSetup &es)
virtual void	endJob ()
	FamosProducer (edm::ParameterSet const &p)
virtual void	produce (edm::Event &e, const edm::EventSetup &c)
virtual	~FamosProducer ()
Private Attributes
HepMC::GenEvent *	evt_
FamosManager *	famosManager_
bool	simulateMuons
edm::InputTag	theBeamSpotLabel
edm::InputTag	theGenParticleLabel
edm::InputTag	theSourceLabel

Detailed Description

Definition at line 15 of file FamosProducer.h.

Constructor & Destructor Documentation

FamosProducer::FamosProducer ( edm::ParameterSet const & p ) [explicit]

Definition at line 33 of file FamosProducer.cc.

References famosManager_, edm::ParameterSet::getParameter(), simulateMuons, theBeamSpotLabel, theGenParticleLabel, and theSourceLabel.

{    

    produces<edm::SimTrackContainer>();
    produces<edm::SimVertexContainer>();
    produces<FSimVertexTypeCollection>("VertexTypes");
    produces<edm::PSimHitContainer>("TrackerHits");
    produces<edm::PCaloHitContainer>("EcalHitsEB");
    produces<edm::PCaloHitContainer>("EcalHitsEE");
    produces<edm::PCaloHitContainer>("EcalHitsES");
    produces<edm::PCaloHitContainer>("HcalHits");
    // Temporary facility to allow for the crossing frame to work...
    simulateMuons = p.getParameter<bool>("SimulateMuons");
    if ( simulateMuons ) produces<edm::SimTrackContainer>("MuonSimTracks");

    // The generator input label
    theSourceLabel = p.getParameter<edm::InputTag>("SourceLabel");
    theGenParticleLabel = p.getParameter<edm::InputTag>("GenParticleLabel");
    theBeamSpotLabel = p.getParameter<edm::InputTag>("BeamSpotLabel");

    famosManager_ = new FamosManager(p);

}

FamosProducer::~FamosProducer ( ) [virtual]

Definition at line 57 of file FamosProducer.cc.

References famosManager_.

{ if ( famosManager_ ) delete famosManager_; }

Member Function Documentation

void FamosProducer::beginRun	(	edm::Run &	run,
		const edm::EventSetup &	es
	)		`[virtual]`

Reimplemented from edm::EDProducer.

Definition at line 61 of file FamosProducer.cc.

References famosManager_, and FamosManager::setupGeometryAndField().

                                                              {
  famosManager_->setupGeometryAndField(run,es);
}

void FamosProducer::endJob ( void ) [virtual]

Reimplemented from edm::EDProducer.

Definition at line 65 of file FamosProducer.cc.

{ 
}

void FamosProducer::produce	(	edm::Event &	e,
		const edm::EventSetup &	c
	)		`[virtual]`

Implements edm::EDProducer.

Definition at line 69 of file FamosProducer.cc.

References FamosManager::calorimetryManager(), gather_cfg::cout, famosManager_, TtGenEvtProducer_cfi::genEvt, edm::Event::getByLabel(), p1, p2, p3, p4, edm::Event::put(), FamosManager::reconstruct(), FBaseSimEvent::setBeamSpot(), FamosManager::simEvent(), simulateMuons, LaserTracksInput_cfi::source, theBeamSpotLabel, theGenParticleLabel, theSourceLabel, patCandidatesForDimuonsSequences_cff::tracker, and FamosManager::trackerManager().

{
   using namespace edm;

   // The beam spot position
   edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
   iEvent.getByLabel(theBeamSpotLabel,recoBeamSpotHandle); 
   math::XYZPoint BSPosition_ = recoBeamSpotHandle->position();

   // Get the generated event(s) from the edm::Event
   // 1. Check if a HepMCProduct exists
   //    a. Take the VtxSmeared if it exists
   //    b. Take the source  otherwise
   // 2. Otherwise go for the CandidateCollection
   Handle<HepMCProduct> theHepMCProduct;

   const HepMC::GenEvent* myGenEvent = 0;
   FSimEvent* fevt = famosManager_->simEvent();
   fevt->setBeamSpot(BSPosition_);
   PrimaryVertexGenerator* theVertexGenerator = fevt->thePrimaryVertexGenerator();

   // Get the generated signal event
   bool source = iEvent.getByLabel(theSourceLabel,theHepMCProduct);
   if ( source ) { 
     myGenEvent = theHepMCProduct->GetEvent();
     // First rotate in case of beam crossing angle (except if done already)
     if ( theVertexGenerator ) { 
       TMatrixD* boost = theVertexGenerator->boost();
       if ( boost ) theHepMCProduct->boostToLab(boost,"momentum");
     }          
     myGenEvent = theHepMCProduct->GetEvent();
   } 

   // In case there is no HepMCProduct, seek a genParticle Candidate Collection
   bool genPart = false;
   const reco::GenParticleCollection* myGenParticles = 0;
   if ( !myGenEvent ) { 
     // Look for the particle CandidateCollection
     Handle<reco::GenParticleCollection> genEvt;
     genPart = iEvent.getByLabel(theGenParticleLabel,genEvt);
     if ( genPart ) myGenParticles = &(*genEvt);
   } 

   if ( !myGenEvent && !genPart )
     std::cout << "There is no generator input for this event, under " 
               << "any form (HepMCProduct, genParticles)" << std::endl
               << "Please check SourceLabel or GenParticleLabel" << std::endl;

   // Get the pile-up events from the pile-up producer
   // There might be no pile-up events, by the way, in that case, just continue
   
   Handle<HepMCProduct> thePileUpEvents;
   bool isPileUp = iEvent.getByLabel("famosPileUp","PileUpEvents",thePileUpEvents);
   const HepMC::GenEvent* thePUEvents = isPileUp ? thePileUpEvents->GetEvent() : 0;

   // .and pass the event to the Famos Manager for propagation and simulation
   if ( myGenEvent || myGenParticles ) 
     famosManager_->reconstruct(myGenEvent,myGenParticles,thePUEvents);
   
   // Set the vertex back to the HepMCProduct (except if it was smeared already)
   if ( myGenEvent ) { 
     if ( theVertexGenerator ) { 
       HepMC::FourVector theVertex(
        (theVertexGenerator->X()-theVertexGenerator->beamSpot().X()+BSPosition_.X())*10.,
        (theVertexGenerator->Y()-theVertexGenerator->beamSpot().Y()+BSPosition_.Y())*10.,
        (theVertexGenerator->Z()-theVertexGenerator->beamSpot().Z()+BSPosition_.Z())*10.,
                                   0.);
       if ( fabs(theVertexGenerator->Z()) > 1E-10 ) theHepMCProduct->applyVtxGen( &theVertex );
     }
   }
   
   CalorimetryManager * calo = famosManager_->calorimetryManager();
   TrajectoryManager * tracker = famosManager_->trackerManager();

   // Save everything in the edm::Event
   std::auto_ptr<edm::SimTrackContainer> p1(new edm::SimTrackContainer);
   std::auto_ptr<edm::SimTrackContainer> m1(new edm::SimTrackContainer);
   std::auto_ptr<edm::SimVertexContainer> p2(new edm::SimVertexContainer);
   std::auto_ptr<FSimVertexTypeCollection> v1(new FSimVertexTypeCollection);
   std::auto_ptr<edm::PSimHitContainer> p3(new edm::PSimHitContainer);
   std::auto_ptr<edm::PCaloHitContainer> p4(new edm::PCaloHitContainer);
   std::auto_ptr<edm::PCaloHitContainer> p5(new edm::PCaloHitContainer);
   std::auto_ptr<edm::PCaloHitContainer> p6(new edm::PCaloHitContainer); 
   std::auto_ptr<edm::PCaloHitContainer> p7(new edm::PCaloHitContainer);

   fevt->load(*p1,*m1);
   fevt->load(*p2);
   fevt->load(*v1);
   //   fevt->print();
   tracker->loadSimHits(*p3);

   //   fevt->print();

   if ( calo ) {  
     calo->loadFromEcalBarrel(*p4);
     calo->loadFromEcalEndcap(*p5);
     calo->loadFromPreshower(*p6);
     calo->loadFromHcal(*p7);
     // update the muon SimTracks
     calo->loadMuonSimTracks(*m1);
   }

   // Write muon first, to allow tracking particles to work... (pending MixingModule fix)
   if ( simulateMuons ) iEvent.put(m1,"MuonSimTracks");
   iEvent.put(p1);
   iEvent.put(p2);
   iEvent.put(p3,"TrackerHits");
   iEvent.put(v1,"VertexTypes");
   iEvent.put(p4,"EcalHitsEB");
   iEvent.put(p5,"EcalHitsEE");
   iEvent.put(p6,"EcalHitsES");
   iEvent.put(p7,"HcalHits");

}