HLTTrackClusterRemover Class Reference

Inheritance diagram for HLTTrackClusterRemover:

Classes
struct	ParamBlock

Public Member Functions
template<typename T >
auto_ptr< edmNew::DetSetVector < T > >	cleanup (const edmNew::DetSetVector< T > &oldClusters, const std::vector< uint8_t > &isGood, reco::ClusterRemovalInfo::Indices &refs, const reco::ClusterRemovalInfo::Indices *oldRefs)
	HLTTrackClusterRemover (const edm::ParameterSet &iConfig)
void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup)
	~HLTTrackClusterRemover ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
virtual	~EDProducer ()
Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
boost::function< void(const BranchDescription &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()

Private Member Functions
template<typename T >
std::auto_ptr < edmNew::DetSetVector< T > >	cleanup (const edmNew::DetSetVector< T > &oldClusters, const std::vector< uint8_t > &isGood, reco::ClusterRemovalInfo::Indices &refs, const reco::ClusterRemovalInfo::Indices *oldRefs)
void	mergeOld (reco::ClusterRemovalInfo::Indices &refs, const reco::ClusterRemovalInfo::Indices &oldRefs)
void	process (const TrackingRecHit *hit, float chi2)
void	process (const SiStripRecHit2D *hit2D, uint32_t subdet)
void	process (const SiStripRecHit1D *hit1D, uint32_t subdet)
void	readPSet (const edm::ParameterSet &iConfig, const std::string &name, int id1=-1, int id2=-1, int id3=-1, int id4=-1, int id5=-1, int id6=-1)

Private Attributes
std::map< uint32_t, std::set < SiPixelRecHit::ClusterRef > >	collectedPixel
std::map< uint32_t, std::set < SiStripRecHit1D::ClusterRegionalRef > >	collectedRegStrip
bool	doPixel_
bool	doStrip_
bool	makeProducts_
bool	mergeOld_
edm::InputTag	oldRemovalInfo_
ParamBlock	pblocks_ [NumberOfParamBlocks]
edm::InputTag	pixelClusters_
std::vector< uint8_t >	pixels
edm::ProductID	pixelSourceProdID
edm::InputTag	stripClusters_
std::vector< uint8_t >	strips
edm::ProductID	stripSourceProdID
edm::InputTag	trajectories_

Static Private Attributes
static const unsigned int	NumberOfParamBlocks = 6

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType
typedef WorkerT< EDProducer >	WorkerType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
Protected Member Functions inherited from edm::EDProducer
CurrentProcessingContext const *	currentContext () const
Protected Member Functions inherited from edm::ProducerBase
template<class TProducer , class TMethod >
void	callWhenNewProductsRegistered (TProducer *iProd, TMethod iMethod)

Detailed Description

Definition at line 33 of file HLTTrackClusterRemover.cc.

Constructor & Destructor Documentation

HLTTrackClusterRemover::HLTTrackClusterRemover

(

const edm::ParameterSet &

iConfig

)

Definition at line 108 of file HLTTrackClusterRemover.cc.

References doPixel_, doStrip_, edm::hlt::Exception, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), i, mergeOld_, NumberOfParamBlocks, oldRemovalInfo_, pblocks_, and readPSet().

                                                                         :
    trajectories_(iConfig.getParameter<InputTag>("trajectories")),
    doStrip_(iConfig.existsAs<bool>("doStrip") ? iConfig.getParameter<bool>("doStrip") : true),
    doPixel_(iConfig.existsAs<bool>("doPixel") ? iConfig.getParameter<bool>("doPixel") : true),
    stripClusters_(doStrip_ ? iConfig.getParameter<InputTag>("stripClusters") : InputTag("NONE")),
    pixelClusters_(doPixel_ ? iConfig.getParameter<InputTag>("pixelClusters") : InputTag("NONE")),
    mergeOld_(false),
    oldRemovalInfo_(edm::InputTag()),
    makeProducts_(true)
{

  if (iConfig.exists("oldClusterRemovalInfo"))
    {
      oldRemovalInfo_=iConfig.getParameter<InputTag>("oldClusterRemovalInfo");
      if (not (oldRemovalInfo_== edm::InputTag())) mergeOld_=true;
    }

    fill(pblocks_, pblocks_+NumberOfParamBlocks, ParamBlock());
    readPSet(iConfig, "Common",-1);
    if (doPixel_) {
        readPSet(iConfig, "Pixel" ,0,1);
        readPSet(iConfig, "PXB" ,0);
        readPSet(iConfig, "PXE" ,1);
    }
    if (doStrip_) {
        readPSet(iConfig, "Strip" ,2,3,4,5);
        readPSet(iConfig, "StripInner" ,2,3);
        readPSet(iConfig, "StripOuter" ,4,5);
        readPSet(iConfig, "TIB" ,2);
        readPSet(iConfig, "TID" ,3);
        readPSet(iConfig, "TOB" ,4);
        readPSet(iConfig, "TEC" ,5);
    }

    bool usingCharge = false;
    for (size_t i = 0; i < NumberOfParamBlocks; ++i) {
        if (!pblocks_[i].isSet_) throw cms::Exception("Configuration Error") << "HLTTrackClusterRemover: Missing configuration for detector with subDetID = " << (i+1);
        if (pblocks_[i].usesCharge_ && !usingCharge) {
            throw cms::Exception("Configuration Error") << "HLTTrackClusterRemover: Configuration for subDetID = " << (i+1) << " uses cluster charge, which is not enabled.";
        }
    }

    produces<edmNew::DetSetVector<SiPixelClusterRefNew> >();
    produces<edmNew::DetSetVector<SiStripRecHit1D::ClusterRegionalRef> >();

}

