LArG4ShowerLibFunctions.TestShowerLib Class Reference

Collaboration diagram for LArG4ShowerLibFunctions.TestShowerLib:

Public Member Functions
def	__init__ (self)
def	fromLibs (self, libs)
def	readFromFile (self, filename)
def	writeToFile (self, filename)
def	printInfo (self)
def	drawHits (self)

Public Attributes
	library
	detector
	particle
	release
	geometry
	geant
	phys
	comment

Detailed Description

Definition at line 35 of file LArG4ShowerLibFunctions.py.

Constructor & Destructor Documentation

__init__()

def LArG4ShowerLibFunctions.TestShowerLib.__init__

(

self

)

Definition at line 36 of file LArG4ShowerLibFunctions.py.

    def __init__(self) :
        self.library = [] # list of StoredTestShower objs
        self.detector= ""
        self.particle= ""
        self.release= ""
        self.geometry= ""
        self.geant= ""
        self.phys= ""
        self.comment= ""

Member Function Documentation

drawHits()

def LArG4ShowerLibFunctions.TestShowerLib.drawHits

(

self

)

Definition at line 222 of file LArG4ShowerLibFunctions.py.

    def drawHits(self):
        from ROOT import TH3F
        from math import sqrt,copysign,log10
        hits = TH3F("HITS","Hits Distrib",50,1,1000,101,-300,300,100,0,500)
        containmentZ = TH3F("CONTZ","ContZ Distrib",50,1,1000,101,-300,300,100,0,500)
        containmentR = TH3F("CONTR","ContR Distrib",50,1,1000,101,-300,300,100,0,500)
        for storedShower in self.library :
            containmentR.Fill(log10(storedShower.momentum.e)*333,storedShower.rsize,storedShower.zsize/2,10)
            containmentR.Fill(log10(storedShower.momentum.e)*333,-storedShower.rsize,storedShower.zsize/2,10)
            containmentZ.Fill(log10(storedShower.momentum.e)*333,0,storedShower.zsize,10)
            for hit in storedShower.shower :
                hits.Fill(log10(storedShower.momentum.e)*333,copysign(sqrt(hit.x*hit.x + hit.y*hit.y),hit.x),hit.z)
        return hits,containmentZ,containmentR
 

fromLibs()

def LArG4ShowerLibFunctions.TestShowerLib.fromLibs	(		self,
			libs
	)

Definition at line 45 of file LArG4ShowerLibFunctions.py.

    def fromLibs(self,libs) :
        for lib in libs :
            if not isinstance(lib,self.__class__):
                print ("ERROR: Different types of libs")
                return False
        self.detector = libs[0].detector
        self.particle = libs[0].particle
        self.release = libs[0].release
        self.geometry = libs[0].geometry
        self.geant = libs[0].geant
        self.phys = libs[0].phys
        for lib in libs :
            if (    self.detector != lib.detector or
                self.particle != lib.particle or
                self.release != lib.release or
                self.geometry != lib.geometry or
                self.geant != lib.geant or
                self.phys != lib.phys ) :
                print ("ERROR: DIFFERENT LIBS!!!")
                return False
        from datetime import datetime
        self.comment = "ADDED "+str(datetime.now())
        for lib in libs :
            self.library += lib.library
        return True

printInfo()

def LArG4ShowerLibFunctions.TestShowerLib.printInfo

(

self

)

Definition at line 220 of file LArG4ShowerLibFunctions.py.

    def printInfo(self) :
        pass

readFromFile()

def LArG4ShowerLibFunctions.TestShowerLib.readFromFile	(		self,
			filename
	)

Definition at line 70 of file LArG4ShowerLibFunctions.py.

    def readFromFile(self,filename) :
        from ROOT import TFile
        tfile = TFile(filename)
        try :
            ver = int(tfile.Get("version").GetVal())
        except Exception:
            print ("Not a TestShowerLib: Broken file")
            tfile.Close()
            return False
        if (ver != 10) : #<<<<<<<<<<<<<<<<<<<<<<-------------- lib ver
            print ("Not a TestShowerLib")
            tfile.Close()
            return False
        meta = tfile.Get("meta")
        libr = tfile.Get("library")
        for event in meta :
            self.detector=str(event.detector)
            self.particle=str(event.particle)
            self.release=str(event.release)
            self.geometry=str(event.geometry)
            self.geant=str(event.geantVersion)
            self.phys=str(event.physicsList)
            self.comment=str(event.comment)
 
        state = 0
 
        #state == 0 : first line of shower header
        #state == 1 : second line of sower header
        #state == 2 : hit
 
        for event in libr : #this is quite unclear, but easy to implement
            if   (state == 0) : #first line
                curShower = StoredTestShower()
                curShower.vertex.x = event.x
                curShower.vertex.y = event.y
                curShower.vertex.z = event.z
                curShower.zsize = event.time
                hitsInCurShower = event.e
                state = 1
            elif (state == 1) : #shower header
                curShower.momentum.x = event.x
                curShower.momentum.y = event.y
                curShower.momentum.z = event.z
                curShower.momentum.e = event.e
                curShower.rsize = event.time
                if (hitsInCurShower > 0) :
                    state = 2
                else :
                    self.library.append(curShower)
                    state = 0
            elif (state == 2) :
                hit = FourVector()
                hit.e = event.e
                hit.x = event.x
                hit.y = event.y
                hit.z = event.z
                hit.time = event.time
                curShower.shower.append(hit)
                hitsInCurShower -= 1
                if (hitsInCurShower == 0) : #last hit
                    self.library.append(curShower)
                    state = 0
        tfile.Close()
        if (state != 0) :
            print ("FILE CORRUPTED!!")
            return False
        return True

