mapnik/tests/python_tests/cairo_test.py

#!/usr/bin/env python

import os
import shutil
import mapnik
from nose.tools import *
from utilities import execution_path, run_all

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('.'))

def make_tmp_map():
    m = mapnik.Map(512,512)
    m.background_color = mapnik.Color('steelblue')
    ds = mapnik.MemoryDatasource()
    context = mapnik.Context()
    context.push('Name')
    f = mapnik.Feature(context,1)
    f['Name'] = 'Hello'
    f.add_geometries_from_wkt('POINT (0 0)')
    ds.add_feature(f)
    s = mapnik.Style()
    r = mapnik.Rule()
    sym = mapnik.MarkersSymbolizer()
    sym.allow_overlap = True
    r.symbols.append(sym)
    s.rules.append(r)
    lyr = mapnik.Layer('Layer')
    lyr.datasource = ds
    lyr.styles.append('style')
    m.append_style('style',s)
    m.layers.append(lyr)
    return m

def draw_title(m,ctx,text,size=10,color=mapnik.Color('black')):
    """ Draw a Map Title near the top of a page."""
    middle = m.width/2.0
    ctx.set_source_rgba(*cairo_color(color))
    ctx.select_font_face("DejaVu Sans Book", cairo.FONT_SLANT_NORMAL, cairo.FONT_WEIGHT_NORMAL)
    ctx.set_font_size(size)
    x_bearing, y_bearing, width, height = ctx.text_extents(text)[:4]
    ctx.move_to(middle - width / 2 - x_bearing, 20.0 - height / 2 - y_bearing)
    ctx.show_text(text)

def draw_neatline(m,ctx):
    w,h = m.width, m.height
    ctx.set_source_rgba(*cairo_color(mapnik.Color('black')))
    outline = [
      [0,0],[w,0],[w,h],[0,h]
    ]
    ctx.set_line_width(1)
    for idx,pt in enumerate(outline):
        if (idx == 0):
          ctx.move_to(*pt)
        else:
          ctx.line_to(*pt)
    ctx.close_path()
    inset = 6
    inline = [
      [inset,inset],[w-inset,inset],[w-inset,h-inset],[inset,h-inset]
    ]
    ctx.set_line_width(inset/2)
    for idx,pt in enumerate(inline):
        if (idx == 0):
          ctx.move_to(*pt)
        else:
          ctx.line_to(*pt)
    ctx.close_path()
    ctx.stroke()

def cairo_color(c):
    """ Return a Cairo color tuple from a Mapnik Color."""
    ctx_c = (c.r/255.0,c.g/255.0,c.b/255.0,c.a/255.0)
    return ctx_c

if mapnik.has_pycairo():
    import cairo

    def test_passing_pycairo_context_svg():
        m = make_tmp_map()
        m.zoom_to_box(mapnik.Box2d(-180,-90,180,90))
        test_cairo_file = '/tmp/mapnik-cairo-context-test.svg'
        surface = cairo.SVGSurface(test_cairo_file, m.width, m.height)
        expected_cairo_file = './images/pycairo/cairo-cairo-expected.svg'
        context = cairo.Context(surface)
        mapnik.render(m,context)
        draw_title(m,context,"Hello Map",size=20)
        draw_neatline(m,context)
        surface.finish()
        if not os.path.exists(expected_cairo_file):
            print 'generated expected cairo surface file %s' % expected_cairo_file
            shutil.copy(test_cairo_file,expected_cairo_file)
        diff = abs(os.stat(expected_cairo_file).st_size-os.stat(test_cairo_file).st_size)
        msg = 'diff in size (%s) between actual (%s) and expected(%s)' % (diff,test_cairo_file,'tests/python_tests/'+ expected_cairo_file)
        eq_( diff < 1500, True, msg)
        os.remove(test_cairo_file)

    def test_passing_pycairo_context_pdf():
        m = make_tmp_map()
        m.zoom_to_box(mapnik.Box2d(-180,-90,180,90))
        test_cairo_file = '/tmp/mapnik-cairo-context-test.pdf'
        surface = cairo.PDFSurface(test_cairo_file, m.width, m.height)
        expected_cairo_file = './images/pycairo/cairo-cairo-expected.pdf'
        context = cairo.Context(surface)
        mapnik.render(m,context)
        draw_title(m,context,"Hello Map",size=20)
        draw_neatline(m,context)
        surface.finish()
        if not os.path.exists(expected_cairo_file):
            print 'generated expected cairo surface file %s' % expected_cairo_file
            shutil.copy(test_cairo_file,expected_cairo_file)
        diff = abs(os.stat(expected_cairo_file).st_size-os.stat(test_cairo_file).st_size)
        msg = 'diff in size (%s) between actual (%s) and expected(%s)' % (diff,test_cairo_file,'tests/python_tests/'+ expected_cairo_file)
        eq_( diff < 1500, True, msg)
        os.remove(test_cairo_file)

    def test_passing_pycairo_context_png():
        m = make_tmp_map()
        m.zoom_to_box(mapnik.Box2d(-180,-90,180,90))
        test_cairo_file = '/tmp/mapnik-cairo-context-test.png'
        surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, m.width, m.height)
        expected_cairo_file = './images/pycairo/cairo-cairo-expected.png'
        expected_cairo_file2 = './images/pycairo/cairo-cairo-expected-reduced.png'
        context = cairo.Context(surface)
        mapnik.render(m,context)
        draw_title(m,context,"Hello Map",size=20)
        draw_neatline(m,context)
        surface.write_to_png(test_cairo_file)
        reduced_color_image = test_cairo_file.replace('png','-mapnik.png')
        im = mapnik.Image.from_cairo(surface)
        im.save(reduced_color_image,'png8')
        surface.finish()
        if not os.path.exists(expected_cairo_file):
            print 'generated expected cairo surface file %s' % expected_cairo_file
            shutil.copy(test_cairo_file,expected_cairo_file)
        diff = abs(os.stat(expected_cairo_file).st_size-os.stat(test_cairo_file).st_size)
        msg = 'diff in size (%s) between actual (%s) and expected(%s)' % (diff,test_cairo_file,'tests/python_tests/'+ expected_cairo_file)
        eq_( diff < 500, True, msg)
        os.remove(test_cairo_file)
        if not os.path.exists(expected_cairo_file2):
            print 'generated expected cairo surface file %s' % expected_cairo_file2
            shutil.copy(reduced_color_image,expected_cairo_file2)
        diff = abs(os.stat(expected_cairo_file2).st_size-os.stat(reduced_color_image).st_size)
        msg = 'diff in size (%s) between actual (%s) and expected(%s)' % (diff,reduced_color_image,'tests/python_tests/'+ expected_cairo_file2)
        eq_( diff < 500, True, msg)
        os.remove(reduced_color_image)

    if 'sqlite' in mapnik.DatasourceCache.plugin_names():
        def _pycairo_surface(type,sym):
                test_cairo_file = '/tmp/mapnik-cairo-surface-test.%s.%s' % (sym,type)
                expected_cairo_file = './images/pycairo/cairo-surface-expected.%s.%s' % (sym,type)
                m = mapnik.Map(256,256)
                mapnik.load_map(m,'../data/good_maps/%s_symbolizer.xml' % sym)
                m.zoom_all()
                if hasattr(cairo,'%sSurface' % type.upper()):
                    surface = getattr(cairo,'%sSurface' % type.upper())(test_cairo_file, m.width,m.height)
                    mapnik.render(m, surface)
                    surface.finish()
                    if not os.path.exists(expected_cairo_file):
                        print 'generated expected cairo surface file %s' % expected_cairo_file
                        shutil.copy(test_cairo_file,expected_cairo_file)
                    diff = abs(os.stat(expected_cairo_file).st_size-os.stat(test_cairo_file).st_size)
                    msg = 'diff in size (%s) between actual (%s) and expected(%s)' % (diff,test_cairo_file,'tests/python_tests/'+ expected_cairo_file)
                    eq_( diff < 2100, True, msg)
                    os.remove(test_cairo_file)
                    return True
                else:
                    print 'skipping cairo.%s test since surface is not available' % type.upper()
                    return True

        def test_pycairo_svg_surface1():
            eq_(_pycairo_surface('svg','point'),True)

        def test_pycairo_svg_surface2():
            eq_(_pycairo_surface('svg','building'),True)

        def test_pycairo_svg_surface3():
            eq_(_pycairo_surface('svg','polygon'),True)

        def test_pycairo_pdf_surface1():
            eq_(_pycairo_surface('pdf','point'),True)

        def test_pycairo_pdf_surface2():
            eq_(_pycairo_surface('pdf','building'),True)

        def test_pycairo_pdf_surface3():
            eq_(_pycairo_surface('pdf','polygon'),True)

if __name__ == "__main__":
    setup()
    run_all(eval(x) for x in dir() if x.startswith("test_"))