HLTTrackClusterRemover::~HLTTrackClusterRemover

(

)

Definition at line 156 of file HLTTrackClusterRemover.cc.

{
}

Member Function Documentation

template<typename T >

std::auto_ptr<edmNew::DetSetVector<T> > HLTTrackClusterRemover::cleanup ( const edmNew::DetSetVector< T > &  oldClusters, const std::vector< uint8_t > &  isGood, reco::ClusterRemovalInfo::Indices &  refs, const reco::ClusterRemovalInfo::Indices *  oldRefs  )

private

template<typename T >

auto_ptr<edmNew::DetSetVector<T> > HLTTrackClusterRemover::cleanup	(	const edmNew::DetSetVector< T > &	oldClusters,
		const std::vector< uint8_t > &	isGood,
		reco::ClusterRemovalInfo::Indices &	refs,
		const reco::ClusterRemovalInfo::Indices *	oldRefs
	)

Definition at line 171 of file HLTTrackClusterRemover.cc.

References edmNew::DetSetVector< T >::begin(), edmNew::DetSetVector< T >::data(), edmNew::DetSetVector< T >::dataSize(), edmNew::DetSetVector< T >::end(), getHLTprescales::index, mergeOld(), convertSQLitetoXML_cfg::output, and edmNew::DetSetVector< T >::size().

                                                                                                   {
    typedef typename edmNew::DetSetVector<T>             DSV;
    typedef typename edmNew::DetSetVector<T>::FastFiller DSF;
    typedef typename edmNew::DetSet<T>                   DS;
    auto_ptr<DSV> output(new DSV());
    output->reserve(oldClusters.size(), oldClusters.dataSize());

    unsigned int countOld=0;
    unsigned int countNew=0;

    // cluster removal loop
    const T * firstOffset = & oldClusters.data().front();
    for (typename DSV::const_iterator itdet = oldClusters.begin(), enddet = oldClusters.end(); itdet != enddet; ++itdet) {
        DS oldDS = *itdet;

        if (oldDS.empty()) continue; // skip empty detsets

        uint32_t id = oldDS.detId();
        DSF outds(*output, id);

        for (typename DS::const_iterator it = oldDS.begin(), ed = oldDS.end(); it != ed; ++it) {
            uint32_t index = ((&*it) - firstOffset);
        countOld++;
            if (isGood[index]) { 
                outds.push_back(*it);
        countNew++;
                refs.push_back(index); 
                //std::cout << "HLTTrackClusterRemover::cleanup " << typeid(T).name() << " reference " << index << " to " << (refs.size() - 1) << std::endl;
            }

    }
        if (outds.empty()) outds.abort(); // not write in an empty DSV
    }
    //    double fraction = countNew  / (double) countOld;
    //    std::cout<<"fraction: "<<fraction<<std::endl;
    if (oldRefs != 0) mergeOld(refs, *oldRefs);
    return output;
}

