retrieve only the needed members of the tree

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);
 

diff --git a/PWG0/dNdEta/AlidNdEtaAnalysisMCSelector.h b/PWG0/dNdEta/AlidNdEtaAnalysisMCSelector.h
index 2ff873d11eb5f41b74d9ea61089937ea43c1e90e..ca1a0d09608949d7f3cafac5234c21524fdae141 100644 (file)
--- a/PWG0/dNdEta/AlidNdEtaAnalysisMCSelector.h
+++ b/PWG0/dNdEta/AlidNdEtaAnalysisMCSelector.h
@@ -8,6 +8,7 @@ class AlidNdEtaAnalysisMCSelector : public AlidNdEtaAnalysisSelector {
     AlidNdEtaAnalysisMCSelector(TTree *tree=0);
     virtual ~AlidNdEtaAnalysisMCSelector();
 
+    virtual void    Init(TTree *tree);
     virtual Bool_t  Process(Long64_t entry);
 
  protected: