Partial plotting correction. May address #277

mementum · mementum · commit f3bba1202652 · 2017-03-12T08:10:59.000+01:00
diff --git a/backtrader/cerebro.py b/backtrader/cerebro.py
@@ -613,7 +613,7 @@ def getbroker(self):
 
     broker = property(getbroker, setbroker)
 
-    def plot(self, plotter=None, numfigs=1, iplot=True, start=0, end=-1,
+    def plot(self, plotter=None, numfigs=1, iplot=True, start=None, end=None,
              **kwargs):
         '''
         Plots the strategies inside cerebro
diff --git a/backtrader/plot/plot.py b/backtrader/plot/plot.py
@@ -113,7 +113,7 @@ def drawtag(self, ax, x, y, facecolor, edgecolor, alpha=0.9, **kwargs):
                       **kwargs)
 
     def plot(self, strategy, figid=0, numfigs=1, iplot=True,
-             start=0, end=-1, **kwargs):
+             start=None, end=None, **kwargs):
         # pfillers={}):
         if not strategy.datas:
             return
@@ -134,12 +134,20 @@ def plot(self, strategy, figid=0, numfigs=1, iplot=True,
         self.calcrows(strategy)
 
         st_dtime = strategy.lines.datetime.plot()
+        if start is None:
+            start = 0
+        if end is None:
+            end = len(st_dtime)
+
         if isinstance(start, datetime.date):
             start = bisect.bisect_left(st_dtime, date2num(start))
 
         if isinstance(end, datetime.date):
             end = bisect.bisect_right(st_dtime, date2num(end))
 
+        if end < 0:
+            end = len(st_dtime) + 1 + end  # -1 =  len() -2 = len() - 1
+
         slen = len(st_dtime[start:end])
         d, m = divmod(slen, numfigs)
         pranges = list()
@@ -194,10 +202,11 @@ def plot(self, strategy, figid=0, numfigs=1, iplot=True,
                         dtidx = bisect.bisect_left(xreal, dt)
                         xdata.append(dtidx)
 
-                    self.pinf.xstart = bisect.bisect_left(
-                        dts, xreal[xdata[0]])
-                    self.pinf.xend = bisect.bisect_right(
-                        dts, xreal[xdata[-1]])
+                    if False:
+                        self.pinf.xstart = bisect.bisect_left(
+                            dts, xreal[xdata[0]])
+                        self.pinf.xend = bisect.bisect_right(
+                            dts, xreal[xdata[-1]])
 
                 for ind in self.dplotsup[data]:
                     self.plotind(
