#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Jan 28 10:54:33 2020

@author: suwako
"""
import mecha
import pygame
import sys
import numpy as np


class Obj():
    def __init__(self, **keyargs):
        self.mecha = mecha.Player(**keyargs)
        self.image = pygame.image.load("assets/planet.gif")
        self.position = self.image.get_rect()
        self.dimension = np.array((self.position.width, self.position.height))

    def dimension(self):
        return np.array((self.position.width, self.position.height),
                        dtype='float64')

    def draw(self,
             scale=1,
             reference=np.array((0, 0), dtype='float64'),
             size=(0, 0)):
        npsize=np.array(size, dtype='float64')/(2*scale)
        scaled_pos = scale * (self.mecha.pos - reference + npsize)
        scaled_dimension = (scale*self.dimension()).astype(int)
        scaled_image = pygame.transform.scale(self.image, scaled_dimension)
        scaled_image = pygame.transform.rotate(scaled_image,
                                               self.mecha.orientation%360)
        scaled_position = scaled_image.get_rect()
        scaled_position.center = tuple(map(int, scaled_pos))
        screen.blit(scaled_image, scaled_position)


class Star(Obj):
    def __init__(self, **keyargs):
        self.mecha = mecha.Player(**keyargs)
        self.image = pygame.image.load("assets/star.gif")
        self.position = self.image.get_rect()


class Planet(Obj):
    def __init__(self, **keyargs):
        self.mecha = mecha.Player(**keyargs)
        self.image = pygame.image.load("assets/planet.gif")
        self.position = self.image.get_rect()


class Player(Obj):
    def __init__(self, **keyargs):
        self.mecha = mecha.Player(**keyargs)
        self.image = pygame.image.load("assets/ship/setup/0").convert_alpha()
        self.position = self.image.get_rect()


if __name__ == "__main__":
    import time
    pygame.init()
    black = (0, 0, 0)
    size = width, height = 1366, 720
    TICK_RATE = 60
    keys = set()
    tick = (time.time(), time.time(), 0)
    screen = pygame.display.set_mode(size,
                                     flags=pygame.RESIZABLE)
    elements = [Player(speed=(-5, 30), pos=(10, 1)), Player()]
    reference = 1
    joy=[1, 1]
    while 1:
        dtick = tick
        tick = (tick[0], time.time(), int((time.time() - tick[0])*TICK_RATE))
        dtick = tick[2] - dtick[2]
        if dtick > 2:
            print(dtick)
        for i in range(dtick):
            size = pygame.display.get_surface().get_size()
            for event in pygame.event.get():
                #print(event.type)
                if event.type == pygame.VIDEORESIZE:
                    size = width, height = event.size
                    screen =screen = pygame.display.set_mode(size,
                                                             flags=pygame.RESIZABLE)
                if event.type == pygame.QUIT: sys.exit()
                if event.type == pygame.KEYDOWN:
                    keys.update({event.key})
                    print(event.key)
                    # if event.key == 102:
                    #     follow = (follow + 1)%len(elements)
                if event.type == pygame.KEYUP:
                    if event.key in keys:
                        keys.remove(event.key)
            screen.fill(black)
            for key in keys:
                if key in [275,276,273,274]:
                    exec("joy" + ("[0]+","[0]-","[1]+","[1]-")[[275,276,273,274].index(key)] + "=1")
                    elements[1].mecha.orientation=joy[0]*(-3)
            for element in elements[::-1]:
                element.mecha.apply_gravity([element.mecha for element in elements])
                element.mecha.tick(origin=elements[0].mecha.pos)
            if i <= 2:
                for element in elements:
                    element.draw(scale=10**(joy[1]*0.05), reference=elements[1].mecha.pos, size=screen.get_size())
                pygame.display.flip()
            # for element in elements:
            #     element.apply(camera=camera, origin=origin)
            #     screen.blit(element.image, element.position)
        time.sleep(0.001)
        if tick[2]%60 == 0 : print(elements[1].mecha.pos)