1

"""

     2

    Raw tile handling.

     3

"""

     4

     5

from pngtile.application import BaseApplication

     6

from werkzeug import Request, Response, exceptions

     7

     8

import pypngtile

     9

    10

## Coordinates

    11

# width of a tile

    12

TILE_SIZE = 256

    13

    14

# maximum zoom out

    15

MAX_ZOOM = 4

    16

    17

# max. output resolution to allow

    18

MAX_PIXELS = 1920 * 1200

    19

    20

def scale (val, zoom):

    21

    """

    22

        Scale dimension by zoom factor

    23

    24

        zl > 0 -> bigger

    25

        zl < 0 -> smaller

    26

    """

    27

    28

    if zoom > 0:

    29

        return val << zoom

    30

    31

    elif zoom < 0:

    32

        return val >> -zoom

    33

    34

    else:

    35

        return val

    36

    37

def scale_center (val, dim, zoom):

    38

    """

    39

        Scale value about center by zoom.

    40

    """

    41

    43

    44

class TileApplication (BaseApplication):

    45

    def __init__ (self, **opts):

    46

        BaseApplication.__init__(self, **opts)

    47

    48

    def render_region (self, request, image):

    49

        """

    50

            Handle request for an image region

    51

        """

    52

    53

        width = int(request.args['w'])

    54

        height = int(request.args['h'])

    55

        x = int(request.args['x'])

    56

        y = int(request.args['y'])

    57

        zoom = int(request.args.get('zoom', "0"))

    58

    59

        # safety limit

    60

        if width * height > MAX_PIXELS:

    61

            raise exceptions.BadRequest("Image size: %d * %d > %d" % (width, height, MAX_PIXELS))

    62

    63

        if zoom > MAX_ZOOM:

    64

            raise exceptions.BadRequest("Image zoom: %d > %d" % (zoom, MAX_ZOOM))

    65

    68

    69

        try:

    70

            return image.tile_mem(width, height, x, y, zoom)

    71

    72

        except pypngtile.Error as error:

    73

            raise exceptions.BadRequest(str(error))

    74

    75

    def render_tile (self, request, image):

    76

        """

    77

            Handle request for image tile

    78

        """

    79

    80

        width = TILE_SIZE

    81

        height = TILE_SIZE

    82

        x = int(request.args['x'])

    83

        y = int(request.args['y'])

    84

        zoom = int(request.args.get('zoom', "0"))

    85

    86

        if zoom > MAX_ZOOM:

    87

            raise exceptions.BadRequest("Image zoom: %d > %d" % (zoom, MAX_ZOOM))

    88

    89

        x = scale(x * width, zoom)

    90

        y = scale(y * height, zoom)

    91

    92

        try:

    93

            return image.tile_mem(width, height, x, y, zoom)

    94

    95

        except pypngtile.Error as error:

    96

            raise exceptions.BadRequest(str(error))

    97

    98

    def handle (self, request):

    99

        """

   100

            Handle request for an image

   101

        """

   102

   103

        try:

   104

            image, name = self.get_image(request.path)

   105

        except pypngtile.Error as error:

   106

            raise exceptions.BadRequest(str(error))

   107

   108

        if 'w' in request.args and 'h' in request.args and 'x' in request.args and 'y' in request.args:

   109

            png = self.render_region(request, image)

   110

   111

        elif 'x' in request.args and 'y' in request.args:

   112

            png = self.render_tile(request, image)

   113

   114

        else:

   115

            raise exceptions.BadRequest("Unknown args")

   116

   117

        return Response(png, content_type='image/png')