CalorimeterHitDigi.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// Copyright (C) 2022 Chao Peng, Wouter Deconinck, Sylvester Joosten, Barak Schmookler, David Lawrence
 
// A general digitization for CalorimeterHit from simulation
// 1. Smear energy deposit with a/sqrt(E/GeV) + b + c/E or a/sqrt(E/GeV) (relative value)
// 2. Digitize the energy with dynamic ADC range and add pedestal (mean +- sigma)
// 3. Time conversion with smearing resolution (absolute value)
// 4. Signal is summed if the SumFields are provided
//
// Author: Chao Peng
// Date: 06/02/2021
 
#pragma once
 
#include <DD4hep/IDDescriptor.h>
#include <algorithms/algorithm.h>
#include <algorithms/geo.h>
#include <edm4eic/MCRecoCalorimeterHitAssociationCollection.h>
#include <edm4eic/EDM4eicVersion.h>
#include <edm4hep/EventHeaderCollection.h>
#include <edm4hep/RawCalorimeterHitCollection.h>
#include <edm4hep/SimCalorimeterHitCollection.h>
#include <stdint.h>
#include <functional>
#include <string>
#include <string_view>
 
#include "CalorimeterHitDigiConfig.h"
#include "algorithms/interfaces/UniqueIDGenSvc.h"
#include "algorithms/interfaces/WithPodConfig.h"
 
#if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
#include <edm4eic/MCRecoCalorimeterHitLinkCollection.h>
#endif
 
namespace eicrecon {
 
using CalorimeterHitDigiAlgorithm = algorithms::Algorithm<
    algorithms::Input<edm4hep::EventHeaderCollection, edm4hep::SimCalorimeterHitCollection>,
    algorithms::Output<edm4hep::RawCalorimeterHitCollection,
#if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
                       edm4eic::MCRecoCalorimeterHitLinkCollection,
#endif
                       edm4eic::MCRecoCalorimeterHitAssociationCollection>>;
 
class CalorimeterHitDigi : public CalorimeterHitDigiAlgorithm,
                           public WithPodConfig<CalorimeterHitDigiConfig> {
 
public:
  CalorimeterHitDigi(std::string_view name)
      : CalorimeterHitDigiAlgorithm{
            name,
            {"eventHeader", "inputHitCollection"},
            {"outputRawHitCollection",
#if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
             "outputHitLinks",
#endif
             "outputRawHitAssociationCollection"},
            "Smear energy deposit, digitize within ADC range, add pedestal, "
            "convert time with smearing resolution, and sum signals."} {
  }
 
  void init() final;
  void process(const Input&, const Output&) const final;
 
private:
  // unitless counterparts of inputs
  double stepTDC{0}, tRes{0};
 
  uint64_t id_mask{0};
 
  std::function<double(const edm4hep::SimCalorimeterHit& h)> corrMeanScale;
 
  dd4hep::IDDescriptor id_spec;
 
  enum readout_enum { kSimpleReadout, kPoissonPhotonReadout, kSipmReadout };
  enum readout_enum readoutType { kSimpleReadout };
 
private:
  const algorithms::GeoSvc& m_geo         = algorithms::GeoSvc::instance();
  const algorithms::UniqueIDGenSvc& m_uid = algorithms::UniqueIDGenSvc::instance();
};
 
} // namespace eicrecon