CSV_InDetExporter.CSVDumper Class Reference

Collaboration diagram for CSV_InDetExporter.CSVDumper:

Public Member Functions
def	__init__ (self, inputAOD, outputDir, dict_variables_types, treename="CollectionTree", nEvents=-1)
def	WriteCSV (self, filename, dictionary)
def	ArrayFloat3_to_CppArray (self, ArrayFloat3)
def	ProcessEvent (self, evt)
def	Run (self)

Public Attributes
	tree
	n_entries
	outputDir
	nEvents
	dict_variables_types

Detailed Description

Definition at line 12 of file CSV_InDetExporter.py.

Constructor & Destructor Documentation

__init__()

def CSV_InDetExporter.CSVDumper.__init__	(		self,
			inputAOD,
			outputDir,
			dict_variables_types,
			treename = "CollectionTree",
			nEvents = -1
	)

Definition at line 14 of file CSV_InDetExporter.py.

    def __init__(self, inputAOD, outputDir, dict_variables_types, treename="CollectionTree", nEvents=-1):
 
        xAOD.Init()    # Setting up ROOT tools
        self.tree            = xAOD.MakeTransientTree( TFile(inputAOD), treename )
        self.n_entries       = self.tree.GetEntriesFast()
        self.outputDir       = outputDir
 
        if nEvents < 0 or nEvents > self.n_entries:   # Getting number of events to run
            self.nEvents = self.n_entries
        else:
            self.nEvents = nEvents
        print(f"Running on {self.nEvents} events")
 
        os.system("mkdir -p "+self.outputDir)   # Creating directory to store files
 
        # Container and Variables with their types that should be stored in csv format. Should be something like:
        #{
        #    "ContainerName": { "var1": "int",
        #                       "var2": "float",
        #                     }
        #}
        self.dict_variables_types = dict_variables_types
 

Member Function Documentation

ArrayFloat3_to_CppArray()

def CSV_InDetExporter.CSVDumper.ArrayFloat3_to_CppArray	(		self,
			ArrayFloat3
	)

Definition at line 45 of file CSV_InDetExporter.py.

    def ArrayFloat3_to_CppArray(self, ArrayFloat3): # Check ArrayFloat3 in https://gitlab.cern.ch/atlas/athena/-/blob/master/Event/xAOD/xAODInDetMeasurement/xAODInDetMeasurement/versions/SpacePoint_v1.h
        try:
            arr = ArrayFloat3.data()
        except Exception:
            arr = ArrayFloat3    
        arr.reshape((3,))
        return list(arr)
 
 

ProcessEvent()

def CSV_InDetExporter.CSVDumper.ProcessEvent	(		self,
			evt
	)

Definition at line 54 of file CSV_InDetExporter.py.

    def ProcessEvent(self, evt):
 
        # Getting aux variables based on the implementation of https://gitlab.cern.ch/atlas/athena/-/commit/fc5baf9fd2bb28c56f115fc2107a3ff159f1945d
        print("--> Event "+str(evt))
        self.tree.GetEntry(evt)
        EventNumber = self.tree.EventInfo.mcEventNumber()
 
        # Check whether the container exists
        for container in self.dict_variables_types.keys():
            dict_lists = {}
            dict_container = self.dict_variables_types[container]
            try:
                tp  = getattr(self.tree, container)
            except Exception:
                print(".. Missing ", container)
                continue
 
            if tp.size() == 0:
                print(".. Empty ", container)
                continue
 
            for var,fmt in dict_container.items():
                try:
                    sp  = tp.getConstDataSpan[ fmt ]( var) 
                except Exception:
                    # This is for the case arrays are either empty or have some trouble when accessed via getConstDataSpan. Used in excepction as makes the code slower
                    print("getConstDataSpan failed for variable ",var,fmt, container)
                    sp  = [ getattr(element, var)() for element in tp ]
 
                try:
                    # Convert list of std::vector<double> to the list of lists
                    list_of_lists = [list(std_vector) for std_vector in sp] 
                    # Find the length of the longest vector
                    max_len = max(len(ll) for ll in list_of_lists)
                    # extend lists to the length of max_len
                    for ll in list_of_lists:
                        ll.extend([np.nan] * (max_len - len(ll)))
                except Exception:
                    list_of_lists = sp  # the sp is not iterable        
 
                # Convert the list of lists to a NumPy array
                sp = np.array(list_of_lists)
 
                if "ArrayFloat3" in fmt and len(sp) > 0:  # Needs extra coding when dealing with xAOD::ArrayFloat3 instead of standard C++ array
                    sp = array( list( map(self.ArrayFloat3_to_CppArray, sp) ) )
 
                if ("unsigned char" in fmt or "uint8" in fmt) and len(sp) > 0:
                    sp = sp.view(np.int32)
 
                if sp.ndim == 1:
                    dict_lists[var] = sp
                else:
                    # Then we have an array. Each element of the array will be a column in the csv file. [We want things flat]
                    for column in range(sp.shape[1]):
                        dict_lists[var+f"_at{column}"] = sp.T[column]
            
            self.WriteCSV( filename=getCSVFilename(self.outputDir, container, EventNumber), dictionary=dict_lists )
 
 

Run()

def CSV_InDetExporter.CSVDumper.Run

(

self

)

Definition at line 113 of file CSV_InDetExporter.py.

    def Run(self):
 
        for evt in range(self.nEvents):
            self.ProcessEvent(evt)
 
 

WriteCSV()

def CSV_InDetExporter.CSVDumper.WriteCSV	(		self,
			filename,
			dictionary
	)

Definition at line 37 of file CSV_InDetExporter.py.

    def WriteCSV(self, filename, dictionary):
 
        with open(filename, "w") as out:
            writer = csv.writer(out)
            writer.writerow(dictionary.keys())
            writer.writerows( zip(*dictionary.values()) )
        print(f"New file saved: {filename}")
 

Member Data Documentation

dict_variables_types

CSV_InDetExporter.CSVDumper.dict_variables_types

Definition at line 35 of file CSV_InDetExporter.py.

n_entries

CSV_InDetExporter.CSVDumper.n_entries

Definition at line 18 of file CSV_InDetExporter.py.

nEvents

CSV_InDetExporter.CSVDumper.nEvents

Definition at line 22 of file CSV_InDetExporter.py.

outputDir

CSV_InDetExporter.CSVDumper.outputDir

Definition at line 19 of file CSV_InDetExporter.py.

tree

CSV_InDetExporter.CSVDumper.tree

Definition at line 17 of file CSV_InDetExporter.py.

---

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

• CSV_InDetExporter.py