80351b68ffc1d942f1430e0152cd9e2fbb89bea5
[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnPairFunctions.cxx
1 //
2 // *** Class AliRsnPairFunctions ***
3 //
4 // "Core" method for defining the work on a pari of particles.
5 // For one analysis, one must setup one of this for each pair he wants to analyze,
6 // adding to it all analysis which he desires to do.
7 // Here he defines the cuts, and the particle types and charges, and can add
8 // functions which do different operations on the same pair, and some binning
9 // with respect to some kinematic variables (eta, momentum)
10 //
11 // authors: A. Pulvirenti (email: alberto.pulvirenti@ct.infn.it)
12 //          M. Vala (email: martin.vala@cern.ch)
13 //
14
15 #include <TList.h>
16
17 #include "AliLog.h"
18
19 #include "AliRsnMother.h"
20 #include "AliRsnEvent.h"
21 #include "AliRsnFunction.h"
22 #include "AliRsnCutSet.h"
23 #include "AliRsnValue.h"
24
25 #include "AliRsnPairFunctions.h"
26
27 ClassImp(AliRsnPairFunctions)
28
29 //_____________________________________________________________________________
30 AliRsnPairFunctions::AliRsnPairFunctions(const char *name, AliRsnPairDef *def) :
31    AliRsnPair(name, def),
32    fFunctions("AliRsnFunction", 0)
33 {
34 //
35 // Default constructor
36 //
37
38    AliDebug(AliLog::kDebug + 2, "<-");
39    AliDebug(AliLog::kDebug + 2, "->");
40 }
41
42 //_____________________________________________________________________________
43 AliRsnPairFunctions::AliRsnPairFunctions(const AliRsnPairFunctions& copy) :
44    AliRsnPair(copy),
45    fFunctions(copy.fFunctions)
46 {
47 //
48 // Default constructor
49 //
50
51    AliDebug(AliLog::kDebug + 2, "<-");
52    AliDebug(AliLog::kDebug + 2, "->");
53 }
54
55 //_____________________________________________________________________________
56 AliRsnPairFunctions& AliRsnPairFunctions::operator=(const AliRsnPairFunctions& copy)
57 {
58    AliRsnPair::operator=(copy);
59
60    Int_t i, n = copy.fFunctions.GetEntries();
61    for (i = 0; i < n; i++) {
62       AliRsnFunction *fcn = (AliRsnFunction*)copy.fFunctions[i];
63       if (fcn) AddFunction(fcn);
64    }
65
66    return (*this);
67 }
68
69 //_____________________________________________________________________________
70 AliRsnPairFunctions::~AliRsnPairFunctions()
71 {
72 //
73 // Destructor
74 //
75
76    AliDebug(AliLog::kDebug + 2, "<-");
77    AliDebug(AliLog::kDebug + 2, "->");
78 }
79
80 //_____________________________________________________________________________
81 void AliRsnPairFunctions::Compute()
82 {
83 //
84 // Makes computations using the two passed daughter objects.
85 // Checks all cuts and then computes the corresponding pair object
86 // and then fill the list of required values using it.
87 //
88
89    AliDebug(AliLog::kDebug + 2, "<-");
90
91    TObjArrayIter   nextFcn(&fFunctions);
92    AliRsnFunction *fcn = 0x0;
93
94    while ((fcn = (AliRsnFunction*)nextFcn())) {
95       fcn->SetPairDef(fPairDef);
96       fcn->SetPair(&fMother);
97       fcn->Fill();
98    }
99
100    AliDebug(AliLog::kDebug + 2, "->");
101 }
102
103 //_____________________________________________________________________________
104 void AliRsnPairFunctions::Init(const char *prefix, TList *list)
105 {
106 //
107 // Generates needed histograms, giving them a name based on
108 // the flags defined here, on the pair definition, and attaches
109 // a prefix to it, according to the argument.
110 //
111 // All generated histograms are stored into the output TList.
112 //
113
114    AliDebug(AliLog::kDebug + 2, "<-");
115
116    Int_t  i;
117    TString hName("");
118    AliRsnFunction *fcn = 0;
119    for (i = 0; i < fFunctions.GetEntries(); i++) {
120       fcn = (AliRsnFunction*)fFunctions.At(i);
121       hName += prefix;
122       hName += '_';
123       hName += GetName();
124       hName += '_';
125       hName += fcn->GetName();
126       if (fcn->IsUsingTH1()) list->Add(fcn->CreateHistogram(hName.Data(), ""));
127       else list->Add(fcn->CreateHistogramSparse(hName.Data(), ""));
128    }
129
130    AliDebug(AliLog::kDebug + 2, "->");
131 }
132
133 //_____________________________________________________________________________
134 void AliRsnPairFunctions::AddFunction(AliRsnFunction *const fcn)
135 {
136 //
137 // Adds a new computing function
138 //
139
140    AliDebug(AliLog::kDebug + 2, "<-");
141
142    fFunctions.Print();
143    Int_t size = fFunctions.GetEntries();
144    new(fFunctions[size]) AliRsnFunction(*fcn);
145
146    AliDebug(AliLog::kDebug + 2, "->");
147 }