- first version of PWGTRD lib

[u/mrichter/AliRoot.git] / PWG / TRD / AliAnalysisTaskTRDgtuSim.cxx

diff --git a/PWG/TRD/AliAnalysisTaskTRDgtuSim.cxx b/PWG/TRD/AliAnalysisTaskTRDgtuSim.cxx
new file mode 100644 (file)
index 0000000..2e9f259
--- /dev/null
+++ b/PWG/TRD/AliAnalysisTaskTRDgtuSim.cxx
@@ -0,0 +1,170 @@
+#include "AliLog.h"
+
+// analysis framework
+#include "AliAnalysisManager.h"
+#include "AliInputEventHandler.h"
+
+// ESD stuff
+#include "AliESDEvent.h"
+#include "AliESDInputHandler.h"
+
+// GTU simulation
+#include "AliTRDgtuParam.h"
+#include "AliTRDgtuSim.h"
+
+#include "AliAnalysisTaskTRDgtuSim.h"
+#include "AliTRDonlineTrackMatching.h"
+
+AliAnalysisTaskTRDgtuSim::AliAnalysisTaskTRDgtuSim(const char *name) :
+  AliAnalysisTaskSE(name),
+  fOutputList(0x0),
+  fHistStat(0x0),
+  fHistDeltaA(0x0),
+  fGtuSim(new AliTRDgtuSim()),
+  fTrackletLabel(-2),
+  fLabel(-10),
+  fDeltaY(12),
+  fDeltaAlpha(16),
+  fLimitNoTracklets(kTRUE),
+  fMaxNoTracklets(62)
+{
+  // ctor
+
+  DefineOutput(1, TList::Class());
+}
+
+AliAnalysisTaskTRDgtuSim::~AliAnalysisTaskTRDgtuSim()
+{
+  // dtor
+
+  delete fGtuSim;
+}
+
+Bool_t AliAnalysisTaskTRDgtuSim::Notify()
+{
+  // Implemented Notify()
+
+  return AliAnalysisTaskSE::Notify();
+}
+
+void AliAnalysisTaskTRDgtuSim::UserCreateOutputObjects()
+{
+  // create output objects
+
+  OpenFile(1);
+  fOutputList = new TList;
+  fOutputList->SetOwner(kTRUE);
+
+  fHistStat = new TH1F("stat", "stat", 4, .5, 4.5);
+  fHistStat->GetXaxis()->SetBinLabel(1, "identical");
+  fHistStat->GetXaxis()->SetBinLabel(2, "mismatch");
+  fHistStat->GetXaxis()->SetBinLabel(3, "sim only");
+  fHistStat->GetXaxis()->SetBinLabel(4, "raw only");
+  fOutputList->Add(fHistStat);
+
+  fHistDeltaA = new TH1F("deltaA", "delta A", 100, -100., 100.);
+  fOutputList->Add(fHistDeltaA);
+
+  PostData(1, fOutputList);
+}
+
+void AliAnalysisTaskTRDgtuSim::UserExec(Option_t * /* option */)
+{
+  // perform actual analysis
+
+  AliESDEvent *esdEvent =
+    dynamic_cast<AliESDEvent*>(this->InputEvent());
+
+  if (!esdEvent)
+    return;
+
+  AliTRDgtuParam::SetDeltaY(fDeltaY);
+  AliTRDgtuParam::SetDeltaAlpha(fDeltaAlpha);
+  AliTRDgtuParam::SetLimitNoTracklets(fLimitNoTracklets);
+  AliTRDgtuParam::SetMaxNoTracklets(fMaxNoTracklets);
+
+  fGtuSim->RunGTU(0x0, esdEvent, fTrackletLabel, fLabel);
+
+  AliTRDonlineTrackMatching trdMatch;
+  trdMatch.ProcessEvent(esdEvent, kTRUE, fLabel);
+
+  // compare to hardware tracks
+  Check(fLabel, -3);
+
+  PostData(1, fOutputList);
+}
+
+
+void AliAnalysisTaskTRDgtuSim::Terminate(const Option_t*)
+{
+
+}
+
+void AliAnalysisTaskTRDgtuSim::Check(Int_t label, Int_t labelRef)
+{
+  // compare tracks with label label with tracks with label labelRef
+
+  AliESDEvent *esdEvent =
+    dynamic_cast<AliESDEvent*>(this->InputEvent());
+
+  if (!esdEvent)
+    return;
+
+  TList tracksCheck;
+  TList tracksRef;
+
+  Int_t nTracks = esdEvent->GetNumberOfTrdTracks();
+  for (Int_t iTrack = 0; iTrack < nTracks; ++iTrack) {
+    AliESDTrdTrack *trk = esdEvent->GetTrdTrack(iTrack);
+    if (trk->GetLabel() == label)
+      tracksCheck.Add(trk);
+    if (trk->GetLabel() == labelRef)
+      tracksRef.Add(trk);
+  }
+
+  TIter iterTrack(&tracksCheck);
+  while (AliESDTrdTrack *trk = (AliESDTrdTrack*) iterTrack()) {
+    TIter iterTrackRef(&tracksRef);
+    Bool_t foundMatch = kFALSE;
+    while (AliESDTrdTrack *trkRef = (AliESDTrdTrack*) iterTrackRef()) {
+      // check for common tracklets
+      Bool_t commonTracklet = kFALSE;
+      Bool_t allEqual = kTRUE;
+      for (Int_t iLayer = 0; iLayer < 6; ++iLayer) {
+       if (trk->GetTracklet(iLayer)) {
+         if (trk->GetTracklet(iLayer) == trkRef->GetTracklet(iLayer))
+           commonTracklet = kTRUE;
+         else
+           allEqual = kFALSE;
+       }
+       else if (trkRef->GetTracklet(iLayer))
+         allEqual = kFALSE;
+      }
+      if (commonTracklet) {
+       // tracks with a common tracklet should be identical
+       if (allEqual) {
+         // identical track composition
+         fHistStat->Fill(1);
+         Int_t deltaA = trk->GetA() - trkRef->GetA();
+         fHistDeltaA->Fill(deltaA);
+       }
+       else {
+         // track with different tracklets
+         fHistStat->Fill(2);
+       }
+       tracksRef.Remove(trkRef);
+       foundMatch = kTRUE;
+       break;
+      }
+    }
+    if (!foundMatch) {
+      // unmatched sim track
+      fHistStat->Fill(3);
+    }
+  }
+  TIter iterTrackRef(&tracksRef);
+  while (iterTrackRef()) {
+    // unmatched raw track
+    fHistStat->Fill(4);
+  }
+}