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

SimDataFormats/CaloAnalysis/interface/MtdSimCluster.h
133 строки4 KB
Matti Kortelainen
Move stored types in io_v1 namespace, reset class versions to 3
22 май 2026, 18:18
22 май 2026, 18:18909f324
// Author: Aurora Perego, Fabio Cossutti - aurora.perego@cern.ch, fabio.cossutti@ts.infn.it
// Date: 05/2023

#ifndef SimDataFormats_CaloAnalysis_MtdSimCluster_h
#define SimDataFormats_CaloAnalysis_MtdSimCluster_h

#include "DataFormats/GeometryVector/interface/LocalPoint.h"
#include "SimDataFormats/CaloAnalysis/interface/SimCluster.h"
#include "SimDataFormats/TrackingHit/interface/PSimHit.h"
#include <vector>

namespace io_v1 {
  class MtdSimCluster : public SimCluster {
    friend std::ostream &operator<<(std::ostream &s, MtdSimCluster const &tp);

  public:
    MtdSimCluster();
    MtdSimCluster(const SimTrack &simtrk);
    MtdSimCluster(EncodedEventId eventID, uint32_t particleID);  // for PU

    // destructor
    ~MtdSimCluster();

    /** @brief add hit time */
    void addHitTime(float time) {
      times_.emplace_back(time);
      ++nsimhits_;
    }

    /** @brief add hit with fraction */
    void addHitAndFraction(uint64_t hit, float fraction) {
      mtdHits_.emplace_back(hit);
      fractions_.emplace_back(fraction);
    }

    /** @brief add hit position*/
    void addHitPosition(LocalPoint pos) { positions_.emplace_back(pos); }

    /** @brief Returns list of hit IDs and fractions for this SimCluster */
    std::vector<std::pair<uint64_t, float>> hits_and_fractions() const {
      assert(mtdHits_.size() == fractions_.size());
      std::vector<std::pair<uint64_t, float>> result;
      result.reserve(mtdHits_.size());
      for (size_t i = 0; i < mtdHits_.size(); ++i) {
        result.emplace_back(mtdHits_[i], fractions_[i]);
      }
      return result;
    }

    /** @brief Returns list of hit IDs and energies for this SimCluster */
    std::vector<std::pair<uint64_t, float>> hits_and_energies() const {
      assert(mtdHits_.size() == energies_.size());
      std::vector<std::pair<uint64_t, float>> result;
      result.reserve(mtdHits_.size());
      for (size_t i = 0; i < mtdHits_.size(); ++i) {
        result.emplace_back(mtdHits_[i], energies_[i]);
      }
      return result;
    }

    /** @brief clear the hits and fractions list */
    void clearHitsAndFractions() {
      std::vector<uint64_t>().swap(mtdHits_);
      std::vector<float>().swap(fractions_);
    }

    /** @brief Returns list of hit IDs and times for this SimCluster */
    std::vector<std::pair<uint64_t, float>> hits_and_times() const {
      assert(mtdHits_.size() == times_.size());
      std::vector<std::pair<uint64_t, float>> result;
      result.reserve(mtdHits_.size());
      for (size_t i = 0; i < mtdHits_.size(); ++i) {
        result.emplace_back(mtdHits_[i], times_[i]);
      }
      return result;
    }

    /** @brief Returns list of hit IDs and times for this SimCluster */
    std::vector<std::pair<uint64_t, LocalPoint>> hits_and_positions() const {
      assert(mtdHits_.size() == times_.size());
      std::vector<std::pair<uint64_t, LocalPoint>> result;
      result.reserve(mtdHits_.size());
      for (size_t i = 0; i < mtdHits_.size(); ++i) {
        result.emplace_back(mtdHits_[i], positions_[i]);
      }
      return result;
    }

    /** @brief Returns list of detIds, rows and columns for this SimCluster */
    std::vector<std::pair<uint32_t, std::pair<uint8_t, uint8_t>>> detIds_and_rows() const {
      std::vector<std::pair<uint32_t, std::pair<uint8_t, uint8_t>>> result;
      result.reserve(mtdHits_.size());
      for (size_t i = 0; i < mtdHits_.size(); ++i) {
        result.emplace_back(
            mtdHits_[i] >> 32,
            std::pair<uint8_t, uint8_t>(static_cast<uint8_t>(mtdHits_[i] >> 16), static_cast<uint8_t>(mtdHits_[i])));
      }
      return result;
    }

    /** @brief clear the times list */
    void clearHitsTime() { std::vector<float>().swap(times_); }

    /** @brief clear the positions list */
    void clearHitsPosition() { std::vector<LocalPoint>().swap(positions_); }

    void clear() {
      clearHitsAndFractions();
      clearHitsEnergy();
      clearHitsTime();
      clearHitsPosition();
    }

    /** @brief add simhit's energy to cluster */
    void addSimHit(const PSimHit &hit) {
      simhit_energy_ += hit.energyLoss();
      ++nsimhits_;
    }

    void setHitProdType(unsigned int offset) { idOffset_ = offset; }

    unsigned int hitProdType() const { return idOffset_; }

  protected:
    std::vector<uint64_t> mtdHits_;
    std::vector<float> times_;
    std::vector<LocalPoint> positions_;
    unsigned int idOffset_{0};
  };
}  // namespace io_v1
using MtdSimCluster = io_v1::MtdSimCluster;

#endif