Make dmprof handle long runs better

tools/deep_memory_profiler/subcommands/policies.py

 # Copyright 2013 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import datetime
 import json
 import logging
 import sys
 
 from lib.bucket import BUCKET_ID, COMMITTED
@@ skipping 281 matching lines @@
       max_components = max(max_components, len(policy_set[label].components))
 
     for label in sorted(policy_set):
       components = policy_set[label].components
       if len(policy_set) > 1:
         out.write('%s%s\n' % (label, ',' * (max_components - 1)))
       out.write('%s%s\n' % (
           ','.join(components), ',' * (max_components - len(components))))
 
       LOGGER.info('Applying a policy %s to...' % label)
-      for dump in dumps:
+      for index, dump in enumerate(dumps):
+        if index == 0:
+          first_dump_time = dump.time
         component_sizes = self._apply_policy(
-            dump, pfn_counts_dict, policy_set[label], bucket_set, dumps[0].time)
+            dump, pfn_counts_dict, policy_set[label], bucket_set,
+            first_dump_time)
         s = []
         for c in components:
           if c in ('hour', 'minute', 'second'):
             if isinstance(component_sizes[c], str):
               s.append('%s' % component_sizes[c])
             else:
               s.append('%05.5f' % (component_sizes[c]))
           else:
             s.append('%05.5f' % (component_sizes[c] / 1024.0 / 1024.0))
         out.write('%s%s\n' % (
@@ skipping 19 matching lines @@
       'policies': {},
     }
 
     for label in sorted(policy_set):
       json_base['policies'][label] = {
         'legends': policy_set[label].components,
         'snapshots': [],
       }
 
       LOGGER.info('Applying a policy %s to...' % label)
-      for dump in dumps:
+      for index, dump in enumerate(dumps):
+        if index == 0:
+          first_dump_time = dump.time
         component_sizes = self._apply_policy(
-            dump, pfn_counts_dict, policy_set[label], bucket_set, dumps[0].time)
+            dump, pfn_counts_dict, policy_set[label], bucket_set,
+            first_dump_time)
         component_sizes['dump_path'] = dump.path
         component_sizes['dump_time'] = datetime.datetime.fromtimestamp(
             dump.time).strftime('%Y-%m-%d %H:%M:%S')
         json_base['policies'][label]['snapshots'].append(component_sizes)
 
       bucket_set.clear_component_cache()
 
     json.dump(json_base, out, indent=2, sort_keys=True)
 
     return 0
@@ skipping 17 matching lines @@
         out.write('%s for %s:\n' % (label, dump.path))
         for c in policy_set[label].components:
           if c in ['hour', 'minute', 'second']:
             out.write('%40s %12.3f\n' % (c, component_sizes[c]))
           else:
             out.write('%40s %12d\n' % (c, component_sizes[c]))
 
       bucket_set.clear_component_cache()
 
     return 0