[u/mrichter/AliRoot.git] / PWGPP / analysisQA / processMakeQA2pc.C

const char * tag = "_Hadrons_QA_PbPb";
Float_t gpTMin = 0.51;
Float_t gpTMax = 49.99;
const char* lastFileName = 0;
void* cacheSameEvent = 0;
void* cacheMixedEvent = 0;

void processMakeQA2pc(const char * filename="AnalysisResults",  //file name without ".root"
		      TString suffix = "eps",
		      const char *outfile="MakeQA2pc_output.root"){
  
  loadlibs();
  if (!gGrid && TString(filename).BeginsWith("alien://"))
    TGrid::Connect("alien://");
  
  //fill CFContainers
  FillParentTHnSparse(Form("%s.root",filename),kFALSE,tag);
  
  Double_t ptTmin[]={0.5,0.5,2.0,2.0};
  Double_t ptTmax[]={1.0,1.0,4.0,4.0};
  Double_t ptAmin[]={0.5,0.5,2.0,2.0};
  Double_t ptAmax[]={1.0,1.0,4.0,4.0};
  Int_t   centBegin[]={  0, 60,  0, 60};
  Int_t     centEnd[]={ 10, 90, 10, 90};
  

  for(Int_t i=0;i<4;i++){
    TCanvas *c=new TCanvas(Form("c%d",i),Form("c%d",i),1200,1200);
    DrawSameMixedSV(Form("%s_zvtx.root",filename),ptTmin[i],ptTmax[i],ptAmin[i],ptAmax[i],centBegin[i],centEnd[i],8,c,i,outfile);
    c->SaveAs(Form("fig_cf_c%d.%s",i,suffix.Data()));
  }
  
  TCanvas *ccorr=new TCanvas("ccorr","ccorr",1200,1200);
  DrawCorrelation(Form("%s_zvtx.root",filename),ccorr,outfile);


  ccorr->SaveAs(Form("fig_cf_ccorr.%s",suffix.Data()));
  
}

void SetupRanges(void* obj)
{
  if (!obj)
    return;
  ((AliUEHistograms*) obj)->SetEtaRange(0, 0);
  ((AliUEHistograms*) obj)->SetPtRange(gpTMin, gpTMax);
  ((AliUEHistograms*) obj)->SetCombineMinMax(kTRUE);
}

void DrawCorrelation(const char* fileName,TCanvas *c,const char *outfile){
  
  AliUEHistograms* h = (AliUEHistograms*) GetUEHistogram(fileName);
  c->Divide(1,3);
  c->cd(1);
  gPad->SetLogz();
  h->((TH2F*)h->GetCorrelationpT())->DrawCopy("colz")->GetYaxis()->SetRangeUser(0.,20.);
  c->cd(2);
  h->((TH2F*)h->GetCorrelationEta())->DrawCopy("colz")->GetYaxis()->SetRangeUser(-1.5,1.5);
  c->cd(3);
  h->((TH2F*)h->GetCorrelationPhi())->DrawCopy("colz");

  //Added by sjena
  TFile *fout = TFile::Open(outfile,"UPDATE");
  fout->ls();

  TDirectoryFile *cdd = NULL;
  cdd = (TDirectoryFile*)fout->Get("CF");
  if(!cdd) {
    Printf("Warning: CF <dir> doesn't exist, creating a new one");
    cdd = (TDirectoryFile*)fout->mkdir("CF");
  }
  cdd->cd();
  cdd->ls();
  

  TH2F *h1 = (TH2F*)h->((TH2F*)h->GetCorrelationpT())->Clone();
  TH2F *h2 = (TH2F*)h->((TH2F*)h->GetCorrelationEta())->Clone();
  TH2F *h3 = (TH2F*)h->((TH2F*)h->GetCorrelationPhi())->Clone();
  h1->Write();
  h2->Write();
  h3->Write();
  fout->Close();

}


