CSV_InDetImporter.CSV_InDetImporterAlg Class Reference

Inheritance diagram for CSV_InDetImporter.CSV_InDetImporterAlg:

Collaboration diagram for CSV_InDetImporter.CSV_InDetImporterAlg:

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

Public Attributes
	indir = None
	trackParticleName = None

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__()

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()

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()

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()

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()

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 = None

Definition at line 24 of file CSV_InDetImporter.py.

trackParticleName

CSV_InDetImporter.CSV_InDetImporterAlg.trackParticleName = None

Definition at line 25 of file CSV_InDetImporter.py.

---

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

• CSV_InDetImporter.py