info.py revision 1165
11049Sbinkertn@umich.edufrom __future__ import division 21049Sbinkertn@umich.eduimport operator, re, types 31049Sbinkertn@umich.edu 41049Sbinkertn@umich.edusource = None 51049Sbinkertn@umich.edudisplay_run = 0 61049Sbinkertn@umich.edu 71049Sbinkertn@umich.edudef issequence(t): 81049Sbinkertn@umich.edu return isinstance(t, types.TupleType) or isinstance(t, types.ListType) 91049Sbinkertn@umich.edu 101049Sbinkertn@umich.edudef total(f): 111049Sbinkertn@umich.edu if isinstance(f, FormulaStat): 121049Sbinkertn@umich.edu v = f.value 131049Sbinkertn@umich.edu else: 141049Sbinkertn@umich.edu v = f 151049Sbinkertn@umich.edu 161049Sbinkertn@umich.edu f = FormulaStat() 171049Sbinkertn@umich.edu if issequence(v): 181049Sbinkertn@umich.edu f.value = reduce(operator.add, v) 191049Sbinkertn@umich.edu else: 201049Sbinkertn@umich.edu f.value = v 211049Sbinkertn@umich.edu 221049Sbinkertn@umich.edu return f 231049Sbinkertn@umich.edu 241049Sbinkertn@umich.edudef unaryop(op, f): 251049Sbinkertn@umich.edu if isinstance(f, FormulaStat): 261049Sbinkertn@umich.edu v = f.value 271049Sbinkertn@umich.edu else: 281049Sbinkertn@umich.edu v = f 291049Sbinkertn@umich.edu 301049Sbinkertn@umich.edu if issequence(v): 311049Sbinkertn@umich.edu return map(op, v) 321049Sbinkertn@umich.edu else: 331049Sbinkertn@umich.edu return op(v) 341049Sbinkertn@umich.edu 351049Sbinkertn@umich.edudef zerodiv(lv, rv): 361049Sbinkertn@umich.edu if rv == 0.0: 371049Sbinkertn@umich.edu return 0.0 381049Sbinkertn@umich.edu else: 391049Sbinkertn@umich.edu return operator.truediv(lv, rv) 401049Sbinkertn@umich.edu 411049Sbinkertn@umich.edudef wrapop(op, lv, rv): 421049Sbinkertn@umich.edu if isinstance(lv, str): 431049Sbinkertn@umich.edu return lv 441049Sbinkertn@umich.edu 451049Sbinkertn@umich.edu if isinstance(rv, str): 461049Sbinkertn@umich.edu return rv 471049Sbinkertn@umich.edu 481049Sbinkertn@umich.edu return op(lv, rv) 491049Sbinkertn@umich.edu 501049Sbinkertn@umich.edudef same(lv, rv): 511049Sbinkertn@umich.edu for lrun,rrun in zip(lv.keys(),rv.keys()): 521049Sbinkertn@umich.edu if lrun != rrun: 531049Sbinkertn@umich.edu print 'lrun != rrun' 541049Sbinkertn@umich.edu print lrun, rrun 551049Sbinkertn@umich.edu print lv.keys() 561049Sbinkertn@umich.edu print rv.keys() 571049Sbinkertn@umich.edu return False 581049Sbinkertn@umich.edu for lx,rx in zip(lv[lrun].keys(),rv[rrun].keys()): 591049Sbinkertn@umich.edu if lx != rx: 601049Sbinkertn@umich.edu print 'lx != rx' 611049Sbinkertn@umich.edu print lx, rx 621049Sbinkertn@umich.edu print lv[lrun].keys() 631049Sbinkertn@umich.edu print rv[rrun].keys() 641049Sbinkertn@umich.edu return False 651049Sbinkertn@umich.edu for ly,ry in zip(lv[lrun][lx].keys(),rv[rrun][rx].keys()): 661049Sbinkertn@umich.edu if ly != ry: 671049Sbinkertn@umich.edu print 'ly != ry' 681049Sbinkertn@umich.edu print ly, ry 691049Sbinkertn@umich.edu print lv[lrun][lx].keys() 701049Sbinkertn@umich.edu print rv[rrun][rx].keys() 711049Sbinkertn@umich.edu return False 721049Sbinkertn@umich.edu return True 731049Sbinkertn@umich.edu 741049Sbinkertn@umich.edu 751049Sbinkertn@umich.edudef binaryop(op, lf, rf): 761049Sbinkertn@umich.edu result = {} 771049Sbinkertn@umich.edu 781049Sbinkertn@umich.edu if isinstance(lf, FormulaStat) and isinstance(rf, FormulaStat): 791049Sbinkertn@umich.edu lv = lf.value 801049Sbinkertn@umich.edu rv = rf.value 811049Sbinkertn@umich.edu 821049Sbinkertn@umich.edu if not same(lv, rv): 831049Sbinkertn@umich.edu raise AttributeError, "run,x,y not identical" 841049Sbinkertn@umich.edu 851049Sbinkertn@umich.edu for run in lv.keys(): 861049Sbinkertn@umich.edu result[run] = {} 871049Sbinkertn@umich.edu for x in lv[run].keys(): 881049Sbinkertn@umich.edu result[run][x] = {} 891049Sbinkertn@umich.edu for y in lv[run][x].keys(): 901049Sbinkertn@umich.edu result[run][x][y] = wrapop(op, lv[run][x][y], 911049Sbinkertn@umich.edu rv[run][x][y]) 921049Sbinkertn@umich.edu elif isinstance(lf, FormulaStat): 931049Sbinkertn@umich.edu lv = lf.value 941049Sbinkertn@umich.edu for run in lv.keys(): 951049Sbinkertn@umich.edu result[run] = {} 961049Sbinkertn@umich.edu for x in lv[run].keys(): 971049Sbinkertn@umich.edu result[run][x] = {} 981049Sbinkertn@umich.edu for y in lv[run][x].keys(): 991049Sbinkertn@umich.edu result[run][x][y] = wrapop(op, lv[run][x][y], rf) 1001049Sbinkertn@umich.edu elif isinstance(rf, FormulaStat): 1011049Sbinkertn@umich.edu rv = rf.value 1021049Sbinkertn@umich.edu for run in rv.keys(): 1031049Sbinkertn@umich.edu result[run] = {} 1041049Sbinkertn@umich.edu for x in rv[run].keys(): 1051049Sbinkertn@umich.edu result[run][x] = {} 1061049Sbinkertn@umich.edu for y in rv[run][x].keys(): 1071049Sbinkertn@umich.edu result[run][x][y] = wrapop(op, lf, rv[run][x][y]) 1081049Sbinkertn@umich.edu 1091049Sbinkertn@umich.edu return result 1101049Sbinkertn@umich.edu 1111049Sbinkertn@umich.edudef sums(x, y): 1121049Sbinkertn@umich.edu if issequence(x): 1131049Sbinkertn@umich.edu return map(lambda x, y: x + y, x, y) 1141049Sbinkertn@umich.edu else: 1151049Sbinkertn@umich.edu return x + y 1161049Sbinkertn@umich.edu 1171049Sbinkertn@umich.edudef alltrue(list): 1181049Sbinkertn@umich.edu return reduce(lambda x, y: x and y, list) 1191049Sbinkertn@umich.edu 1201049Sbinkertn@umich.edudef allfalse(list): 1211049Sbinkertn@umich.edu return not reduce(lambda x, y: x or y, list) 1221049Sbinkertn@umich.edu 1231049Sbinkertn@umich.edudef enumerate(list): 1241049Sbinkertn@umich.edu return map(None, range(len(list)), list) 1251049Sbinkertn@umich.edu 1261049Sbinkertn@umich.edudef cmp(a, b): 1271049Sbinkertn@umich.edu if a < b: 1281049Sbinkertn@umich.edu return -1 1291049Sbinkertn@umich.edu elif a == b: 1301049Sbinkertn@umich.edu return 0 1311049Sbinkertn@umich.edu else: 1321049Sbinkertn@umich.edu return 1 1331049Sbinkertn@umich.edu 1341049Sbinkertn@umich.educlass Statistic(object): 1351049Sbinkertn@umich.edu def __init__(self, data): 1361049Sbinkertn@umich.edu self.__dict__.update(data.__dict__) 1371049Sbinkertn@umich.edu if not self.__dict__.has_key('value'): 1381049Sbinkertn@umich.edu self.__dict__['value'] = None 1391049Sbinkertn@umich.edu if not self.__dict__.has_key('bins'): 1401049Sbinkertn@umich.edu self.__dict__['bins'] = None 1411049Sbinkertn@umich.edu if not self.__dict__.has_key('ticks'): 1421049Sbinkertn@umich.edu self.__dict__['ticks'] = None 1431049Sbinkertn@umich.edu 1441049Sbinkertn@umich.edu def __getattribute__(self, attr): 1451049Sbinkertn@umich.edu if attr == 'value': 1461049Sbinkertn@umich.edu if self.__dict__['value'] == None: 1471049Sbinkertn@umich.edu self.__dict__['value'] = self.getValue() 1481049Sbinkertn@umich.edu return self.__dict__['value'] 1491049Sbinkertn@umich.edu else: 1501049Sbinkertn@umich.edu return super(Statistic, self).__getattribute__(attr) 1511049Sbinkertn@umich.edu 1521049Sbinkertn@umich.edu def __setattr__(self, attr, value): 1531049Sbinkertn@umich.edu if attr == 'bins' or attr == 'ticks': 1541049Sbinkertn@umich.edu if attr == 'bins': 1551049Sbinkertn@umich.edu if value is not None: 1561165Sbinkertn@umich.edu value = source.getBin(value) 1571049Sbinkertn@umich.edu elif attr == 'samples' and type(value) is str: 1581049Sbinkertn@umich.edu value = [ int(x) for x in value.split() ] 1591049Sbinkertn@umich.edu 1601049Sbinkertn@umich.edu self.__dict__[attr] = value 1611049Sbinkertn@umich.edu self.__dict__['value'] = None 1621049Sbinkertn@umich.edu else: 1631049Sbinkertn@umich.edu super(Statistic, self).__setattr__(attr, value) 1641049Sbinkertn@umich.edu 1651049Sbinkertn@umich.edu def getValue(self): 1661049Sbinkertn@umich.edu raise AttributeError, 'getValue() must be defined' 1671049Sbinkertn@umich.edu 1681049Sbinkertn@umich.edu def zero(self): 1691049Sbinkertn@umich.edu return False 1701049Sbinkertn@umich.edu 1711049Sbinkertn@umich.edu def __ne__(self, other): 1721049Sbinkertn@umich.edu return not (self == other) 1731049Sbinkertn@umich.edu 1741049Sbinkertn@umich.edu def __str__(self): 1751049Sbinkertn@umich.edu return '%f' % (float(self)) 1761049Sbinkertn@umich.edu 1771049Sbinkertn@umich.educlass FormulaStat(object): 1781049Sbinkertn@umich.edu def __add__(self, other): 1791049Sbinkertn@umich.edu f = FormulaStat() 1801049Sbinkertn@umich.edu f.value = binaryop(operator.add, self, other) 1811049Sbinkertn@umich.edu return f 1821049Sbinkertn@umich.edu def __sub__(self, other): 1831049Sbinkertn@umich.edu f = FormulaStat() 1841049Sbinkertn@umich.edu f.value = binaryop(operator.sub, self, other) 1851049Sbinkertn@umich.edu return f 1861049Sbinkertn@umich.edu def __mul__(self, other): 1871049Sbinkertn@umich.edu f = FormulaStat() 1881049Sbinkertn@umich.edu f.value = binaryop(operator.mul, self, other) 1891049Sbinkertn@umich.edu return f 1901049Sbinkertn@umich.edu def __truediv__(self, other): 1911049Sbinkertn@umich.edu f = FormulaStat() 1921049Sbinkertn@umich.edu f.value = binaryop(zerodiv, self, other) 1931049Sbinkertn@umich.edu return f 1941049Sbinkertn@umich.edu def __mod__(self, other): 1951049Sbinkertn@umich.edu f = FormulaStat() 1961049Sbinkertn@umich.edu f.value = binaryop(operator.mod, self, other) 1971049Sbinkertn@umich.edu return f 1981049Sbinkertn@umich.edu def __radd__(self, other): 1991049Sbinkertn@umich.edu f = FormulaStat() 2001049Sbinkertn@umich.edu f.value = binaryop(operator.add, other, self) 2011049Sbinkertn@umich.edu return f 2021049Sbinkertn@umich.edu def __rsub__(self, other): 2031049Sbinkertn@umich.edu f = FormulaStat() 2041049Sbinkertn@umich.edu f.value = binaryop(operator.sub, other, self) 2051049Sbinkertn@umich.edu return f 2061049Sbinkertn@umich.edu def __rmul__(self, other): 2071049Sbinkertn@umich.edu f = FormulaStat() 2081049Sbinkertn@umich.edu f.value = binaryop(operator.mul, other, self) 2091049Sbinkertn@umich.edu return f 2101049Sbinkertn@umich.edu def __rtruediv__(self, other): 2111049Sbinkertn@umich.edu f = FormulaStat() 2121049Sbinkertn@umich.edu f.value = binaryop(zerodiv, other, self) 2131049Sbinkertn@umich.edu return f 2141049Sbinkertn@umich.edu def __rmod__(self, other): 2151049Sbinkertn@umich.edu f = FormulaStat() 2161049Sbinkertn@umich.edu f.value = binaryop(operator.mod, other, self) 2171049Sbinkertn@umich.edu return f 2181049Sbinkertn@umich.edu def __neg__(self): 2191049Sbinkertn@umich.edu f = FormulaStat() 2201049Sbinkertn@umich.edu f.value = unaryop(operator.neg, self) 2211049Sbinkertn@umich.edu return f 2221049Sbinkertn@umich.edu def __getitem__(self, idx): 2231049Sbinkertn@umich.edu f = FormulaStat() 2241049Sbinkertn@umich.edu f.value = {} 2251049Sbinkertn@umich.edu for key in self.value.keys(): 2261049Sbinkertn@umich.edu f.value[key] = {} 2271049Sbinkertn@umich.edu f.value[key][0] = {} 2281049Sbinkertn@umich.edu f.value[key][0][0] = self.value[key][idx][0] 2291049Sbinkertn@umich.edu return f 2301049Sbinkertn@umich.edu 2311049Sbinkertn@umich.edu def __float__(self): 2321049Sbinkertn@umich.edu if isinstance(self.value, FormulaStat): 2331049Sbinkertn@umich.edu return float(self.value) 2341049Sbinkertn@umich.edu if not self.value.has_key(display_run): 2351049Sbinkertn@umich.edu return (1e300*1e300) 2361049Sbinkertn@umich.edu if len(self.value[display_run]) == 1: 2371049Sbinkertn@umich.edu return self.value[display_run][0][0] 2381049Sbinkertn@umich.edu else: 2391049Sbinkertn@umich.edu #print self.value[display_run] 2401049Sbinkertn@umich.edu return self.value[display_run][4][0] 2411049Sbinkertn@umich.edu #raise ValueError 2421049Sbinkertn@umich.edu 2431049Sbinkertn@umich.edu def display(self): 2441049Sbinkertn@umich.edu import display 2451049Sbinkertn@umich.edu d = display.VectorDisplay() 2461049Sbinkertn@umich.edu d.flags = 0 2471049Sbinkertn@umich.edu d.precision = 1 2481049Sbinkertn@umich.edu d.name = 'formula' 2491049Sbinkertn@umich.edu d.desc = 'formula' 2501049Sbinkertn@umich.edu val = self.value[display_run] 2511049Sbinkertn@umich.edu d.value = [ val[x][0] for x in val.keys() ] 2521049Sbinkertn@umich.edu d.display() 2531049Sbinkertn@umich.edu 2541049Sbinkertn@umich.edu 2551049Sbinkertn@umich.educlass Scalar(Statistic,FormulaStat): 2561049Sbinkertn@umich.edu def getValue(self): 2571049Sbinkertn@umich.edu return source.data(self, self.bins) 2581049Sbinkertn@umich.edu 2591049Sbinkertn@umich.edu def display(self): 2601049Sbinkertn@umich.edu import display 2611049Sbinkertn@umich.edu p = display.Print() 2621049Sbinkertn@umich.edu p.name = self.name 2631049Sbinkertn@umich.edu p.desc = self.desc 2641049Sbinkertn@umich.edu p.value = float(self) 2651049Sbinkertn@umich.edu p.flags = self.flags 2661049Sbinkertn@umich.edu p.precision = self.precision 2671049Sbinkertn@umich.edu if display.all or (self.flags & flags.printable): 2681049Sbinkertn@umich.edu p.display() 2691049Sbinkertn@umich.edu 2701049Sbinkertn@umich.edu def comparable(self, other): 2711049Sbinkertn@umich.edu return self.name == other.name 2721049Sbinkertn@umich.edu 2731049Sbinkertn@umich.edu def __eq__(self, other): 2741049Sbinkertn@umich.edu return self.value == other.value 2751049Sbinkertn@umich.edu 2761049Sbinkertn@umich.edu def __isub__(self, other): 2771049Sbinkertn@umich.edu self.value -= other.value 2781049Sbinkertn@umich.edu return self 2791049Sbinkertn@umich.edu 2801049Sbinkertn@umich.edu def __iadd__(self, other): 2811049Sbinkertn@umich.edu self.value += other.value 2821049Sbinkertn@umich.edu return self 2831049Sbinkertn@umich.edu 2841049Sbinkertn@umich.edu def __itruediv__(self, other): 2851049Sbinkertn@umich.edu if not other: 2861049Sbinkertn@umich.edu return self 2871049Sbinkertn@umich.edu self.value /= other 2881049Sbinkertn@umich.edu return self 2891049Sbinkertn@umich.edu 2901049Sbinkertn@umich.educlass Vector(Statistic,FormulaStat): 2911049Sbinkertn@umich.edu def getValue(self): 2921049Sbinkertn@umich.edu return source.data(self, self.bins); 2931049Sbinkertn@umich.edu 2941049Sbinkertn@umich.edu def display(self): 2951049Sbinkertn@umich.edu import display 2961049Sbinkertn@umich.edu if not display.all and not (self.flags & flags.printable): 2971049Sbinkertn@umich.edu return 2981049Sbinkertn@umich.edu 2991049Sbinkertn@umich.edu d = display.VectorDisplay() 3001049Sbinkertn@umich.edu d.__dict__.update(self.__dict__) 3011049Sbinkertn@umich.edu d.display() 3021049Sbinkertn@umich.edu 3031049Sbinkertn@umich.edu def comparable(self, other): 3041049Sbinkertn@umich.edu return self.name == other.name and \ 3051049Sbinkertn@umich.edu len(self.value) == len(other.value) 3061049Sbinkertn@umich.edu 3071049Sbinkertn@umich.edu def __eq__(self, other): 3081049Sbinkertn@umich.edu if issequence(self.value) != issequence(other.value): 3091049Sbinkertn@umich.edu return false 3101049Sbinkertn@umich.edu 3111049Sbinkertn@umich.edu if issequence(self.value): 3121049Sbinkertn@umich.edu if len(self.value) != len(other.value): 3131049Sbinkertn@umich.edu return False 3141049Sbinkertn@umich.edu else: 3151049Sbinkertn@umich.edu for v1,v2 in zip(self.value, other.value): 3161049Sbinkertn@umich.edu if v1 != v2: 3171049Sbinkertn@umich.edu return False 3181049Sbinkertn@umich.edu return True 3191049Sbinkertn@umich.edu else: 3201049Sbinkertn@umich.edu return self.value == other.value 3211049Sbinkertn@umich.edu 3221049Sbinkertn@umich.edu def __isub__(self, other): 3231049Sbinkertn@umich.edu self.value = binaryop(operator.sub, self.value, other.value) 3241049Sbinkertn@umich.edu return self 3251049Sbinkertn@umich.edu 3261049Sbinkertn@umich.edu def __iadd__(self, other): 3271049Sbinkertn@umich.edu self.value = binaryop(operator.add, self.value, other.value) 3281049Sbinkertn@umich.edu return self 3291049Sbinkertn@umich.edu 3301049Sbinkertn@umich.edu def __itruediv__(self, other): 3311049Sbinkertn@umich.edu if not other: 3321049Sbinkertn@umich.edu return self 3331049Sbinkertn@umich.edu if issequence(self.value): 3341049Sbinkertn@umich.edu for i in xrange(len(self.value)): 3351049Sbinkertn@umich.edu self.value[i] /= other 3361049Sbinkertn@umich.edu else: 3371049Sbinkertn@umich.edu self.value /= other 3381049Sbinkertn@umich.edu return self 3391049Sbinkertn@umich.edu 3401049Sbinkertn@umich.educlass Formula(Vector): 3411049Sbinkertn@umich.edu def getValue(self): 3421049Sbinkertn@umich.edu formula = re.sub(':', '__', self.formula) 3431049Sbinkertn@umich.edu x = eval(formula, source.stattop) 3441049Sbinkertn@umich.edu return x.value 3451049Sbinkertn@umich.edu 3461049Sbinkertn@umich.edu def comparable(self, other): 3471049Sbinkertn@umich.edu return self.name == other.name and \ 3481049Sbinkertn@umich.edu compare(self.dist, other.dist) 3491049Sbinkertn@umich.edu 3501049Sbinkertn@umich.edu def __eq__(self, other): 3511049Sbinkertn@umich.edu return self.value == other.value 3521049Sbinkertn@umich.edu 3531049Sbinkertn@umich.edu def __isub__(self, other): 3541049Sbinkertn@umich.edu return self 3551049Sbinkertn@umich.edu 3561049Sbinkertn@umich.edu def __iadd__(self, other): 3571049Sbinkertn@umich.edu return self 3581049Sbinkertn@umich.edu 3591049Sbinkertn@umich.edu def __itruediv__(self, other): 3601049Sbinkertn@umich.edu if not other: 3611049Sbinkertn@umich.edu return self 3621049Sbinkertn@umich.edu return self 3631049Sbinkertn@umich.edu 3641049Sbinkertn@umich.educlass SimpleDist(object): 3651049Sbinkertn@umich.edu def __init__(self, sums, squares, samples): 3661049Sbinkertn@umich.edu self.sums = sums 3671049Sbinkertn@umich.edu self.squares = squares 3681049Sbinkertn@umich.edu self.samples = samples 3691049Sbinkertn@umich.edu 3701049Sbinkertn@umich.edu def getValue(self): 3711049Sbinkertn@umich.edu return 0.0 3721049Sbinkertn@umich.edu 3731049Sbinkertn@umich.edu def display(self, name, desc, flags, precision): 3741049Sbinkertn@umich.edu import display 3751049Sbinkertn@umich.edu p = display.Print() 3761049Sbinkertn@umich.edu p.flags = flags 3771049Sbinkertn@umich.edu p.precision = precision 3781049Sbinkertn@umich.edu 3791049Sbinkertn@umich.edu if self.samples > 0: 3801049Sbinkertn@umich.edu p.name = name + ".mean" 3811049Sbinkertn@umich.edu p.value = self.sums / self.samples 3821049Sbinkertn@umich.edu p.display() 3831049Sbinkertn@umich.edu 3841049Sbinkertn@umich.edu p.name = name + ".stdev" 3851049Sbinkertn@umich.edu if self.samples > 1: 3861049Sbinkertn@umich.edu var = (self.samples * self.squares - self.sums ** 2) \ 3871049Sbinkertn@umich.edu / (self.samples * (self.samples - 1)) 3881049Sbinkertn@umich.edu if var >= 0: 3891049Sbinkertn@umich.edu p.value = math.sqrt(var) 3901049Sbinkertn@umich.edu else: 3911049Sbinkertn@umich.edu p.value = 'NaN' 3921049Sbinkertn@umich.edu else: 3931049Sbinkertn@umich.edu p.value = 0.0 3941049Sbinkertn@umich.edu p.display() 3951049Sbinkertn@umich.edu 3961049Sbinkertn@umich.edu p.name = name + ".samples" 3971049Sbinkertn@umich.edu p.value = self.samples 3981049Sbinkertn@umich.edu p.display() 3991049Sbinkertn@umich.edu 4001049Sbinkertn@umich.edu def comparable(self, other): 4011049Sbinkertn@umich.edu return True 4021049Sbinkertn@umich.edu 4031049Sbinkertn@umich.edu def __eq__(self, other): 4041049Sbinkertn@umich.edu return self.sums == other.sums and self.squares == other.squares and \ 4051049Sbinkertn@umich.edu self.samples == other.samples 4061049Sbinkertn@umich.edu 4071049Sbinkertn@umich.edu def __isub__(self, other): 4081049Sbinkertn@umich.edu self.sums -= other.sums 4091049Sbinkertn@umich.edu self.squares -= other.squares 4101049Sbinkertn@umich.edu self.samples -= other.samples 4111049Sbinkertn@umich.edu return self 4121049Sbinkertn@umich.edu 4131049Sbinkertn@umich.edu def __iadd__(self, other): 4141049Sbinkertn@umich.edu self.sums += other.sums 4151049Sbinkertn@umich.edu self.squares += other.squares 4161049Sbinkertn@umich.edu self.samples += other.samples 4171049Sbinkertn@umich.edu return self 4181049Sbinkertn@umich.edu 4191049Sbinkertn@umich.edu def __itruediv__(self, other): 4201049Sbinkertn@umich.edu if not other: 4211049Sbinkertn@umich.edu return self 4221049Sbinkertn@umich.edu self.sums /= other 4231049Sbinkertn@umich.edu self.squares /= other 4241049Sbinkertn@umich.edu self.samples /= other 4251049Sbinkertn@umich.edu return self 4261049Sbinkertn@umich.edu 4271049Sbinkertn@umich.educlass FullDist(SimpleDist): 4281049Sbinkertn@umich.edu def __init__(self, sums, squares, samples, minval, maxval, 4291049Sbinkertn@umich.edu under, vec, over, min, max, bsize, size): 4301049Sbinkertn@umich.edu self.sums = sums 4311049Sbinkertn@umich.edu self.squares = squares 4321049Sbinkertn@umich.edu self.samples = samples 4331049Sbinkertn@umich.edu self.minval = minval 4341049Sbinkertn@umich.edu self.maxval = maxval 4351049Sbinkertn@umich.edu self.under = under 4361049Sbinkertn@umich.edu self.vec = vec 4371049Sbinkertn@umich.edu self.over = over 4381049Sbinkertn@umich.edu self.min = min 4391049Sbinkertn@umich.edu self.max = max 4401049Sbinkertn@umich.edu self.bsize = bsize 4411049Sbinkertn@umich.edu self.size = size 4421049Sbinkertn@umich.edu 4431049Sbinkertn@umich.edu def getValue(self): 4441049Sbinkertn@umich.edu return 0.0 4451049Sbinkertn@umich.edu 4461049Sbinkertn@umich.edu def display(self, name, desc, flags, precision): 4471049Sbinkertn@umich.edu import display 4481049Sbinkertn@umich.edu p = display.Print() 4491049Sbinkertn@umich.edu p.flags = flags 4501049Sbinkertn@umich.edu p.precision = precision 4511049Sbinkertn@umich.edu 4521049Sbinkertn@umich.edu p.name = name + '.min_val' 4531049Sbinkertn@umich.edu p.value = self.minval 4541049Sbinkertn@umich.edu p.display() 4551049Sbinkertn@umich.edu 4561049Sbinkertn@umich.edu p.name = name + '.max_val' 4571049Sbinkertn@umich.edu p.value = self.maxval 4581049Sbinkertn@umich.edu p.display() 4591049Sbinkertn@umich.edu 4601049Sbinkertn@umich.edu p.name = name + '.underflow' 4611049Sbinkertn@umich.edu p.value = self.under 4621049Sbinkertn@umich.edu p.display() 4631049Sbinkertn@umich.edu 4641049Sbinkertn@umich.edu i = self.min 4651049Sbinkertn@umich.edu for val in self.vec[:-1]: 4661049Sbinkertn@umich.edu p.name = name + '[%d:%d]' % (i, i + self.bsize - 1) 4671049Sbinkertn@umich.edu p.value = val 4681049Sbinkertn@umich.edu p.display() 4691049Sbinkertn@umich.edu i += self.bsize 4701049Sbinkertn@umich.edu 4711049Sbinkertn@umich.edu p.name = name + '[%d:%d]' % (i, self.max) 4721049Sbinkertn@umich.edu p.value = self.vec[-1] 4731049Sbinkertn@umich.edu p.display() 4741049Sbinkertn@umich.edu 4751049Sbinkertn@umich.edu 4761049Sbinkertn@umich.edu p.name = name + '.overflow' 4771049Sbinkertn@umich.edu p.value = self.over 4781049Sbinkertn@umich.edu p.display() 4791049Sbinkertn@umich.edu 4801049Sbinkertn@umich.edu SimpleDist.display(self, name, desc, flags, precision) 4811049Sbinkertn@umich.edu 4821049Sbinkertn@umich.edu def comparable(self, other): 4831049Sbinkertn@umich.edu return self.min == other.min and self.max == other.max and \ 4841049Sbinkertn@umich.edu self.bsize == other.bsize and self.size == other.size 4851049Sbinkertn@umich.edu 4861049Sbinkertn@umich.edu def __eq__(self, other): 4871049Sbinkertn@umich.edu return self.sums == other.sums and self.squares == other.squares and \ 4881049Sbinkertn@umich.edu self.samples == other.samples 4891049Sbinkertn@umich.edu 4901049Sbinkertn@umich.edu def __isub__(self, other): 4911049Sbinkertn@umich.edu self.sums -= other.sums 4921049Sbinkertn@umich.edu self.squares -= other.squares 4931049Sbinkertn@umich.edu self.samples -= other.samples 4941049Sbinkertn@umich.edu 4951049Sbinkertn@umich.edu if other.samples: 4961049Sbinkertn@umich.edu self.minval = min(self.minval, other.minval) 4971049Sbinkertn@umich.edu self.maxval = max(self.maxval, other.maxval) 4981049Sbinkertn@umich.edu self.under -= under 4991049Sbinkertn@umich.edu self.vec = map(lambda x,y: x - y, self.vec, other.vec) 5001049Sbinkertn@umich.edu self.over -= over 5011049Sbinkertn@umich.edu return self 5021049Sbinkertn@umich.edu 5031049Sbinkertn@umich.edu def __iadd__(self, other): 5041049Sbinkertn@umich.edu if not self.samples and other.samples: 5051049Sbinkertn@umich.edu self = other 5061049Sbinkertn@umich.edu return self 5071049Sbinkertn@umich.edu 5081049Sbinkertn@umich.edu self.sums += other.sums 5091049Sbinkertn@umich.edu self.squares += other.squares 5101049Sbinkertn@umich.edu self.samples += other.samples 5111049Sbinkertn@umich.edu 5121049Sbinkertn@umich.edu if other.samples: 5131049Sbinkertn@umich.edu self.minval = min(self.minval, other.minval) 5141049Sbinkertn@umich.edu self.maxval = max(self.maxval, other.maxval) 5151049Sbinkertn@umich.edu self.under += other.under 5161049Sbinkertn@umich.edu self.vec = map(lambda x,y: x + y, self.vec, other.vec) 5171049Sbinkertn@umich.edu self.over += other.over 5181049Sbinkertn@umich.edu return self 5191049Sbinkertn@umich.edu 5201049Sbinkertn@umich.edu def __itruediv__(self, other): 5211049Sbinkertn@umich.edu if not other: 5221049Sbinkertn@umich.edu return self 5231049Sbinkertn@umich.edu self.sums /= other 5241049Sbinkertn@umich.edu self.squares /= other 5251049Sbinkertn@umich.edu self.samples /= other 5261049Sbinkertn@umich.edu 5271049Sbinkertn@umich.edu if self.samples: 5281049Sbinkertn@umich.edu self.under /= other 5291049Sbinkertn@umich.edu for i in xrange(len(self.vec)): 5301049Sbinkertn@umich.edu self.vec[i] /= other 5311049Sbinkertn@umich.edu self.over /= other 5321049Sbinkertn@umich.edu return self 5331049Sbinkertn@umich.edu 5341049Sbinkertn@umich.educlass Dist(Statistic): 5351049Sbinkertn@umich.edu def getValue(self): 5361049Sbinkertn@umich.edu return 0.0 5371049Sbinkertn@umich.edu 5381049Sbinkertn@umich.edu def display(self): 5391049Sbinkertn@umich.edu import display 5401049Sbinkertn@umich.edu if not display.all and not (self.flags & flags.printable): 5411049Sbinkertn@umich.edu return 5421049Sbinkertn@umich.edu 5431049Sbinkertn@umich.edu self.dist.display(self.name, self.desc, self.flags, self.precision) 5441049Sbinkertn@umich.edu 5451049Sbinkertn@umich.edu def comparable(self, other): 5461049Sbinkertn@umich.edu return self.name == other.name and \ 5471049Sbinkertn@umich.edu self.dist.compareable(other.dist) 5481049Sbinkertn@umich.edu 5491049Sbinkertn@umich.edu def __eq__(self, other): 5501049Sbinkertn@umich.edu return self.dist == other.dist 5511049Sbinkertn@umich.edu 5521049Sbinkertn@umich.edu def __isub__(self, other): 5531049Sbinkertn@umich.edu self.dist -= other.dist 5541049Sbinkertn@umich.edu return self 5551049Sbinkertn@umich.edu 5561049Sbinkertn@umich.edu def __iadd__(self, other): 5571049Sbinkertn@umich.edu self.dist += other.dist 5581049Sbinkertn@umich.edu return self 5591049Sbinkertn@umich.edu 5601049Sbinkertn@umich.edu def __itruediv__(self, other): 5611049Sbinkertn@umich.edu if not other: 5621049Sbinkertn@umich.edu return self 5631049Sbinkertn@umich.edu self.dist /= other 5641049Sbinkertn@umich.edu return self 5651049Sbinkertn@umich.edu 5661049Sbinkertn@umich.educlass VectorDist(Statistic): 5671049Sbinkertn@umich.edu def getValue(self): 5681049Sbinkertn@umich.edu return 0.0 5691049Sbinkertn@umich.edu 5701049Sbinkertn@umich.edu def display(self): 5711049Sbinkertn@umich.edu import display 5721049Sbinkertn@umich.edu if not display.all and not (self.flags & flags.printable): 5731049Sbinkertn@umich.edu return 5741049Sbinkertn@umich.edu 5751049Sbinkertn@umich.edu if isinstance(self.dist, SimpleDist): 5761049Sbinkertn@umich.edu return 5771049Sbinkertn@umich.edu 5781049Sbinkertn@umich.edu for dist,sn,sd,i in map(None, self.dist, self.subnames, self.subdescs, 5791049Sbinkertn@umich.edu range(len(self.dist))): 5801049Sbinkertn@umich.edu if len(sn) > 0: 5811049Sbinkertn@umich.edu name = '%s.%s' % (self.name, sn) 5821049Sbinkertn@umich.edu else: 5831049Sbinkertn@umich.edu name = '%s[%d]' % (self.name, i) 5841049Sbinkertn@umich.edu 5851049Sbinkertn@umich.edu if len(sd) > 0: 5861049Sbinkertn@umich.edu desc = sd 5871049Sbinkertn@umich.edu else: 5881049Sbinkertn@umich.edu desc = self.desc 5891049Sbinkertn@umich.edu 5901049Sbinkertn@umich.edu dist.display(name, desc, self.flags, self.precision) 5911049Sbinkertn@umich.edu 5921049Sbinkertn@umich.edu if (self.flags & flags.total) or 1: 5931049Sbinkertn@umich.edu if isinstance(self.dist[0], SimpleDist): 5941049Sbinkertn@umich.edu disttotal = SimpleDist( \ 5951049Sbinkertn@umich.edu reduce(sums, [d.sums for d in self.dist]), 5961049Sbinkertn@umich.edu reduce(sums, [d.squares for d in self.dist]), 5971049Sbinkertn@umich.edu reduce(sums, [d.samples for d in self.dist])) 5981049Sbinkertn@umich.edu else: 5991049Sbinkertn@umich.edu disttotal = FullDist( \ 6001049Sbinkertn@umich.edu reduce(sums, [d.sums for d in self.dist]), 6011049Sbinkertn@umich.edu reduce(sums, [d.squares for d in self.dist]), 6021049Sbinkertn@umich.edu reduce(sums, [d.samples for d in self.dist]), 6031049Sbinkertn@umich.edu min([d.minval for d in self.dist]), 6041049Sbinkertn@umich.edu max([d.maxval for d in self.dist]), 6051049Sbinkertn@umich.edu reduce(sums, [d.under for d in self.dist]), 6061049Sbinkertn@umich.edu reduce(sums, [d.vec for d in self.dist]), 6071049Sbinkertn@umich.edu reduce(sums, [d.over for d in self.dist]), 6081049Sbinkertn@umich.edu dist[0].min, 6091049Sbinkertn@umich.edu dist[0].max, 6101049Sbinkertn@umich.edu dist[0].bsize, 6111049Sbinkertn@umich.edu dist[0].size) 6121049Sbinkertn@umich.edu 6131049Sbinkertn@umich.edu name = '%s.total' % (self.name) 6141049Sbinkertn@umich.edu desc = self.desc 6151049Sbinkertn@umich.edu disttotal.display(name, desc, self.flags, self.precision) 6161049Sbinkertn@umich.edu 6171049Sbinkertn@umich.edu def comparable(self, other): 6181049Sbinkertn@umich.edu return self.name == other.name and \ 6191049Sbinkertn@umich.edu alltrue(map(lambda x, y : x.comparable(y), 6201049Sbinkertn@umich.edu self.dist, 6211049Sbinkertn@umich.edu other.dist)) 6221049Sbinkertn@umich.edu 6231049Sbinkertn@umich.edu def __eq__(self, other): 6241049Sbinkertn@umich.edu return alltrue(map(lambda x, y : x == y, self.dist, other.dist)) 6251049Sbinkertn@umich.edu 6261049Sbinkertn@umich.edu def __isub__(self, other): 6271049Sbinkertn@umich.edu if issequence(self.dist) and issequence(other.dist): 6281049Sbinkertn@umich.edu for sd,od in zip(self.dist, other.dist): 6291049Sbinkertn@umich.edu sd -= od 6301049Sbinkertn@umich.edu else: 6311049Sbinkertn@umich.edu self.dist -= other.dist 6321049Sbinkertn@umich.edu return self 6331049Sbinkertn@umich.edu 6341049Sbinkertn@umich.edu def __iadd__(self, other): 6351049Sbinkertn@umich.edu if issequence(self.dist) and issequence(other.dist): 6361049Sbinkertn@umich.edu for sd,od in zip(self.dist, other.dist): 6371049Sbinkertn@umich.edu sd += od 6381049Sbinkertn@umich.edu else: 6391049Sbinkertn@umich.edu self.dist += other.dist 6401049Sbinkertn@umich.edu return self 6411049Sbinkertn@umich.edu 6421049Sbinkertn@umich.edu def __itruediv__(self, other): 6431049Sbinkertn@umich.edu if not other: 6441049Sbinkertn@umich.edu return self 6451049Sbinkertn@umich.edu if issequence(self.dist): 6461049Sbinkertn@umich.edu for dist in self.dist: 6471049Sbinkertn@umich.edu dist /= other 6481049Sbinkertn@umich.edu else: 6491049Sbinkertn@umich.edu self.dist /= other 6501049Sbinkertn@umich.edu return self 6511049Sbinkertn@umich.edu 6521049Sbinkertn@umich.educlass Vector2d(Statistic): 6531049Sbinkertn@umich.edu def getValue(self): 6541049Sbinkertn@umich.edu return 0.0 6551049Sbinkertn@umich.edu 6561049Sbinkertn@umich.edu def display(self): 6571049Sbinkertn@umich.edu import display 6581049Sbinkertn@umich.edu if not display.all and not (self.flags & flags.printable): 6591049Sbinkertn@umich.edu return 6601049Sbinkertn@umich.edu 6611049Sbinkertn@umich.edu d = display.VectorDisplay() 6621049Sbinkertn@umich.edu d.__dict__.update(self.__dict__) 6631049Sbinkertn@umich.edu 6641049Sbinkertn@umich.edu if self.__dict__.has_key('ysubnames'): 6651049Sbinkertn@umich.edu ysubnames = list(self.ysubnames) 6661049Sbinkertn@umich.edu slack = self.x - len(ysubnames) 6671049Sbinkertn@umich.edu if slack > 0: 6681049Sbinkertn@umich.edu ysubnames.extend(['']*slack) 6691049Sbinkertn@umich.edu else: 6701049Sbinkertn@umich.edu ysubnames = range(self.x) 6711049Sbinkertn@umich.edu 6721049Sbinkertn@umich.edu for x,sname in enumerate(ysubnames): 6731049Sbinkertn@umich.edu o = x * self.y 6741049Sbinkertn@umich.edu d.value = self.value[o:o+self.y] 6751049Sbinkertn@umich.edu d.name = '%s[%s]' % (self.name, sname) 6761049Sbinkertn@umich.edu d.display() 6771049Sbinkertn@umich.edu 6781049Sbinkertn@umich.edu if self.flags & flags.total: 6791049Sbinkertn@umich.edu d.value = [] 6801049Sbinkertn@umich.edu for y in range(self.y): 6811049Sbinkertn@umich.edu xtot = 0.0 6821049Sbinkertn@umich.edu for x in range(self.x): 6831049Sbinkertn@umich.edu xtot += self.value[y + x * self.x] 6841049Sbinkertn@umich.edu d.value.append(xtot) 6851049Sbinkertn@umich.edu 6861049Sbinkertn@umich.edu d.name = self.name + '.total' 6871049Sbinkertn@umich.edu d.display() 6881049Sbinkertn@umich.edu 6891049Sbinkertn@umich.edu def comparable(self, other): 6901049Sbinkertn@umich.edu return self.name == other.name and self.x == other.x and \ 6911049Sbinkertn@umich.edu self.y == other.y 6921049Sbinkertn@umich.edu 6931049Sbinkertn@umich.edu def __eq__(self, other): 6941049Sbinkertn@umich.edu return True 6951049Sbinkertn@umich.edu 6961049Sbinkertn@umich.edu def __isub__(self, other): 6971049Sbinkertn@umich.edu return self 6981049Sbinkertn@umich.edu 6991049Sbinkertn@umich.edu def __iadd__(self, other): 7001049Sbinkertn@umich.edu return self 7011049Sbinkertn@umich.edu 7021049Sbinkertn@umich.edu def __itruediv__(self, other): 7031049Sbinkertn@umich.edu if not other: 7041049Sbinkertn@umich.edu return self 7051049Sbinkertn@umich.edu return self 7061049Sbinkertn@umich.edu 7071049Sbinkertn@umich.edudef NewStat(data): 7081049Sbinkertn@umich.edu stat = None 7091049Sbinkertn@umich.edu if data.type == 'SCALAR': 7101049Sbinkertn@umich.edu stat = Scalar(data) 7111049Sbinkertn@umich.edu elif data.type == 'VECTOR': 7121049Sbinkertn@umich.edu stat = Vector(data) 7131049Sbinkertn@umich.edu elif data.type == 'DIST': 7141049Sbinkertn@umich.edu stat = Dist(data) 7151049Sbinkertn@umich.edu elif data.type == 'VECTORDIST': 7161049Sbinkertn@umich.edu stat = VectorDist(data) 7171049Sbinkertn@umich.edu elif data.type == 'VECTOR2D': 7181049Sbinkertn@umich.edu stat = Vector2d(data) 7191049Sbinkertn@umich.edu elif data.type == 'FORMULA': 7201049Sbinkertn@umich.edu stat = Formula(data) 7211049Sbinkertn@umich.edu 7221049Sbinkertn@umich.edu return stat 7231049Sbinkertn@umich.edu 724