import random
from PIL import Image

n0 = [
    [1,1,1],
    [1,0,1],
    [1,0,1],
    [1,0,1],
    [1,1,1]
]

n1 = [
    [0,1,0],
    [1,1,0],
    [0,1,0],
    [0,1,0],
    [1,1,1]
]

n2 = [
    [1,1,1],
    [0,0,1],
    [1,1,1],
    [1,0,0],
    [1,1,1]
]

n3 = [
    [1,1,1],
    [0,0,1],
    [1,1,1],
    [0,0,1],
    [1,1,1]
]

n4 = [
    [1,0,1],
    [1,0,1],
    [1,1,1],
    [0,0,1],
    [0,0,1]
]

n5 = [
    [1,1,1],
    [1,0,0],
    [1,1,1],
    [0,0,1],
    [1,1,1]
]

n6 = [
    [1,1,1],
    [1,0,0],
    [1,1,1],
    [1,0,1],
    [1,1,1]
]

n7 = [
    [1,1,1],
    [0,0,1],
    [0,0,1],
    [0,0,1],
    [0,0,1]
]

n8 = [
    [1,1,1],
    [1,0,1],
    [1,1,1],
    [1,0,1],
    [1,1,1]
]

n9 = [
    [1,1,1],
    [1,0,1],
    [1,1,1],
    [0,0,1],
    [1,1,1]
]

### FONT: https://github.com/teryror/pixel-fonts/tree/master

a_s = [
    [0,1,1,1,1,0],
    [1,1,0,0,1,1],
    [1,1,0,0,0,0],
    [0,1,1,1,0,0],
    [0,0,1,1,1,0],
    [0,0,0,0,1,1],
    [1,1,0,0,1,1],
    [0,1,1,1,1,0]
]

a_c = [
    [0,1,1,1,1,0],
    [1,1,1,0,1,1],
    [1,1,0,0,0,0],
    [1,1,0,0,0,0],
    [1,1,0,0,0,0],
    [1,1,0,0,0,0],
    [1,1,1,0,1,1],
    [0,1,1,1,1,0]
]

a_o = [
    [0,1,1,1,1,0],
    [1,1,0,0,1,1],
    [1,1,0,0,1,1],
    [1,1,0,0,1,1],
    [1,1,0,0,1,1],
    [1,1,0,0,1,1],
    [1,1,0,0,1,1],
    [0,1,1,1,1,0]
]

a_r = [
    [1,1,1,1,1,0],
    [1,1,0,0,1,1],
    [1,1,0,0,1,1],
    [1,1,0,0,1,1],
    [1,1,1,1,1,0],
    [1,1,0,1,1,1],
    [1,1,0,0,1,1],
    [1,1,0,0,1,1]
]

a_e = [
    [1,1,1,1,1,1],
    [1,1,0,0,0,0],
    [1,1,0,0,0,0],
    [1,1,1,1,1,1],
    [1,1,0,0,0,0],
    [1,1,0,0,0,0],
    [1,1,0,0,0,0],
    [1,1,1,1,1,1]
]

class color:
    def __init__(self, color=None, red=None, green=None, blue=None, alpha=255) -> None:
        # Allow hex string input like "#ff8800" or "ff8800"
        if isinstance(color, str):
            hexcode = color.lstrip("#")

            if len(hexcode) not in (6, 8):
                raise ValueError("Hex color must be 6 or 8 characters long")

            self.r = int(hexcode[0:2], 16)
            self.g = int(hexcode[2:4], 16)
            self.b = int(hexcode[4:6], 16)

            if len(hexcode) == 8:
                self.a = int(hexcode[6:8], 16)
            else:
                self.a = alpha

        else:
            # Regular RGB(A) input
            self.r = red
            self.g = green
            self.b = blue
            self.a = alpha
    def get(self) -> tuple[int, int, int]:
        return (self.r, self.g, self.b)
    def sub(self, n) -> tuple[int, int, int]:
        if self.r:
            r=self.r-n
        else:
            r=0
        if self.g:
            g=self.g-n
        else:
            g=0
        if self.b:
            b=self.b-n
        else:
            b=0
        return (r,g,b)
        
    def multiply(self, other) -> tuple[int, int, int]:
        r = other[0]
        g = other[1]
        b = other[2]
        return (((self.r * r) >> 8), ((self.g * g) >> 8), ((self.b * b)>>8))
    def shiftr(self, n):
        return ((self.r  >> n), (self.g  >> n), (self.b >>n))
    def mask(self, n):
        return ((self.r & n), (self.g & n), (self.b & n))

