# Copyright (c) 2013-2016 Jeffrey Pfau
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from ._pylib import ffi, lib
from . import image

class Tile:
    def __init__(self, data):
        self.buffer = data

    def toImage(self):
        i = image.Image(8, 8)
        self.composite(i, 0, 0)
        return i

    def composite(self, i, x, y):
        for iy in range(8):
            for ix in range(8):
                i.buffer[ix + x + (iy + y) * i.stride] = image.u16ToColor(self.buffer[ix + iy * 8])

class TileView:
    def __init__(self, core):
        self.core = core
        self.cache = ffi.gc(ffi.new("struct mTileCache*"), core._deinitTileCache)
        core._initTileCache(self.cache)
        lib.mTileCacheSetPalette(self.cache, 0)
        self.paletteSet = 0

    def getTile(self, tile, palette):
        return Tile(lib.mTileCacheGetTile(self.cache, tile, palette))

    def setPalette(self, paletteSet):
        if paletteSet > 1 or paletteSet < 0:
            raise IndexError("Palette Set ID out of bounds")
        lib.mTileCacheSetPalette(self.cache, paletteSet)
        self.paletteSet = paletteSet

class Sprite(object):
    TILE_BASE = 0, 0
    PALETTE_BASE = 0, 0

    def constitute(self, tileView, tilePitch, paletteSet):
        oldPaletteSet = tileView.paletteSet
        tileView.setPalette(paletteSet)
        i = image.Image(self.width, self.height)
        tileId = self.tile + self.TILE_BASE[paletteSet]
        for y in range(self.height // 8):
            for x in range(self.width // 8):
                tile = tileView.getTile(tileId, self.paletteId + self.PALETTE_BASE[paletteSet])
                tile.composite(i, x * 8, y * 8)
                tileId += 1
            if tilePitch:
                tileId += tilePitch - self.width // 8
        self.image = i
        tileView.setPalette(oldPaletteSet)