// ------------ method called once each job just before starting event loop ------------
void
LYZ::beginJob()
{
double Vmax[nbin], eps[nbin];
for(int ibin=0; ibin<nbin ; ibin++) {
Vmax[ibin]=0.065*(trkbin[ibin]+30);
eps[ibin]=0.00025*(trkbin[ibin]+30);
}
Nevent.ResizeTo(nbin);
totmultall.ResizeTo(nbin), tottrk.ResizeTo(nbin);;
Nevent.Zero();
totmultall.Zero(), tottrk.Zero();
for(int ibin=0; ibin<nbin; ibin++) {
for(int iptbin=0; iptbin<nptV; iptbin++) {
Qx2[ibin][iptbin]=0;
Qy2[ibin][iptbin]=0;
}
}
for(int itheta=0; itheta<ntheta; itheta++) {
theta[itheta]=itheta*TMath::Pi()/ntheta/nn;
for(int ibin=0; ibin<nbin; ibin++) {
for(int iptbin=0; iptbin<nptV; iptbin++) {
GRe[ibin][iptbin][itheta].ResizeTo(nstepr);
GRe[ibin][iptbin][itheta].Zero();
GIm[ibin][iptbin][itheta].ResizeTo(nstepr);
GIm[ibin][iptbin][itheta].Zero();
}
}
}
for(int ibin=0; ibin<nbin; ibin++) {
for(int ir=0; ir<nstepr; ir++)
if(isSimple==0) r[ibin][ir]=j01/(Vmax[ibin]-eps[ibin]*ir);
else r[ibin][ir]=0.00025*20*(ir+1);
Qx1[ibin].ResizeTo(nptV);
Qy1[ibin].ResizeTo(nptV);
Q2[ibin].ResizeTo(nptV);
Qx1[ibin].Zero();
Qy1[ibin].Zero();
Q2[ibin].Zero();
totmult[ibin].ResizeTo(nptv);
totmult[ibin].Zero();
totpt[ibin].ResizeTo(nptv);
totpt[ibin].Zero();
dDRe[ibin].ResizeTo(ntheta);
dDRe[ibin].Zero();
dDIm[ibin].ResizeTo(ntheta);
dDIm[ibin].Zero();
for(int itheta=0; itheta<ntheta; itheta++) {
dNRe[ibin][itheta].ResizeTo(nptv);
dNRe[ibin][itheta].Zero();
dNIm[ibin][itheta].ResizeTo(nptv);
dNIm[ibin][itheta].Zero();
}
}
}
void merge(){
TVectorD Nevent; Nevent.ResizeTo(nbin); Nevent.Zero();
// TVectorD totmultall; totmultall.ResizeTo(nbin); totmultall.Zero();
// TVectorD avgmultall; avgmultall.ResizeTo(nbin); avgmultall.Zero();
TVectorD tottrk; tottrk.ResizeTo(nbin); tottrk.Zero();
// TVectorD totptall; totptall.ResizeTo(nbin); totptall.Zero();
// TVectorD totetaall; totetaall.ResizeTo(nbin); totetaall.Zero();
TVectorD avgtrk; avgtrk.ResizeTo(nbin); avgtrk.Zero();
// TVectorD avgmult; avgmult.ResizeTo(nbin);
TH2F* s[nbin];
TH2F* b[nbin];
for(int ibin=0;ibin<nbin;ibin++){
s[ibin] = new TH2F(Form("s_%d",ibin),Form("signal",ibin),detastep,detamin,detamax,dphistep,dphimin,dphimax);
s[ibin]->Sumw2();
b[ibin] = new TH2F(Form("b_%d",ibin), "background",detastep,detamin,detamax,dphistep,dphimin,dphimax);
b[ibin]->Sumw2();
}
TFile *fout = new TFile(Form("Anav3_merged.root"),"Recreate");
TFile *f[nFileAll];
for(int ifile=; ifile<63; ifile++){
f[ifile] = TFile::Open(Form("%s/Anav3_%d.root",outdir.Data(),ifile));
if(!f[ifile]) continue;
TVectorD* Nevent_t = (TVectorD*)f[ifile]->Get(Form("Nevent"));
// TVectorD* totmultall_t = (TVectorD*)f[ifile]->Get(Form("totmultall"));
TVectorD* tottrk_t = (TVectorD*)f[ifile]->Get(Form("tottrk"));
// TVectorD* totptall_t = (TVectorD*)f[ifile]->Get(Form("totptall"));
// TVectorD* totetaall_t = (TVectorD*)f[ifile]->Get(Form("totetaall"));
for(int ibin=0;ibin<nbin;ibin++){
// totptall[ibin] += (*totptall_t)[ibin];
// totetaall[ibin] += (*totetaall_t)[ibin];
Nevent[ibin] += (*Nevent_t)[ibin];
// totmultall[ibin] += (*totmultall_t)[ibin];
tottrk[ibin] += (*tottrk_t)[ibin];
TH1D* s_t = (TH1D*)f[ifile]->Get(Form("signal_%d",ibin));
TH1D* b_t = (TH1D*)f[ifile]->Get(Form("background_%d",ibin));
s[ibin]->Add(s_t);
b[ibin]->Add(b_t);
}
f[ifile]->Close();
}
for(int ibin=0;ibin<nbin;ibin++){
avgtrk[ibin] = tottrk[ibin]/Nevent[ibin];
TH1D* hr = (TH1D*)s[ibin]->Clone("hr");
hr->Divide(b[ibin]);
hr->Scale(b[ibin]->GetBinContent(b[ibin]->FindBin(0,0)));
fout->cd();
Nevent.Write("Nevent");
avgtrk.Write("avgtrk");
TDirectory *dir0 = (TDirectory*)fout->mkdir(Form("D_%d",ibin));
dir0->cd();
s[ibin]->Write("s");
b[ibin]->Write("b");
hr->Write();
}
}
// ------------ method called once each job just before starting event loop ------------
void
LYZ::beginJob(int ispt_)
{
double Vmax[nbin], eps[nbin];
for(int ibin=0; ibin<nbin ;ibin++){
Vmax[ibin]=0.065*(trkbin[ibin]+30);
eps[ibin]=0.00025*(trkbin[ibin]+30);
}
ispt = ispt_;
if(ispt_){ nvv = nptv; binv = ptbinv;}
else{ nvv = netav; binv = etabinv;}
Nevent.ResizeTo(nbin); totmultall.ResizeTo(nbin), tottrk.ResizeTo(nbin), totptall.ResizeTo(nbin), totetaall.ResizeTo(nbin); totwall.ResizeTo(nbin);
Qx1.ResizeTo(nbin); Qy1.ResizeTo(nbin); Q2.ResizeTo(nbin);
Nevent.Zero(); totmultall.Zero(); totwall.Zero(); tottrk.Zero(); totptall.Zero(); totetaall.Zero();
Qx1.Zero(); Qy1.Zero(); Q2.Zero();
for(int ibin=0; ibin<nbin; ibin++){
Qx2[ibin]=0; Qy2[ibin]=0;
}
for(int itheta=0;itheta<ntheta;itheta++){
theta[itheta]=itheta*TMath::Pi()/ntheta/nn;
for(int ibin=0; ibin<nbin; ibin++){
GRe[ibin][itheta].ResizeTo(nstepr); GRe[ibin][itheta].Zero();
GIm[ibin][itheta].ResizeTo(nstepr); GIm[ibin][itheta].Zero();
}
}
for(int ibin=0; ibin<nbin; ibin++){
for(int ir=0; ir<nstepr; ir++)
if(isSimple==0) r[ibin][ir]=j01/(Vmax[ibin]-eps[ibin]*ir);
else r[ibin][ir]=0.00025*20*(ir+1);
totmultv[ibin].ResizeTo(nvv); totmultv[ibin].Zero();
totptv[ibin].ResizeTo(nvv); totptv[ibin].Zero();
totetav[ibin].ResizeTo(nvv); totetav[ibin].Zero();
dDRe[ibin].ResizeTo(ntheta); dDRe[ibin].Zero();
dDIm[ibin].ResizeTo(ntheta); dDIm[ibin].Zero();
for(int itheta=0;itheta<ntheta;itheta++){
dNRe[ibin][itheta].ResizeTo(nvv); dNRe[ibin][itheta].Zero();
dNIm[ibin][itheta].ResizeTo(nvv); dNIm[ibin][itheta].Zero();
}
}
}
// ------------ method called once each job just before starting event loop ------------
void
qfit::beginJob(int ispt_)
{
ispt = ispt_;
if(ispt_){ nvv = nptv; binv = ptbinv;}
else{ nvv = netav; binv = etabinv;}
Nevent.ResizeTo(nbin); totmultall.ResizeTo(nbin), tottrk.ResizeTo(nbin), totptall.ResizeTo(nbin), totptall2.ResizeTo(nbin); totetaall.ResizeTo(nbin);
Nevent.Zero(); totmultall.Zero(), tottrk.Zero(); totptall.Zero(); totptall2.Zero(); totetaall.Zero();
q22.ResizeTo(nbin);q24.ResizeTo(nbin);
q22.Zero();q24.Zero();
for(int itheta=0;itheta<ntheta;itheta++){
theta[itheta]=itheta*TMath::Pi()/ntheta/nn;
}
for(int ibin=0; ibin<nbin; ibin++){
totmultv[ibin].ResizeTo(nvv); totmultv[ibin].Zero();
totptv[ibin].ResizeTo(nvv); totptv[ibin].Zero();
totetav[ibin].ResizeTo(nvv); totetav[ibin].Zero();
for(int itheta=0;itheta<ntheta;itheta++){
hq[ibin][itheta] = new TH1D(Form("hq_%d_%d",ibin,itheta),Form("hq_%d_%d",ibin,itheta),1000,0,10);
hq[ibin][itheta]->Sumw2();
}
hqx[ibin] = new TH1D(Form("hqx_%d",ibin),Form("hqx_%d",ibin),1000,0,10);
hqy[ibin] = new TH1D(Form("hqy_%d",ibin),Form("hqy_%d",ibin),1000,0,10);
hq2[ibin] = new TH1D(Form("hq2_%d",ibin),Form("hq2_%d",ibin),1000,0,10);
hvq2[ibin] = new TH1D(Form("hvq2_%d",ibin),Form("hvq2_%d",ibin),1000,0,1);
hq2nonf[ibin] = new TH1D(Form("hq2nonf_%d",ibin),Form("hq2nonf_%d",ibin),1000,0,10);
hqx[ibin]->Sumw2();
hqy[ibin]->Sumw2();
hq2[ibin]->Sumw2();
hvq2[ibin]->Sumw2();
hq2nonf[ibin]->Sumw2();
}
}
void getResV(){
//------------------------Deal with the number and get the output stored in txt files and root files
string SumorProd = getenv("SUMORPROD");
double Vmax[nbin], eps[nbin];
for(int ibin=0; ibin<nbin ;ibin++){
//Vmax[ibin]=0.065*(trkbin[ibin]+30);
Vmax[ibin]=0.065*avgmultIn;
//eps[ibin]=0.00025*(trkbin[ibin]+30);
eps[ibin]=0.00025*avgmultIn;
}
ofstream fstrV;
double theta[ntheta];
TVectorD avgmult[nbin], avgpt[nbin], totmult[nbin], totpt[nbin];
TVectorD Qx1[nbin], Qy1[nbin], Q2[nbin];
TVectorD Gmod2[nbin][nptV][ntheta];
TVectorD sigma2[nbin][nptV],deltaV[nbin][nptV];
TVectorD sigma2_[nbin],chi_[nbin];
TVectorD deltaVmean[nbin], Vmean[nbin];
TVectorD r[nbin];
TVectorD r0[nbin][nptV], V[nbin][nptV], chi[nbin][nptV];
TVectorD GRe[nbin][nptV][ntheta]; TVectorD* GRe_t[nbin][nptV][ntheta];
TVectorD GIm[nbin][nptV][ntheta]; TVectorD* GIm_t[nbin][nptV][ntheta];
TComplex G[nbin][nptV][ntheta][nstepr];
if(SumorProd=="Sum") fstrV.open("V_Sum.txt");
else fstrV.open("V_Prod.txt");
TFile *f[nFileAll];
for(int ibin=0; ibin<nbin; ibin++){
r[ibin].ResizeTo(nstepr);
for(int ir=0; ir<nstepr; ir++){
if(isSimple==0) r[ibin][ir]=j01/(Vmax[ibin]-eps[ibin]*ir);
else r[ibin][ir]=0.00025*20*(ir+1);
}
}
TVectorD Nevent; Nevent.ResizeTo(nbin); Nevent.Zero();
TVectorD totmultall; totmultall.ResizeTo(nbin); totmultall.Zero();
TVectorD avgmultall; avgmultall.ResizeTo(nbin); avgmultall.Zero();
TVectorD tottrk; tottrk.ResizeTo(nbin); tottrk.Zero();
TVectorD avgtrk; avgtrk.ResizeTo(nbin); avgtrk.Zero();
for(int itheta=0;itheta<ntheta;itheta++)
theta[itheta]=itheta*TMath::Pi()/ntheta/nn;
for(int ibin=0;ibin<nbin;ibin++){
for(int iptbin=0;iptbin<nptV;iptbin++){
r0[ibin][iptbin].ResizeTo(ntheta);
sigma2[ibin][iptbin].ResizeTo(ntheta);
V[ibin][iptbin].ResizeTo(ntheta);
deltaV[ibin][iptbin].ResizeTo(ntheta);
chi[ibin][iptbin].ResizeTo(ntheta);
for(int itheta=0;itheta<ntheta;itheta++){
Gmod2[ibin][iptbin][itheta].ResizeTo(nstepr);
GRe[ibin][iptbin][itheta].ResizeTo(nstepr);
GRe[ibin][iptbin][itheta].Zero();
GIm[ibin][iptbin][itheta].ResizeTo(nstepr);
GIm[ibin][iptbin][itheta].Zero();
}
}
avgmult[ibin].ResizeTo(nptV); deltaVmean[ibin].ResizeTo(nptV); Vmean[ibin].ResizeTo(nptV);
totpt[ibin].ResizeTo(nptV); totpt[ibin].Zero();
totmult[ibin].ResizeTo(nptV); totmult[ibin].Zero();
avgpt[ibin].ResizeTo(nptV);
Qx1[ibin].ResizeTo(nptV); Qx1[ibin].Zero();
Qy1[ibin].ResizeTo(nptV); Qy1[ibin].Zero();
Q2[ibin].ResizeTo(nptV); Q2[ibin].Zero();
sigma2_[ibin].ResizeTo(nptV);sigma2_[ibin].Zero();
chi_[ibin].ResizeTo(nptV);chi_[ibin].Zero();
}
for(int ifile=0; ifile<nFileAll; ifile++){
if(SumorProd=="Sum") f[ifile] = TFile::Open(Form("/lio/lfs/cms/store/user/qixu/flow/pbsjoboutput/PFcandpt03to6/M300260/AnaV_Sum_%d.root",ifile));
else f[ifile] = TFile::Open(Form("/lio/lfs/cms/store/user/qixu/flow/pbsjoboutput/PFcandpt03to6/M300260/AnaV_Prod_%d.root",ifile));
TVectorD* Nevent_t = (TVectorD*)f[ifile]->Get(Form("Nevent"));
TVectorD* totmultall_t = (TVectorD*)f[ifile]->Get(Form("totmultall"));
TVectorD* tottrk_t = (TVectorD*)f[ifile]->Get(Form("tottrk"));
for(int ibin=0;ibin<nbin;ibin++){
for(int iptbin=0;iptbin<nptV;iptbin++){
for(int itheta=0;itheta<ntheta;itheta++){
GRe_t[ibin][iptbin][itheta] = (TVectorD*)f[ifile]->Get(Form("GRe_%d_%d_%d",ibin,iptbin,itheta));
GIm_t[ibin][iptbin][itheta] = (TVectorD*)f[ifile]->Get(Form("GIm_%d_%d_%d",ibin,iptbin,itheta));
for(ir=0; ir<nstepr; ir++){
GRe[ibin][iptbin][itheta][ir] += (*GRe_t[ibin][iptbin][itheta])[ir];
GIm[ibin][iptbin][itheta][ir] += (*GIm_t[ibin][iptbin][itheta])[ir];
}
}
}
TVectorD* Qx1_t = (TVectorD*)f[ifile]->Get(Form("Qx1_%d",ibin));
TVectorD* Qy1_t = (TVectorD*)f[ifile]->Get(Form("Qy1_%d",ibin));
TVectorD* Q2_t = (TVectorD*)f[ifile]->Get(Form("Q2_%d",ibin));
TVectorD* totmult_t = (TVectorD*)f[ifile]->Get(Form("totmult_%d",ibin));
TVectorD* totpt_t = (TVectorD*)f[ifile]->Get(Form("totpt_%d",ibin));
for(int iptbin=0;iptbin<nptV;iptbin++){
Qx1[ibin][iptbin] += (*Qx1_t)[iptbin];
Qy1[ibin][iptbin] += (*Qy1_t)[iptbin];
Q2[ibin][iptbin] += (*Q2_t)[iptbin];
totmult[ibin][iptbin] += (*totmult_t)[iptbin];
totpt[ibin][iptbin] += (*totpt_t)[iptbin];
}
Nevent[ibin] += (*Nevent_t)[ibin];
totmultall[ibin] += (*totmultall_t)[ibin];
tottrk[ibin] += (*tottrk_t)[ibin];
}
f[ifile]->Close();
}
fstrV<<setprecision(4)<<fixed;
fstrV<<"ibin"<<"\t"<<"iptbin"<<"\t"<<"itheta"<<"\t"<<"r0"<<"\t"<<"V"<<"\t"<<"sigma2"<<"\t"<<"chi"<<"\t"<<"Vn Errors"<<endl;
for(int ibin=0;ibin<nbin;ibin++){
avgmultall[ibin]=totmultall[ibin]/Nevent[ibin];
avgtrk[ibin]=tottrk[ibin]/Nevent[ibin];
for(int iptbin=0;iptbin<nptV;iptbin++){
for(int itheta=0;itheta<ntheta;itheta++){
for(ir=0; ir<nstepr; ir++){
G[ibin][iptbin][itheta][ir]=TComplex(GRe[ibin][iptbin][itheta][ir],GIm[ibin][iptbin][itheta][ir]);
G[ibin][iptbin][itheta][ir]/=Nevent[ibin];
Gmod2[ibin][iptbin][itheta][ir]=TMath::Power(TComplex::Abs(G[ibin][iptbin][itheta][ir]),2);
}
for(ir=0; ir<nstepr-1; ir++)
if(ir!=0 && Gmod2[ibin][iptbin][itheta][ir]<=Gmod2[ibin][iptbin][itheta][ir-1] && Gmod2[ibin][iptbin][itheta][ir]<=Gmod2[ibin][iptbin][itheta][ir+1]) break;
if(ir!=0 && ir<nstepr-1) r0[ibin][iptbin][itheta]=r[ibin][ir];
else if(ir==0) {cout<<"ibin="<<ibin<<"\t"<<"iptbin="<<iptbin<<"\t"<<"itheta="<<itheta<<"\tminimum lies on ir = 0, please select proper range!"<<endl; continue;}
else {cout<<"ibin="<<ibin<<"\t"<<"iptbin="<<iptbin<<"\t"<<"itheta="<<itheta<<"\tminimum lies on ir = maximum "<<nstepr-1<<", please select proper range!"<<endl; continue;}
avgmult[ibin][iptbin]=1.0*totmult[ibin][iptbin]/Nevent[ibin];
avgpt[ibin][iptbin]=1.0*totpt[ibin][iptbin]/totmult[ibin][iptbin];
if(isSimple==0) V[ibin][iptbin][itheta]=Vmax[ibin]-ir*eps[ibin]+eps[ibin]*(Gmod2[ibin][iptbin][itheta][ir+1]-Gmod2[ibin][iptbin][itheta][ir-1])/2./(Gmod2[ibin][iptbin][itheta][ir-1]-2*Gmod2[ibin][iptbin][itheta][ir]+Gmod2[ibin][iptbin][itheta][ir+1]);
else V[ibin][iptbin][itheta]=j01/r0[ibin][iptbin][itheta]; //simple method
V[ibin][iptbin][itheta]/=avgmult[ibin][iptbin];
sigma2[ibin][iptbin][itheta]=Q2[ibin][iptbin]/Nevent[ibin]-(Qx1[ibin][iptbin]/Nevent[ibin])*(Qx1[ibin][iptbin]/Nevent[ibin])-(Qy1[ibin][iptbin]/Nevent[ibin])*(Qy1[ibin][iptbin]/Nevent[ibin])-(V[ibin][iptbin][itheta]*avgmult[ibin][iptbin])*(V[ibin][iptbin][itheta]*avgmult[ibin][iptbin]);
sigma2_[ibin][iptbin]+=sigma2[ibin][iptbin][itheta];
Vmean[ibin][iptbin]+=V[ibin][iptbin][itheta];
chi[ibin][iptbin][itheta]=V[ibin][iptbin][itheta]*avgmult[ibin][iptbin]/TMath::Sqrt(sigma2[ibin][iptbin][itheta]);
deltaV[ibin][iptbin][itheta]=V[ibin][iptbin][itheta]/j01/TMath::BesselJ1(j01)*TMath::Sqrt((TMath::Exp(j01*j01/2./chi[ibin][iptbin][itheta]/chi[ibin][iptbin][itheta])+TMath::Exp(-j01*j01/2./chi[ibin][iptbin][itheta]/chi[ibin][iptbin][itheta])*TMath::BesselJ0(2*j01))/2./Nevent[ibin]);
fstrV<<ibin<<"\t"<<iptbin<<"\t"<<itheta<<"\t"<<r0[ibin][iptbin][itheta]<<"\t"<<V[ibin][iptbin][itheta]<<"\t"<<sigma2[ibin][iptbin][itheta]<<"\t"<<chi[ibin][iptbin][itheta]<<"\t"<<deltaV[ibin][iptbin][itheta]<<endl;
}
//deltaVmean[ibin][iptbin]+=TMath::Exp(j01*j01/2./chi[ibin][iptbin][itheta]/chi[ibin][iptbin][itheta]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Sin(nn*theta[itheta]/2.))+TMath::Exp(-j01*j01/2./chi[ibin][iptbin][itheta]/chi[ibin][iptbin][itheta]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Cos(nn*theta[itheta]/2.));
sigma2_[ibin][iptbin]/=ntheta;
Vmean[ibin][iptbin]/=ntheta;
sigma2_[ibin][iptbin]-=TMath::Power(Vmean[ibin][iptbin]*avgmult[ibin][iptbin],2);
chi_[ibin][iptbin]=Vmean[ibin][iptbin]*avgmult[ibin][iptbin]/TMath::Sqrt(sigma2_[ibin][iptbin]);
for(int itheta=0;itheta<ntheta;itheta++){
deltaVmean[ibin][iptbin]+=TMath::Exp(j01*j01/2./chi_[ibin][iptbin]/chi_[ibin][iptbin]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Sin(nn*theta[itheta]/2.))+TMath::Exp(-j01*j01/2./chi_[ibin][iptbin]/chi_[ibin][iptbin]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Cos(nn*theta[itheta]/2.));
}
fstrV<<endl;
deltaVmean[ibin][iptbin]=Vmean[ibin][iptbin]/j01/TMath::BesselJ1(j01)*TMath::Sqrt(deltaVmean[ibin][iptbin]/ntheta/2./Nevent[ibin]);
}
fstrV<<endl;
}
fstrV<<"ibin"<<"\t"<<"iptbin"<<"\t"<<"avgmult"<<"\t"<<"avgpt"<<"\t"<<"Vn mean"<<"\t"<<"Vn mean Error"<<endl;
for(int ibin=0;ibin<nbin;ibin++){
for(int iptbin=0;iptbin<nptV;iptbin++){
fstrV<<ibin<<"\t"<<iptbin<<"\t"<<avgmult[ibin][iptbin]<<"\t"<<avgpt[ibin][iptbin]<<"\t"<<Vmean[ibin][iptbin]<<"\t"<<deltaVmean[ibin][iptbin]<<endl;
}
fstrV<<endl;
}
fstrV<<"ibin"<<"\t"<<"Nevent"<<"\t"<<"totmultall"<<"\t"<<"avgmultall"<<"\t"<<"avgtrk"<<endl;
for(int ibin=0;ibin<nbin;ibin++){
fstrV<<ibin<<"\t"<<Nevent[ibin]<<"\t"<<totmultall[ibin]<<"\t"<<avgmultall[ibin]<<"\t"<<avgtrk[ibin]<<endl;
fstrV<<endl;
}
if(SumorProd=="Sum") TString outname = "mergedV_Sum.root";
else TString outname="mergedV_Prod.root";
TFile *outf = new TFile(outname,"Recreate");
Nevent.Write("Nevent");
totmultall.Write("totmultall");
avgmultall.Write("avgmultall");
tottrk.Write("tottrk");
avgtrk.Write("avgtrk");
for(int ibin=0;ibin<nbin;ibin++){
TDirectory *dir0 = outf->mkdir(Form("D_%d",ibin,itheta));dir0->cd();
r[ibin].Write("r");
Qx1[ibin].Write("Qx1");
Qy1[ibin].Write("Qy1");
Q2[ibin].Write("Q2");
totmult[ibin].Write("totmult");
totpt[ibin].Write("totpt");
avgmult[ibin].Write("avgmult");
avgpt[ibin].Write("avgpt");
Vmean[ibin].Write("Vmean");
deltaVmean[ibin].Write("deltaVmean");
for(int iptbin=0;iptbin<nptV;iptbin++){
TDirectory *dir1 = dir0->mkdir(Form("D_%d",iptbin));dir1->cd();
sigma2[ibin][iptbin].Write("sigma2"); chi[ibin][iptbin].Write("chi"); deltaV[ibin][iptbin].Write("deltaV");
r0[ibin][iptbin].Write("r0"); V[ibin][iptbin].Write("V");
for(int itheta=0;itheta<ntheta;itheta++){
TDirectory *dir2 = dir1->mkdir(Form("D_%d",itheta));dir2->cd();
GRe[ibin][iptbin][itheta].Write(Form("GRe"));
GIm[ibin][iptbin][itheta].Write(Form("GIm"));
Gmod2[ibin][iptbin][itheta].Write(Form("G2"));
}
}
}
outf->Close();
}
void getResV(){
//------------------------Deal with the number and get the output stored in txt files and root files
TF1 *ffit[nbin];
TF1 *ffit0[nbin][ntheta];
TFile *f[nFileAll];
double theta[ntheta];
TVectorD Nevent; Nevent.ResizeTo(nbin); Nevent.Zero();
TVectorD totmultall; totmultall.ResizeTo(nbin); totmultall.Zero();
TVectorD avgmultall; avgmultall.ResizeTo(nbin); avgmultall.Zero();
TVectorD tottrk; tottrk.ResizeTo(nbin); tottrk.Zero();
TVectorD totptall; totptall.ResizeTo(nbin); totptall.Zero();
TVectorD totetaall; totetaall.ResizeTo(nbin); totetaall.Zero();
TVectorD avgtrk; avgtrk.ResizeTo(nbin); avgtrk.Zero();
TVectorD avgpt; avgpt.ResizeTo(nbin);
TVectorD avgeta; avgeta.ResizeTo(nbin);
TH1D* hq[nbin][ntheta];
TH1D* hqx[nbin];
TH1D* hqy[nbin];
TH1D* hq2[nbin];
TH1D* hq2nonf[nbin];
string dir = getenv("DIR");
for(int itheta=0;itheta<ntheta;itheta++)
theta[itheta]=itheta*TMath::Pi()/ntheta/nn;
for(int ibin=0;ibin<nbin;ibin++){
for(int itheta=0;itheta<ntheta;itheta++){
hq[ibin][itheta] = new TH1D(Form("hq"),Form("hq"),1000,0,10);
}
hqx[ibin] = new TH1D(Form("hqx"),Form("hqx"),1000,0,10);
hqy[ibin] = new TH1D(Form("hqy"),Form("hqy"),1000,0,10);
hq2[ibin] = new TH1D(Form("hq2"),Form("hq2"),1000,0,10);
hq2nonf[ibin] = new TH1D(Form("hq2nonf"),Form("hq2nonf"),1000,0,10);
}
for(int ifile=0; ifile<nFileAll; ifile++){
f[ifile] = TFile::Open(Form("/scratch/xuq7/flow/pbsjoboutput/qfit/tracknormcpt03to6/%s/AnaV_Sum_%d.root",dir.c_str(),ifile));
TVectorD* Nevent_t = (TVectorD*)f[ifile]->Get(Form("Nevent"));
TVectorD* totmultall_t = (TVectorD*)f[ifile]->Get(Form("totmultall"));
TVectorD* tottrk_t = (TVectorD*)f[ifile]->Get(Form("tottrk"));
TVectorD* totptall_t = (TVectorD*)f[ifile]->Get(Form("totptall"));
TVectorD* totetaall_t = (TVectorD*)f[ifile]->Get(Form("totetaall"));
for(int ibin=0;ibin<nbin;ibin++){
TH1D* hqx_t = (TH1D*)f[ifile]->Get(Form("hqx_%d",ibin));
hqx[ibin]->Add(hqx_t);
TH1D* hqy_t = (TH1D*)f[ifile]->Get(Form("hqy_%d",ibin));
hqy[ibin]->Add(hqy_t);
TH1D* hq2_t = (TH1D*)f[ifile]->Get(Form("hq2_%d",ibin));
hq2[ibin]->Add(hq2_t);
TH1D* hq2nonf_t = (TH1D*)f[ifile]->Get(Form("hq2nonf_%d",ibin));
hq2nonf[ibin]->Add(hq2nonf_t);
for(int itheta=0;itheta<ntheta;itheta++){
TH1D* hq_t = (TH1D*)f[ifile]->Get(Form("hq_%d_%d",ibin,itheta));
hq[ibin][itheta]->Add(hq_t);
}
totptall[ibin] += (*totptall_t)[ibin];
totetaall[ibin] += (*totetaall_t)[ibin];
Nevent[ibin] += (*Nevent_t)[ibin];
totmultall[ibin] += (*totmultall_t)[ibin];
tottrk[ibin] += (*tottrk_t)[ibin];
}
f[ifile]->Close();
}
for(int ibin=0;ibin<nbin;ibin++){
avgmultall[ibin]=totmultall[ibin]/Nevent[ibin];
avgtrk[ibin]=tottrk[ibin]/Nevent[ibin];
for(int itheta=0;itheta<ntheta;itheta++){
avgpt[ibin]=1.0*totptall[ibin]/totmultall[ibin];
avgeta[ibin]=1.0*totetaall[ibin]/totmultall[ibin];
}
}
TString outname = "mergedV_Sum.root";
TFile *outf = new TFile(outname,"Recreate");
Nevent.Write("Nevent");
totmultall.Write("totmultall");
avgmultall.Write("avgmultall");
tottrk.Write("tottrk");
avgtrk.Write("avgtrk");
avgpt.Write("avgpt");
avgeta.Write("avgeta");
for(int ibin=0;ibin<nbin;ibin++){
TDirectory *dir0 = outf->mkdir(Form("D_%d",ibin));dir0->cd();
hqx[ibin]->Write(); hqy[ibin]->Write();hq2[ibin]->Write();hq2nonf[ibin]->Write();
for(int itheta=0;itheta<ntheta;itheta++){
TDirectory *dir1 = dir0->mkdir(Form("D_%d",itheta));dir1->cd();
hq[ibin][itheta]->Write();
}
}
outf->Close();
}
void getResV(){
//------------------------Deal with the number and get the output stored in txt files and root files
double Vmax[nbin], eps[nbin];
for(int ibin=0; ibin<nbin ;ibin++){
Vmax[ibin]=0.065*(trkbin[ibin]+30);
eps[ibin]=0.00025*(trkbin[ibin]+30);
}
ofstream fstrV;
double theta[ntheta];
TVectorD Gmod2[nbin][ntheta];
TVectorD sigma2[nbin],deltaV[nbin];
TVectorD r[nbin];
TVectorD r0[nbin], r01[nbin], V[nbin], chi[nbin];
TVectorD GRe[nbin][ntheta]; TVectorD* GRe_t[nbin][ntheta];
TVectorD GIm[nbin][ntheta]; TVectorD* GIm_t[nbin][ntheta];
TComplex G[nbin][ntheta][nstepr];
TF1 *ffit[nbin];
TF1 *ffit0[nbin][ntheta];
if(isSum) fstrV.open("V_Sum.txt");
else fstrV.open("V_Prod.txt");
TFile *f[nFileAll];
for(int ibin=0; ibin<nbin; ibin++){
r[ibin].ResizeTo(nstepr);
for(int ir=0; ir<nstepr; ir++){
if(isSimple==0) r[ibin][ir]=j01/(Vmax[ibin]-eps[ibin]*ir);
else r[ibin][ir]=0.00025*20*(ir+1);
}
}
TVectorD Nevent; Nevent.ResizeTo(nbin); Nevent.Zero();
TVectorD totmultall; totmultall.ResizeTo(nbin); totmultall.Zero();
TVectorD avgmultall; avgmultall.ResizeTo(nbin); avgmultall.Zero();
TVectorD tottrk; tottrk.ResizeTo(nbin); tottrk.Zero();
TVectorD totptall; totptall.ResizeTo(nbin); totptall.Zero();
TVectorD totetaall; totetaall.ResizeTo(nbin); totetaall.Zero();
TVectorD avgtrk; avgtrk.ResizeTo(nbin); avgtrk.Zero();
TVectorD deltaVmean; deltaVmean.ResizeTo(nbin);
TVectorD Vmean; Vmean.ResizeTo(nbin);
TVectorD avgpt; avgpt.ResizeTo(nbin);
TVectorD avgeta; avgeta.ResizeTo(nbin);
TVectorD Qx1; Qx1.ResizeTo(nbin); Qx1.Zero();
TVectorD Qy1; Qy1.ResizeTo(nbin); Qy1.Zero();
TVectorD Q2; Q2.ResizeTo(nbin); Q2.Zero();
TVectorD sigma2_; sigma2_.ResizeTo(nbin); sigma2_.Zero();
TVectorD chi_; chi_.ResizeTo(nbin); chi_.Zero();
TH1D* hq[nbin][ntheta];
TH1D* hqx[nbin];
TH1D* hqy[nbin];
TH1D* hq2[nbin];
TH1D* hq2nonf[nbin];
for(int itheta=0;itheta<ntheta;itheta++)
theta[itheta]=itheta*TMath::Pi()/ntheta/nn;
for(int ibin=0;ibin<nbin;ibin++){
r0[ibin].ResizeTo(ntheta);
r01[ibin].ResizeTo(ntheta);
sigma2[ibin].ResizeTo(ntheta);
V[ibin].ResizeTo(ntheta);
deltaV[ibin].ResizeTo(ntheta);
chi[ibin].ResizeTo(ntheta);
for(int itheta=0;itheta<ntheta;itheta++){
Gmod2[ibin][itheta].ResizeTo(nstepr);
GRe[ibin][itheta].ResizeTo(nstepr);
GRe[ibin][itheta].Zero();
GIm[ibin][itheta].ResizeTo(nstepr);
GIm[ibin][itheta].Zero();
hq[ibin][itheta] = new TH1D(Form("hq"),Form("hq"),1000,0,10);
}
hqx[ibin] = new TH1D(Form("hqx"),Form("hqx"),1000,0,10);
hqy[ibin] = new TH1D(Form("hqy"),Form("hqy"),1000,0,10);
hq2[ibin] = new TH1D(Form("hq2"),Form("hq2"),1000,0,10);
hq2nonf[ibin] = new TH1D(Form("hq2nonf"),Form("hq2nonf"),1000,0,10);
}
for(int ifile=0; ifile<nFileAll; ifile++){
if(isSum) f[ifile] = TFile::Open(Form("/scratch/xuq7/flow/pbsjoboutput/Bessfun/pPbDataV200m30/AnaV_Sum_%d.root",ifile));
else f[ifile] = TFile::Open(Form("/scratch/xuq7/flow/pbsjoboutput/Bessfun/pPbDataV200m30/AnaV_Prod_%d.root",ifile));
TVectorD* Nevent_t = (TVectorD*)f[ifile]->Get(Form("Nevent"));
TVectorD* totmultall_t = (TVectorD*)f[ifile]->Get(Form("totmultall"));
TVectorD* tottrk_t = (TVectorD*)f[ifile]->Get(Form("tottrk"));
TVectorD* totptall_t = (TVectorD*)f[ifile]->Get(Form("totptall"));
TVectorD* totetaall_t = (TVectorD*)f[ifile]->Get(Form("totetaall"));
TVectorD* Qx1_t = (TVectorD*)f[ifile]->Get(Form("Qx1"));
TVectorD* Qy1_t = (TVectorD*)f[ifile]->Get(Form("Qy1"));
TVectorD* Q2_t = (TVectorD*)f[ifile]->Get(Form("Q2"));
for(int ibin=0;ibin<nbin;ibin++){
TH1D* hqx_t = (TH1D*)f[ifile]->Get(Form("hqx_%d",ibin));
hqx[ibin]->Add(hqx_t);
TH1D* hqy_t = (TH1D*)f[ifile]->Get(Form("hqy_%d",ibin));
hqy[ibin]->Add(hqy_t);
TH1D* hq2_t = (TH1D*)f[ifile]->Get(Form("hq2_%d",ibin));
hq2[ibin]->Add(hq2_t);
TH1D* hq2nonf_t = (TH1D*)f[ifile]->Get(Form("hq2nonf_%d",ibin));
hq2nonf[ibin]->Add(hq2nonf_t);
for(int itheta=0;itheta<ntheta;itheta++){
GRe_t[ibin][itheta] = (TVectorD*)f[ifile]->Get(Form("GRe_%d_%d",ibin,itheta));
GIm_t[ibin][itheta] = (TVectorD*)f[ifile]->Get(Form("GIm_%d_%d",ibin,itheta));
TH1D* hq_t = (TH1D*)f[ifile]->Get(Form("hq_%d_%d",ibin,itheta));
hq[ibin][itheta]->Add(hq_t);
for(ir=0; ir<nstepr; ir++){
GRe[ibin][itheta][ir] += (*GRe_t[ibin][itheta])[ir];
GIm[ibin][itheta][ir] += (*GIm_t[ibin][itheta])[ir];
}
}
Qx1[ibin] += (*Qx1_t)[ibin];
Qy1[ibin] += (*Qy1_t)[ibin];
Q2[ibin] += (*Q2_t)[ibin];
totptall[ibin] += (*totptall_t)[ibin];
totetaall[ibin] += (*totetaall_t)[ibin];
Nevent[ibin] += (*Nevent_t)[ibin];
totmultall[ibin] += (*totmultall_t)[ibin];
tottrk[ibin] += (*tottrk_t)[ibin];
}
f[ifile]->Close();
}
fstrV<<setprecision(4)<<fixed;
fstrV<<"ibin"<<"\t"<<"itheta"<<"\t"<<"r0"<<"\t"<<"V"<<"\t"<<"sigma2"<<"\t"<<"chi"<<"\t"<<"Vn Errors"<<endl;
for(int ibin=0;ibin<nbin;ibin++){
avgmultall[ibin]=totmultall[ibin]/Nevent[ibin];
avgtrk[ibin]=tottrk[ibin]/Nevent[ibin];
for(int itheta=0;itheta<ntheta;itheta++){
for(ir=0; ir<nstepr; ir++){
G[ibin][itheta][ir]=TComplex(GRe[ibin][itheta][ir],GIm[ibin][itheta][ir]);
G[ibin][itheta][ir]/=Nevent[ibin];
Gmod2[ibin][itheta][ir]=TMath::Power(TComplex::Abs(G[ibin][itheta][ir]),2);
}
for(ir=0; ir<nstepr-1; ir++)
if(ir!=0 && Gmod2[ibin][itheta][ir]<=Gmod2[ibin][itheta][ir-1] && Gmod2[ibin][itheta][ir]<=Gmod2[ibin][itheta][ir+1]) break;
if(ir!=0 && ir<nstepr-1) r01[ibin][itheta]=r[ibin][ir];
else if(ir==0) {cout<<"ibin="<<ibin<<"\t"<<"itheta="<<itheta<<"\tminimum lies on ir = 0, please select proper range!"<<endl; continue;}
else {cout<<"ibin="<<ibin<<"\t"<<"itheta="<<itheta<<"\tminimum lies on ir = maximum "<<nstepr-1<<", please select proper range!"<<endl; continue;}
avgpt[ibin]=1.0*totptall[ibin]/totmultall[ibin];
avgeta[ibin]=1.0*totetaall[ibin]/totmultall[ibin];
if(isSimple==0) V[ibin][itheta]=Vmax[ibin]-ir*eps[ibin]+eps[ibin]*(Gmod2[ibin][itheta][ir+1]-Gmod2[ibin][itheta][ir-1])/2./(Gmod2[ibin][itheta][ir-1]-2*Gmod2[ibin][itheta][ir]+Gmod2[ibin][itheta][ir+1]);
else V[ibin][itheta]=j01/r01[ibin][itheta]; //simple method
r0[ibin][itheta]=j01/V[ibin][itheta];
V[ibin][itheta]/=avgmultall[ibin];
sigma2[ibin][itheta]=Q2[ibin]/Nevent[ibin]-(Qx1[ibin]/Nevent[ibin])*(Qx1[ibin]/Nevent[ibin])-(Qy1[ibin]/Nevent[ibin])*(Qy1[ibin]/Nevent[ibin]);
sigma2_[ibin]+=sigma2[ibin][itheta];
Vmean[ibin]+=V[ibin][itheta];
chi[ibin][itheta]=V[ibin][itheta]*avgmultall[ibin]/TMath::Sqrt(sigma2[ibin][itheta]);
}
//deltaVmean[ibin]+=TMath::Exp(j01*j01/2./chi[ibin][itheta]/chi[ibin][itheta]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Sin(nn*theta[itheta]/2.))+TMath::Exp(-j01*j01/2./chi[ibin][itheta]/chi[ibin][itheta]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Cos(nn*theta[itheta]/2.));
sigma2_[ibin]/=ntheta;
Vmean[ibin]/=ntheta;
sigma2_[ibin]-=TMath::Power(Vmean[ibin]*avgmultall[ibin],2);
chi_[ibin]=Vmean[ibin]*avgmultall[ibin]/TMath::Sqrt(sigma2_[ibin]);
for(int itheta=0;itheta<ntheta;itheta++){
deltaV[ibin][itheta]=V[ibin][itheta]/j01/TMath::BesselJ1(j01)*TMath::Sqrt((TMath::Exp(j01*j01/2./chi_[ibin]/chi_[ibin])+TMath::Exp(-j01*j01/2./chi_[ibin]/chi_[ibin])*TMath::BesselJ0(2*j01))/2./Nevent[ibin]);
deltaVmean[ibin]+=TMath::Exp(j01*j01/2./chi_[ibin]/chi_[ibin]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Sin(nn*theta[itheta]/2.))+TMath::Exp(-j01*j01/2./chi_[ibin]/chi_[ibin]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Cos(nn*theta[itheta]/2.));
fstrV<<ibin<<"\t"<<itheta<<"\t"<<r0[ibin][itheta]<<"\t"<<V[ibin][itheta]<<"\t"<<sigma2[ibin][itheta]<<"\t"<<chi[ibin][itheta]<<"\t"<<deltaV[ibin][itheta]<<endl;
}
fstrV<<"\t\t\t\t\t"<<sigma2_[ibin]<<"\t"<<chi_[ibin]<<endl;
deltaVmean[ibin]=Vmean[ibin]/j01/TMath::BesselJ1(j01)*TMath::Sqrt(deltaVmean[ibin]/ntheta/2./Nevent[ibin]);
fstrV<<endl;
}
fstrV<<"ibin"<<"\t"<<"avgmult"<<"\t"<<"avgpt"<<"\t"<<"avgeta"<<"\t"<<"Vn mean"<<"\t"<<"Vn mean Error"<<endl;
for(int ibin=0;ibin<nbin;ibin++){
fstrV<<ibin<<"\t"<<avgmultall[ibin]<<"\t"<<avgpt[ibin]<<"\t"<<avgeta[ibin]<<"\t"<<Vmean[ibin]<<"\t"<<deltaVmean[ibin]<<endl;
fstrV<<endl;
}
fstrV<<"ibin"<<"\t"<<"Nevent"<<"\t"<<"totmultall"<<"\t"<<"avgmultall"<<"\t"<<"avgtrk"<<endl;
for(int ibin=0;ibin<nbin;ibin++){
fstrV<<ibin<<"\t"<<Nevent[ibin]<<"\t"<<totmultall[ibin]<<"\t"<<avgmultall[ibin]<<"\t"<<avgtrk[ibin]<<endl;
fstrV<<endl;
}
if(isSum) TString outname = "mergedV_Sum.root";
else TString outname="mergedV_Prod.root";
TFile *outf = new TFile(outname,"Recreate");
Nevent.Write("Nevent");
totmultall.Write("totmultall");
avgmultall.Write("avgmultall");
tottrk.Write("tottrk");
avgtrk.Write("avgtrk");
Qx1.Write("Qx1");
Qy1.Write("Qy1");
Q2.Write("Q2");
avgpt.Write("avgpt");
avgeta.Write("avgeta");
Vmean.Write("Vmean");
deltaVmean.Write("deltaVmean");
chi_.Write("chi");
sigma2_.Write("sigma2");
for(int ibin=0;ibin<nbin;ibin++){
TDirectory *dir0 = outf->mkdir(Form("D_%d",ibin));dir0->cd();
r[ibin].Write("r");
sigma2[ibin].Write("sigma2"); chi[ibin].Write("chi0"); deltaV[ibin].Write("deltaV");
r0[ibin].Write("r0"); r01[ibin].Write("r01"); V[ibin].Write("V");
hqx[ibin]->Write(); hqy[ibin]->Write();hq2[ibin]->Write();hq2nonf[ibin]->Write();
for(int itheta=0;itheta<ntheta;itheta++){
TDirectory *dir1 = dir0->mkdir(Form("D_%d",itheta));dir1->cd();
GRe[ibin][itheta].Write(Form("GRe"));
GIm[ibin][itheta].Write(Form("GIm"));
Gmod2[ibin][itheta].Write(Form("G2"));
hq[ibin][itheta]->Write();
}
}
outf->Close();
}
void
Analyze(const TString mode="CLOSED",
UShort_t maxH=6,
Bool_t /*doLoops*/=false,
Int_t ifile=0 )
{
#ifdef __CINT__
gROOT->LoadMacro("correlations/Types.hh++");
gROOT->LoadMacro("correlations/Result.hh++");
gROOT->LoadMacro("correlations/QVector.hh++");
gROOT->LoadMacro("correlations/recursive/FromQVector.hh++");
gROOT->LoadMacro("correlations/recurrence/FromQVector.hh++");
gROOT->LoadMacro("correlations/closed/FromQVector.hh++");
gROOT->LoadMacro("correlations/test/ReadData.hh++");
#endif
// --- Setup of harmonics, etc -------------------------------------
gRandom->SetSeed(54321);
UShort_t emode = 0;
if (mode.EqualTo("closed", TString::kIgnoreCase)) emode = 0;
else if (mode.EqualTo("recurrence", TString::kIgnoreCase)) emode = 1;
else if (mode.EqualTo("recursive", TString::kIgnoreCase)) emode = 2;
else
Warning("Analyze", "Mode %s unknown, assuming CLOSED", mode.Data());
correlations::QVector q[nbin];
correlations::FromQVector* c[nbin];
correlations::HarmonicVector h(maxH);
for (UShort_t i = 0; i < maxH; i++) {
// Generate random harmonicx
// h[i] = -6 + gRandom->Integer(12);
h[0] = 2;
h[1] = -2;
h[2] = 2;
h[3] = -2;
h[4] = 2;
h[5] = -2;
h[6] = -2;
h[7] = 2;
// Printf("h_%d:\t%d", i, h[i]);
}
// Resize the Q-vector to fit the harmonics
for(int ibin=0;ibin<nbin;ibin++){
q[ibin] = correlations::QVector(0,0,false);
q[ibin].resize(h);
switch (emode) {
case 0: c[ibin] = new correlations::closed::FromQVector(q[ibin]); break;
case 1: c[ibin] = new correlations::recurrence::FromQVector(q[ibin]); break;
case 2: c[ibin] = new correlations::recursive::FromQVector(q[ibin]); break;
}
}
//Printf("Correlator: %s", c->name());
// --- Some histograms ---------------------------------------------
TH1* sumreals[nbin];
TH1* sumimags[nbin];
TH1* weights[nbin];
TVectorD tottrk;
TVectorD Nevent;
tottrk.ResizeTo(nbin);
tottrk.Zero();
Nevent.ResizeTo(nbin);
Nevent.Zero();
//TH1* reals = new TH1D("reals", "Re(C{n})", maxH-2+1, 2+.5, maxH+1+.5);
//TH1* imags = static_cast<TH1*>(reals->Clone("imags"));
for(int ibin=0;ibin<nbin;ibin++){
sumreals[ibin] = new TH1D(Form("sumreals_%d",ibin), "Re(C{n})", maxH-2+1, 2+.5, maxH+1+.5);
sumimags[ibin] = static_cast<TH1*>(sumreals[ibin]->Clone(Form("sumimags_%d",ibin)));
weights[ibin] = static_cast<TH1*>(sumreals[ibin]->Clone(Form("weights_%d",ibin)));
}
TProfile* timing = new TProfile("timing", "Timing", maxH-2+1, 2+.5,maxH+1+.5);
TH1* hs = new TH1I("harmonics", "Harmonics", maxH, 1.5, maxH+1.5);
/*reals->SetFillColor(kGreen+1);
reals->SetFillStyle(3001);
reals->SetStats(0);
imags->SetTitle("Im(C{n})");
imags->SetFillColor(kBlue+1);
imags->SetFillStyle(3001);
imags->SetStats(0);
timing->SetFillColor(kRed+1);
timing->SetFillStyle(3001);
timing->SetStats(0);
hs->SetFillColor(kMagenta+1);
hs->SetFillStyle(3001);
hs->SetStats(0);*/
for (UShort_t i = 0; i < maxH-1; i++) {
TString label = TString::Format("C{%d}", i+2);
// reals->GetXaxis()->SetBinLabel(i+1, label);
// imags->GetXaxis()->SetBinLabel(i+1, label);
timing->GetXaxis()->SetBinLabel(i+1, label);
hs->GetXaxis()->SetBinLabel(i+1,Form("h_{%d}", i+1));
hs->SetBinContent(i+1, h[i]);
}
hs->GetXaxis()->SetBinLabel(maxH,Form("h_{%d}", maxH));
hs->SetBinContent(maxH, h[maxH-1]);
TStopwatch timer;
// --- Setup input ------------------------------------------------
TFile* file = TFile::Open(Form("%s/vndata_50k_%d.root",dir.Data(),ifile), "READ");
TTree* tree = static_cast<TTree*>(file->Get("tree"));
// TArrayD phis(0);
Int_t M;
Float_t phi[10000],pt[10000],eta[10000];
// TArrayD weights(0);
// Double_t phiR = 0;
// TArrayD* pPhis = &phis;
// TArrayD* pWeights = &weights;
tree->SetBranchAddress("phig", phi);
tree->SetBranchAddress("ptg",pt);
tree->SetBranchAddress("etag",eta);
tree->SetBranchAddress("n", &M);
// tree->SetBranchAddress("weight", &pWeights);
// tree->SetBranchAddress("event", &phiR);
// --- The results -------------------------------------------------
const UShort_t nQ = maxH - 1;
correlations::ResultVector qs[nbin];
for(int ibin=0;ibin<nbin;ibin++)
qs[ibin] = correlations::ResultVector(nQ);
// --- Event loop --------------------------------------------------
Int_t nEvents = tree->GetEntries();
for (Int_t event = 0; event < nEvents; event++) {
tree->GetEntry(event);
int ntrk = M; int xbin=-1;
for(int j=0;j<nbin;j++)
if(ntrk<trkbin[j]&&ntrk>=trkbin[j+1])
xbin=j;
if(xbin<0 || xbin==nbin) continue;
tottrk[xbin]+=ntrk;
q[xbin].reset();
// printf("Event # %4u %4d particles ", event++, phis.GetSize());
for (UShort_t pa = 0; pa < M; pa++){
if(fabs(eta[pa])>etamax) continue;
if(pt[pa]<ptmin||pt[pa]>ptmax) continue; //event selection
// phis.Set(n,pPhis);
q[xbin].fill(phi[pa], 1.);
}
for (UShort_t i = 0; i < nQ; i++) {
UShort_t n = i + 2;
// printf("%s%d", i == 0 ? "" : "..", n);
timer.Reset();
timer.Start();
qs[xbin][i] += c[xbin]->calculate(n, h);
timer.Stop();
timing->Fill(n+.5, timer.RealTime());
}
// printf(" done\n");
Nevent[xbin]++;
}
file->Close();
for(int ibin=0;ibin<nbin;ibin++){
for (UShort_t i = 0; i < nQ; i++) {
// UShort_t iq = i+2;
// Double_t t = timing->GetBinContent(i+1);
// correlations::Complex rc = qs[i].eval();
// Printf("QC{%2d}: %12g + %12gi <t>: %10gs",
// iq, rc.real(), rc.imag(), t);
// if(i==0)Printf("v2{%2d}: %3g\n",2,sqrt(qs[0].eval().real()));
// if(i==2)Printf("v2{%2d}: %3g\n",4,TMath::Power(fabs(qs[2].eval().real()),1./4));
// if(i==4)Printf("v2{%2d}: %3g\n",6,TMath::Power(fabs(qs[4].eval().real()),1./6));
sumreals[ibin]->SetBinContent(i+1,qs[ibin][i]._sum.real());
sumimags[ibin]->SetBinContent(i+1,qs[ibin][i]._sum.imag());
weights[ibin]->SetBinContent(i+1,qs[ibin][i]._weights);
//reals->SetBinContent(i+1, rc.real());
//imags->SetBinContent(i+1, rc.imag());
}
}
/*
TCanvas* can = new TCanvas("C", "C");
can->SetTopMargin(0.15);
can->SetBottomMargin(0.15);
can->SetRightMargin(0.03);
can->Divide(1,3, 0, 0);
DrawInPad(can, 3, timing, true);
DrawInPad(can, 1, reals);
DrawInPad(can, 2, imags);
can->cd(0);
TLatex* ltx = new TLatex(0.5,0.995,c->name());
ltx->SetNDC(true);
ltx->SetTextAlign(23);
ltx->SetTextSize(0.04);
ltx->Draw();
can->Modified();
can->Update();
can->cd();
*/
TString out(mode);
out.ToLower();
file = TFile::Open(Form("%s/%s_%d.root",outdir.Data(),out.Data(),ifile), "RECREATE");
for(int ibin=0;ibin<nbin;ibin++){
sumimags[ibin]->Write();
sumreals[ibin]->Write();
weights[ibin]->Write();
}
Nevent.Write("Nevent");
tottrk.Write("tottrk");
timing->Write();
hs->Write();
file->Write();
file->Close();
for(int ibin=0;ibin<nbin;ibin++){
delete sumimags[ibin];
delete sumreals[ibin];
delete weights[ibin];
}
delete timing;
delete hs;
}
void merge(){
TVectorD Nevent; Nevent.ResizeTo(nbin); Nevent.Zero();
// TVectorD totmultall; totmultall.ResizeTo(nbin); totmultall.Zero();
// TVectorD avgmultall; avgmultall.ResizeTo(nbin); avgmultall.Zero();
TVectorD tottrk; tottrk.ResizeTo(nbin); tottrk.Zero();
// TVectorD totptall; totptall.ResizeTo(nbin); totptall.Zero();
// TVectorD totetaall; totetaall.ResizeTo(nbin); totetaall.Zero();
TVectorD avgtrk; avgtrk.ResizeTo(nbin); avgtrk.Zero();
// TVectorD avgmult; avgmult.ResizeTo(nbin);
TVectorD qc22, qc24, qc26, qc28;
qc22.ResizeTo(nbin); qc22.Zero();
qc24.ResizeTo(nbin); qc24.Zero();
qc26.ResizeTo(nbin); qc26.Zero();
qc28.ResizeTo(nbin); qc28.Zero();
TVectorD c22, c24, c26, c28;
c22.ResizeTo(nbin); c22.Zero();
c24.ResizeTo(nbin); c24.Zero();
c26.ResizeTo(nbin); c26.Zero();
c28.ResizeTo(nbin); c28.Zero();
TVectorD v22, v24, v26, v28;
v22.ResizeTo(nbin); v22.Zero();
v24.ResizeTo(nbin); v24.Zero();
v26.ResizeTo(nbin); v26.Zero();
v28.ResizeTo(nbin); v28.Zero();
TH1* sumreals[nbin];
TH1* sumimags[nbin];
TH1* weights[nbin];
for(int ibin=0;ibin<nbin;ibin++){
sumreals[ibin] = new TH1D(Form("sumreals",ibin), "Re(C{n})", maxN-2+1, 2+.5, maxN+1+.5);
sumimags[ibin] = static_cast<TH1*>(sumreals[ibin]->Clone(Form("sumimags",ibin)));
weights[ibin] = static_cast<TH1*>(sumreals[ibin]->Clone(Form("weights",ibin)));
}
TFile *fout = new TFile(Form("%s_merged.root",method.Data()),"Recreate");
TFile *f[nFileAll];
for(int ifile=0; ifile<nFileAll; ifile++){
f[ifile] = TFile::Open(Form("%s/%s_%d.root",outdir.Data(),method.Data(),ifile));
TVectorD* Nevent_t = (TVectorD*)f[ifile]->Get(Form("Nevent"));
// TVectorD* totmultall_t = (TVectorD*)f[ifile]->Get(Form("totmultall"));
TVectorD* tottrk_t = (TVectorD*)f[ifile]->Get(Form("tottrk"));
// TVectorD* totptall_t = (TVectorD*)f[ifile]->Get(Form("totptall"));
// TVectorD* totetaall_t = (TVectorD*)f[ifile]->Get(Form("totetaall"));
for(int ibin=0;ibin<nbin;ibin++){
// totptall[ibin] += (*totptall_t)[ibin];
// totetaall[ibin] += (*totetaall_t)[ibin];
Nevent[ibin] += (*Nevent_t)[ibin];
// totmultall[ibin] += (*totmultall_t)[ibin];
tottrk[ibin] += (*tottrk_t)[ibin];
TH1D* sumreals_t = (TH1D*)f[ifile]->Get(Form("sumreals_%d",ibin));
TH1D* sumimags_t = (TH1D*)f[ifile]->Get(Form("sumimags_%d",ibin));
TH1D* weights_t = (TH1D*)f[ifile]->Get(Form("weights_%d",ibin));
sumreals[ibin]->Add(sumreals_t);
sumimags[ibin]->Add(sumimags_t);
weights[ibin]->Add(weights_t);
}
f[ifile]->Close();
}
cout<<"v2{gen} = "<<inV2<<endl;
for(int ibin=0;ibin<nbin;ibin++){
avgtrk[ibin] = tottrk[ibin]/Nevent[ibin];
TH1D* hQCr = (TH1D*)sumreals[ibin]->Clone("hQCr");
TH1D* hQCi = (TH1D*)sumimags[ibin]->Clone("hQCi");
hQCr->Divide(weights[ibin]);
hQCi->Divide(weights[ibin]);
complex<double> qc22com(hQCr->GetBinContent(1),hQCi->GetBinContent(1));
complex<double> qc24com(hQCr->GetBinContent(3),hQCi->GetBinContent(3));
complex<double> qc26com(hQCr->GetBinContent(5),hQCi->GetBinContent(5));
qc22[ibin] = abs(qc22com);
qc24[ibin] = abs(qc24com);
qc26[ibin] = abs(qc26com);
c22[ibin] = qc22[ibin];
c24[ibin] = qc24[ibin] - 2*qc22[ibin]*qc22[ibin];
c26[ibin] = qc26[ibin] - 9*qc22[ibin]*qc24[ibin] + 12*qc22[ibin]*qc22[ibin]*qc22[ibin];
v22[ibin] = sqrt(c22[ibin]);
v24[ibin] = TMath::Power(-c24[ibin],1./4);
v26[ibin] = TMath::Power(1./4*c26[ibin],1./6);
//v24[ibin] = TMath::Power(qc24[ibin],1./4);
//v26[ibin] = TMath::Power(qc26[ibin],1./6);
cout<<"avgtrk = "<<avgtrk[ibin]<<endl;
cout<<"v2{2} = "<<v22[ibin]<<endl;
cout<<"v2{4} = "<<v24[ibin]<<endl;
cout<<"v2{6} = "<<v26[ibin]<<endl;
fout->cd();
Nevent.Write("Nevent");
avgtrk.Write("avgtrk");
qc22.Write("qc22"); qc24.Write("qc24"); qc26.Write("qc26"); qc28.Write("qc28");
c22.Write("c22"); c24.Write("c24"); c26.Write("c26"); c28.Write("c28");
v22.Write("v22"); v24.Write("v24"); v26.Write("v26"); v28.Write("v28");
TDirectory *dir0 = (TDirectory*)fout->mkdir(Form("D_%d",ibin));
dir0->cd();
sumreals[ibin]->Write();
sumimags[ibin]->Write();
weights[ibin]->Write();
}
}
void getResv2() {
ifstream f_("merged.root");
if(!f_.good()) {
system("hadd merged.root /scratch/xuq7/flow/EPAnalyzer/pPbDataV205m150/AnaEP_*.root");
}
TFile *fmerged = TFile::Open("merged.root");
TH1D* hEPPhip[nbin][neta];
TH1D* hEPPhim[nbin][neta];
TH1D* hEPPhic[nbin][neta];
for(int ibin=0; ibin<nbin; ibin++) {
for(int ieta=0; ieta<neta; ieta++) {
hEPPhip[ibin][ieta] = (TH1D*)fmerged->Get(Form("hEPPhip_%d_%d",ibin,ieta));
hEPPhim[ibin][ieta] = (TH1D*)fmerged->Get(Form("hEPPhim_%d_%d",ibin,ieta));
hEPPhic[ibin][ieta] = (TH1D*)fmerged->Get(Form("hEPPhic_%d_%d",ibin,ieta));
}
}
ofstream fstrV;
fstrV.open("v2.txt");
TFile *f[nFileAll];
TVectorD Nevent;
Nevent.ResizeTo(nbin);
Nevent.Zero();
TVectorD totmultall;
totmultall.ResizeTo(nbin);
totmultall.Zero();
TVectorD EPR[nbin], EPRpc[nbin], EPRpm[nbin], EPRcm[nbin];
for(int ibin=0; ibin<nbin; ibin++) {
EPR[ibin].ResizeTo(neta);
EPR[ibin].Zero();
EPRpc[ibin].ResizeTo(neta);
EPRpc[ibin].Zero();
EPRpm[ibin].ResizeTo(neta);
EPRpm[ibin].Zero();
EPRcm[ibin].ResizeTo(neta);
EPRcm[ibin].Zero();
}
TVectorD avgmult;
avgmult.ResizeTo(nbin);
TVectorD totmult[nbin], totpt[nbin], avgpt[nbin];
TVectorD Vobs[nbin][neta], Vcorr[nbin][neta], Vcorr3[nbin][neta];
for(int ibin=0; ibin<nbin; ibin++) {
totmult[ibin].ResizeTo(npt);
totmult[ibin].Zero();
totpt[ibin].ResizeTo(npt);
totpt[ibin].Zero();
avgpt[ibin].ResizeTo(npt);
avgpt[ibin].Zero();
for(int ieta=0; ieta<neta; ieta++) {
Vobs[ibin][ieta].ResizeTo(npt);
Vobs[ibin][ieta].Zero();
Vcorr[ibin][ieta].ResizeTo(npt);
Vcorr[ibin][ieta].Zero();
Vcorr3[ibin][ieta].ResizeTo(npt);
Vcorr3[ibin][ieta].Zero();
}
}
for(int ifile=0; ifile<nFileAll; ifile++) {
f[ifile] = TFile::Open(Form("/scratch/xuq7/flow/EPAnalyzer/pPbDataV205m150/AnaEP_%d.root",ifile));
TVectorD* Nevent_t = (TVectorD*)f[ifile]->Get(Form("Nevent"));
TVectorD* totmultall_t = (TVectorD*)f[ifile]->Get(Form("totmultall"));
for(int ibin=0; ibin<nbin; ibin++) {
TVectorD* EPR_t = (TVectorD*)f[ifile]->Get(Form("EPR_%d",ibin));
TVectorD* EPRpc_t = (TVectorD*)f[ifile]->Get(Form("EPRpc_%d",ibin));
TVectorD* EPRpm_t = (TVectorD*)f[ifile]->Get(Form("EPRpm_%d",ibin));
TVectorD* EPRcm_t = (TVectorD*)f[ifile]->Get(Form("EPRcm_%d",ibin));
TVectorD* totmult_t = (TVectorD*)f[ifile]->Get(Form("totmult_%d",ibin));
TVectorD* totpt_t = (TVectorD*)f[ifile]->Get(Form("totpt_%d",ibin));
Nevent[ibin] += (*Nevent_t)[ibin];
totmultall[ibin] += (*totmultall_t)[ibin];
for(int iptbin=0; iptbin<npt; iptbin++) {
totmult[ibin][iptbin] += (*totmult_t)[iptbin];
totpt[ibin][iptbin] += (*totpt_t)[iptbin];
}
for(int ieta=0; ieta<neta; ieta++) {
EPR[ibin][ieta] += (*EPR_t)[ieta];
EPRpc[ibin][ieta] += (*EPRpc_t)[ieta];
EPRpm[ibin][ieta] += (*EPRpm_t)[ieta];
EPRcm[ibin][ieta] += (*EPRcm_t)[ieta];
TVectorD* Vobs_t = (TVectorD*)f[ifile]->Get(Form("v2obs_%d_%d",ibin,ieta));
for(int iptbin=0; iptbin<npt; iptbin++) {
Vobs[ibin][ieta][iptbin] += (*Vobs_t)[iptbin];
}
}
}
f[ifile]->Close();
}
TFile *outf = new TFile("mergedVobs.root","Recreate");
Nevent.Write("Nevent");
totmultall.Write("totmultall");
for(int ibin=0; ibin<nbin; ibin++) {
for(int ieta=0; ieta<neta; ieta++) {
hEPPhip[ibin][ieta]->Write();
hEPPhim[ibin][ieta]->Write();
hEPPhic[ibin][ieta]->Write();
}
}
fstrV<<setprecision(4)<<fixed;
fstrV<<"ibin"<<"\t"<<"ieta"<<"\t"<<"iptbin"<<"\t"<<"avgpt"<<"\t"<<"v2obs"<<"\t"<<"v2corr"<<"\t"<<"v2corr3"<<endl;
for(int ibin=0; ibin<nbin; ibin++) {
avgmult[ibin]=(double)totmultall[ibin]/Nevent[ibin];
for(int iptbin=0; iptbin<npt; iptbin++) {
avgpt[ibin][iptbin]=totpt[ibin][iptbin]/totmult[ibin][iptbin];
}
for(int ieta=0; ieta<neta; ieta++) {
//EPR[ibin][ieta]=EPR[ibin][ieta]/Nevent[ibin];
EPR[ibin][ieta]=TMath::Sqrt(EPR[ibin][ieta]/Nevent[ibin]);
EPRpc[ibin][ieta]=EPRpc[ibin][ieta]/Nevent[ibin];
EPRcm[ibin][ieta]=EPRcm[ibin][ieta]/Nevent[ibin];
EPRpm[ibin][ieta]=EPRpm[ibin][ieta]/Nevent[ibin];
for(int iptbin=0; iptbin<npt; iptbin++) {
Vobs[ibin][ieta][iptbin]/=totmult[ibin][iptbin];
Vcorr[ibin][ieta][iptbin]=Vobs[ibin][ieta][iptbin]/EPR[ibin][ieta];
Vcorr3[ibin][ieta][iptbin]=Vobs[ibin][ieta][iptbin]/sqrt(EPRpc[ibin][ieta]*EPRpm[ibin][ieta]/EPRcm[ibin][ieta]);
fstrV<<trkbin[ibin+1]<<"-"<<trkbin[ibin]<<"\t"<<etap[ieta]<<"-"<<etan[ieta]<<"\t"<<ptbin[iptbin]<<"-"<<ptbin[iptbin+1]<<"\t"<<avgpt[ibin][iptbin]<<"\t"<<Vobs[ibin][ieta][iptbin]<<"\t"<<Vcorr[ibin][ieta][iptbin]<<"\t"<<Vcorr3[ibin][ieta][iptbin]<<endl;
}
fstrV<<endl;
}
TDirectory *dir0 = outf->mkdir(Form("D_%d",ibin));
dir0->cd();
totmult[ibin].Write("totmult");
avgpt[ibin].Write("avgpt");
EPR[ibin].Write("EPR");
EPRpc[ibin].Write("EPRpc");
EPRpm[ibin].Write("EPRpm");
EPRcm[ibin].Write("EPRcm");
for(int ieta=0; ieta<neta; ieta++) {
TDirectory *dir1 = dir0->mkdir(Form("E_%d",ieta));
dir1->cd();
Vobs[ibin][ieta].Write("v2obs");
Vcorr[ibin][ieta].Write("v2");
Vcorr3[ibin][ieta].Write("v23sub");
}
}
outf->cd();
avgmult.Write("avgmult");
outf->Close();
remove("merged.root");
}
//------------------------Deal with the number and get the output stored in txt files and root files
void getResVsub(){
double Vmax[nbin], eps[nbin];
for(int ibin=0; ibin<nbin ;ibin++){
Vmax[ibin]=0.065*(trkbin[ibin]+30);
eps[ibin]=0.00025*(trkbin[ibin]+30);
}
ofstream fstrV;
double theta[ntheta];
TVectorD avgmult, avgpt, totptall;
TVectorD Qx1, Qy1, Q2;
TVectorD Gmod2[nbin][ntheta];
TVectorD sigma2[nbin],deltaV[nbin];
TVectorD sigma2_,chi_;
TVectorD deltaVmean, Vmean;
TVectorD deltaVmeanmean, Vmeanmean, sigmaVmeanmean;
TVectorD r[nbin];
TVectorD r0[nbin], r01[nbin], V[nbin], chi[nbin];
TVectorD GRe[nbin][ntheta]; TVectorD* GRe_t[nbin][ntheta];
TVectorD GIm[nbin][ntheta]; TVectorD* GIm_t[nbin][ntheta];
TComplex G[nbin][ntheta][nstepr];
if(isSum) fstrV.open("V_Sum_sub.txt");
else fstrV.open("V_Prod_sub.txt");
if(isSum) TString outname = "mergedV_Sum_sub.root";
else TString outname="mergedV_Prod_sub.root";
TFile *outf = new TFile(outname,"Recreate");
int subsample = (int)(nFileAll/nsamples);
TFile *f[nFileAll];
for(int ibin=0; ibin<nbin; ibin++){
r[ibin].ResizeTo(nstepr);
for(int ir=0; ir<nstepr; ir++){
if(isSimple==0) r[ibin][ir]=j01/(Vmax[ibin]-eps[ibin]*ir);
else r[ibin][ir]=0.00025*20*(ir+1);
}
}
Vmeanmean.ResizeTo(nbin); Vmeanmean.Zero();
deltaVmeanmean.ResizeTo(nbin); deltaVmeanmean.Zero();
sigmaVmeanmean.ResizeTo(nbin); sigmaVmeanmean.Zero();
for(int isample=0;isample<nsamples;isample++){
cout<<"start with "<<isample<<"\t";
int start=isample*subsample;
int end=(isample+1)*subsample;
TVectorD Nevent; Nevent.ResizeTo(nbin); Nevent.Zero();
TVectorD totmultall; totmultall.ResizeTo(nbin); totmultall.Zero();
for(int itheta=0;itheta<ntheta;itheta++)
theta[itheta]=itheta*TMath::Pi()/ntheta/nn;
for(int ibin=0;ibin<nbin;ibin++){
r0[ibin].ResizeTo(ntheta);
r01[ibin].ResizeTo(ntheta);
sigma2[ibin].ResizeTo(ntheta);
V[ibin].ResizeTo(ntheta);
deltaV[ibin].ResizeTo(ntheta);
chi[ibin].ResizeTo(ntheta);
for(int itheta=0;itheta<ntheta;itheta++){
Gmod2[ibin][itheta].ResizeTo(nstepr);
GRe[ibin][itheta].ResizeTo(nstepr);
GRe[ibin][itheta].Zero();
GIm[ibin][itheta].ResizeTo(nstepr);
GIm[ibin][itheta].Zero();
}
}
avgmult.ResizeTo(nbin); deltaVmean.ResizeTo(nbin); Vmean.ResizeTo(nbin);
deltaVmean.Zero(); Vmean.Zero(); deltaVmean.Zero();
totptall.ResizeTo(nbin); totptall.Zero();
avgpt.ResizeTo(nbin);
Qx1.ResizeTo(nbin); Qx1.Zero();
Qy1.ResizeTo(nbin); Qy1.Zero();
Q2.ResizeTo(nbin); Q2.Zero();
sigma2_.ResizeTo(nbin);sigma2_.Zero();
chi_.ResizeTo(nbin);chi_.Zero();
for(int ifile=start; ifile<end; ifile++){
if(isSum) f[ifile] = TFile::Open(Form("/scratch/xuq7/flow/pbsjoboutput/ToyMCHole/pPbDataV205m185/AnaV_Sum_%d.root",ifile));
else f[ifile] = TFile::Open(Form("/scratch/xuq7/flow/pbsjoboutput/ToyMCHole/pPbDataV205m185/AnaV_Prod_%d.root",ifile));
TVectorD* Nevent_t = (TVectorD*)f[ifile]->Get(Form("Nevent"));
TVectorD* totmultall_t = (TVectorD*)f[ifile]->Get(Form("totmultall"));
TVectorD* Qx1_t = (TVectorD*)f[ifile]->Get(Form("Qx1"));
TVectorD* Qy1_t = (TVectorD*)f[ifile]->Get(Form("Qy1"));
TVectorD* Q2_t = (TVectorD*)f[ifile]->Get(Form("Q2"));
TVectorD* totptall_t = (TVectorD*)f[ifile]->Get(Form("totptall"));
for(int ibin=0;ibin<nbin;ibin++){
for(int itheta=0;itheta<ntheta;itheta++){
GRe_t[ibin][itheta] = (TVectorD*)f[ifile]->Get(Form("GRe_%d_%d",ibin,itheta));
GIm_t[ibin][itheta] = (TVectorD*)f[ifile]->Get(Form("GIm_%d_%d",ibin,itheta));
for(ir=0; ir<nstepr; ir++){
GRe[ibin][itheta][ir] += (*GRe_t[ibin][itheta])[ir];
GIm[ibin][itheta][ir] += (*GIm_t[ibin][itheta])[ir];
}
}
Qx1[ibin] += (*Qx1_t)[ibin];
Qy1[ibin] += (*Qy1_t)[ibin];
Q2[ibin] += (*Q2_t)[ibin];
totptall[ibin] += (*totptall_t)[ibin];
Nevent[ibin] += (*Nevent_t)[ibin];
totmultall[ibin] += (*totmultall_t)[ibin];
}
f[ifile]->Close();
}
fstrV<<setprecision(4)<<fixed;
fstrV<<"ibin"<<"\t"<<"itheta"<<"\t"<<"r0"<<"\t"<<"V"<<"\t"<<"sigma2"<<"\t"<<"chi"<<"\t"<<"Vn Errors"<<endl;
for(int ibin=0;ibin<nbin;ibin++){
for(int itheta=0;itheta<ntheta;itheta++){
for(ir=0; ir<nstepr; ir++){
G[ibin][itheta][ir]=TComplex(GRe[ibin][itheta][ir],GIm[ibin][itheta][ir]);
G[ibin][itheta][ir]/=Nevent[ibin];
Gmod2[ibin][itheta][ir]=TMath::Power(TComplex::Abs(G[ibin][itheta][ir]),2);
}
for(ir=0; ir<nstepr-1; ir++)
if(ir!=0 && Gmod2[ibin][itheta][ir]<=Gmod2[ibin][itheta][ir-1] && Gmod2[ibin][itheta][ir]<=Gmod2[ibin][itheta][ir+1]) break;
if(ir!=0 && ir<nstepr-1) r01[ibin][itheta]=r[ibin][ir];
else if(ir==0) {cout<<"ibin="<<ibin<<"\t"<<"itheta="<<itheta<<"\tminimum lies on ir = 0, please select proper range!"<<endl; continue;}
else {cout<<"ibin="<<ibin<<"\t"<<"itheta="<<itheta<<"\tminimum lies on ir = maximum "<<nstepr-1<<", please select proper range!"<<endl; continue;}
avgmult[ibin]=1.0*totmultall[ibin]/Nevent[ibin];
avgpt[ibin]=1.0*totptall[ibin]/totmultall[ibin];
if(isSimple==0) V[ibin][itheta]=Vmax[ibin]-ir*eps[ibin]+eps[ibin]*(Gmod2[ibin][itheta][ir+1]-Gmod2[ibin][itheta][ir-1])/2./(Gmod2[ibin][itheta][ir-1]-2*Gmod2[ibin][itheta][ir]+Gmod2[ibin][itheta][ir+1]);
else V[ibin][itheta]=j01/r0[ibin][itheta]; //simple method
r0[ibin][itheta]=j01/V[ibin][itheta];
V[ibin][itheta]/=avgmult[ibin];
//sigma2[ibin][itheta]=Q2[ibin]/Nevent[ibin]-(Qx1[ibin]/Nevent[ibin])*(Qx1[ibin]/Nevent[ibin])-(Qy1[ibin]/Nevent[ibin])*(Qy1[ibin]/Nevent[ibin])-(V[ibin][itheta]*avgmult[ibin])*(V[ibin][itheta]*avgmult[ibin]);
sigma2[ibin][itheta]=Q2[ibin]/Nevent[ibin]-(Qx1[ibin]/Nevent[ibin])*(Qx1[ibin]/Nevent[ibin])-(Qy1[ibin]/Nevent[ibin])*(Qy1[ibin]/Nevent[ibin]);
sigma2_[ibin]+=sigma2[ibin][itheta];
Vmean[ibin]+=V[ibin][itheta];
chi[ibin][itheta]=V[ibin][itheta]*avgmult[ibin]/TMath::Sqrt(sigma2[ibin][itheta]);
//deltaV[ibin][itheta]=V[ibin][itheta]/j01/TMath::BesselJ1(j01)*TMath::Sqrt((TMath::Exp(j01*j01/2./chi[ibin][itheta]/chi[ibin][itheta])+TMath::Exp(-j01*j01/2./chi[ibin][itheta]/chi[ibin][itheta])*TMath::BesselJ0(2*j01))/2./Nevent[ibin]);
}
sigma2_[ibin]/=ntheta;
Vmean[ibin]/=ntheta;
sigma2_[ibin]-=TMath::Power(Vmean[ibin]*avgmult[ibin],2);
chi_[ibin]=Vmean[ibin]*avgmult[ibin]/TMath::Sqrt(sigma2_[ibin]);
//deltaVmean[ibin]+=TMath::Exp(j01*j01/2./chi[ibin][itheta]/chi[ibin][itheta]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Sin(nn*theta[itheta]/2.))+TMath::Exp(-j01*j01/2./chi[ibin][itheta]/chi[ibin][itheta]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Cos(nn*theta[itheta]/2.));
for(int itheta=0;itheta<ntheta;itheta++){
deltaV[ibin][itheta]=V[ibin][itheta]/j01/TMath::BesselJ1(j01)*TMath::Sqrt((TMath::Exp(j01*j01/2./chi_[ibin]/chi_[ibin])+TMath::Exp(-j01*j01/2./chi_[ibin]/chi_[ibin])*TMath::BesselJ0(2*j01))/2./Nevent[ibin]);
deltaVmean[ibin]+=TMath::Exp(j01*j01/2./chi_[ibin]/chi_[ibin]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Sin(nn*theta[itheta]/2.))+TMath::Exp(-j01*j01/2./chi_[ibin]/chi_[ibin]*TMath::Cos(nn*theta[itheta]))*TMath::BesselJ0(2*j01*TMath::Cos(nn*theta[itheta]/2.));
fstrV<<ibin<<"\t"<<itheta<<"\t"<<r0[ibin][itheta]<<"\t"<<V[ibin][itheta]<<"\t"<<sigma2[ibin][itheta]<<"\t"<<chi[ibin][itheta]<<"\t"<<deltaV[ibin][itheta]<<endl;
}
fstrV<<endl;
deltaVmean[ibin]=Vmean[ibin]/j01/TMath::BesselJ1(j01)*TMath::Sqrt(deltaVmean[ibin]/ntheta/2./Nevent[ibin]);
}
fstrV<<endl;
fstrV<<"ibin"<<"\t"<<"avgmult"<<"\t"<<"avgpt"<<"\t"<<"Vn mean"<<"\t"<<"Vn mean Error"<<endl;
for(int ibin=0;ibin<nbin;ibin++){
fstrV<<ibin<<"\t"<<avgmult[ibin]<<"\t"<<avgpt[ibin]<<"\t"<<Vmean[ibin]<<"\t"<<deltaVmean[ibin]<<endl;
}
fstrV<<endl;
TDirectory *dirsample = outf->mkdir(Form("s_%d",isample));
dirsample->cd();
Nevent.Write("Nevent");
totmultall.Write("totmultall");
Qx1.Write("Qx1");
Qy1.Write("Qy1");
Q2.Write("Q2");
totptall.Write("totptall");
avgmult.Write("avgmult");
avgpt.Write("avgpt");
Vmean.Write("Vmean");
deltaVmean.Write("deltaVmean");
chi_.Write("chi");
for(int ibin=0;ibin<nbin;ibin++){
TDirectory *dir0 = dirsample->mkdir(Form("D_%d",ibin));dir0->cd();
r[ibin].Write("r");
Vmeanmean[ibin]+=Vmean[ibin]/nsamples;
deltaVmeanmean[ibin]+=deltaVmean[ibin]/nsamples;
sigmaVmeanmean[ibin]+=TMath::Power(Vmean[ibin]/nsamples,2);
sigma2[ibin].Write("sigma2"); chi[ibin].Write("chi0"); deltaV[ibin].Write("deltaV");
r0[ibin].Write("r0"); r01[ibin].Write("r01"); V[ibin].Write("V");
for(int itheta=0;itheta<ntheta;itheta++){
TDirectory *dir1 = dir0->mkdir(Form("D_%d",itheta));dir1->cd();
GRe[ibin][itheta].Write(Form("GRe"));
GIm[ibin][itheta].Write(Form("GIm"));
Gmod2[ibin][itheta].Write(Form("G2"));
}
}
}
fstrV<<"ibin"<<"\t"<<"Vn mean mean"<<"\t"<<"Vn mean mean Error"<<"\t"<<"Vn mean deviation"<<endl;
for(int ibin=0;ibin<nbin;ibin++){
sigmaVmeanmean[ibin]=TMath::Sqrt(sigmaVmeanmean[ibin]*nsamples-Vmeanmean[ibin]*Vmeanmean[ibin]);
fstrV<<endl<<ibin<<"\t"<<Vmeanmean[ibin]<<"\t"<<deltaVmeanmean[ibin]<<"\t"<<sigmaVmeanmean[ibin]<<endl;
}
outf->cd();
Vmeanmean.Write("Vmeanmean");
deltaVmeanmean.Write("deltaVmeanmean");
sigmaVmeanmean.Write("sigmaVmeanmean");
cout<<endl;
outf->Close();
}
