Index: gm/gm_json.py |
diff --git a/gm/gm_json.py b/gm/gm_json.py |
index 1aac6632f5674b0adec9f7c915c82d2837ec3308..6f4b324d46bbd2ce7606671eeab55851b5ce7848 100644 |
--- a/gm/gm_json.py |
+++ b/gm/gm_json.py |
@@ -15,12 +15,14 @@ __author__ = 'Elliot Poger' |
import io |
import json |
import os |
+import posixpath |
+import re |
# Key strings used in GM results JSON files (both expected-results.json and |
# actual-results.json). |
# |
-# These constants must be kept in sync with the kJsonKey_ constants in |
+# NOTE: These constants must be kept in sync with the kJsonKey_ constants in |
# gm_expectations.cpp ! |
@@ -93,6 +95,14 @@ SKIMAGE_EXPECTATIONS_ROOT = os.path.join('expectations', 'skimage') |
# Pattern used to assemble each image's filename |
IMAGE_FILENAME_PATTERN = '(.+)_(.+)\.png' # matches (testname, config) |
+# Pattern used to create image URLs, relative to some base URL. |
+GM_RELATIVE_URL_FORMATTER = '%s/%s/%s.png' # pass in (hash_type, test_name, |
+ # hash_digest) |
+GM_RELATIVE_URL_PATTERN = '(.+)/(.+)/(.+).png' # matches (hash_type, test_name, |
+ # hash_digest) |
+GM_RELATIVE_URL_RE = re.compile(GM_RELATIVE_URL_PATTERN) |
+ |
+ |
def CreateGmActualUrl(test_name, hash_type, hash_digest, |
gm_actuals_root_url=GM_ACTUALS_ROOT_HTTP_URL): |
"""Return the URL we can use to download a particular version of |
@@ -104,10 +114,40 @@ def CreateGmActualUrl(test_name, hash_type, hash_digest, |
hash_digest: the hash digest of the image to retrieve |
gm_actuals_root_url: root url where actual images are stored |
""" |
- # TODO(epoger): Maybe use url_or_path.join() so that, for testing, this can |
- # return either a URL or a local filepath? |
- return '%s/%s/%s/%s.png' % (gm_actuals_root_url, hash_type, test_name, |
- hash_digest) |
+ return posixpath.join( |
+ gm_actuals_root_url, CreateGmRelativeUrl( |
+ test_name=test_name, hash_type=hash_type, hash_digest=hash_digest)) |
+ |
+ |
+def CreateGmRelativeUrl(test_name, hash_type, hash_digest): |
+ """Returns a relative URL pointing at a test result's image. |
+ |
+ Returns the URL we can use to download a particular version of |
+ the actually-generated image for this particular GM test, |
+ relative to the URL root. |
+ |
+ Args: |
+ test_name: name of the test, e.g. 'perlinnoise' |
+ hash_type: string indicating the hash type used to generate hash_digest, |
+ e.g. JSONKEY_HASHTYPE_BITMAP_64BITMD5 |
+ hash_digest: the hash digest of the image to retrieve |
+ """ |
+ return GM_RELATIVE_URL_FORMATTER % (hash_type, test_name, hash_digest) |
+ |
+ |
+def SplitGmRelativeUrl(url): |
+ """Splits the relative URL into (test_name, hash_type, hash_digest) tuple. |
+ |
+ This is the inverse of CreateGmRelativeUrl(). |
+ |
+ Args: |
+ url: a URL generated with CreateGmRelativeUrl(). |
+ |
+ Returns: (test_name, hash_type, hash_digest) tuple. |
+ """ |
+ hash_type, test_name, hash_digest = GM_RELATIVE_URL_RE.match(url).groups() |
+ return (test_name, hash_type, hash_digest) |
+ |
def LoadFromString(file_contents): |
"""Loads the JSON summary written out by the GM tool. |
@@ -119,6 +159,7 @@ def LoadFromString(file_contents): |
json_dict = json.loads(file_contents) |
return json_dict |
+ |
def LoadFromFile(file_path): |
"""Loads the JSON summary written out by the GM tool. |
Returns a dictionary keyed by the values listed as JSONKEY_ constants |
@@ -126,6 +167,7 @@ def LoadFromFile(file_path): |
file_contents = open(file_path, 'r').read() |
return LoadFromString(file_contents) |
+ |
def WriteToFile(json_dict, file_path): |
"""Writes the JSON summary in json_dict out to file_path. |