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
            ###weed out [10]-like stuff just in case they exist
#                elif len(w)==1:
#                    expandedvalues.append(w[0])
            ##weed out [10,10]-like stuff
#                elif len(w)==2 and w[0]==w[1]:
#                    expandedvalues.append(w[0])
#                else:
#                    for i in range(w[0],w[1]+1):
#                        expandedvalues.append(i)
            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)