# coding=utf8
import os

from nose.tools import eq_
from nose.plugins.skip import SkipTest

import mapnik

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

if 'shape' in mapnik.DatasourceCache.plugin_names():

    #@raises(RuntimeError)
    def test_zoom_all_will_fail():
        if not os.path.exists('../data/good_maps/wgs842merc_reprojection.xml'):
            raise SkipTest

        m = mapnik.Map(512, 512)
        mapnik.load_map(m, '../data/good_maps/wgs842merc_reprojection.xml')
        m.zoom_all()

    def test_zoom_all_will_work_with_max_extent():
        if not os.path.exists('../data/good_maps/wgs842merc_reprojection.xml'):
            raise SkipTest

        m = mapnik.Map(512, 512)
        mapnik.load_map(m, '../data/good_maps/wgs842merc_reprojection.xml')
        merc_bounds = mapnik.Box2d(-20037508.34, -
                                   20037508.34, 20037508.34, 20037508.34)
        m.maximum_extent = merc_bounds
        m.zoom_all()
        # note - fixAspectRatio is being called, then re-clipping to maxextent
        # which makes this hard to predict
        # eq_(m.envelope(),merc_bounds)

        #m = mapnik.Map(512,512)
        # mapnik.load_map(m,'../data/good_maps/wgs842merc_reprojection.xml')
        #merc_bounds = mapnik.Box2d(-20037508.34,-20037508.34,20037508.34,20037508.34)
        # m.zoom_to_box(merc_bounds)
        # eq_(m.envelope(),merc_bounds)

    def test_visual_zoom_all_rendering1():
        if not os.path.exists('../data/good_maps/wgs842merc_reprojection.xml'):
            raise SkipTest

        m = mapnik.Map(512, 512)
        mapnik.load_map(m, '../data/good_maps/wgs842merc_reprojection.xml')
        merc_bounds = mapnik.Box2d(-20037508.34, -
                                   20037508.34, 20037508.34, 20037508.34)
        m.maximum_extent = merc_bounds
        m.zoom_all()
        im = mapnik.Image(512, 512)
        mapnik.render(m, im)
        actual = '/tmp/mapnik-wgs842merc-reprojection-render.png'
        expected = 'images/support/mapnik-wgs842merc-reprojection-render.png'
        im.save(actual, 'png32')
        expected_im = mapnik.Image.open(expected)
        eq_(im.tostring('png32'),
            expected_im.tostring('png32'),
            'failed comparing actual (%s) and expected (%s)' % (actual,
                                                                'test/python_tests/' + expected))

    def test_visual_zoom_all_rendering2():
        if not os.path.exists('../data/good_maps/wgs842merc_reprojection.xml'):
            raise SkipTest

        m = mapnik.Map(512, 512)
        mapnik.load_map(m, '../data/good_maps/merc2wgs84_reprojection.xml')
        m.zoom_all()
        im = mapnik.Image(512, 512)
        mapnik.render(m, im)
        actual = '/tmp/mapnik-merc2wgs84-reprojection-render.png'
        expected = 'images/support/mapnik-merc2wgs84-reprojection-render.png'
        im.save(actual, 'png32')
        expected_im = mapnik.Image.open(expected)
        eq_(im.tostring('png32'),
            expected_im.tostring('png32'),
            'failed comparing actual (%s) and expected (%s)' % (actual,
                                                                'test/python_tests/' + expected))

    # maximum-extent read from map.xml
    def test_visual_zoom_all_rendering3():
        if not os.path.exists('../data/good_maps/bounds_clipping.xml'):
            raise SkipTest

        m = mapnik.Map(512, 512)
        mapnik.load_map(m, '../data/good_maps/bounds_clipping.xml')
        m.zoom_all()
        im = mapnik.Image(512, 512)
        mapnik.render(m, im)
        actual = '/tmp/mapnik-merc2merc-reprojection-render1.png'
        expected = 'images/support/mapnik-merc2merc-reprojection-render1.png'
        im.save(actual, 'png32')
        expected_im = mapnik.Image.open(expected)
        eq_(im.tostring('png32'),
            expected_im.tostring('png32'),
            'failed comparing actual (%s) and expected (%s)' % (actual,
                                                                'test/python_tests/' + expected))

    # no maximum-extent
    def test_visual_zoom_all_rendering4():
        if not os.path.exists('../data/good_maps/bounds_clipping.xml'):
            raise SkipTest

        m = mapnik.Map(512, 512)
        mapnik.load_map(m, '../data/good_maps/bounds_clipping.xml')
        m.maximum_extent = None
        m.zoom_all()
        im = mapnik.Image(512, 512)
        mapnik.render(m, im)
        actual = '/tmp/mapnik-merc2merc-reprojection-render2.png'
        expected = 'images/support/mapnik-merc2merc-reprojection-render2.png'
        im.save(actual, 'png32')
        expected_im = mapnik.Image.open(expected)
        eq_(im.tostring('png32'),
            expected_im.tostring('png32'),
            'failed comparing actual (%s) and expected (%s)' % (actual,
                                                                'test/python_tests/' + expected))

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