""" 粒子系统插件主文件 """ from plugins.plugin_manager import BasePlugin from .core.particle_manager import ParticleManager from .editor.particle_editor import ParticleEditor from .tools.performance_monitor import PerformanceMonitor from .editor.preset_manager import PresetManager class Plugin(BasePlugin): """ 粒子系统插件 提供完整的3D粒子特效功能 """ def __init__(self, plugin_manager, name): super().__init__(plugin_manager, name) self.config = { "version": "1.0.0", "author": "EG Team", "description": "高性能3D粒子系统插件，支持火焰、烟雾、爆炸等多种特效" } # 核心组件 self.particle_manager = None self.particle_editor = None self.performance_monitor = None self.preset_manager = None # GUI元素 self.gui_elements = [] # 事件处理器 self.event_handlers = {} # 插件状态 self.is_initialized = False self.is_enabled = False def initialize(self) -> bool: """ 初始化插件 """ try: if self.is_initialized: return True print(f"初始化粒子系统插件: {self.name}") # 初始化粒子管理器 self.particle_manager = ParticleManager(self.plugin_manager.world) # 初始化性能监控器 self.performance_monitor = PerformanceMonitor(self.particle_manager) # 初始化粒子编辑器 self.particle_editor = ParticleEditor( self.plugin_manager.world, self.particle_manager ) # 初始化预设管理器 self.preset_manager = PresetManager() self.is_initialized = True print("✓ 粒子系统插件初始化完成") return True except Exception as e: print(f"✗ 粒子系统插件初始化失败: {e}") return False def enable(self) -> bool: """ 启用插件 """ if not super().enable(): return False if self.is_enabled: return True try: print(f"启用粒子系统插件: {self.name}") # 启用粒子管理器 if self.particle_manager: self.particle_manager.enable() # 注册事件处理器 self._register_event_handlers() # 创建GUI界面 self._create_gui() # 启动性能监控 if self.performance_monitor: self.performance_monitor.start() self.is_enabled = True print("✓ 粒子系统插件启用成功") return True except Exception as e: print(f"✗ 粒子系统插件启用失败: {e}") return False def disable(self) -> bool: """ 禁用插件 """ if not super().disable(): return False if not self.is_enabled: return True try: print(f"禁用粒子系统插件: {self.name}") # 停止性能监控 if self.performance_monitor: self.performance_monitor.stop() # 清理GUI界面 self._cleanup_gui() # 移除事件处理器 self._unregister_event_handlers() # 禁用粒子管理器 if self.particle_manager: self.particle_manager.disable() self.is_enabled = False print("✓ 粒子系统插件禁用成功") return True except Exception as e: print(f"✗ 粒子系统插件禁用失败: {e}") return False def finalize(self): """ 清理插件资源 """ print(f"清理粒子系统插件资源: {self.name}") # 清理所有组件 if self.particle_editor: self.particle_editor.cleanup() self.particle_editor = None if self.performance_monitor: self.performance_monitor.cleanup() self.performance_monitor = None if self.particle_manager: self.particle_manager.cleanup() self.particle_manager = None if self.preset_manager: self.preset_manager = None self.is_initialized = False self.is_enabled = False def _register_event_handlers(self): """ 注册事件处理器 """ world = self.plugin_manager.world # 粒子系统控制快捷键 self.event_handlers['f5'] = self.toggle_particle_editor self.event_handlers['f6'] = self.toggle_performance_monitor self.event_handlers['f7'] = self.create_test_particles self.event_handlers['f8'] = self.clear_all_particles self.event_handlers['f9'] = self.create_fire_effect self.event_handlers['f10'] = self.create_smoke_effect self.event_handlers['f11'] = self.create_explosion_effect self.event_handlers['f12'] = self.create_water_effect # 注册事件 for event, handler in self.event_handlers.items(): world.accept(event, handler) world.accept(event.upper(), handler) # 同时注册大写事件 def _unregister_event_handlers(self): """ 移除事件处理器 """ world = self.plugin_manager.world for event in self.event_handlers.keys(): world.ignore(event) world.ignore(event.upper()) self.event_handlers.clear() def _create_gui(self): """ 创建GUI界面 """ try: world = self.plugin_manager.world # 检查是否有GUI管理器 if not hasattr(world, 'gui_manager') or not world.gui_manager: print("⚠ GUI管理器不可用，跳过GUI创建") return gui_manager = world.gui_manager # 创建粒子系统控制面板 panel_button = gui_manager.createGUIButton( pos=(0.02, 0, 0.95), text="粒子系统", size=0.06 ) self.gui_elements.append(panel_button) # 创建快速效果按钮 effects = [ ("火焰 [F9]", self.create_fire_effect), ("烟雾 [F10]", self.create_smoke_effect), ("爆炸 [F11]", self.create_explosion_effect), ("水花 [F12]", self.create_water_effect), ("测试 [F7]", self.create_test_particles), ("清除 [F8]", self.clear_all_particles) ] for i, (name, callback) in enumerate(effects): button = gui_manager.createGUIButton( pos=(0.02, 0, 0.88 - i * 0.07), text=name, size=0.05 ) self.gui_elements.append(button) world.accept(f"particle_effect_{name}", callback) # 创建工具按钮 tools = [ ("编辑器 [F5]", self.toggle_particle_editor), ("性能监控 [F6]", self.toggle_performance_monitor) ] for i, (name, callback) in enumerate(tools): button = gui_manager.createGUIButton( pos=(0.02, 0, 0.40 - i * 0.07), text=name, size=0.05 ) self.gui_elements.append(button) except Exception as e: print(f"⚠ GUI创建失败: {e}") def _cleanup_gui(self): """ 清理GUI界面 """ try: world = self.plugin_manager.world if hasattr(world, 'gui_manager') and world.gui_manager: gui_manager = world.gui_manager # 删除所有GUI元素 for element in self.gui_elements: gui_manager.deleteGUIElement(element) self.gui_elements.clear() except Exception as e: print(f"⚠ GUI清理失败: {e}") # 事件处理方法 def toggle_particle_editor(self): """切换粒子编辑器显示""" if self.particle_editor: self.particle_editor.toggle_visibility() def toggle_performance_monitor(self): """切换性能监控器显示""" if self.performance_monitor: self.performance_monitor.toggle_display() def create_test_particles(self): """创建测试粒子""" if self.particle_manager: self.particle_manager.create_test_effect() def clear_all_particles(self): """清除所有粒子""" if self.particle_manager: self.particle_manager.clear_all_particles() def create_fire_effect(self): """创建火焰效果""" if self.particle_manager: self.particle_manager.create_fire_effect() def create_smoke_effect(self): """创建烟雾效果""" if self.particle_manager: self.particle_manager.create_smoke_effect() def create_explosion_effect(self): """创建爆炸效果""" if self.particle_manager: self.particle_manager.create_explosion_effect() def create_water_effect(self): """创建水花效果""" if self.particle_manager: self.particle_manager.create_water_effect() def create_magic_effect(self): """创建魔法效果""" if self.particle_manager: self.particle_manager.create_magic_effect() def create_snow_effect(self): """创建雪花效果""" if self.particle_manager: self.particle_manager.create_snow_effect() def get_info(self) -> dict: """ 获取插件信息 """ info = super().get_info() info.update({ "version": self.config.get("version", "1.0.0"), "author": self.config.get("author", "EG Team"), "description": self.config.get("description", "粒子系统插件"), "particle_count": self.particle_manager.get_particle_count() if self.particle_manager else 0, "active_emitters": self.particle_manager.get_emitter_count() if self.particle_manager else 0, "is_enabled": self.is_enabled, "is_initialized": self.is_initialized }) return info # 高级功能接口 def load_preset(self, preset_name: str): """加载预设效果""" if self.preset_manager and self.particle_manager: config = self.preset_manager.get_preset_config(preset_name) if config: # 在摄像机前方创建效果 camera_pos = self.plugin_manager.world.camera.getPos() camera_forward = self.plugin_manager.world.camera.getNetTransform().getMat().getRow3(1) position = camera_pos + camera_forward * 5 self.particle_manager.create_emitter( config.get('emitter_type', 'default'), position, config ) print(f"✓ 加载预设: {preset_name}") else: print(f"✗ 预设未找到: {preset_name}") def save_current_as_preset(self, preset_name: str, description: str = ""): """保存当前效果为预设""" if self.preset_manager and self.particle_editor: config = self.particle_editor.get_current_config() if config: success = self.preset_manager.save_preset( preset_name, config, description ) if success: print(f"✓ 保存预设: {preset_name}") else: print(f"✗ 保存预设失败: {preset_name}") else: print("✗ 没有可保存的配置") def list_presets(self): """列出所有预设""" if self.preset_manager: presets = self.preset_manager.list_presets() print("可用预设:") for preset in presets: print(f" - {preset}") return presets return [] def set_global_time_scale(self, scale: float): """设置全局时间缩放""" if self.particle_manager: self.particle_manager.set_global_time_scale(scale) def set_max_particles(self, max_count: int): """设置最大粒子数""" if self.particle_manager: self.particle_manager.set_max_particles(max_count) def set_emit_rate_multiplier(self, multiplier: float): """设置发射速率倍增器""" if self.particle_manager: self.particle_manager.set_emit_rate_multiplier(multiplier) def set_particle_size_multiplier(self, multiplier: float): """设置粒子尺寸倍增器""" if self.particle_manager: self.particle_manager.set_particle_size_multiplier(multiplier)