--- UserCode/LEP3/lepTree.py	2012/05/27 21:18:51	1.1
+++ UserCode/LEP3/lepTree.py	2012/08/31 14:33:22	1.5
@@ -1,18 +1,42 @@
 #!/usr/bin/env python
 import ROOT
+from array import array
 
 class lepTree:
     def __init__(self):
         self.tree = ROOT.TTree("lep3Tree", "lep3Tree")
+        # doubles
+        self.leaves = {}
+        for label in ['processID','type',
+                      'lq1','lq2','lq3','lq4',
+                      'mu1IsoCH','mu1IsoEM','mu1IsoNH',
+                      'mu2IsoCH','mu2IsoEM','mu2IsoNH',
+                      'mu3IsoCH','mu3IsoEM','mu3IsoNH',
+                      'mu4IsoCH','mu4IsoEM','mu4IsoNH',
+                      'ele1IsoCH','ele1IsoEM','ele1IsoNH',
+                      'ele2IsoCH','ele2IsoEM','ele2IsoNH',
+                      'ele3IsoCH','ele3IsoEM','ele3IsoNH',
+                      'ele4IsoCH','ele4IsoEM','ele4IsoNH',
+                      'visibleE','chargeMulti',
+                      ]:
+            self.leaves[label] = array('d',[-999]) 
         # four vectors
         self.fourVectorLeaves = {}
-        for label in ['muon1', 'muon2','electron1', 'electron2','dilep']:
+        for label in ['muonicZ1','muonicZ2','electronicZ1','electronicZ2',
+                      'lepton1', 'lepton2','lepton3','lepton4',
+                      'photon1','photon2',
+                      'jet1','jet2','jet3','jet4','jet5','jet6',
+                      'missingE']:
             self.fourVectorLeaves[label] = ROOT.TLorentzVector()
         self.setBranches()
         
     def setBranches(self):
+        for leaf in self.leaves:
+            self.tree.Branch(leaf,self.leaves[leaf],leaf+'/D') 
         for leaf in self.fourVectorLeaves:
             self.tree.Branch(leaf, self.fourVectorLeaves[leaf])
 
     def reset(self):
+        for leaf in self.leaves:
+            self.leaves[leaf][0] = -999
         for leaf in self.fourVectorLeaves: self.fourVectorLeaves[leaf].SetPxPyPzE(0,0,0,0)