summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--read_layers.py117
1 files changed, 63 insertions, 54 deletions
diff --git a/read_layers.py b/read_layers.py
index 15675b3..de1bd47 100644
--- a/read_layers.py
+++ b/read_layers.py
@@ -24,90 +24,98 @@ def makedirs(dirs):
pass
os.makedirs(dirs)
+
class Layer(object):
- def __init__(self, name, surface):
- self.name = name
- self._surface = surface
+ def __init__(self, info, path):
+ self.name = info.get('name', None)
+ if self.name is None:
+ self.name = os.path.splitext(os.path.basename(path))[0]
+
+ self._scale = info.get('scale', 1.0)
+ self._x_offset = info.get('x-offset', 0.0)
+ self._y_offset = info.get('y-offset', 0.0)
+
+ self._load_file(path)
@property
def width(self):
- return self._surface.get_width()
+ return self._x_offset + (self._orig_width * self._scale)
@property
def height(self):
- return self._surface.get_height()
+ return self._y_offset + (self._orig_height * self._scale)
def draw(self, context):
- context.set_source_surface(self._surface)
- context.paint()
+ context.transform(cairo.Matrix(x0=self._x_offset, y0=self._y_offset))
+ context.transform(cairo.Matrix(xx=self._scale, yy=self._scale))
-class LayerReader(object):
- def __init__(self, layer_type=None):
- self.layer_type = Layer if layer_type is None else layer_type
+ def _load_file(self, path):
+ raise NotImplementedError
- def read(self, path):
- info_file = '{0}.txt'.format(path)
- if os.path.exists(info_file):
- with open(info_file, 'r') as f:
- info = yaml.safe_load(f)
- else:
- info = {}
- if path.endswith('.pdf'):
- from gi.repository import Poppler
- document = Poppler.Document.new_from_file('file://{0}'.format(path), None)
- page = document.get_page(0)
- orig_width, orig_height = page.get_size()
+class PdfLayer(Layer):
+ def _load_file(self, path):
+ from gi.repository import Poppler
+ document = Poppler.Document.new_from_file('file://{0}'.format(path), None)
+ self._page = document.get_page(0)
- def draw_layer(context):
- page.render(context)
- elif path.endswith('.png'):
- image = cairo.ImageSurface.create_from_png(path)
- orig_width, orig_height = image.get_width(), image.get_height()
+ @property
+ def _orig_width(self):
+ return self._page.get_size()[0]
- def draw_layer(context):
- context.set_source_surface(image)
- context.paint()
- else:
- raise RuntimeError("Unsupported Format for '{0}'".format(path))
+ @property
+ def _orig_height(self):
+ return self._page.get_size()[1]
- layer_name = info.get('name', None)
- if layer_name is None:
- layer_name = os.path.splitext(os.path.basename(path))[0]
+ def draw(self, context):
+ super(PdfLayer, self).draw(context)
+ self._page.render(context)
- scale = info.get('scale', 1.0)
- x_offset = info.get('x-offset', 0.0)
- y_offset = info.get('y-offset', 0.0)
- width = int(orig_width * scale + x_offset)
- height = int(orig_height * scale + y_offset)
+class PngLayer(Layer):
+ def _load_file(self, path):
+ self._image = cairo.ImageSurface.create_from_png(path)
- layer_surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, width, height)
- layer_context = cairo.Context(layer_surface)
+ @property
+ def _orig_width(self):
+ return self._image.get_width()
- # Init surface to white
- layer_context.set_source_rgba(1.0, 1.0, 1.0, 1.0)
- layer_context.paint()
+ @property
+ def _orig_height(self):
+ return self._image.get_height()
- layer_context.translate(x_offset, y_offset)
- layer_context.scale(scale, scale)
+ def draw(self, context):
+ super(PngLayer, self).draw(context)
+ context.set_source_surface(self._image)
+ context.paint()
- draw_layer(layer_context)
- return self.layer_type(layer_name, layer_surface)
class LayerLoader(object):
- def __init__(self, reader=None):
- self.reader = LayerReader() if reader is None else reader
-
def load(self, layer_path):
layers = []
for layer_file in sorted(os.listdir(layer_path)):
if layer_file.endswith('.txt'):
continue
layer_file = os.path.join(layer_path, layer_file)
- layers.append(self.reader.read(layer_file))
+ layers.append(self.read(layer_file))
return layers
+ def read(self, path):
+ info_file = '{0}.txt'.format(path)
+ if os.path.exists(info_file):
+ with open(info_file, 'r') as f:
+ info = yaml.safe_load(f)
+ else:
+ info = {}
+
+ if path.endswith('.pdf'):
+ return PdfLayer(info, path)
+ elif path.endswith('.png'):
+ return PngLayer(info, path)
+ else:
+ raise RuntimeError("Unsupported Format for '{0}'".format(path))
+
+
class TileGenerator(object):
def __init__(self, layer, width=None, height=None, tile_size=None, scale=0):
self.layer = layer
@@ -156,14 +164,15 @@ class TileGenerator(object):
tile_path = os.path.join(path, str(max_zoom_level - zoom_level), str(x), '{0}.png'.format(y))
tile_transform = column_transform * cairo.Matrix(y0=-self.tile_size * y)
tile_context.set_matrix(tile_transform)
- print "Matrix for %s is %r" % (tile_path, tile_context.get_matrix())
tile_context.set_source_rgba(1.0, 1.0, 1.0, 1.0)
tile_context.paint()
layer.draw(tile_context)
+
tile_surface.write_to_png(tile_path)
+
if __name__ == '__main__':
layer_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'layers')
tiles_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'tiles')