from gamedata.objects.base import BaseObject
import random,math

class Lemming(BaseObject):
    def __init__(self,x,y,game,speedmargin=5):
        super().__init__(x,y,game,w=60,h=60)
        self.direction = random.randint(0,360)
        self.holdrect = self.rect.copy()
        self.holdradius = 65
        self.holdrect = self.holdrect.inflate(self.holdradius,self.holdradius)
        self.basespeed = 40+random.randint(-speedmargin,speedmargin) # Speed that he normally walks by
        self.holdtimer = game.lib.Timer(2.5) # Max seconds of holding
        self.normalspeed = self.basespeed # Speed "objective"
        self.speed = 0 # Current speed, leaning towards objective speed

        self.selected = False # If beeing redirected
        self.cachedrel = [] # Storing relative movement of mouse
        self.cachedrelsize = 10
        self.mincachedsize = 5

        # Used for movement
        self.restx = 0
        self.resty = 0

        # Sprites
        self.orientations = ["Right","Left","Left","Right"]
        self.sprites = {}
        for i in self.orientations:
            self.sprites[i] = game.getSpriteDir("lemmings/"+i+"/")
        self.spriteindex = 0
        self.animspeed = 0.2

    def step(self):

        # Depth updating based on position
        self.depth = 1+self.rect[1]/100

        # Lean towards the normal speed
        if self.selected:
            self.normalspeed = 0
        else:
            self.normalspeed = self.basespeed
        self.speed += (self.normalspeed-self.speed)*self.game.dt*3

        # Getting x and y velocity
        diffx = math.cos(math.radians(self.direction))*self.speed
        diffy = math.sin(math.radians(self.direction))*self.speed

        self.move(diffx*self.game.dt,diffy*self.game.dt)

        # Mouse selection
        mouse=self.game.inputs["mouse"]
        if self.selected:
            # Caching mouse relative movement
            self.cachedrel.append(mouse["rel"])
            if len(self.cachedrel)>self.cachedrelsize:
                self.cachedrel.pop(0)
            # Releasing it
            if mouse["click"]==0 or not self.holdrect.collidepoint(mouse["campos"]):
                self.launch()
            else:
                if self.holdtimer.tick(self.game.dt):
                    self.selected = False
                    self.holdtimer.reset()
                    self.normalspeed = self.basespeed
        if self.game.inputs["mouse"]["click"]==1:
            if self.rect.collidepoint(mouse["campos"]):
                self.selected = True

        # Animation
        self.spriteindex+=self.game.dt*self.animspeed*self.speed
    def launch(self):
        # Launch itself in the mouse direction
        if len(self.cachedrel)>self.mincachedsize:
            averagerel = [sum([x[i] for x in self.cachedrel])/len(self.cachedrel) for i in range(2)]
            self.direction = math.degrees(math.atan2(averagerel[1],averagerel[0]))

            self.speed = self.basespeed*4
            self.cachedrel = []
        self.selected = False

    def move(self,x,y):
        # Remember the digits, pygame rects only move with integers
        velx = x+self.restx
        vely = y+self.resty
        self.restx = velx-int(velx)
        self.resty = vely-int(vely)
        self.rect[0]+=int(velx)
        self.rect[1]+=int(vely)
        self.holdrect[0]+=int(velx)
        self.holdrect[1]+=int(vely)

    def draw(self):
        orientation = self.orientations[int(self.direction%360/361*4)]
        sprites = self.sprites[orientation]
        self.game.lib.drawcenter(self.game,self.game.sprite_lib["lemmings/shadow.png"],self.rect.center[0],self.rect.center[1])
        self.game.lib.drawcenter(self.game,sprites[int(self.spriteindex)%len(sprites)],self.rect.center[0],self.rect.center[1])
        if self.game.globals["debug"]:
            s = self.game.pygame.Surface(self.rect.size)
            s.fill([255,0,0])
            s.set_alpha(30)
            self.game.window.blit(s,self.rect[:2])
            s = self.game.pygame.Surface(self.holdrect.size)
            s.fill([0,255,0])
            s.set_alpha(30)
            self.game.window.blit(s,self.holdrect[:2])