BeamHaloPairGenerator.cc

Go to the documentation of this file.

#include "RecoTracker/SpecialSeedGenerators/interface/BeamHaloPairGenerator.h"
typedef SeedingHitSet::ConstRecHitPointer SeedingHit;

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "TrackingTools/TransientTrackingRecHit/interface/SeedingLayerSetsHits.h"
using namespace ctfseeding;


BeamHaloPairGenerator::BeamHaloPairGenerator(const edm::ParameterSet& conf, edm::ConsumesCollector& iC): 
  theSeedingLayerToken(iC.consumes<SeedingLayerSetsHits>(conf.getParameter<edm::InputTag>("LayerSrc")))
{
    edm::LogInfo("CtfSpecialSeedGenerator|BeamHaloPairGenerator") << "Constructing BeamHaloPairGenerator";
    theMaxTheta=conf.getParameter<double>("maxTheta");
    theMaxTheta=fabs(sin(theMaxTheta));
} 


const OrderedSeedingHits& BeamHaloPairGenerator::run(const TrackingRegion& region,
                                        const edm::Event& e,
                                            const edm::EventSetup& es){
    hitPairs.clear();
        edm::Handle<SeedingLayerSetsHits> hlayers;
        e.getByToken(theSeedingLayerToken, hlayers);
        const SeedingLayerSetsHits& layers = *hlayers;
        if(layers.numberOfLayersInSet() != 2)
          throw cms::Exception("CtfSpecialSeedGenerator") << "You are using " << layers.numberOfLayersInSet() <<" layers in set instead of 2 ";
        for(SeedingLayerSetsHits::SeedingLayerSet ls: layers) {
        auto innerHits  = region.hits(es, ls[0]);
        auto outerHits  = region.hits(es, ls[1]);
        
        for (auto iOuterHit = outerHits.begin(); iOuterHit != outerHits.end(); iOuterHit++){
          for (auto iInnerHit = innerHits.begin(); iInnerHit != innerHits.end(); iInnerHit++){
            //do something in there... if necessary
            SeedingHitSet::ConstRecHitPointer  crhpi =  &(**iInnerHit);
            SeedingHitSet::ConstRecHitPointer  crhpo =  &(**iOuterHit);
            GlobalVector d=crhpo->globalPosition() - crhpi->globalPosition();
            double ABSsinDtheta = fabs(sin(d.theta()));
            LogDebug("BeamHaloPairGenerator")<<"position1: "<<crhpo->globalPosition()
                             <<" position2: "<<crhpi->globalPosition()
                             <<" |sin(Dtheta)|: "<< ABSsinDtheta <<((ABSsinDtheta>theMaxTheta)?" skip":" keep");
            
              if (ABSsinDtheta>theMaxTheta) {;continue;}

              hitPairs.push_back(OrderedHitPair(crhpi,
                                crhpo));
            }
        }
        }
    return hitPairs;
}