1

from __future__ import absolute_import

     2

     3

from django.conf import settings

     4

import django.utils.http

     5

     6

# the top-level pngtile package

     7

from pngtile.store import (

     8

        Error,

     9

        NotFound,

    10

        InvalidImage,

    11

        UncachedImage,

    12

        PNGTileStore,

    13

)

    14

    15

import os.path

    16

    17

class Store (PNGTileStore):

    18

    TILE_SIZE = 256

    19

    TILE_ZOOM = 4

    20

    21

    def __init__ (self, title, tileserver, **opts):

    22

        super(Store, self).__init__(**opts)

    23

    24

        self._title = title

    25

        self.tileserver = tileserver

    26

    27

    def title (self, name=None):

    28

        if not name:

    29

            return self._title

    30

    31

        if '/'  in name:

    32

            _, name = name.rsplit('/', 1)

    33

    34

        if '.' in name:

    35

            name, ext = name.split('.', 1)

    36

    37

        return name

    38

    39

    def breadcrumb (self, name):

    40

        """

    41

            Yield (name, title) tuples for breadcrumb to name

    42

        """

    43

    44

        path = [ ]

    45

    46

        for part in name.split('/'):

    47

            path.append(part)

    48

    49

            yield '/'.join(path), part

    50

    51

    def tree (self, name):

    52

        """

    53

            Yield [(item, name)] lists for given path.

    54

        """

    55

    56

        # always, starting from root

    57

        parts = []

    58

    59

        if name:

    60

            # also including the dir itself... but not the root twice

    61

            name += '/'

    62

    63

        # walk up the tree from root the dir itself

    64

        for tree_item in name.split('/'):

    65

            tree_name = '/'.join(parts)

    66

    67

            dirs = [ ]

    68

            items = [ ]

    69

    70

            # walk through the items within this tree node

    71

            for item, type in self.list(tree_name):

    72

                if tree_name:

    73

                    item_name = tree_name + '/' + item

    74

                else:

    75

                    item_name = item

    76

    77

                if type:

    78

                    items.append((item, item_name))

    79

                else:

    80

                    dirs.append((item, item_name))

    81

    82

            # tree of parent + item within parent

    83

            yield sorted(dirs), tree_item

    84

    85

            parts.append(tree_item)

    86

    87

    def items (self, name):

    88

        """

    89

            Yield (item, name) for given path.

    90

        """

    91

    92

        for item, type in self.list(name):

    93

            if not type:

    94

                # skip dirs

    95

                continue

    96

    97

            if name:

    98

                item_name = name + '/' + item

    99

            else:

   100

                item_name = item

   101

   102

            # open Image

   103

            image, image_name = self.open(item_name)

   104

            image_info = image.info()

   105

   106

            # preview <img>

   107

            x = image_info['img_width'] / 2

   108

            y = image_info['img_height'] / 2

   109

            zoom = 2

   110

   111

            img_src = self.tiles_url(image_name, w=640, h=320,

   112

                    t       = image_info['cache_mtime'],

   113

                    x       = x >> zoom,

   114

                    y       = y >> zoom,

   115

                    zoom    = zoom,

   116

            )

   117

   118

            # overlay title

   119

            item_title = self.title(item_name)

   120

   121

            yield item_title, item_name, img_src

   122

   123

    def tiles_url (self, name=None, **query):

   124

        """

   125

            Return a tileserver URL.

   126

        """

   127

   128

        url = self.tileserver

   129

   130

        if name:

   131

            url += name

   132

   133

        if query:

   134

            url += '?' + django.utils.http.urlencode(query)

   135

   136

        return url

   137

   138

STORE = Store(

   139

    # PNGTileStore

   140

    image_root  = settings.QMSK_PNGTILE_ROOT,

   141

   142

    # Store

   143

    title       = settings.QMSK_PNGTILE_TITLE,

   144

    tileserver  = settings.QMSK_PNGTILE_SERVER,

   145

)