void DrawSameMixed(const char* fileName,Double_t ptTmin,Double_t ptTmax,Double_t ptAmin,Double_t ptAmax,Int_t centBegin,Int_t centEnd,Int_t step = 8,TCanvas *c)
{
  c->Divide(3, 3);
  TPaveText* paveText = new TPaveText(0.2, 0.9, 1., 1., "BRNDC");
  paveText->SetTextSize(0.04);
  paveText->SetFillColor(0);
  paveText->SetShadowColor(0);
  paveText->SetBorderSize(0);
  paveText->SetFillStyle(0);
  paveText->AddText(Form("%.1f<p_{T,trig}<%.1f, %.1f<p_{T,ass}<%.1f, %d-%d %%",ptTmin,ptTmax,ptAmin,ptAmax,centBegin,centEnd));
  
  loadlibs();
  
  AliUEHistograms* h = (AliUEHistograms*) GetUEHistogram(fileName);
  hMixed = (AliUEHistograms*) GetUEHistogram(fileName, 0, kTRUE);
  
  gpTMin=ptAmin;
  gpTMax=ptAmax;
  
  SetupRanges(h);
  SetupRanges(hMixed);
  
  TH1* hist1 = 0;
  TH1* hist2 = 0;
  GetDistAndFlow(h, &hist1, step, centBegin,  centEnd, ptTmin, ptTmax); 
  
  NormalizeToBinWidth(hist1);
  
  c->cd(1);
  gPad->SetLeftMargin(0.15);
  hist1->SetTitle("");
  hist1->GetYaxis()->SetRangeUser(-1.79, 1.79);
  hist1->GetZaxis()->SetTitleOffset(1.8);
  hist1->GetXaxis()->SetTitleOffset(1.5);
  hist1->GetYaxis()->SetTitleOffset(2);
  hist1->GetZaxis()->SetTitle("same event pairs (a.u.)");
  hist1->SetStats(kFALSE);
  hist1->DrawCopy("SURF1");
  paveText->Draw();
  c->cd(4);
  ((TH2*) hist1)->ProjectionX()->DrawCopy();
  c->cd(7);
  ((TH2*) hist1)->ProjectionY()->DrawCopy();
  
  hist2 = hist1;
  
  GetDistAndFlow(hMixed, &hist1, step, centBegin,  centEnd, ptTmin, ptTmax); 
  NormalizeToBinWidth(hist1);
  
  c->cd(2);
  gPad->SetLeftMargin(0.15);
  hist1->SetTitle("");
  hist1->GetYaxis()->SetRangeUser(-1.79, 1.79);
  hist1->GetZaxis()->SetTitleOffset(1.8);
  hist1->GetXaxis()->SetTitleOffset(1.5);
  hist1->GetYaxis()->SetTitleOffset(2);
  hist1->GetZaxis()->SetTitle("mixed event pairs (a.u.)");
  hist1->SetStats(kFALSE);
  hist1->DrawCopy("SURF1");
  c->cd(5);
  ((TH2*) hist1)->ProjectionX()->DrawCopy();
  c->cd(8);
  ((TH2*) hist1)->ProjectionY()->DrawCopy();
  
  c->cd(3);
  gPad->SetLeftMargin(0.15);
  hist2->GetZaxis()->SetTitle("same/mixed event pairs (a.u.)");
  hist2->Divide(hist1);
  hist2->DrawCopy("SURF1");
  c->cd(6);
  ((TH2*) hist2)->ProjectionX()->DrawCopy();
  c->cd(9);
  ((TH2*) hist2)->ProjectionY()->DrawCopy();
  
}


