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 duplicate 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 copy_config.py gl gles '.*Mac10.7.*' | |
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/copy_config.py 8888 8888-copy '.*Mac10.7.*' \ | |
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 Copier(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 duplications.""" | |
50 for path in self._get_file_list(): | |
51 self._duplicate_config(path=path, | |
52 old=self._args.old_config_name, | |
53 new=self._args.new_config_name) | |
54 | |
55 def _duplicate_config(self, path, old, new): | |
56 """Duplicates 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[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 we want to duplicate.')) | |
89 parser.add_argument('new_config_name', | |
90 help=('Name of the new config we want to create.')) | |
91 parser.add_argument('builder_name_pattern', | |
92 help=('Regex pattern describing which builders we want ' | |
93 'to make the duplication for; \'.*\' to perform ' | |
94 'the duplication 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 copier = Copier(args) | |
101 copier.run() | |
102 | |
103 if __name__ == '__main__': | |
104 main() | |
OLD | NEW |