FitDigitizerAlgo.h

Go to the documentation of this file.

/*
 * FitDigitizerAlgo.h
 *
 *  Created on: 20 Feb 2020
 *      Author: Valerio Formato (based on code by Junjing Wang)
 */

#ifndef HERD_FITDIGITIZERALGO_H_
#define HERD_FITDIGITIZERALGO_H_

// HerdSoftware headers
#include "algorithms/digitization/FitDigitizationTools/AttenuationTool.h"
#include "algorithms/digitization/FitDigitizationTools/DistributionChannelTool.h"
#include "algorithms/digitization/FitDigitizationTools/DistributionFiberTool.h"
#include "algorithms/digitization/FitDigitizationTools/PhotonTool.h"
#include "algorithms/digitization/FitDigitizationTools/SiPMTool.h"
#include "dataobjects/FitChannelInfoColl.h"
#include "dataobjects/FitGeoParamsColl.h"
#include "dataobjects/FitHitsColl.h"
#include "dataobjects/FitParticleHitsColl.h"

// EventAnalysis headers
#include "algorithm/Algorithm.h"

using namespace EA;

namespace Herd {

// clang-format off
// clang-format on

class FitDigitizerAlgo : public Algorithm {
public:
  FitDigitizerAlgo(const std::string &name);

  bool Initialize() override;

  bool Process() override;

private:
  StorePtr m_evStore, m_globStore;

  FitHitsColl m_fitParticleHitsColl;
  observer_ptr<FitGeoParamsColl> m_origFitGeoParamsColl;
  std::shared_ptr<FitGeoParamsColl> m_digFitGeoParamsColl;
  std::shared_ptr<FitChannelInfoColl> m_digFitChannelInfoColl;

  FitGeoParams createGeoParams(const FitGeoParams &origGeoParams);
  FitChannelInfo createChannelInfo(const FitGeoParams &digGeoParams);
  std::vector<float> generateSipmGains(const FitGeoParams &digGeoParams);

  SidesArray<std::vector<std::vector<float>>> m_sipmGains;

  // Parameters
  bool m_publishChannelInfo = true;
  bool m_simulateNoise = true;
  bool m_simulateIntraChannelXTalk = true;
  unsigned int m_nTopMatsPerLayer = 11;
  unsigned int m_nSideShortMatsPerLayer = 10;
  unsigned int m_nSideLongMatsPerLayer = 7;
  float m_topMatsSize = defaultValue<float>;
  float m_sideShortMatsSize = defaultValue<float>;
  float m_sideLongMatsSize = defaultValue<float>;

  double m_sipmGainVariation = 0.0125; // relative fluctuation of the gain
  double m_adcNoise = 0.1;             // Sigma of the noise in the electronics ADC conversion

  // utility tools
  std::unique_ptr<FitDigitizationTools::PhotonTool> m_photonTool;
  std::unique_ptr<FitDigitizationTools::DistributionFiberTool> m_distrFiberTool;
  std::unique_ptr<FitDigitizationTools::SiPMTool> m_sipmTool;

  std::mt19937 m_generator;
  std::normal_distribution<double> m_gauss_dist;
};

} // namespace Herd

#endif /* HERD_PSDGEOMETRICDIGITIZERALGO_H_ */