HGCalClusteringImpl Class Reference

#include <HGCalClusteringImpl.h>

Public Member Functions
void	clusterizeDR (const edm::PtrVector< l1t::HGCalTriggerCell > &triggerCellsPtrs, l1t::HGCalClusterBxCollection &clusters)
void	clusterizeNN (const edm::PtrVector< l1t::HGCalTriggerCell > &triggerCellsPtrs, l1t::HGCalClusterBxCollection &clusters, const HGCalTriggerGeometryBase &triggerGeometry)
	HGCalClusteringImpl (const edm::ParameterSet &conf)
bool	isPertinent (const l1t::HGCalTriggerCell &tc, const l1t::HGCalCluster &clu, double distXY) const
void	mergeClusters (l1t::HGCalCluster &main_cluster, const l1t::HGCalCluster &secondary_cluster) const
void	NNKernel (const std::vector< edm::Ptr< l1t::HGCalTriggerCell >> &reshuffledTriggerCells, l1t::HGCalClusterBxCollection &clusters, const HGCalTriggerGeometryBase &triggerGeometry)

Private Member Functions
void	triggerCellReshuffling (const edm::PtrVector< l1t::HGCalTriggerCell > &triggerCellsPtrs, std::array< std::array< std::vector< edm::Ptr< l1t::HGCalTriggerCell >>, kLayers_ >, kNSides_ > &reshuffledTriggerCells)

Private Attributes
std::string	clusteringAlgorithmType_
double	dr_
double	seedThreshold_
double	triggerCellThreshold_

Static Private Attributes
static const unsigned	kLayers_ = kLayersEE_+kLayersFH_
static const unsigned	kLayersEE_ = 28
static const unsigned	kLayersFH_ = 12
static const unsigned	kNSides_ = 2

Detailed Description

Definition at line 13 of file HGCalClusteringImpl.h.

Constructor & Destructor Documentation

HGCalClusteringImpl::HGCalClusteringImpl

(

const edm::ParameterSet &

conf

)

Definition at line 8 of file HGCalClusteringImpl.cc.

References clusteringAlgorithmType_, seedThreshold_, and triggerCellThreshold_.

                                                                    :
    seedThreshold_(conf.getParameter<double>("seeding_threshold")),
    triggerCellThreshold_(conf.getParameter<double>("clustering_threshold")),
    dr_(conf.getParameter<double>("dR_cluster")),
    clusteringAlgorithmType_(conf.getParameter<string>("clusterType"))
{    
    edm::LogInfo("HGCalClusterParameters") << "C2d Clustering Algorithm selected : " << clusteringAlgorithmType_ ; 
    edm::LogInfo("HGCalClusterParameters") << "C2d seeding Thr: " << seedThreshold_ ; 
    edm::LogInfo("HGCalClusterParameters") << "C2d clustering Thr: " << triggerCellThreshold_ ; 
 
}

Member Function Documentation

void HGCalClusteringImpl::clusterizeDR	(	const edm::PtrVector< l1t::HGCalTriggerCell > &	triggerCellsPtrs,
		l1t::HGCalClusterBxCollection &	clusters
	)

Definition at line 42 of file HGCalClusteringImpl.cc.

