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