|
|
|
#include <SETMuonSeedProducer.h>
Public Types | |
| typedef MuonTransientTrackingRecHit::MuonRecHitContainer | MuonRecHitContainer |
| typedef std::vector< Trajectory * > | TrajectoryContainer |
Public Member Functions | |
| virtual void | produce (edm::Event &, const edm::EventSetup &) |
| SETMuonSeedProducer (const edm::ParameterSet &) | |
| Constructor with Parameter set. | |
| virtual | ~SETMuonSeedProducer () |
| Destructor. | |
Private Member Functions | |
| SETFilter * | filter () const |
| pre-filter | |
| void | setEvent (const edm::Event &) |
Private Attributes | |
| bool | apply_prePruning |
| edm::InputTag | theBeamSpotTag |
| SETFilter * | theFilter |
| SETPatternRecognition | thePatternRecognition |
| SETSeedFinder | theSeedFinder |
| MuonServiceProxy * | theService |
| bool | useSegmentsInTrajectory |
I. Bloch, E. James, S. Stoynev
Definition at line 29 of file SETMuonSeedProducer.h.
Definition at line 32 of file SETMuonSeedProducer.h.
| typedef std::vector<Trajectory*> SETMuonSeedProducer::TrajectoryContainer |
Definition at line 33 of file SETMuonSeedProducer.h.
| SETMuonSeedProducer::SETMuonSeedProducer | ( | const edm::ParameterSet & | parameterSet | ) |
Constructor with Parameter set.
Definition at line 31 of file SETMuonSeedProducer.cc.
References edm::ParameterSet::addUntrackedParameter(), apply_prePruning, edm::ParameterSet::getParameter(), LogTrace, metname, MuonServiceProxy_cff::MuonServiceProxy, SETPatternRecognition::setServiceProxy(), SETSeedFinder::setServiceProxy(), theFilter, thePatternRecognition, theSeedFinder, theService, and useSegmentsInTrajectory.
: thePatternRecognition(parameterSet), theSeedFinder(parameterSet), theBeamSpotTag(parameterSet.getParameter<edm::InputTag>("beamSpotTag")) { const string metname = "Muon|RecoMuon|SETMuonSeedSeed"; //std::cout<<" The SET SEED producer started."<<std::endl; ParameterSet serviceParameters = parameterSet.getParameter<ParameterSet>("ServiceParameters"); theService = new MuonServiceProxy(serviceParameters); thePatternRecognition.setServiceProxy(theService); theSeedFinder.setServiceProxy(theService); // Parameter set for the Builder ParameterSet trajectoryBuilderParameters = parameterSet.getParameter<ParameterSet>("SETTrajBuilderParameters"); LogTrace(metname) << "constructor called" << endl; apply_prePruning = trajectoryBuilderParameters.getParameter<bool>("Apply_prePruning"); useSegmentsInTrajectory = trajectoryBuilderParameters.getParameter<bool>("UseSegmentsInTrajectory"); // The inward-outward fitter (starts from seed state) ParameterSet filterPSet = trajectoryBuilderParameters.getParameter<ParameterSet>("FilterParameters"); filterPSet.addUntrackedParameter("UseSegmentsInTrajectory", useSegmentsInTrajectory); theFilter = new SETFilter(filterPSet,theService); //---- produces<TrajectorySeedCollection>(); }
| SETMuonSeedProducer::~SETMuonSeedProducer | ( | ) | [virtual] |
Destructor.
Definition at line 63 of file SETMuonSeedProducer.cc.
References LogTrace, theFilter, and theService.
{
LogTrace("Muon|RecoMuon|SETMuonSeedProducer")
<< "SETMuonSeedProducer destructor called" << endl;
if(theFilter) delete theFilter;
if (theService) delete theService;
}
| SETFilter* SETMuonSeedProducer::filter | ( | ) | const [inline, private] |
pre-filter
Definition at line 50 of file SETMuonSeedProducer.h.
References theFilter.
Referenced by produce().
{return theFilter;}
| void SETMuonSeedProducer::produce | ( | edm::Event & | event, |
| const edm::EventSetup & | eventSetup | ||
| ) | [virtual] |
Implements edm::EDProducer.
Definition at line 72 of file SETMuonSeedProducer.cc.
References alongMomentum, apply_prePruning, SiPixelRawToDigiRegional_cfi::beamSpot, SETFilter::buildTrajectoryMeasurements(), debug, dir, MuonPatternRecoDumper::dumpLayer(), SETSeedFinder::fillSeedCandidates(), filter(), SETSeedFinder::findAllValidSets(), SETFilter::fwfit_SET(), edm::HandleBase::isValid(), SETFilter::layers(), SETSeedFinder::limitCombinatorics(), LogTrace, metname, oppositeToMomentum, convertSQLitetoXML_cfg::output, trajectoryStateTransform::persistentState(), SETPatternRecognition::produce(), edm::OwnVector< T, P >::push_back(), SETFilter::reset(), MuonSeedVFinder::setBeamSpot(), setEvent(), SETSeedFinder::sortByLayer(), theBeamSpotTag, theFilter, thePatternRecognition, theSeedFinder, theService, SETFilter::transform(), SETFilter::transformLight(), useSegmentsInTrajectory, SETSeedFinder::validSetsPrePruning(), reco::BeamSpot::x0(), reco::BeamSpot::y0(), and reco::BeamSpot::z0().
{
//std::cout<<" start producing..."<<std::endl;
const string metname = "Muon|RecoMuon|SETMuonSeedSeed";
MuonPatternRecoDumper debug;
//Get the CSC Geometry :
theService->update(eventSetup);
std::auto_ptr<TrajectorySeedCollection> output(new TrajectorySeedCollection());
Handle<View<TrajectorySeed> > seeds;
setEvent(event);
reco::BeamSpot beamSpot;
edm::Handle<reco::BeamSpot> beamSpotHandle;
event.getByLabel(theBeamSpotTag, beamSpotHandle);
if ( beamSpotHandle.isValid() )
{
beamSpot = *beamSpotHandle;
} else
{
edm::LogInfo("MuonSeedGenerator")
<< "No beam spot available from EventSetup \n";
}
// make it a vector so we can subtract it from position vectors
GlobalVector gv(beamSpot.x0(), beamSpot.y0(), beamSpot.z0());
theSeedFinder.setBeamSpot(gv);
bool fwFitFailed = true;
std::vector <SeedCandidate> seedCandidates_AllChosen;
std::vector< MuonRecHitContainer > MuonRecHitContainer_clusters;
//---- this is "clustering"; later a trajectory can not use hits from different clusters
thePatternRecognition.produce(event, eventSetup, MuonRecHitContainer_clusters);
//std::cout<<"We have formed "<<MuonRecHitContainer_clusters.size()<<" clusters"<<std::endl;
//---- for each cluster,
for(unsigned int cluster = 0; cluster < MuonRecHitContainer_clusters.size(); ++cluster) {
//std::cout<<" This is cluster number : "<<cluster<<std::endl;
std::vector <SeedCandidate> seedCandidates_inCluster;
//---- group hits in detector layers (if in same layer); the idea is that
//---- some hits could not belong to a track simultaneously - these will be in a
//---- group; two hits from one and the same group will not go to the same track
std::vector< MuonRecHitContainer > MuonRecHitContainer_perLayer = theSeedFinder.sortByLayer(MuonRecHitContainer_clusters[cluster]);
//---- Add protection against huge memory consumption
//---- Delete busy layers if needed (due to combinatorics)
theSeedFinder.limitCombinatorics(MuonRecHitContainer_perLayer);
//---- build all possible combinations (valid sets)
std::vector <MuonRecHitContainer> allValidSets = theSeedFinder.findAllValidSets(MuonRecHitContainer_perLayer);
if(apply_prePruning){
//---- remove "wild" segments from the combination
theSeedFinder.validSetsPrePruning(allValidSets);
}
//---- build the appropriate output: seedCandidates_inCluster
//---- if too many (?) valid sets in a cluster - skip it
if(allValidSets.size()<500){// hardcoded - remove it
seedCandidates_inCluster = theSeedFinder.fillSeedCandidates(allValidSets);
}
//---- find the best valid combinations using simple (no propagation errors) chi2-fit
//std::cout<<" Found "<<seedCandidates_inCluster.size()<<" valid sets in the current cluster."<<std::endl;
if(!seedCandidates_inCluster.empty()){
//---- this is the forward fitter (segments); choose which of the SETs in a cluster to be considered further
std::vector < SeedCandidate> bestSets_inCluster;
fwFitFailed = !(filter()->fwfit_SET(seedCandidates_inCluster, bestSets_inCluster));
//---- has the fit failed? continue to the next cluster instead of returning the empty trajectoryContainer and stop the loop IBL 080903
if(fwFitFailed){
//std::cout<<" fwfit_SET failed!"<<std::endl;
continue;
}
for(unsigned int iSet = 0; iSet<bestSets_inCluster.size();++iSet){
seedCandidates_AllChosen.push_back(bestSets_inCluster[iSet]);
}
}
}
//---- loop over all the SETs candidates
for(unsigned int iMuon = 0;iMuon<seedCandidates_AllChosen.size();++iMuon){
//std::cout<<" chosen iMuon = "<<iMuon<<std::endl;
Trajectory::DataContainer finalCandidate;
SeedCandidate * aFinalSet = &(seedCandidates_AllChosen[iMuon]);
fwFitFailed = !(filter()->buildTrajectoryMeasurements(aFinalSet, finalCandidate));
if(fwFitFailed ){
//std::cout<<" buildTrajectoryMeasurements failed!"<<std::endl;
continue;
}
//---- are there measurements (or detLayers) used at all?
if( filter()->layers().size() )
LogTrace(metname) << debug.dumpLayer( filter()->lastDetLayer());
else {
continue;
}
//std::cout<<" chambers used - all : "<<filter()->getTotalChamberUsed()<<", DT : "<<filter()->getDTChamberUsed()<<
//", CSC : "<<filter()->getCSCChamberUsed()<<", RPC : "<<filter()->getRPCChamberUsed()<<std::endl;
//---- ask for some "reasonable" conditions to build a STA muon;
//---- (totalChambers >= 2, dtChambers + cscChambers >0)
if (filter()->goodState()) {
TransientTrackingRecHit::ConstRecHitContainer hitContainer;
TrajectoryStateOnSurface firstTSOS;
bool conversionPassed = false;
if(!useSegmentsInTrajectory){
//---- transforms set of segment measurements to a set of rechit measurements
conversionPassed = filter()->transform(finalCandidate, hitContainer, firstTSOS);
}
else{
//---- transforms set of segment measurements to a set of segment measurements
conversionPassed = filter()->transformLight(finalCandidate, hitContainer, firstTSOS);
}
if ( conversionPassed && !finalCandidate.empty() && !hitContainer.empty()) {
//---- doesn't work...
//output->push_back( theSeedFinder.makeSeed(firstTSOS, hitContainer) );
edm::OwnVector<TrackingRecHit> recHitsContainer;
for(unsigned int iHit = 0;iHit < hitContainer.size();++iHit){
recHitsContainer.push_back(hitContainer.at(iHit)->hit()->clone());
}
PropagationDirection dir = oppositeToMomentum;
if(useSegmentsInTrajectory){
dir = alongMomentum;// why forward (for rechits) later?
}
PTrajectoryStateOnDet seedTSOS =
trajectoryStateTransform::persistentState( firstTSOS, hitContainer.at(0)->geographicalId().rawId());
TrajectorySeed seed(seedTSOS,recHitsContainer,dir);
//MuonPatternRecoDumper debug;
//std::cout<<" firstTSOS (not IP) = "<<debug.dumpTSOS(firstTSOS)<<std::endl;
//std::cout<<" hits(from range) = "<<range.second-range.first<<" hits (from size) = "<<hitContainer.size()<<std::endl;
//for(unsigned int iRH=0;iRH<hitContainer.size();++iRH){
//std::cout<<" RH = "<<iRH+1<<" globPos = "<<hitContainer.at(iRH)->globalPosition()<<std::endl;
//}
output->push_back(seed);
}
else{
//std::cout<<" Transformation from TrajectoryMeasurements to RecHitContainer faild - skip "<<std::endl;
continue;
}
}else{
//std::cout<<" Not enough (as defined) measurements to build trajectory - skip"<<std::endl;
continue;
}
}
event.put(output);
theFilter->reset();
}
| void SETMuonSeedProducer::setEvent | ( | const edm::Event & | event | ) | [private] |
Definition at line 223 of file SETMuonSeedProducer.cc.
References SETFilter::setEvent(), and theFilter.
Referenced by produce().
bool SETMuonSeedProducer::apply_prePruning [private] |
Definition at line 60 of file SETMuonSeedProducer.h.
Referenced by produce(), and SETMuonSeedProducer().
Definition at line 67 of file SETMuonSeedProducer.h.
Referenced by produce().
SETFilter* SETMuonSeedProducer::theFilter [private] |
Definition at line 56 of file SETMuonSeedProducer.h.
Referenced by filter(), produce(), setEvent(), SETMuonSeedProducer(), and ~SETMuonSeedProducer().
Definition at line 64 of file SETMuonSeedProducer.h.
Referenced by produce(), and SETMuonSeedProducer().
Definition at line 65 of file SETMuonSeedProducer.h.
Referenced by produce(), and SETMuonSeedProducer().
MuonServiceProxy* SETMuonSeedProducer::theService [private] |
Definition at line 62 of file SETMuonSeedProducer.h.
Referenced by produce(), SETMuonSeedProducer(), and ~SETMuonSeedProducer().
bool SETMuonSeedProducer::useSegmentsInTrajectory [private] |
Definition at line 61 of file SETMuonSeedProducer.h.
Referenced by produce(), and SETMuonSeedProducer().
1.7.3