// revised by : A. Pulvirenti [alberto.pulvirenti@ct.infn.it]
//
-#include <Riostream.h>
+#include <TH1.h>
#include <TROOT.h>
#include "AliLog.h"
+#include "AliStack.h"
#include "AliVEvent.h"
#include "AliMCEvent.h"
#include "AliRsnEvent.h"
-#include "AliRsnPair.h"
+#include "AliRsnPairFunctions.h"
+#include "AliRsnPairNtuple.h"
#include "AliRsnAnalysisManager.h"
//_____________________________________________________________________________
AliRsnAnalysisManager::AliRsnAnalysisManager(const char*name) :
TNamed(name, ""),
- fPairs(0)
+ fList(0x0),
+ fPairs(0),
+ fGlobalTrackCuts()
{
//
// Default constructor
//
+}
- AliDebug(AliLog::kDebug+2, "<-");
- AliDebug(AliLog::kDebug+2, "->");
+//_____________________________________________________________________________
+AliRsnAnalysisManager::AliRsnAnalysisManager(const AliRsnAnalysisManager& copy) :
+ TNamed(copy),
+ fList(copy.fList),
+ fPairs(copy.fPairs),
+ fGlobalTrackCuts(copy.fGlobalTrackCuts)
+{
+//
+// Copy constructor
+//
+}
+
+//_____________________________________________________________________________
+AliRsnAnalysisManager& AliRsnAnalysisManager::operator=(const AliRsnAnalysisManager& copy)
+{
+//
+// Assignment operator
+//
+
+ TNamed::operator=(copy);
+
+ fList = copy.fList;
+ fPairs = copy.fPairs;
+ fGlobalTrackCuts = copy.fGlobalTrackCuts;
+
+ return (*this);
}
//_____________________________________________________________________________
AliDebug(AliLog::kDebug+2,"<-");
- if (!pair) {
+ if (!pair)
+ {
AliWarning(Form("AliRsnPairManager is %p. Skipping ...", pair));
return;
}
AliDebug(AliLog::kDebug+1, Form("Adding %s [%d]...", pair->GetName(), fPairs.GetEntries()));
- fPairs.Add((AliRsnPair*)pair);
+ fPairs.Add(pair);
AliDebug(AliLog::kDebug+2,"->");
}
AliDebug(AliLog::kDebug+2,"<-");
-// TList *list = new TList();
-// list->SetName(GetName());
-// list->SetOwner();
-
AliRsnPair *pair = 0;
TObjArrayIter next(&fPairs);
Int_t i = 0;
while ((pair = (AliRsnPair*)next()))
{
- if (!pair) continue;
AliDebug(AliLog::kDebug+1, Form("InitAllPairs of the PairManager(%s) [%d] ...", pair->GetName(), i++));
- pair->Init("", list);
+// pair->Init("", list);
+ pair->Init(GetName(), list);
+
+ // add a counter for used/unused events for each pair
+ TH1I *hPairUsed = new TH1I(Form("%s_%s_USED", GetName(),pair->GetName()), "Used events for pair", 2, 0, 2);
+ list->Add(hPairUsed);
}
+
+ fList = list;
+
AliDebug(AliLog::kDebug+2, "->");
-// return list;
}
//_____________________________________________________________________________
-void AliRsnAnalysisManager::ProcessAllPairs(AliRsnEvent *ev0, AliRsnEvent *ev1)
+void AliRsnAnalysisManager::ProcessAllPairs()
{
//
// Process one or two events for all pair managers.
//
+ static Int_t evnum = 0;
+ evnum++;
+
AliDebug(AliLog::kDebug+2,"<-");
- if (!ev1) ev1 = ev0;
+ // skip if the global event pointers are NULL
+ if (!AliRsnEvent::IsCurrentEvent1()) return;
+ if (!AliRsnEvent::IsCurrentEvent2()) return;
- Int_t nTracks[2], nV0[2], nTot[2];
- nTracks[0] = ev0->GetRef()->GetNumberOfTracks();
- nV0[0] = ev0->GetRef()->GetNumberOfV0s();
- nTracks[1] = ev1->GetRef()->GetNumberOfTracks();
- nV0[1] = ev1->GetRef()->GetNumberOfV0s();
- nTot[0] = nTracks[0] + nV0[0];
- nTot[1] = nTracks[1] + nV0[1];
+ // for better readability, reference two pointers to the current events
+ AliRsnEvent *ev0 = AliRsnEvent::GetCurrentEvent1();
+ AliRsnEvent *ev1 = AliRsnEvent::GetCurrentEvent2();
- // external loop
- // joins the loop on tracks and v0s, by looping the indexes from 0
- // to the sum of them, and checking what to take depending of its value
- Int_t i0, i1, i;
+ // count total number of candidates per event
+ // (sum of tracks, V0s and cascades)
+ Int_t nTot[2];
+ nTot[0] = AliRsnEvent::GetCurrentEvent1()->GetAbsoluteSum();
+ nTot[1] = AliRsnEvent::GetCurrentEvent2()->GetAbsoluteSum();
+
+ // variables
+ Int_t i0, i1, i, start, index0, index1;
AliRsnDaughter daughter0, daughter1;
AliRsnPair *pair = 0x0;
TObjArrayIter next(&fPairs);
+ AliRsnDaughter::ERefType type0, type1;
+
+ // reset all counters which tell us
+ // how many entries were added now
+ while ((pair = (AliRsnPair*)next()))
+ {
+ pair->ResetCount();
+ }
+ // external loop
for (i0 = 0; i0 < nTot[0]; i0++)
{
// assign first track
- if (i0 < nTracks[0]) ev0->SetDaughter(daughter0, i0, AliRsnDaughter::kTrack);
- else ev0->SetDaughter(daughter0, i0 - nTracks[0], AliRsnDaughter::kV0);
+ if (!ev0->ConvertAbsoluteIndex(i0, index0, type0)) continue;
+ ev0->SetDaughter(daughter0, index0, type0);
+
+ // check global cuts
+ if (!fGlobalTrackCuts.IsSelected(&daughter0)) continue;
+
+ // define start depending if we are processing one or two events
+ start = (AliRsnEvent::SameEvent() ? i0 + 1 : 0);
// internal loop (same criterion)
- for (i1 = 0; i1 < nTot[1]; i1++)
+ for (i1 = start; i1 < nTot[1]; i1++)
{
// if looking same event, skip the case when the two indexes are equal
- if (ev0 == ev1 && i0 == i1) continue;
+ // if (AliRsnEvent::SameEvent() && i0 == i1) continue;
// assign second track
- if (i1 < nTracks[1]) ev1->SetDaughter(daughter1, i1, AliRsnDaughter::kTrack);
- else ev1->SetDaughter(daughter1, i1 - nTracks[1], AliRsnDaughter::kV0);
+ if (!ev1->ConvertAbsoluteIndex(i1, index1, type1)) continue;
+ ev1->SetDaughter(daughter1, index1, type1);
+
+ // check global cuts
+ if (!fGlobalTrackCuts.IsSelected(&daughter1)) continue;
// loop over all pairs and make computations
next.Reset();
// if the pair is a like-sign, skip the case when i1 < i0,
// in order not to double count each like-sign pair
// (equivalent to looping from i0+1 to ntracks)
- if (pair->GetPairDef()->IsLikeSign() && i1 < i0) continue;
+ // if (AliRsnEvent::SameEvent() && pair->GetPairDef()->IsLikeSign() && i1 < i0) continue;
// process the two tracks
- if (!pair->Fill(&daughter0, &daughter1, ev0, ev1)) continue;
- pair->Compute();
+ if (pair->Fill(&daughter0, &daughter1))
+ {
+ pair->Compute();
+ }
+ else if (pair->Fill(&daughter1, &daughter0))
+ {
+ pair->Compute();
+ }
}
}
}
+
+ // update all count histograms counters
+ next.Reset();
+ if (!fList) return;
+ while ((pair = (AliRsnPair*)next()))
+ {
+ TH1I *hist = (TH1I*)fList->FindObject(Form("_%s_USED", pair->GetName()));
+ if (!hist) continue;
+ if (pair->GetCount() > 0) hist->Fill(1); else hist->Fill(0);
+ }
AliDebug(AliLog::kDebug+2,"->");
}
//_____________________________________________________________________________
-void AliRsnAnalysisManager::ProcessAllPairsMC(AliRsnEvent *ev0, AliRsnEvent *ev1)
+void AliRsnAnalysisManager::ProcessAllPairsMC()
{
//
// Process one or two events for all pair managers.
AliDebug(AliLog::kDebug+2,"<-");
- if (!ev1) ev1 = ev0;
+ // skip if the global event pointers are NULL
+ if (!AliRsnEvent::IsCurrentEvent1()) return;
+ if (!AliRsnEvent::IsCurrentEvent2()) return;
+
+ // for better readability, reference two pointers to the current events
+ AliRsnEvent *ev0 = AliRsnEvent::GetCurrentEvent1();
+ AliRsnEvent *ev1 = AliRsnEvent::GetCurrentEvent2();
+ // this time the number of tracks comes from MC
Int_t nTracks[2];
nTracks[0] = ev0->GetRefMC()->GetNumberOfTracks();
nTracks[1] = ev1->GetRefMC()->GetNumberOfTracks();
// external loop
// joins the loop on tracks and v0s, by looping the indexes from 0
// to the sum of them, and checking what to take depending of its value
- Int_t i0, i1, i;
+ Int_t i0, i1, start, i;
+ Bool_t filled;
AliRsnDaughter daughter0, daughter1;
AliRsnPair *pair = 0x0;
TObjArrayIter next(&fPairs);
+ // reset all counters
+ while ((pair = (AliRsnPair*)next()))
+ {
+ pair->ResetCount();
+ }
+
for (i0 = 0; i0 < nTracks[0]; i0++)
{
+ // skip not physical primaries
+ if (!ev0->GetRefMCESD()->Stack()->IsPhysicalPrimary(i0)) continue;
+
// assign first track
- if (i0 < nTracks[0]) ev0->SetDaughter(daughter0, i0, AliRsnDaughter::kTrack);
- else ev0->SetDaughter(daughter0, i0 - nTracks[0], AliRsnDaughter::kV0);
+ ev0->SetDaughterMC(daughter0, i0);
+
+ // define start depending if we are processing one or two events
+ start = (AliRsnEvent::SameEvent() ? i0 + 1 : 0);
// internal loop (same criterion)
- for (i1 = 0; i1 < nTracks[1]; i1++)
+ for (i1 = start; i1 < nTracks[1]; i1++)
{
// if looking same event, skip the case when the two indexes are equal
- if (ev0 == ev1 && i0 == i1) continue;
+ if (AliRsnEvent::SameEvent() && i0 == i1) continue;
+
+ // skip not physical primaries
+ if (!ev1->GetRefMCESD()->Stack()->IsPhysicalPrimary(i1)) continue;
// assign second track
- if (i1 < nTracks[1]) ev1->SetDaughter(daughter1, i1, AliRsnDaughter::kTrack);
- else ev1->SetDaughter(daughter1, i1 - nTracks[1], AliRsnDaughter::kV0);
+ ev1->SetDaughterMC(daughter1, i1);
// loop over all pairs and make computations
next.Reset();
if (pair->GetPairDef()->IsLikeSign() && i1 < i0) continue;
// process the two tracks
- if (!pair->Fill(&daughter0, &daughter1, ev0, ev1)) continue;
+ filled = pair->Fill(&daughter0, &daughter1);
+ if (!filled) continue;
pair->Compute();
}
}
}
+
+ // update all count histograms counters
+ next.Reset();
+ if (!fList) return;
+ while ((pair = (AliRsnPair*)next()))
+ {
+ TH1I *hist = (TH1I*)fList->FindObject(Form("_%s_USED", pair->GetName()));
+ if (!hist) continue;
+ if (pair->GetCount() > 0) hist->Fill(1); else hist->Fill(0);
+ }
AliDebug(AliLog::kDebug+2,"->");
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff