Python 植物大戰殭屍程式碼實現(2):植物卡片選擇和種植

marble_xu發表於2020-01-23

功能介紹

最近一直在給這個植物大戰殭屍遊戲新增新的植物和殭屍, 因為網上的圖片資源有限,能加的植物和殭屍比較少, 目前進展如下:
功能實現如下:

  • 支援的植物型別:太陽花,豌豆射手,寒冰射手,堅果,櫻桃炸彈。新增加植物:雙重豌豆射手,三重豌豆射手,食人花 ,小噴菇,土豆地雷,倭瓜。
  • 支援的殭屍型別:普通殭屍,棋子殭屍,路障殭屍,鐵桶殭屍。新增加讀報殭屍。
  • 使用json檔案儲存關卡資訊,設定殭屍出現的時間和位置。
  • 增加每關開始時選擇上場植物。
  • 增加除草機。

Python 植物大戰殭屍程式碼實現(1):圖片載入和顯示切換

下面是遊戲的截圖:
圖1:新增的植物和殭屍
demo1
圖2:每關開始選擇上場植物卡片
demo2圖3:選擇植物在哪裡種植
demo3

植物卡片選擇和種植

如圖3所示,遊戲中可以種植物的方格一共有45個(有5行,每行9列)。
這篇文章要介紹的是:

  • 上方植物卡片欄的實現。
  • 點選植物卡片,滑鼠切換為植物圖片。
  • 滑鼠移動時,判斷當前在哪個方格中,並顯示半透明的植物作為提示。

完整程式碼

遊戲實現程式碼的github連結 植物大戰殭屍
這邊是csdn的下載連結 植物大戰殭屍

程式碼實現

所有的植物卡片的名稱和屬性都儲存在單獨的list中,每個list index都對應一種植物。
比如list index 0 就是太陽花:

  • card_name_list[0] 是太陽花卡片的名字,用來獲取太陽花卡片的圖片。
  • plant_name_list[0] 是太陽花的名字,用來獲取太陽花卡片的圖片。
  • plant_sun_list[0] 是種植太陽花需要花費的太陽點數。
  • plant_frozen_time_list[0] 是太陽花的冷卻時間。

程式碼在source\component\menubar.py中

card_name_list = [c.CARD_SUNFLOWER, c.CARD_PEASHOOTER, c.CARD_SNOWPEASHOOTER, c.CARD_WALLNUT,
                  c.CARD_CHERRYBOMB, c.CARD_THREEPEASHOOTER, c.CARD_REPEATERPEA, c.CARD_CHOMPER,
                  c.CARD_PUFFMUSHROOM, c.CARD_POTATOMINE, c.CARD_SQUASH]
plant_name_list = [c.SUNFLOWER, c.PEASHOOTER, c.SNOWPEASHOOTER, c.WALLNUT,
                   c.CHERRYBOMB, c.THREEPEASHOOTER, c.REPEATERPEA, c.CHOMPER,
                   c.PUFFMUSHROOM, c.POTATOMINE, c.SQUASH]
plant_sun_list = [50, 100, 175, 50, 150, 325, 200, 150, 0, 25, 50]
plant_frozen_time_list = [0, 5000, 5000, 10000, 5000, 5000, 5000, 5000, 8000, 8000, 8000]
all_card_list = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

植物卡片類

程式碼在source\component\menubar.py中,每個植物卡片是一個單獨的Card類,用來顯示這個植物。

  • checkMouseClick函式:判斷滑鼠是否點選到這個卡片
  • canClick:判斷這個卡片是否能種植(有沒有足夠的點數,是否還在冷卻時間內)
  • update 函式:通過設定圖片的透明度來表示這個卡片是否能選擇
class Card():
    def __init__(self, x, y, name_index, scale=0.78):
        self.loadFrame(card_name_list[name_index], scale)
        self.rect = self.image.get_rect()
        self.rect.x = x
        self.rect.y = y
        
        self.name_index = name_index
        self.sun_cost = plant_sun_list[name_index]
        self.frozen_time = plant_frozen_time_list[name_index]
        self.frozen_timer = -self.frozen_time
        self.select = True

    def loadFrame(self, name, scale):
        frame = tool.GFX[name]
        rect = frame.get_rect()
        width, height = rect.w, rect.h

        self.image = tool.get_image(frame, 0, 0, width, height, c.BLACK, scale)
    
    def checkMouseClick(self, mouse_pos):
        x, y = mouse_pos
        if(x >= self.rect.x and x <= self.rect.right and
           y >= self.rect.y and y <= self.rect.bottom):
            return True
        return False

    def canClick(self, sun_value, current_time):
        if self.sun_cost <= sun_value and (current_time - self.frozen_timer) > self.frozen_time:
            return True
        return False

    def canSelect(self):
        return self.select

    def setSelect(self, can_select):
        self.select = can_select
        if can_select:
            self.image.set_alpha(255)
        else:
            self.image.set_alpha(128)

    def setFrozenTime(self, current_time):
        self.frozen_timer = current_time

    def update(self, sun_value, current_time):
        if (self.sun_cost > sun_value or
            (current_time - self.frozen_timer) <= self.frozen_time):
            self.image.set_alpha(128)
        else:
            self.image.set_alpha(255)

    def draw(self, surface):
        surface.blit(self.image, self.rect)