void DrawSameMixedSV(const char* fileName,Double_t ptTmin,Double_t ptTmax,Double_t ptAmin,Double_t ptAmax,Int_t centBegin,Int_t centEnd,Int_t step = 8,TCanvas *c, Int_t key, const char *outfile)
{
  c->Divide(3, 3);
  TPaveText* paveText = new TPaveText(0.2, 0.9, 1., 1., "BRNDC");
  paveText->SetTextSize(0.04);
  paveText->SetFillColor(0);
  paveText->SetShadowColor(0);
  paveText->SetBorderSize(0);
  paveText->SetFillStyle(0);
  paveText->AddText(Form("%.1f<p_{T,trig}<%.1f, %.1f<p_{T,ass}<%.1f, %d-%d %%",ptTmin,ptTmax,ptAmin,ptAmax,centBegin,centEnd));
  
  loadlibs();
  
  AliUEHistograms* h = (AliUEHistograms*) GetUEHistogram(fileName);
  hMixed = (AliUEHistograms*) GetUEHistogram(fileName, 0, kTRUE);
  
  gpTMin=ptAmin;
  gpTMax=ptAmax;
  
  SetupRanges(h);
  SetupRanges(hMixed);
  
  TH1* hist1 = 0;
  TH1* hist2 = 0;
  GetDistAndFlow(h, &hist1, step, centBegin,  centEnd, ptTmin, ptTmax); 
  
  NormalizeToBinWidth(hist1);
  
  c->cd(1);
  gPad->SetLeftMargin(0.15);
  hist1->SetTitle("");
  hist1->GetYaxis()->SetRangeUser(-1.79, 1.79);
  hist1->GetZaxis()->SetTitleOffset(1.8);
  hist1->GetXaxis()->SetTitleOffset(1.5);
  hist1->GetYaxis()->SetTitleOffset(2);
  hist1->GetZaxis()->SetTitle("same event pairs (a.u.)");
  hist1->SetStats(kFALSE);
  hist1->DrawCopy("SURF1");
  
  //added by sjena
  TFile *fout = TFile::Open(outfile,"UPDATE");
  fout->ls();
  
  TDirectoryFile *cdd = NULL;
  cdd = (TDirectoryFile*)fout->Get("CF");
  if(!cdd) {
    Printf("Warning: CF <dir> doesn't exist, creating a new one");
    cdd = (TDirectoryFile*)fout->mkdir("CF");
  }
  cdd->cd();
  cdd->ls();
  

  TH1D *h1 = (TH1D*)hist1->Clone();
  h1->SetName(Form("fig_cf_cr1_%d",key));
  h1->SetTitle(hist1->GetName());
  h1->Write();
  

  paveText->Draw();
  c->cd(4);
  ((TH2*) hist1)->ProjectionX()->DrawCopy();
 
  TH2D *h2 =  (TH2D*)((TH2*) hist1)->ProjectionX();
  h2->SetTitle(h2->GetName());
  h2->SetName(Form("fig_cf_cr2_%d", key));
  h2->Write();

  c->cd(7);
  ((TH2*) hist1)->ProjectionY()->DrawCopy();
  //ofile->cd();
  TH2D *h3 =  (TH2D*)((TH2*) hist1)->ProjectionY();
  h3->SetTitle(h3->GetName());
  h3->SetName(Form("fig_cf_cr3_%d",key));
  h3->Write();


  hist2 = hist1;
  
  GetDistAndFlow(hMixed, &hist1, step, centBegin,  centEnd, ptTmin, ptTmax); 
  NormalizeToBinWidth(hist1);
  
  c->cd(2);
  gPad->SetLeftMargin(0.15);
  hist1->SetTitle("");
  hist1->GetYaxis()->SetRangeUser(-1.79, 1.79);
  hist1->GetZaxis()->SetTitleOffset(1.8);
  hist1->GetXaxis()->SetTitleOffset(1.5);
  hist1->GetYaxis()->SetTitleOffset(2);
  hist1->GetZaxis()->SetTitle("mixed event pairs (a.u.)");
  hist1->SetStats(kFALSE);
  hist1->DrawCopy("SURF1");
  //ofile->cd();
   TH1D *h4 =  (TH1D*)hist1->Clone();
  h4->SetTitle(hist1->GetName());
  h4->SetName(Form("fig_cf_cr4_%d",key));
  h4->Write();


  c->cd(5);
  ((TH2*) hist1)->ProjectionX()->DrawCopy();
  //ofile->cd();
  TH2D *h5 =  (TH2D*)((TH2*) hist1)->ProjectionX();
  h5->SetTitle(h5->GetName());
  h5->SetName(Form("fig_cf_cr5_%d",key));
  h5->Write();


  c->cd(8);
  ((TH2*) hist1)->ProjectionY()->DrawCopy();
  //ofile->cd();
  TH2D *h6 =  (TH2D*)((TH2*) hist1)->ProjectionY();
  h6->SetTitle(h6->GetName());
   h6->SetName(Form("fig_cf_cr6_%d",key));
   h6->Write();  


  c->cd(3);
  gPad->SetLeftMargin(0.15);
  hist2->GetZaxis()->SetTitle("same/mixed event pairs (a.u.)");
  hist2->Divide(hist1);
  hist2->DrawCopy("SURF1");
  //ofile->cd();
 
  TH1D *h7 =  (TH1D*)hist2->Clone();
  h7->SetTitle(hist2->GetName());
  h7->SetName(Form("fig_cf_cr7_%d",key));
  h7->Write();


  c->cd(6);
  ((TH2*) hist2)->ProjectionX()->DrawCopy();
  //ofile->cd();
  TH2D *h8 =  (TH2D*)((TH2*) hist2)->ProjectionX();
  h8->SetTitle(h8->GetName());
  h8->SetName(Form("fig_cf_cr8_%d",key));
  h8->Write();  
  

  c->cd(9);
  ((TH2*) hist2)->ProjectionY()->DrawCopy();
  //ofile->cd();
  TH2D *h9 =  (TH2D*)((TH2*) hist2)->ProjectionY();
  h9->SetTitle(h9->GetName());
  h9->SetName(Form("fig_cf_cr9_%d",key));
  h9->Write();  

  fout->Close();
}


void loadlibs()
{
  gSystem->Load("libANALYSIS");
  gSystem->Load("libANALYSISalice");
  gSystem->Load("libCORRFW");
  gSystem->Load("libPWGTools");
  gSystem->Load("libPWGCFCorrelationsBase");
}

void FillParentTHnSparse(const char* fileName, Bool_t reduce = kFALSE, const char* tag = "")
{
  if (TString(fileName).BeginsWith("alien:"))
    TGrid::Connect("alien:");
  
  loadlibs();

  TList* list = 0;
  
  AliUEHistograms* h = (AliUEHistograms*) GetUEHistogram(fileName, &list, kFALSE, tag);

  
  Printf("We have %d axes", ((AliTHn*) h->GetUEHist(2)->GetTrackHist(0)->GetNVar()));
  
  if (reduce)
    ((AliTHn*) h->GetUEHist(2)->GetTrackHist(0))->ReduceAxis();
  ((AliTHn*) h->GetUEHist(2)->GetTrackHist(0))->FillParent();
  ((AliTHn*) h->GetUEHist(2)->GetTrackHist(0))->DeleteContainers();
  
  AliUEHistograms* hMixed = (AliUEHistograms*) GetUEHistogram(fileName, 0, kTRUE, tag);
  if (reduce)
    ((AliTHn*) hMixed->GetUEHist(2)->GetTrackHist(0))->ReduceAxis();
  ((AliTHn*) hMixed->GetUEHist(2)->GetTrackHist(0))->FillParent();
  ((AliTHn*) hMixed->GetUEHist(2)->GetTrackHist(0))->DeleteContainers();
  
  TString newFileName(fileName);

  if (TString(fileName).BeginsWith("alien:"))
    newFileName = gSystem->BaseName(newFileName);
  
  newFileName.ReplaceAll(".root", "");
  if (reduce)
    newFileName += "_.root";
  else
    newFileName += "_zvtx.root";

  file3 = TFile::Open(newFileName, "RECREATE");
  file3->mkdir("PWG4_PhiCorrelations");
  file3->cd("PWG4_PhiCorrelations");
  list->Write("histosPhiCorrelations", TObject::kSingleKey);
  file3->Close();
}