writeToFile()

def LArG4ShowerLibFunctions.TestShowerLib.writeToFile	(		self,
			filename
	)

Definition at line 137 of file LArG4ShowerLibFunctions.py.

    def writeToFile(self,filename) :
        from ROOT import TFile,TTree,TParameter
        from ROOT import gROOT, addressof
        gROOT.ProcessLine(
            "struct MyMetaStruct {\
            Char_t   detector[40];\
            Char_t   release[40];\
            Char_t   geometry[40];\
            Char_t   geant[40];\
            Char_t   phys[40];\
            Char_t   comment[400];\
            Int_t    particle;\
        };" )
        from ROOT import MyMetaStruct
        gROOT.ProcessLine(
            "struct MyStruct {\
            Float_t   x;\
            Float_t   y;\
            Float_t   z;\
            Float_t   e;\
            Float_t   time;\
        };" )
        from ROOT import MyStruct
 
        tfile = TFile(filename,"RECREATE")
 
        ver = TParameter(int)("version",10) #<<<<<<<<<<<<<<<<<<<<<<-------------- lib ver
        ver.Write("version")
 
        meta = TTree()
        libr = TTree()
 
        mmstruct = MyMetaStruct()
 
        mmstruct.detector = "%s" % (str(self.detector))
        mmstruct.particle = int(self.particle)
        mmstruct.release = "%s" % (str(self.release))
        mmstruct.geometry = "%s" % (str(self.geometry))
        mmstruct.geant = "%s" % (str(self.geant))
        mmstruct.phys = "%s" % (str(self.phys))
        mmstruct.comment = "%s" % (str(self.comment))
 
        meta.Branch("detector",addressof(mmstruct,"detector"),"detector/C")
        meta.Branch("particle",addressof(mmstruct,"particle"),"particle/I")
        meta.Branch("release",addressof(mmstruct,"release"),"release/C")
        meta.Branch("geometry",addressof(mmstruct,"geometry"),"geometry/C")
        meta.Branch("geantVersion",addressof(mmstruct,"geant"),"geantVersion/C")
        meta.Branch("physicsList",addressof(mmstruct,"phys"),"physicsList/C")
        meta.Branch("comment",addressof(mmstruct,"comment"),"physicsList/C")
 
        meta.Fill()
 
        mstruct = MyStruct()
 
        libr.Branch("x",addressof(mstruct,"x"),"x/F")
        libr.Branch("y",addressof(mstruct,"y"),"y/F")
        libr.Branch("z",addressof(mstruct,"z"),"z/F")
        libr.Branch("e",addressof(mstruct,"e"),"e/F")
        libr.Branch("time",addressof(mstruct,"time"),"time/F")
 
        for storedShower in self.library :
            mstruct.x = storedShower.vertex.x
            mstruct.y = storedShower.vertex.y
            mstruct.z = storedShower.vertex.z
            mstruct.time = storedShower.zsize
            mstruct.e = len(storedShower.shower)
            libr.Fill()
            mstruct.x = storedShower.momentum.x
            mstruct.y = storedShower.momentum.y
            mstruct.z = storedShower.momentum.z
            mstruct.e = storedShower.momentum.e
            mstruct.time = storedShower.rsize
            libr.Fill()
            for hit in storedShower.shower:
                mstruct.e = hit.e
                mstruct.x = hit.x
                mstruct.y = hit.y
                mstruct.z = hit.z
                mstruct.time = hit.time
                libr.Fill()
        meta.Write("meta")
        libr.Write("library")
        tfile.Close()

Member Data Documentation

comment

LArG4ShowerLibFunctions.TestShowerLib.comment

Definition at line 44 of file LArG4ShowerLibFunctions.py.

detector

LArG4ShowerLibFunctions.TestShowerLib.detector

Definition at line 38 of file LArG4ShowerLibFunctions.py.

geant

LArG4ShowerLibFunctions.TestShowerLib.geant

Definition at line 42 of file LArG4ShowerLibFunctions.py.

geometry

LArG4ShowerLibFunctions.TestShowerLib.geometry

Definition at line 41 of file LArG4ShowerLibFunctions.py.

library

LArG4ShowerLibFunctions.TestShowerLib.library

Definition at line 37 of file LArG4ShowerLibFunctions.py.

particle

LArG4ShowerLibFunctions.TestShowerLib.particle

Definition at line 39 of file LArG4ShowerLibFunctions.py.

phys

LArG4ShowerLibFunctions.TestShowerLib.phys

Definition at line 43 of file LArG4ShowerLibFunctions.py.

release

LArG4ShowerLibFunctions.TestShowerLib.release

Definition at line 40 of file LArG4ShowerLibFunctions.py.

---

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

• LArG4ShowerLibFunctions.py