ParticleID.h

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

#ifndef INCLUDE_EICSMEAR_SMEAR_PARTICLEID_H_
#define INCLUDE_EICSMEAR_SMEAR_PARTICLEID_H_

#include <cmath>
#include <fstream>
#include <iostream>
#include <sstream>
#include <vector>

#include <TF1.h>
#include <TF2.h>
#include <TLorentzVector.h>
#include <TRandom3.h>
#include <TROOT.h>
#include <TSystem.h>
#include <TString.h>

#include "eicsmear/smear/Device.h"
#include "eicsmear/erhic/Kinematics.h"
#include "eicsmear/erhic/Particle.h"
#include "eicsmear/erhic/ParticleIdentifier.h"
#include "eicsmear/smear/Smear.h"
#include "eicsmear/smear/Smearer.h"
#include "eicsmear/erhic/VirtualParticle.h"

namespace Smear {

struct ParticleID : public Smearer {
  ParticleID();

  explicit ParticleID(TString filename);

  virtual ~ParticleID();

  void SetPMatrixPath(TString);

  void SetPIDUseMC(bool useMc);

  TRandom3& GetRandomGenerator();

  void SetRanSeed(int seed);

  void GetAcceptanceFromDevice(const Device&);

  void SetPMatrixSize();

  void SetupProbabilityArray();

  virtual ParticleID* Clone(const char* = "") const;

  int Wild(int pbin, int trueID);

  int InListOfTrue(int ID);

  int InListOfFalse(int ID);

  void ReadP(TString filename);

  void Smear(const erhic::VirtualParticle&, ParticleMCS&);

  void Speak();

  virtual void Clear(Option_t* = "");

  virtual void Print(Option_t* = "") const;

  TRandom3 Ran;
  TString PMatPath;
  std::vector<int> TrueIdent;
  std::vector<int> FalseIdent;
  std::vector<double> PMin;
  std::vector<double> PMax;
  // Indices are for momentum bin, true ID, and false ID
  // PMatrix[i][j][k] stores the probability for a particle of type
  // TrueIdent[j] in the momentum bin (PMin[i], PMax[i]) to be
  // identified as a particle of type FalseIdent[k].
  std::vector< std::vector<std::vector<double> > > PMatrix;
  // Range is the cumulative probability distribution of PMatrix.
  std::vector< std::vector<std::vector<double> > > Range;
  bool bUseMC;

  ClassDef(Smear::ParticleID, 1)
};

inline void ParticleID::SetPMatrixPath(TString str) {
  PMatPath = str;
  ReadP(PMatPath);
}

inline void ParticleID::SetPIDUseMC(bool d) {
  bUseMC = d;
}

inline TRandom3& ParticleID::GetRandomGenerator() {
  return Ran;
}

inline void ParticleID::SetRanSeed(int n) {
  Ran.SetSeed(n);
}

inline void ParticleID::GetAcceptanceFromDevice(const Device& dev) {
  Accept = dev.Accept;
}

inline ParticleID* ParticleID::Clone(const char*) const {
  // const char* argument comes from TObject::Clone(), usused here.
  return new ParticleID(*this);
}

}  // namespace Smear

#endif  // INCLUDE_EICSMEAR_SMEAR_PARTICLEID_H_