class coord:
    def __init__(self, x, y, width, height) -> None:
        self.x = x
        self.y = y
        self.width = width
        self.height = height
    
def make_next_board():
    next_board = [
        [0, 0, 0, 0],
        [0, 0, 0, 0],
        [0, 1, 0, 0],
        [1, 1, 1, 0],
    ]
    return next_board

def make_board(n: int):
    board_height = 20
    board_width = 10

    board = []

    for _ in range(2):
        line = []

        for _ in range(board_width):
            line.append(0)

        board.append(line)

    board.append([0,1,2,3,4,5,6,7,0,0])

    for _ in range(board_height - 3):
        line = []

        for _ in range(board_width):
            line.append(random.randint(0, n-1))

        board.append(line)

    return board_height, board_width, board

def make_sprite():
    sprite = Image.open("img/bitmap.png")

    sprite_pixels = sprite.load()

    sprite_width, sprite_height = sprite.size

    return sprite_pixels, sprite_width, sprite_height

def make_score():
    score = random.randint(0, 999999999)
    return score

black    = color("#000000")
white    = color("#ffffff")

red      = color("#ff0000")
cyan     = color("#00ffff")

blue     = color("#0000ff")
yellow   = color("#fff200")

green    = color("#00ff00")
magenta  = color("#ff00ff")

orange   = color("#ff8000")
sky_blue = color("#0080ff")

purple   = color("#9d00ff")
lime     = color("#5fdf5f")

colors = [
    black,
    cyan,
    blue,
    orange,
    yellow,
    green,
    purple,
    red
]

colors2 = [
    black,
    color("#0086f3"),
    color("#3d4fdc"),
    color("#fdab01"),
    color("#abcb04"),
    color("#02bdcd"),
    color("#d20f63"),
    color("#fe4b0f"),
]


text_bg = black

next_board = make_next_board()
nb1 = coord(88, 96, 128, 128)
nb2 = coord(1064, 96, 128, 128)

board_height, board_width, board = make_board(len(colors))
b1 = coord(304, 40, 320, 640)
board2_height, board2_width, board2 = make_board(len(colors2))
b2 = coord(656, 40, 320, 640)

score1 = make_score()
s1  = coord(84, 454, 136, 32)
sn1 = coord(74, 490, 156, 20)
score2 = make_score()
s2  = coord(1060, 454, 136, 32)
sn2 = coord(1050, 490, 156, 20)

sprite_pixels, sprite_width, sprite_height = make_sprite()

width = 1280
height = 720

img = Image.new("RGB", (width, height))

pixels = img.load()

