mapnik/tests/visual_tests/test.py

434 lines
19 KiB
Python
Raw Normal View History

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import mapnik
mapnik.logger.set_severity(mapnik.severity_type.None)
#mapnik.logger.set_severity(mapnik.severity_type.Debug)
import shutil
import sys
import os.path
2013-03-25 03:56:14 +01:00
from compare import compare, compare_grids
try:
import json
except ImportError:
import simplejson as json
visual_output_dir = "/tmp/mapnik-visual-images"
2012-03-20 22:06:22 +01:00
defaults = {
'sizes': [(500, 100)],
2013-03-25 03:56:14 +01:00
'scales':[1.0,2.0],
2013-05-20 21:45:06 +02:00
'agg': True,
'cairo': mapnik.has_cairo(),
'grid': mapnik.has_grid_renderer()
2012-03-20 22:06:22 +01:00
}
2013-05-31 23:03:53 +02:00
cairo_threshold = 10
if 'Linux' == os.uname()[0]:
# we assume if linux then you are running packaged cairo
# which is older than the 1.12.14 version we used on OS X
# to generate the expected images, so we'll rachet back the threshold
# https://github.com/mapnik/mapnik/issues/1868
cairo_threshold = 181
2013-05-31 23:03:53 +02:00
def render_cairo(m, output, scale_factor):
mapnik.render_to_file(m, output, 'ARGB32', scale_factor)
# open and re-save as png8 to save space
new_im = mapnik.Image.open(output)
new_im.save(output, 'png8:m=h')
def render_grid(m, output, scale_factor):
grid = mapnik.Grid(m.width, m.height)
mapnik.render_layer(m, grid, layer=0)
utf1 = grid.encode('utf', resolution=4)
open(output,'wb').write(json.dumps(utf1, indent=1))
renderers = [
{ 'name': 'agg',
'render': lambda m, output, scale_factor: mapnik.render_to_file(m, output, 'png8:m=h', scale_factor),
'compare': lambda actual, reference: compare(actual, reference, alpha=True),
'threshold': 0,
'filetype': 'png',
'dir': 'images'
},
{ 'name': 'cairo',
'render': render_cairo,
'compare': lambda actual, reference: compare(actual, reference, alpha=False),
'threshold': cairo_threshold,
'filetype': 'png',
'dir': 'images'
},
{ 'name': 'grid',
'render': render_grid,
'compare': lambda actual, reference: compare_grids(actual, reference, alpha=False),
'threshold': 0,
'filetype': 'json',
'dir': 'grids'
}
]
2012-03-20 22:06:22 +01:00
sizes_many_in_big_range = [(800, 100), (600, 100), (400, 100),
(300, 100), (250, 100), (150, 100), (100, 100)]
sizes_few_square = [(800, 800), (600, 600), (400, 400), (200, 200)]
sizes_many_in_small_range = [(490, 100), (495, 100), (497, 100), (498, 100),
(499, 100), (500, 100), (501, 100), (502, 100), (505, 100), (510, 100)]
default_text_box = mapnik.Box2d(-0.05, -0.01, 0.95, 0.01)
merc_z1_bboxes = {
'0,0':mapnik.Box2d(-20037508.342,0,0,20037508.342), # upper left
'1,0':mapnik.Box2d(0,0,20037508.342,20037508.342), # upper right
'0,1':mapnik.Box2d(-20037508.342,-20037508.342,0,0), # lower left
'1,1':mapnik.Box2d(0,-20037508.342,20037508.342,0) # lower right
}
2012-03-09 02:01:51 +01:00
dirname = os.path.dirname(__file__)
2013-05-20 21:45:06 +02:00
files = {
'list': {'sizes': sizes_many_in_big_range,'bbox':default_text_box},
'simple': {'sizes': sizes_many_in_big_range,'bbox':default_text_box},
'lines-1': {'sizes': sizes_few_square,'bbox':default_text_box},
'lines-2': {'sizes': sizes_few_square,'bbox':default_text_box},
'lines-3': {'sizes': sizes_few_square,'bbox':default_text_box},
2013-10-24 08:59:14 +02:00
'lines-4': {'sizes': sizes_few_square,'bbox':default_text_box},
'lines-5': {'sizes': sizes_few_square,'bbox':default_text_box},
'lines-6': {'sizes': sizes_few_square,'bbox':default_text_box},
'lines-7': {'sizes': sizes_few_square,'bbox':mapnik.Box2d(-1.2, -1.2, 1.2, 1.2)},
2014-02-10 15:03:50 +01:00
'lines-multi-layout-1': {'sizes': [(800,800)],'bbox':default_text_box},
'lines-multi-layout-2': {'sizes': [(800,800)],'bbox':mapnik.Box2d(-1.2, -1.2, 1.2, 1.2)},
'lines-multi-layout-shield': {'sizes': [(800,800)],'bbox':default_text_box},
2013-05-20 21:45:06 +02:00
'lines-shield': {'sizes': sizes_few_square,'bbox':default_text_box},
'collision': {'sizes':[(600,400)]},
'shield-on-polygon': {'sizes':[(600,400)]},
'shield-on-line-spacing-eq-width': {'sizes':[(600,400)]},
2013-06-28 23:22:31 +02:00
'geometry-transform-translate': {'sizes':[(200,200)]},
'geometry-transform-translate-patterns': {'sizes':[(200,200)]},
2013-05-20 21:45:06 +02:00
'marker-svg-opacity':{},
'marker-svg-opacity2':{},
2013-07-10 23:23:17 +02:00
'marker-svg-empty-g-element':{},
2013-05-20 21:45:06 +02:00
'marker-multi-policy': {'sizes':[(600,400)]},
'marker-on-line': {'sizes':[(600,400)],
'bbox': mapnik.Box2d(-10, 0, 15, 20)},
'marker-on-line-and-line-placement': {'sizes':[(600,400)],
'bbox': mapnik.Box2d(-10, 0, 15, 20)},
2013-05-20 21:45:06 +02:00
'marker-on-line-spacing-eq-width': {'sizes':[(600,400)]},
'marker-on-line-spacing-eq-width-overlap': {'sizes':[(600,400)]},
'marker_line_placement_on_points':{},
'marker-with-background-image': {'sizes':[(600,400),(400,600),(257,256)]},
'marker-with-background-image-and-hsla-transform': {'sizes':[(600,400),(400,600),(257,256)]},
2013-05-20 21:45:06 +02:00
'marker-on-hex-grid': {'sizes':[(600,400),(400,600),(257,256)]},
'whole-centroid': {'sizes':[(600,400)],
'bbox': mapnik.Box2d(736908, 4390316, 2060771, 5942346)},
2013-05-20 21:45:06 +02:00
'text-halo-rasterizer': {'sizes':[(600,400)]},
'simple-E': {'bbox':mapnik.Box2d(-0.05, -0.01, 0.95, 0.01)},
'simple-NE': {'bbox':default_text_box},
'simple-NW': {'bbox':default_text_box},
'simple-N': {'bbox':default_text_box},
'simple-SE': {'bbox':default_text_box},
'simple-SW': {'bbox':default_text_box},
'simple-S': {'bbox':default_text_box},
'simple-W': {'bbox':default_text_box},
'formatting-1': {'bbox':default_text_box},
'formatting-2': {'bbox':default_text_box},
'formatting-3': {'bbox':default_text_box},
'formatting-4': {'bbox':default_text_box},
'expressionformat': {'bbox':default_text_box},
'shieldsymbolizer-1': {'sizes': sizes_many_in_small_range,'bbox':default_text_box},
2013-10-24 08:59:14 +02:00
'shieldsymbolizer-2': {'sizes': sizes_many_in_small_range,'bbox':default_text_box},
'shieldsymbolizer-3': {'sizes': sizes_many_in_small_range,'bbox':default_text_box},
'shieldsymbolizer-4': {'sizes': sizes_many_in_small_range,'bbox':default_text_box},
'orientation': {'sizes': [(800, 200)], 'bbox': default_text_box},
'harfbuzz': {'sizes': [(800, 200)], 'bbox': default_text_box},
'hb-fontsets': {'sizes': [(800, 200)], 'bbox': default_text_box},
'khmer': {'sizes': [(800, 200)], 'bbox': default_text_box},
'charspacing': {'sizes': [(200, 400)], 'bbox': default_text_box},
'charspacing-lines': {'sizes': [(300, 300)], 'bbox': default_text_box},
'line_break': {'sizes': [(800, 800)], 'bbox': default_text_box},
2013-05-20 21:45:06 +02:00
'rtl-point': {'sizes': [(200, 200)],'bbox':default_text_box},
'jalign-auto': {'sizes': [(200, 200)],'bbox':default_text_box},
'line-offset': {'sizes':[(900, 250)],'bbox': mapnik.Box2d(-5.192, 50.189, -5.174, 50.195)},
2013-10-24 08:59:14 +02:00
'text-bug1532': {'sizes': [(600, 165)]},
'text-bug1533': {'sizes': [(600, 600)]},
'text-bug1820-1': {'sizes': [(600, 300)], 'bbox': default_text_box},
'text-bug1820+0': {'sizes': [(600, 300)], 'bbox': default_text_box},
'text-bug1820+1': {'sizes': [(600, 300)], 'bbox': default_text_box},
'text-bug2037': {'sizes': [(800, 300)], 'bbox': default_text_box},
'text-expressionformat-color': {'sizes': [(800, 100)], 'bbox': default_text_box},
'text-halign': {'sizes': [(800,800)], 'bbox': default_text_box},
'text-malayalam': {'sizes': [(800, 100)], 'bbox': default_text_box},
'text-bengali': {'sizes': [(800, 100)], 'bbox': default_text_box},
'text-multi-layout-1': {'sizes': [(512,512)], 'bbox':mapnik.Box2d(-1, -1, 1, 1)},
'text-multi-layout-2': {'sizes': [(512,512)], 'bbox':mapnik.Box2d(-1, -1, 1, 1)},
2013-08-21 17:33:24 +02:00
'line-pattern-symbolizer': {'sizes':[(900, 250)],'bbox': mapnik.Box2d(-5.192, 50.189, -5.174, 50.195)},
2013-05-20 21:45:06 +02:00
'tiff-alpha-gdal': {'sizes':[(600,400)]},
'tiff-alpha-broken-assoc-alpha-gdal': {'sizes':[(600,400)]},
'tiff-alpha-gradient-gdal': {'sizes':[(600,400)]},
'tiff-nodata-edge-gdal': {'sizes':[(600,400),(969,793)]},
'tiff-opaque-edge-gdal': {'sizes':[(256,256),(969,793)]},
'tiff-opaque-edge-gdal2': {'sizes':[(600,400),(969,793)]},
'tiff-opaque-edge-raster2': {'sizes':[(600,400),(969,793)]},
'tiff-resampling': {'sizes':[(600,400)]},
2013-07-20 00:56:50 +02:00
'gdal-filter-factor': {'sizes':[(600,400)]},
# https://github.com/mapnik/mapnik/issues/1622
2013-05-20 21:45:06 +02:00
'tiff-edge-alignment-gdal1': {'sizes':[(256,256),(255,257)],
'bbox':mapnik.Box2d(-13267022.12540147,4618019.500877209,-13247454.246160466,4637587.380118214)
},
2013-05-20 21:45:06 +02:00
'tiff-edge-alignment-gdal2': {'sizes':[(256,256),(255,257)],
'bbox':mapnik.Box2d(-13267022.12540147,4598451.621636203,-13247454.246160466,4618019.500877209)
},
'tiff-reprojection-1': {'sizes':[(250,250)]},
2013-07-19 06:39:22 +02:00
# disabled since fixing is not actionable: https://github.com/mapnik/mapnik/issues/1913
#'tiff-reprojection-2': {'sizes':[(250,250)]},
# https://github.com/mapnik/mapnik/issues/1520
# commented because these are not critical failures
2013-05-20 21:45:06 +02:00
#'tiff-alpha-raster': {'sizes':[(600,400)]},
#'tiff-alpha-broken-assoc-alpha-raster': {'sizes':[(600,400)]},
#'tiff-nodata-edge-raster': {'sizes':[(600,400)]},
#'tiff-opaque-edge-raster': {'sizes':[(256,256)]},
'road-casings-grouped-rendering': {'sizes':[(600,600)],
'bbox':mapnik.Box2d(1477001.12245,6890242.37746,1480004.49012,6892244.62256)
},
'road-casings-non-grouped-rendering': {'sizes':[(600,600)],
'bbox':mapnik.Box2d(1477001.12245,6890242.37746,1480004.49012,6892244.62256)
},
'style-level-compositing-tiled-0,0':{'sizes':[(512,512)],'bbox':merc_z1_bboxes['0,0']},
'style-level-compositing-tiled-1,0':{'sizes':[(512,512)],'bbox':merc_z1_bboxes['1,0']},
'style-level-compositing-tiled-0,1':{'sizes':[(512,512)],'bbox':merc_z1_bboxes['0,1']},
2013-07-24 02:00:42 +02:00
'style-level-compositing-tiled-1,1':{'sizes':[(512,512)],'bbox':merc_z1_bboxes['1,1']},
'marker-path-expression':{},
'map-background-image-compositing':{'sizes':[(512,512)]},
'building-symbolizer-opacity':{'sizes':[(512,512)]},
'line-pattern-symbolizer-opacity':{'sizes':[(512,512)]},
'dst-over-compositing':{'sizes':[(512,512)]},
'tiff_colortable':{'sizes':[(256,256)]},
'tiff_colortable_custom_nodata':{'sizes':[(256,256)]},
'vrt_colortable':{'sizes':[(256,256)]},
2013-09-25 23:08:47 +02:00
'raster_colorizer':{'sizes':[(512,512)]},
'raster_symbolizer':{'sizes':[(512,512)]},
'raster-color-to-alpha1':{'sizes':[(512,512)]},
'raster-color-to-alpha2':{'sizes':[(512,512)]},
'raster-color-to-alpha3':{'sizes':[(512,512)]},
'raster-color-to-alpha4':{'sizes':[(512,512)]},
'raster-color-to-alpha5':{'sizes':[(512,512)]},
'colorize-alpha1':{'sizes':[(512,512)]},
'colorize-alpha2':{'sizes':[(512,512)]},
2013-10-04 01:32:38 +02:00
'colorize-alpha3':{'sizes':[(512,512)]},
'image-filters-galore':{'sizes':[(512,512)]},
'image-filters-multi-blur':{'sizes':[(512,512)]},
'image-filters-multi-blur-inflate':{'sizes':[(512,512)]},
'line-opacity-multi-render':{'sizes':[(512,512)]},
2014-01-18 04:41:17 +01:00
'tiff-nodata-rgb':{'sizes':[(512,512)]},
'tiff-nodata-rgba':{'sizes':[(512,512)]},
'tiff-nodata-tolerance':{'sizes':[(512,512)]},
'tiff-nodata-edge-rgba':{'sizes':[(512,512)]},
'marker-vs-point':{'sizes':[(512,512)]},
'line-symbolizer-expressions':{'sizes':[(256,256)]},
'line-symbolizer-expressions-all':{'sizes':[(256,256)]},
'point-symbolizer-expressions':{'sizes':[(256,256)]},
'point-symbolizer-expressions-all':{'sizes':[(256,256)]},
'polygon-symbolizer-expressions':{'sizes':[(256,256)]},
'polygon-symbolizer-expressions-all':{'sizes':[(256,256)]},
2013-05-20 21:45:06 +02:00
}
2013-03-25 03:56:14 +01:00
class Reporting:
DIFF = 1
NOT_FOUND = 2
OTHER = 3
REPLACE = 4
def __init__(self, quiet, overwrite_failures = False):
2013-03-25 03:56:14 +01:00
self.quiet = quiet
self.passed = 0
self.failed = 0
self.overwrite_failures = overwrite_failures
self.errors = [ #(type, actual, expected, diff, message)
]
2013-09-19 04:55:02 +02:00
2013-03-25 03:56:14 +01:00
def result_fail(self, actual, expected, diff):
self.failed += 1
if self.quiet:
2012-12-07 05:14:14 +01:00
sys.stderr.write('\x1b[31m.\x1b[0m')
else:
print '\x1b[31m✘\x1b[0m (\x1b[34m%u different pixels\x1b[0m)' % diff
2013-03-25 03:56:14 +01:00
if self.overwrite_failures:
2013-03-25 03:56:14 +01:00
self.errors.append((self.REPLACE, actual, expected, diff, None))
contents = open(actual, 'r').read()
open(expected, 'wb').write(contents)
else:
self.errors.append((self.DIFF, actual, expected, diff, None))
2013-09-19 04:55:02 +02:00
2013-03-25 03:56:14 +01:00
def result_pass(self, actual, expected, diff):
self.passed += 1
if self.quiet:
2012-12-07 05:14:14 +01:00
sys.stderr.write('\x1b[32m.\x1b[0m')
else:
print '\x1b[32m✓\x1b[0m'
2013-03-25 03:56:14 +01:00
def not_found(self, actual, expected):
self.failed += 1
self.errors.append((self.NOT_FOUND, actual, expected, 0, None))
if self.quiet:
sys.stderr.write('\x1b[33m.\x1b[0m')
else:
print '\x1b[33m?\x1b[0m (\x1b[34mReference file not found, creating\x1b[0m)'
contents = open(actual, 'r').read()
open(expected, 'wb').write(contents)
2013-03-25 03:56:14 +01:00
def other_error(self, expected, message):
self.failed += 1
self.errors.append((self.OTHER, None, expected, 0, message))
if self.quiet:
sys.stderr.write('\x1b[31m.\x1b[0m')
else:
print '\x1b[31m✘\x1b[0m (\x1b[34m%s\x1b[0m)' % message
def make_html_item(self,actual,expected,diff):
item = '''
<div class="expected">
<a href="%s">
<img src="%s" width="100%s">
</a>
</div>
''' % (expected,expected,'%')
item += '<div class="text">%s</div>' % (diff)
item += '''
<div class="actual">
<a href="%s">
<img src="%s" width="100%s">
</a>
</div>
''' % (actual,actual,'%')
return item
def summary(self):
2013-03-25 03:56:14 +01:00
if len(self.errors) == 0:
print '\nAll %s visual tests passed: \x1b[1;32m✓ \x1b[0m' % self.passed
return 0
sortable_errors = []
2013-03-25 03:56:14 +01:00
print "\nVisual rendering: %s failed / %s passed" % (len(self.errors), self.passed)
for idx, error in enumerate(self.errors):
if error[0] == self.OTHER:
print str(idx+1) + ") \x1b[31mfailure to run test:\x1b[0m %s (\x1b[34m%s\x1b[0m)" % (error[2],error[4])
2013-03-25 03:56:14 +01:00
elif error[0] == self.NOT_FOUND:
print str(idx+1) + ") Generating reference image: '%s'" % error[2]
2013-03-25 03:56:14 +01:00
continue
elif error[0] == self.DIFF:
print str(idx+1) + ") \x1b[34m%s different pixels\x1b[0m:\n\t%s (\x1b[31mactual\x1b[0m)\n\t%s (\x1b[32mexpected\x1b[0m)" % (error[3], error[1], error[2])
2013-05-21 06:32:32 +02:00
if '.png' in error[1]: # ignore grids
sortable_errors.append((error[3],error))
2013-03-25 03:56:14 +01:00
elif error[0] == self.REPLACE:
print str(idx+1) + ") \x1b[31mreplaced reference with new version:\x1b[0m %s" % error[2]
if len(sortable_errors):
# drop failure results in folder
vdir = os.path.join(visual_output_dir,'visual-test-results')
if not os.path.exists(vdir):
os.makedirs(vdir)
html_template = open(os.path.join(dirname,'html_report_template.html'),'r').read()
name = 'comparison.html'
failures_realpath = os.path.join(vdir,name)
html_out = open(failures_realpath,'w+')
sortable_errors.sort(reverse=True)
html_body = ''
for item in sortable_errors:
# copy images into single directory
actual = item[1][1]
expected = item[1][2]
diff = item[0]
actual_new = os.path.join(vdir,os.path.basename(actual))
shutil.copy(actual,actual_new)
expected_new = os.path.join(vdir,os.path.basename(expected))
shutil.copy(expected,expected_new)
html_body += self.make_html_item(os.path.relpath(actual_new,vdir),os.path.relpath(expected_new,vdir),diff)
html_out.write(html_template.replace('{{RESULTS}}',html_body))
print 'View failures by opening %s' % failures_realpath
2013-03-25 03:56:14 +01:00
return 1
2013-05-20 21:45:06 +02:00
def render(filename,config, width, height, bbox, scale_factor, reporting):
m = mapnik.Map(width, height)
2013-03-25 03:56:14 +01:00
postfix = "%s-%d-%d-%s" % (filename, width, height, scale_factor)
try:
mapnik.load_map(m, os.path.join(dirname, "styles", "%s.xml" % filename), True)
if bbox is not None:
m.zoom_to_box(bbox)
else:
m.zoom_all()
except Exception, e:
if 'Could not create datasource' in str(e):
return m
2013-03-25 03:56:14 +01:00
reporting.other_error(filename, repr(e))
return m
2013-09-19 04:55:02 +02:00
2013-03-25 03:56:14 +01:00
for renderer in renderers:
# TODO - grid renderer does not support scale_factor yet via python
if renderer['name'] == 'grid' and scale_factor != 1.0:
continue
if config.get(renderer['name'], True):
expected = os.path.join(dirname, renderer['dir'], '%s-%s-reference.%s' %
(postfix, renderer['name'], renderer['filetype']))
actual = os.path.join(visual_output_dir, '%s-%s.%s' %
(postfix, renderer['name'], renderer['filetype']))
if not quiet:
print "\"%s\" with %s..." % (postfix, renderer['name']),
try:
renderer['render'](m, actual, scale_factor)
if not os.path.exists(expected):
reporting.not_found(actual, expected)
else:
2013-03-25 03:56:14 +01:00
diff = renderer['compare'](actual, expected)
if diff > renderer['threshold']:
reporting.result_fail(actual, expected, diff)
else:
reporting.result_pass(actual, expected, diff)
except Exception, e:
reporting.other_error(expected, repr(e))
return m
if __name__ == "__main__":
if '-q' in sys.argv:
quiet = True
sys.argv.remove('-q')
else:
quiet = False
if '--overwrite' in sys.argv:
overwrite_failures = True
sys.argv.remove('--overwrite')
else:
overwrite_failures = False
2013-05-20 21:45:06 +02:00
select_files = {}
if len(sys.argv) > 1:
2012-07-30 01:25:21 +02:00
for name in sys.argv[1:]:
2013-05-20 21:45:06 +02:00
if name in files:
select_files[name]=files[name]
else:
select_files[name]={}
if len(select_files) > 0:
files = select_files
if not os.path.exists(visual_output_dir):
os.makedirs(visual_output_dir)
reporting = Reporting(quiet, overwrite_failures)
for filename in files:
config = dict(defaults)
config.update(files[filename])
for size in config['sizes']:
for scale_factor in config['scales']:
m = render(filename,
config,
size[0],
size[1],
config.get('bbox'),
scale_factor,
reporting)
mapnik.save_map(m, os.path.join(dirname, 'xml_output', "%s-out.xml" % filename))
2012-01-25 00:00:09 +01:00
sys.exit(reporting.summary())