pileupParser.py

Go to the documentation of this file.

#!/usr/bin/env python
from __future__ import print_function
import json
class pileupParser(object):
    def __init__(self,selectStr):
        self.__result={}
        self.__strresult={}
        strresult=json.loads(selectStr)
        for k,v in strresult.items():
#            expandedvalues=[]
#            print v
            ldict={}
            for elems in v:
                llist = [ elems[1],elems[2],elems[3] ]
                ldict[int(elems[0])] = llist


#            for w in v:
#                if len(w)==0:
#                    self.__result[int(k)]=expandedvalues
#                    self.__strresult[k]=[]
#                    continue
            
            self.__result[int(k)]=ldict
#            self.__strresult[k]=[str(x) for x in expandedvalues]
    def runs(self):
        return self.__result.keys()
    def runsandls(self):
        '''return expanded {run:lslist}
        '''
        return self.__result
    def runsandlsStr(self):
        '''return expanded {'run':lslist}
        '''
        return self.__strresult
    def numruns(self):
        return len(self.__result)
    def numls(self,run):
        return len(self.__result[run])
if __name__ == "__main__":
#    s=pileupParser('{"1":[[3,45]],"2":[[4,8],[10,10]],"3":[[]]}')
    s=pileupParser('{"179434": [[21,38.140,7.2,2.220e-04],[22,49.032,7.2,2.218e-04]],"179435": [[21,38.140,7.2,2.220e-04],[22,49.032,7.2,2.218e-04]]}')

    print('runs : ',s.runs())
    print('full result : ',s.runsandls())
#    print 'str result : ',s.runsandlsStr()
#    print 'num runs : ',s.numruns()
#    print 'numls in run : ',s.numls(1)