""" 高级地形编辑工具 提供噪声生成、侵蚀模拟、傅里叶地形等高级功能 """ import math import random import numpy as np from panda3d.core import PNMImage class AdvancedTerrainTools: """ 高级地形编辑工具类 提供噪声生成、侵蚀模拟、傅里叶地形等高级功能 """ def __init__(self, world): self.world = world def generate_perlin_noise(self, terrain_info, scale=1.0, octaves=4, persistence=0.5, lacunarity=2.0): """ 生成Perlin噪声地形 """ try: terrain_node = terrain_info['node'] heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return False # 获取高度图尺寸 width = heightfield.getXSize() height = heightfield.getYSize() # 生成Perlin噪声 for y in range(height): for x in range(width): # 计算噪声坐标 nx = x / width * scale ny = y / height * scale # 生成分形噪声 noise_value = self._fractal_noise(nx, ny, octaves, persistence, lacunarity) # 将噪声值映射到0-1范围 normalized_value = (noise_value + 1) / 2 heightfield.setRed(x, y, normalized_value) heightfield.setGreen(x, y, normalized_value) heightfield.setBlue(x, y, normalized_value) # 更新地形 return self._update_terrain(terrain_info) except Exception as e: print(f"生成Perlin噪声时出错: {e}") return False def _fractal_noise(self, x, y, octaves, persistence, lacunarity): """ 生成分形噪声 """ total = 0.0 frequency = 1.0 amplitude = 1.0 maxValue = 0.0 # 用于归一化 for i in range(octaves): total += self._perlin_noise(x * frequency, y * frequency) * amplitude maxValue += amplitude amplitude *= persistence frequency *= lacunarity return total / maxValue def _perlin_noise(self, x, y): """ 生成Perlin噪声 """ # 获取整数部分 X = int(x) & 255 Y = int(y) & 255 # 获取小数部分 x -= int(x) y -= int(y) # 计算平滑曲线 u = self._fade(x) v = self._fade(y) # 计算梯度 a = self._gradient(X, Y, x, y) b = self._gradient(X + 1, Y, x - 1, y) c = self._gradient(X, Y + 1, x, y - 1) d = self._gradient(X + 1, Y + 1, x - 1, y - 1) # 插值 return self._lerp(v, self._lerp(u, a, b), self._lerp(u, c, d)) def _fade(self, t): """ 平滑曲线函数 """ return t * t * t * (t * (t * 6 - 15) + 10) def _lerp(self, t, a, b): """ 线性插值 """ return a + t * (b - a) def _gradient(self, hash_val, x, y): """ 梯度函数 """ h = hash_val & 15 u = x if h < 8 else y v = y if h < 4 else (x if h in [12, 14] else 0) return (u if (h & 1) == 0 else -u) + (v if (h & 2) == 0 else -v) def simulate_hydraulic_erosion(self, terrain_info, iterations=100, rainfall=0.1, solubility=0.01, evaporation=0.01): """ 模拟水力侵蚀 """ try: heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return False width = heightfield.getXSize() height = heightfield.getYSize() # 初始化水流和沉积物数组 water = np.zeros((width, height)) sediment = np.zeros((width, height)) # 执行侵蚀迭代 for iteration in range(iterations): # 添加降雨 water += rainfall # 计算水流 for y in range(height): for x in range(width): current_height = heightfield.getRed(x, y) + water[x, y] # 查找最低邻居 lowest_height = current_height lowest_x, lowest_y = x, y for dx in [-1, 0, 1]: for dy in [-1, 0, 1]: if dx == 0 and dy == 0: continue nx, ny = x + dx, y + dy if 0 <= nx < width and 0 <= ny < height: neighbor_height = heightfield.getRed(nx, ny) + water[nx, ny] if neighbor_height < lowest_height: lowest_height = neighbor_height lowest_x, lowest_y = nx, ny # 移动水流 if lowest_x != x or lowest_y != y: flow_amount = min(water[x, y], (current_height - lowest_height) * 0.5) water[x, y] -= flow_amount water[lowest_x, lowest_y] += flow_amount # 溶解物质 dissolve_amount = flow_amount * solubility sediment[x, y] -= dissolve_amount sediment[lowest_x, lowest_y] += dissolve_amount # 沉积和蒸发 for y in range(height): for x in range(width): # 蒸发 evaporation_amount = min(water[x, y], evaporation) water[x, y] -= evaporation_amount # 沉积 if sediment[x, y] > 0: deposit_amount = min(sediment[x, y], 0.01) sediment[x, y] -= deposit_amount new_height = heightfield.getRed(x, y) + deposit_amount heightfield.setRed(x, y, new_height) heightfield.setGreen(x, y, new_height) heightfield.setBlue(x, y, new_height) # 更新地形 return self._update_terrain(terrain_info) except Exception as e: print(f"模拟水力侵蚀时出错: {e}") return False def generate_voronoi_terrain(self, terrain_info, points=100, feature_size=10.0): """ 生成Voronoi图地形 """ try: heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return False width = heightfield.getXSize() height = heightfield.getYSize() # 生成随机点 voronoi_points = [] for i in range(points): x = random.randint(0, width - 1) y = random.randint(0, height - 1) height_value = random.random() voronoi_points.append((x, y, height_value)) # 为每个像素分配最近点的高度 for y in range(height): for x in range(width): min_distance = float('inf') closest_height = 0.0 for px, py, ph in voronoi_points: distance = math.sqrt((x - px)**2 + (y - py)**2) if distance < min_distance: min_distance = distance closest_height = ph # 应用特征大小 normalized_distance = min(1.0, min_distance / feature_size) final_height = closest_height * (1.0 - normalized_distance) heightfield.setRed(x, y, final_height) heightfield.setGreen(x, y, final_height) heightfield.setBlue(x, y, final_height) # 更新地形 return self._update_terrain(terrain_info) except Exception as e: print(f"生成Voronoi地形时出错: {e}") return False def generate_fourier_terrain(self, terrain_info, frequency_count=10): """ 生成傅里叶地形 """ try: heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return False width = heightfield.getXSize() height_value = heightfield.getYSize() # 初始化高度图 for y in range(height_value): for x in range(width): heightfield.setRed(x, y, 0.5) heightfield.setGreen(x, y, 0.5) heightfield.setBlue(x, y, 0.5) # 添加不同频率的波 for i in range(frequency_count): frequency = 2 ** i amplitude = 1.0 / frequency for y in range(height_value): for x in range(width): # 添加正弦波 wave_x = math.sin(2 * math.pi * x * frequency / width) * amplitude wave_y = math.sin(2 * math.pi * y * frequency / height_value) * amplitude current_height = heightfield.getRed(x, y) new_height = current_height + (wave_x + wave_y) * 0.1 new_height = max(0.0, min(1.0, new_height)) heightfield.setRed(x, y, new_height) heightfield.setGreen(x, y, new_height) heightfield.setBlue(x, y, new_height) # 更新地形 return self._update_terrain(terrain_info) except Exception as e: print(f"生成傅里叶地形时出错: {e}") return False def generate_fault_formation(self, terrain_info, iterations=50, displacement=0.1): """ 生成断层地形 """ try: heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return False width = heightfield.getXSize() height_value = heightfield.getYSize() # 执行断层迭代 for i in range(iterations): # 随机生成断层线 angle = random.random() * 2 * math.pi center_x = random.randint(0, width - 1) center_y = random.randint(0, height_value - 1) # 计算断层方向向量 dx = math.cos(angle) dy = math.sin(angle) # 移动断层一侧的点 for y in range(height_value): for x in range(width): # 计算点到断层线的距离和方向 distance = (x - center_x) * dx + (y - center_y) * dy if distance > 0: # 提升这一侧的高度 current_height = heightfield.getRed(x, y) new_height = current_height + displacement * (iterations - i) / iterations new_height = max(0.0, min(1.0, new_height)) heightfield.setRed(x, y, new_height) heightfield.setGreen(x, y, new_height) heightfield.setBlue(x, y, new_height) # 更新地形 return self._update_terrain(terrain_info) except Exception as e: print(f"生成断层地形时出错: {e}") return False def generate_thermal_erosion(self, terrain_info, iterations=100, talus_angle=0.1): """ 生成热侵蚀地形 """ try: heightfield = terrain_info['heightfield'] if not heightfield: print("无法获取地形高度图数据") return False width = heightfield.getXSize() height_value = heightfield.getYSize() # 执行热侵蚀迭代 for iteration in range(iterations): # 创建高度变化数组 height_changes = np.zeros((width, height_value)) # 计算每个点的侵蚀 for y in range(height_value): for x in range(width): current_height = heightfield.getRed(x, y) # 检查周围点 for dx in [-1, 0, 1]: for dy in [-1, 0, 1]: if dx == 0 and dy == 0: continue nx, ny = x + dx, y + dy if 0 <= nx < width and 0 <= ny < height_value: neighbor_height = heightfield.getRed(nx, ny) height_diff = current_height - neighbor_height # 如果高度差超过临界角度，发生滑坡 if height_diff > talus_angle: # 移动部分物质到邻居 move_amount = (height_diff - talus_angle) * 0.1 height_changes[x, y] -= move_amount height_changes[nx, ny] += move_amount * 0.5 # 部分沉积 # 应用高度变化 for y in range(height_value): for x in range(width): if height_changes[x, y] != 0: current_height = heightfield.getRed(x, y) new_height = current_height + height_changes[x, y] new_height = max(0.0, min(1.0, new_height)) heightfield.setRed(x, y, new_height) heightfield.setGreen(x, y, new_height) heightfield.setBlue(x, y, new_height) # 更新地形 return self._update_terrain(terrain_info) except Exception as e: print(f"生成热侵蚀地形时出错: {e}") return False def _update_terrain(self, terrain_info): """ 更新地形 """ try: terrain = terrain_info['terrain'] terrain.setHeightfield(terrain_info['heightfield']) terrain.generate() # 重新创建碰撞体 self._recreate_collision(terrain_info['node']) return True except Exception as e: print(f"更新地形时出错: {e}") return False 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}")