void HLTTrackClusterRemover::mergeOld ( reco::ClusterRemovalInfo::Indices &  refs, const reco::ClusterRemovalInfo::Indices &  oldRefs  )

private

Definition at line 160 of file HLTTrackClusterRemover.cc.

References i, and n.

Referenced by cleanup().

{
        for (size_t i = 0, n = refs.size(); i < n; ++i) {
            refs[i] = oldRefs[refs[i]];
       }
}

void HLTTrackClusterRemover::process ( const TrackingRecHit *  hit, float  chi2  )

inlineprivate

Definition at line 230 of file HLTTrackClusterRemover.cc.

References SiPixelRecHit::cluster(), collectedPixel, cond::rpcobgas::detid, doPixel_, doStrip_, edm::hlt::Exception, TrackingRecHit::geographicalId(), edm::Ref< C, T, F >::id(), HLTTrackClusterRemover::ParamBlock::maxSize_, SiStripMatchedRecHit2D::monoHit(), NumberOfParamBlocks, ProjectedSiStripRecHit2D::originalHit(), pblocks_, PixelSubdetector::PixelBarrel, PixelSubdetector::PixelEndcap, pixelSourceProdID, DetId::rawId(), SiStripMatchedRecHit2D::stereoHit(), and DetId::subdetId().

Referenced by produce().

                                                                          {
    DetId detid = hit->geographicalId(); 
    uint32_t subdet = detid.subdetId();

    assert ((subdet > 0) && (subdet <= NumberOfParamBlocks));

    // chi2 cut
    if (chi2 > pblocks_[subdet-1].maxChi2_) return;

    if ((subdet == PixelSubdetector::PixelBarrel) || (subdet == PixelSubdetector::PixelEndcap)) {
      //      std::cout<<"process pxl hit"<<std::endl;
        if (!doPixel_) return;
        // this is a pixel, and i *know* it is
        const SiPixelRecHit *pixelHit = static_cast<const SiPixelRecHit *>(hit);

        SiPixelRecHit::ClusterRef cluster = pixelHit->cluster();

        if (cluster.id() != pixelSourceProdID) throw cms::Exception("Inconsistent Data") << 
                "HLTTrackClusterRemover: pixel cluster ref from Product ID = " << cluster.id() << 
                " does not match with source cluster collection (ID = " << pixelSourceProdID << ")\n.";

        assert(cluster.id() == pixelSourceProdID);
//DBG// cout << "HIT NEW PIXEL DETID = " << detid.rawId() << ", Cluster [ " << cluster.key().first << " / " <<  cluster.key().second << " ] " << endl;

        // if requested, cut on cluster size
        if (pblocks_[subdet-1].usesSize_ && (cluster->pixels().size() > pblocks_[subdet-1].maxSize_)) return;

        // mark as used
    //        pixels[cluster.key()] = false;
    
    collectedPixel[detid.rawId()].insert(cluster);
    } else { // aka Strip
        if (!doStrip_) return;
        const type_info &hitType = typeid(*hit);
        if (hitType == typeid(SiStripRecHit2D)) {
            const SiStripRecHit2D *stripHit = static_cast<const SiStripRecHit2D *>(hit);
//DBG//     cout << "Plain RecHit 2D: " << endl;
            process(stripHit,subdet);}
    else if (hitType == typeid(SiStripRecHit1D)) {
      const SiStripRecHit1D *hit1D = static_cast<const SiStripRecHit1D *>(hit);
      process(hit1D,subdet);
        } else if (hitType == typeid(SiStripMatchedRecHit2D)) {
            const SiStripMatchedRecHit2D *matchHit = static_cast<const SiStripMatchedRecHit2D *>(hit);
//DBG//     cout << "Matched RecHit 2D: " << endl;
            process(matchHit->monoHit(),subdet);
            process(matchHit->stereoHit(),subdet);
        } else if (hitType == typeid(ProjectedSiStripRecHit2D)) {
            const ProjectedSiStripRecHit2D *projHit = static_cast<const ProjectedSiStripRecHit2D *>(hit);
//DBG//     cout << "Projected RecHit 2D: " << endl;
            process(&projHit->originalHit(),subdet);
        } else throw cms::Exception("NOT IMPLEMENTED") << "Don't know how to handle " << hitType.name() << " on detid " << detid.rawId() << "\n";
    }
}

