# HG changeset patch # User Nicolas Saunier # Date 1343346396 14400 # Node ID 7a3bf04cf01614e8437634cb9062a21f3fa37fa8 # Parent a04a6af4b81032a5a1153272d4db0ba65beb204b added plotting of indicators diff -r a04a6af4b810 -r 7a3bf04cf016 python/extrapolation.py --- a/python/extrapolation.py Thu Jul 26 03:54:41 2012 -0400 +++ b/python/extrapolation.py Thu Jul 26 19:46:36 2012 -0400 @@ -167,16 +167,18 @@ saveSafetyPoints(outCZ, obj1.num, obj2.num, i, crossingZones[i]) if debug: - from matplotlib.pyplot import figure, axis + from matplotlib.pyplot import figure, axis, title figure() obj1.draw('r') obj2.draw('b') for et in extrapolatedTrajectories1: et.predictPosition(timeHorizon) et.draw('rx') + for et in extrapolatedTrajectories2: et.predictPosition(timeHorizon) et.draw('bx') + title('instant {0}'.format(i)) axis('equal') diff -r a04a6af4b810 -r 7a3bf04cf016 python/indicators.py --- a/python/indicators.py Thu Jul 26 03:54:41 2012 -0400 +++ b/python/indicators.py Thu Jul 26 19:46:36 2012 -0400 @@ -75,6 +75,14 @@ else: return values + def plot(self, options = '', **kwargs): + from matplotlib.pylab import plot + if not self.timeInterval and type(self.values)==dict: + time = sorted(self.values.keys()) + plot(time, [self.values[i] for i in time], options, **kwargs) + else: + plot(list(getTimeInterval()), self.values, options, **kwargs) + class SeverityIndicator(TemporalIndicator): '''Class for severity indicators field mostSevereIsMax is True