for y in range(height):
    for x in range(width):
        ####################### player 1
        ### Board
        if x>=b1.x and x<b1.x+b1.width and y>=b1.y and y<b1.y+b1.height:
            xd = x-b1.x
            yd = y-b1.y
            # pixels[x, y] = colors[board[yd>>5][xd>>5]].multiply(sprite_pixels[yd&0b0000000000011111, xd&0b0000000000011111])
            xs = xd&0x1F
            ys = yd&0x1F
            if (xs > 7 and xs < 24) and (ys > 7 and ys < 24): # middle
                pixels[x, y] = colors[board[yd>>5][xd>>5]].get()
            elif (xs > 5 and xs < 26) and (ys > 5 and ys < 26): # mid boarder
                pixels[x, y] = colors[board[yd>>5][xd>>5]].mask(0xFFFFFFB0)
            # elif xs == ys or xs+ys==31:
            #     pixels[x, y] = colors[board[yd>>5][xd>>5]].shiftr(2)
            elif xs+ys<31: # top left edge
                if xs>ys:
                    pixels[x, y] = colors[board[yd>>5][xd>>5]].get()
                else:
                    pixels[x, y] = colors[board[yd>>5][xd>>5]].mask(0xFFFFFFEF)
            else:
                if xs>ys:
                    pixels[x, y] = colors[board[yd>>5][xd>>5]].mask(0xFFFFFFD7)
                else:
                    pixels[x, y] = colors[board[yd>>5][xd>>5]].mask(0xFFFFFFC0)
        ### Next Piece
        elif x>=nb1.x and x<nb1.x+nb1.width and y>=nb1.y and y<nb1.y+nb1.height:
            xd = x-nb1.x
            yd = y-nb1.y
            xs = xd&0x1F
            ys = yd&0x1F
            if (xs == 31) or (ys==31):
                if (xs == 0) or (ys == 0):
                    pixels[x, y] = colors[next_board[yd>>5][xd>>5]].shiftr(1)
                else:
                    pixels[x, y] = colors[next_board[yd>>5][xd>>5]].shiftr(2)
            elif (not xs) or (not ys):
                pixels[x, y] = colors[next_board[yd>>5][xd>>5]].get()
            else:
                pixels[x, y] = colors[next_board[yd>>5][xd>>5]].mask(0xFFFFFFC0)
        ### Score
        elif x>=s1.x and x<s1.x+s1.width and y>=s1.y and y<s1.y+s1.height:
            yd = y-s1.y
            if x < (24+s1.x) and a_s[yd>>2][(x-s1.x)>>2]:
                pixels[x, y] = white.get()
            elif x<(52+s1.x) and x > (28+s1.x) and a_c[yd>>2][(x-(28+s1.x))>>2]:
                pixels[x, y] = white.get()
            elif x<(80+s1.x) and x > (56+s1.x) and a_o[yd>>2][(x-(56+s1.x))>>2]:
                pixels[x, y] = white.get()
            elif x<(108+s1.x) and x > (84+s1.x) and a_r[yd>>2][(x-(84+s1.x))>>2]:
                pixels[x, y] = white.get()
            elif x > (112+s1.x) and a_e[yd>>2][(x-(112+s1.x))>>2]:
                pixels[x, y] = white.get()
            else:
                pixels[x, y] = text_bg.get()
        elif x>=sn1.x and x<sn1.x+sn1.width and y>=sn1.y and y<sn1.y+sn1.height:
            pixels[x, y] = text_bg.get()
        ####################### player 2
        ### Board
        elif x>=b2.x and x<b2.x+b2.width and y>=b2.y and y<b2.y+b2.height:
            # xd = x-b2.x
            # yd = y-b2.y
            # pixels[x, y] = colors2[board2[yd>>5][xd>>5]].multiply(sprite_pixels[yd&0b0000000000011111, xd&0b0000000000011111])
            xd = x-b2.x
            yd = y-b2.y
            xs = xd&0x1F
            ys = yd&0x1F
            if (xs == 31) or (ys == 31):
                if (xs == 0) or (ys == 0):
                    pixels[x, y] = colors2[board2[yd>>5][xd>>5]].shiftr(1)
                else:
                    pixels[x, y] = colors2[board2[yd>>5][xd>>5]].shiftr(2)
            elif (not xs) or (not ys):
                pixels[x, y] = colors2[board2[yd>>5][xd>>5]].get()
            else:
                pixels[x, y] = colors2[board2[yd>>5][xd>>5]].mask(0xFFFFFFC0)
        ### Next Piece
        elif x>=nb2.x and x<nb2.x+nb2.width and y>=nb2.y and y<nb2.y+nb2.height:
            xd = x-nb2.x
            yd = y-nb2.y
            pixels[x, y] = colors2[next_board[yd>>5][xd>>5]].multiply(sprite_pixels[yd&0b0000000000011111, xd&0b0000000000011111])
        ### Score
        elif x>=s2.x and x<s2.x+s2.width and y>=s2.y and y<s2.y+s2.height:
            yd = y-s2.y
            if x < (24+s2.x) and a_s[yd>>2][(x-s2.x)>>2]:
                pixels[x, y] = white.get()
            elif x<(52+s2.x) and x > (28+s2.x) and a_c[yd>>2][(x-(28+s2.x))>>2]:
                pixels[x, y] = white.get()
            elif x<(80+s2.x) and x > (56+s2.x) and a_o[yd>>2][(x-(56+s2.x))>>2]:
                pixels[x, y] = white.get()
            elif x<(108+s2.x) and x > (84+s2.x) and a_r[yd>>2][(x-(84+s2.x))>>2]:
                pixels[x, y] = white.get()
            elif x > (112+s2.x) and a_e[yd>>2][(x-(112+s2.x))>>2]:
                pixels[x, y] = white.get()
            else:
                pixels[x, y] = text_bg.get()
        elif x>=sn2.x and x<sn2.x+sn2.width and y>=sn2.y and y<sn2.y+sn2.height:
            pixels[x, y] = text_bg.get()
        else: 
            if (x+y)&1:
                pixels[x, y] = (0, 0, 255)
            else:
                pixels[x, y] = ((x+y)>>3, (((1280+720)>>3)-((x+y)>>3)), 0)

img.save("output/alpha.png")
img.show()