void HLTTrackClusterRemover::process ( const SiStripRecHit2D *  hit2D, uint32_t  subdet  )

inlineprivate

Definition at line 212 of file HLTTrackClusterRemover.cc.

References SiStripRecHit2D::cluster_regional(), collectedRegStrip, TrackingRecHit::geographicalId(), edm::Ref< C, T, F >::id(), and stripSourceProdID.

                                                                                {
  SiStripRecHit2D::ClusterRegionalRef clusterReg = hit->cluster_regional();
  if (clusterReg.id() != stripSourceProdID) throw cms::Exception("Inconsistent Data") <<
    "HLTTrackClusterRemover: strip cluster ref from Product ID = " << clusterReg.id() <<
    " does not match with source cluster collection (ID = " << stripSourceProdID << ")\n.";
  collectedRegStrip[hit->geographicalId()].insert(clusterReg);
}

void HLTTrackClusterRemover::process ( const SiStripRecHit1D *  hit1D, uint32_t  subdet  )

inlineprivate

Definition at line 220 of file HLTTrackClusterRemover.cc.

References SiStripRecHit1D::cluster_regional(), collectedRegStrip, TrackingRecHit::geographicalId(), edm::Ref< C, T, F >::id(), and stripSourceProdID.

                                                                                {
  SiStripRecHit2D::ClusterRegionalRef clusterReg = hit->cluster_regional();
  if (clusterReg.id() != stripSourceProdID) throw cms::Exception("Inconsistent Data") << 
    "HLTTrackClusterRemover: strip cluster ref from Product ID = " << clusterReg.id() << 
    " does not match with source cluster collection (ID = " << stripSourceProdID << ")\n.";
  collectedRegStrip[hit->geographicalId()].insert(clusterReg);
}

void HLTTrackClusterRemover::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

virtual

Implements edm::EDProducer.

Definition at line 296 of file HLTTrackClusterRemover.cc.

References edmNew::DetSetVector< T >::FastFiller::abort(), collectedPixel, collectedRegStrip, doPixel_, doStrip_, edmNew::DetSetVector< T >::FastFiller::empty(), edm::Event::getByLabel(), edm::HandleBase::id(), TrackingRecHit::isValid(), LogDebug, python.multivaluedict::map(), Trajectory::measurements(), mergeOld_, oldRemovalInfo_, pixelClusters_, pixelSourceProdID, process(), edmNew::DetSetVector< T >::FastFiller::push_back(), edm::Event::put(), stripClusters_, stripSourceProdID, and trajectories_.

