ATLAS Offline Software
Public Member Functions | Public Attributes | List of all members
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 24 of file eFEXNTuple.py.

Constructor & Destructor Documentation

◆ __init__()

def eFEXNTuple.eFEXNTupleLoader.__init__ (   self,
  filename 
)

Definition at line 25 of file eFEXNTuple.py.

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

Member Function Documentation

◆ __len__()

def eFEXNTuple.eFEXNTupleLoader.__len__ (   self)

Definition at line 60 of file eFEXNTuple.py.

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

◆ entries()

def eFEXNTuple.eFEXNTupleLoader.entries (   self)

Definition at line 29 of file eFEXNTuple.py.

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

Member Data Documentation

◆ tfile

eFEXNTuple.eFEXNTupleLoader.tfile

Definition at line 26 of file eFEXNTuple.py.

◆ tree

eFEXNTuple.eFEXNTupleLoader.tree

Definition at line 27 of file eFEXNTuple.py.


The documentation for this class was generated from the following file:
CaloCellPos2Ntuple.int
int
Definition: CaloCellPos2Ntuple.py:24
Get
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,...
Definition: comparitor.cxx:178
python.processes.powheg.ZZ.ZZ.__init__
def __init__(self, base_directory, **kwargs)
Constructor: all process options are set here.
Definition: ZZ.py:18
generate::GetEntries
double GetEntries(TH1D *h, int ilow, int ihi)
Definition: rmsFrac.cxx:20
entries
double entries
Definition: listroot.cxx:49