retrieve only the needed members of the tree

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

diff --git a/PWG0/dNdEta/AlidNdEtaAnalysisMCSelector.cxx b/PWG0/dNdEta/AlidNdEtaAnalysisMCSelector.cxx
index c4cf94fdf85fe91b6072d6b5f80c4281313642f7..f62fb3f8405eaae5c7e8fd268a44d8c2f88ab830 100644 (file)
--- a/PWG0/dNdEta/AlidNdEtaAnalysisMCSelector.cxx
+++ b/PWG0/dNdEta/AlidNdEtaAnalysisMCSelector.cxx
@@ -32,6 +32,13 @@ AlidNdEtaAnalysisMCSelector::~AlidNdEtaAnalysisMCSelector()
   //
 }
 
+void AlidNdEtaAnalysisMCSelector::Init(TTree *tree)
+{
+   AlidNdEtaAnalysisSelector::Init(tree);
+
+  tree->SetBranchStatus("ESD", 0);
+}
+
 Bool_t AlidNdEtaAnalysisMCSelector::Process(Long64_t entry)
 {
   //
@@ -49,6 +56,13 @@ Bool_t AlidNdEtaAnalysisMCSelector::Process(Long64_t entry)
   TArrayF vtxMC(3);
   genHeader->PrimaryVertex(vtxMC);
 
+  particleTree->SetBranchStatus("*", 0);
+  particleTree->SetBranchStatus("fDaughter[2]", 1);
+  particleTree->SetBranchStatus("fPdgCode", 1);
+  particleTree->SetBranchStatus("fPx", 1);
+  particleTree->SetBranchStatus("fPy", 1);
+  particleTree->SetBranchStatus("fPz", 1);
+  
   TParticle* particle = 0;
   particleTree->SetBranchAddress("Particles", &particle);