rebaseline_server live queries: allow comparison against summary files within the Skia repo

gm/rebaseline_server/compare_rendered_pictures.py

 #!/usr/bin/python
 
 """
 Copyright 2014 Google Inc.
 
 Use of this source code is governed by a BSD-style license that can be
 found in the LICENSE file.
 
 Compare results of two render_pictures runs.
 
@@ skipping 37 matching lines @@
 FREEFORM_COLUMN_IDS = [
     COLUMN__TILENUM,
 ]
 ORDERED_COLUMN_IDS = [
     COLUMN__RESULT_TYPE,
     COLUMN__SOURCE_SKP,
     COLUMN__TILED_OR_WHOLE,
     COLUMN__TILENUM,
 ]
 
+# A special "repo:" URL type that we use to refer to Skia repo contents.
+# (Useful for comparing against expectations files we store in our repo.)
+REPO_URL_PREFIX = 'repo:'
+REPO_BASEPATH = os.path.abspath(os.path.join(
+    os.path.dirname(os.path.abspath(__file__)), os.pardir, os.pardir))
+
 
 class RenderedPicturesComparisons(results.BaseComparisons):
   """Loads results from multiple render_pictures runs into an ImagePairSet.
   """
 
   def __init__(self, setA_dirs, setB_dirs, image_diff_db,
                image_base_gs_url=DEFAULT_IMAGE_BASE_GS_URL,
                diff_base_url=None, setA_label='setA',
                setB_label='setB', gs=None,
                truncate_results=False):
     """
     Args:
       setA_dirs: list of root directories to copy all JSON summaries from,
-          and to use as setA within the comparisons
+          and to use as setA within the comparisons. These directories may be
+          gs:// URLs, special "repo:" URLs, or local filepaths.
       setB_dirs: list of root directories to copy all JSON summaries from,
-          and to use as setB within the comparisons
+          and to use as setB within the comparisons. These directories may be
+          gs:// URLs, special "repo:" URLs, or local filepaths.
       image_diff_db: ImageDiffDB instance
       image_base_gs_url: "gs://" URL pointing at the Google Storage bucket/dir
           under which all render_pictures result images can
           be found; this will be used to read images for comparison within
           this code, and included in the ImagePairSet (as an HTTP URL) so its
           consumers know where to download the images from
       diff_base_url: base URL within which the client should look for diff
           images; if not specified, defaults to a "file:///" URL representation
           of image_diff_db's storage_root
       setA_label: description to use for results in setA
@@ skipping 234 matching lines @@
           'got exception while creating ImagePair for'
           ' urlPair=("%s","%s"), source_skp_name="%s", tilenum="%s"' % (
               imageA_relative_url, imageB_relative_url, source_skp_name,
               tilenum))
       return None
 
   def _copy_dir_contents(self, source_dir, dest_dir):
     """Copy all contents of source_dir into dest_dir, recursing into subdirs.
 
     Args:
-      source_dir: path to source dir (GS URL or local filepath)
+      source_dir: path to source dir (GS URL, local filepath, or a special
+          "repo:" URL type that points at a file within our Skia checkout)
       dest_dir: path to destination dir (local filepath)
 
     The copy operates as a "merge with overwrite": any files in source_dir will
     be "overlaid" on top of the existing content in dest_dir.  Existing files
     with the same names will be overwritten.
     """
     if gs_utils.GSUtils.is_gs_url(source_dir):
       (bucket, path) = gs_utils.GSUtils.split_gs_url(source_dir)
       self._gs.download_dir_contents(source_bucket=bucket, source_dir=path,
                                      dest_dir=dest_dir)
+    elif source_dir.lower().startswith(REPO_URL_PREFIX):
+      repo_dir = os.path.join(REPO_BASEPATH, source_dir[len(REPO_URL_PREFIX):])
+      shutil.copytree(repo_dir, dest_dir)
     else:
       shutil.copytree(source_dir, dest_dir)