pygame之2048

最新推荐文章于 2024-09-09 00:00:00 发布

星眺北海

最新推荐文章于 2024-09-09 00:00:00 发布

阅读量81

点赞数

分类专栏： game 文章标签： pygame python 开发语言

本文链接：https://blog.csdn.net/longforone/article/details/131711146

版权

game 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

pygame之2048

先放链接 代码

主要难点在于移动和合并

我的思路是这样的

横向：

移动的本质就是把0移到另一边，所以我们可以利用python列表很简单的从中间取出0然后插到两边
合并直接把相邻相等的数字合并到一格即可，这时会产生新的0
再进行一次移动操作即可

所以步骤就是 移动-合并-移动

纵向：
在这个方向上移动是在几个嵌套列表之间的，实现较为复杂，但是可以直接对原来的矩阵求逆，就可以直接按照横向的思路来做，注意操作完成之后要还原

所以步骤就是 求逆-移动-合并-移动-求逆

使用方式：

env=game2048()
env.start()

# 1
上下左右控制

# 2
env.step(action)
"""
action:
    0:左
    1:右
    2:上
    3:下

"""

本来是想用这个做强化学习的，但是鸭脖我呀，做不出来QAQ

import random
import sys
import time

import pygame


class game2048:
    data = None
    old_data = None
    over = None
    # 颜色自行设定，gpt填的好丑
    color = {
        0: (0, 0, 0),
        2: (20, 30, 0),
        4: (40, 20, 90),
        8: (60, 30, 15),
        16: (80, 40, 20),
        32: (100, 50, 25),
        64: (120, 60, 30),
        128: (140, 70, 35),
        256: (160, 10, 40),
        512: (180, 90, 45),
        1024: (70, 100, 50),
        2048: (220, 180, 55),
        
    }
    
    def __init__(self):
        pygame.init()
        pygame.display.set_caption("2048")
        self.screen = pygame.display.set_mode((400, 400))
        self.clock = pygame.time.Clock()
        self.font = pygame.font.Font(None, 50)
        self.reset()
    
    def draw(self):
        
        for i in range(4):
            for j in range(4):
                pos = pygame.Rect(i * 100, j * 100, 100, 100)
                color = self.color[self.data[i][j]]
                pygame.draw.rect(self.screen, color, pos, 50)
                if self.data[i][j]!=0:
                    text = self.font.render(str(self.data[i][j]), True, (255, 255, 255))
                    text_rect = text.get_rect(center=pos.center)
                    self.screen.blit(text, text_rect)
        pygame.display.flip()
    
    def get_T(self):
        new_data = []
        for i in range(4):
            new_data.append([])
            for j in range(4):
                new_data[i].append(self.data[j][i])
        self.data = new_data
    
    def show_data(self):
        print("-" * 25)
        for i in self.data:
            print(end="|")
            for j in i:
                print("{:^5}".format(j), end="|")
            
            print('\n', "-" * 25, sep='')
        print()
    
    def merge(self, direction):
        self.zero_to_end(direction)
        reward = -1
        if direction == 0:
            
            for i in range(3):
                for j in range(4):
                    if self.data[i][j] != 0 and self.data[i][j] == self.data[i + 1][j]:
                        reward += 1
                        self.data[i][j] <<= 1
                        self.data[i + 1][j] = 0
        
        elif direction == 1:
            for i in reversed(range(1, 4)):
                for j in range(4):
                    if self.data[i][j] != 0 and self.data[i][j] == self.data[i - 1][j]:
                        reward += 1
                        self.data[i][j] <<= 1
                        self.data[i - 1][j] = 0
        
        elif direction == 2:
            for j in range(3):
                for i in range(4):
                    if self.data[i][j] != 0 and self.data[i][j] == self.data[i][j + 1]:
                        reward += 1
                        self.data[i][j] <<= 1
                        self.data[i][j + 1] = 0
        
        elif direction == 3:
            for j in reversed(range(1, 4)):
                for i in range(4):
                    if self.data[i][j] != 0 and self.data[i][j] == self.data[i][j - 1]:
                        reward += 1
                        self.data[i][j] <<= 1
                        self.data[i][j - 1] = 0
        
        self.zero_to_end(direction)
        return reward
    
    def zero_to_end(self, direction):
        
        if direction == 0:
            self.get_T()
            for i in range(4):
                j = 0
                n = 3
                while j < n:
                    if self.data[i][j] == 0:
                        self.data[i].pop(j)
                        self.data[i].append(0)
                        n -= 1
                        continue
                    j += 1
            self.get_T()
        elif direction == 1:
            self.get_T()
            for i in range(4):
                j = 3
                n = 0
                while j > n:
                    if self.data[i][j] == 0:
                        self.data[i].pop(j)
                        self.data[i].insert(0, 0)
                        n += 1
                        continue
                    j -= 1
            self.get_T()
        elif direction == 2:
            for i in range(4):
                j = 0
                n = 3
                while j < n:
                    if self.data[i][j] == 0:
                        self.data[i].pop(j)
                        self.data[i].append(0)
                        n -= 1
                        continue
                    j += 1
        elif direction == 3:
            for i in range(4):
                j = 3
                n = 0
                while j > n:
                    if self.data[i][j] == 0:
                        self.data[i].pop(j)
                        self.data[i].insert(0, 0)
                        n += 1
                        continue
                    j -= 1
    
    def check(self):
        if self.data[3][3] == 0:
            return False
        for i in range(3):
            for j in range(3):
                if self.data[i][j] == 0 or self.data[i][j] == self.data[i + 1][j] or self.data[i][j] == self.data[i][j + 1]:
                    return False
            if self.data[i][3] == 0 or self.data[i][3] == self.data[i + 1][3]:
                return False
            if self.data[3][i] == 0 or self.data[3][i] == self.data[3][i + 1]:
                return False
        else:
            return True
    
    def reset(self):
        self.data = [[0, 0, 0, 0] for i in range(4)]
        self.old_data = self.data.copy()
        self.random_fill()
        self.random_fill()
        self.over = False
        self.draw()
        return self.data.copy()
    
    def random_fill(self):
        unfilled = [(i, j) for i in range(4) for j in range(4) if self.data[i][j] == 0]
        if unfilled:
            pos = random.choice(unfilled)
            self.data[pos[0]][pos[1]] = random.choice([2, 4])
        else:
            self.over = True
    
    # def is_changed(self):
    #     for i in range(4):
    #         for j in range(4):
    #             if self.data[i][j] != self.old_data[i][j]:
    #                 return True
    #     return False
    
    def step(self, action):
        
        reward = self.merge(action)
        
        self.over = self.check()
        # changed = self.is_changed()
        if self.over:
            reward -= 10
        else:
            self.update()
            self.show_data()
        # return self.data.copy(), reward, self.over, changed
        return self.data.copy(), reward, self.over

    def update(self):
        self.random_fill()
        
        self.draw()
    
    def start(self):
        
        while True:
            self.clock.tick(30)
            
            if self.over:
                self.reset()
            
            for event in pygame.event.get():
                
                if event.type == pygame.QUIT:
                    pygame.quit()
                    sys.exit()
                if event.type==pygame.KEYDOWN:
                    if event.key==pygame.K_LEFT:
                        self.step(0)
                    elif event.key==pygame.K_RIGHT:
                        self.step(1)
                    elif event.key==pygame.K_UP:
                        self.step(2)
                    elif event.key==pygame.K_DOWN:
                        self.step(3)


a = game2048()
a.start()