OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/python |
| 2 # Copyright (c) 2014 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. |
| 5 |
| 6 """Utility to rename a config in some subset of our GM expectation files. |
| 7 |
| 8 Created for http://skbug.com/2752 ('split existing "gpu" GM results into "gl" |
| 9 and "gles"') |
| 10 |
| 11 Run with -h to see usage. |
| 12 |
| 13 Example command lines: |
| 14 rename_config.py gpu gles '.*Android.*' |
| 15 |
| 16 TODO(epoger): Once https://codereview.chromium.org/397103003/ is committed, |
| 17 we should add a unittest. Until then, we can test this as follows: |
| 18 |
| 19 OLD=expectations/gm && NEW=/tmp/expectations && \ |
| 20 rm -rf $NEW && \ |
| 21 cp -a $OLD $NEW && \ |
| 22 gm/rename_config.py msaa4 gles-msaa4 '.*Android.*' \ |
| 23 --expectations-root $NEW && \ |
| 24 diff --recursive $OLD $NEW |
| 25 """ |
| 26 __author__ = 'Elliot Poger' |
| 27 |
| 28 import argparse |
| 29 import os |
| 30 import re |
| 31 |
| 32 import gm_json |
| 33 |
| 34 DEFAULT_EXPECTATIONS_ROOT = os.path.join( |
| 35 os.path.dirname(__file__), os.pardir, 'expectations', 'gm') |
| 36 IMAGE_FILENAME_RE = re.compile(gm_json.IMAGE_FILENAME_PATTERN) |
| 37 |
| 38 |
| 39 class Renamer(object): |
| 40 |
| 41 def __init__(self, args): |
| 42 """ |
| 43 Params: |
| 44 args: the Namespace object generated by argparse.parse_args() |
| 45 """ |
| 46 self._args = args |
| 47 |
| 48 def run(self): |
| 49 """Perform all the subsitutions.""" |
| 50 for path in self._get_file_list(): |
| 51 self._rename_config(path=path, |
| 52 old=self._args.old_config_name, |
| 53 new=self._args.new_config_name) |
| 54 |
| 55 def _rename_config(self, path, old, new): |
| 56 """Renames all instances of a config within a GM expectations file. |
| 57 |
| 58 Params: |
| 59 path: path to file which will be modified in place |
| 60 old: old config name |
| 61 new: new config name |
| 62 """ |
| 63 dic = gm_json.LoadFromFile(file_path=path) |
| 64 expected_results = dic[gm_json.JSONKEY_EXPECTEDRESULTS] |
| 65 orig_keys = expected_results.keys() |
| 66 for key in orig_keys: |
| 67 result = expected_results.pop(key) |
| 68 (testname, config) = IMAGE_FILENAME_RE.match(key).groups() |
| 69 if config == old: |
| 70 config = new |
| 71 key = '%s_%s.png' % (testname, config) |
| 72 expected_results[key] = result |
| 73 gm_json.WriteToFile(json_dict=dic, file_path=path) |
| 74 |
| 75 def _get_file_list(self): |
| 76 """Returns the list of files we want to operate on (the complete path |
| 77 to each file).""" |
| 78 root = self._args.expectations_root |
| 79 regex = re.compile(self._args.builder_name_pattern) |
| 80 return [os.path.join(root, builder, 'expected-results.json') |
| 81 for builder in os.listdir(root) |
| 82 if regex.match(builder)] |
| 83 |
| 84 |
| 85 def main(): |
| 86 parser = argparse.ArgumentParser() |
| 87 parser.add_argument('old_config_name', |
| 88 help=('Config name we want to replace.')) |
| 89 parser.add_argument('new_config_name', |
| 90 help=('Config name we want to replace the old one with.')) |
| 91 parser.add_argument('builder_name_pattern', |
| 92 help=('Regex pattern describing which builders we want ' |
| 93 'to make the substitution for; \'.*\' to perform ' |
| 94 'the replacement on all builders.')) |
| 95 parser.add_argument('--expectations-root', |
| 96 default=DEFAULT_EXPECTATIONS_ROOT, |
| 97 help=('Root of the GM expectations dir; defaults to ' |
| 98 '%(default)s')) |
| 99 args = parser.parse_args() |
| 100 renamer = Renamer(args) |
| 101 renamer.run() |
| 102 |
| 103 if __name__ == '__main__': |
| 104 main() |
OLD | NEW |