#include "TMath.h"
#include "TArrayD.h"
#include "AliEMCALPIDUtils.h"
#include "AliLog.h"
ClassImp(AliEMCALPIDUtils)
AliEMCALPIDUtils::AliEMCALPIDUtils():
fPrintInfo(kFALSE), fProbGamma(0.),fProbPiZero(0.),fProbHadron(0.), fWeightHadronEnergy(1.), fWeightGammaEnergy(1.),fWeightPiZeroEnergy(1.)
{
InitParameters();
}
void AliEMCALPIDUtils::ComputePID(Double_t energy, Double_t lambda0)
{
Double_t weightGammaEnergy = DistEnergy(energy, 1);
Double_t weightPiZeroEnergy = DistEnergy(energy, 2);
Double_t weightHadronEnergy = DistEnergy(energy, 3);
Double_t energyhadron=energy;
if(energyhadron<1.)energyhadron=1.;
if (energy<2){energy =2;}
if (energy>55){
energy =55.;
energyhadron=55.;
}
TArrayD paramDistribGamma = DistLambda0(energy, 1);
TArrayD paramDistribPiZero = DistLambda0(energy, 2);
TArrayD paramDistribHadron = DistLambda0(energyhadron, 3);
Bool_t norm = kFALSE;
fProbGamma = TMath::Gaus(lambda0, paramDistribGamma[1], paramDistribGamma[2], norm) * paramDistribGamma[0];
fProbGamma += TMath::Landau(((1-paramDistribGamma[4])-lambda0),paramDistribGamma[4],paramDistribGamma[5],norm)* paramDistribGamma[3];
if(fProbGamma<0.)fProbGamma=0.;
fProbGamma = fProbGamma*weightGammaEnergy;
if(energy>10. || energy < 55.){
fProbPiZero = TMath::Gaus(lambda0, paramDistribPiZero[1], paramDistribPiZero[2], norm) * paramDistribPiZero[0];
fProbPiZero += TMath::Landau(lambda0, paramDistribPiZero[4], paramDistribPiZero[5], norm) * paramDistribPiZero[3];
if(fProbPiZero<0. || energy<5.)fProbPiZero=0.;
fProbPiZero = fProbPiZero*weightPiZeroEnergy;
}
else {
fProbPiZero = 0.;
}
fProbHadron = TMath::Gaus(lambda0, paramDistribHadron[1], paramDistribHadron[2], norm) * paramDistribHadron[0];
fProbHadron += TMath::Landau(lambda0, paramDistribHadron[4], paramDistribHadron[5], norm) * paramDistribHadron[3];
if(fProbHadron<0.)fProbHadron=0.;
fProbHadron = fProbHadron*weightHadronEnergy;
if( (fProbGamma + fProbPiZero + fProbHadron)>0.){
fPIDWeight[0] = fProbGamma / (fProbGamma + fProbPiZero + fProbHadron);
fPIDWeight[1] = fProbPiZero / (fProbGamma+fProbPiZero+fProbHadron);
fPIDWeight[2] = fProbHadron / (fProbGamma+fProbPiZero+fProbHadron);
}
else{
fPIDWeight[0] = 0.;
fPIDWeight[1] = 0.;
fPIDWeight[2] = 0.;
}
SetPID(fPIDWeight[0], 0);
SetPID(fPIDWeight[1], 1);
SetPID(fPIDWeight[2], 2);
if (fPrintInfo) {
AliInfo(Form( "Energy in loop = %f", energy) );
AliInfo(Form( "Lambda0 in loop = %f", lambda0) );
AliInfo(Form( "fProbGamma in loop = %f", fProbGamma) );
AliInfo(Form( "fProbaPiZero = %f", fProbPiZero ));
AliInfo(Form( "fProbaHadron = %f", fProbHadron) );
AliInfo(Form( "PIDWeight in loop = %f ||| %f ||| %f", fPIDWeight[0] , fPIDWeight[1], fPIDWeight[2]) );
AliInfo("********************************************************" );
}
fPIDFinal[AliPID::kElectron] = fPIDWeight[0]/2;
fPIDFinal[AliPID::kMuon] = fPIDWeight[2]/8;
fPIDFinal[AliPID::kPion] = fPIDWeight[2]/8;
fPIDFinal[AliPID::kKaon] = fPIDWeight[2]/8;
fPIDFinal[AliPID::kProton] = fPIDWeight[2]/8;
fPIDFinal[AliPID::kDeuteron] = 0;
fPIDFinal[AliPID::kTriton] = 0;
fPIDFinal[AliPID::kHe3] = 0;
fPIDFinal[AliPID::kAlpha] = 0;
fPIDFinal[AliPID::kPhoton] = fPIDWeight[0]/2;
fPIDFinal[AliPID::kPi0] = fPIDWeight[1] ;
fPIDFinal[AliPID::kNeutron] = fPIDWeight[2]/8;
fPIDFinal[AliPID::kKaon0] = fPIDWeight[2]/8;
fPIDFinal[AliPID::kEleCon] = fPIDWeight[2]/8;
fPIDFinal[AliPID::kUnknown] = fPIDWeight[2]/8;
}
TArrayD AliEMCALPIDUtils::DistLambda0(const Double_t energy, const Int_t type)
{
Double_t constGauss = 0., meanGauss = 0., sigmaGauss = 0.;
Double_t constLandau=0., mpvLandau=0., sigmaLandau=0.;
TArrayD distributionParam(6);
switch (type) {
case 1:
constGauss = PolynomialMixed2(energy, fGamma[0]);
meanGauss = PolynomialMixed2(energy, fGamma[1]);
sigmaGauss = PolynomialMixed2(energy, fGamma[2]);
constLandau = PolynomialMixed2(energy, fGamma[3]);
mpvLandau = PolynomialMixed2(energy, fGamma[4]);
sigmaLandau = PolynomialMixed2(energy, fGamma[5]);
break;
case 2:
constGauss = PolynomialMixed2(energy, fPiZero[0]);
meanGauss = PolynomialMixed2(energy, fPiZero[1]);
sigmaGauss = PolynomialMixed2(energy, fPiZero[2]);
constLandau = PolynomialMixed2(energy, fPiZero[3]);
mpvLandau = PolynomialMixed2(energy, fPiZero[4]);
sigmaLandau = PolynomialMixed2(energy, fPiZero[5]);
break;
case 3:
constGauss = PolynomialMixed2(energy, fHadron[0]);
meanGauss = PolynomialMixed2(energy, fHadron[1]);
sigmaGauss = PolynomialMixed2(energy, fHadron[2]);
constLandau = PolynomialMixed2(energy, fHadron[3]);
mpvLandau = PolynomialMixed2(energy, fHadron[4]);
sigmaLandau = PolynomialMixed2(energy, fHadron[5]);
break;
}
distributionParam[0] = constGauss;
distributionParam[1] = meanGauss;
distributionParam[2] = sigmaGauss;
distributionParam[3] = constLandau;
distributionParam[4] = mpvLandau;
distributionParam[5] = sigmaLandau;
return distributionParam;
}
Double_t AliEMCALPIDUtils::DistEnergy(const Double_t energy, const Int_t type)
{
Double_t constante = 0.;
switch (type) {
case 1:
constante = 1.;
break;
case 2:
constante = 1.;
break;
case 3:
constante = PowerExp(energy, fHadronEnergyProb);
break;
}
return constante;
}
Double_t AliEMCALPIDUtils::Polynomial(const Double_t x, const Double_t *params) const
{
Double_t y = params[0];
y += params[1] * x;
y += params[2] * x * x;
y += params[3] * x * x * x;
y += params[4] * x * x * x * x;
y += params[5] * x * x * x * x * x;
return y;
}
Double_t AliEMCALPIDUtils::Polynomial0(const Double_t *params) const
{
Double_t y = params[0];
return y;
}
Double_t AliEMCALPIDUtils::Polynomialinv(const Double_t x, const Double_t *params) const
{
Double_t y=0.;
if(x>0){
y = params[0];
y += params[1] / x;
y += params[2] / (x * x);
y += params[3] / (x * x * x);
y += params[4] / (x * x * x * x);
y += params[5] / (x * x * x * x * x);
}
return y;
}
Double_t AliEMCALPIDUtils::PolynomialMixed1(const Double_t x, const Double_t *params) const
{
Double_t y=0.;
if(x>0){
y = params[0] / x;
y += params[1] ;
y += params[2] * x ;
}
return y;
}
Double_t AliEMCALPIDUtils::PolynomialMixed2(const Double_t x, const Double_t *params) const
{
Double_t y=0.;
if(x>0){
y = params[0] / ( x * x);
y += params[1] / x;
y += params[2] ;
y += params[3] * x ;
y += params[4] * x * x ;
}
return y;
}
Double_t AliEMCALPIDUtils::PowerExp(const Double_t x, const Double_t *params) const
{
Double_t y = params[0] *TMath::Power( x,params[1]);
y += params[2] *TMath::Exp((x-params[3])*params[4]);
return y;
}
void AliEMCALPIDUtils::InitParameters()
{
fPIDWeight[0] = -1;
fPIDWeight[1] = -1;
fPIDWeight[2] = -1;
for(Int_t i=0; i<AliPID::kSPECIESCN+1; i++)
fPIDFinal[i]= 0;
SetHighFluxParam();
}
void AliEMCALPIDUtils::SetLowFluxParam()
{
Int_t i=0, j=0;
for (i = 0; i < 6; i++) {
for (j = 0; j < 6; j++) {
fGamma[i][j] = fHadron[i][j] = fPiZero[i][j] = 0.;
fGamma1to10[i][j] = fHadron1to10[i][j] = 0.;
}
}
fGamma[0][0] = -7.656908e-01;
fGamma[0][1] = 2.352536e-01;
fGamma[0][2] = 1.555996e-02;
fGamma[0][3] = 2.243525e-04;
fGamma[0][4] = -2.560087e-06;
fGamma[1][0] = 6.500216e+00;
fGamma[1][1] = -2.564958e-01;
fGamma[1][2] = 1.967894e-01;
fGamma[1][3] = -3.982273e-04;
fGamma[1][4] = 2.797737e-06;
fGamma[2][0] = 2.416489e+00;
fGamma[2][1] = -1.601258e-01;
fGamma[2][2] = 3.126839e-02;
fGamma[2][3] = 3.387532e-04;
fGamma[2][4] = -4.089145e-06;
fGamma[3][0] = 0.;
fGamma[3][1] = -2.696008e+00;
fGamma[3][2] = 6.920305e-01;
fGamma[3][3] = -2.281122e-03;
fGamma[3][4] = 0.;
fGamma[4][0] = 2.281564e-01;
fGamma[4][1] = -7.575040e-02;
fGamma[4][2] = 3.813423e-01;
fGamma[4][3] = -1.243854e-04;
fGamma[4][4] = 1.232045e-06;
fGamma[5][0] = -3.290107e-01;
fGamma[5][1] = 3.707545e-02;
fGamma[5][2] = 2.917397e-03;
fGamma[5][3] = 4.695306e-05;
fGamma[5][4] = -3.572981e-07;
fHadron[0][0] = 9.482243e-01;
fHadron[0][1] = -2.780896e-01;
fHadron[0][2] = 2.223507e-02;
fHadron[0][3] = 7.294263e-04;
fHadron[0][4] = -5.665872e-06;
fHadron[1][0] = 0.;
fHadron[1][1] = 0.;
fHadron[1][2] = 2.483298e-01;
fHadron[1][3] = 0.;
fHadron[1][4] = 0.;
fHadron[2][0] = -5.601199e+00;
fHadron[2][1] = 2.097382e+00;
fHadron[2][2] = -2.307965e-01;
fHadron[2][3] = 9.206871e-03;
fHadron[2][4] = -8.887548e-05;
fHadron[3][0] = 6.543101e+00;
fHadron[3][1] = -2.305203e+00;
fHadron[3][2] = 2.761673e-01;
fHadron[3][3] = -5.465855e-03;
fHadron[3][4] = 2.784329e-05;
fHadron[4][0] = -2.443530e+01;
fHadron[4][1] = 8.902578e+00 ;
fHadron[4][2] = -5.265901e-01;
fHadron[4][3] = 2.549111e-02;
fHadron[4][4] = -2.196801e-04;
fHadron[5][0] = 2.102007e-01;
fHadron[5][1] = -3.844418e-02;
fHadron[5][2] = 1.234682e-01;
fHadron[5][3] = -3.866733e-03;
fHadron[5][4] = 3.362719e-05 ;
fPiZero[0][0] = 5.072157e-01;
fPiZero[0][1] = -5.352747e-01;
fPiZero[0][2] = 8.499259e-02;
fPiZero[0][3] = -3.687401e-03;
fPiZero[0][4] = 5.482280e-05;
fPiZero[1][0] = 4.590137e+02;
fPiZero[1][1] = -7.079341e+01;
fPiZero[1][2] = 4.990735e+00;
fPiZero[1][3] = -1.241302e-01;
fPiZero[1][4] = 1.065772e-03;
fPiZero[2][0] = 1.376415e+02;
fPiZero[2][1] = -3.031577e+01;
fPiZero[2][2] = 2.474338e+00;
fPiZero[2][3] = -6.903410e-02;
fPiZero[2][4] = 6.244089e-04;
fPiZero[3][0] = 0.;
fPiZero[3][1] = 1.145983e+00;
fPiZero[3][2] = -2.476052e-01;
fPiZero[3][3] = 1.367373e-02;
fPiZero[3][4] = 0.;
fPiZero[4][0] = -2.097586e+02;
fPiZero[4][1] = 6.300800e+01;
fPiZero[4][2] = -4.038906e+00;
fPiZero[4][3] = 1.088543e-01;
fPiZero[4][4] = -9.362485e-04;
fPiZero[5][0] = -1.671477e+01;
fPiZero[5][1] = 2.995415e+00;
fPiZero[5][2] = -6.040360e-02;
fPiZero[5][3] = -6.137459e-04;
fPiZero[5][4] = 1.847328e-05;
fHadronEnergyProb[0] = 4.767543e-02;
fHadronEnergyProb[1] = -1.537523e+00;
fHadronEnergyProb[2] = 2.956727e-01;
fHadronEnergyProb[3] = -3.051022e+01;
fHadronEnergyProb[4] =-6.036931e-02;
}
void AliEMCALPIDUtils::SetHighFluxParam()
{
Int_t i=0, j=0;
for (i = 0; i < 6; i++) {
for (j = 0; j < 6; j++) {
fGamma[i][j] = fHadron[i][j] = fPiZero[i][j] = 0.;
fGamma1to10[i][j] = fHadron1to10[i][j] = 0.;
}
fGammaEnergyProb[i] = 0.;
fPiZeroEnergyProb[i] = 0.;
fHadronEnergyProb[i] = 0.;
}
fGamma[0][0] = -7.656908e-01;
fGamma[0][1] = 2.352536e-01;
fGamma[0][2] = 1.555996e-02;
fGamma[0][3] = 2.243525e-04;
fGamma[0][4] = -2.560087e-06;
fGamma[1][0] = 6.500216e+00;
fGamma[1][1] = -2.564958e-01;
fGamma[1][2] = 1.967894e-01;
fGamma[1][3] = -3.982273e-04;
fGamma[1][4] = 2.797737e-06;
fGamma[2][0] = 2.416489e+00;
fGamma[2][1] = -1.601258e-01;
fGamma[2][2] = 3.126839e-02;
fGamma[2][3] = 3.387532e-04;
fGamma[2][4] = -4.089145e-06;
fGamma[3][0] = 0.;
fGamma[3][1] = -2.696008e+00;
fGamma[3][2] = 6.920305e-01;
fGamma[3][3] = -2.281122e-03;
fGamma[3][4] = 0.;
fGamma[4][0] = 2.281564e-01;
fGamma[4][1] = -7.575040e-02;
fGamma[4][2] = 3.813423e-01;
fGamma[4][3] = -1.243854e-04;
fGamma[4][4] = 1.232045e-06;
fGamma[5][0] = -3.290107e-01;
fGamma[5][1] = 3.707545e-02;
fGamma[5][2] = 2.917397e-03;
fGamma[5][3] = 4.695306e-05;
fGamma[5][4] = -3.572981e-07;
fHadron[0][0] = 1.519112e-01;
fHadron[0][1] = -8.267603e-02;
fHadron[0][2] = 1.914574e-02;
fHadron[0][3] = -2.677921e-04;
fHadron[0][4] = 5.447939e-06;
fHadron[1][0] = 0.;
fHadron[1][1] = -7.549870e-02;
fHadron[1][2] = 3.930087e-01;
fHadron[1][3] = -2.368500e-03;
fHadron[1][4] = 0.;
fHadron[2][0] = 0.;
fHadron[2][1] = -2.463152e-02;
fHadron[2][2] = 1.349257e-01;
fHadron[2][3] = -1.089440e-03;
fHadron[2][4] = 0.;
fHadron[3][0] = 0.;
fHadron[3][1] = 5.101560e-01;
fHadron[3][2] = 1.458679e-01;
fHadron[3][3] = 4.903068e-04;
fHadron[3][4] = 0.;
fHadron[4][0] = 0.;
fHadron[4][1] = -6.693943e-03;
fHadron[4][2] = 2.444753e-01;
fHadron[4][3] = -5.553749e-05;
fHadron[4][4] = 0.;
fHadron[5][0] = -4.414030e-01;
fHadron[5][1] = 2.292277e-01;
fHadron[5][2] = -2.433737e-02;
fHadron[5][3] = 1.758422e-03;
fHadron[5][4] = -3.001493e-05;
fPiZero[0][0] = 5.072157e-01;
fPiZero[0][1] = -5.352747e-01;
fPiZero[0][2] = 8.499259e-02;
fPiZero[0][3] = -3.687401e-03;
fPiZero[0][4] = 5.482280e-05;
fPiZero[1][0] = 4.590137e+02;
fPiZero[1][1] = -7.079341e+01;
fPiZero[1][2] = 4.990735e+00;
fPiZero[1][3] = -1.241302e-01;
fPiZero[1][4] = 1.065772e-03;
fPiZero[2][0] = 1.376415e+02;
fPiZero[2][1] = -3.031577e+01;
fPiZero[2][2] = 2.474338e+00;
fPiZero[2][3] = -6.903410e-02;
fPiZero[2][4] = 6.244089e-04;
fPiZero[3][0] = 0.;
fPiZero[3][1] = 1.145983e+00;
fPiZero[3][2] = -2.476052e-01;
fPiZero[3][3] = 1.367373e-02;
fPiZero[3][4] = 0.;
fPiZero[4][0] = -2.097586e+02;
fPiZero[4][1] = 6.300800e+01;
fPiZero[4][2] = -4.038906e+00;
fPiZero[4][3] = 1.088543e-01;
fPiZero[4][4] = -9.362485e-04;
fPiZero[5][0] = -1.671477e+01;
fPiZero[5][1] = 2.995415e+00;
fPiZero[5][2] = -6.040360e-02;
fPiZero[5][3] = -6.137459e-04;
fPiZero[5][4] = 1.847328e-05;
fHadronEnergyProb[0] = 0.;
fHadronEnergyProb[1] = 0.;
fHadronEnergyProb[2] = 6.188452e-02;
fHadronEnergyProb[3] = 2.030230e+00;
fHadronEnergyProb[4] = -6.402242e-02;
}