Changeset 13 for trunk/modules/MadGraphMatchingTreeWriter.h

Timestamp:

Sep 15, 2008, 10:59:33 AM (16 years ago)

Author:

Pavel Demin

Message:

add configuration for branches

File:

• trunk/modules/MadGraphMatchingTreeWriter.h (modified) (2 diffs)

trunk/modules/MadGraphMatchingTreeWriter.h

@@ -4 +4 @@
 #include "ExRootAnalysis/ExRootModule.h"
 
-#include "TString.h"
-
 #include <map>
 
-class TH1;
+class TClass;
 class TIterator;
-class TObjArray;
-class TClonesArray;
+class ExRootTreeBranch;
 
 class MadGraphMatchingTreeWriter: public ExRootModule
@@ -26 +23 @@
 private:
 
-  Double_t fJetPTMin, fJetEtaMax;
+  void ProcessPartons(ExRootTreeBranch *branch, TIterator *iterator);
+  void ProcessMatching(ExRootTreeBranch *branch, TIterator *iterator);
+  void ProcessJets(ExRootTreeBranch *branch, TIterator *iterator);
 
-  const TObjArray *fInputArrayPartonJets; //!
-  const TObjArray *fInputArrayHadronJets; //!
-  const TObjArray *fInputArrayMatching; //!
-  const TObjArray *fInputArrayPartons; //!
+  Double_t fJetPTMin, fJetEtaMax; //!
+#ifndef __CINT__
+  typedef void (MadGraphMatchingTreeWriter::*TProcessMethod)(ExRootTreeBranch*, TIterator*); //!
 
-  TIterator *fItInputArrayPartonJets; //!
-  TIterator *fItInputArrayHadronJets; //!
-  TIterator *fItInputArrayMatching; //!
-  TIterator *fItInputArrayPartons; //!
-  
-  ExRootTreeBranch *fBranchPartonJets; //!
-  ExRootTreeBranch *fBranchHadronJets; //!
-  ExRootTreeBranch *fBranchMatching; //!
-  ExRootTreeBranch *fBranchPartons; //!
+  typedef std::map<ExRootTreeBranch *, std::pair<TProcessMethod, TIterator*> > TBranchMap; //!
 
+  TBranchMap fBranchMap; //!
+
+  std::map< TClass *, TProcessMethod > fClassMap; //!
+#endif
   ClassDef(MadGraphMatchingTreeWriter, 1)
 };