#ifndef XSAEXTRACTOR_HH
#define XSAEXTRACTOR_HH

#include "DiskImageUtils.hh"

#include "MemBuffer.hh"

#include <array>
#include <cstdint>
#include <span>

namespace openmsx {

class XSAExtractor
{
public:
	explicit XSAExtractor(std::span<const uint8_t> file);
	MemBuffer<SectorBuffer> extractData() &&;

private:
	static constexpr int MAX_STR_LEN = 254;
	static constexpr int TBL_SIZE = 16;
	static constexpr int MAX_HUF_CNT = 127;

	[[nodiscard]] inline uint8_t charIn();
	void chkHeader();
	void unLz77();
	[[nodiscard]] unsigned rdStrLen();
	[[nodiscard]] int rdStrPos();
	[[nodiscard]] bool bitIn();
	void initHufInfo();
	void mkHufTbl();

	struct HufNode {
		HufNode* child1;
		HufNode* child2;
		int weight;
	};

private:
	std::span<const uint8_t> file; // the not-yet-consumed part of the file
	MemBuffer<SectorBuffer> output;

	int updHufCnt;
	std::array<int, TBL_SIZE + 1> cpDist;
	std::array<int, TBL_SIZE> tblSizes;
	std::array<HufNode, 2 * TBL_SIZE - 1> hufTbl;

	uint8_t bitFlg; // flag with the bits
	uint8_t bitCnt; // nb bits left

	static constexpr std::array<uint8_t, TBL_SIZE> cpdExt = { // Extra bits for distance codes
		  0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12
	};
};

} // namespace openmsx

#endif