Exemplo n.º 1
0
    def __init__(self, main, host=True):
        Screen.__init__(self, main)

        self.my_index = 0

        self.player_colors = [(0xFF, 0, 0), (0, 0xFF, 0), (0xFF, 0, 0xFF), (0xFF, 0xFF, 0),
                              (0, 0xFF, 0xFF), (0xFF, 180, 0), (180, 0, 0xFF),
                              (0, 0xFF, 100), (0xFF, 0xCC, 0xCC), (0xCC, 0xCC, 0xFF)]
        self.color_pick = 0
        self.largefont = pygame.font.Font(None, 70)
        self.font = pygame.font.Font(None, 30)

        self.waves = Waves()

        self.player_menu = pygame.Surface((415, 120), pygame.SRCALPHA)
        self.player_menu_pos = (0, 0)

        self.players_panel = pygame.Surface((496, 578), pygame.SRCALPHA)
        self.players_panel.fill((0, 0, 0, 128))
        self.players = []

        self.start_timeout = time.time()

        self.players.append(["Host", pygame.Surface((474, 48), pygame.SRCALPHA), False])
        for _ in xrange(1, 10):
            self.players.append([CLOSED, pygame.Surface((474, 48), pygame.SRCALPHA), False])
        self.redraw_players()
        self.chat_panel = pygame.Surface((696, 578), pygame.SRCALPHA)
        self.chat_panel.fill((0, 0, 0, 128))
        send_button = pygame.Surface((67, 30), pygame.SRCALPHA)
        send_button.fill((0, 0, 0, 64))
        txt = self.font.render("Send", True, (0xCC, )*3)    # TODO: nope
        send_button.blit(txt, (10, 7))
        self.chat_panel.blit(send_button, (20+582+5, 577-38))
        self.textbox = pygame.Surface((1820, 30), pygame.SRCALPHA)
        self.textbox.fill((0, 0, 0, 64))
        self.base_panel = pygame.Surface((1215, 96), pygame.SRCALPHA)
        self.base_panel.fill((0, 0, 0, 128))
        txt_leave_lobby = self.largefont.render("Leave Lobby", True, COLORS["white"])
        self.button_leave_lobby = pygame.Surface((331, 76), pygame.SRCALPHA)
        self.button_leave_lobby.fill((0, 0, 0, 64))
        self.button_leave_lobby.blit(txt_leave_lobby, (15, 15))
        self.base_panel.blit(self.button_leave_lobby, (10, 10))

        if host:
            def get_ip():
                try:
                    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
                    s.connect(('google.com', 80))
                    ip = s.getsockname()[0]
                    s.close()

                    txt_ip = self.largefont.render(ip, True, COLORS["white"])
                    self.base_panel.blit(txt_ip, (450, 25))

                    def copy_ip(mpos):
                        try:
                            pygame.scrap.init()
                            pygame.scrap.put(pygame.SCRAP_TEXT, ip)
                        except:
                            pass  # fixes!

                    self.clickbox.append((476, 706, txt_ip.get_width(), txt_ip.get_height()), copy_ip)
                except (urllib2.URLError, urllib2.HTTPError, IOError):
                    print "Error fetching IP"
            threading.Thread(target=get_ip).start()

        self.button_start = pygame.Surface((331, 76), pygame.SRCALPHA)
        self.button_start.fill((0, 0, 0, 64))
        txt_start_game = self.largefont.render("Start Game", True, COLORS["white"])
        self.button_start.blit(txt_start_game, (30, 15))
        self.base_panel.blit(self.button_start, (870, 10))

        def start_game(mpos):
            if time.time() - self.start_timeout > 3:
                self.main.client.start_game()
                self.start_timeout = time.time()
        self.clickbox.append((900, 690, 331, 76), start_game)

        self.text_input = ""
        self.msgpanel = MessagePanel((652, 509), 23, self.font)
        self.startable = False

        def to_main(mpos):
            self.main.change_screen("main")
        self.clickbox.append((37, 692, 329, 74), to_main)

        def ready_button_click(i):
            def anon(mpos):
                if i == self.my_index:
                    self.main.client.set_ready(int(not self.players[i][2]))
                else:
                    print "ready button clicked!!"
            return anon
        for x in xrange(10):
            self.clickbox.append((37, 75+54*x, 48, 48), ready_button_click(x))

        def player_panel_click(i):
            def anon(mpos):
                if self.my_index == 0:      # only the host can kick
                    if i == self.my_index:
                        pass
                    elif isinstance(self.players[i][0], int):
                        self.player_menu_pos = (95, 75+54*i+48)
                        options = ["Closed", "Open", "AI"]
                        self.player_menu.fill((0, 0, 0, 128))

                        def clearworking(mpos):
                            self.player_menu_pos = (0, 0)
                            for k in xrange(3):
                                self.clickbox.remove((95, 75+54*i+48+k*40+5))
                            self.clickbox.remove((0, 0))
                        self.clickbox.append((0, 0, 1280, 800), clearworking, z=2)
                        for j, option in enumerate(options):
                            def wrapper(toWhat):
                                def changeto(mpos):
                                    self.main.client.set_slot(i, toWhat+1)
                                    self.player_menu_pos = (0, 0)
                                    for k in xrange(3):
                                        self.clickbox.remove((95, 75+54*i+48+k*40+5))
                                    self.clickbox.remove((0, 0))
                                return changeto
                            self.clickbox.append((95, 75+54*i+48+j*40+5, 415, 40), wrapper(j), z=3)
                            text = self.font.render(option, True, COLORS["white"])
                            self.player_menu.blit(text, (5, 5+j*40))
                    else:   # it's a person
                        self.player_menu_pos = (95, 75+54*i+48)
                        self.player_menu.fill((0, 0, 0, 0))
                        pygame.draw.rect(self.player_menu, (0, 0, 0, 128), (0, 0, 415, 30))

                        def clearworking(mpos):
                            self.player_menu_pos = (0, 0)
                            self.clickbox.remove((95, 75+54*i+48+5))
                            self.clickbox.remove((0, 0))
                        self.clickbox.append((0, 0, 1280, 800), clearworking, z=2)

                        def kickhim(mpos):
                            self.main.client.kick_player(str(i))
                            self.player_menu_pos = (0, 0)
                            self.clickbox.remove((95, 75+54*i+48+5))
                            self.clickbox.remove((0, 0))
                        self.clickbox.append((95, 75+54*i+48, 415, 40), kickhim, z=3)
                        text = self.font.render("Kick", True, COLORS["white"])
                        self.player_menu.blit(text, (5, 5))
            return anon
        for x in xrange(10):
            self.clickbox.append((95, 75+54*x, 415, 48), player_panel_click(x))

        def wrapper(mpos):
            self.message()
            self.textbox.fill((0, 0, 0, 64))
            txt = self.font.render(self.text_input, True, COLORS["white"])
            self.textbox.blit(txt, (5, 5))
        self.clickbox.append((1154, 580, 64, 25), wrapper)
