HGVHistoProducerAlgo.h

Go to the documentation of this file.

#ifndef Validation_HGCalValidation_HGVHistoProducerAlgo_h
#define Validation_HGCalValidation_HGVHistoProducerAlgo_h

/* \author HGCal
 */

#include <iostream>
#include <fstream>
#include <vector>
#include <unordered_map>

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"

#include "DataFormats/CaloRecHit/interface/CaloClusterFwd.h"
#include "DataFormats/HGCRecHit/interface/HGCRecHitCollections.h"
#include "DataFormats/ForwardDetId/interface/HGCalDetId.h"
#include "DataFormats/HGCRecHit/interface/HGCRecHit.h"
#include "DataFormats/ParticleFlowReco/interface/HGCalMultiCluster.h"

#include "RecoLocalCalo/HGCalRecAlgos/interface/RecHitTools.h"
#include "SimDataFormats/CaloAnalysis/interface/CaloParticle.h"
#include "SimDataFormats/Vertex/interface/SimVertex.h"
#include "RecoLocalCalo/HGCalRecProducers/interface/HGCalClusteringAlgoBase.h"

#include "DQMServices/Core/interface/DQMStore.h"

struct HGVHistoProducerAlgoHistograms {
  //Info
  //To be able to spot any issues both in -z and +z a layer id was introduced
  //that spans from 0 to 103 for hgcal_v9 geometry. The mapping for hgcal_v9 is:
  //-z: 0->51
  //+z: 52->103
  //We will pick the numbers below from RecHitTools just to avoid future problems
  dqm::reco::MonitorElement* lastLayerEEzm;  // last layer of EE -z
  dqm::reco::MonitorElement* lastLayerFHzm;  // last layer of FH -z
  dqm::reco::MonitorElement* maxlayerzm;     // last layer of BH -z
  dqm::reco::MonitorElement* lastLayerEEzp;  // last layer of EE +z
  dqm::reco::MonitorElement* lastLayerFHzp;  // last layer of FH +z
  dqm::reco::MonitorElement* maxlayerzp;     // last layer of BH +z

  //1D
  std::vector<dqm::reco::MonitorElement*> h_cluster_eta;
  std::vector<dqm::reco::MonitorElement*> h_mixedhitscluster_zminus;
  std::vector<dqm::reco::MonitorElement*> h_mixedhitscluster_zplus;
  std::vector<dqm::reco::MonitorElement*> h_energyclustered_zminus;
  std::vector<dqm::reco::MonitorElement*> h_energyclustered_zplus;
  std::vector<dqm::reco::MonitorElement*> h_longdepthbarycentre_zminus;
  std::vector<dqm::reco::MonitorElement*> h_longdepthbarycentre_zplus;

  std::unordered_map<int, dqm::reco::MonitorElement*> h_clusternum_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_energyclustered_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_score_layercl2caloparticle_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_score_caloparticle2layercl_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_energy_vs_score_caloparticle2layercl_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_energy_vs_score_layercl2caloparticle_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_sharedenergy_caloparticle2layercl_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_sharedenergy_caloparticle2layercl_vs_eta_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_sharedenergy_caloparticle2layercl_vs_phi_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_sharedenergy_layercl2caloparticle_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_sharedenergy_layercl2caloparticle_vs_eta_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_sharedenergy_layercl2caloparticle_vs_phi_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_num_caloparticle_eta_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_numDup_caloparticle_eta_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_denom_caloparticle_eta_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_num_caloparticle_phi_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_numDup_caloparticle_phi_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_denom_caloparticle_phi_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_num_layercl_eta_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_numMerge_layercl_eta_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_denom_layercl_eta_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_num_layercl_phi_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_numMerge_layercl_phi_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_denom_layercl_phi_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_cellAssociation_perlayer;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_clusternum_perthick;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_cellsenedens_perthick;

