#pragma once
#include "IComponent.h"

enum class AnimationMode : uint
{
	Play,
	Stop,
	Pause
};

class AnimatorComponent final : public IComponent
{
public:
	AnimatorComponent(class Context* const context, class Actor* const actor, class Transform* const transform);
	~AnimatorComponent() = default;

	void Initialize() override;
	void Update() override;
	void Destroy() override;

	const AnimationMode& GetAnimationMode() const { return animation_mode; }
	void SetAnimationMode(const AnimationMode& mode) { this->animation_mode = mode; }

	const uint& GetCurrentFrameNumber() const { return current_frame_number; }
	void SetCurrentFrameNumber(const uint& number) { this->current_frame_number = number; }

	const std::shared_ptr<class Animation> GetCurrentAnimation();
	void SetCurrentAnimation(const std::string& animation_name);

	const std::map<std::string, std::shared_ptr<class Animation>>& GetAnimations() const { return animations; }
	const Keyframe* GetCurrentKeyframe() const;

	void AddAniamtion(const std::string& animation_name, const std::shared_ptr<class Animation>& animation);

	void Play();
	void Stop();
	void Pause();

	bool IsPlaying() const { return animation_mode == AnimationMode::Play; }

private:
	class Timer* timer = nullptr;
	AnimationMode animation_mode = AnimationMode::Play;
	uint current_frame_number = 0;
	float frame_counter = 0.0f;

	std::weak_ptr<class Animation> current_animation;
	std::map<std::string, std::shared_ptr<class Animation>> animations;
};

void AnimatorComponent::Update()
{
	if (current_animation.expired() || !IsPlaying())
	{
		return;
	}

	frame_counter += timer->GetDeltaTimeMS();

	if (frame_counter > GetCurrentKeyframe()->time)
	{
		current_frame_number++;
		
		switch (current_animation.lock()->GetRepeatType())
		{
		case RepeatType::Once:
			if (current_frame_number >= current_animation.lock()->GetKeyframeCount())
			{
				current_frame_number = current_animation.lock()->GetKeyframeCount() - 1;
				Pause();
			}
			break;
		case RepeatType::Loop:
			current_frame_number %= current_animation.lock()->GetKeyframeCount();
			break;
		}

		frame_counter = 0.0f;
	}
}

void AnimatorComponent::Play()
{
	animation_mode = AnimationMode::Play;

	frame_counter = 0.0f;
}

void AnimatorComponent::Stop()
{
	animation_mode = AnimationMode::Stop;

	current_frame_number = 0;
	frame_counter = 0.0f;
}

void AnimatorComponent::Pause()
{
	animation_mode = AnimationMode::Pause;
}