//
// Created by Hanno Gödecke on 29.02.24.
//

#pragma once

#include "RNFFilamentAssetWrapper.h"
#include "RNFFilamentInstanceWrapper.h"
#include "jsi/RNFHybridObject.h"
#include <gltfio/Animator.h>
#include <utils/NameComponentManager.h>

#include <map>

namespace margelo {

using namespace filament::gltfio;
using EntityNameMap = std::map<std::string, Entity>;

class AnimatorWrapper : public HybridObject {
public:
  explicit AnimatorWrapper(Animator* animator, FilamentInstance* instance, std::shared_ptr<NameComponentManager> nameComponentManager)
      : HybridObject("AnimatorWrapper"), _animator(animator), _instance(instance), _nameComponentManager(nameComponentManager),
        _entityMap(createEntityNameMap(instance)) {}

  void loadHybridMethods() override;

private: // Exposed JS API
  void applyAnimation(int animationIndex, double time);
  void applyCrossFade(int previousAnimationIndex, double previousAnimationTime, double alpha);
  void updateBoneMatrices();
  void resetBoneMatrices();
  int getAnimationCount();
  double getAnimationDuration(int animationIndex);
  std::string getAnimationName(int animationIndex);
  int addToSyncList(std::shared_ptr<FilamentInstanceWrapper> instanceWrapper);
  void removeFromSyncList(int instanceId);

private: // Internal
  /**
   * Creates a map of entities and their names
   */
  EntityNameMap createEntityNameMap(FilamentInstance* instance);
  /**
   * Will take all entities associates with this animator instance and apply it to the provided entities
   */
  void applyAnimationTo(EntityNameMap entitiesNameMap);
  /**
   * Loops through all instances in the sync list and applies the current transform of all entities to the instance
   * using `applyAnimationTo`, then updates the bone matrices of the instance using `instanceAnimator->updateBoneMatrices`.
   */
  void syncInstances();

  TransformManager& getTransformManager();

protected:
  std::mutex _mutex;
  Animator* _animator;
  FilamentInstance* _instance;
  std::shared_ptr<NameComponentManager> _nameComponentManager;
  // The entity map of this class's FilamentInstance
  EntityNameMap _entityMap;
  int _syncId = 0;
  std::map<int, FilamentInstance*> _syncMap;
  std::map<int, EntityNameMap> _instanceEntityMap;
  TransformManager& _transformManager = getTransformManager();

private:
  static auto constexpr TAG = "AnimatorWrapper";
};

} // namespace margelo