#pragma once

#include <cstdint>
#include <memory>
#include <string>
#include <vector>

#include "ZResource.h"

enum class TextureAnimationParamsType
{
	/* 0 */ SingleScroll,
	/* 1 */ DualScroll,
	/* 2 */ ColorChange,
	/* 3 */ ColorChangeLERP,
	/* 4 */ ColorChangeLagrange,
	/* 5 */ TextureCycle,
	/* 6 */ Empty  // An empty TextureAnimation has the form 00 00 00 06 00000000
};

class ZTextureAnimationParams : public ZResource
{
public:
	ZTextureAnimationParams(ZFile* parent);

	void ExtractFromBinary(uint32_t nRawDataIndex);
	virtual void ExtractFromBinary(uint32_t nRawDataIndex, int count);

	virtual std::string GetDefaultName(const std::string& prefix) const;
	ZResourceType GetResourceType() const;

	TextureAnimationParamsType type;
};

struct TextureScrollingParamsEntry
{
	int8_t xStep;
	int8_t yStep;
	uint8_t width;
	uint8_t height;
};

class TextureScrollingParams : public ZTextureAnimationParams
{
public:
	TextureScrollingParams(ZFile* parent);

	void ParseRawData() override;
	void ExtractFromBinary(uint32_t nRawDataIndex, int count) override;

	std::string GetSourceTypeName() const override;
	std::string GetDefaultName(const std::string& prefix) const override;
	size_t GetRawDataSize() const override;

	Declaration* DeclareVar(const std::string& prefix, const std::string& bodyStr) override;
	std::string GetBodySourceCode() const override;

	int count;                            // 1 for Single, 2 for Dual
	TextureScrollingParamsEntry rows[2];  // Too small to make a vector worth it
};

struct F3DPrimColor
{
	uint8_t r;
	uint8_t g;
	uint8_t b;
	uint8_t a;
	uint8_t lodFrac;
};

struct F3DEnvColor
{
	uint8_t r;
	uint8_t g;
	uint8_t b;
	uint8_t a;
};

class TextureColorChangingParams : public ZTextureAnimationParams
{
public:
	TextureColorChangingParams(ZFile* parent);

	void ParseRawData() override;

	std::string GetSourceTypeName() const override;
	std::string GetDefaultName(const std::string& prefix) const override;
	size_t GetRawDataSize() const override;

	void DeclareReferences(const std::string& prefix) override;

	std::string GetBodySourceCode() const override;

	uint16_t animLength;  // size of list for type 2
	uint16_t colorListCount;
	segptr_t primColorListAddress;
	segptr_t envColorListAddress;
	segptr_t frameDataListAddress;
	std::vector<F3DPrimColor> primColorList;
	std::vector<F3DEnvColor> envColorList;
	std::vector<uint16_t> frameDataList;
};

class TextureCyclingParams : public ZTextureAnimationParams
{
public:
	TextureCyclingParams(ZFile* parent);

	void ParseRawData() override;

	std::string GetSourceTypeName() const override;
	std::string GetDefaultName(const std::string& prefix) const override;
	size_t GetRawDataSize() const override;

	void DeclareReferences(const std::string& prefix) override;

	std::string GetBodySourceCode() const override;

	uint16_t cycleLength;
	segptr_t textureListAddress;
	segptr_t textureIndexListAddress;
	std::vector<segptr_t> textureList;
	std::vector<uint8_t> textureIndexList;
};

struct TextureAnimationEntry
{
	int8_t segment;
	TextureAnimationParamsType type;
	segptr_t paramsPtr;
};

class ZTextureAnimation : public ZResource
{
public:
	ZTextureAnimation(ZFile* nParent);

	void ParseRawData() override;
	void DeclareReferences(const std::string& prefix) override;

	std::string GetSourceTypeName() const override;
	ZResourceType GetResourceType() const override;
	size_t GetRawDataSize() const override;
	std::string GetDefaultName(const std::string& prefix) const override;

	Declaration* DeclareVar(const std::string& prefix, const std::string& bodyStr) override;
	std::string GetBodySourceCode() const override;

private:
	std::vector<TextureAnimationEntry> entries;
};