void* GetUEHistogram(const char* fileName, TList** listRef = 0, Bool_t mixed = kFALSE, const char* tag = "")
{
  if (!lastFileName || strcmp(lastFileName, fileName) != 0)
    {
      lastFileName = fileName;
      file = TFile::Open(fileName);
      if (!file)
	return 0;
      
      list = (TList*) gFile->Get("PWG4_LeadingTrackUE/histosLeadingTrackUE");
      if (!list)
	list = (TList*) gFile->Get(Form("PWG4_PhiCorrelations/histosPhiCorrelations%s", tag));
      if (!list)
	list = (TList*) gFile->Get("PWG4_PhiCorrelations/histosPhiCorrelations_Syst");
      
      if (!list)
	return 0;
      
      if (listRef)
	*listRef = list;
      
      cacheMixedEvent = list->FindObject("AliUEHistogramsMixed");
      cacheSameEvent = list->FindObject("AliUEHistogramsSame");

      if (mixed)
	return cacheMixedEvent;
    
      if (list->FindObject("AliUEHistograms"))
	return list->FindObject("AliUEHistograms");
      
      return cacheSameEvent;
    }
  else
    {
      Printf("GetUEHistogram --> Using cache for %s", fileName);
    
      if (mixed)
	return cacheMixedEvent;
      else
	return cacheSameEvent;
    }
}


Int_t gHistCount = 0;

void GetDistAndFlow(void* hVoid, TH1** hist, Int_t step, Int_t centralityBegin, Int_t centralityEnd, Float_t ptBegin, Float_t ptEnd)
{
  h = (AliUEHistograms*) hVoid;
  
  Int_t centralityBeginBin = 0;
  Int_t centralityEndBin = -1;
  
  if (centralityEnd >= centralityBegin)
    {
      centralityBeginBin = h->GetUEHist(2)->GetEventHist()->GetGrid(step)->GetGrid()->GetAxis(1)->FindBin(0.01 + centralityBegin);
      centralityEndBin = h->GetUEHist(2)->GetEventHist()->GetGrid(step)->GetGrid()->GetAxis(1)->FindBin(-0.01 + centralityEnd);
    }
  
  // 2d same and mixed event
  TH2* sameTwoD  = h->GetUEHist(2)->GetUEHist(step, 0, ptBegin, ptEnd, centralityBeginBin, centralityEndBin, 1, kFALSE);
  
  TString histName;
  histName.Form("GetDistAndFlow%d", gHistCount++);
  
  *hist = sameTwoD;
    
  TString str;
  str.Form("%.1f < p_{T,trig} < %.1f", ptBegin - 0.01, ptEnd + 0.01);
  
  TString str2;
  str2.Form("%.2f < p_{T,assoc} < %.2f", gpTMin - 0.01, gpTMax + 0.01);
  
  TString newTitle;
  newTitle.Form("%s - %s - %d-%d%%", str.Data(), str2.Data(), centralityBegin, centralityEnd);
  (*hist)->SetTitle(newTitle);
  
}

