1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
# #!/usr/bin/env python
# # -*- coding: utf-8 -*-
# import os
# import math
# import mapnik
# import sys
# from utilities import execution_path, run_all
# from nose.tools import *
# def setup():
# # All of the paths used are relative, if we run the tests
# # from another directory we need to chdir()
# os.chdir(execution_path('.'))
# class PointDatasource(mapnik.PythonDatasource):
# def __init__(self):
# super(PointDatasource, self).__init__(
# geometry_type = mapnik.DataGeometryType.Point,
# envelope = mapnik.Box2d(0,-10,100,110),
# data_type = mapnik.DataType.Vector
# )
# def features(self, query):
# return mapnik.PythonDatasource.wkt_features(
# keys = ('label',),
# features = (
# ( 'POINT (5 6)', { 'label': 'foo-bar'} ),
# ( 'POINT (60 50)', { 'label': 'buzz-quux'} ),
# )
# )
# class ConcentricCircles(object):
# def __init__(self, centre, bounds, step=1):
# self.centre = centre
# self.bounds = bounds
# self.step = step
# class Iterator(object):
# def __init__(self, container):
# self.container = container
# centre = self.container.centre
# bounds = self.container.bounds
# step = self.container.step
# self.radius = step
# def next(self):
# points = []
# for alpha in xrange(0, 361, 5):
# x = math.sin(math.radians(alpha)) * self.radius + self.container.centre[0]
# y = math.cos(math.radians(alpha)) * self.radius + self.container.centre[1]
# points.append('%s %s' % (x,y))
# circle = 'POLYGON ((' + ','.join(points) + '))'
# # has the circle grown so large that the boundary is entirely within it?
# tl = (self.container.bounds.maxx, self.container.bounds.maxy)
# tr = (self.container.bounds.maxx, self.container.bounds.maxy)
# bl = (self.container.bounds.minx, self.container.bounds.miny)
# br = (self.container.bounds.minx, self.container.bounds.miny)
# def within_circle(p):
# delta_x = p[0] - self.container.centre[0]
# delta_y = p[0] - self.container.centre[0]
# return delta_x*delta_x + delta_y*delta_y < self.radius*self.radius
# if all(within_circle(p) for p in (tl,tr,bl,br)):
# raise StopIteration()
# self.radius += self.container.step
# return ( circle, { } )
# def __iter__(self):
# return ConcentricCircles.Iterator(self)
# class CirclesDatasource(mapnik.PythonDatasource):
# def __init__(self, centre_x=-20, centre_y=0, step=10):
# super(CirclesDatasource, self).__init__(
# geometry_type = mapnik.DataGeometryType.Polygon,
# envelope = mapnik.Box2d(-180, -90, 180, 90),
# data_type = mapnik.DataType.Vector
# )
# # note that the plugin loader will set all arguments to strings and will not try to parse them
# centre_x = int(centre_x)
# centre_y = int(centre_y)
# step = int(step)
# self.centre_x = centre_x
# self.centre_y = centre_y
# self.step = step
# def features(self, query):
# centre = (self.centre_x, self.centre_y)
# return mapnik.PythonDatasource.wkt_features(
# keys = (),
# features = ConcentricCircles(centre, query.bbox, self.step)
# )
# if 'python' in mapnik.DatasourceCache.plugin_names():
# # make sure we can load from ourself as a module
# sys.path.append(execution_path('.'))
# def test_python_point_init():
# ds = mapnik.Python(factory='python_plugin_test:PointDatasource')
# e = ds.envelope()
# assert_almost_equal(e.minx, 0, places=7)
# assert_almost_equal(e.miny, -10, places=7)
# assert_almost_equal(e.maxx, 100, places=7)
# assert_almost_equal(e.maxy, 110, places=7)
# def test_python_circle_init():
# ds = mapnik.Python(factory='python_plugin_test:CirclesDatasource')
# e = ds.envelope()
# assert_almost_equal(e.minx, -180, places=7)
# assert_almost_equal(e.miny, -90, places=7)
# assert_almost_equal(e.maxx, 180, places=7)
# assert_almost_equal(e.maxy, 90, places=7)
# def test_python_circle_init_with_args():
# ds = mapnik.Python(factory='python_plugin_test:CirclesDatasource', centre_x=40, centre_y=7)
# e = ds.envelope()
# assert_almost_equal(e.minx, -180, places=7)
# assert_almost_equal(e.miny, -90, places=7)
# assert_almost_equal(e.maxx, 180, places=7)
# assert_almost_equal(e.maxy, 90, places=7)
# def test_python_point_rendering():
# m = mapnik.Map(512,512)
# mapnik.load_map(m,'../data/python_plugin/python_point_datasource.xml')
# m.zoom_all()
# im = mapnik.Image(512,512)
# mapnik.render(m,im)
# actual = '/tmp/mapnik-python-point-render1.png'
# expected = 'images/support/mapnik-python-point-render1.png'
# im.save(actual)
# expected_im = mapnik.Image.open(expected)
# eq_(im.tostring('png32'),expected_im.tostring('png32'),
# 'failed comparing actual (%s) and expected (%s)' % (actual,'tests/python_tests/'+ expected))
# def test_python_circle_rendering():
# m = mapnik.Map(512,512)
# mapnik.load_map(m,'../data/python_plugin/python_circle_datasource.xml')
# m.zoom_all()
# im = mapnik.Image(512,512)
# mapnik.render(m,im)
# actual = '/tmp/mapnik-python-circle-render1.png'
# expected = 'images/support/mapnik-python-circle-render1.png'
# im.save(actual)
# expected_im = mapnik.Image.open(expected)
# eq_(im.tostring('png32'),expected_im.tostring('png32'),
# 'failed comparing actual (%s) and expected (%s)' % (actual,'tests/python_tests/'+ expected))
# if __name__ == "__main__":
# setup()
# run_all(eval(x) for x in dir() if x.startswith("test_"))
|
