//________________________________________________________________________________________
AliRsnFunction::AliRsnFunction(Bool_t useTH1) :
- TObject(),
- fPairDef(0x0),
- fAxisList("AliRsnValue", 0),
- fPair(0x0),
- fEvent(0x0),
- fUseTH1(useTH1),
- fSize(0),
- fH1(0x0),
- fHSparse(0x0)
+ TObject(),
+ fPairDef(0x0),
+ fAxisList("AliRsnValue", 0),
+ fPair(0x0),
+ fEvent(0x0),
+ fUseTH1(useTH1),
+ fSize(0),
+ fH1(0x0),
+ fHSparse(0x0)
{
//
// Constructor.
//________________________________________________________________________________________
AliRsnFunction::AliRsnFunction(const AliRsnFunction ©) :
- TObject(copy),
- fPairDef(copy.fPairDef),
- fAxisList(copy.fAxisList),
- fPair(copy.fPair),
- fEvent(copy.fEvent),
- fUseTH1(copy.fUseTH1),
- fSize(copy.fSize),
- fH1(0x0),
- fHSparse(0x0)
+ TObject(copy),
+ fPairDef(copy.fPairDef),
+ fAxisList(copy.fAxisList),
+ fPair(copy.fPair),
+ fEvent(copy.fEvent),
+ fUseTH1(copy.fUseTH1),
+ fSize(copy.fSize),
+ fH1(0x0),
+ fHSparse(0x0)
{
//
// Copy constructor.
// Assignment operator.
//
- fPairDef = copy.fPairDef;
- fPair = copy.fPair;
- fEvent = copy.fEvent;
- fUseTH1 = copy.fUseTH1;
- fSize = copy.fSize;
+ fPairDef = copy.fPairDef;
+ fPair = copy.fPair;
+ fEvent = copy.fEvent;
+ fUseTH1 = copy.fUseTH1;
+ fSize = copy.fSize;
- if (fH1) delete fH1;
- fH1 = 0x0;
-
- if (fHSparse) delete fHSparse;
- fHSparse = 0x0;
+ if (fH1) delete fH1;
+ fH1 = 0x0;
- return (*this);
+ if (fHSparse) delete fHSparse;
+ fHSparse = 0x0;
+
+ return (*this);
}
//________________________________________________________________________________________
// the function type and binning
//
- TString name("");
+ TString name("");
- TObjArrayIter next(&fAxisList);
- AliRsnValue *axis = 0;
+ TObjArrayIter next(&fAxisList);
+ AliRsnValue *axis = 0;
- while ((axis = (AliRsnValue*)next()))
- {
- if (name.Length() > 1) name += '_';
- name += axis->GetName();
- }
+ while ((axis = (AliRsnValue*)next())) {
+ if (name.Length() > 1) name += '_';
+ name += axis->GetName();
+ }
- return name.Data();
+ return name.Data();
}
//________________________________________________________________________________________
// NOTE: this can cause large files and high memory occupancy.
//
- RSNTARGET target = axis->GetTargetType();
- if (target != AliRsnTarget::kMother && target != AliRsnTarget::kEvent)
- {
- AliError(Form("Allowed targets are mothers and events; cannot use axis '%s' which has target '%s'", axis->GetName(), axis->GetTargetTypeName()));
- return kFALSE;
- }
-
- Int_t size = fAxisList.GetEntries();
- new (fAxisList[size]) AliRsnValue(*axis);
-
- if (fAxisList.GetEntries() > 3)
- {
- AliWarning("Adding more than 3 axes will produce a THnSparseD output.");
- fUseTH1 = kFALSE;
- }
-
- return kTRUE;
+ RSNTARGET target = axis->GetTargetType();
+ if (target != AliRsnTarget::kMother && target != AliRsnTarget::kEvent) {
+ AliError(Form("Allowed targets are mothers and events; cannot use axis '%s' which has target '%s'", axis->GetName(), axis->GetTargetTypeName()));
+ return kFALSE;
+ }
+
+ Int_t size = fAxisList.GetEntries();
+ new(fAxisList[size]) AliRsnValue(*axis);
+
+ if (fAxisList.GetEntries() > 3) {
+ AliWarning("Adding more than 3 axes will produce a THnSparseD output.");
+ fUseTH1 = kFALSE;
+ }
+
+ return kTRUE;
}
//________________________________________________________________________________________
// This version produces a THnSparseF.
//
- fSize = fAxisList.GetEntries();
- if (!fSize)
- {
- AliError("No axes defined");
- return 0x0;
- }
- else if (fSize > 3)
- {
- AliError("Too many axes defined for a TH1 object");
- return 0x0;
- }
-
- // retrieve binnings for main and secondary axes
- AliRsnValue *fcnAxis;
- TArrayD array[3];
- for (Int_t i = 0; i < fSize; i++)
- {
- fcnAxis = (AliRsnValue*)fAxisList.At(i);
- if (!fcnAxis)
- {
- AliError("Empty axis");
- array[i].Set(2);
- array[i][0] = -1E5;
- array[i][1] = -1E5;
- continue;
- }
- else
- {
- array[i] = fcnAxis->GetArray();
- }
- }
-
- // create histogram depending on the number of axes
- switch (fSize)
- {
- case 1:
- fH1 = new TH1F(histoName, histoTitle, array[0].GetSize() - 1, array[0].GetArray());
- break;
- case 2:
- fH1 = new TH2F(histoName, histoTitle, array[0].GetSize() - 1, array[0].GetArray(), array[1].GetSize() - 1, array[1].GetArray());
- break;
- case 3:
- fH1 = new TH3F(histoName, histoTitle, array[0].GetSize() - 1, array[0].GetArray(), array[1].GetSize() - 1, array[1].GetArray(), array[2].GetSize() - 1, array[2].GetArray());
- break;
- }
- fH1->Sumw2();
-
- return fH1;
+ fSize = fAxisList.GetEntries();
+ if (!fSize) {
+ AliError("No axes defined");
+ return 0x0;
+ } else if (fSize > 3) {
+ AliError("Too many axes defined for a TH1 object");
+ return 0x0;
+ }
+
+ // retrieve binnings for main and secondary axes
+ AliRsnValue *fcnAxis;
+ TArrayD array[3];
+ for (Int_t i = 0; i < fSize; i++) {
+ fcnAxis = (AliRsnValue*)fAxisList.At(i);
+ if (!fcnAxis) {
+ AliError("Empty axis");
+ array[i].Set(2);
+ array[i][0] = -1E5;
+ array[i][1] = -1E5;
+ continue;
+ } else {
+ array[i] = fcnAxis->GetArray();
+ }
+ }
+
+ // create histogram depending on the number of axes
+ switch (fSize) {
+ case 1:
+ fH1 = new TH1F(histoName, histoTitle, array[0].GetSize() - 1, array[0].GetArray());
+ break;
+ case 2:
+ fH1 = new TH2F(histoName, histoTitle, array[0].GetSize() - 1, array[0].GetArray(), array[1].GetSize() - 1, array[1].GetArray());
+ break;
+ case 3:
+ fH1 = new TH3F(histoName, histoTitle, array[0].GetSize() - 1, array[0].GetArray(), array[1].GetSize() - 1, array[1].GetArray(), array[2].GetSize() - 1, array[2].GetArray());
+ break;
+ }
+ fH1->Sumw2();
+
+ return fH1;
}
//________________________________________________________________________________________
// This version produces a THnSparseF.
//
- fSize = fAxisList.GetEntries();
- if (!fSize)
- {
- AliError("No axes defined");
- return 0x0;
- }
-
- // initialize the array of number of bins for each axis
- // taking it from the stored values, while for the bins
- // they are set as summied and defined later
- Double_t dummyD;
- Int_t *nbins = new Int_t[fSize];
- AliRsnValue *fcnAxis = 0;
- for (Int_t i = 0; i < fSize; i++)
- {
- fcnAxis = (AliRsnValue*)fAxisList.At(i);
- if (!fcnAxis)
- {
- nbins[i] = 1;
- AliError("Empty axis");
- continue;
- }
- nbins[i] = fcnAxis->GetArray().GetSize() - 1;
- }
-
- // create histogram
- fHSparse = new THnSparseF(histoName, histoTitle, fSize, nbins, &dummyD, &dummyD);
- fHSparse->Sumw2();
-
- // update the various axes using the definitions given in the array of axes here
- for (Int_t i = 0; i < fSize; i++)
- {
- fcnAxis = (AliRsnValue*)fAxisList.At(i);
- if (!fcnAxis) {
- AliError("Empty axis: doing unique bin betweeen -100000 and 100000");
- continue;
- }
- fHSparse->SetBinEdges(i, fcnAxis->GetArray().GetArray());
- }
-
- delete [] nbins;
-
- return fHSparse;
+ fSize = fAxisList.GetEntries();
+ if (!fSize) {
+ AliError("No axes defined");
+ return 0x0;
+ }
+
+ // initialize the array of number of bins for each axis
+ // taking it from the stored values, while for the bins
+ // they are set as summied and defined later
+ Double_t dummyD;
+ Int_t *nbins = new Int_t[fSize];
+ AliRsnValue *fcnAxis = 0;
+ for (Int_t i = 0; i < fSize; i++) {
+ fcnAxis = (AliRsnValue*)fAxisList.At(i);
+ if (!fcnAxis) {
+ nbins[i] = 1;
+ AliError("Empty axis");
+ continue;
+ }
+ nbins[i] = fcnAxis->GetArray().GetSize() - 1;
+ }
+
+ // create histogram
+ fHSparse = new THnSparseF(histoName, histoTitle, fSize, nbins, &dummyD, &dummyD);
+ fHSparse->Sumw2();
+
+ // update the various axes using the definitions given in the array of axes here
+ for (Int_t i = 0; i < fSize; i++) {
+ fcnAxis = (AliRsnValue*)fAxisList.At(i);
+ if (!fcnAxis) {
+ AliError("Empty axis: doing unique bin betweeen -100000 and 100000");
+ continue;
+ }
+ fHSparse->SetBinEdges(i, fcnAxis->GetArray().GetArray());
+ }
+
+ delete [] nbins;
+
+ return fHSparse;
}
// Fill function histogram with values computed from given input object.
//
- AliDebug(AliLog::kDebug +2,"->");
-
- Int_t i;
- Double_t *values = new Double_t[fSize];
- Bool_t computeOK = kFALSE;
-
- AliRsnValue *fcnAxis = 0;
- for (i = 0; i < fSize; i++)
- {
- values[i] = 0.0;
- fcnAxis = (AliRsnValue*)fAxisList.At(i);
- if (!fcnAxis) continue;
- switch (fcnAxis->GetTargetType())
- {
- case AliRsnTarget::kMother:
- fcnAxis->SetSupportObject(fPairDef);
- computeOK = fcnAxis->Eval(fPair);
- break;
- case AliRsnTarget::kEvent:
- computeOK = fcnAxis->Eval(fEvent);
- break;
- default:
- AliError(Form("Allowed targets are mothers and events; cannot use axis '%s' which has target '%s'", fcnAxis->GetName(), fcnAxis->GetTargetTypeName()));
- computeOK = kFALSE;
- }
- if (computeOK) values[i] = ((Float_t)fcnAxis->GetComputedValue());
- }
-
- // fill histogram
- if (fUseTH1)
- {
- // check presence of output histogram
- if (!fH1)
- {
- AliError("Required a TH1 which is not initialized");
- delete [] values;
- return kFALSE;
- }
-
- // fill according to dimensions
- switch (fSize)
- {
- case 1:
- {
- TH1F *h1 = (TH1F*)fH1;
- h1->Fill(values[0]);
- }
- break;
- case 2:
- {
- TH2F *h2 = (TH2F*)fH1;
- h2->Fill(values[0], values[1]);
- }
- break;
- case 3:
- {
- TH3F *h3 = (TH3F*)fH1;
- h3->Fill(values[0], values[1], values[2]);
- }
- break;
- default:
- AliError(Form("Wrong size : %d", fSize));
- delete [] values;
- return kFALSE;
- }
- }
- else
- {
- // check presence of output histogram
- if (!fHSparse)
- {
- AliError("Required a THnSparseF which is not initialized");
- delete [] values;
- return kFALSE;
- }
-
- fHSparse->Fill(values);
- }
-
- delete [] values;
-
- AliDebug(AliLog::kDebug +2,"->");
- return kTRUE;
+ AliDebug(AliLog::kDebug + 2, "->");
+
+ Int_t i;
+ Double_t *values = new Double_t[fSize];
+ Bool_t computeOK = kFALSE;
+
+ AliRsnValue *fcnAxis = 0;
+ for (i = 0; i < fSize; i++) {
+ values[i] = 0.0;
+ fcnAxis = (AliRsnValue*)fAxisList.At(i);
+ if (!fcnAxis) continue;
+ switch (fcnAxis->GetTargetType()) {
+ case AliRsnTarget::kMother:
+ fcnAxis->SetSupportObject(fPairDef);
+ computeOK = fcnAxis->Eval(fPair);
+ break;
+ case AliRsnTarget::kEvent:
+ computeOK = fcnAxis->Eval(fEvent);
+ break;
+ default:
+ AliError(Form("Allowed targets are mothers and events; cannot use axis '%s' which has target '%s'", fcnAxis->GetName(), fcnAxis->GetTargetTypeName()));
+ computeOK = kFALSE;
+ }
+ if (computeOK) values[i] = ((Float_t)fcnAxis->GetComputedValue());
+ }
+
+ // fill histogram
+ if (fUseTH1) {
+ // check presence of output histogram
+ if (!fH1) {
+ AliError("Required a TH1 which is not initialized");
+ delete [] values;
+ return kFALSE;
+ }
+
+ // fill according to dimensions
+ switch (fSize) {
+ case 1: {
+ TH1F *h1 = (TH1F*)fH1;
+ h1->Fill(values[0]);
+ }
+ break;
+ case 2: {
+ TH2F *h2 = (TH2F*)fH1;
+ h2->Fill(values[0], values[1]);
+ }
+ break;
+ case 3: {
+ TH3F *h3 = (TH3F*)fH1;
+ h3->Fill(values[0], values[1], values[2]);
+ }
+ break;
+ default:
+ AliError(Form("Wrong size : %d", fSize));
+ delete [] values;
+ return kFALSE;
+ }
+ } else {
+ // check presence of output histogram
+ if (!fHSparse) {
+ AliError("Required a THnSparseF which is not initialized");
+ delete [] values;
+ return kFALSE;
+ }
+
+ fHSparse->Fill(values);
+ }
+
+ delete [] values;
+
+ AliDebug(AliLog::kDebug + 2, "->");
+ return kTRUE;
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff