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):
-    """Finds a rule whose name is |component_name|. """
+  def find_rule(self, component_name, after_rule=None):
+    """Finds a rule whose name is |component_name|.
+
+    If there are multiple rules with same component name,
+    use |after_rule| to search the rule after it.
+    """
+    found_after_rule = False
     for rule in self._rules:
+      if after_rule and not found_after_rule:
+        if rule == after_rule:
+          found_after_rule = 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)