TH1* GetOne(UShort_t sNN, const TString& trigger)
{
Long_t p = gROOT->ProcessLine(Form("Drawer::GetStack(0, \"pp\", %d, "
"\"%s\", false, true)",
sNN, trigger.Data()));
THStack* s = (THStack*)p;
TList* l = s->GetHists();
TH1* h = 0;
TIter n(l);
l->ls();
while ((h = static_cast<TH1*>(n()))) {
TString m(h->GetName());
if (m.EqualTo("dndetaForward_all")) break;
}
if (h) {
switch (sNN) {
case 900: h->SetTitle("900GeV"); h->SetMarkerColor(kRed+2); break;
case 2760: h->SetTitle("2.76TeV"); h->SetMarkerColor(kGreen+2); break;
case 7000: h->SetTitle("7TeV"); h->SetMarkerColor(kBlue+2); break;
case 8000: h->SetTitle("8TeV"); h->SetMarkerColor(kBlack); break;
}
}
return h;
}
void QAcentrality(const Char_t *fdata)
{
style();
TFile *fin = TFile::Open(fdata);
TList *lin = (TList *)fin->Get("clist");
lin->ls();
TH1 *hin = (TH1 *)lin->FindObject("EvCentrDist");
Float_t sum = 1.2 * hin->Integral(hin->FindBin(0.1), hin->FindBin(79.9));
hin->Scale(1. / sum);
SetHistoStyle(hin, 20, kRed+1);
TCanvas *c = new TCanvas("cQAcentrality", "cQAcentrality", 800, 800);
TH1 * hfr = c->DrawFrame(0., 0.005, 100., 0.015);
hfr->SetTitle(";centrality percentile;events");
hin->Draw("same");
c->SaveAs(canvasPrefix+"centrality.pdf");
TH2 *hinv0 = (TH2 *)lin->FindObject("V0");
TCanvas *cv0 = new TCanvas("cQAcentralityV0", "cQAcentralityV0", 800, 800);
cv0->SetLogx();
cv0->SetLogz();
// TH1 * hfrv0 = cv0->DrawFrame(100., -0.5, 50000., 10.5);
// DrawBinLabelsY(hfrv0, kTRUE);
// hfrv0->SetTitle(";V0 signal;");
//hinv0->Draw("same,col");
hinv0->Draw("col");
cv0->SaveAs(canvasPrefix+"centralityV0.pdf");
}
TList *QtMultiFileDialog(const char *style="") {
// This is a small ROOT macro to use Qt 3.3 class :begin_html <a href="http://doc.trolltech.com/3.3/qfiledialog.html">QFileDialog</a> end_html
// See: begin_html <a href="http://doc.trolltech.com/3.3/qfiledialog.html#getOpenFileNames">http://doc.trolltech.com/3.3/qfiledialog.html#getOpenFileNames</a> end_html
//
// To use, invoke ACLiC from the ROOT prompt:
// root [] .x QtMultiFileDialog.C++
//
// To use it with no ACLiC, omit the trailing "++"
// root [] .x QtMultiFileDialog.C
//
// The QtMultiFileDialog creates TList of TObjString objects and
// returns its pointer.
//
// The "QtFileDialog.C" macro provides the simplified version of the "QtMultiFileDialog.C"
//
// Option: you can change the look and feel of the Qt file dialog
// ======= by providing the optional parameter "style":
// The number of the available styles is defined by your local
// Qt installation.
// Try: "windows", "motif", "kde", "platinum" etc
//
// The full list of the Qt classes available from Cint is defined by
// begin_html <a href="http://root.bnl.gov/QtRoot/htmldoc/src/qtclasses.h.html">by $ROOTSYS/cint/lib/qtclasses.h</a> end_html
//
// All Qt classes can be used from ACLiC though.
#ifdef __CINT__
// Load the qt cint dictionary.
// One is recommended to do that at once somewhere.
// For example from one's custom rootlogon.C script
gSystem->Load("$ROOTSYS/cint/cint/include/qtcint");
#endif
QStyle *saveStyle = 0;
if (!QString(style).isEmpty()) {
saveStyle = QApplication::style();
QApplication::setStyle(style);
}
TList *listOfNames = new TList();
QStringList files = QFileDialog::getOpenFileNames ();
QStringList::Iterator it = files.begin();
while ( it != files.end() ) {
std::string flnm = (*it).toStdString();
printf ("Next file selected: %s\n", flnm.c_str() );
// Convert QString to TObjString and add it to the output
listOfNames->Add(new TObjString(flnm.c_str()));
++it;
}
// Restore the style
if (saveStyle) QApplication::setStyle(saveStyle);
printf ("\nThe TList of the file names contains:");
printf ("\n-------------------------------------\n");
listOfNames->ls();
return listOfNames;
}
void TestTListIter(){
// This macros tests the various methods of TFileIter class.
gSystem->Load("libRootKernel");
TList testList;
TString member = "member";
int i = 0;
for (i = 0; i < 10; i++) {
// Fill list
TString nextM = member;
nextM += i;
testList.Add(new TNamed(nextM,"Title"));
}
testList.ls();
TLIST::iterator iter(&testList);
i = 0;
while (*iter != 0 ) {
printf(" Name = %s\n", (*iter)->GetName() );
iter++; i++;
}
if ( i == 10) printf(" Ok ! \n");
else printf(" Failed ! \n");
}
TList *QtMultiFileDialog(const char *style="") {
QStyle *saveStyle = 0;
if (!QString(style).isEmpty()) {
saveStyle = QApplication::style();
QApplication::setStyle(style);
}
TList *listOfNames = new TList();
QStringList files = QFileDialog::getOpenFileNames ();
QStringList::Iterator it = files.begin();
while ( it != files.end() ) {
std::string flnm = (*it).toStdString();
printf ("Next file selected: %s\n", flnm.c_str() );
// Convert QString to TObjString and add it to the output
listOfNames->Add(new TObjString(flnm.c_str()));
++it;
}
// Restore the style
if (saveStyle) QApplication::setStyle(saveStyle);
printf ("\nThe TList of the file names contains:");
printf ("\n-------------------------------------\n");
listOfNames->ls();
return listOfNames;
}
void QAtracklets(const Char_t *fdata, const Char_t *fmc)
{
style();
TFile *fdtin = TFile::Open(fdata);
TList *ldtin = (TList *)fdtin->Get("clist");
TH2 *hdtin = (TH2 *)ldtin->FindObject("etaphiTracklets");
TH1 *pdtin = (TH1 *)hdtin->ProjectionY("pdtin_tracklets");
pdtin->SetMarkerStyle(20);
pdtin->SetMarkerSize(1.);
pdtin->SetMarkerColor(kAzure-3);
hdtin->Scale(1. / hdtin->GetEntries());
pdtin->Scale(1. / hdtin->GetEntries());
TFile *fmcin = TFile::Open(fmc);
TList *lmcin = (TList *)fmcin->Get("clist");
if(!lmcin) {
std::cout << "NOLIST" << std::endl;
}
lmcin->ls();
TH2 *hmcin = (TH2 *)lmcin->FindObject("etaphiTracklets");
if(!hmcin) {
std::cout << "NO H!! etaphiTracklets" << std::endl;
}
TH1 *pmcin = (TH1 *)hmcin->ProjectionY("pmcin_tracklets");
pmcin->SetLineColor(kRed+1);
pmcin->SetFillStyle(1001);
pmcin->SetFillColorAlpha(kRed+1, 0.1);
hmcin->Scale(1. / hmcin->GetEntries());
pmcin->Scale(1. / hmcin->GetEntries());
/*
pdtin->Scale(pmcin->Integral(pmcin->FindBin(0. + 0.001),
pmcin->FindBin(1. - 0.001))
/ pdtin->Integral(pdtin->FindBin(0. + 0.001),
pdtin->FindBin(1. - 0.001)));
*/
TCanvas *cData = new TCanvas("cTrackletData", "cTrackletData", 800, 800);
// cData->SetLogz();
TH1 * hfr = cData->DrawFrame(-2.5, 0., 2.5, 2. * TMath::Pi());
hfr->SetTitle(";#eta;#varphi");
hdtin->Draw("same,col");
cData->SaveAs(canvasPrefix+"trackletData.pdf");
TCanvas *cMC = new TCanvas("cTrackletMC", "cTrackletMC", 800, 800);
// cMC->SetLogz();
hfr = cMC->DrawFrame(-2.5, 0., 2.5, 2. * TMath::Pi());
hfr->SetTitle(";#eta;#varphi");
hmcin->Draw("same,col");
cMC->SaveAs(canvasPrefix+"trackletMC.pdf");
TCanvas *cPhi = new TCanvas("cTrackletPhi", "cTrackletPhi", 800, 800);
// cPhi->SetLogy();
hfr = cPhi->DrawFrame(0., 0., 2. * TMath::Pi(), 0.01);
hfr->SetTitle(";#varphi;");
pdtin->DrawCopy("same");
pmcin->DrawCopy("same,histo");
TLegend *legend = new TLegend(0.20, 0.18+0.63, 0.50, 0.30+0.63);
legend->SetFillColor(0);
legend->SetBorderSize(0);
legend->SetTextFont(42);
legend->SetTextSize(0.04);
legend->AddEntry(pdtin, "data", "pl");
legend->AddEntry(pmcin, "Monte Carlo", "l");
legend->Draw("same");
cPhi->SaveAs(canvasPrefix+"trackletPhi.pdf");
TCanvas *cPhir = new TCanvas("cTrackletPhir", "cTrackletPhir", 800, 800);
// cPhi->SetLogy();
hfr = cPhir->DrawFrame(0., 0.5, 2. * TMath::Pi(), 1.5);
hfr->SetTitle(";#varphi;data / Monte Carlo");
pdtin->Divide(pmcin);
pdtin->Draw("same");
cPhir->SaveAs(canvasPrefix+"trackletPhir.pdf");
}
TProfile *extractFlowVZEROsingle(Int_t icentr,Int_t spec,Int_t arm,Bool_t isMC,Float_t pTh,Int_t addbin,const char *nameSp,Float_t detMin,Float_t detMax,Int_t chMin,Int_t chMax){
LoadLib();
pTh += 0.00001;
// NUA correction currently are missing
char name[100];
char stringa[200];
snprintf(name,100,"AnalysisResults.root");
if(!fo) fo = new TFile(name);
snprintf(name,100,"contVZEROv%i",arm);
TList *cont = (TList *) fo->Get(name);
cont->ls();
Float_t xMin[5] = {icentr/*centrality bin*/,chMin/*charge*/,pTh/*prob*/,-TMath::Pi()/arm/*Psi*/,detMin/*PID mask*/};
Float_t xMax[5] = {icentr+addbin,chMax,1.0,TMath::Pi()/arm,detMax};
cont->ls();
TProfile *p1 = cont->At(2);
TProfile *p2 = cont->At(3);
TProfile *p3 = cont->At(4);
TH2F *hPsi2DA = cont->At(5);
TH2F *hPsi2DC = cont->At(6);
TH1D *hPsiA = hPsi2DA->ProjectionY("PsiA",icentr+1,icentr+addbin+1);
TH1D *hPsiC = hPsi2DC->ProjectionY("PsiC",icentr+1,icentr+addbin+1);
if(!fPsi) fPsi = new TF1("fPsi","pol0",-TMath::Pi()/arm,TMath::Pi()/arm);
hPsiA->Fit(fPsi,"0");
Float_t offsetA = fPsi->GetParameter(0);
hPsiC->Fit(fPsi,"0");
Float_t offsetC = fPsi->GetParameter(0);
Int_t nbinPsi = hPsiA->GetNbinsX();
Float_t *NUAcorrA = new Float_t[nbinPsi];
Float_t *NUAcorrC = new Float_t[nbinPsi];
for(Int_t i=0;i < nbinPsi;i++){
NUAcorrA[i] = offsetA/(hPsiA->GetBinContent(i+1));
NUAcorrC[i] = offsetC/(hPsiC->GetBinContent(i+1));
}
Float_t res1=0,res2=0,res3=0;
Float_t eres1=0,eres2=0,eres3=0;
for(Int_t i = icentr; i <= icentr+addbin;i++){
if(p1->GetBinError(i+1)){
eres1 += 1./p1->GetBinError(i+1)/p1->GetBinError(i+1);
res1 += p1->GetBinContent(i+1)/p1->GetBinError(i+1)/p1->GetBinError(i+1);
}
if(p2->GetBinError(i+1)){
eres2 += 1./p2->GetBinError(i+1)/p2->GetBinError(i+1);
res2 += p2->GetBinContent(i+1)/p2->GetBinError(i+1)/p2->GetBinError(i+1);
}
if(p3->GetBinError(i+1)){
eres3 += 1./p3->GetBinError(i+1)/p3->GetBinError(i+1);
res3 += p3->GetBinContent(i+1)/p3->GetBinError(i+1)/p3->GetBinError(i+1);
}
}
res1 /= eres1;
res2 /= eres2;
res3 /= eres3;
eres1 = sqrt(1./eres1);
eres2 = sqrt(1./eres2);
eres3 = sqrt(1./eres3);
AliFlowVZEROResults *a = (AliFlowVZEROResults *) cont->At(0);
AliFlowVZEROResults *b = (AliFlowVZEROResults *) cont->At(1);
TProfile *pp,*pp2;
if(kNUAcorr){ // with NUA corrections
pp = a->GetV2reweight(spec,xMin,xMax,3,NUAcorrA);
pp2 = b->GetV2reweight(spec,xMin,xMax,3,NUAcorrC);
}
else{
pp = a->GetV2(spec,xMin,xMax);
pp2 = b->GetV2(spec,xMin,xMax);
}
Float_t scaling = sqrt(res1*res3/res2);
if(kVZEROrescorr){
pp->Scale(1./scaling);
}
Float_t err1_2 = eres1*eres1/res1/res1/4 +
eres2*eres2/res2/res2/4 +
eres3*eres3/res3/res3/4;
Float_t err2_2 = err1_2;
err1_2 /= scaling*scaling;
printf("resolution V0A = %f +/- %f\n",scaling,err1_2);
scaling = sqrt(res2*res3/res1);
err2_2 /= scaling*scaling;
if(kVZEROrescorr){
pp2->Scale(1./scaling);
}
printf("resolution V0C =%f +/- %f\n",scaling,err2_2);
pp->SetName("V0A");
pp2->SetName("V0C");
if(! kCleanMemory){
new TCanvas();
pp->Draw();
pp2->Draw("SAME");
}
TProfile *pData = new TProfile(*pp);
pData->Add(pp2);
snprintf(stringa,100,"%sData",nameSp);
pData->SetName(stringa);
TProfile *pMc = NULL;
TProfile *ppMC;
TProfile *ppMC2;
if(isMC){
snprintf(name,100,"contVZEROmc");
cont = (TList *) fo->Get(name);
cont->ls();
if(arm == 2){
AliFlowVZEROResults *c = (AliFlowVZEROResults *) cont->At(0);
if(! kCleanMemory) c->GetV2(spec,xMin,xMax)->Draw("SAME");
}
AliFlowVZEROResults *cA;
if(fo->Get("contVZEROv2")) cA = (AliFlowVZEROResults *) cont->At(1+2*(arm==3));
else cA = (AliFlowVZEROResults *) cont->At(0);
AliFlowVZEROResults *cC;
if(fo->Get("contVZEROv2")) cC = (AliFlowVZEROResults *) cont->At(2+2*(arm==3));
else cC = (AliFlowVZEROResults *) cont->At(1);
TProfile *p1mc = cont->At(5+3*(arm==3));
TProfile *p2mc = cont->At(6+3*(arm==3));
TProfile *p3mc = cont->At(7+3*(arm==3));
Float_t resMC1=0,resMC2=0,resMC3=0;
Float_t eresMC1=0,eresMC2=0,eresMC3=0;
for(Int_t i = icentr; i <= icentr+addbin;i++){
if(p1mc->GetBinError(i+1)){
eresMC1 += 1./p1mc->GetBinError(i+1)/p1mc->GetBinError(i+1);
resMC1 += p1mc->GetBinContent(i+1)/p1mc->GetBinError(i+1)/p1mc->GetBinError(i+1);
}
if(p2mc->GetBinError(i+1)){
eresMC2 += 1./p2mc->GetBinError(i+1)/p2mc->GetBinError(i+1);
resMC2 += p2mc->GetBinContent(i+1)/p2mc->GetBinError(i+1)/p2mc->GetBinError(i+1);
}
if(p3mc->GetBinError(i+1)){
eresMC3 += 1./p3mc->GetBinError(i+1)/p3mc->GetBinError(i+1);
resMC3 += p3mc->GetBinContent(i+1)/p3mc->GetBinError(i+1)/p3mc->GetBinError(i+1);
}
}
resMC1 /= eresMC1;
resMC2 /= eresMC2;
resMC3 /= eresMC3;
eresMC1 = sqrt(1./eresMC1);
eresMC2 = sqrt(1./eresMC2);
eresMC3 = sqrt(1./eresMC3);
ppMC = cA->GetV2(spec,xMin,xMax);
ppMC2 = cC->GetV2(spec,xMin,xMax);
scaling = sqrt(resMC1*resMC3/resMC2);
ppMC->Scale(1./scaling);
err1_2 = eresMC1*eresMC1/resMC1/resMC1/4 +
eresMC2*eresMC2/resMC2/resMC2/4 +
eresMC3*eresMC3/resMC3/resMC3/4;
err2_2 = err1_2;
err1_2 /= scaling*scaling;
printf("resolution V0A (MC) = %f +/- %f\n",scaling,err1_2);
scaling = sqrt(resMC2*resMC3/resMC1);
err2_2 /= scaling*scaling;
ppMC2->Scale(1./scaling);
printf("resolution V0C (MC) =%f +/- %f\n",scaling,err2_2);
ppMC->SetName("V0Amc");
ppMC2->SetName("V0Cmc");
if(! kCleanMemory){
ppMC->Draw("SAME");
ppMC2->Draw("SAME");
}
pMc = new TProfile(*ppMC);
pMc->Add(ppMC2);
snprintf(stringa,100,"%sMC",nameSp);
pMc->SetName(stringa);
pMc->SetLineColor(2);
}
if(! kCleanMemory){
new TCanvas();
pData->Draw();
}
if(pMc && !kCleanMemory){
pMc->Draw("SAME");
TH1D *hData = pData->ProjectionX();
TH1D *hMc = pMc->ProjectionX();
hData->Divide(hMc);
new TCanvas();
hData->Draw();
}
delete[] NUAcorrA;
delete[] NUAcorrC;
if(kCleanMemory){
if(pp) delete pp;
if(pp2) delete pp2;
if(ppMC) delete ppMC;
if(ppMC2) delete ppMC2;
}
delete cont;
if(isMC) return pMc;
return pData;
}