MuonSeedProducer.cc

Go to the documentation of this file.

#include "RecoMuon/MuonSeedGenerator/plugins/MuonSeedProducer.h"
#include "RecoMuon/MuonSeedGenerator/src/MuonSeedBuilder.h"


// Data Formats 
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"

// Magnetic Field
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

// Geometry
#include "Geometry/CommonDetUnit/interface/GeomDet.h"
#include "TrackingTools/DetLayers/interface/DetLayer.h"

#include "RecoMuon/DetLayers/interface/MuonDetLayerGeometry.h"
#include "RecoMuon/Records/interface/MuonRecoGeometryRecord.h"

// Framework
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include <DataFormats/Common/interface/Handle.h>
// C++
#include <vector>



/* 
 * Constructor
 */
MuonSeedProducer::MuonSeedProducer(const edm::ParameterSet& pset){

  // Register what this produces
  produces<TrajectorySeedCollection>(); 

  // Local Debug flag
  debug              = pset.getParameter<bool>("DebugMuonSeed");

  // Builder which returns seed collection 
  muonSeedBuilder_   = new MuonSeedBuilder( pset ); 

}


/*
 * Destructor
 */
MuonSeedProducer::~MuonSeedProducer(){

  delete muonSeedBuilder_;

}


/*
 * Producer (the main)
 */ 
void MuonSeedProducer::produce(edm::Event& event, const edm::EventSetup& eSetup){

  // Muon Geometry
  edm::ESHandle<MuonDetLayerGeometry> muonLayers;
  eSetup.get<MuonRecoGeometryRecord>().get(muonLayers);
  const MuonDetLayerGeometry* lgeom = &*muonLayers;
  muonSeedBuilder_->setGeometry( lgeom );

  // Magnetic field
  edm::ESHandle<MagneticField> field;
  eSetup.get<IdealMagneticFieldRecord>().get(field);
  const MagneticField* theField = &*field;
  muonSeedBuilder_->setBField( theField );

   // Create pointer to the seed container

  std::auto_ptr<TrajectorySeedCollection> output(new TrajectorySeedCollection() );

  int nSeeds = 0;
  nSeeds = muonSeedBuilder_->build( event, eSetup, *output);

  // Append muon seed collection to event
  event.put( output );

}