ParticleID.h

Go to the documentation of this file.

 
#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_