卡片欄類

程式碼在source\component\menubar.py中,MenuBar類顯示圖3中的植物卡片欄,

  • self.sun_value:當前採集的太陽點數
  • self.card_list: 植物卡片的list
  • setupCards函式:遍歷初始化__init__函式中傳入這個關卡選好的植物卡片list,依次建立Card類,設定每個卡片的顯示位置。
  • checkCardClick函式:檢查滑鼠是否點選了卡片欄上的某個植物卡片,如果選擇了一個可種植的卡片,返回結果。
class MenuBar():
    def __init__(self, card_list, sun_value):
        self.loadFrame(c.MENUBAR_BACKGROUND)
        self.rect = self.image.get_rect()
        self.rect.x = 10
        self.rect.y = 0
        
        self.sun_value = sun_value
        self.card_offset_x = 32
        self.setupCards(card_list)

    def loadFrame(self, name):
        frame = tool.GFX[name]
        rect = frame.get_rect()
        frame_rect = (rect.x, rect.y, rect.w, rect.h)

        self.image = tool.get_image(tool.GFX[name], *frame_rect, c.WHITE, 1)

    def update(self, current_time):
        self.current_time = current_time
        for card in self.card_list:
            card.update(self.sun_value, self.current_time)

    def createImage(self, x, y, num):
        if num == 1:
            return
        img = self.image
        rect = self.image.get_rect()
        width = rect.w
        height = rect.h
        self.image = pg.Surface((width * num, height)).convert()
        self.rect = self.image.get_rect()
        self.rect.x = x
        self.rect.y = y
        for i in range(num):
            x = i * width
            self.image.blit(img, (x,0))
        self.image.set_colorkey(c.BLACK)
    
    def setupCards(self, card_list):
        self.card_list = []
        x = self.card_offset_x
        y = 8
        for index in card_list:
            x += 55
            self.card_list.append(Card(x, y, index))

    def checkCardClick(self, mouse_pos):
        result = None
        for card in self.card_list:
            if card.checkMouseClick(mouse_pos):
                if card.canClick(self.sun_value, self.current_time):
                    result = (plant_name_list[card.name_index], card.sun_cost)
                break
        return result
    
    def checkMenuBarClick(self, mouse_pos):
        x, y = mouse_pos
        if(x >= self.rect.x and x <= self.rect.right and
           y >= self.rect.y and y <= self.rect.bottom):
            return True
        return False

    def decreaseSunValue(self, value):
        self.sun_value -= value

    def increaseSunValue(self, value):
        self.sun_value += value

    def setCardFrozenTime(self, plant_name):
        for card in self.card_list:
            if plant_name_list[card.name_index] == plant_name:
                card.setFrozenTime(self.current_time)
                break

    def drawSunValue(self):
        self.value_image = getSunValueImage(self.sun_value)
        self.value_rect = self.value_image.get_rect()
        self.value_rect.x = 21
        self.value_rect.y = self.rect.bottom - 21
        
        self.image.blit(self.value_image, self.value_rect)

    def draw(self, surface):
        self.drawSunValue()
        surface.blit(self.image, self.rect)
        for card in self.card_list:
            card.draw(surface)

滑鼠圖片切換

