#include "AliHMPIDRecon.h"
#include "AliHMPIDCluster.h"
#include <TRotation.h> //TracePhot()
#include <TH1D.h> //HoughResponse()
#include <TClonesArray.h> //CkovAngle()
#include <AliESDtrack.h>
#include <AliESDfriendTrack.h>
AliHMPIDRecon::AliHMPIDRecon():
TNamed("RichRec","RichPat"),
fPhotCnt(-1),
fPhotFlag(0x0),
fPhotClusIndex(0x0),
fPhotCkov(0x0),
fPhotPhi(0x0),
fPhotWei(0x0),
fCkovSigma2(0),
fIsWEIGHT(kFALSE),
fDTheta(0.001),
fWindowWidth(0.045),
fRingArea(0),
fRingAcc(0),
fTrkDir(0,0,1),
fTrkPos(30,40),
fMipPos(0,0),
fPc(0,0),
fParam(AliHMPIDParam::Instance())
{
fParam->SetRefIdx(fParam->MeanIdxRad());
}
void AliHMPIDRecon::InitVars(Int_t n)
{
if(n<=0) return;
fPhotFlag = new Int_t[n];
fPhotClusIndex = new Int_t[n];
fPhotCkov = new Double_t[n];
fPhotPhi = new Double_t[n];
fPhotWei = new Double_t[n];
}
void AliHMPIDRecon::DeleteVars()const
{
delete [] fPhotFlag;
delete [] fPhotClusIndex;
delete [] fPhotCkov;
delete [] fPhotPhi;
delete [] fPhotWei;
}
void AliHMPIDRecon::CkovAngle(AliESDtrack *pTrk,TClonesArray *pCluLst,Int_t index,Double_t nmean,Float_t xRa,Float_t yRa)
{
const Int_t nMinPhotAcc = 3;
Int_t nClusTot = pCluLst->GetEntries();
InitVars(nClusTot);
Float_t xPc,yPc,th,ph;
pTrk->GetHMPIDtrk(xPc,yPc,th,ph);
SetTrack(xRa,yRa,th,ph);
fParam->SetRefIdx(nmean);
Float_t mipX=-1,mipY=-1;
Int_t chId=-1,mipQ=-1,sizeClu = -1;
fPhotCnt=0;
Int_t nPads = 0;
for (Int_t iClu=0; iClu<pCluLst->GetEntriesFast();iClu++){
AliHMPIDCluster *pClu=(AliHMPIDCluster*)pCluLst->UncheckedAt(iClu);
nPads+=pClu->Size();
if(iClu == index) {
mipX = pClu->X();
mipY = pClu->Y();
mipQ=(Int_t)pClu->Q();
sizeClu=pClu->Size();
continue;
}
chId=pClu->Ch();
if(pClu->Q()>2*fParam->QCut()) continue;
Double_t thetaCer,phiCer;
if(FindPhotCkov(pClu->X(),pClu->Y(),thetaCer,phiCer)){
fPhotCkov[fPhotCnt]=thetaCer;
fPhotPhi [fPhotCnt]=phiCer;
fPhotClusIndex[fPhotCnt]=iClu;
fPhotCnt++;
}
}
pTrk->SetHMPIDmip(mipX,mipY,mipQ,fPhotCnt);
pTrk->SetHMPIDcluIdx(chId,index+1000*sizeClu);
if(fPhotCnt<nMinPhotAcc) {
pTrk->SetHMPIDsignal(kNoPhotAccept);
return;
}
fMipPos.Set(mipX,mipY);
if(fPhotCnt>fParam->MultCut()) fIsWEIGHT = kTRUE;
else fIsWEIGHT = kFALSE;
Int_t iNrec=FlagPhot(HoughResponse(),pCluLst,pTrk);
pTrk->SetHMPIDmip(mipX,mipY,mipQ,iNrec);
if(iNrec<nMinPhotAcc){
pTrk->SetHMPIDsignal(kNoPhotAccept);
return;
}
Int_t occupancy = (Int_t)(1000*(nPads/(6.*80.*48.)));
Double_t thetaC = FindRingCkov(pCluLst->GetEntries());
pTrk->SetHMPIDsignal(thetaC+occupancy);
pTrk->SetHMPIDchi2(fCkovSigma2);
DeleteVars();
}
Bool_t AliHMPIDRecon::FindPhotCkov(Double_t cluX,Double_t cluY,Double_t &thetaCer,Double_t &phiCer)
{
TVector3 dirCkov;
Double_t zRad= -0.5*fParam->RadThick()-0.5*fParam->WinThick();
TVector3 rad(fTrkPos.X(),fTrkPos.Y(),zRad);
TVector3 pc(cluX,cluY,0.5*fParam->WinThick()+fParam->GapIdx());
Double_t cluR = TMath::Sqrt((cluX-fTrkPos.X())*(cluX-fTrkPos.X())+
(cluY-fTrkPos.Y())*(cluY-fTrkPos.Y()));
Double_t phi=(pc-rad).Phi();
Double_t ckov1=0;
Double_t ckov2=0.75+fTrkDir.Theta();
const Double_t kTol=0.01;
Int_t iIterCnt = 0;
while(1){
if(iIterCnt>=50) return kFALSE;
Double_t ckov=0.5*(ckov1+ckov2);
dirCkov.SetMagThetaPhi(1,ckov,phi);
TVector2 posC=TraceForward(dirCkov);
Double_t dist=cluR-(posC-fTrkPos).Mod();
if(posC.X()==-999) dist = - 999;
iIterCnt++;
if (dist> kTol) ckov1=ckov;
else if(dist<-kTol) ckov2=ckov;
else{
dirCkov.SetMagThetaPhi(1,ckov,phi);
Lors2Trs(dirCkov,thetaCer,phiCer);
return kTRUE;
}
}
}
TVector2 AliHMPIDRecon::TraceForward(TVector3 dirCkov)const
{
TVector2 pos(-999,-999);
Double_t thetaCer = dirCkov.Theta();
if(thetaCer > TMath::ASin(1./fParam->GetRefIdx())) return pos;
Double_t zRad= -0.5*fParam->RadThick()-0.5*fParam->WinThick();
TVector3 posCkov(fTrkPos.X(),fTrkPos.Y(),zRad);
Propagate(dirCkov,posCkov, -0.5*fParam->WinThick());
Refract (dirCkov, fParam->GetRefIdx(),fParam->WinIdx());
Propagate(dirCkov,posCkov, 0.5*fParam->WinThick());
Refract (dirCkov, fParam->WinIdx(),fParam->GapIdx());
Propagate(dirCkov,posCkov,0.5*fParam->WinThick()+fParam->GapThick());
pos.Set(posCkov.X(),posCkov.Y());
return pos;
}
void AliHMPIDRecon::Lors2Trs(TVector3 dirCkov,Double_t &thetaCer,Double_t &phiCer)const
{
TRotation mtheta; mtheta.RotateY(-fTrkDir.Theta());
TRotation mphi; mphi.RotateZ(-fTrkDir.Phi());
TRotation mrot=mtheta*mphi;
TVector3 dirCkovTRS;
dirCkovTRS=mrot*dirCkov;
phiCer = dirCkovTRS.Phi();
thetaCer= dirCkovTRS.Theta();
}
void AliHMPIDRecon::Trs2Lors(TVector3 dirCkov,Double_t &thetaCer,Double_t &phiCer)const
{
TRotation mtheta; mtheta.RotateY(fTrkDir.Theta());
TRotation mphi; mphi.RotateZ(fTrkDir.Phi());
TRotation mrot=mphi*mtheta;
TVector3 dirCkovLORS;
dirCkovLORS=mrot*dirCkov;
phiCer = dirCkovLORS.Phi();
thetaCer= dirCkovLORS.Theta();
}
void AliHMPIDRecon::FindRingGeom(Double_t ckovAng,Int_t level)
{
Int_t kN=50*level;
Int_t nPoints = 0;
Double_t area=0;
Bool_t first=kFALSE;
TVector2 pos1;
for(Int_t i=0;i<kN;i++){
if(!first) {
pos1=TracePhot(ckovAng,Double_t(TMath::TwoPi()*(i+1)/kN));
if(pos1.X()==-999) continue;
if(!fParam->IsInside(pos1.X(),pos1.Y(),0)) {
pos1 = IntWithEdge(fMipPos,pos1);
} else {
if(!AliHMPIDParam::IsInDead(pos1.X(),pos1.Y())) nPoints++;
}
first=kTRUE;
continue;
}
TVector2 pos2=TracePhot(ckovAng,Double_t(TMath::TwoPi()*(i+1)/kN));
if(pos2.X()==-999) continue;
if(!fParam->IsInside(pos2.X(),pos2.Y(),0)) {
pos2 = IntWithEdge(fMipPos,pos2);
} else {
if(!AliHMPIDParam::IsInDead(pos2.X(),pos2.Y())) nPoints++;
}
area+=TMath::Abs((pos1-fMipPos).X()*(pos2-fMipPos).Y()-(pos1-fMipPos).Y()*(pos2-fMipPos).X());
pos1 = pos2;
}
fRingAcc = (Double_t)nPoints/(Double_t)kN;
area*=0.5;
fRingArea = area;
}
TVector2 AliHMPIDRecon::IntWithEdge(TVector2 p1,TVector2 p2)const
{
Double_t xmin = (p1.X()<p2.X())? p1.X():p2.X();
Double_t xmax = (p1.X()<p2.X())? p2.X():p1.X();
Double_t ymin = (p1.Y()<p2.Y())? p1.Y():p2.Y();
Double_t ymax = (p1.Y()<p2.Y())? p2.Y():p1.Y();
Double_t m = TMath::Tan((p2-p1).Phi());
TVector2 pint;
pint.Set((Double_t)(p1.X() + (0-p1.Y())/m),0.);
if(pint.X()>=0 && pint.X()<=fParam->SizeAllX() &&
pint.X()>=xmin && pint.X()<=xmax &&
pint.Y()>=ymin && pint.Y()<=ymax) return pint;
pint.Set((Double_t)(p1.X() + (fParam->SizeAllY()-p1.Y())/m),(Double_t)(fParam->SizeAllY()));
if(pint.X()>=0 && pint.X()<=fParam->SizeAllX() &&
pint.X()>=xmin && pint.X()<=xmax &&
pint.Y()>=ymin && pint.Y()<=ymax) return pint;
pint.Set(0.,(Double_t)(p1.Y() + m*(0-p1.X())));
if(pint.Y()>=0 && pint.Y()<=fParam->SizeAllY() &&
pint.Y()>=ymin && pint.Y()<=ymax &&
pint.X()>=xmin && pint.X()<=xmax) return pint;
pint.Set((Double_t)(fParam->SizeAllX()),(Double_t)(p1.Y() + m*(fParam->SizeAllX()-p1.X())));
if(pint.Y()>=0 && pint.Y()<=fParam->SizeAllY() &&
pint.Y()>=ymin && pint.Y()<=ymax &&
pint.X()>=xmin && pint.X()<=xmax) return pint;
return p1;
}
Double_t AliHMPIDRecon::FindRingCkov(Int_t)
{
Double_t wei = 0.;
Double_t weightThetaCerenkov = 0.;
Double_t ckovMin=9999.,ckovMax=0.;
Double_t sigma2 = 0;
for(Int_t i=0;i<fPhotCnt;i++){
if(fPhotFlag[i] == 2){
if(fPhotCkov[i]<ckovMin) ckovMin=fPhotCkov[i];
if(fPhotCkov[i]>ckovMax) ckovMax=fPhotCkov[i];
weightThetaCerenkov += fPhotCkov[i]*fPhotWei[i];
wei += fPhotWei[i];
sigma2 += 1./fParam->Sigma2(fTrkDir.Theta(),fTrkDir.Phi(),fPhotCkov[i],fPhotPhi[i]);
}
}
if(sigma2>0) fCkovSigma2=1./sigma2;
else fCkovSigma2=1e10;
if(wei != 0.) weightThetaCerenkov /= wei; else weightThetaCerenkov = 0.;
return weightThetaCerenkov;
}
Int_t AliHMPIDRecon::FlagPhot(Double_t ckov,TClonesArray *pCluLst, AliESDtrack *pTrk)
{
Int_t steps = (Int_t)((ckov )/ fDTheta);
Double_t tmin = (Double_t)(steps - 1)*fDTheta;
Double_t tmax = (Double_t)(steps)*fDTheta;
Double_t tavg = 0.5*(tmin+tmax);
tmin = tavg - 0.5*fWindowWidth; tmax = tavg + 0.5*fWindowWidth;
Int_t iInsideCnt = 0;
for(Int_t i=0;i<fPhotCnt;i++){
fPhotFlag[i] = 0;
if(fPhotCkov[i] >= tmin && fPhotCkov[i] <= tmax) {
fPhotFlag[i]=2;
AddObjectToFriends(pCluLst,i,pTrk);
iInsideCnt++;
}
}
return iInsideCnt;
}
void AliHMPIDRecon::AddObjectToFriends(TClonesArray *pCluLst, Int_t photonIndex, AliESDtrack *pTrk)
{
AliHMPIDCluster *pClu=(AliHMPIDCluster*)pCluLst->UncheckedAt(fPhotClusIndex[photonIndex]);
AliHMPIDCluster *pClus = new AliHMPIDCluster(*pClu);
pClus->SetChi2(fPhotCkov[photonIndex]);
pTrk->AddCalibObject(pClus);
}
TVector2 AliHMPIDRecon::TracePhot(Double_t ckovThe,Double_t ckovPhi)const
{
Double_t theta,phi;
TVector3 dirTRS,dirLORS;
dirTRS.SetMagThetaPhi(1,ckovThe,ckovPhi);
Trs2Lors(dirTRS,theta,phi);
dirLORS.SetMagThetaPhi(1,theta,phi);
return TraceForward(dirLORS);
}
void AliHMPIDRecon::Propagate(const TVector3 dir,TVector3 &pos,Double_t z)const
{
static TVector3 nrm(0,0,1);
TVector3 pnt(0,0,z);
TVector3 diff=pnt-pos;
Double_t sint=(nrm*diff)/(nrm*dir);
pos+=sint*dir;
}
void AliHMPIDRecon::Refract(TVector3 &dir,Double_t n1,Double_t n2)const
{
Double_t sinref=(n1/n2)*TMath::Sin(dir.Theta());
if(TMath::Abs(sinref)>1.) dir.SetXYZ(-999,-999,-999);
else dir.SetTheta(TMath::ASin(sinref));
}
Double_t AliHMPIDRecon::HoughResponse()
{
Double_t kThetaMax=0.75;
Int_t nChannels = (Int_t)(kThetaMax/fDTheta+0.5);
TH1D *phots = new TH1D("Rphot" ,"phots" ,nChannels,0,kThetaMax);
TH1D *photsw = new TH1D("RphotWeighted" ,"photsw" ,nChannels,0,kThetaMax);
TH1D *resultw = new TH1D("resultw","resultw" ,nChannels,0,kThetaMax);
Int_t nBin = (Int_t)(kThetaMax/fDTheta);
Int_t nCorrBand = (Int_t)(fWindowWidth/(2*fDTheta));
for (Int_t i=0; i< fPhotCnt; i++){
Double_t angle = fPhotCkov[i]; if(angle<0||angle>kThetaMax) continue;
phots->Fill(angle);
Int_t bin = (Int_t)(0.5+angle/(fDTheta));
Double_t weight=1.;
if(fIsWEIGHT){
Double_t lowerlimit = ((Double_t)bin)*fDTheta - 0.5*fDTheta; Double_t upperlimit = ((Double_t)bin)*fDTheta + 0.5*fDTheta;
FindRingGeom(lowerlimit);
Double_t areaLow = GetRingArea();
FindRingGeom(upperlimit);
Double_t areaHigh = GetRingArea();
Double_t diffArea = areaHigh - areaLow;
if(diffArea>0) weight = 1./diffArea;
}
photsw->Fill(angle,weight);
fPhotWei[i]=weight;
}
for (Int_t i=1; i<=nBin;i++){
Int_t bin1= i-nCorrBand;
Int_t bin2= i+nCorrBand;
if(bin1<1) bin1=1;
if(bin2>nBin)bin2=nBin;
Double_t sumPhots=phots->Integral(bin1,bin2);
if(sumPhots<3) continue;
Double_t sumPhotsw=photsw->Integral(bin1,bin2);
if((Double_t)((i+0.5)*fDTheta)>0.7) continue;
resultw->Fill((Double_t)((i+0.5)*fDTheta),sumPhotsw);
}
Double_t *pVec = resultw->GetArray();
Int_t locMax = TMath::LocMax(nBin,pVec);
delete phots;delete photsw;delete resultw;
return (Double_t)(locMax*fDTheta+0.5*fDTheta);
}
Double_t AliHMPIDRecon::FindRingExt(Double_t ckov,Int_t ch,Double_t xPc,Double_t yPc,Double_t thRa,Double_t phRa)
{
Double_t xRa = xPc - (fParam->RadThick()+fParam->WinThick()+fParam->GapThick())*TMath::Cos(phRa)*TMath::Tan(thRa);
Double_t yRa = yPc - (fParam->RadThick()+fParam->WinThick()+fParam->GapThick())*TMath::Sin(phRa)*TMath::Tan(thRa);
Int_t nStep = 500;
Int_t nPhi = 0;
Int_t ipc,ipadx,ipady;
if(ckov>0){
SetTrack(xRa,yRa,thRa,phRa);
for(Int_t j=0;j<nStep;j++){
TVector2 pos; pos=TracePhot(ckov,j*TMath::TwoPi()/(Double_t)(nStep-1));
if(fParam->IsInDead(pos.X(),pos.Y())) continue;
fParam->Lors2Pad(pos.X(),pos.Y(),ipc,ipadx,ipady);
ipadx+=(ipc%2)*fParam->kPadPcX;
ipady+=(ipc/2)*fParam->kPadPcY;
if(ipadx<0 || ipady>160 || ipady<0 || ipady>144 || ch<0 || ch>6) continue;
if(fParam->IsDeadPad(ipadx,ipady,ch)) continue;
nPhi++;
}
return ((Double_t)nPhi/(Double_t)nStep);
}
return -1;
}