void NormalizeToBinWidth(TH1* hist)
{
  //
  // normalizes a 1-d histogram to its bin width
  //

  if (hist->GetDimension() == 1)
    {
      for (Int_t i=1; i<=hist->GetNbinsX(); ++i)
	{
	  hist->SetBinContent(i, hist->GetBinContent(i) / hist->GetBinWidth(i));
	  hist->SetBinError(i, hist->GetBinError(i) / hist->GetBinWidth(i));
	}
    }
  else if (hist->GetDimension() == 2)
    {
      for (Int_t i=1; i<=hist->GetNbinsX(); ++i)
	{
	  for (Int_t j=1; j<=hist->GetNbinsY(); ++j)
	    {
	      hist->SetBinContent(i, j, hist->GetBinContent(i, j) / hist->GetXaxis()->GetBinWidth(i) / hist->GetYaxis()->GetBinWidth(j));
	      hist->SetBinError(i, j, hist->GetBinError(i) / hist->GetXaxis()->GetBinWidth(i) / hist->GetYaxis()->GetBinWidth(j));
	    }
	}
    }
}
Commit	Line	Data
948b38b2	1	const char * tag = "_Hadrons_QA_PbPb";
	2	Float_t gpTMin = 0.51;
	3	Float_t gpTMax = 49.99;
	4	const char* lastFileName = 0;
	5	void* cacheSameEvent = 0;
	6	void* cacheMixedEvent = 0;
	7
	8	void processMakeQA2pc(const char * filename="AnalysisResults", //file name without ".root"
	9	TString suffix = "eps",
	10	const char *outfile="MakeQA2pc_output.root"){
	11
	12	loadlibs();
	13	if (!gGrid && TString(filename).BeginsWith("alien://"))
	14	TGrid::Connect("alien://");
	15
	16	//fill CFContainers
	17	FillParentTHnSparse(Form("%s.root",filename),kFALSE,tag);
	18
	19	Double_t ptTmin[]={0.5,0.5,2.0,2.0};
	20	Double_t ptTmax[]={1.0,1.0,4.0,4.0};
	21	Double_t ptAmin[]={0.5,0.5,2.0,2.0};
	22	Double_t ptAmax[]={1.0,1.0,4.0,4.0};
	23	Int_t centBegin[]={ 0, 60, 0, 60};
	24	Int_t centEnd[]={ 10, 90, 10, 90};
	25
	26
	27
	28	for(Int_t i=0;i<4;i++){
	29	TCanvas *c=new TCanvas(Form("c%d",i),Form("c%d",i),1200,1200);
	30	DrawSameMixedSV(Form("%s_zvtx.root",filename),ptTmin[i],ptTmax[i],ptAmin[i],ptAmax[i],centBegin[i],centEnd[i],8,c,i,outfile);
	31	c->SaveAs(Form("fig_cf_c%d.%s",i,suffix.Data()));
	32	}
	33
	34	TCanvas *ccorr=new TCanvas("ccorr","ccorr",1200,1200);
	35	DrawCorrelation(Form("%s_zvtx.root",filename),ccorr,outfile);
	36
	37
	38	ccorr->SaveAs(Form("fig_cf_ccorr.%s",suffix.Data()));
	39
	40	}
	41
	42	void SetupRanges(void* obj)
	43	{
	44	if (!obj)
	45	return;
	46	((AliUEHistograms*) obj)->SetEtaRange(0, 0);
	47	((AliUEHistograms*) obj)->SetPtRange(gpTMin, gpTMax);
	48	((AliUEHistograms*) obj)->SetCombineMinMax(kTRUE);
	49	}
	50
	51	void DrawCorrelation(const char* fileName,TCanvas c,const char outfile){
	52
	53	AliUEHistograms* h = (AliUEHistograms*) GetUEHistogram(fileName);
	54	c->Divide(1,3);
	55	c->cd(1);
	56	gPad->SetLogz();
	57	h->((TH2F*)h->GetCorrelationpT())->DrawCopy("colz")->GetYaxis()->SetRangeUser(0.,20.);
	58	c->cd(2);
	59	h->((TH2F*)h->GetCorrelationEta())->DrawCopy("colz")->GetYaxis()->SetRangeUser(-1.5,1.5);
	60	c->cd(3);
	61	h->((TH2F*)h->GetCorrelationPhi())->DrawCopy("colz");
	62
	63	//Added by sjena
	64	TFile *fout = TFile::Open(outfile,"UPDATE");
65	fout->ls();
66
67	TDirectoryFile *cdd = NULL;
68	cdd = (TDirectoryFile*)fout->Get("CF");
69	if(!cdd) {
70	Printf("Warning: CF <dir> doesn't exist, creating a new one");
71	cdd = (TDirectoryFile*)fout->mkdir("CF");
72	}
73	cdd->cd();
74	cdd->ls();
75
76
77	TH2F h1 = (TH2F)h->((TH2F*)h->GetCorrelationpT())->Clone();
78	TH2F h2 = (TH2F)h->((TH2F*)h->GetCorrelationEta())->Clone();
79	TH2F h3 = (TH2F)h->((TH2F*)h->GetCorrelationPhi())->Clone();
80	h1->Write();
81	h2->Write();
82	h3->Write();
83	fout->Close();
84
85	}
86
87
88	void DrawSameMixed(const char* fileName,Double_t ptTmin,Double_t ptTmax,Double_t ptAmin,Double_t ptAmax,Int_t centBegin,Int_t centEnd,Int_t step = 8,TCanvas *c)
89	{
90	c->Divide(3, 3);
91	TPaveText* paveText = new TPaveText(0.2, 0.9, 1., 1., "BRNDC");
92	paveText->SetTextSize(0.04);
93	paveText->SetFillColor(0);
94	paveText->SetShadowColor(0);
95	paveText->SetBorderSize(0);
96	paveText->SetFillStyle(0);
97	paveText->AddText(Form("%.1f<p_{T,trig}<%.1f, %.1f<p_{T,ass}<%.1f, %d-%d %%",ptTmin,ptTmax,ptAmin,ptAmax,centBegin,centEnd));
98
99	loadlibs();
100
101	AliUEHistograms* h = (AliUEHistograms*) GetUEHistogram(fileName);
102	hMixed = (AliUEHistograms*) GetUEHistogram(fileName, 0, kTRUE);
103
104	gpTMin=ptAmin;
105	gpTMax=ptAmax;
106
107	SetupRanges(h);
108	SetupRanges(hMixed);
109
110	TH1* hist1 = 0;
111	TH1* hist2 = 0;
112	GetDistAndFlow(h, &hist1, step, centBegin, centEnd, ptTmin, ptTmax);
113
114	NormalizeToBinWidth(hist1);
115
116	c->cd(1);
117	gPad->SetLeftMargin(0.15);
118	hist1->SetTitle("");
119	hist1->GetYaxis()->SetRangeUser(-1.79, 1.79);
120	hist1->GetZaxis()->SetTitleOffset(1.8);
121	hist1->GetXaxis()->SetTitleOffset(1.5);
122	hist1->GetYaxis()->SetTitleOffset(2);
123	hist1->GetZaxis()->SetTitle("same event pairs (a.u.)");
124	hist1->SetStats(kFALSE);
125	hist1->DrawCopy("SURF1");
126	paveText->Draw();
127	c->cd(4);
128	((TH2*) hist1)->ProjectionX()->DrawCopy();
129	c->cd(7);
130	((TH2*) hist1)->ProjectionY()->DrawCopy();
131
132	hist2 = hist1;
133
134	GetDistAndFlow(hMixed, &hist1, step, centBegin, centEnd, ptTmin, ptTmax);
135	NormalizeToBinWidth(hist1);
136
137	c->cd(2);
138	gPad->SetLeftMargin(0.15);
139	hist1->SetTitle("");
140	hist1->GetYaxis()->SetRangeUser(-1.79, 1.79);
141	hist1->GetZaxis()->SetTitleOffset(1.8);
142	hist1->GetXaxis()->SetTitleOffset(1.5);
143	hist1->GetYaxis()->SetTitleOffset(2);
144	hist1->GetZaxis()->SetTitle("mixed event pairs (a.u.)");
145	hist1->SetStats(kFALSE);
146	hist1->DrawCopy("SURF1");
147	c->cd(5);
148	((TH2*) hist1)->ProjectionX()->DrawCopy();
149	c->cd(8);
150	((TH2*) hist1)->ProjectionY()->DrawCopy();
151
152	c->cd(3);
153	gPad->SetLeftMargin(0.15);
154	hist2->GetZaxis()->SetTitle("same/mixed event pairs (a.u.)");
155	hist2->Divide(hist1);
156	hist2->DrawCopy("SURF1");
157	c->cd(6);
158	((TH2*) hist2)->ProjectionX()->DrawCopy();
159	c->cd(9);
160	((TH2*) hist2)->ProjectionY()->DrawCopy();
161
162	}
163
164
165	void DrawSameMixedSV(const char* fileName,Double_t ptTmin,Double_t ptTmax,Double_t ptAmin,Double_t ptAmax,Int_t centBegin,Int_t centEnd,Int_t step = 8,TCanvas c, Int_t key, const char outfile)
166	{
167	c->Divide(3, 3);
168	TPaveText* paveText = new TPaveText(0.2, 0.9, 1., 1., "BRNDC");
169	paveText->SetTextSize(0.04);
170	paveText->SetFillColor(0);
171	paveText->SetShadowColor(0);
172	paveText->SetBorderSize(0);
173	paveText->SetFillStyle(0);
174	paveText->AddText(Form("%.1f<p_{T,trig}<%.1f, %.1f<p_{T,ass}<%.1f, %d-%d %%",ptTmin,ptTmax,ptAmin,ptAmax,centBegin,centEnd));
175
176	loadlibs();
177
178	AliUEHistograms* h = (AliUEHistograms*) GetUEHistogram(fileName);
179	hMixed = (AliUEHistograms*) GetUEHistogram(fileName, 0, kTRUE);
180
181	gpTMin=ptAmin;
182	gpTMax=ptAmax;
183
184	SetupRanges(h);
185	SetupRanges(hMixed);
186
187	TH1* hist1 = 0;
188	TH1* hist2 = 0;
189	GetDistAndFlow(h, &hist1, step, centBegin, centEnd, ptTmin, ptTmax);
190
191	NormalizeToBinWidth(hist1);
192
193	c->cd(1);
194	gPad->SetLeftMargin(0.15);
195	hist1->SetTitle("");
196	hist1->GetYaxis()->SetRangeUser(-1.79, 1.79);
197	hist1->GetZaxis()->SetTitleOffset(1.8);
198	hist1->GetXaxis()->SetTitleOffset(1.5);
199	hist1->GetYaxis()->SetTitleOffset(2);
200	hist1->GetZaxis()->SetTitle("same event pairs (a.u.)");
201	hist1->SetStats(kFALSE);
202	hist1->DrawCopy("SURF1");
203
204	//added by sjena
205	TFile *fout = TFile::Open(outfile,"UPDATE");
206	fout->ls();
207
208	TDirectoryFile *cdd = NULL;
209	cdd = (TDirectoryFile*)fout->Get("CF");
210	if(!cdd) {
211	Printf("Warning: CF <dir> doesn't exist, creating a new one");
212	cdd = (TDirectoryFile*)fout->mkdir("CF");
213	}
214	cdd->cd();
215	cdd->ls();
216
217
218	TH1D h1 = (TH1D)hist1->Clone();
219	h1->SetName(Form("fig_cf_cr1_%d",key));
220	h1->SetTitle(hist1->GetName());
221	h1->Write();
222
223
224	paveText->Draw();
225	c->cd(4);
226	((TH2*) hist1)->ProjectionX()->DrawCopy();
227
228	TH2D h2 = (TH2D)((TH2*) hist1)->ProjectionX();
229	h2->SetTitle(h2->GetName());
230	h2->SetName(Form("fig_cf_cr2_%d", key));
231	h2->Write();
232
233	c->cd(7);
234	((TH2*) hist1)->ProjectionY()->DrawCopy();
235	//ofile->cd();
236	TH2D h3 = (TH2D)((TH2*) hist1)->ProjectionY();
237	h3->SetTitle(h3->GetName());
238	h3->SetName(Form("fig_cf_cr3_%d",key));
239	h3->Write();
240
241
242
243
244	hist2 = hist1;
245
246	GetDistAndFlow(hMixed, &hist1, step, centBegin, centEnd, ptTmin, ptTmax);
247	NormalizeToBinWidth(hist1);
248
249	c->cd(2);
250	gPad->SetLeftMargin(0.15);
251	hist1->SetTitle("");
252	hist1->GetYaxis()->SetRangeUser(-1.79, 1.79);
253	hist1->GetZaxis()->SetTitleOffset(1.8);
254	hist1->GetXaxis()->SetTitleOffset(1.5);
255	hist1->GetYaxis()->SetTitleOffset(2);
256	hist1->GetZaxis()->SetTitle("mixed event pairs (a.u.)");
257	hist1->SetStats(kFALSE);
258	hist1->DrawCopy("SURF1");
259	//ofile->cd();
260	TH1D h4 = (TH1D)hist1->Clone();
261	h4->SetTitle(hist1->GetName());
262	h4->SetName(Form("fig_cf_cr4_%d",key));
263	h4->Write();
264
265
266
267
268
269	c->cd(5);
270	((TH2*) hist1)->ProjectionX()->DrawCopy();
271	//ofile->cd();
272	TH2D h5 = (TH2D)((TH2*) hist1)->ProjectionX();
273	h5->SetTitle(h5->GetName());
274	h5->SetName(Form("fig_cf_cr5_%d",key));
275	h5->Write();
276
277
278
279
280	c->cd(8);
281	((TH2*) hist1)->ProjectionY()->DrawCopy();
282	//ofile->cd();
283	TH2D h6 = (TH2D)((TH2*) hist1)->ProjectionY();
284	h6->SetTitle(h6->GetName());
285	h6->SetName(Form("fig_cf_cr6_%d",key));
286	h6->Write();
287
288
289	c->cd(3);
290	gPad->SetLeftMargin(0.15);
291	hist2->GetZaxis()->SetTitle("same/mixed event pairs (a.u.)");
292	hist2->Divide(hist1);
293	hist2->DrawCopy("SURF1");
294	//ofile->cd();
295
296	TH1D h7 = (TH1D)hist2->Clone();
297	h7->SetTitle(hist2->GetName());
298	h7->SetName(Form("fig_cf_cr7_%d",key));
299	h7->Write();
300
301
302	c->cd(6);
303	((TH2*) hist2)->ProjectionX()->DrawCopy();
304	//ofile->cd();
305	TH2D h8 = (TH2D)((TH2*) hist2)->ProjectionX();
306	h8->SetTitle(h8->GetName());
307	h8->SetName(Form("fig_cf_cr8_%d",key));
308	h8->Write();
309
310
311	c->cd(9);
312	((TH2*) hist2)->ProjectionY()->DrawCopy();
313	//ofile->cd();
314	TH2D h9 = (TH2D)((TH2*) hist2)->ProjectionY();
315	h9->SetTitle(h9->GetName());
316	h9->SetName(Form("fig_cf_cr9_%d",key));
317	h9->Write();
318
319	fout->Close();
320	}
321
322
323	void loadlibs()
324	{
325	gSystem->Load("libANALYSIS");
326	gSystem->Load("libANALYSISalice");
327	gSystem->Load("libCORRFW");
328	gSystem->Load("libPWGTools");
329	gSystem->Load("libPWGCFCorrelationsBase");
330	}
331
332	void FillParentTHnSparse(const char* fileName, Bool_t reduce = kFALSE, const char* tag = "")
333	{
334	if (TString(fileName).BeginsWith("alien:"))
335	TGrid::Connect("alien:");
336
337	loadlibs();
338
339	TList* list = 0;
340
341	AliUEHistograms* h = (AliUEHistograms*) GetUEHistogram(fileName, &list, kFALSE, tag);
342
343
344	Printf("We have %d axes", ((AliTHn*) h->GetUEHist(2)->GetTrackHist(0)->GetNVar()));
345
346	if (reduce)
347	((AliTHn*) h->GetUEHist(2)->GetTrackHist(0))->ReduceAxis();
348	((AliTHn*) h->GetUEHist(2)->GetTrackHist(0))->FillParent();
349	((AliTHn*) h->GetUEHist(2)->GetTrackHist(0))->DeleteContainers();
350
351	AliUEHistograms* hMixed = (AliUEHistograms*) GetUEHistogram(fileName, 0, kTRUE, tag);
352	if (reduce)
353	((AliTHn*) hMixed->GetUEHist(2)->GetTrackHist(0))->ReduceAxis();
354	((AliTHn*) hMixed->GetUEHist(2)->GetTrackHist(0))->FillParent();
355	((AliTHn*) hMixed->GetUEHist(2)->GetTrackHist(0))->DeleteContainers();
356
357	TString newFileName(fileName);
358
359	if (TString(fileName).BeginsWith("alien:"))
360	newFileName = gSystem->BaseName(newFileName);
361
362	newFileName.ReplaceAll(".root", "");
363	if (reduce)
364	newFileName += "_.root";
365	else
366	newFileName += "_zvtx.root";
367
368	file3 = TFile::Open(newFileName, "RECREATE");
369	file3->mkdir("PWG4_PhiCorrelations");
370	file3->cd("PWG4_PhiCorrelations");
371	list->Write("histosPhiCorrelations", TObject::kSingleKey);
372	file3->Close();
373	}
374
375	void* GetUEHistogram(const char* fileName, TList** listRef = 0, Bool_t mixed = kFALSE, const char* tag = "")
376	{
377	if (!lastFileName \|\| strcmp(lastFileName, fileName) != 0)
378	{
379	lastFileName = fileName;
380	file = TFile::Open(fileName);
381	if (!file)
382	return 0;
383
384	list = (TList*) gFile->Get("PWG4_LeadingTrackUE/histosLeadingTrackUE");
385	if (!list)
386	list = (TList*) gFile->Get(Form("PWG4_PhiCorrelations/histosPhiCorrelations%s", tag));
387	if (!list)
388	list = (TList*) gFile->Get("PWG4_PhiCorrelations/histosPhiCorrelations_Syst");
389
390	if (!list)
391	return 0;
392
393	if (listRef)
394	*listRef = list;
395
396	cacheMixedEvent = list->FindObject("AliUEHistogramsMixed");
397	cacheSameEvent = list->FindObject("AliUEHistogramsSame");
398
399	if (mixed)
400	return cacheMixedEvent;
401
402	if (list->FindObject("AliUEHistograms"))
403	return list->FindObject("AliUEHistograms");
404
405	return cacheSameEvent;
406	}
407	else
408	{
409	Printf("GetUEHistogram --> Using cache for %s", fileName);
410
411	if (mixed)
412	return cacheMixedEvent;
413	else
414	return cacheSameEvent;
415	}
416	}
417
418
419	Int_t gHistCount = 0;
420
421	void GetDistAndFlow(void* hVoid, TH1** hist, Int_t step, Int_t centralityBegin, Int_t centralityEnd, Float_t ptBegin, Float_t ptEnd)
422	{
423	h = (AliUEHistograms*) hVoid;
424
425	Int_t centralityBeginBin = 0;
426	Int_t centralityEndBin = -1;
427
428	if (centralityEnd >= centralityBegin)
429	{
430	centralityBeginBin = h->GetUEHist(2)->GetEventHist()->GetGrid(step)->GetGrid()->GetAxis(1)->FindBin(0.01 + centralityBegin);
431	centralityEndBin = h->GetUEHist(2)->GetEventHist()->GetGrid(step)->GetGrid()->GetAxis(1)->FindBin(-0.01 + centralityEnd);
432	}
433
434	// 2d same and mixed event
435	TH2* sameTwoD = h->GetUEHist(2)->GetUEHist(step, 0, ptBegin, ptEnd, centralityBeginBin, centralityEndBin, 1, kFALSE);
436
437	TString histName;
438	histName.Form("GetDistAndFlow%d", gHistCount++);
439
440	*hist = sameTwoD;
441
442	TString str;
443	str.Form("%.1f < p_{T,trig} < %.1f", ptBegin - 0.01, ptEnd + 0.01);
444
445	TString str2;
446	str2.Form("%.2f < p_{T,assoc} < %.2f", gpTMin - 0.01, gpTMax + 0.01);
447
448	TString newTitle;
449	newTitle.Form("%s - %s - %d-%d%%", str.Data(), str2.Data(), centralityBegin, centralityEnd);
450	(*hist)->SetTitle(newTitle);
451
452	}
453
454	void NormalizeToBinWidth(TH1* hist)
455	{
456	//
457	// normalizes a 1-d histogram to its bin width
458	//
459
460	if (hist->GetDimension() == 1)
461	{
462	for (Int_t i=1; i<=hist->GetNbinsX(); ++i)
463	{
464	hist->SetBinContent(i, hist->GetBinContent(i) / hist->GetBinWidth(i));
465	hist->SetBinError(i, hist->GetBinError(i) / hist->GetBinWidth(i));
466	}
467	}
468	else if (hist->GetDimension() == 2)
469	{
470	for (Int_t i=1; i<=hist->GetNbinsX(); ++i)
471	{
472	for (Int_t j=1; j<=hist->GetNbinsY(); ++j)
473	{
474	hist->SetBinContent(i, j, hist->GetBinContent(i, j) / hist->GetXaxis()->GetBinWidth(i) / hist->GetYaxis()->GetBinWidth(j));
475	hist->SetBinError(i, j, hist->GetBinError(i) / hist->GetXaxis()->GetBinWidth(i) / hist->GetYaxis()->GetBinWidth(j));
476	}
477	}
478	}
479	}