00001 #include "SeedGeneratorFromRegionHitsEDProducer.h"
00002
00003 #include "FWCore/Framework/interface/Event.h"
00004 #include "FWCore/Framework/interface/EventSetup.h"
00005
00006 #include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
00007
00008 #include "RecoTracker/TkTrackingRegions/interface/TrackingRegionProducerFactory.h"
00009 #include "RecoTracker/TkTrackingRegions/interface/TrackingRegionProducer.h"
00010 #include "RecoTracker/TkTrackingRegions/interface/TrackingRegion.h"
00011
00012 #include "RecoTracker/TkTrackingRegions/interface/OrderedHitsGeneratorFactory.h"
00013 #include "RecoTracker/TkTrackingRegions/interface/OrderedHitsGenerator.h"
00014
00015 #include "RecoTracker/TkSeedingLayers/interface/SeedComparitorFactory.h"
00016 #include "RecoTracker/TkSeedingLayers/interface/SeedComparitor.h"
00017
00018 #include "RecoTracker/TkSeedGenerator/interface/SeedGeneratorFromRegionHits.h"
00019
00020
00021 SeedGeneratorFromRegionHitsEDProducer::SeedGeneratorFromRegionHitsEDProducer(
00022 const edm::ParameterSet& cfg)
00023 : theConfig(cfg), theGenerator(0), theRegionProducer(0)
00024 {
00025 produces<TrajectorySeedCollection>();
00026 }
00027
00028 SeedGeneratorFromRegionHitsEDProducer::~SeedGeneratorFromRegionHitsEDProducer()
00029 {
00030 }
00031
00032 void SeedGeneratorFromRegionHitsEDProducer::endRun(edm::Run &run, const edm::EventSetup& es) {
00033 delete theRegionProducer;
00034 delete theGenerator;
00035 }
00036
00037 void SeedGeneratorFromRegionHitsEDProducer::beginRun(edm::Run &run, const edm::EventSetup& es)
00038 {
00039 edm::ParameterSet regfactoryPSet =
00040 theConfig.getParameter<edm::ParameterSet>("RegionFactoryPSet");
00041 std::string regfactoryName = regfactoryPSet.getParameter<std::string>("ComponentName");
00042 theRegionProducer = TrackingRegionProducerFactory::get()->create(regfactoryName,regfactoryPSet);
00043
00044 edm::ParameterSet hitsfactoryPSet =
00045 theConfig.getParameter<edm::ParameterSet>("OrderedHitsFactoryPSet");
00046 std::string hitsfactoryName = hitsfactoryPSet.getParameter<std::string>("ComponentName");
00047 OrderedHitsGenerator* hitsGenerator =
00048 OrderedHitsGeneratorFactory::get()->create( hitsfactoryName, hitsfactoryPSet);
00049
00050 edm::ParameterSet comparitorPSet =
00051 theConfig.getParameter<edm::ParameterSet>("SeedComparitorPSet");
00052 std::string comparitorName = comparitorPSet.getParameter<std::string>("ComponentName");
00053 SeedComparitor * aComparitor = (comparitorName == "none") ?
00054 0 : SeedComparitorFactory::get()->create( comparitorName, comparitorPSet);
00055
00056 theGenerator = new SeedGeneratorFromRegionHits(hitsGenerator, theConfig, aComparitor);
00057
00058 }
00059
00060 void SeedGeneratorFromRegionHitsEDProducer::produce(edm::Event& ev, const edm::EventSetup& es)
00061 {
00062 std::auto_ptr<TrajectorySeedCollection> result(new TrajectorySeedCollection());
00063
00064 typedef std::vector<TrackingRegion* > Regions;
00065 typedef Regions::const_iterator IR;
00066 Regions regions = theRegionProducer->regions(ev,es);
00067
00068 for (IR ir=regions.begin(), irEnd=regions.end(); ir < irEnd; ++ir) {
00069 const TrackingRegion & region = **ir;
00070
00071
00072
00073 theGenerator->run(*result, region, ev,es);
00074 }
00075
00076
00077 for (IR ir=regions.begin(), irEnd=regions.end(); ir < irEnd; ++ir) delete (*ir);
00078
00079
00080 ev.put(result);
00081 }