// ------------------------------------------------------
//
+#include <TDirectory.h>
+#include <TH1F.h>
+#include <TH2F.h>
#include <TH3F.h>
+#include <TString.h>
+#include <TMath.h>
#include <AliLog.h>
#include "AliCorrectionMatrix3D.h"
+#include "AliCorrectionMatrix2D.h"
#include "AliPWG0Helper.h"
//____________________________________________________________________
((AliCorrectionMatrix3D &)c).Copy(*this);
}
+//____________________________________________________________________
+AliCorrectionMatrix3D &AliCorrectionMatrix3D::operator=(const AliCorrectionMatrix3D &c)
+{
+ // assigment operator
+
+ if (this != &c)
+ ((AliCorrectionMatrix3D &) c).Copy(*this);
+
+ return *this;
+}
+
//____________________________________________________________________
AliCorrectionMatrix3D::AliCorrectionMatrix3D(const Char_t* name, const Char_t* title,
Int_t nBinX, Float_t Xmin, Float_t Xmax,
// constructor
//
- fhMeas = new TH3F(Form("meas_%s",name), Form("meas_%s",title), nBinX, Xmin, Xmax, nBinY, Ymin, Ymax, nBinZ, Zmin, Zmax);
- fhGene = new TH3F(Form("gene_%s",name), Form("gene_%s",title), nBinX, Xmin, Xmax, nBinY, Ymin, Ymax, nBinZ, Zmin, Zmax);
- fhCorr = new TH3F(Form("corr_%s",name), Form("corr_%s",title), nBinX, Xmin, Xmax, nBinY, Ymin, Ymax, nBinZ, Zmin, Zmax);
+ Float_t* binLimitsX = new Float_t[nBinX+1];
+ for (Int_t i=0; i<=nBinX; ++i)
+ binLimitsX[i] = Xmin + (Xmax - Xmin) / nBinX * i;
- fhMeas->Sumw2();
- fhGene->Sumw2();
- fhCorr->Sumw2();
+ Float_t* binLimitsY = new Float_t[nBinY+1];
+ for (Int_t i=0; i<=nBinY; ++i)
+ binLimitsY[i] = Ymin + (Ymax - Ymin) / nBinY * i;
+
+ Float_t* binLimitsZ = new Float_t[nBinZ+1];
+ for (Int_t i=0; i<=nBinZ; ++i)
+ binLimitsZ[i] = Zmin + (Zmax - Zmin) / nBinZ * i;
+
+ CreateHists(nBinX, binLimitsX, nBinY, binLimitsY, nBinZ, binLimitsZ);
+
+ delete[] binLimitsX;
+ delete[] binLimitsY;
+ delete[] binLimitsZ;
}
AliCorrectionMatrix3D::AliCorrectionMatrix3D(const Char_t* name, const Char_t* title,
for (Int_t i=0; i<=nBinY; ++i)
binLimitsY[i] = Ymin + (Ymax - Ymin) / nBinY * i;
- fhMeas = new TH3F(Form("meas_%s",name), Form("meas_%s",title), nBinX, binLimitsX, nBinY, binLimitsY, nBinZ, zbins);
- fhGene = new TH3F(Form("gene_%s",name), Form("gene_%s",title), nBinX, binLimitsX, nBinY, binLimitsY, nBinZ, zbins);
- fhCorr = new TH3F(Form("corr_%s",name), Form("corr_%s",title), nBinX, binLimitsX, nBinY, binLimitsY, nBinZ, zbins);
+ CreateHists(nBinX, binLimitsX, nBinY, binLimitsY, nBinZ, zbins);
delete[] binLimitsX;
delete[] binLimitsY;
+}
+
+AliCorrectionMatrix3D::AliCorrectionMatrix3D(const Char_t* name, const Char_t* title, TH3* hBinning)
+ : AliCorrectionMatrix(name, title)
+{
+ // constructor with variable bin sizes (uses binning of hBinning)
+
+ // do not add this hists to the directory
+ Bool_t oldStatus = TH1::AddDirectoryStatus();
+ TH1::AddDirectory(kFALSE);
+
+ fhMeas = (TH3F*)hBinning->Clone("measured");
+ fhGene = (TH3F*)hBinning->Clone("generated");
+ fhCorr = (TH3F*)hBinning->Clone("correction");
+
+ fhMeas->SetTitle(Form("%s measured", GetTitle()));
+ fhGene->SetTitle(Form("%s generated", GetTitle()));
+ fhCorr->SetTitle(Form("%s correction", GetTitle()));
+
+ fhMeas->Reset();
+ fhGene->Reset();
+ fhCorr->Reset();
+
+ TH1::AddDirectory(oldStatus);
+
+ fhMeas->Sumw2();
+ fhGene->Sumw2();
+ fhCorr->Sumw2();
+}
+
+//____________________________________________________________________
+void AliCorrectionMatrix3D::CreateHists(Int_t nBinX, const Float_t* binLimitsX,
+ Int_t nBinY, const Float_t* binLimitsY,
+ Int_t nBinZ, const Float_t* binLimitsZ)
+{
+ // create the histograms
+
+ // do not add this hists to the directory
+ Bool_t oldStatus = TH1::AddDirectoryStatus();
+ TH1::AddDirectory(kFALSE);
+
+ fhMeas = new TH3F("measured", Form("%s measured",GetTitle()), nBinX, binLimitsX, nBinY, binLimitsY, nBinZ, binLimitsZ);
+ fhGene = new TH3F("generated", Form("%s generated",GetTitle()), nBinX, binLimitsX, nBinY, binLimitsY, nBinZ, binLimitsZ);
+ fhCorr = new TH3F("correction", Form("%s correction",GetTitle()), nBinX, binLimitsX, nBinY, binLimitsY, nBinZ, binLimitsZ);
fhMeas->Sumw2();
fhGene->Sumw2();
fhCorr->Sumw2();
+
+ TH1::AddDirectory(oldStatus);
}
+
//____________________________________________________________________
AliCorrectionMatrix3D::~AliCorrectionMatrix3D()
{
}
//____________________________________________________________________
-TH3F* AliCorrectionMatrix3D::GetGeneratedHistogram()
+TH3* AliCorrectionMatrix3D::GetGeneratedHistogram()
{
// return generated histogram casted to correct type
return dynamic_cast<TH3F*> (fhGene);
}
//____________________________________________________________________
-TH3F* AliCorrectionMatrix3D::GetMeasuredHistogram()
+TH3* AliCorrectionMatrix3D::GetMeasuredHistogram()
{
// return measured histogram casted to correct type
return dynamic_cast<TH3F*> (fhMeas);
}
//____________________________________________________________________
-TH3F* AliCorrectionMatrix3D::GetCorrectionHistogram()
+TH3* AliCorrectionMatrix3D::GetCorrectionHistogram()
{
// return correction histogram casted to correct type
return dynamic_cast<TH3F*> (fhCorr);
}
//____________________________________________________________________
-void AliCorrectionMatrix3D::FillMeas(Float_t ax, Float_t ay, Float_t az)
+AliCorrectionMatrix2D* AliCorrectionMatrix3D::Get2DCorrection(Option_t* opt, Float_t aMin, Float_t aMax)
+{
+ // returns a 2D projection of this correction
+
+ TString option = opt;
+
+ // unzoom
+ fhMeas->GetXaxis()->SetRange(0, 0);
+ fhMeas->GetYaxis()->SetRange(0, 0);
+ fhMeas->GetZaxis()->SetRange(0, 0);
+
+ fhGene->GetXaxis()->SetRange(0, 0);
+ fhGene->GetYaxis()->SetRange(0, 0);
+ fhGene->GetZaxis()->SetRange(0, 0);
+
+ if (aMin<aMax) {
+ if (option.Contains("xy") || option.Contains("yx")) {
+ Int_t bMin = fhMeas->GetZaxis()->FindBin(aMin);
+ Int_t bMax = fhMeas->GetZaxis()->FindBin(aMax);
+ fhGene->GetZaxis()->SetRange(bMin, bMax);
+ fhMeas->GetZaxis()->SetRange(bMin, bMax);
+ }
+ else if (option.Contains("xz") || option.Contains("zx")) {
+ Int_t bMin = fhMeas->GetYaxis()->FindBin(aMin);
+ Int_t bMax = fhMeas->GetYaxis()->FindBin(aMax);
+ fhGene->GetYaxis()->SetRange(bMin, bMax);
+ fhMeas->GetYaxis()->SetRange(bMin, bMax);
+ }
+ else if (option.Contains("yz") || option.Contains("zy")) {
+ Int_t bMin = fhMeas->GetXaxis()->FindBin(aMin);
+ Int_t bMax = fhMeas->GetXaxis()->FindBin(aMax);
+ fhGene->GetXaxis()->SetRange(bMin, bMax);
+ fhMeas->GetXaxis()->SetRange(bMin, bMax);
+ }
+ else {
+ AliDebug(AliLog::kWarning, Form("WARNING: unknown projection option %s", opt));
+ return 0;
+ }
+ }
+
+ AliCorrectionMatrix2D* corr2D = new AliCorrectionMatrix2D(Form("%s_%s",GetName(),opt),Form("%s projection %s",GetName(),opt),100,0,100,100,0,100);
+
+ TH2F* meas = (TH2F*) ((TH3F*)fhMeas)->Project3D(option)->Clone(Form("%s_meas", corr2D->GetName()));
+ TH2F* gene = (TH2F*) ((TH3F*)fhGene)->Project3D(option)->Clone(Form("%s_gene", corr2D->GetName()));
+
+ // set errors
+ for (Int_t x = 0; x<=meas->GetNbinsX()+1; x++)
+ for (Int_t y = 0; y<=meas->GetNbinsY()+1; y++)
+ {
+ gene->SetBinError(x, y, TMath::Sqrt(gene->GetBinContent(x, y)));
+ meas->SetBinError(x, y, TMath::Sqrt(meas->GetBinContent(x, y)));
+ }
+
+ TH2F* corr = (TH2F*)gene->Clone(Form("%s_corr", corr2D->GetName()));
+ corr->Reset();
+
+ corr2D->SetGeneratedHistogram(gene);
+ corr2D->SetMeasuredHistogram(meas);
+ corr2D->SetCorrectionHistogram(corr);
+
+ corr2D->Divide();
+
+ // unzoom
+ fhMeas->GetXaxis()->SetRange(0, 0);
+ fhMeas->GetYaxis()->SetRange(0, 0);
+ fhMeas->GetZaxis()->SetRange(0, 0);
+
+ fhGene->GetXaxis()->SetRange(0, 0);
+ fhGene->GetYaxis()->SetRange(0, 0);
+ fhGene->GetZaxis()->SetRange(0, 0);
+
+ return corr2D;
+}
+
+//____________________________________________________________________
+TH1* AliCorrectionMatrix3D::Get1DCorrectionHistogram(Option_t* opt, Float_t aMin1, Float_t aMax1, Float_t aMin2, Float_t aMax2)
+{
+ // returns a 1D projection of this correction
+
+ AliCorrectionMatrix2D* corr2D = 0;
+ if (strcmp(opt,"x")==0) {
+ corr2D = Get2DCorrection("yx",aMin2,aMax2);
+ return corr2D->Get1DCorrectionHistogram("x",aMin1,aMax1);
+ }
+ if (strcmp(opt,"y")==0) {
+ corr2D = Get2DCorrection("xy",aMin2,aMax2);
+ return corr2D->Get1DCorrectionHistogram("x",aMin1,aMax1);
+ }
+ if (strcmp(opt,"z")==0) {
+ corr2D = Get2DCorrection("yz",aMin1,aMax1);
+ return corr2D->Get1DCorrectionHistogram("x",aMin2,aMax2);
+ }
+ AliDebug(AliLog::kWarning, Form("WARNING: unknown projection option %s (should be x,y or z)", opt));
+
+ return 0;
+}
+
+//____________________________________________________________________
+void AliCorrectionMatrix3D::FillMeas(Float_t ax, Float_t ay, Float_t az, Double_t weight)
{
// add value to measured histogram
- GetMeasuredHistogram()->Fill(ax, ay, az);
+ GetMeasuredHistogram()->Fill(ax, ay, az, weight);
}
//____________________________________________________________________
-void AliCorrectionMatrix3D::FillGene(Float_t ax, Float_t ay, Float_t az)
+void AliCorrectionMatrix3D::FillGene(Float_t ax, Float_t ay, Float_t az, Double_t weight)
{
// add value to generated histogram
- GetGeneratedHistogram()->Fill(ax, ay, az);
+ GetGeneratedHistogram()->Fill(ax, ay, az, weight);
}
//____________________________________________________________________
AliCorrectionMatrix::SaveHistograms();
- //AliPWG0Helper::CreateProjections(GetMeasuredHistogram());
- //AliPWG0Helper::CreateProjections(GetGeneratedHistogram());
+ if (GetGeneratedHistogram() && GetMeasuredHistogram())
+ {
+ gDirectory->cd(GetName());
+
+ AliPWG0Helper::CreateDividedProjections(GetGeneratedHistogram(), GetMeasuredHistogram(), 0, kFALSE, kTRUE);
+
+ gDirectory->cd("..");
+ }
+}
+
+//____________________________________________________________________
+Int_t AliCorrectionMatrix3D::CheckEmptyBins(Float_t xmin, Float_t xmax, Float_t ymin, Float_t ymax, Float_t zmin, Float_t zmax, Bool_t quiet)
+{
+ //
+ // counts the number of empty Bins in a given region
+ //
+
+ TH3* hist = GetGeneratedHistogram();
+ if (!hist)
+ return -1;
+
+ Int_t emptyBins = 0;
+ for (Int_t x=hist->GetXaxis()->FindBin(xmin); x<=hist->GetXaxis()->FindBin(xmax); ++x)
+ for (Int_t y=hist->GetYaxis()->FindBin(ymin); y<=hist->GetYaxis()->FindBin(ymax); ++y)
+ for (Int_t z=hist->GetZaxis()->FindBin(zmin); z<=hist->GetZaxis()->FindBin(zmax); ++z)
+ if (hist->GetBinContent(x, y, z) == 0)
+ {
+ if (!quiet)
+ printf("Empty bin in %s at vtx = %f, eta = %f, pt = %f\n", GetName(), hist->GetXaxis()->GetBinCenter(x), hist->GetYaxis()->GetBinCenter(y), hist->GetZaxis()->GetBinCenter(z));
+ ++emptyBins;
+ }
+
+ return emptyBins;
+}
+
+//____________________________________________________________________
+TH1* AliCorrectionMatrix3D::PlotBinErrors(Float_t xmin, Float_t xmax, Float_t ymin, Float_t ymax, Float_t zmin, Float_t zmax)
+{
+ //
+ // makes a 1d plots of the relative bin errors
+ //
+
+ TH3* hist = GetCorrectionHistogram();
+ if (!hist)
+ return 0;
+
+ TH1F* target = new TH1F("relerrors", "relerrors", 100, 0, 10);
- if (GetCorrectionHistogram())
- AliPWG0Helper::CreateDividedProjections(GetMeasuredHistogram(), GetGeneratedHistogram());
+ for (Int_t x=hist->GetXaxis()->FindBin(xmin); x<=hist->GetXaxis()->FindBin(xmax); ++x)
+ for (Int_t y=hist->GetYaxis()->FindBin(ymin); y<=hist->GetYaxis()->FindBin(ymax); ++y)
+ for (Int_t z=hist->GetZaxis()->FindBin(zmin); z<=hist->GetZaxis()->FindBin(zmax); ++z)
+ if (hist->GetBinContent(x, y, z) != 0)
+ target->Fill(100 * hist->GetBinError(x, y, z) / hist->GetBinContent(x, y, z));
+
+ return target;
}
+