{
    ProductID pixelOldProdID, stripOldProdID;

    Handle<edmNew::DetSetVector<SiPixelCluster> > pixelClusters;
    if (doPixel_) {
        iEvent.getByLabel(pixelClusters_, pixelClusters);
        pixelSourceProdID = pixelClusters.id();
    }

    Handle<edm::LazyGetter<SiStripCluster> > stripClusters;
    if (doStrip_) {
        iEvent.getByLabel(stripClusters_, stripClusters);
        stripSourceProdID = stripClusters.id();
    }

    //Handle<TrajTrackAssociationCollection> trajectories; 
    Handle<vector<Trajectory> > trajectories; 
    iEvent.getByLabel(trajectories_, trajectories);


    //for (TrajTrackAssociationCollection::const_iterator it = trajectories->begin(), ed = trajectories->end(); it != ed; ++it)  {
    //    const Trajectory &tj = * it->key;
    for (vector<Trajectory>::const_iterator it = trajectories->begin(), ed = trajectories->end(); it != ed; ++it)  {
        const Trajectory &tj = * it;
        const vector<TrajectoryMeasurement> &tms = tj.measurements();
        vector<TrajectoryMeasurement>::const_iterator itm, endtm;
        for (itm = tms.begin(), endtm = tms.end(); itm != endtm; ++itm) {
            const TrackingRecHit *hit = itm->recHit()->hit();
            if (!hit->isValid()) continue; 
        //      std::cout<<"process hit"<<std::endl;
            process( hit, itm->estimate() );
        }
    }

    

    auto_ptr<edmNew::DetSetVector<SiPixelClusterRefNew> > removedPixelClsuterRefs(new edmNew::DetSetVector<SiPixelClusterRefNew>());
    auto_ptr<edmNew::DetSetVector<SiStripRecHit1D::ClusterRegionalRef> > removedStripClsuterRegRefs(new edmNew::DetSetVector<SiStripRecHit1D::ClusterRegionalRef>());

    edm::Handle<edmNew::DetSetVector<SiPixelClusterRefNew> > oldPxlRef;
    edm::Handle<edmNew::DetSetVector<SiStripRecHit1D::ClusterRegionalRef> > oldStrRegRef;      

    if (mergeOld_){
      iEvent.getByLabel(oldRemovalInfo_,oldPxlRef);
      iEvent.getByLabel(oldRemovalInfo_,oldStrRegRef);
      LogDebug("TrackClusterRemover")<<"to merge in, "<<oldStrRegRef->size()<<" strp and "<<oldPxlRef->size()<<" pxl";
    }

    if (mergeOld_){
      for (edmNew::DetSetVector<SiStripRecHit1D::ClusterRegionalRef>::const_iterator itOld=oldStrRegRef->begin();itOld!=oldStrRegRef->end();++itOld){
    uint32_t id = itOld->detId();
    collectedRegStrip[id].insert(itOld->begin(),itOld->end());
      }
    }
    for (std::map<uint32_t, std::set< SiStripRecHit1D::ClusterRegionalRef > >::iterator itskiped= collectedRegStrip.begin();
     itskiped!=collectedRegStrip.end();++itskiped){
      edmNew::DetSetVector<SiStripRecHit1D::ClusterRegionalRef>::FastFiller fill(*removedStripClsuterRegRefs, itskiped->first);
      for (std::set< SiStripRecHit1D::ClusterRegionalRef >::iterator topush = itskiped->second.begin();
       topush!=itskiped->second.end();++topush){
    fill.push_back(*topush);
      }
      if (fill.empty()) fill.abort();
      itskiped->second.clear();
    }
    iEvent.put( removedStripClsuterRegRefs );

    if (mergeOld_){
      for (edmNew::DetSetVector<SiPixelClusterRefNew>::const_iterator itOld=oldPxlRef->begin();itOld!=oldPxlRef->end();++itOld){
    uint32_t id = itOld->detId();
    collectedPixel[id].insert(itOld->begin(),itOld->end());
      }
    }
    
    for (std::map<uint32_t, std::set< SiPixelRecHit::ClusterRef  > >::iterator itskiped= collectedPixel.begin();
     itskiped!=collectedPixel.end();++itskiped){  
      edmNew::DetSetVector<SiPixelClusterRefNew>::FastFiller fill(*removedPixelClsuterRefs, itskiped->first);
      for (std::set< SiPixelRecHit::ClusterRef  >::iterator topush = itskiped->second.begin(); 
       topush!=itskiped->second.end();++topush){ 
    fill.push_back(*topush); 
      }
      if (fill.empty()) fill.abort();   
      itskiped->second.clear();
    }
    iEvent.put( removedPixelClsuterRefs );
    
}