References edm::PtrVector< T >::begin(), edmIntegrityCheck::d, dr_, edm::PtrVector< T >::end(), mps_fire::i, isPertinent(), BXVector< T >::resize(), seedThreshold_, BXVector< T >::set(), edm::PtrVectorBase::size(), and triggerCellThreshold_.

     {

    bool isSeed[triggerCellsPtrs.size()];
    
    /* search for cluster seeds */
    int itc(0);
    for( edm::PtrVector<l1t::HGCalTriggerCell>::const_iterator tc = triggerCellsPtrs.begin(); tc != triggerCellsPtrs.end(); ++tc,++itc ){
        isSeed[itc] = ( (*tc)->mipPt() > seedThreshold_) ? true : false;
    }
    
    /* clustering the TCs */
    std::vector<l1t::HGCalCluster> clustersTmp;

    itc=0;
    for( edm::PtrVector<l1t::HGCalTriggerCell>::const_iterator tc = triggerCellsPtrs.begin(); tc != triggerCellsPtrs.end(); ++tc,++itc ){
            
        if( (*tc)->mipPt() < triggerCellThreshold_ ){
            continue;
        }
        
        /* searching for TC near the center of the cluster  */
        int iclu=0;
        vector<int> tcPertinentClusters; 
        for( const auto& clu : clustersTmp){
            if( this->isPertinent(**tc, clu, dr_) ){
                tcPertinentClusters.push_back(iclu);
            }
            ++iclu;
        }
        if( tcPertinentClusters.size() == 0 && isSeed[itc] ){
            clustersTmp.emplace_back( *tc );
        }
        else if ( tcPertinentClusters.size() > 0 ){
         
            unsigned minDist(300);
            unsigned targetClu(0);
                        
            for( int iclu : tcPertinentClusters){
                double d = clustersTmp.at(iclu).distance(**tc);
                if( d < minDist ){
                    minDist = d;
                    targetClu = iclu;
                }
            } 

            clustersTmp.at(targetClu).addConstituent( *tc );                    

        }
    }

    /* store clusters in the persistent collection */
    clusters.resize(0, clustersTmp.size());
    for( unsigned i(0); i<clustersTmp.size(); ++i ){
        clusters.set( 0, i, clustersTmp.at(i) );
    }
    
}

void HGCalClusteringImpl::clusterizeNN	(	const edm::PtrVector< l1t::HGCalTriggerCell > &	triggerCellsPtrs,
		l1t::HGCalClusterBxCollection &	clusters,
		const HGCalTriggerGeometryBase &	triggerGeometry
	)

Definition at line 251 of file HGCalClusteringImpl.cc.

References kLayers_, kNSides_, NNKernel(), and triggerCellReshuffling().

     {

    std::array< std::array< std::vector<edm::Ptr<l1t::HGCalTriggerCell> >,kLayers_>,kNSides_> reshuffledTriggerCells; 
    triggerCellReshuffling( triggerCellsPtrs, reshuffledTriggerCells );

    for(unsigned iec=0; iec<kNSides_; ++iec){
        for(unsigned il=0; il<kLayers_; ++il){
            NNKernel( reshuffledTriggerCells[iec][il], clusters, triggerGeometry );
        }
    }

}

bool HGCalClusteringImpl::isPertinent	(	const l1t::HGCalTriggerCell &	tc,
		const l1t::HGCalCluster &	clu,
		double	distXY
	)		const

Definition at line 22 of file HGCalClusteringImpl.cc.

References l1t::HGCalTriggerCell::detId(), l1t::HGCalClusterT< C >::detId(), and l1t::HGCalClusterT< C >::distance().

Referenced by clusterizeDR().

{

    HGCalDetId tcDetId( tc.detId() );
    HGCalDetId cluDetId( clu.detId() );
    if( (tcDetId.layer() != cluDetId.layer()) ||
        (tcDetId.subdetId() != cluDetId.subdetId()) ||
        (tcDetId.zside() != cluDetId.zside()) ){
        return false;
    }   
    if ( clu.distance((tc)) < distXY ){
        return true;
    }
    return false;

}

void HGCalClusteringImpl::mergeClusters	(	l1t::HGCalCluster &	main_cluster,
		const l1t::HGCalCluster &	secondary_cluster
	)		const

Definition at line 135 of file HGCalClusteringImpl.cc.

References l1t::HGCalClusterT< C >::addConstituent(), edm::PtrVector< T >::begin(), l1t::HGCalClusterT< C >::constituents(), and edm::PtrVector< T >::end().

Referenced by NNKernel().

