import ROOT

class PyGetCdcPars :
    def __init__(self):
        print "PyGetCdcPars.__init__"
        self.t0s={}
        self.lorentzs={}
        self.vdrifts={}
        self.fileLoaded=False

    def loadInputFile(self, fileName):
        print "opening file",fileName
        f=open(fileName,'r')
        for line in f:
            split=line.split(',')
            run=int(split[0])
            self.t0s[run]=float(split[1])
            self.lorentzs[run]=float(split[3])
            self.vdrifts[run]=float(split[2])
        self.fileLoaded=True
        print "loaded data from ",len(self.t0s),"runs"
    
    def getT0(self,run):
        if not self.fileLoaded:
            print "############################################################################################"
            print "#"
            print "# File not loaded!"
            print "#"
            print "############################################################################################"
            return 0
        if run in self.t0s:
            return self.t0s[run]
        else:
            print "############################################################################################"
            print "#"
            print "# Warning trying to get cdc parameters for run ",run," for which there does not exist data"
            print "#"
            print "############################################################################################"
            return 0
        
    def getVDrift(self,run):
        if self.fileLoaded==False:
            print "############################################################################################"
            print "#"
            print "# File not loaded!"
            print "#"
            print "############################################################################################"
            return 0
        if run in self.vdrifts:
            return self.vdrifts[run]
        else:
            print "############################################################################################"
            print "#"
            print "# Warning trying to get cdc parameters for run ",run," for which there does not exist data"
            print "#"
            print "############################################################################################"
            return 0
        
    def getLorentzAngle(self,run):
        if self.fileLoaded==False:
            print "############################################################################################"
            print "#"
            print "# File not loaded!"
            print "#"
            print "############################################################################################"
            return 0
        if run in self.lorentzs:
            return self.lorentzs[run]
        else:
            print "############################################################################################"
            print "#"
            print "# Warning trying to get cdc parameters for run ",run," for which there does not exist data"
            print "#"
            print "############################################################################################"            
            return 0