Exemplo n.º 2
0
class LobbyScreen(Screen):
    def __init__(self, main, host=True):
        Screen.__init__(self, main)

        self.my_index = 0

        self.player_colors = [(0xFF, 0, 0), (0, 0xFF, 0), (0xFF, 0, 0xFF), (0xFF, 0xFF, 0),
                              (0, 0xFF, 0xFF), (0xFF, 180, 0), (180, 0, 0xFF),
                              (0, 0xFF, 100), (0xFF, 0xCC, 0xCC), (0xCC, 0xCC, 0xFF)]
        self.color_pick = 0
        self.largefont = pygame.font.Font(None, 70)
        self.font = pygame.font.Font(None, 30)

        self.waves = Waves()

        self.player_menu = pygame.Surface((415, 120), pygame.SRCALPHA)
        self.player_menu_pos = (0, 0)

        self.players_panel = pygame.Surface((496, 578), pygame.SRCALPHA)
        self.players_panel.fill((0, 0, 0, 128))
        self.players = []

        self.start_timeout = time.time()

        self.players.append(["Host", pygame.Surface((474, 48), pygame.SRCALPHA), False])
        for _ in xrange(1, 10):
            self.players.append([CLOSED, pygame.Surface((474, 48), pygame.SRCALPHA), False])
        self.redraw_players()
        self.chat_panel = pygame.Surface((696, 578), pygame.SRCALPHA)
        self.chat_panel.fill((0, 0, 0, 128))
        send_button = pygame.Surface((67, 30), pygame.SRCALPHA)
        send_button.fill((0, 0, 0, 64))
        txt = self.font.render("Send", True, (0xCC, )*3)    # TODO: nope
        send_button.blit(txt, (10, 7))
        self.chat_panel.blit(send_button, (20+582+5, 577-38))
        self.textbox = pygame.Surface((1820, 30), pygame.SRCALPHA)
        self.textbox.fill((0, 0, 0, 64))
        self.base_panel = pygame.Surface((1215, 96), pygame.SRCALPHA)
        self.base_panel.fill((0, 0, 0, 128))
        txt_leave_lobby = self.largefont.render("Leave Lobby", True, COLORS["white"])
        self.button_leave_lobby = pygame.Surface((331, 76), pygame.SRCALPHA)
        self.button_leave_lobby.fill((0, 0, 0, 64))
        self.button_leave_lobby.blit(txt_leave_lobby, (15, 15))
        self.base_panel.blit(self.button_leave_lobby, (10, 10))

        if host:
            def get_ip():
                try:
                    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
                    s.connect(('google.com', 80))
                    ip = s.getsockname()[0]
                    s.close()

                    txt_ip = self.largefont.render(ip, True, COLORS["white"])
                    self.base_panel.blit(txt_ip, (450, 25))

                    def copy_ip(mpos):
                        try:
                            pygame.scrap.init()
                            pygame.scrap.put(pygame.SCRAP_TEXT, ip)
                        except:
                            pass  # fixes!

                    self.clickbox.append((476, 706, txt_ip.get_width(), txt_ip.get_height()), copy_ip)
                except (urllib2.URLError, urllib2.HTTPError, IOError):
                    print "Error fetching IP"
            threading.Thread(target=get_ip).start()

        self.button_start = pygame.Surface((331, 76), pygame.SRCALPHA)
        self.button_start.fill((0, 0, 0, 64))
        txt_start_game = self.largefont.render("Start Game", True, COLORS["white"])
        self.button_start.blit(txt_start_game, (30, 15))
        self.base_panel.blit(self.button_start, (870, 10))

        def start_game(mpos):
            if time.time() - self.start_timeout > 3:
                self.main.client.start_game()
                self.start_timeout = time.time()
        self.clickbox.append((900, 690, 331, 76), start_game)

        self.text_input = ""
        self.msgpanel = MessagePanel((652, 509), 23, self.font)
        self.startable = False

        def to_main(mpos):
            self.main.change_screen("main")
        self.clickbox.append((37, 692, 329, 74), to_main)

        def ready_button_click(i):
            def anon(mpos):
                if i == self.my_index:
                    self.main.client.set_ready(int(not self.players[i][2]))
                else:
                    print "ready button clicked!!"
            return anon
        for x in xrange(10):
            self.clickbox.append((37, 75+54*x, 48, 48), ready_button_click(x))

        def player_panel_click(i):
            def anon(mpos):
                if self.my_index == 0:      # only the host can kick
                    if i == self.my_index:
                        pass
                    elif isinstance(self.players[i][0], int):
                        self.player_menu_pos = (95, 75+54*i+48)
                        options = ["Closed", "Open", "AI"]
                        self.player_menu.fill((0, 0, 0, 128))

                        def clearworking(mpos):
                            self.player_menu_pos = (0, 0)
                            for k in xrange(3):
                                self.clickbox.remove((95, 75+54*i+48+k*40+5))
                            self.clickbox.remove((0, 0))
                        self.clickbox.append((0, 0, 1280, 800), clearworking, z=2)
                        for j, option in enumerate(options):
                            def wrapper(toWhat):
                                def changeto(mpos):
                                    self.main.client.set_slot(i, toWhat+1)
                                    self.player_menu_pos = (0, 0)
                                    for k in xrange(3):
                                        self.clickbox.remove((95, 75+54*i+48+k*40+5))
                                    self.clickbox.remove((0, 0))
                                return changeto
                            self.clickbox.append((95, 75+54*i+48+j*40+5, 415, 40), wrapper(j), z=3)
                            text = self.font.render(option, True, COLORS["white"])
                            self.player_menu.blit(text, (5, 5+j*40))
                    else:   # it's a person
                        self.player_menu_pos = (95, 75+54*i+48)
                        self.player_menu.fill((0, 0, 0, 0))
                        pygame.draw.rect(self.player_menu, (0, 0, 0, 128), (0, 0, 415, 30))

                        def clearworking(mpos):
                            self.player_menu_pos = (0, 0)
                            self.clickbox.remove((95, 75+54*i+48+5))
                            self.clickbox.remove((0, 0))
                        self.clickbox.append((0, 0, 1280, 800), clearworking, z=2)

                        def kickhim(mpos):
                            self.main.client.kick_player(str(i))
                            self.player_menu_pos = (0, 0)
                            self.clickbox.remove((95, 75+54*i+48+5))
                            self.clickbox.remove((0, 0))
                        self.clickbox.append((95, 75+54*i+48, 415, 40), kickhim, z=3)
                        text = self.font.render("Kick", True, COLORS["white"])
                        self.player_menu.blit(text, (5, 5))
            return anon
        for x in xrange(10):
            self.clickbox.append((95, 75+54*x, 415, 48), player_panel_click(x))

        def wrapper(mpos):
            self.message()
            self.textbox.fill((0, 0, 0, 64))
            txt = self.font.render(self.text_input, True, COLORS["white"])
            self.textbox.blit(txt, (5, 5))
        self.clickbox.append((1154, 580, 64, 25), wrapper)

    def redraw_players(self):
        self.players_panel.fill((0, 0, 0, 128))
        for i, (name, surf, ready) in enumerate(self.players):
            surf.fill((0, 0, 0, 64))
            if ready:
                pygame.draw.rect(surf, (135, 221, 68, 128), (0, 0, 48, 48))
            pygame.draw.rect(surf, (0, 0, 0, 0), (48, 0, 12, 48))  # clear
            color = self.player_colors[i]
            if isinstance(name, int):
                color = (0x80, 0x80, 0x80)
                if name == CLOSED:
                    name = "Closed"
                elif name == OPEN:
                    name = "Open"
                elif name == AI:
                    name = "AI Player"
            surf.blit(self.font.render(name, True, color), (65, 15))
            self.players_panel.blit(surf, (10, 37+i*54))

    def display(self, screen):
        Screen.display(self, screen)
        # I feel like the following code is flipping me off...
        self.waves.display(screen, False)
        screen.blit(self.players_panel, (26, 38))
        screen.blit(self.chat_panel, (545, 38))
        screen.blit(self.textbox, (565, 577), (min(self.textbox.get_width()-582, max(0, (self.font.size(self.text_input)[0])-582+10)), 0, 582, 30))
        screen.blit(self.base_panel, (26, 681))
        screen.blit(self.msgpanel.view, (566, 63))

        if self.player_menu_pos != (0, 0):
            screen.blit(self.player_menu, (self.player_menu_pos[0], self.player_menu_pos[1]))

    def message(self, data=None):
        index = self.my_index
        msg = self.text_input
        if not (data is None):  # gotta parse the data
            data = data.split(" ")
            index, msg = int(data[0]), ' '.join(data[1:])
        if msg:  # needs to like... say something.
            if isinstance(self.players[index][0], int):
                #print "Why is", self.players[index][0], "trying to talk?"
                return
            if index == self.my_index:
                self.text_input = ""
                if msg[:6] == "/nick ":
                    newnick = msg[6:]
                    if self.main.valid_nick(newnick):
                        self.send_nick_change(newnick)
                    return
                if msg[:6] == "/kick ":
                    newkick = msg[6:].lower()
                    for i, (name, _, _) in enumerate(self.players):
                        if not isinstance(name, int):
                            if name.lower() == newkick:
                                self.main.client.kick_player(str(i))
                                self.recv_kick_player(str(i))
                    return
                self.main.client.send_message(msg)
            self.msgpanel.message(self.players[index][0], msg, self.player_colors[index])

    def reload_server_data(self, data):
        """ loading player data after joining a lobby """
        self.main.race_cond = True
        self.my_index = int(data[0])    # what slot should you get
        data = data.split("\n")[1:]     # split on newlines (ignore the first slot)
        #print data
        for i, d in enumerate(data):     #
            num = int(d[0])             # are you a player or a certain value
            if num:                     # if not a human
                self.players[i][0] = num    # make my "name" that value
                self.players[i][2] = False  # also i'm 'unready'
            else:                           # 0 c Name (c is 0 if unready, 1 if ready)
                self.players[i][0] = d[4:]  # the rest of the string after the first two characters and spaces is the name
                self.players[i][2] = bool(int(d[2]))    # char space _char_ is readiness
        if self.main.player_name is None:
            self.main.player_name = self.players[self.my_index][0]
        self.redraw_players()           # redraw that board

    def ready_up(self, data):
        #print data
        index, ready = int(data[0]), bool(int(data[2]))
        self.players[index][2] = ready
        self.redraw_players()

    def send_nick_change(self, nick):
        self.main.player_name = nick
        self.main.client.change_name(nick)

    def recv_nick_change(self, data=None):
        data = data.split(" ")
        i, name = int(data[0]), ' '.join(data[1:])
        if i == self.my_index:
            self.main.player_name = name
        try:
            name = int(name)
        except ValueError:
            pass
        if not (isinstance(self.players[i][0], int) is isinstance(name, int) is False):
            self.ready_up(str(i)+" 0")
        self.players[i][0] = name
        self.redraw_players()

    def recv_kick_player(self, data=None):
        kickme = int(data[0])
        if kickme == self.my_index:  # D:
            self.main.client = None
            self.main.change_screen("main")
            return
        self.players[kickme][0] = OPEN
        self.players[kickme][2] = False
        self.redraw_players()

    def recv_start_game(self, players):
        #print "game starting!!"
        seed, _, players = players.partition(" ")
        self.main.rand.seed(int(seed))
        players = players.split("\t")
        self.main.reset_screen("game", players, False)
        self.main.reset_screen("shop")
        self.main.change_screen("game")

    def notify_key(self, inkey):
        if inkey.key == pygame.K_BACKSPACE:
            self.text_input = self.text_input[:-1]
        elif inkey.key == pygame.K_TAB:
            try:    # tab complete
                mname = self.text_input.split(' ')[-1].lower()
                if mname == "":  # don't tab complete on nothing
                    return
                matches = []    # people whose names we've matched
                for i, (name, _, _) in enumerate(self.players):
                    if not isinstance(name, int):   # don't check slots that aren't players
                        if mname == name[:len(mname)].lower():  # if the first part of their name is what you've typed
                            matches += [name]       # they're a match
                if len(matches) == 1:
                    self.text_input = self.text_input[:-len(mname)]+matches[0]
                elif len(matches) > 1:
                    def get_most_letters():
                        text = ""
                        for x in xrange(len(mname), 22):
                            letter = None
                            for i in matches:
                                if letter is None:
                                    letter = i[x:x+1]
                                    if letter == "":
                                        return text
                                    continue
                                if i[x:x+1] != letter:
                                    return text
                            text += letter
                    self.text_input += get_most_letters()
            except IndexError:
                pass
        elif inkey.key == pygame.K_RETURN or inkey.key == pygame.K_KP_ENTER:
            self.message()  # send my current message
        elif inkey.unicode and self.font.size(self.text_input+inkey.unicode)[0] <= self.textbox.get_width()-10 and len(self.text_input) < 250:
            self.text_input += inkey.unicode    # send the unicode character you pressed
        else:
            return  # if none of these things happened, no need to redraw
        self.textbox.fill((0, 0, 0, 64))
        txt = self.font.render(self.text_input, True, COLORS["white"])
        self.textbox.blit(txt, (5, 5))