ECCEhpDIRCFastPIDMap.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ECCEhpDIRCFastPIDMap.h

// $Id: $

#ifndef ECCEHPDIRCFASTPIDMAP_H_
#define ECCEHPDIRCFASTPIDMAP_H_

#include "ECCEFastPIDMap.h"

#include <string>

class TH2F;
class TF1;

class ECCEhpDIRCFastPIDMap : public ECCEFastPIDMap {
public:
  ECCEhpDIRCFastPIDMap();
  virtual ~ECCEhpDIRCFastPIDMap();

  PIDCandidate_LogLikelihood_map
  getFastSmearLogLikelihood(int truth_pid, const double momentum,
                            const double theta) const override;

  void ReadMap(const std::string &name);

  TH2F *GetTrrMap() { return fTrrMap; }

private:
  struct DrcPidInfo {
    double probability[5] = {0};
    double sigma[5] = {0};
    double cangle = 0;
    double cctr = 0;
  };

  static int get_pid(int pdg);
  static EICPIDDefs::PIDCandidate get_PIDCandidate(int id);

  TH2F *fTrrMap = nullptr;
  double fMass[5] = {0};
  TF1 *fMs_mom = nullptr;
  TF1 *fMs_thickness = nullptr;
  double fMs_thickness_max = (0);
};

#endif /* ECCEHPDIRCFASTPIDMAP_H_ */