void HLTTrackClusterRemover::readPSet ( const edm::ParameterSet &  iConfig, const std::string &  name, int  id1 = -1, int  id2 = -1, int  id3 = -1, int  id4 = -1, int  id5 = -1, int  id6 = -1  )

private

Definition at line 90 of file HLTTrackClusterRemover.cc.

References edm::ParameterSet::exists(), and edm::ParameterSet::getParameter().

Referenced by HLTTrackClusterRemover().

{
    if (iConfig.exists(name)) {
        ParamBlock pblock(iConfig.getParameter<ParameterSet>(name));
        if (id1 == -1) {
            fill(pblocks_, pblocks_+NumberOfParamBlocks, pblock);
        } else {
            pblocks_[id1] = pblock;
            if (id2 != -1) pblocks_[id2] = pblock;
            if (id3 != -1) pblocks_[id3] = pblock;
            if (id4 != -1) pblocks_[id4] = pblock;
            if (id5 != -1) pblocks_[id5] = pblock;
            if (id6 != -1) pblocks_[id6] = pblock;
        }
    }
}

Member Data Documentation

std::map<uint32_t, std::set< SiPixelRecHit::ClusterRef > > HLTTrackClusterRemover::collectedPixel

private

Definition at line 81 of file HLTTrackClusterRemover.cc.

Referenced by process(), and produce().

std::map<uint32_t, std::set< SiStripRecHit1D::ClusterRegionalRef > > HLTTrackClusterRemover::collectedRegStrip

private

Definition at line 80 of file HLTTrackClusterRemover.cc.

Referenced by process(), and produce().

bool HLTTrackClusterRemover::doPixel_

private

Definition at line 55 of file HLTTrackClusterRemover.cc.

Referenced by HLTTrackClusterRemover(), process(), and produce().

bool HLTTrackClusterRemover::doStrip_

private

Definition at line 55 of file HLTTrackClusterRemover.cc.

Referenced by HLTTrackClusterRemover(), process(), and produce().

bool HLTTrackClusterRemover::makeProducts_

private

Definition at line 79 of file HLTTrackClusterRemover.cc.

bool HLTTrackClusterRemover::mergeOld_

private

Definition at line 57 of file HLTTrackClusterRemover.cc.

Referenced by HLTTrackClusterRemover(), and produce().

const unsigned int HLTTrackClusterRemover::NumberOfParamBlocks = 6

staticprivate

Definition at line 52 of file HLTTrackClusterRemover.cc.

Referenced by HLTTrackClusterRemover(), and process().

edm::InputTag HLTTrackClusterRemover::oldRemovalInfo_

private

Definition at line 58 of file HLTTrackClusterRemover.cc.

Referenced by HLTTrackClusterRemover(), and produce().

ParamBlock HLTTrackClusterRemover::pblocks_[NumberOfParamBlocks]

private

Definition at line 60 of file HLTTrackClusterRemover.cc.

Referenced by HLTTrackClusterRemover(), and process().

edm::InputTag HLTTrackClusterRemover::pixelClusters_

private

Definition at line 56 of file HLTTrackClusterRemover.cc.

Referenced by produce().

std::vector<uint8_t> HLTTrackClusterRemover::pixels

private

Definition at line 64 of file HLTTrackClusterRemover.cc.

edm::ProductID HLTTrackClusterRemover::pixelSourceProdID

private

Definition at line 65 of file HLTTrackClusterRemover.cc.

Referenced by process(), and produce().

edm::InputTag HLTTrackClusterRemover::stripClusters_

private

Definition at line 56 of file HLTTrackClusterRemover.cc.

Referenced by produce().

std::vector<uint8_t> HLTTrackClusterRemover::strips

private

Definition at line 64 of file HLTTrackClusterRemover.cc.

edm::ProductID HLTTrackClusterRemover::stripSourceProdID

private

Definition at line 65 of file HLTTrackClusterRemover.cc.

Referenced by process(), and produce().

edm::InputTag HLTTrackClusterRemover::trajectories_

private

Definition at line 54 of file HLTTrackClusterRemover.cc.

Referenced by produce().