// Copyright (c) 2013 - 2026 kio@little-bat.de
// BSD-2-Clause license
// https://opensource.org/licenses/BSD-2-Clause

#pragma once
#include "Templates/Array.h"

namespace zxsp
{

extern uint16 crc16(const uint8* q, uint count);

enum {
	sf_id_crc_wrong	  = 1 << 0,
	sf_data_crc_wrong = 1 << 1,
	sf_idam_notfound  = 1 << 2,
	sf_dam_notfound	  = 1 << 3,
	sf_ddam			  = 1 << 4
};

struct SectorFormatInfo
{
	uint8  side_id;	  // normally: 0/1 for side 0/1
	uint8  track_id;  // normally: starting at 0
	uint8  sector_id; // starting at 1 (DOS), 0x41 (CP/M)
	uint8  sector_sz; // log2(sector_length) - 7
	uint8  gap3;	  // gap *before* this sectors; GAP1=50; GAP3=54 (DOS)
	uint8  databyte;  // if data==NULL
	uint8  special;
	uint8* data;

	uint totalsize() { return gap3 + 12 + 10 + 22 + 12 + 4 + (0x80 << sector_sz) + 2; }
};

using TrackFormatInfo = Array<SectorFormatInfo>;


struct DiskFormatInfo
{
	uint sides;
	uint tracks;
	uint sectors;
	uint side0;
	uint track0;
	uint sector0;
	uint sector_sz; // log2(SZ)-7
	uint databyte;
	uint gap4a; // 80 (FDC765) or 92 (TMS279X)
	uint gap3;
	uint interleaved; // 1 3 5 7 9 2 4 6 8
					  // uint	head_skew;
					  // uint	track_skew;
};


class FloppyDisk
{
public:
	uint		  bytes_per_track;
	bool		  writeprotected;
	bool		  modified;
	cstr		  filepath;
	Array<uint8*> sides[2];

public:
	explicit FloppyDisk(uint bytes_per_track = 6250); // 6250: mfm, 300rpm
	FloppyDisk(uint sides, uint tracks, uint sectors, bool interleaved, uint bytes_per_track = 6250);
	explicit FloppyDisk(cstr filepath);
	~FloppyDisk();

	void makeSingleSided();
	void truncateTracks(uint n);
	void saveAs(cstr path);
	void saveDisk();
	bool fileIsWritable();
	bool isModified() { return modified; }
	bool isWriteProtected() { return writeprotected; }
	cstr getFilepath() { return filepath; }
	uint getBytesPerTrack() { return bytes_per_track; }
	int	 setWriteProtected(bool);
	void setFilepath(cstr);

	void   formatDisk(const DiskFormatInfo&);
	void   writeTrack(Array<SectorFormatInfo>&, uint8*& track, uint gap4a);
	uint8* writeSector(SectorFormatInfo&, uint8*);

	uint8*& getTrack(uint head, uint track)
	{
		sides[head].grow(track + 1);
		return sides[head][track];
	} // may be NULL
	uint8 readByte(uint head, uint track, uint bytepos);
	void  writeByte(uint head, uint track, uint bytepos, uint8 byte);

private:
	cstr read_dsk_file(uint8* bu, uint32 sz);
	cstr read_extended_dsk_file(uint8* bu, uint32 sz);
	void parse_disk(Array<Array<SectorFormatInfo>> trackinfo[]); // !!! ObjArray
	void parse_track(uint8* track, TrackFormatInfo& sectorinfo) const;
	void write_extended_disk_file(FD& fd) const;
};

} // namespace zxsp