Merge pull request #69 from WHug0/master

Add return_calculated_values parameter

Author: DanielGoldfarb

src/mplfinance/plotting.py

@@ -127,6 +127,9 @@ def _valid_plot_kwargs():
 
         'returnfig'   : { 'Default'     : False, 
                           'Validator'   : lambda value: isinstance(value,bool) },
+
+        'return_calculated_values': {'Default': None,
+                                     'Validator': lambda value: isinstance(value, dict) and len(value) == 0},
 
     }
 
@@ -299,6 +302,17 @@ def plot( data, **kwargs ):
             else:
                 ax1.plot(xdates, mavprices)
 
+    if config['return_calculated_values'] is not None:
+        retdict = config['return_calculated_values']
+        if ptype == 'renko':
+            retdict['renko_bricks'] = brick_values
+            retdict['renko_dates'] = mdates.num2date(new_dates)
+            if config['volume']:
+                retdict['renko_volumes'] = volumes
+        if mavgs is not None:
+            for i in range(0, len(mavgs)):
+                retdict['mav' + str(mavgs[i])] = mavprices
+
     avg_dist_between_points = (xdates[-1] - xdates[0]) / float(len(xdates))
     minx = xdates[0]  - avg_dist_between_points
     maxx = xdates[-1] + avg_dist_between_points