CSV_InDetImporter.CSV_InDetImporterAlg Class Reference

Inheritance diagram for CSV_InDetImporter.CSV_InDetImporterAlg:

Collaboration diagram for CSV_InDetImporter.CSV_InDetImporterAlg:

Public Member Functions
def	__init__ (self, name)
def	initialize (self)
def	execute (self)
def	getEventNumber (self)
def	readTrackParticle (self)

Public Attributes
	indir
	trackParticleName

Detailed Description

Algorithm to load InDet tracks data from CSV files to xAOD collections
Conventions in files naming:
 - the files should reside in one dir and alg needs to be pointed to it through indir property, the naming convention: container_event_XYZ.csv should be used
 - files CSV format needs to be identical to the one used in export
 - if a given file is missing the data is not loaded, just reporting in the log about it
 - If need to modify this behavior, hey, go for it, it is python

Definition at line 13 of file CSV_InDetImporter.py.

Constructor & Destructor Documentation

__init__()

def CSV_InDetImporter.CSV_InDetImporterAlg.__init__	(		self,
			name
	)

Definition at line 22 of file CSV_InDetImporter.py.

    def __init__(self, name):
        Alg.__init__ (self, name)
        self.indir = None
        self.trackParticleName = None
        return
 

Member Function Documentation

execute()

def CSV_InDetImporter.CSV_InDetImporterAlg.execute

(

self

)

Definition at line 38 of file CSV_InDetImporter.py.

    def execute(self):
        if self.trackParticleName:
            if not self.readTrackParticle():
                return StatusCode.Failure
 
        return StatusCode.Success
 

getEventNumber()

def CSV_InDetImporter.CSV_InDetImporterAlg.getEventNumber

(

self

)

Definition at line 45 of file CSV_InDetImporter.py.

    def getEventNumber(self):
        ei = self.evtStore.retrieve("xAOD::EventInfo", "EventInfo")
        return ei.eventNumber()
 

initialize()

def CSV_InDetImporter.CSV_InDetImporterAlg.initialize

(

self

)

Definition at line 28 of file CSV_InDetImporter.py.

    def initialize(self):
        if self.indir is None:
            self.msg.error("Input directory not configured")
            return StatusCode.Failure
        if not os.path.exists(self.indir):
            self.msg.error("Missing input directory %s", self.indir)
            return StatusCode.Failure
 
        return StatusCode.Success
 

readTrackParticle()

def CSV_InDetImporter.CSV_InDetImporterAlg.readTrackParticle

(

self

)

Definition at line 49 of file CSV_InDetImporter.py.

    def readTrackParticle(self):
        inputFileName = getCSVFilename(self.indir, self.trackParticleName, self.getEventNumber())
        if not os.path.exists(inputFileName):
            self.msg.warning("Missing file %s, this will result in missing collections in certain events which is not allowed by POOL &ROOT  ", inputFileName)
            return StatusCode.Recoverable
 
 
        c = ROOT.xAOD.TrackParticleContainer()
        aux = ROOT.xAOD.TrackParticleAuxContainer()
        c.setStore (aux)
        ROOT.SetOwnership (c,False)
        ROOT.SetOwnership (aux,False)
 
        # data reading
        with open(inputFileName, "r") as f:
            reader = csv.DictReader(f)
            for k in reader.fieldnames:
                if k not in CSV_DictFormats["InDetTrackParticles"].keys():
                    self.msg.error("A key: %s found in data that does not seem to be known", k)
                    return StatusCode.Failure
 
            for data in reader:
                tp = ROOT.xAOD.TrackParticle()
                c.push_back(tp)
                ROOT.SetOwnership (tp, False)
                covm = ROOT.xAOD.ParametersCovMatrix_t()
                covm.setZero()
                theta = 2.0 * math.atan( math.exp(-float(data['eta'])))
                tp.setDefiningParameters(float(data['d0']), float(data['z0']), float(data['phi']),
                                         theta,
                                         float(data['charge']) * math.sin(theta)/float(data['pt']))
                tp.setDefiningParametersCovMatrix(covm)
                pass
 
        if not self.evtStore.record (c, self.trackParticleName, False):
            return StatusCode.Failure
        if not self.evtStore.record (aux, f'{self.trackParticleName}Aux.', False):
            return StatusCode.Failure
 
        return StatusCode.Success
 
 

Member Data Documentation

indir

CSV_InDetImporter.CSV_InDetImporterAlg.indir

Definition at line 24 of file CSV_InDetImporter.py.

trackParticleName

CSV_InDetImporter.CSV_InDetImporterAlg.trackParticleName

Definition at line 25 of file CSV_InDetImporter.py.

---

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

• CSV_InDetImporter.py