""" 地形编辑工具模块 """ import math import random from panda3d.core import PNMImage class TerrainTools: """ 地形编辑工具类 提供完整的地形编辑功能，包括抬高、降低、平坦化、平滑、噪声等工具 """ def __init__(self, world): self.world = world self.current_tool = "抬高工具" self.radius = 10.0 self.strength = 50.0 self.operation = "添加" self.is_editing = False self.last_edit_pos = None self.brush_spacing = 1.0 # 画笔间隔，用于连续编辑 # 工具参数 self.smooth_factor = 0.5 self.noise_scale = 1.0 self.erosion_factor = 0.1 self.thermal_erosion_factor = 0.05 def set_tool(self, tool_name): """ 设置当前工具 """ valid_tools = ["抬高工具", "降低工具", "平坦化工具", "平滑工具", "噪声工具", "侵蚀工具", "热侵蚀工具"] if tool_name in valid_tools: self.current_tool = tool_name print(f"设置地形编辑工具: {tool_name}") return True else: print(f"无效的工具名称: {tool_name}") return False def set_radius(self, radius): """ 设置工具半径 """ self.radius = max(0.1, min(100.0, float(radius))) def set_strength(self, strength): """ 设置工具强度 """ self.strength = max(0.0, min(100.0, float(strength))) def set_operation(self, operation): """ 设置操作类型 """ valid_operations = ["添加", "减去", "设置"] if operation in valid_operations: self.operation = operation else: print(f"无效的操作类型: {operation}") def set_smooth_factor(self, factor): """ 设置平滑因子 """ self.smooth_factor = max(0.0, min(1.0, float(factor))) def set_noise_scale(self, scale): """ 设置噪声缩放 """ self.noise_scale = max(0.1, min(10.0, float(scale))) def set_erosion_factor(self, factor): """ 设置侵蚀因子 """ self.erosion_factor = max(0.0, min(1.0, float(factor))) def set_thermal_erosion_factor(self, factor): """ 设置热侵蚀因子 """ self.thermal_erosion_factor = max(0.0, min(1.0, float(factor))) def set_brush_spacing(self, spacing): """ 设置画笔间隔 """ self.brush_spacing = max(0.1, min(10.0, float(spacing))) def start_editing(self, x, y): """ 开始编辑地形 """ self.is_editing = True self.last_edit_pos = (x, y) self._apply_tool(x, y) def update_editing(self, x, y): """ 更新编辑过程 """ if self.is_editing: # 检查是否需要应用工具（基于画笔间隔） if self.last_edit_pos: distance = math.sqrt((x - self.last_edit_pos[0])**2 + (y - self.last_edit_pos[1])**2) if distance >= self.brush_spacing: self._apply_tool(x, y) self.last_edit_pos = (x, y) else: self._apply_tool(x, y) self.last_edit_pos = (x, y) def stop_editing(self): """ 停止编辑地形 """ self.is_editing = False self.last_edit_pos = None def _apply_tool(self, x, y): """ 应用当前工具到指定位置 """ # 获取当前选中的地形 selected_terrain = self._get_selected_terrain() if not selected_terrain: # 如果没有选中地形，尝试获取场景中的第一个地形 selected_terrain = self._get_first_terrain() if not selected_terrain: print("没有可用的地形") return # 根据当前工具类型应用不同的操作 if self.current_tool == "抬高工具": self._raise_terrain(selected_terrain, x, y) elif self.current_tool == "降低工具": self._lower_terrain(selected_terrain, x, y) elif self.current_tool == "平坦化工具": self._flatten_terrain(selected_terrain, x, y) elif self.current_tool == "平滑工具": self._smooth_terrain(selected_terrain, x, y) elif self.current_tool == "噪声工具": self._noise_terrain(selected_terrain, x, y) elif self.current_tool == "侵蚀工具": self._erode_terrain(selected_terrain, x, y) elif self.current_tool == "热侵蚀工具": self._thermal_erode_terrain(selected_terrain, x, y) def _get_selected_terrain(self): """ 获取当前选中的地形 """ # 如果有选择系统，获取选中的地形 if hasattr(self.world, 'selection') and self.world.selection.selectedNode: selected_node = self.world.selection.selectedNode # 检查选中的节点是否是地形节点 if selected_node.getTag("tree_item_type") == "TERRAIN_NODE": # 在地形管理器中查找对应的地形信息 if hasattr(self.world, 'terrain_manager'): for terrain_info in self.world.terrain_manager.terrains: if terrain_info['node'] == selected_node: return terrain_info return None def _get_first_terrain(self): """ 获取场景中的第一个地形 """ if hasattr(self.world, 'terrain_manager') and self.world.terrain_manager.terrains: return self.world.terrain_manager.terrains[0] return None def _raise_terrain(self, terrain_info, x, y): """ 抬高地形 """ if hasattr(self.world, 'terrain_manager'): operation = "add" if self.operation == "减去": operation = "subtract" elif self.operation == "设置": operation = "set" self.world.terrain_manager.modifyTerrainHeight( terrain_info, x, y, self.radius, self.strength, operation ) def _lower_terrain(self, terrain_info, x, y): """ 降低地形 """ if hasattr(self.world, 'terrain_manager'): operation = "subtract" if self.operation == "添加": operation = "add" elif self.operation == "设置": operation = "set" self.world.terrain_manager.modifyTerrainHeight( terrain_info, x, y, self.radius, self.strength, operation ) def _flatten_terrain(self, terrain_info, x, y): """ 平坦化地形 """ if hasattr(self.world, 'terrain_manager'): # 平坦化工具使用特殊的"set"操作 self.world.terrain_manager.modifyTerrainHeight( terrain_info, x, y, self.radius, self.strength, "set" ) def _smooth_terrain(self, terrain_info, x, y): """ 平滑地形 """ if not hasattr(self.world, 'terrain_manager'): return try: terrain_node = terrain_info['node'] heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return # 获取高度图尺寸 width = heightfield.getXSize() height = heightfield.getYSize() # 获取地形节点信息 terrain_pos = terrain_node.getPos() terrain_scale = terrain_node.getScale() # 坐标转换 relative_x = x - terrain_pos.getX() relative_y = y - terrain_pos.getY() scaled_x = relative_x / terrain_scale.getX() scaled_y = relative_y / terrain_scale.getY() center_offset = (width - 1) / 2 center_x = int(scaled_x + center_offset) center_y = int(scaled_y + center_offset) # 检查中心点是否在有效范围内 if not (0 <= center_x < width and 0 <= center_y < height): return # 计算有效编辑半径 effective_radius = max(1, int(self.radius / max(terrain_scale.getX(), terrain_scale.getY()))) effective_radius = min(effective_radius, width // 4, height // 4) # 创建临时高度图用于存储平滑结果 temp_heightfield = PNMImage(width, height) temp_heightfield.copyFrom(heightfield) # 应用平滑算法 modified = False for dx in range(-effective_radius, effective_radius + 1): for dy in range(-effective_radius, effective_radius + 1): distance = math.sqrt(dx*dx + dy*dy) if distance <= effective_radius: target_x = center_x + dx target_y = center_y + dy # 检查边界 if 0 <= target_x < width and 0 <= target_y < height: # 计算邻域平均值 total_height = 0.0 count = 0 # 检查周围的8个邻居 for nx in range(-1, 2): for ny in range(-1, 2): neighbor_x = target_x + nx neighbor_y = target_y + ny # 检查邻居是否在边界内 if 0 <= neighbor_x < width and 0 <= neighbor_y < height: total_height += heightfield.getRed(neighbor_x, neighbor_y) count += 1 if count > 0: average_height = total_height / count current_height = heightfield.getRed(target_x, target_y) # 应用平滑因子 new_height = current_height * (1 - self.smooth_factor) + average_height * self.smooth_factor # 设置新高度到临时高度图 temp_heightfield.setRed(target_x, target_y, new_height) temp_heightfield.setGreen(target_x, target_y, new_height) temp_heightfield.setBlue(target_x, target_y, new_height) modified = True # 如果有修改，则更新地形 if modified: # 更新高度图 terrain_info['heightfield'] = temp_heightfield # 重新设置高度图并生成地形 terrain = terrain_info['terrain'] terrain.setHeightfield(temp_heightfield) terrain.generate() # 重新创建碰撞体 self._recreate_collision(terrain_node) print(f"✓ 地形已平滑: 位置({x}, {y}), 半径{self.radius}") except Exception as e: print(f"平滑地形时出错: {e}") def _noise_terrain(self, terrain_info, x, y): """ 添加噪声到地形 """ if not hasattr(self.world, 'terrain_manager'): return try: terrain_node = terrain_info['node'] heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return # 获取高度图尺寸 width = heightfield.getXSize() height = heightfield.getYSize() # 获取地形节点信息 terrain_pos = terrain_node.getPos() terrain_scale = terrain_node.getScale() # 坐标转换 relative_x = x - terrain_pos.getX() relative_y = y - terrain_pos.getY() scaled_x = relative_x / terrain_scale.getX() scaled_y = relative_y / terrain_scale.getY() center_offset = (width - 1) / 2 center_x = int(scaled_x + center_offset) center_y = int(scaled_y + center_offset) # 检查中心点是否在有效范围内 if not (0 <= center_x < width and 0 <= center_y < height): return # 计算有效编辑半径 effective_radius = max(1, int(self.radius / max(terrain_scale.getX(), terrain_scale.getY()))) effective_radius = min(effective_radius, width // 4, height // 4) # 应用噪声算法 modified = False for dx in range(-effective_radius, effective_radius + 1): for dy in range(-effective_radius, effective_radius + 1): distance = math.sqrt(dx*dx + dy*dy) if distance <= effective_radius: # 计算影响权重 weight = 1.0 - (distance / effective_radius) if effective_radius > 0 else 1.0 target_x = center_x + dx target_y = center_y + dy # 检查边界 if 0 <= target_x < width and 0 <= target_y < height: # 生成噪声值 noise_value = random.uniform(-1.0, 1.0) * self.noise_scale # 获取当前高度 current_height = heightfield.getRed(target_x, target_y) # 应用噪声 new_height = max(0.0, min(1.0, current_height + noise_value * weight * self.strength * 0.01)) # 设置新高度 heightfield.setRed(target_x, target_y, new_height) heightfield.setGreen(target_x, target_y, new_height) heightfield.setBlue(target_x, target_y, new_height) modified = True # 如果有修改，则更新地形 if modified: # 重新设置高度图并生成地形 terrain = terrain_info['terrain'] terrain.setHeightfield(heightfield) terrain.generate() # 重新创建碰撞体 self._recreate_collision(terrain_node) print(f"✓ 地形已添加噪声: 位置({x}, {y}), 半径{self.radius}") except Exception as e: print(f"添加噪声到地形时出错: {e}") def _erode_terrain(self, terrain_info, x, y): """ 水力侵蚀地形 """ if not hasattr(self.world, 'terrain_manager'): return try: terrain_node = terrain_info['node'] heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return # 获取高度图尺寸 width = heightfield.getXSize() height = heightfield.getYSize() # 获取地形节点信息 terrain_pos = terrain_node.getPos() terrain_scale = terrain_node.getScale() # 坐标转换 relative_x = x - terrain_pos.getX() relative_y = y - terrain_pos.getY() scaled_x = relative_x / terrain_scale.getX() scaled_y = relative_y / terrain_scale.getY() center_offset = (width - 1) / 2 center_x = int(scaled_x + center_offset) center_y = int(scaled_y + center_offset) # 检查中心点是否在有效范围内 if not (0 <= center_x < width and 0 <= center_y < height): return # 计算有效编辑半径 effective_radius = max(1, int(self.radius / max(terrain_scale.getX(), terrain_scale.getY()))) effective_radius = min(effective_radius, width // 4, height // 4) # 应用简单的侵蚀算法 modified = False for dx in range(-effective_radius, effective_radius + 1): for dy in range(-effective_radius, effective_radius + 1): distance = math.sqrt(dx*dx + dy*dy) if distance <= effective_radius: # 计算影响权重 weight = 1.0 - (distance / effective_radius) if effective_radius > 0 else 1.0 target_x = center_x + dx target_y = center_y + dy # 检查边界 if 0 <= target_x < width and 0 <= target_y < height: # 简单的侵蚀模型：降低高点，填充低点 current_height = heightfield.getRed(target_x, target_y) # 根据高度确定侵蚀效果 # 高度越高，侵蚀越强 erosion_amount = self.erosion_factor * weight * (1.0 - current_height) * self.strength * 0.01 # 应用侵蚀 new_height = max(0.0, min(1.0, current_height - erosion_amount)) # 设置新高度 heightfield.setRed(target_x, target_y, new_height) heightfield.setGreen(target_x, target_y, new_height) heightfield.setBlue(target_x, target_y, new_height) modified = True # 如果有修改，则更新地形 if modified: # 重新设置高度图并生成地形 terrain = terrain_info['terrain'] terrain.setHeightfield(heightfield) terrain.generate() # 重新创建碰撞体 self._recreate_collision(terrain_node) print(f"✓ 地形已应用侵蚀: 位置({x}, {y}), 半径{self.radius}") except Exception as e: print(f"侵蚀地形时出错: {e}") def _thermal_erode_terrain(self, terrain_info, x, y): """ 热侵蚀地形（模拟重力作用） """ if not hasattr(self.world, 'terrain_manager'): return try: terrain_node = terrain_info['node'] heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return # 获取高度图尺寸 width = heightfield.getXSize() height = heightfield.getYSize() # 获取地形节点信息 terrain_pos = terrain_node.getPos() terrain_scale = terrain_node.getScale() # 坐标转换 relative_x = x - terrain_pos.getX() relative_y = y - terrain_pos.getY() scaled_x = relative_x / terrain_scale.getX() scaled_y = relative_y / terrain_scale.getY() center_offset = (width - 1) / 2 center_x = int(scaled_x + center_offset) center_y = int(scaled_y + center_offset) # 检查中心点是否在有效范围内 if not (0 <= center_x < width and 0 <= center_y < height): return # 计算有效编辑半径 effective_radius = max(1, int(self.radius / max(terrain_scale.getX(), terrain_scale.getY()))) effective_radius = min(effective_radius, width // 4, height // 4) # 应用热侵蚀算法 modified = False for dx in range(-effective_radius, effective_radius + 1): for dy in range(-effective_radius, effective_radius + 1): distance = math.sqrt(dx*dx + dy*dy) if distance <= effective_radius: # 计算影响权重 weight = 1.0 - (distance / effective_radius) if effective_radius > 0 else 1.0 target_x = center_x + dx target_y = center_y + dy # 检查边界 if 0 <= target_x < width and 0 <= target_y < height: # 检查周围的邻居，找到最低点 current_height = heightfield.getRed(target_x, target_y) lowest_neighbor_height = current_height lowest_neighbor_x = target_x lowest_neighbor_y = target_y # 检查8个邻居 for nx in range(-1, 2): for ny in range(-1, 2): if nx == 0 and ny == 0: continue # 跳过中心点 neighbor_x = target_x + nx neighbor_y = target_y + ny # 检查邻居是否在边界内 if 0 <= neighbor_x < width and 0 <= neighbor_y < height: neighbor_height = heightfield.getRed(neighbor_x, neighbor_y) if neighbor_height < lowest_neighbor_height: lowest_neighbor_height = neighbor_height lowest_neighbor_x = neighbor_x lowest_neighbor_y = neighbor_y # 如果有更低的邻居，移动部分物质 height_diff = current_height - lowest_neighbor_height if height_diff > 0: # 移动物质 move_amount = min(height_diff, self.thermal_erosion_factor * weight * self.strength * 0.01) # 降低当前点 new_height = max(0.0, current_height - move_amount) heightfield.setRed(target_x, target_y, new_height) heightfield.setGreen(target_x, target_y, new_height) heightfield.setBlue(target_x, target_y, new_height) # 提高最低邻居点 neighbor_new_height = min(1.0, lowest_neighbor_height + move_amount * 0.5) heightfield.setRed(lowest_neighbor_x, lowest_neighbor_y, neighbor_new_height) heightfield.setGreen(lowest_neighbor_x, lowest_neighbor_y, neighbor_new_height) heightfield.setBlue(lowest_neighbor_x, lowest_neighbor_y, neighbor_new_height) modified = True # 如果有修改，则更新地形 if modified: # 重新设置高度图并生成地形 terrain = terrain_info['terrain'] terrain.setHeightfield(heightfield) terrain.generate() # 重新创建碰撞体 self._recreate_collision(terrain_node) print(f"✓ 地形已应用热侵蚀: 位置({x}, {y}), 半径{self.radius}") except Exception as e: print(f"热侵蚀地形时出错: {e}") def _recreate_collision(self, terrain_node): """ 重新创建地形碰撞体 """ try: from panda3d.core import BitMask32 # 移除旧的地形碰撞体 for child in terrain_node.getChildren(): if child.getName().startswith("terrain_collision_"): child.removeNode() # 设置地形节点的碰撞掩码 terrain_node.setCollideMask(BitMask32.bit(2)) # 为地形的每个子节点也设置碰撞掩码 for child in terrain_node.getChildren(): child.setCollideMask(BitMask32.bit(2)) except Exception as e: print(f"重新创建碰撞体时出错: {e}") def apply_brush(self, terrain_info, x, y, brush_data): """ 应用画笔数据到地形 """ if not hasattr(self.world, 'terrain_manager'): return try: terrain_node = terrain_info['node'] heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return # 获取高度图尺寸 width = heightfield.getXSize() height = heightfield.getYSize() # 获取地形节点信息 terrain_pos = terrain_node.getPos() terrain_scale = terrain_node.getScale() # 坐标转换 relative_x = x - terrain_pos.getX() relative_y = y - terrain_pos.getY() scaled_x = relative_x / terrain_scale.getX() scaled_y = relative_y / terrain_scale.getY() center_offset = (width - 1) / 2 center_x = int(scaled_x + center_offset) center_y = int(scaled_y + center_offset) # 检查中心点是否在有效范围内 if not (0 <= center_x < width and 0 <= center_y < height): return # 应用画笔数据 modified = False brush_size = len(brush_data) brush_radius = brush_size // 2 for dx in range(-brush_radius, brush_radius + 1): for dy in range(-brush_radius, brush_radius + 1): if 0 <= brush_radius + dx < brush_size and 0 <= brush_radius + dy < brush_size: brush_value = brush_data[brush_radius + dy][brush_radius + dx] target_x = center_x + dx target_y = center_y + dy # 检查边界 if 0 <= target_x < width and 0 <= target_y < height: # 获取当前高度 current_height = heightfield.getRed(target_x, target_y) # 应用画笔值 new_height = max(0.0, min(1.0, current_height + brush_value * self.strength * 0.01)) # 设置新高度 heightfield.setRed(target_x, target_y, new_height) heightfield.setGreen(target_x, target_y, new_height) heightfield.setBlue(target_x, target_y, new_height) modified = True # 如果有修改，则更新地形 if modified: # 重新设置高度图并生成地形 terrain = terrain_info['terrain'] terrain.setHeightfield(heightfield) terrain.generate() # 重新创建碰撞体 self._recreate_collision(terrain_node) print(f"✓ 应用画笔数据到地形: 位置({x}, {y})") except Exception as e: print(f"应用画笔数据到地形时出错: {e}")