{

    const edm::PtrVector<l1t::HGCalTriggerCell>& pertinentTC = secondary_cluster.constituents();
    
    for( edm::PtrVector<l1t::HGCalTriggerCell>::iterator tc = pertinentTC.begin(); tc != pertinentTC.end(); ++tc ){
        main_cluster.addConstituent(*tc);
    }

}

void HGCalClusteringImpl::NNKernel	(	const std::vector< edm::Ptr< l1t::HGCalTriggerCell >> &	reshuffledTriggerCells,
		l1t::HGCalClusterBxCollection &	clusters,
		const HGCalTriggerGeometryBase &	triggerGeometry
	)

Definition at line 148 of file HGCalClusteringImpl.cc.

References Exception, HGCalTriggerGeometryBase::getNeighborsFromTriggerCell(), mergeClusters(), neighbor(), BXVector< T >::push_back(), seedThreshold_, and triggerCellThreshold_.

Referenced by clusterizeNN().

     {
   
    /* declaring the clusters vector */
    std::vector<l1t::HGCalCluster> clustersTmp;

    // map TC id -> cluster index in clustersTmp
    std::unordered_map<uint32_t, unsigned> cluNNmap;

    /* loop over the trigger-cells */
    for( const auto& tc_ptr : reshuffledTriggerCells ){
                
        if( tc_ptr->mipPt() < triggerCellThreshold_ ){
            continue;
        }
        
        // Check if the neighbors of that TC are already included in a cluster
        // If this is the case, add the TC to the first (arbitrary) neighbor cluster
        // Otherwise create a new cluster
        bool createNewC2d(true);
        const auto neighbors = triggerGeometry.getNeighborsFromTriggerCell(tc_ptr->detId());
        for( const auto neighbor : neighbors ){
            auto tc_cluster_itr = cluNNmap.find(neighbor);
            if(tc_cluster_itr!=cluNNmap.end()){ 
                createNewC2d = false;
                if( tc_cluster_itr->second < clustersTmp.size()){
                    clustersTmp.at(tc_cluster_itr->second).addConstituent(tc_ptr);
                    // map TC id to the existing cluster
                    cluNNmap.emplace(tc_ptr->detId(), tc_cluster_itr->second);
                }
                else{
                    throw cms::Exception("HGCTriggerUnexpected")
                        << "Trying to access a non-existing cluster. But it should exist...\n";
                }                
                break;
            }
        }
        if(createNewC2d){
            clustersTmp.emplace_back(tc_ptr);
            clustersTmp.back().setValid(true);
            // map TC id to the cluster index (size - 1)
            cluNNmap.emplace(tc_ptr->detId(), clustersTmp.size()-1);
        }
    }
    
    /* declaring the vector with possible clusters merged */
    // Merge neighbor clusters together
    for( auto& cluster1 : clustersTmp){
        // If the cluster has been merged into another one, skip it
        if( !cluster1.valid() ) continue;
        // Fill a set containing all TC included in the clusters
        // as well as all neighbor TC
        std::unordered_set<uint32_t> cluTcSet;
        for(const auto& tc_clu1 : cluster1.constituents()){ 
            cluTcSet.insert( tc_clu1->detId() );
            const auto neighbors = triggerGeometry.getNeighborsFromTriggerCell( tc_clu1->detId() );
            for(const auto neighbor : neighbors){
                cluTcSet.insert( neighbor );
            }
        }        
            
        for( auto& cluster2 : clustersTmp ){
            // If the cluster has been merged into another one, skip it
            if( cluster1.detId()==cluster2.detId() ) continue;
            if( !cluster2.valid() ) continue;
            // Check if the TC in clu2 are in clu1 or its neighbors
            // If yes, merge the second cluster into the first one
            for(const auto& tc_clu2 : cluster2.constituents()){ 
                if( cluTcSet.find(tc_clu2->detId())!=cluTcSet.end() ){
                    mergeClusters( cluster1, cluster2 );                    
                    cluTcSet.insert( tc_clu2->detId() );
                    const auto neighbors = triggerGeometry.getNeighborsFromTriggerCell( tc_clu2->detId() );
                    for(const auto neighbor : neighbors){
                        cluTcSet.insert( neighbor );
                    }                    
                    cluster2.setValid(false);
                    break;
                }
            }
        }
    }

    /* store clusters in the persistent collection */
    // only if the cluster contain a TC above the seed threshold
    for( auto& cluster : clustersTmp ){
        if( !cluster.valid() ) continue;
        bool saveInCollection(false);
        for( const auto& tc_ptr : cluster.constituents() ){
            /* threshold in transverse-mip */
            if( tc_ptr->mipPt() > seedThreshold_ ){
                saveInCollection = true;
                break;
            }
        }
        if(saveInCollection){
            clusters.push_back( 0, cluster );
        }
    }
}

void HGCalClusteringImpl::triggerCellReshuffling ( const edm::PtrVector< l1t::HGCalTriggerCell > &  triggerCellsPtrs, std::array< std::array< std::vector< edm::Ptr< l1t::HGCalTriggerCell >>, kLayers_ >, kNSides_ > &  reshuffledTriggerCells  )

private

Definition at line 107 of file HGCalClusteringImpl.cc.

References edm::PtrVector< T >::begin(), edm::PtrVector< T >::end(), makeMuonMisalignmentScenario::endcap, HGCEE, HGCHEB, HGCHEF, kLayersEE_, and DetId::subdetId().

Referenced by clusterizeNN().

     {

    for( edm::PtrVector<l1t::HGCalTriggerCell>::const_iterator tc = triggerCellsPtrs.begin(); tc != triggerCellsPtrs.end(); ++tc){
        int endcap = (*tc)->zside() == -1 ? 0 : 1 ;
        HGCalDetId tcDetId( (*tc)->detId() );
        int subdet = tcDetId.subdetId();
        int layer = -1;
        
        if( subdet == HGCEE ){ 
            layer = (*tc)->layer();
        }
        else if( subdet == HGCHEF ){
            layer = (*tc)->layer() + kLayersEE_;
        }
        else if( subdet == HGCHEB ){
            edm::LogWarning("DataNotFound") << "WARNING: the BH trgCells are not yet implemented";            
        }
        
        reshuffledTriggerCells[endcap][layer-1].emplace_back(*tc);
        
    }

}

Member Data Documentation

std::string HGCalClusteringImpl::clusteringAlgorithmType_

private

Definition at line 57 of file HGCalClusteringImpl.h.

Referenced by HGCalClusteringImpl().

double HGCalClusteringImpl::dr_

private

Definition at line 56 of file HGCalClusteringImpl.h.

Referenced by clusterizeDR().

const unsigned HGCalClusteringImpl::kLayers_ = kLayersEE_+kLayersFH_

staticprivate

Definition at line 21 of file HGCalClusteringImpl.h.

Referenced by clusterizeNN().

const unsigned HGCalClusteringImpl::kLayersEE_ = 28

staticprivate

Definition at line 19 of file HGCalClusteringImpl.h.

Referenced by triggerCellReshuffling().

const unsigned HGCalClusteringImpl::kLayersFH_ = 12

staticprivate

Definition at line 20 of file HGCalClusteringImpl.h.

const unsigned HGCalClusteringImpl::kNSides_ = 2

staticprivate

Definition at line 16 of file HGCalClusteringImpl.h.

Referenced by clusterizeNN().

double HGCalClusteringImpl::seedThreshold_

private

Definition at line 54 of file HGCalClusteringImpl.h.

Referenced by clusterizeDR(), HGCalClusteringImpl(), and NNKernel().

double HGCalClusteringImpl::triggerCellThreshold_

private

Definition at line 55 of file HGCalClusteringImpl.h.

Referenced by clusterizeDR(), HGCalClusteringImpl(), and NNKernel().