补一下之前写的python合成大罗马

之前python课程设计的作业
没有什么新意，就是用python写了个合成大罗马
不过自己对库的使用理解更深了

from pygame.draw import rect
import pymunk
import random
import pygame
import pickle
from moviepy.editor import VideoFileClip
import math
import time
from PIL import Image
from pymunk import pygame_util
import sys

from pymunk.vec2d import Vec2d

space = pymunk.Space()
space.gravity = (0,900)
Gsp = []
Fruit = [5, 10, 15, 21, 23, 29, 35, 37, 50, 59, 60, 78]
clock = pygame.time.Clock()

def initgame():
    pygame.font.init()
    pygame.init()
    pygame.mixer.init()
    screen = pygame.display.set_mode((640,800))
    pygame.display.set_caption("合成正统罗马")
    return screen

class CircleSprite(pygame.sprite.Sprite):
    def __init__(self, target):
        pygame.sprite.Sprite.__init__(self)
        self.image = None


class ball(CircleSprite):
    def __init__(self, num, position, space, friction=0.1):
        pygame.sprite.Sprite.__init__(self)
        self.mass = Fruit[num] * 2
        self.r = Fruit[num] * 2
        self.angle_degree = 0
        self.images = 'image\\罗马\\'+str(num)+'.png'
        self.image = pygame.image.load(self.images)
        self.image = pygame.transform.smoothscale(self.image, (self.r * 2, self.r * 2))
        self.rect = self.image.get_rect()
        self.rect.x,self.rect.y = position
        self.radius = num
        self.num = num
        self.position = position
        moment = pymunk.moment_for_circle(self.mass, 0, self.r)
        self.body = pymunk.Body(self.mass, moment)
        self.body.position = position
        self.shape = pymunk.Circle(self.body, self.r)
        self.shape.friction = friction
        self.elasticity = 0.95
        self.shape.collision_type = 1
        self.shape.sprite = self
        space.add(self.body, self.shape)
        self.space = space
        self.count = 0
        self.gameover = 0
        self.isclash = 0

    def draw(self, screen):
        screen.blit(self.image, (self.body.position.x - self.image.get_width() / 2, self.body.position.y - self.image.get_height() / 2))

    def update_position(self, x, y, angle_degree=0):
        self.rect.x = self.body.position[0]
        self.rect.y = self.body.position[1]
    def update(self):
        img2 = pygame.transform.rotate(self.image,-math.degrees(self.body.angle))
        self.update_position(self.position[0],self.position[1])
        screen.blit(img2, (self.body.position.x - img2.get_width() / 2, self.body.position.y - img2.get_height() / 2))
        if(self.body.position.y < 120):
            self.count += 1
def load_data():
    # 加载历史最高分
    file_path = "score.txt"
    with open(file_path, "r") as f:
        try:
            high_score = int(f.read())
        except:
            high_score = 0
    return high_score
    
lines = pygame.image.load("image\\line.png")

static_lines = [
        pymunk.Segment(space.static_body, (0, 0), (0, 1000), 0.0),
        pymunk.Segment(space.static_body, (640, 0), (640, 800), 0.0),
        pymunk.Segment(space.static_body, (0, 800), (640, 800), 0.0)
        ]

for line in static_lines:
            line.elasticity = 0.4  # 弹性系数
            line.friction = 0.2  # 摩擦系数
            space.add(line)

if __name__ == '__main__':
    screen = initgame()
    over = pygame.mixer.Sound("image\\ceddin deden.mp3")
    music = pygame.mixer.Sound("sound.wav")
    bg_img = pygame.image.load("back.png")
    rd = [2, 4, 8, 16, 32, 64]
    num = 1
    maxtype = 1
    flag = 0
    GaO = 0
    high_score = load_data()
    randomnum = 1
    count = 0
    ico = pygame.image.load("苏丹.ico")
    pygame.display.set_icon(ico) # 可以填img
    gameover = pygame.image.load('image\\GO.png')
    myfont = pygame.font.Font('Astrid.ttf', 60)
    clip = VideoFileClip('抬棺.mp4') #视频加载
    score = 0
    nextfruit = pygame.image.load('image\\罗马\\'+str(num)+'.png')
    while 1:
        if(GaO == 0):
            space.step(1/60) 
            if flag == 1:
                x,y = pygame.mouse.get_pos()
                b = ball(num, (x, 0), space)
                Gsp.append(b)
                flag = 0
            for event in pygame.event.get():
                if event.type == pygame.QUIT:
                    pygame.quit()
                    sys.exit()
                if event.type == pygame.MOUSEBUTTONDOWN:
                    flag = 1
                    num = randomnum
                    randomnum = random.randint(1,maxtype)
                    nextfruit = pygame.image.load('image\\罗马\\'+str(randomnum)+'.png')
            draw_options = pygame_util.DrawOptions(screen)
            scoreimage = myfont.render("score: " + str(score), True, (255,192,203))
            space.debug_draw(draw_options)
            screen.blit(bg_img, (0, 0))
            screen.blit(lines, (0,120))
            screen.blit(nextfruit, (0, 0))
            screen.blit(scoreimage, (270, 0))
            for i in range(0,len(Gsp)):
                Gsp[i].update()
                if(Gsp[i].radius > maxtype and maxtype<=5):
                    maxtype = Gsp[i].radius
                if Gsp[i].count > 100:
                    GaO = 1
            for i in Gsp:
                for j in Gsp:
                    det_x = j.body.position.x - i.body.position.x
                    det_y = j.body.position.y - i.body.position.y
                    if i != j and math.pow(det_x, 2) + math.pow(det_y, 2) < math.pow(i.r + j.r + 3, 2) and i.num == j.num and i.isclash==0 and j.isclash==0:
                        music.play()
                        score += i.num
                        i.isclash = 1 # 添加标记，标记过的就不再触发碰撞事件
                        j.isclash = 1
                        c = ball(i.num+1, i.body.position, space)
                        Gsp.append(c)
                        space.remove(i.body, i.shape) 
                        Gsp.remove(i) 
                        space.remove(j.body, j.shape)
                        Gsp.remove(j)     
        else:
            for i in Gsp:
                if i.isclash == 0:
                    # music.play() #结束音乐    
                    score += i.num
                    i.isclash = 1
                    space.remove(i.body, i.shape)
                    i.update()
                space.debug_draw(draw_options)
                screen.blit(bg_img, (0, 0))
            scoreimage = myfont.render("score: " + str(score), True, (255,255,255))
            screen.blit(gameover, (0, 0))
            # over.play() #结束图片
            clip.preview() #视频的播放
            if score > high_score:
                high_score = score
                screen.blit(myfont.render("New High Score: " + str(high_score), True, (255,255,255)),(85, 120))
                with open("score.txt", "w") as f:
                    f.write(str(high_score))
            else:
                screen.blit(myfont.render("High Score: " + str(high_score), True, (255,255,255)),(85, 120))
            screen.blit(scoreimage, (170, 50))       
        clock.tick(60)
        pygame.display.flip()