Extra2D/Extra2D/include/extra2d/resource/resource_manager.h

#pragma once

#include <extra2d/audio/sound.h>
#include <extra2d/core/types.h>
#include <extra2d/graphics/alpha_mask.h>
#include <extra2d/graphics/font.h>
#include <extra2d/graphics/texture.h>
#include <mutex>
#include <string>
#include <unordered_map>

namespace extra2d {

// ============================================================================
// 资源管理器 - 统一管理纹理、字体、音效等资源
// ============================================================================
class ResourceManager {
public:
  // ------------------------------------------------------------------------
  // 单例访问
  // ------------------------------------------------------------------------
  static ResourceManager &getInstance();

  // ------------------------------------------------------------------------
  // 纹理资源
  // ------------------------------------------------------------------------

  /// 加载纹理（带缓存）
  Ptr<Texture> loadTexture(const std::string &filepath);

  /// 加载纹理并生成Alpha遮罩（用于不规则形状图片）
  Ptr<Texture> loadTextureWithAlphaMask(const std::string &filepath);

  /// 通过key获取已缓存的纹理
  Ptr<Texture> getTexture(const std::string &key) const;

  /// 检查纹理是否已缓存
  bool hasTexture(const std::string &key) const;

  /// 卸载指定纹理
  void unloadTexture(const std::string &key);

  // ------------------------------------------------------------------------
  // Alpha遮罩资源
  // ------------------------------------------------------------------------

  /// 获取纹理的Alpha遮罩（如果已生成）
  const AlphaMask *getAlphaMask(const std::string &textureKey) const;

  /// 为已加载的纹理生成Alpha遮罩
  bool generateAlphaMask(const std::string &textureKey);

  /// 检查纹理是否有Alpha遮罩
  bool hasAlphaMask(const std::string &textureKey) const;

  // ------------------------------------------------------------------------
  // 字体图集资源
  // ------------------------------------------------------------------------

  /// 加载字体图集（带缓存）
  Ptr<FontAtlas> loadFont(const std::string &filepath, int fontSize,
                          bool useSDF = false);

  /// 通过key获取已缓存的字体图集
  Ptr<FontAtlas> getFont(const std::string &key) const;

  /// 检查字体是否已缓存
  bool hasFont(const std::string &key) const;

  /// 卸载指定字体
  void unloadFont(const std::string &key);

  // ------------------------------------------------------------------------
  // 音效资源
  // ------------------------------------------------------------------------

  /// 加载音效（带缓存）
  Ptr<Sound> loadSound(const std::string &filepath);
  Ptr<Sound> loadSound(const std::string &name, const std::string &filepath);

  /// 通过key获取已缓存的音效
  Ptr<Sound> getSound(const std::string &key) const;

  /// 检查音效是否已缓存
  bool hasSound(const std::string &key) const;

  /// 卸载指定音效
  void unloadSound(const std::string &key);

  // ------------------------------------------------------------------------
  // 缓存清理
  // ------------------------------------------------------------------------

  /// 清理所有失效的弱引用（自动清理已释放的资源）
  void purgeUnused();

  /// 清理指定类型的所有缓存
  void clearTextureCache();
  void clearFontCache();
  void clearSoundCache();

  /// 清理所有资源缓存
  void clearAllCaches();

  /// 获取各类资源的缓存数量
  size_t getTextureCacheSize() const;
  size_t getFontCacheSize() const;
  size_t getSoundCacheSize() const;

  ResourceManager();
  ~ResourceManager();
  ResourceManager(const ResourceManager &) = delete;
  ResourceManager &operator=(const ResourceManager &) = delete;

  // 生成字体缓存key
  std::string makeFontKey(const std::string &filepath, int fontSize,
                          bool useSDF) const;

  // 互斥锁保护缓存
  mutable std::mutex textureMutex_;
  mutable std::mutex fontMutex_;
  mutable std::mutex soundMutex_;

  // 资源缓存 - 使用弱指针实现自动清理
  std::unordered_map<std::string, WeakPtr<Texture>> textureCache_;
  std::unordered_map<std::string, WeakPtr<FontAtlas>> fontCache_;
  std::unordered_map<std::string, WeakPtr<Sound>> soundCache_;
};

} // namespace extra2d