dmprof expand command supports to analyze multiple components with same name in different allocators

tools/deep_memory_profiler/lib/policy.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 json
 import logging
 import os
 import re
 
 
@@ skipping 107 matching lines @@
     return self._rules
 
   @property
   def version(self):
     return self._version
 
   @property
   def components(self):
     return self._components
 
-  def find_rule(self, component_name):
+  def find_rule(self, component_name, follow_rule=None):

[kouhei (in TOK), 2014/11/04 18:17:03]

follow_rule -> after?

[wensheng, 2014/11/05 01:46:17]

Yes

     """Finds a rule whose name is |component_name|. """

[kouhei (in TOK), 2014/11/04 18:17:03]

Please update comment.

[wensheng, 2014/11/05 01:46:17]

Done.

+    flag = False

[kouhei (in TOK), 2014/11/04 18:17:03]

please use more descriptive variable name.

[wensheng, 2014/11/05 01:46:17]

Done.

     for rule in self._rules:
+      if follow_rule and not flag:
+        if rule == follow_rule:
+          flag = True
+        continue
+
       if rule.name == component_name:
         return rule
     return None
 
   def find_malloc(self, bucket):
     """Finds a matching component name which a given |bucket| belongs to.
 
     Args:
         bucket: A Bucket object to be searched for.
 
@@ skipping 258 matching lines @@
   @staticmethod
   def _load_policies(directory):
     LOGGER.info('Loading policy files.')
     policies = {}
     for label in directory:
       LOGGER.info('  %s: %s' % (label, directory[label]['file']))
       loaded = Policy.load(directory[label]['file'], directory[label]['format'])
       if loaded:
         policies[label] = loaded
     return PolicySet(policies)