HGCalClusteringImpl.h

Go to the documentation of this file.

#ifndef __L1Trigger_L1THGCal_HGCalClusteringImpl_h__
#define __L1Trigger_L1THGCal_HGCalClusteringImpl_h__

#include <array> 
#include <string>
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "L1Trigger/L1THGCal/interface/HGCalTriggerGeometryBase.h"
#include "DataFormats/L1THGCal/interface/HGCalTriggerCell.h"
#include "DataFormats/ForwardDetId/interface/HGCalDetId.h"
#include "DataFormats/L1THGCal/interface/HGCalCluster.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "L1Trigger/L1THGCal/interface/HGCalTriggerTools.h"


bool distanceSorter (pair<edm::Ptr<l1t::HGCalTriggerCell>,float> i,pair<edm::Ptr<l1t::HGCalTriggerCell>,float> j) { return (i.second<j.second); }


class HGCalClusteringImpl{

private:
    static const unsigned kNSides_ = 2;

public:
  
    HGCalClusteringImpl( const edm::ParameterSet & conf);    

    void eventSetup(const edm::EventSetup& es) {triggerTools_.eventSetup(es);}

    /* dR-algorithms */
    bool isPertinent( const l1t::HGCalTriggerCell & tc, 
                      const l1t::HGCalCluster & clu, 
                      double distXY) const;

    void clusterizeDR( const std::vector<edm::Ptr<l1t::HGCalTriggerCell>> & triggerCellsPtrs,
                       l1t::HGCalClusterBxCollection & clusters
        );

    /* NN-algorithms */    
    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
        );
        
    void clusterizeNN( const std::vector<edm::Ptr<l1t::HGCalTriggerCell>> & triggerCellsPtrs,
                     l1t::HGCalClusterBxCollection & clusters,
                     const HGCalTriggerGeometryBase & triggerGeometry 
        );

    /* FW-algorithms */
    void clusterizeDRNN( const std::vector<edm::Ptr<l1t::HGCalTriggerCell>> & triggerCellsPtrs,
                         l1t::HGCalClusterBxCollection & clusters,
                         const HGCalTriggerGeometryBase & triggerGeometry
        );

    

private:
    
    double siliconSeedThreshold_;
    double siliconTriggerCellThreshold_;
    double scintillatorSeedThreshold_;
    double scintillatorTriggerCellThreshold_;
    double dr_;
    std::string clusteringAlgorithmType_;
    double calibSF_;
    std::vector<double> layerWeights_;
    bool applyLayerWeights_;
    HGCalTriggerTools triggerTools_;

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

    bool areTCneighbour( uint32_t detIDa, uint32_t detIDb, const HGCalTriggerGeometryBase & triggerGeometry );
    
    void removeUnconnectedTCinCluster( l1t::HGCalCluster & cluster,
                                        const HGCalTriggerGeometryBase & triggerGeometry
        );
    
    void calibratePt( l1t::HGCalCluster & cluster );

};

#endif