SiliconDetectorDriftDiffusionAlgo.h

Go to the documentation of this file.

/*
 * SiliconDetectorDriftDiffusionAlgo.h
 *
 *  Created on: 23 Sep 2020
 *      Author: Valerio Formato
 */

#ifndef HERD_SILICONDETECTORDRIFTDIFFUSIONALGO_H_
#define HERD_SILICONDETECTORDRIFTDIFFUSIONALGO_H_

// HerdSoftware headers
#include "dataobjects/SiliconDetectorGeoParamsColl.h"
#include "dataobjects/SiliconDetectorHitsColl.h"
#include "dataobjects/SiliconDetectorParticleHitsColl.h"

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

using namespace EA;

namespace Herd {

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

namespace SiliconDetectorDigitizationTools {
class SDChargePacket;
}

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

  bool Initialize() override;

  bool Process() override;

  std::vector<std::string> FreeObjects(const std::vector<std::string> &objs, Memory::Status memStatus) override;

protected:
  // Parameters
  float m_pitch;
  std::string m_detName; 

  float m_minStepSize;

  // diffusion parameters
  std::string m_diffusionType;

  // - Gaussian drift parameters
  float m_posEffDiffCoeff;
  float m_negEffDiffCoeff;

  // - non-linear Gaussian drift parameters
  float m_biasVoltage;
  float m_intrinsicVoltage;

private:
  enum class DiffusionType { Gaussian, GaussianNonLinear };
  DiffusionType m_diffType = DiffusionType::GaussianNonLinear;

  SiliconDetectorHitsColl m_siliconDetectorDigHitsColl;
  observer_ptr<SiliconDetectorGeoParamsColl> m_origSiliconDetectorGeoParamsColl;
  std::shared_ptr<SiliconDetectorGeoParamsColl> m_digSiliconDetectorGeoParamsColl;

  StorePtr m_evStore, m_globStore;

  // Helper functions

  void GenerateHits(const SiliconDetectorParticleHitsColl &particleHitsColl, const SiliconDetectorHitsColl &hitsColl,
                    SiliconDetectorHitsColl &digHitsColl);

  SiliconDetectorGeoParams CreateGeoParams(const SiliconDetectorGeoParams &origGeoParams);

  std::vector<Point> PartitionTrajectory(const ParticleHit &hit);

  void AccumulateSteps(std::vector<Hit> &layerHits, const SiliconDetectorGeoParams &geoParams, unsigned int firstStrip,
                       unsigned int lastStrip, const SiliconDetectorDigitizationTools::SDChargePacket &packet,
                       float edep);

  // Helper function to generate different kinds of charge packets
  std::unique_ptr<SiliconDetectorDigitizationTools::SDChargePacket>
  CreatePacket(Point position, DiffusionType diffType, const SiliconDetectorGeoParams &geoParams);
};

} // namespace Herd

#endif /* HERD_SILICONDETECTORDriftDiffusionALGO_H_ */