lumiTime.py

Go to the documentation of this file.

import os,sys,time,calendar,pytz
from datetime import datetime,timedelta

class lumiTime(object):
    def __init__(self):
        self.coraltimefm='MM/DD/YY HH24:MI:SS'
        self.pydatetimefm='%m/%d/%y %H:%M:%S'
        self.nbx=3564
        self.bunchspace_us=0.02495 #in microseconds
        self.bunchspace_s=24.95e-09 #in seconds
        
    def timestampTodatetimeUTC(self,ts):
        return datetime.fromtimestamp(ts,tz=pytz.utc)
    
    def LSDuration(self,norbits):
        return timedelta(microseconds=(self.nbx*norbits*self.bunchspace_us))
    
    def OrbitDuration(self):
        return timedelta(microseconds=(self.nbx*self.bunchspace_us))
    
    def OrbitToTimeStr(self,begStrTime,orbitnumber,begorbit=0,customfm=''):
        '''
        given a orbit number, return its corresponding time. Assuming begin time has orbit=0
        '''
        return self.DatetimeToStr(self.StrToDatetime(begStrTime)+(orbitnumber-begorbit)*self.OrbitDuration(),customfm=customfm)
    def OrbitToTime(self,begStrTime,orbitnumber,begorbit=0,customfm=''):
        '''
        given a orbit number, return its corresponding time. Default run begin time counting from orbit=0
        '''
        return self.StrToDatetime(begStrTime,customfm=customfm)+(orbitnumber-begorbit)*self.OrbitDuration()
    def OrbitToLocalTimestamp(self,begStrTime,orbitnumber,begorbit=0,customfm=''):
        '''
        given a orbit number, return its corresponding unixtimestamp. Default run begin time counting from orbit=0
        '''
        os.environ['TZ']='CET'
        time.tzset()
        orbittime=self.OrbitToTime(begStrTime,orbitnumber,begorbit=begorbit,customfm=customfm)
        return time.mktime(orbittime.timetuple())+orbittime.microsecond/1e6

    def OrbitToUTCTimestamp(self,begStrTime,orbitnumber,begorbit=0,customfm=''):
        '''
        given a orbit number, return its corresponding unixtimestamp. Default run begin time counting from orbit=0
        '''
        os.environ['TZ']='UTC'
        time.tzset()
        orbittime=self.OrbitToTime(begStrTime,orbitnumber,begorbit=begorbit,customfm=customfm)
        return time.mktime(orbittime.timetuple())+(orbittime.microsecond/1e6)
    def StrToDatetime(self,strTime,customfm=''):
        '''convert string timestamp to python datetime
        '''
        result=''
        try:
            if not customfm:
                result=datetime.strptime(strTime,self.pydatetimefm)
            else:
                result=datetime.strptime(strTime,customfm)
        except Exception,er:
            print str(er)
        return result
    def DatetimeToStr(self,timeValue,customfm=''):
        '''convert python datetime to string timestamp
        '''
        result=''
        try:
            if not customfm:
                result=timeValue.strftime(self.pydatetimefm)
            else:
                result=timeValue.strftime(customfm)
        except Exception,er:
            print str(er)
        return result
if __name__=='__main__':
    begTimeStr='03/30/10 10:10:01'
    c=lumiTime()
    print 'orbit 0 : ',c.OrbitToTime(begTimeStr,0,0)
    print 'orbit 1 : ',c.OrbitToTime(begTimeStr,1,0)
    print 'orbit 262144 : ',c.OrbitToTime(begTimeStr,262144,0)
    print 'orbit 0 : ',c.OrbitToUTCTimestamp(begTimeStr,0,0);
    print 'orbit 0 : ',c.OrbitToLocalTimestamp(begTimeStr,0,0);