From 045773264bb0b2ea6b1ec06679bc1c7e88913ac3 Mon Sep 17 00:00:00 2001
From: fawce <fawce@quantopian.com>
Date: Fri, 8 Mar 2013 14:21:51 -0500
Subject: [PATCH] ENH: Adds a flag for optionally not serializing positions.

So that both computational and memory overhead is reduced,
this turns off serializing positions for cumulative performance.

Positions were essentially being doubled up by being stored
in both cumalative and daily.
---
 zipline/finance/performance.py | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/zipline/finance/performance.py b/zipline/finance/performance.py
index 0a7de692..2c239146 100644
--- a/zipline/finance/performance.py
+++ b/zipline/finance/performance.py
@@ -188,7 +188,11 @@ class PerformanceTracker(object):
             # the cumulative period will be calculated over the entire test.
             self.period_start,
             self.period_end,
+            # don't save the transactions for the cumulative
+            # period
             keep_transactions=False,
+            # don't serialize positions for cumualtive period
+            serialize_positions=False
         )
 
         # this performance period will span just the current market day
@@ -198,8 +202,8 @@ class PerformanceTracker(object):
             # the daily period will be calculated for the market day
             self.market_open,
             self.market_close,
-            # save the transactions for the daily periods
-            keep_transactions=True
+            keep_transactions=True,
+            serialize_positions=True
         )
 
     def __repr__(self):
@@ -446,7 +450,8 @@ class PerformancePeriod(object):
             starting_cash,
             period_open=None,
             period_close=None,
-            keep_transactions=True):
+            keep_transactions=True,
+            serialize_positions=True):
 
         self.period_open = period_open
         self.period_close = period_close
@@ -479,6 +484,7 @@ class PerformancePeriod(object):
         # So as not to avoid creating a new object for each event
         self._portfolio_store = zp.Portfolio()
         self._positions_store = zp.Positions()
+        self.serialize_positions = serialize_positions
 
     def rollover(self):
         self.starting_value = self.ending_value
@@ -621,8 +627,10 @@ class PerformancePeriod(object):
         period. See header comments for a detailed description.
         """
         rval = self.__core_dict()
-        positions = self.get_positions_list()
-        rval['positions'] = positions
+
+        if self.serialize_positions:
+            positions = self.get_positions_list()
+            rval['positions'] = positions
 
         # we want the key to be absent, not just empty
         if self.keep_transactions: