o) adding log tags to all files

[u/mrichter/AliRoot.git] / PWG0 / dNdEta / AlidNdEtaAnalysisSelector.cxx

diff --git a/PWG0/dNdEta/AlidNdEtaAnalysisSelector.cxx b/PWG0/dNdEta/AlidNdEtaAnalysisSelector.cxx
index 75076d08fa0ca8f39c811db52affa0045cd488ee..26e2a6ea9cf34d68bcf9adfeade836f519a64505 100644 (file)
--- a/PWG0/dNdEta/AlidNdEtaAnalysisSelector.cxx
+++ b/PWG0/dNdEta/AlidNdEtaAnalysisSelector.cxx
@@ -1,24 +1,23 @@
+/* $Id$ */
+
 #include "AlidNdEtaAnalysisSelector.h"
 
 #include <TStyle.h>
 #include <TSystem.h>
 #include <TCanvas.h>
 #include <TVector3.h>
+#include <TH2F.h>
 
 #include <AliLog.h>
 #include <AliGenEventHeader.h>
 
-#include "esdTrackCuts/AliESDtrackCuts.h"
-#include "dNdEtaCorrection.h"
 #include "dNdEtaAnalysis.h"
 
 ClassImp(AlidNdEtaAnalysisSelector)
 
-AlidNdEtaAnalysisSelector::AlidNdEtaAnalysisSelector(TTree *) :
+AlidNdEtaAnalysisSelector::AlidNdEtaAnalysisSelector() :
   AliSelector(),
-  fEsdTrackCuts(0),
   fdNdEtaAnalysis(0),
-  fdNdEtaCorrection(0),
   fdNdEtaAnalysisFinal(0)
 {
   //
@@ -45,102 +44,6 @@ void AlidNdEtaAnalysisSelector::SlaveBegin(TTree * tree)
   AliSelector::SlaveBegin(tree);
 
   fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta");
-
-  if (fChain)
-  {
-    fEsdTrackCuts = dynamic_cast<AliESDtrackCuts*> (fChain->GetUserInfo()->FindObject("AliESDtrackCuts"));
-    fdNdEtaCorrection = dynamic_cast<dNdEtaCorrection*> (fChain->GetUserInfo()->FindObject("dNdEtaCorrection"));
-  }
-
-  if (!fEsdTrackCuts)
-     AliDebug(AliLog::kError, "ERROR: Could not read EsdTrackCuts from user info.");
-
-  if (!fEsdTrackCuts)
-     AliDebug(AliLog::kWarning, "ERROR: Could not read dNdEtaCorrection from user info.");
-
-  AliLog::SetClassDebugLevel("AliESDtrackCuts", 1);
-}
-
-Bool_t AlidNdEtaAnalysisSelector::Process(Long64_t entry)
-{
-  // The Process() function is called for each entry in the tree (or possibly
-  // keyed object in the case of PROOF) to be processed. The entry argument
-  // specifies which entry in the currently loaded tree is to be processed.
-  // It can be passed to either TTree::GetEntry() or TBranch::GetEntry()
-  // to read either all or the required parts of the data. When processing
-  // keyed objects with PROOF, the object is already loaded and is available
-  // via the fObject pointer.
-  //
-  // This function should contain the "body" of the analysis. It can contain
-  // simple or elaborate selection criteria, run algorithms on the data
-  // of the event and typically fill histograms.
-
-  // WARNING when a selector is used with a TChain, you must use
-  //  the pointer to the current TTree to call GetEntry(entry).
-  //  The entry is always the local entry number in the current tree.
-  //  Assuming that fChain is the pointer to the TChain being processed,
-  //  use fChain->GetTree()->GetEntry(entry).
-
-  if (AliSelector::Process(entry) == kFALSE)
-    return kFALSE;
-
-  // Check prerequisites
-  if (!fESD || !fEsdTrackCuts)
-    return kFALSE;
-
-  // ########################################################
-  // get the EDS vertex
-  const AliESDVertex* vtxESD = fESD->GetVertex();
-
-  // the vertex should be reconstructed
-  if (strcmp(vtxESD->GetName(),"default")==0)
-    return kTRUE;
-
-  Double_t vtx_res[3];
-  vtx_res[0] = vtxESD->GetXRes();
-  vtx_res[1] = vtxESD->GetYRes();
-  vtx_res[2] = vtxESD->GetZRes();
-
-  // the resolution should be reasonable???
-  if (vtx_res[2]==0 || vtx_res[2]>0.1)
-    return kTRUE;
-
-  Double_t vtx[3];
-  vtxESD->GetXYZ(vtx);
-
-  // ########################################################
-  // loop over esd tracks
-  Int_t nTracks = fESD->GetNumberOfTracks();
-  for (Int_t t=0; t<nTracks; t++)
-  {
-    AliESDtrack* esdTrack = fESD->GetTrack(t);
-    if (!esdTrack)
-    {
-      AliDebug(AliLog::kError, Form("ERROR: Could not retrieve track %d.", t));
-      continue;
-    }
-
-    // cut the esd track?
-    if (!fEsdTrackCuts->AcceptTrack(esdTrack))
-      continue;
-
-    Double_t p[3];
-    esdTrack->GetConstrainedPxPyPz(p); // ### TODO or GetInnerPxPyPy / GetOuterPxPyPy
-    TVector3 vector(p);
-
-    Float_t theta = vector.Theta();
-    Float_t eta   = -TMath::Log(TMath::Tan(theta/2.));
-
-    Float_t correction = fdNdEtaCorrection->GetCorrection(vtx[2], eta);
-
-    fdNdEtaAnalysis->FillTrack(vtx[2], eta, correction);
-
-  } // end of track loop
-
-  // for event count per vertex
-  fdNdEtaAnalysis->FillEvent(vtx[2]);
-
-  return kTRUE;
 }
 
 void AlidNdEtaAnalysisSelector::SlaveTerminate()
@@ -193,17 +96,18 @@ void AlidNdEtaAnalysisSelector::Terminate()
   fdNdEtaAnalysisFinal->Finish();
 
   TFile* fout = new TFile("out.root","RECREATE");
-
-  if (fEsdTrackCuts)
-    fEsdTrackCuts->SaveHistograms("esd_tracks_cuts");
-
-  if (fdNdEtaCorrection)
-    fdNdEtaCorrection->SaveHistograms();
-
-  fdNdEtaAnalysisFinal->SaveHistograms();
-
+  WriteObjects();
   fout->Write();
   fout->Close();
 
   fdNdEtaAnalysisFinal->DrawHistograms();
 }
+
+void AlidNdEtaAnalysisSelector::WriteObjects()
+{
+  // Write objects to output file
+  // this is an extra function to be overloaded...
+  //
+
+  fdNdEtaAnalysisFinal->SaveHistograms();
+}