SimDataFormats/CaloAnalysis/interface/SimTauCPLink.h - master - githubmirror/cmssw

SimDataFormats/CaloAnalysis/interface/SimTauCPLink.h
167 строк5 KB
Matti Kortelainen
Move stored types in io_v1 namespace, reset class versions to 3
22 май 2026, 18:18
22 май 2026, 18:18909f324
#ifndef SimDataFormats_SimTauCPLink_h
#define SimDataFormats_SimTauCPLink_h

#include "DataFormats/HGCalReco/interface/TICLCandidate.h"
#include "SimDataFormats/CaloAnalysis/interface/CaloParticleFwd.h"
#include "SimDataFormats/CaloAnalysis/interface/CaloParticle.h"
#include "SimDataFormats/CaloAnalysis/interface/SimCluster.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

namespace io_v1 {
  class SimTauCPLink {
  public:
    SimTauCPLink() {}
    ~SimTauCPLink() {}
    struct DecayNav {
      int pdgId_;
      int resonance_idx_;
      int calo_particle_idx_;
      int gen_particle_idx_;
      int pdgId() const { return pdgId_; }
      int resonance_idx() const { return resonance_idx_; }
      int calo_particle_idx() const { return calo_particle_idx_; }
      int gen_particle_idx() const { return gen_particle_idx_; }

    private:
    };

    std::vector<std::pair<int, int>> resonances;
    std::vector<DecayNav> leaves;
    CaloParticleRefVector calo_particle_leaves;
    int decayMode;

    enum decayModes {
      kNull = -1,
      kOneProng0PiZero,    // 0
      kOneProng1PiZero,    // 1
      kOneProng2PiZero,    // 2
      kOneProng3PiZero,    // 3
      kOneProngNPiZero,    // 4
      kTwoProng0PiZero,    // 5
      kTwoProng1PiZero,    // 6
      kTwoProng2PiZero,    // 7
      kTwoProng3PiZero,    // 8
      kTwoProngNPiZero,    // 9
      kThreeProng0PiZero,  // 10
      kThreeProng1PiZero,  // 11
      kThreeProng2PiZero,  // 12
      kThreeProng3PiZero,  // 13
      kThreeProngNPiZero,  // 14
      kRareDecayMode,      // 15
      kElectron,           // 16
      kMuon                // 17
    };

    void dump(void) const {
      LogDebug("SimTauProducer")
          .format("Decay mode: {} ", buildDecayModes())
          .format("Leaves: {} ", leaves.size())
          .format("Resonances: {}", resonances.size());
      for (auto const &l : leaves) {
        LogDebug("SimTauProducer")
            .format("L {} {} CP: {} GenP idx: {}",
                    l.pdgId(),
                    l.resonance_idx(),
                    (int)((l.calo_particle_idx() == -1) ? -1 : calo_particle_leaves[l.calo_particle_idx()].key()),
                    l.gen_particle_idx());
      }
      for (auto const &r : resonances) {
        LogDebug("SimTauProducer").format("R {} {}", r.first, r.second);
      }
    }

    void dumpDecay(const std::pair<int, int> &entry) const {
      if (entry.second == -1) {  // No intermediate mother.
        LogDebug("SimTauProducer").format("{} {}", entry.first, entry.second);
      } else {
        LogDebug("SimTauProducer").format("{} {} coming from: ", entry.first, entry.second);
        auto const &mother = resonances[entry.second];
        dumpDecay(mother);
      }
    }

    void dumpFullDecay(void) const {
      for (auto const &leaf : leaves) {
        dumpDecay({leaf.pdgId(), leaf.resonance_idx()});
      }
    }

    int buildDecayModes() const {
      int numElectrons = 0;
      int numMuons = 0;
      int numHadrons = 0;
      int numPhotons = 0;
      for (auto leaf : leaves) {
        int pdg_id = abs(leaf.pdgId());
        switch (pdg_id) {
          case 22:
            numPhotons++;
            break;
          case 11:
            numElectrons++;
            break;
          case 13:
            numMuons++;
            break;
          case 16:
            break;
          default:
            numHadrons++;
        }
      }

      if (numElectrons == 1)
        return kElectron;
      else if (numMuons == 1)
        return kMuon;
      switch (numHadrons) {
        case 1:
          switch (numPhotons) {
            case 0:
              return kOneProng0PiZero;
            case 2:
              return kOneProng1PiZero;
            case 4:
              return kOneProng2PiZero;
            case 6:
              return kOneProng3PiZero;
            default:
              return kOneProngNPiZero;
          }
        case 2:
          switch (numPhotons) {
            case 0:
              return kTwoProng0PiZero;
            case 2:
              return kTwoProng1PiZero;
            case 4:
              return kTwoProng2PiZero;
            case 6:
              return kTwoProng3PiZero;
            default:
              return kTwoProngNPiZero;
          }
        case 3:
          switch (numPhotons) {
            case 0:
              return kThreeProng0PiZero;
            case 2:
              return kThreeProng1PiZero;
            case 4:
              return kThreeProng2PiZero;
            case 6:
              return kThreeProng3PiZero;
            default:
              return kThreeProngNPiZero;
          }
        default:
          return kRareDecayMode;
      }
    }

  private:
  };
}  // namespace io_v1
using SimTauCPLink = io_v1::SimTauCPLink;

#endif  //SimTauCPLink