mapnik/tests/visual_tests/test.py
2012-03-23 12:42:11 -07:00

82 lines
2.6 KiB
Python
Executable file

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import mapnik
import sys
import os.path
from compare import compare, summary
defaults = {
'sizes': [(500, 100)],
'bbox': mapnik.Box2d(-0.05, -0.01, 0.95, 0.01)
}
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)]
dirname = os.path.dirname(__file__)
files = [
{'name': "list", 'sizes': sizes_many_in_big_range},
{'name': "simple", 'sizes': sizes_many_in_big_range},
{'name': "lines-1", 'sizes': sizes_few_square},
{'name': "lines-2", 'sizes': sizes_few_square},
{'name': "lines-3", 'sizes': sizes_few_square},
{'name': "lines-shield", 'sizes': sizes_few_square},
{'name': "simple-E"},
{'name': "simple-NE"},
{'name': "simple-NW"},
{'name': "simple-N"},
{'name': "simple-SE"},
{'name': "simple-SW"},
{'name': "simple-S"},
{'name': "simple-W"},
{'name': "formatting-1"},
{'name': "formatting-2"},
{'name': "formatting-3"},
{'name': "formatting-4"},
{'name': "expressionformat"},
{'name': "shieldsymbolizer-1", 'sizes': sizes_many_in_small_range},
{'name': "rtl-point", 'sizes': [(200, 200)]},
{'name': "jalign-auto", 'sizes': [(200, 200)]},
{'name': "line-offset", 'sizes':[(900, 250)],
'bbox': mapnik.Box2d(-5.192, 50.189, -5.174, 50.195)}
]
def render(filename, width, height, bbox):
print "-"*80
print "Rendering style \"%s\" with size %dx%d ... " % (filename, width, height)
print "-"*80
m = mapnik.Map(width, height)
mapnik.load_map(m, os.path.join(dirname, "styles", "%s.xml" % filename), False)
if bbox is not None:
m.zoom_to_box(bbox)
else:
m.zoom_all()
basefn = os.path.join(dirname, "images", '%s-%d' % (filename, width))
mapnik.render_to_file(m, basefn+'-agg.png')
diff = compare(basefn + '-agg.png', basefn + '-reference.png')
if diff > 0:
print "-"*80
print 'Error: %u different pixels' % diff
print "-"*80
return m
if len(sys.argv) == 2:
files = [(sys.argv[1], (500, 500))]
elif len(sys.argv) > 2:
files = [sys.argv[1:]]
for f in files:
config = dict(defaults)
config.update(f)
for size in config['sizes']:
m = render(config['name'], size[0], size[1], config['bbox'])
mapnik.save_map(m, os.path.join(dirname, 'xml_output', "%s-out.xml" % config['name']))
summary()