  std::unordered_map<std::string, dqm::reco::MonitorElement*> h_cellsnum_perthickperlayer;
  std::unordered_map<std::string, dqm::reco::MonitorElement*> h_distancetoseedcell_perthickperlayer;
  std::unordered_map<std::string, dqm::reco::MonitorElement*> h_distancetoseedcell_perthickperlayer_eneweighted;
  std::unordered_map<std::string, dqm::reco::MonitorElement*> h_distancetomaxcell_perthickperlayer;
  std::unordered_map<std::string, dqm::reco::MonitorElement*> h_distancetomaxcell_perthickperlayer_eneweighted;
  std::unordered_map<std::string, dqm::reco::MonitorElement*> h_distancebetseedandmaxcell_perthickperlayer;
  std::unordered_map<std::string, dqm::reco::MonitorElement*>
      h_distancebetseedandmaxcellvsclusterenergy_perthickperlayer;

  std::unordered_map<int, dqm::reco::MonitorElement*> h_caloparticle_eta;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_caloparticle_eta_Zorigin;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_caloparticle_energy;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_caloparticle_pt;
  std::unordered_map<int, dqm::reco::MonitorElement*> h_caloparticle_phi;

  //For multiclusters
  std::vector<dqm::reco::MonitorElement*> h_score_multicl2caloparticle;
  std::vector<dqm::reco::MonitorElement*> h_score_caloparticle2multicl;
  std::vector<dqm::reco::MonitorElement*> h_energy_vs_score_multicl2caloparticle;
  std::vector<dqm::reco::MonitorElement*> h_energy_vs_score_caloparticle2multicl;
  std::vector<dqm::reco::MonitorElement*> h_num_multicl_eta;
  std::vector<dqm::reco::MonitorElement*> h_num_multicl_phi;
  std::vector<dqm::reco::MonitorElement*> h_numMerge_multicl_eta;
  std::vector<dqm::reco::MonitorElement*> h_numMerge_multicl_phi;
  std::vector<dqm::reco::MonitorElement*> h_sharedenergy_multicl2caloparticle;
  std::vector<dqm::reco::MonitorElement*> h_sharedenergy_caloparticle2multicl;
  std::vector<dqm::reco::MonitorElement*> h_sharedenergy_multicl2caloparticle_vs_eta;
  std::vector<dqm::reco::MonitorElement*> h_sharedenergy_multicl2caloparticle_vs_phi;
  std::vector<dqm::reco::MonitorElement*> h_sharedenergy_caloparticle2multicl_vs_eta;
  std::vector<dqm::reco::MonitorElement*> h_sharedenergy_caloparticle2multicl_vs_phi;
  std::vector<dqm::reco::MonitorElement*> h_denom_multicl_eta;
  std::vector<dqm::reco::MonitorElement*> h_denom_multicl_phi;
  std::vector<dqm::reco::MonitorElement*> h_num_caloparticle_eta;
  std::vector<dqm::reco::MonitorElement*> h_num_caloparticle_phi;
  std::vector<dqm::reco::MonitorElement*> h_numDup_multicl_eta;
  std::vector<dqm::reco::MonitorElement*> h_numDup_multicl_phi;
  std::vector<dqm::reco::MonitorElement*> h_denom_caloparticle_eta;
  std::vector<dqm::reco::MonitorElement*> h_denom_caloparticle_phi;
  std::vector<dqm::reco::MonitorElement*> h_multiclusternum;
  std::vector<dqm::reco::MonitorElement*> h_contmulticlusternum;
  std::vector<dqm::reco::MonitorElement*> h_noncontmulticlusternum;
  std::vector<dqm::reco::MonitorElement*> h_clusternum_in_multicluster;
  std::vector<std::unordered_map<int, dqm::reco::MonitorElement*>> h_clusternum_in_multicluster_perlayer;
  std::vector<dqm::reco::MonitorElement*> h_multiplicityOfLCinMCL;
  std::vector<dqm::reco::MonitorElement*> h_multiplicity_numberOfEventsHistogram;
  std::vector<dqm::reco::MonitorElement*> h_multiplicity_zminus_numberOfEventsHistogram;
  std::vector<dqm::reco::MonitorElement*> h_multiplicity_zplus_numberOfEventsHistogram;
  std::vector<dqm::reco::MonitorElement*> h_multiplicityOfLCinMCL_vs_layercluster;
  std::vector<dqm::reco::MonitorElement*> h_multiplicityOfLCinMCL_vs_layercluster_zminus;
  std::vector<dqm::reco::MonitorElement*> h_multiplicityOfLCinMCL_vs_layercluster_zplus;
  std::vector<dqm::reco::MonitorElement*> h_multiplicityOfLCinMCL_vs_layerclusterenergy;
  std::vector<dqm::reco::MonitorElement*> h_clusternum_in_multicluster_vs_layer;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_pt;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_eta;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_phi;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_energy;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_x;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_y;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_z;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_firstlayer;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_lastlayer;
  std::vector<dqm::reco::MonitorElement*> h_multicluster_layersnum;
};

using Density = hgcal_clustering::Density;

class HGVHistoProducerAlgo {
public:
  typedef dqm::legacy::DQMStore DQMStore;
  typedef dqm::legacy::MonitorElement MonitorElement;

  HGVHistoProducerAlgo(const edm::ParameterSet& pset);
  ~HGVHistoProducerAlgo();

  using Histograms = HGVHistoProducerAlgoHistograms;

  void bookInfo(DQMStore::IBooker& ibook, Histograms& histograms);
  void bookCaloParticleHistos(DQMStore::IBooker& ibook, Histograms& histograms, int pdgid);

  void bookClusterHistos(DQMStore::IBooker& ibook,
                         Histograms& histograms,
                         unsigned layers,
                         std::vector<int> thicknesses,
                         std::string pathtomatbudfile);
  void bookMultiClusterHistos(DQMStore::IBooker& ibook, Histograms& histograms, unsigned layers);
  void layerClusters_to_CaloParticles(const Histograms& histograms,
                                      const reco::CaloClusterCollection& clusters,
                                      std::vector<CaloParticle> const& cP,
                                      std::vector<size_t> const& cPIndices,
                                      std::map<DetId, const HGCRecHit*> const&,
                                      unsigned layers) const;
  void multiClusters_to_CaloParticles(const Histograms& histograms,
                                      int count,
                                      const std::vector<reco::HGCalMultiCluster>& multiClusters,
                                      std::vector<CaloParticle> const& cP,
                                      std::vector<size_t> const& cPIndices,
                                      std::map<DetId, const HGCRecHit*> const&,
                                      unsigned layers) const;
  void fill_info_histos(const Histograms& histograms, unsigned layers) const;
  void fill_caloparticle_histos(const Histograms& histograms,
                                int pdgid,
                                const CaloParticle& caloparticle,
                                std::vector<SimVertex> const& simVertices) const;
  void fill_cluster_histos(const Histograms& histograms, int count, const reco::CaloCluster& cluster) const;
  void fill_generic_cluster_histos(const Histograms& histograms,
                                   int count,
                                   const reco::CaloClusterCollection& clusters,
                                   const Density& densities,
                                   std::vector<CaloParticle> const& cP,
                                   std::vector<size_t> const& cPIndices,
                                   std::map<DetId, const HGCRecHit*> const&,
                                   std::map<double, double> cummatbudg,
                                   unsigned layers,
                                   std::vector<int> thicknesses) const;
  void fill_multi_cluster_histos(const Histograms& histograms,
                                 int count,
                                 const std::vector<reco::HGCalMultiCluster>& multiClusters,
                                 std::vector<CaloParticle> const& cP,
                                 std::vector<size_t> const& cPIndices,
                                 std::map<DetId, const HGCRecHit*> const&,
                                 unsigned layers) const;
  double distance2(const double x1, const double y1, const double x2, const double y2) const;
  double distance(const double x1, const double y1, const double x2, const double y2) const;

  void setRecHitTools(std::shared_ptr<hgcal::RecHitTools> recHitTools);

  DetId findmaxhit(const reco::CaloCluster& cluster, std::map<DetId, const HGCRecHit*> const&) const;