程式碼在source\state\level.py中,setupMouseImage 函式實現滑鼠圖片切換為選中的植物

  • self.mouse_image :根據 plant_name 獲取選中的植物圖片
  • self.mouse_rect:選中植物圖片的位置,在drawMouseShow函式中,需要將植物圖片的位置設定成當前滑鼠的位置
  • pg.mouse.set_visible(False):隱藏預設的滑鼠顯示,這樣效果就是滑鼠圖片切換為選中的植物了。
    def setupMouseImage(self, plant_name, plant_cost):
        frame_list = tool.GFX[plant_name]
        if plant_name in tool.PLANT_RECT:
            data = tool.PLANT_RECT[plant_name]
            x, y, width, height = data['x'], data['y'], data['width'], data['height']
        else:
            x, y = 0, 0
            rect = frame_list[0].get_rect()
            width, height = rect.w, rect.h

        if plant_name == c.POTATOMINE or plant_name == c.SQUASH:
            color = c.WHITE
        else:
            color = c.BLACK
        self.mouse_image = tool.get_image(frame_list[0], x, y, width, height, color, 1)
        self.mouse_rect = self.mouse_image.get_rect()
        pg.mouse.set_visible(False)
        self.drag_plant = True
        self.plant_name = plant_name
        self.plant_cost = plant_cost

    def drawMouseShow(self, surface):
        if self.hint_plant:
            surface.blit(self.hint_image, self.hint_rect)
        x, y = pg.mouse.get_pos()
        self.mouse_rect.centerx = x
        self.mouse_rect.centery = y
        surface.blit(self.mouse_image, self.mouse_rect)

提示種在哪個方格中

先看下map類,程式碼在source\component\map.py 中

  • self.map:二維list,用來儲存每個方格的狀態。每個entry初始化為 0, 表示可以種植物,值為1時表示這個方格已經種了植物。
  • getMapIndex 函式:傳入引數是遊戲中的座標位置(比如當前滑鼠的位置),返回該位置在地圖的哪個方格中。
  • getMapGridPos 函式:傳入一個方格的index,返回在該方格中種植物的座標位置。
  • showPlant 函式:根據傳入的座標位置,判斷該位置所在的方格是否能種植物,如果能種,就返回返回在該方格中種植物的座標位置。
MAP_EMPTY = 0
MAP_EXIST = 1

class Map():
    def __init__(self, width, height):
        self.width = width
        self.height = height
        self.map = [[0 for x in range(self.width)] for y in range(self.height)]

    def isValid(self, map_x, map_y):
        if (map_x < 0 or map_x >= self.width or
            map_y < 0 or map_y >= self.height):
            return False
        return True
    
    def isMovable(self, map_x, map_y):
        return (self.map[map_y][map_x] == c.MAP_EMPTY)
    
    def getMapIndex(self, x, y):
        x -= c.MAP_OFFSET_X
        y -= c.MAP_OFFSET_Y
        return (x // c.GRID_X_SIZE, y // c.GRID_Y_SIZE)
    
    def getMapGridPos(self, map_x, map_y):
        return (map_x * c.GRID_X_SIZE + c.GRID_X_SIZE//2 + c.MAP_OFFSET_X,
                map_y * c.GRID_Y_SIZE + c.GRID_Y_SIZE//5 * 3 + c.MAP_OFFSET_Y)
    
    def setMapGridType(self, map_x, map_y, type):
        self.map[map_y][map_x] = type

    def getRandomMapIndex(self):
        map_x = random.randint(0, self.width-1)
        map_y = random.randint(0, self.height-1)
        return (map_x, map_y)

    def showPlant(self, x, y):
        pos = None
        map_x, map_y = self.getMapIndex(x, y)
        if self.isValid(map_x, map_y) and self.isMovable(map_x, map_y):
            pos = self.getMapGridPos(map_x, map_y)
        return pos

程式碼在source\state\level.py中,

  • canSeedPlant 函式:判斷當前滑鼠位置能否種植物
  • setupHintImage 函式:如果當前滑鼠位置能種植物,且有選擇了一個植物卡片,則設定self.hint_image 顯示當前會在哪一個方格中種植物,self.hint_rect 是植物種的座標位置。
    def canSeedPlant(self):
        x, y = pg.mouse.get_pos()
        return self.map.showPlant(x, y)
 
    def setupHintImage(self):
        pos = self.canSeedPlant()
        if pos and self.mouse_image:
            if (self.hint_image and pos[0] == self.hint_rect.x and
                pos[1] == self.hint_rect.y):
                return
            width, height = self.mouse_rect.w, self.mouse_rect.h
            image = pg.Surface([width, height])
            image.blit(self.mouse_image, (0, 0), (0, 0, width, height))
            image.set_colorkey(c.BLACK)
            image.set_alpha(128)
            self.hint_image = image
            self.hint_rect = image.get_rect()
            self.hint_rect.centerx = pos[0]
            self.hint_rect.bottom = pos[1]
            self.hint_plant = True
        else:
            self.hint_plant = False

編譯環境

python3.7 + pygame1.9

相關文章