Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Improvements to generate mask script
- Rendered images can now be fetched from file:// urls
- Control image path can be deduced by just specifying a partial
match for the control image name instead of the full control
image path
  • Loading branch information
nyalldawson committed Jun 28, 2015
1 parent b6ae2e7 commit 3f6b5fa
Showing 1 changed file with 32 additions and 1 deletion.
33 changes: 32 additions & 1 deletion scripts/generate_test_mask_image.py
Expand Up @@ -12,6 +12,7 @@
from PyQt4.QtGui import *
import struct
import urllib2
import glob

def error ( msg ):
print msg
Expand All @@ -25,7 +26,7 @@ def colorDiff( c1, c2 ):
return max( redDiff, greenDiff, blueDiff, alphaDiff )

def imageFromPath(path):
if ( path[:7] == 'http://' ):
if ( path[:7] == 'http://' or path[:7] == 'file://' ):
#fetch remote image
data = urllib2.urlopen(path).read()
image = QImage()
Expand All @@ -34,6 +35,34 @@ def imageFromPath(path):
image = QImage( path )
return image

def getControlImagePath(path):
if os.path.isfile(path):
return path

#else try and find matching test image
script_folder = os.path.dirname(os.path.realpath(sys.argv[0]))
control_images_folder = os.path.join( script_folder, '../tests/testdata/control_images')

matching_control_images = [x[0] for x in os.walk(control_images_folder) if path in x[0]]
if len(matching_control_images) > 1:
error('Found multiple matching control images for {}'.format(path))
elif len(matching_control_images) == 0:
error('No matching control images found for {}'.format(path))

found_control_image_path = matching_control_images[0]

#check for a single matching expected image
images = glob.glob( os.path.join(found_control_image_path, '*.png') )
filtered_images = [i for i in images if not i[-9:] == '_mask.png']
if len(filtered_images) > 1:
error('Found multiple matching control images for {}'.format(path))
elif len(filtered_images) == 0:
error('No matching control images found for {}'.format(path))

found_image = filtered_images[0]
print 'Found matching control image: {}'.format(found_image)
return found_image

def updateMask(control_image_path, rendered_image_path, mask_image_path):
control_image = imageFromPath( control_image_path )
if not control_image:
Expand Down Expand Up @@ -92,6 +121,8 @@ def updateMask(control_image_path, rendered_image_path, mask_image_path):
parser.add_argument('mask_image', nargs='?', default=None)
args = parser.parse_args()

args.control_image = getControlImagePath( args.control_image)

if not args.mask_image:
args.mask_image = args.control_image[:-4] + '_mask.png'

Expand Down

0 comments on commit 3f6b5fa

Please sign in to comment.