TrackTrim.hh

Go to the documentation of this file.

#ifndef G_TRACK_TRIM_H
#define G_TRACK_TRIM_H
 
#include <array>
#include <string>
#include <vector>
 
#include "Garfield/Track.hh"
 
namespace Garfield {

 
class TrackTrim : public Track {
 public:
  TrackTrim() : TrackTrim(nullptr) {}
  TrackTrim(Sensor* sensor);
  virtual ~TrackTrim() {}

  bool ReadFile(const std::string& file, const unsigned int nIons = 0,
                const unsigned int nSkip = 0);
  void Print();

  void SetCharge(const double q) {
    if (fabs(q) > 0.) m_q = q;
  }

  void SetWorkFunction(const double w) { m_work = w; }
  double GetWorkFunction() const { return m_work; }
  void SetFanoFactor(const double f) {
    m_fano = f;
    m_fset = true;
  }

  void UnsetFanoFactor() { m_fset = false; }
  double GetFanoFactor() const { return m_fano; }

  void EnableStepSizeLimit(const double dmax) { m_maxStepSize = dmax; }
  void DisableStepSizeLimit() { m_maxStepSize = -1.; }
  void EnableEnergyLossLimit(const double emax) { m_maxLossPerStep = emax; }
  void DisableEnergyLossLimit() { m_maxLossPerStep = -1.; }
 
  void SetParticle(const std::string& part) override;
 
  struct Cluster {
    double x, y, z;  
    double t;        
    double energy;   
    double ekin;     
    int n;           
  };
 
  bool NewTrack(const double x0, const double y0, const double z0,
                const double t0, const double dx0, const double dy0,
                const double dz0) override;
  const std::vector<Cluster>& GetClusters() const { return m_clusters; }
  bool GetCluster(double& xc, double& yc, double& zc, double& tc, int& nc,
                  double& ec, double& extra);
 
 protected:
  double m_work = -1.;
  bool m_fset = false;
  double m_fano = -1.;

  double m_ekin = 0.;
  std::vector<std::vector<std::array<float, 6> > > m_ions;
  size_t m_ion = 0;

  std::vector<Cluster> m_clusters;
  size_t m_cluster = 0;

  double m_maxStepSize = -1.;
  double m_maxLossPerStep = -1.;
 
  void AddIon(const std::vector<float>& x, const std::vector<float>& y,
              const std::vector<float>& z, const std::vector<float>& dedx,
              const std::vector<float>& ekin);
};
}  // namespace Garfield
 
#endif