from gamedata.objects.base import BaseObject

class Levels(BaseObject):

    def __init__(self,x,y,game,nodes,customvalues):

        super().__init__(0,0,game)
        self.nodes = nodes
        self.customvalues = customvalues

        self.blueflags = self.game.getSpriteDir("flags/blue/")
        self.greenflags= self.game.getSpriteDir("flags/green/")
        self.flagbase = self.game.sprite_lib["flags/base.png"]

        self.playerwalking = self.game.getSpriteDir("player/walking/")
        self.playerstill = self.game.sprite_lib["player/still.png"]
        self.flip = False
        self.moving = False
        self.spriteindex = 0

        self.flagsindex = []
        self.names = []
        self.textnumbers = []
        for i,v in enumerate(self.nodes):
            v["x"]+=8 # Center on tiles
            v["y"]+=8
            self.flagsindex.append(self.game.lib.randint(1,3))
                
            # Get sprites for "Level N"
            sprites = self.game.getchars("Level "+str(i+1))
            text = self.game.pygame.Surface((len(sprites)*9,9),flags=self.game.pygame.SRCALPHA)
            for j,s in enumerate(sprites):
                if s:
                    text.blit(s,[j*9,0])
            self.textnumbers.append(text)

            # Get sprites for the level's name
            if str(i) in customvalues.keys():
                sprites = self.game.getchars(customvalues[str(i)])
                text = self.game.pygame.Surface((len(sprites)*9,9),flags=self.game.pygame.SRCALPHA)
                for i,v in enumerate(sprites):
                    if v:
                        text.blit(v,[i*9,0])
                self.names.append(text)
            else:
                self.names.append(None)

        self.animspeed = 7
        self.cursor = 0 # Which level is selected
        self.playerx = self.nodes[0]["x"]
        self.playery = self.nodes[1]["y"]

        self.tileset = self.game.gameloop.findname("TilesetRenderer")[0]
        self.maxwidth = self.tileset.layers[0]["surface"].get_width()

    def step(self):
        for i in range(len(self.flagsindex)):
            self.flagsindex[i]+=self.animspeed*self.game.dt

        if not self.moving:
            if self.game.inputs["keys"]["right"]["timer"]>0:
                self.cursor+=1
                self.flip = False
            if self.game.inputs["keys"]["left"]["timer"]>0:
                self.cursor-=1
                self.flip = True
            if self.game.inputs["keys"]["up"]["timer"]>0:
                # Launch the level
                self.game.scenes.ingame(self.game,level="Level "+str(self.cursor+1))
        self.cursor = min(self.cursor,len(self.nodes)-1)
        self.cursor = max(0,self.cursor)

        speedx = (self.nodes[self.cursor]["x"]-self.playerx)
        speedy = (self.nodes[self.cursor]["y"]-self.playery)

        self.moving = (abs(speedx)+1)*(abs(speedy)+1)>3
        if self.moving:
            self.spriteindex += self.animspeed*self.game.dt
        else:
            self.spriteindex = 0
        self.playerx += speedx*4*self.game.dt
        self.playery += speedy*4*self.game.dt

        self.game.globals["camerax"] = self.playerx-self.game.globals["cameraw"]/2
        self.game.globals["camerax"] = max(0,self.game.globals["camerax"])
        self.game.globals["camerax"] = min(self.maxwidth-self.game.globals["cameraw"],self.game.globals["camerax"])

    def draw(self):
        for i,v in enumerate(self.nodes): # Draw flags
            sprites = self.blueflags
            sprite = sprites[int(self.flagsindex[i])%len(sprites)]
            x = v["x"]-sprite.get_width()/2-self.game.globals["camerax"]+8
            y = v["y"]-sprite.get_height()-self.game.globals["cameray"]
            self.game.window.blit(self.flagbase,[v["x"]-self.game.globals["camerax"]-8,v["y"]-self.game.globals["cameray"]-2])
            self.game.window.blit(sprite,[x,y])

        # Draw player
        if self.moving:
            sprite = self.playerwalking[int(self.spriteindex)%len(self.playerwalking)]
        else:
            sprite = self.playerstill
        sprite = self.game.pygame.transform.flip(sprite,self.flip,False)
        x = self.playerx-self.game.globals["camerax"]-sprite.get_width()/2
        y = self.playery-self.game.globals["cameray"]-sprite.get_height()/2-6
        self.game.window.blit(sprite,[x,y])

        # Draw level's number
        sprite = self.textnumbers[self.cursor]
        x = self.nodes[self.cursor]["x"]-self.game.globals["camerax"]-sprite.get_width()/2
        y = self.nodes[self.cursor]["y"]-self.game.globals["cameray"]-sprite.get_height()/2-50
        self.game.window.blit(sprite,[x,y])
        # Draw the level's name
        if self.names[self.cursor]:
            sprite = self.names[self.cursor]
            x = self.nodes[self.cursor]["x"]-self.game.globals["camerax"]-sprite.get_width()/2
            y = self.nodes[self.cursor]["y"]-self.game.globals["cameray"]-sprite.get_height()/2-40
            self.game.window.blit(sprite,[x,y])