ADD file via upload

upgrade.py

@@ -0,0 +1,184 @@
+import pygame
+from settings import *
+
+class Upgrade:
+	def __init__(self,player):
+
+		# 获取显示表面
+		self.display_surface = pygame.display.get_surface()
+		# 绑定玩家实例
+		self.player = player
+		# 属性数量
+		self.attribute_nr = len(player.stats)
+		# 属性名称列表
+		self.attribute_names = list(player.stats.keys())
+		# 属性最大值列表
+		self.max_values = list(player.max_stats.values())
+		# 设置字体
+		self.font = pygame.font.Font(UI_FONT, UI_FONT_SIZE)
+
+		# 创建升级选项
+		# 设置界面高度
+		self.height = self.display_surface.get_size()[1] * 0.8
+		# 设置界面宽度
+		self.width = self.display_surface.get_size()[0] // 6
+		# 创建升级选项
+		self.create_items()
+
+		# 选择系统
+		# 当前选择的属性索引
+		self.selection_index = 0
+		# 选择操作时间戳
+		self.selection_time = None
+		# 是否可以移动选择
+		self.can_move = True
+
+	def input(self):
+		# 获取按键状态
+		keys = pygame.key.get_pressed()
+		# 处理按键输入
+		if self.can_move:
+			# 右键选择下一个属性
+			if keys[pygame.K_RIGHT] and self.selection_index < self.attribute_nr - 1:
+				self.selection_index += 1
+				self.can_move = False
+				self.selection_time = pygame.time.get_ticks()
+			# 左键选择上一个属性
+			elif keys[pygame.K_LEFT] and self.selection_index >= 1:
+				self.selection_index -= 1
+				self.can_move = False
+				self.selection_time = pygame.time.get_ticks()
+			# 空格键触发当前选择的升级项
+			if keys[pygame.K_SPACE]:
+				self.can_move = False
+				self.selection_time = pygame.time.get_ticks()
+				self.item_list[self.selection_index].trigger(self.player)
+
+	def selection_cooldown(self):
+		# 处理选择操作的冷却
+		if not self.can_move:
+			# 获取当前时间
+			current_time = pygame.time.get_ticks()
+			# 检查是否达到冷却时间
+			if current_time - self.selection_time >= 300:
+				# 可以继续移动选择
+				self.can_move = True
+
+	def create_items(self):
+		# 初始化升级项列表
+		self.item_list = []
+		# 计算升级项的水平位置
+		for item, index in enumerate(range(self.attribute_nr)):
+			# horizontal position
+			full_width = self.display_surface.get_size()[0]
+			increment = full_width // self.attribute_nr
+			left = (item * increment) + (increment - self.width) // 2
+			# 计算升级项的垂直位置
+			top = self.display_surface.get_size()[1] * 0.1
+			# 创建升级项对象并添加到列表中
+			item = Item(left,top,self.width,self.height,index,self.font)
+			self.item_list.append(item)
+
+	def display(self):
+		# 处理输入
+		self.input()
+		# 处理选择操作的冷却
+		self.selection_cooldown()
+
+		for index, item in enumerate(self.item_list):
+
+			# 获取属性信息
+			name = self.attribute_names[index]
+			value = self.player.get_value_by_index(index)
+			max_value = self.max_values[index]
+			cost = self.player.get_cost_by_index(index)
+			# 在界面上显示升级项
+			item.display(self.display_surface,self.selection_index,name,value,max_value,cost)
+
+class Item:
+	def __init__(self,l,t,w,h,index,font):
+		# 初始化方法，设置升级项的矩形区域、索引和字体
+		# 设置升级项的矩形区域
+		self.rect = pygame.Rect(l,t,w,h)
+		# 升级项的索引，对应于玩家的属性
+		self.index = index
+		# 字体对象，用于绘制文本
+		self.font = font
+
+	def display_names(self,surface,name,cost,selected):
+		# 显示升级项的名称和升级成本
+		# 根据是否选中设置颜色
+		color = TEXT_COLOR_SELECTED if selected else TEXT_COLOR
+
+		# 绘制标题（名称）
+		# 渲染标题文本
+		title_surf = self.font.render(name,False,color)
+		# 设置标题文本的位置
+		title_rect = title_surf.get_rect(midtop = self.rect.midtop + pygame.math.Vector2(0,20))
+
+		# 绘制升级成本
+		# 渲染成本文本
+		cost_surf = self.font.render(f'{int(cost)}',False,color)
+		# 设置成本文本的位置
+		cost_rect = cost_surf.get_rect(midbottom = self.rect.midbottom - pygame.math.Vector2(0,20))
+
+		# 在界面上绘制标题和成本文本
+		surface.blit(title_surf,title_rect)
+		surface.blit(cost_surf,cost_rect)
+
+	def display_bar(self,surface,value,max_value,selected):
+		# 显示升级项的数值条
+		# 数值条的顶部位置
+		top = self.rect.midtop + pygame.math.Vector2(0,60)
+		# 数值条的底部位置
+		bottom = self.rect.midbottom - pygame.math.Vector2(0,60)
+		# 根据是否选中设置颜色
+		color = BAR_COLOR_SELECTED if selected else BAR_COLOR
+
+		# 计算数值条的长度和位置
+		# 数值条的总长度
+		full_height = bottom[1] - top[1]
+		# 数值的相对位置
+		relative_number = (value / max_value) * full_height
+		# 数值条的矩形区域
+		value_rect = pygame.Rect(top[0] - 15,bottom[1] - relative_number,30,10)
+
+		# 绘制数值条的背景线和填充
+		# 绘制数值条的背景线
+		pygame.draw.line(surface,color,top,bottom,5)
+		# 填充数值条
+		pygame.draw.rect(surface,color,value_rect)
+
+	def trigger(self,player):
+		# 触发升级项效果
+		# 获取对应的属性名称
+		upgrade_attribute = list(player.stats.keys())[self.index]
+		# 检查是否满足升级条件
+		if player.exp >= player.upgrade_cost[upgrade_attribute] and player.stats[upgrade_attribute] < player.max_stats[upgrade_attribute]:
+			# 扣除升级成本
+			player.exp -= player.upgrade_cost[upgrade_attribute]
+			# 属性增加20%
+			player.stats[upgrade_attribute] *= 1.2
+			# 升级成本增加40%
+			player.upgrade_cost[upgrade_attribute] *= 1.4
+		# 防止属性超过最大值
+		if player.stats[upgrade_attribute] > player.max_stats[upgrade_attribute]:
+			player.stats[upgrade_attribute] = player.max_stats[upgrade_attribute]
+
+	def display(self,surface,selection_num,name,value,max_value,cost):
+		# 在界面上显示升级项
+		if self.index == selection_num:
+			# 绘制选中效果
+			pygame.draw.rect(surface,UPGRADE_BG_COLOR_SELECTED,self.rect)
+			# 绘制边框
+			pygame.draw.rect(surface,UI_BORDER_COLOR,self.rect,4)
+		else:
+			# 绘制背景
+			pygame.draw.rect(surface,UI_BG_COLOR,self.rect)
+			# 绘制边框
+			pygame.draw.rect(surface,UI_BORDER_COLOR,self.rect,4)
+		# 显示升级项的名称、数值条和成本
+		# 显示名称和成本
+		self.display_names(surface,name,cost,self.index == selection_num)
+		# 显示数值条
+		self.display_bar(surface,value,max_value,self.index == selection_num)