Extra2D/src/scene/scene_manager.cpp

#include <algorithm>
#include <app/application.h>
#include <graphics/render_backend.h>
#include <graphics/render_command.h>
#include <platform/input.h>
#include <scene/scene_manager.h>
#include <scene/transition_box_scene.h>
#include <scene/transition_fade_scene.h>
#include <scene/transition_flip_scene.h>
#include <scene/transition_scale_scene.h>
#include <scene/transition_scene.h>
#include <scene/transition_slide_scene.h>
#include <utils/logger.h>

namespace extra2d {

namespace {

Node *hitTestTopmost(const Ptr<Node> &node, const Vec2 &worldPos) {
  if (!node || !node->visible()) {
    return nullptr;
  }

  std::vector<Ptr<Node>> children = node->children();
  std::stable_sort(children.begin(), children.end(),
                   [](const Ptr<Node> &a, const Ptr<Node> &b) {
                     return a->zOrder() < b->zOrder();
                   });

  for (auto it = children.rbegin(); it != children.rend(); ++it) {
    if (Node *hit = hitTestTopmost(*it, worldPos)) {
      return hit;
    }
  }

  if (node->getEventDispatcher().getTotalListenerCount() == 0) {
    return nullptr;
  }

  Rect bounds = node->getBoundingBox();
  if (!bounds.empty() && bounds.containsPoint(worldPos)) {
    return node.get();
  }

  return nullptr;
}

void dispatchToNode(Node *node, Event &event) {
  if (!node) {
    return;
  }
  node->getEventDispatcher().dispatch(event);
}

} // namespace

SceneManager &SceneManager::getInstance() {
  static SceneManager instance;
  return instance;
}

void SceneManager::runWithScene(Ptr<Scene> scene) {
  if (!scene) {
    return;
  }

  if (!sceneStack_.empty()) {
    E2D_LOG_WARN("SceneManager: runWithScene should only be called once");
    return;
  }

  scene->onEnter();
  scene->onAttachToScene(scene.get());
  sceneStack_.push(scene);
}

void SceneManager::replaceScene(Ptr<Scene> scene) {
  if (!scene || isTransitioning_) {
    return;
  }

  if (sceneStack_.empty()) {
    runWithScene(scene);
    return;
  }

  // Pop current scene
  auto oldScene = sceneStack_.top();
  oldScene->onExit();
  oldScene->onDetachFromScene();
  sceneStack_.pop();

  // Push new scene
  scene->onEnter();
  scene->onAttachToScene(scene.get());
  sceneStack_.push(scene);
}

void SceneManager::enterScene(Ptr<Scene> scene) {
  if (!scene || isTransitioning_) {
    return;
  }

  if (sceneStack_.empty()) {
    runWithScene(scene);
  } else {
    replaceScene(scene);
  }
}

void SceneManager::enterScene(Ptr<Scene> scene,
                              Ptr<TransitionScene> transitionScene) {
  if (!scene || isTransitioning_) {
    return;
  }

  // 如果没有过渡场景，使用无过渡切换
  if (!transitionScene) {
    enterScene(scene);
    return;
  }

  auto current = getCurrentScene();
  if (!current) {
    enterScene(scene);
    return;
  }

  // 在过渡开始前，发送 UIHoverExit 给当前悬停的节点，重置按钮状态
  if (hoverTarget_) {
    Event evt;
    evt.type = EventType::UIHoverExit;
    evt.data = CustomEvent{0, hoverTarget_};
    dispatchToNode(hoverTarget_, evt);
    hoverTarget_ = nullptr;
  }
  captureTarget_ = nullptr;
  hasLastPointerWorld_ = false;

  // 设置过渡场景
  transitionScene->setOutScene(current);
  transitionScene->setFinishCallback([this]() { finishTransition(); });

  // 暂停当前场景
  current->pause();

  // 推入过渡场景（作为中介场景）
  transitionScene->onEnter();
  transitionScene->onAttachToScene(transitionScene.get());
  sceneStack_.push(transitionScene);

  isTransitioning_ = true;
  activeTransitionScene_ = transitionScene;
  transitionStackAction_ = [this, transitionScene]() {
    // 退出旧场景
    auto outScene = transitionScene->getOutScene();
    if (!sceneStack_.empty() && outScene) {
      // 过渡场景已经在栈顶，弹出它
      if (sceneStack_.top().get() == transitionScene.get()) {
        sceneStack_.pop();
      }
      outScene->onExit();
      outScene->onDetachFromScene();
    }

    // 推入新场景
    auto inScene = transitionScene->getInScene();
    if (inScene) {
      inScene->onAttachToScene(inScene.get());
      sceneStack_.push(inScene);
    }
  };
}

void SceneManager::replaceScene(Ptr<Scene> scene, TransitionType transition,
                                float duration) {
  if (!scene || isTransitioning_) {
    return;
  }

  if (sceneStack_.empty()) {
    runWithScene(scene);
    return;
  }

  auto oldScene = sceneStack_.top();

  startTransition(oldScene, scene, transition, duration, [this]() {
    // 过渡完成后，退出旧场景并从堆栈中移除
    if (!sceneStack_.empty() && activeTransitionScene_) {
      // 弹出过渡场景
      if (sceneStack_.top().get() == activeTransitionScene_.get()) {
        sceneStack_.pop();
      }

      // 退出旧场景
      auto outScene = activeTransitionScene_->getOutScene();
      if (outScene) {
        outScene->onExit();
        outScene->onDetachFromScene();
      }
    }

    // 将新场景推入堆栈
    if (activeTransitionScene_) {
      auto inScene = activeTransitionScene_->getInScene();
      if (inScene) {
        inScene->onAttachToScene(inScene.get());
        sceneStack_.push(inScene);
      }
    }
  });
}

void SceneManager::pushScene(Ptr<Scene> scene) {
  if (!scene || isTransitioning_) {
    return;
  }

  // Pause current scene
  if (!sceneStack_.empty()) {
    sceneStack_.top()->pause();
  }

  // Push new scene
  scene->onEnter();
  scene->onAttachToScene(scene.get());
  sceneStack_.push(scene);
}

void SceneManager::pushScene(Ptr<Scene> scene, TransitionType transition,
                             float duration) {
  if (!scene || isTransitioning_) {
    return;
  }

  if (sceneStack_.empty()) {
    runWithScene(scene);
    return;
  }

  // 暂停当前场景
  sceneStack_.top()->pause();

  auto currentScene = sceneStack_.top();

  startTransition(currentScene, scene, transition, duration, [this]() {
    // 过渡完成后，将新场景推入堆栈
    if (!sceneStack_.empty() && activeTransitionScene_) {
      // 弹出过渡场景
      if (sceneStack_.top().get() == activeTransitionScene_.get()) {
        sceneStack_.pop();
      }
    }

    if (activeTransitionScene_) {
      auto inScene = activeTransitionScene_->getInScene();
      if (inScene) {
        inScene->onAttachToScene(inScene.get());
        sceneStack_.push(inScene);
      }
    }
  });
}

void SceneManager::popScene() {
  if (sceneStack_.size() <= 1 || isTransitioning_) {
    return;
  }

  auto current = sceneStack_.top();
  current->onExit();
  current->onDetachFromScene();
  sceneStack_.pop();

  // Resume previous scene
  if (!sceneStack_.empty()) {
    sceneStack_.top()->resume();
  }
}

void SceneManager::popScene(TransitionType transition, float duration) {
  if (sceneStack_.size() <= 1 || isTransitioning_) {
    return;
  }

  auto current = sceneStack_.top();
  auto previous = getPreviousScene();

  startTransition(current, previous, transition, duration, [this]() {
    // 过渡完成后，退出当前场景并从堆栈中移除
    if (!sceneStack_.empty() && activeTransitionScene_) {
      // 弹出过渡场景
      if (sceneStack_.top().get() == activeTransitionScene_.get()) {
        sceneStack_.pop();
      }

      // 退出当前场景
      auto outScene = activeTransitionScene_->getOutScene();
      if (outScene) {
        outScene->onExit();
        outScene->onDetachFromScene();
      }
    }

    // 恢复前一个场景
    if (activeTransitionScene_) {
      auto inScene = activeTransitionScene_->getInScene();
      if (inScene && !sceneStack_.empty() && sceneStack_.top() == inScene) {
        inScene->resume();
      }
    }
  });
}

void SceneManager::popToRootScene() {
  if (sceneStack_.size() <= 1 || isTransitioning_) {
    return;
  }

  // Exit all scenes except root
  while (sceneStack_.size() > 1) {
    auto scene = sceneStack_.top();
    scene->onExit();
    scene->onDetachFromScene();
    sceneStack_.pop();
  }

  // Resume root
  sceneStack_.top()->resume();
}

void SceneManager::popToRootScene(TransitionType transition, float duration) {
  if (sceneStack_.size() <= 1 || isTransitioning_) {
    return;
  }

  auto root = getRootScene();
  auto current = sceneStack_.top();

  startTransition(current, root, transition, duration, [this, root]() {
    // 退出所有场景直到根场景
    while (!sceneStack_.empty() && sceneStack_.top().get() != root.get()) {
      auto scene = sceneStack_.top();
      scene->onExit();
      scene->onDetachFromScene();
      sceneStack_.pop();
    }

    // 恢复根场景
    if (!sceneStack_.empty() && sceneStack_.top().get() == root.get()) {
      root->resume();
    }
  });
}

void SceneManager::popToScene(const std::string &name) {
  if (isTransitioning_) {
    return;
  }

  // Find target scene in stack
  std::stack<Ptr<Scene>> tempStack;
  Ptr<Scene> target = nullptr;

  while (!sceneStack_.empty()) {
    auto scene = sceneStack_.top();
    if (scene->name() == name) {
      target = scene;
      break;
    }
    scene->onExit();
    scene->onDetachFromScene();
    sceneStack_.pop();
  }

  if (target) {
    target->resume();
  }
}

void SceneManager::popToScene(const std::string &name,
                              TransitionType transition, float duration) {
  if (isTransitioning_) {
    return;
  }

  auto target = getSceneByName(name);
  if (target && target != sceneStack_.top()) {
    auto current = sceneStack_.top();

    startTransition(current, target, transition, duration, [this, target]() {
      // 退出所有场景直到目标场景
      while (!sceneStack_.empty() && sceneStack_.top() != target) {
        auto scene = sceneStack_.top();
        scene->onExit();
        scene->onDetachFromScene();
        sceneStack_.pop();
      }

      // 恢复目标场景
      if (!sceneStack_.empty() && sceneStack_.top() == target) {
        target->resume();
      }
    });
  }
}

Ptr<Scene> SceneManager::getCurrentScene() const {
  if (sceneStack_.empty()) {
    return nullptr;
  }
  return sceneStack_.top();
}

Ptr<Scene> SceneManager::getPreviousScene() const {
  if (sceneStack_.size() < 2) {
    return nullptr;
  }

  // Copy stack to access second top
  auto tempStack = sceneStack_;
  tempStack.pop();
  return tempStack.top();
}

Ptr<Scene> SceneManager::getRootScene() const {
  if (sceneStack_.empty()) {
    return nullptr;
  }

  // Copy stack to access bottom
  auto tempStack = sceneStack_;
  Ptr<Scene> root;
  while (!tempStack.empty()) {
    root = tempStack.top();
    tempStack.pop();
  }
  return root;
}

Ptr<Scene> SceneManager::getSceneByName(const std::string &name) const {
  auto it = namedScenes_.find(name);
  if (it != namedScenes_.end()) {
    return it->second;
  }

  // Search in stack
  auto tempStack = sceneStack_;
  while (!tempStack.empty()) {
    auto scene = tempStack.top();
    if (scene->name() == name) {
      return scene;
    }
    tempStack.pop();
  }

  return nullptr;
}

bool SceneManager::hasScene(const std::string &name) const {
  return getSceneByName(name) != nullptr;
}

void SceneManager::update(float dt) {
  if (isTransitioning_) {
    // 过渡场景在栈顶，正常更新即可
    hoverTarget_ = nullptr;
    captureTarget_ = nullptr;
    hasLastPointerWorld_ = false;
  }

  if (!sceneStack_.empty()) {
    auto &scene = *sceneStack_.top();
    scene.updateScene(dt);
    if (!isTransitioning_) {
      dispatchPointerEvents(scene);
    }
  }
}

void SceneManager::render(RenderBackend &renderer) {
  Color clearColor = Colors::Black;
  if (!sceneStack_.empty()) {
    clearColor = sceneStack_.top()->getBackgroundColor();
  }

  E2D_LOG_TRACE("SceneManager::render - beginFrame with color({}, {}, {})",
                clearColor.r, clearColor.g, clearColor.b);
  renderer.beginFrame(clearColor);

  if (!sceneStack_.empty()) {
    E2D_LOG_TRACE("SceneManager::render - rendering scene content");
    sceneStack_.top()->renderContent(renderer);
  } else {
    E2D_LOG_WARN("SceneManager::render - no scene to render");
  }

  renderer.endFrame();
  E2D_LOG_TRACE("SceneManager::render - endFrame");
}

void SceneManager::collectRenderCommands(std::vector<RenderCommand> &commands) {
  if (!sceneStack_.empty()) {
    sceneStack_.top()->collectRenderCommands(commands, 0);
  }
}

void SceneManager::end() {
  while (!sceneStack_.empty()) {
    auto scene = sceneStack_.top();
    scene->onExit();
    scene->onDetachFromScene();
    sceneStack_.pop();
  }
  namedScenes_.clear();
}

void SceneManager::purgeCachedScenes() { namedScenes_.clear(); }

void SceneManager::startTransition(Ptr<Scene> from, Ptr<Scene> to,
                                   TransitionType type, float duration,
                                   Function<void()> stackAction) {
  if (!from || !to) {
    return;
  }

  // 创建过渡场景
  auto transitionScene = createTransitionScene(type, duration, to);
  if (!transitionScene) {
    // 回退到无过渡切换
    replaceScene(to);
    return;
  }

  // 在过渡开始前，发送 UIHoverExit 给当前悬停的节点，重置按钮状态
  if (hoverTarget_) {
    Event evt;
    evt.type = EventType::UIHoverExit;
    evt.data = CustomEvent{0, hoverTarget_};
    dispatchToNode(hoverTarget_, evt);
    hoverTarget_ = nullptr;
  }
  captureTarget_ = nullptr;
  hasLastPointerWorld_ = false;

  // 设置过渡场景
  transitionScene->setOutScene(from);
  transitionScene->setFinishCallback([this]() { finishTransition(); });

  // 暂停当前场景
  from->pause();

  // 推入过渡场景（作为中介场景）
  transitionScene->onEnter();
  transitionScene->onAttachToScene(transitionScene.get());
  sceneStack_.push(transitionScene);

  isTransitioning_ = true;
  currentTransition_ = type;
  activeTransitionScene_ = transitionScene;
  transitionStackAction_ = std::move(stackAction);
}

Ptr<TransitionScene> SceneManager::createTransitionScene(TransitionType type,
                                                         float duration,
                                                         Ptr<Scene> inScene) {
  if (!inScene) {
    return nullptr;
  }

  switch (type) {
  case TransitionType::Fade:
    return TransitionFadeScene::create(duration, inScene);
  case TransitionType::SlideLeft:
    return TransitionSlideScene::create(duration, inScene,
                                        TransitionDirection::Left);
  case TransitionType::SlideRight:
    return TransitionSlideScene::create(duration, inScene,
                                        TransitionDirection::Right);
  case TransitionType::SlideUp:
    return TransitionSlideScene::create(duration, inScene,
                                        TransitionDirection::Up);
  case TransitionType::SlideDown:
    return TransitionSlideScene::create(duration, inScene,
                                        TransitionDirection::Down);
  case TransitionType::Scale:
    return TransitionScaleScene::create(duration, inScene);
  case TransitionType::Flip:
    return TransitionFlipScene::create(duration, inScene);
  case TransitionType::Box:
    return TransitionBoxScene::create(duration, inScene);
  default:
    // 默认使用淡入淡出
    return TransitionFadeScene::create(duration, inScene);
  }
}

void SceneManager::finishTransition() {
  // 先保存当前悬停的节点，然后在 transitionStackAction_ 之后发送 UIHoverExit
  Node *lastHoverTarget = hoverTarget_;

  isTransitioning_ = false;
  hoverTarget_ = nullptr;
  captureTarget_ = nullptr;
  hasLastPointerWorld_ = false;

  if (transitionStackAction_) {
    transitionStackAction_();
  }

  // 在 transitionStackAction_ 之后发送 UIHoverExit，确保旧场景仍然有效
  if (lastHoverTarget) {
    Event evt;
    evt.type = EventType::UIHoverExit;
    evt.data = CustomEvent{0, lastHoverTarget};
    dispatchToNode(lastHoverTarget, evt);
  }

  activeTransitionScene_.reset();
  transitionStackAction_ = nullptr;

  if (transitionCallback_) {
    transitionCallback_();
    transitionCallback_ = nullptr;
  }
}

void SceneManager::dispatchPointerEvents(Scene &scene) {
  auto &input = Application::instance().input();
  Vec2 screenPos = input.getMousePosition();

  Vec2 worldPos = screenPos;
  if (auto *camera = scene.getActiveCamera()) {
    worldPos = camera->screenToWorld(screenPos);
  }

  Ptr<Node> root = scene.shared_from_this();
  Node *newHover = hitTestTopmost(root, worldPos);

  if (newHover != hoverTarget_) {
    if (hoverTarget_) {
      Event evt;
      evt.type = EventType::UIHoverExit;
      evt.data = CustomEvent{0, hoverTarget_};
      dispatchToNode(hoverTarget_, evt);
    }
    hoverTarget_ = newHover;
    if (hoverTarget_) {
      Event evt;
      evt.type = EventType::UIHoverEnter;
      evt.data = CustomEvent{0, hoverTarget_};
      dispatchToNode(hoverTarget_, evt);
    }
  }

  if (!hasLastPointerWorld_) {
    lastPointerWorld_ = worldPos;
    hasLastPointerWorld_ = true;
  }

  Vec2 delta = worldPos - lastPointerWorld_;
  if (hoverTarget_ && (delta.x != 0.0f || delta.y != 0.0f)) {
    Event evt = Event::createMouseMove(worldPos, delta);
    dispatchToNode(hoverTarget_, evt);
  }

  float scrollDelta = input.getMouseScrollDelta();
  if (hoverTarget_ && scrollDelta != 0.0f) {
    Event evt = Event::createMouseScroll(Vec2(0.0f, scrollDelta), worldPos);
    dispatchToNode(hoverTarget_, evt);
  }

  if (input.isMousePressed(MouseButton::Left)) {
    captureTarget_ = hoverTarget_;
    if (captureTarget_) {
      Event evt = Event::createMouseButtonPress(
          static_cast<int>(MouseButton::Left), 0, worldPos);
      dispatchToNode(captureTarget_, evt);

      Event pressed;
      pressed.type = EventType::UIPressed;
      pressed.data = CustomEvent{0, captureTarget_};
      dispatchToNode(captureTarget_, pressed);
    }
  }

  if (input.isMouseReleased(MouseButton::Left)) {
    Node *target = captureTarget_ ? captureTarget_ : hoverTarget_;
    if (target) {
      Event evt = Event::createMouseButtonRelease(
          static_cast<int>(MouseButton::Left), 0, worldPos);
      dispatchToNode(target, evt);

      Event released;
      released.type = EventType::UIReleased;
      released.data = CustomEvent{0, target};
      dispatchToNode(target, released);
    }

    if (captureTarget_ && captureTarget_ == hoverTarget_) {
      Event clicked;
      clicked.type = EventType::UIClicked;
      clicked.data = CustomEvent{0, captureTarget_};
      dispatchToNode(captureTarget_, clicked);
    }

    captureTarget_ = nullptr;
  }

  lastPointerWorld_ = worldPos;
}

} // namespace extra2d