  struct detIdInfoInCluster {
    bool operator==(const detIdInfoInCluster& o) const { return clusterId == o.clusterId; };
    long unsigned int clusterId;
    float fraction;
  };

  struct detIdInfoInMultiCluster {
    bool operator==(const detIdInfoInMultiCluster& o) const { return multiclusterId == o.multiclusterId; };
    unsigned int multiclusterId;
    long unsigned int clusterId;
    float fraction;
  };

  struct caloParticleOnLayer {
    unsigned int caloParticleId;
    float energy = 0;
    std::vector<std::pair<DetId, float>> hits_and_fractions;
    std::unordered_map<int, std::pair<float, float>> layerClusterIdToEnergyAndScore;
  };

private:
  double getEta(double eta) const;

  std::shared_ptr<hgcal::RecHitTools> recHitTools_;

  //private data members
  double minEta_, maxEta_;
  int nintEta_;
  bool useFabsEta_;
  double minEne_, maxEne_;
  int nintEne_;
  double minPt_, maxPt_;
  int nintPt_;
  double minPhi_, maxPhi_;
  int nintPhi_;
  double minMixedHitsCluster_, maxMixedHitsCluster_;
  int nintMixedHitsCluster_;
  double minEneCl_, maxEneCl_;
  int nintEneCl_;
  double minLongDepBary_, maxLongDepBary_;
  int nintLongDepBary_;
  double minZpos_, maxZpos_;
  int nintZpos_;
  double minTotNClsperlay_, maxTotNClsperlay_;
  int nintTotNClsperlay_;
  double minEneClperlay_, maxEneClperlay_;
  int nintEneClperlay_;
  double minScore_, maxScore_;
  int nintScore_;
  double minSharedEneFrac_, maxSharedEneFrac_;
  int nintSharedEneFrac_;
  double minMCLSharedEneFrac_, maxMCLSharedEneFrac_;
  int nintMCLSharedEneFrac_;
  double minTotNClsperthick_, maxTotNClsperthick_;
  int nintTotNClsperthick_;
  double minTotNcellsperthickperlayer_, maxTotNcellsperthickperlayer_;
  int nintTotNcellsperthickperlayer_;
  double minDisToSeedperthickperlayer_, maxDisToSeedperthickperlayer_;
  int nintDisToSeedperthickperlayer_;
  double minDisToSeedperthickperlayerenewei_, maxDisToSeedperthickperlayerenewei_;
  int nintDisToSeedperthickperlayerenewei_;
  double minDisToMaxperthickperlayer_, maxDisToMaxperthickperlayer_;
  int nintDisToMaxperthickperlayer_;
  double minDisToMaxperthickperlayerenewei_, maxDisToMaxperthickperlayerenewei_;
  int nintDisToMaxperthickperlayerenewei_;
  double minDisSeedToMaxperthickperlayer_, maxDisSeedToMaxperthickperlayer_;
  int nintDisSeedToMaxperthickperlayer_;
  double minClEneperthickperlayer_, maxClEneperthickperlayer_;
  int nintClEneperthickperlayer_;
  double minCellsEneDensperthick_, maxCellsEneDensperthick_;
  int nintCellsEneDensperthick_;
  double minTotNMCLs_, maxTotNMCLs_;
  int nintTotNMCLs_;
  double minTotNClsinMCLs_, maxTotNClsinMCLs_;
  int nintTotNClsinMCLs_;
  double minTotNClsinMCLsperlayer_, maxTotNClsinMCLsperlayer_;
  int nintTotNClsinMCLsperlayer_;
  double minMplofLCs_, maxMplofLCs_;
  int nintMplofLCs_;
  double minSizeCLsinMCLs_, maxSizeCLsinMCLs_;
  int nintSizeCLsinMCLs_;
  double minClEnepermultiplicity_, maxClEnepermultiplicity_;
  int nintClEnepermultiplicity_;
  double minX_, maxX_;
  int nintX_;
  double minY_, maxY_;
  int nintY_;
  double minZ_, maxZ_;
  int nintZ_;
};

#endif