eFEXNTuple.eFEXNTupleLoader Class Reference

Collaboration diagram for eFEXNTuple.eFEXNTupleLoader:

Public Member Functions
def	__init__ (self, filename)
def	entries (self)
def	__len__ (self)

Public Attributes
	tfile
	tree

Detailed Description

Definition at line 23 of file eFEXNTuple.py.

Constructor & Destructor Documentation

__init__()

def eFEXNTuple.eFEXNTupleLoader.__init__	(		self,
			filename
	)

Definition at line 24 of file eFEXNTuple.py.

    def __init__(self, filename):
        self.tfile = ROOT.TFile(filename)
        self.tree = self.tfile.Get("data;1")
 

Member Function Documentation

__len__()

def eFEXNTuple.eFEXNTupleLoader.__len__

(

self

)

Definition at line 59 of file eFEXNTuple.py.

    def __len__(self):
        return self.tree.GetEntries() - 1
 

entries()

def eFEXNTuple.eFEXNTupleLoader.entries

(

self

)

Definition at line 28 of file eFEXNTuple.py.

    def entries(self):
        for each_entry in self.tree:
            entry_dict = {}
            entry_dict["truth_e_ET"] = np.asarray(each_entry.truth_e_ET)/1000.
            entry_dict["truth_e_phi"] = np.asarray(each_entry.truth_e_phi)
            entry_dict["truth_e_eta"] = np.asarray(each_entry.truth_e_eta)
            entry_dict["eg_eta"] = np.asarray(each_entry.eg_eta) * 0.1 #+ 0.05
            entry_dict["eg_phi"] = np.asarray(each_entry.eg_phi) * math.pi/32. #+ math.pi / 64# - math.pi)
            entry_dict["eg_ET"] = np.asarray(each_entry.eg_ET)/1000.
            entry_dict["eg_wstotnum"] = np.asarray(each_entry.eg_wstotnum)
            entry_dict["eg_wstotden"] = np.asarray(each_entry.eg_wstotden)
            entry_dict["eg_rhadnum"] = np.asarray(each_entry.eg_wstotnum)
            entry_dict["eg_rhadden"] = np.asarray(each_entry.eg_wstotden)
            entry_dict["eg_retanum"] = np.asarray(each_entry.eg_wstotnum)
            entry_dict["eg_retaden"] = np.asarray(each_entry.eg_wstotden)
            entry_dict["eg_haveseed"] = np.asarray(each_entry.eg_haveseed).astype(dtype=bool)
            entry_dict["eg_nTOBs"] = int(each_entry.eg_nTOBs)
            
            entry_dict["truth_tauvisible_ET"] = np.asarray(each_entry.truth_tauvisible_ET)/1000.
            entry_dict["truth_tauvisible_phi"] = np.asarray(each_entry.truth_tauvisible_phi)
            entry_dict["truth_tauvisible_eta"] = np.asarray(each_entry.truth_tauvisible_eta)
 
            entry_dict["eg_tauOregon_Et"] = np.asarray(each_entry.eg_tauOregon_Et)/1000.
            entry_dict["eg_tauOregon_Iso"] = np.asarray(each_entry.eg_tauOregon_Iso)
            entry_dict["eg_tauTLV_Et"] = np.asarray(each_entry.eg_tauTLV_Et)/1000.
            entry_dict["eg_tauTLV_Iso"] = np.asarray(each_entry.eg_tauTLV_Iso)
            # print(entry_dict["truth_e_phi"][0], entry_dict["truth_e_eta"][0])
            # i = np.argmax(entry_dict["eg_ET"])
            # print(entry_dict["eg_phi"][i], entry_dict["eg_eta"][i])
            yield eFEXAnalyzer(entry_dict)
 

Member Data Documentation

tfile

eFEXNTuple.eFEXNTupleLoader.tfile

Definition at line 25 of file eFEXNTuple.py.

tree

eFEXNTuple.eFEXNTupleLoader.tree

Definition at line 26 of file eFEXNTuple.py.

---

The documentation for this class was generated from the following file:

• eFEXNTuple.py