ATLAS Offline Software
Loading...
Searching...
No Matches
eFEXNTuple.eFEXNTupleLoader Class Reference
Collaboration diagram for eFEXNTuple.eFEXNTupleLoader:

Public Member Functions

 __init__ (self, filename)
 entries (self)
 __len__ (self)

Public Attributes

 tfile = ROOT.TFile(filename)
 tree = self.tfile.Get("data;1")

Detailed Description

Definition at line 23 of file eFEXNTuple.py.

Constructor & Destructor Documentation

◆ __init__()

eFEXNTuple.eFEXNTupleLoader.__init__ ( self,
filename )

Definition at line 24 of file eFEXNTuple.py.

24 def __init__(self, filename):
25 self.tfile = ROOT.TFile(filename)
26 self.tree = self.tfile.Get("data;1")
27
T * Get(TFile &f, const std::string &n, const std::string &dir="", const chainmap_t *chainmap=0, std::vector< std::string > *saved=0)
get a histogram given a path, and an optional initial directory if histogram is not found,...

Member Function Documentation

◆ __len__()

eFEXNTuple.eFEXNTupleLoader.__len__ ( self)

Definition at line 59 of file eFEXNTuple.py.

59 def __len__(self):
60 return self.tree.GetEntries() - 1
61
TGraphErrors * GetEntries(TH2F *histo)

◆ entries()

eFEXNTuple.eFEXNTupleLoader.entries ( self)

Definition at line 28 of file eFEXNTuple.py.

28 def entries(self):
29 for each_entry in self.tree:
30 entry_dict = {}
31 entry_dict["truth_e_ET"] = np.asarray(each_entry.truth_e_ET)/1000.
32 entry_dict["truth_e_phi"] = np.asarray(each_entry.truth_e_phi)
33 entry_dict["truth_e_eta"] = np.asarray(each_entry.truth_e_eta)
34 entry_dict["eg_eta"] = np.asarray(each_entry.eg_eta) * 0.1 #+ 0.05
35 entry_dict["eg_phi"] = np.asarray(each_entry.eg_phi) * math.pi/32. #+ math.pi / 64# - math.pi)
36 entry_dict["eg_ET"] = np.asarray(each_entry.eg_ET)/1000.
37 entry_dict["eg_wstotnum"] = np.asarray(each_entry.eg_wstotnum)
38 entry_dict["eg_wstotden"] = np.asarray(each_entry.eg_wstotden)
39 entry_dict["eg_rhadnum"] = np.asarray(each_entry.eg_wstotnum)
40 entry_dict["eg_rhadden"] = np.asarray(each_entry.eg_wstotden)
41 entry_dict["eg_retanum"] = np.asarray(each_entry.eg_wstotnum)
42 entry_dict["eg_retaden"] = np.asarray(each_entry.eg_wstotden)
43 entry_dict["eg_haveseed"] = np.asarray(each_entry.eg_haveseed).astype(dtype=bool)
44 entry_dict["eg_nTOBs"] = int(each_entry.eg_nTOBs)
45
46 entry_dict["truth_tauvisible_ET"] = np.asarray(each_entry.truth_tauvisible_ET)/1000.
47 entry_dict["truth_tauvisible_phi"] = np.asarray(each_entry.truth_tauvisible_phi)
48 entry_dict["truth_tauvisible_eta"] = np.asarray(each_entry.truth_tauvisible_eta)
49
50 entry_dict["eg_tauOregon_Et"] = np.asarray(each_entry.eg_tauOregon_Et)/1000.
51 entry_dict["eg_tauOregon_Iso"] = np.asarray(each_entry.eg_tauOregon_Iso)
52 entry_dict["eg_tauTLV_Et"] = np.asarray(each_entry.eg_tauTLV_Et)/1000.
53 entry_dict["eg_tauTLV_Iso"] = np.asarray(each_entry.eg_tauTLV_Iso)
54 # print(entry_dict["truth_e_phi"][0], entry_dict["truth_e_eta"][0])
55 # i = np.argmax(entry_dict["eg_ET"])
56 # print(entry_dict["eg_phi"][i], entry_dict["eg_eta"][i])
57 yield eFEXAnalyzer(entry_dict)
58
double entries
Definition listroot.cxx:49

Member Data Documentation

◆ tfile

eFEXNTuple.eFEXNTupleLoader.tfile = ROOT.TFile(filename)

Definition at line 25 of file eFEXNTuple.py.

◆ tree

eFEXNTuple.eFEXNTupleLoader.tree = self.tfile.Get("data;1")

Definition at line 26 of file eFEXNTuple.py.


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