Smear.h

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

#ifndef INCLUDE_EICSMEAR_SMEAR_SMEAR_H_
#define INCLUDE_EICSMEAR_SMEAR_SMEAR_H_

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

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

#include "eicsmear/erhic/Kinematics.h"
#include "eicsmear/erhic/Particle.h"
#include "eicsmear/erhic/ParticleIdentifier.h"
#include "eicsmear/erhic/VirtualParticle.h"
#include "eicsmear/smear/SmearConstants.h"
#include "eicsmear/smear/ParticleMCS.h"

namespace Smear {

inline int PGenre(const erhic::VirtualParticle& prt) {
  int genre(kAll);
  const int id = abs(prt.Id());  // Sign doesn't matter
  if (1 == prt.GetStatus()) {  // Only check stable particles.
    if (id == 11 || id == 22) {
      genre = kElectromagnetic;
    } else if (id >110) {
      genre = kHadronic;
    }  // if
  }  // if
  return genre;
}

inline double FixTheta(double theta) {
  // std::cout << theta << std::endl;
  while (theta < 0. || theta > TMath::Pi()) {
    if (theta < 0.) {
      theta = -theta;
    }  // if
    if (theta > TMath::Pi()) {
      theta = TMath::TwoPi() - theta;
    }  // if
  }
  return theta;
}

inline double FixPhi(double phi) {
  return TVector2::Phi_0_2pi(phi);
}

inline double GetVariable(const erhic::VirtualParticle& prt, KinType kin) {
  double z(0.);
  switch (kin) {
    case kE:
      z = prt.GetE(); break;
    case kP:
      z = prt.GetP(); break;
    case kTheta:
      z = prt.GetTheta(); break;
    case kPhi:
      z = prt.GetPhi(); break;
    case kPz:
      z = prt.GetPz(); break;
    case kPt:
      z = prt.GetPt(); break;
    default:
      break;
  }  // switch
  return z;
}

inline bool IsCoreType(KinType kin) {
  if (kin == kE || kin == kP || kin == kTheta || kin == kPhi) return true;
  return false;
}

int ParseInputFunction(TString &s, KinType &kin1, KinType &kin2);

}  // namespace Smear

#endif  // INCLUDE_EICSMEAR_SMEAR_SMEAR_H_