//---------------------------------------------------------------------------
//
//	X68000 EMULATOR "XM6"
//
//	Copyright (C) 2001-2006 ＰＩ．(ytanaka@ipc-tokai.or.jp)
//	Copyright (C) 2014-2020 GIMONS
//
//	XM6i
//	Copyright (C) 2010-2015 isaki@NetBSD.org
//
//  	Imported sava's Anex86/T98Next image and MO format support patch.
//  	Comments translated to english by akuker.
//
//	[ Disk ]
//
//---------------------------------------------------------------------------

#pragma once

#include "log.h"
#include "scsi.h"
#include "controllers/scsidev_ctrl.h"
#include "device.h"
#include "device_factory.h"
#include "disk_image/disk_image_handle_factory.h"
#include "file_support.h"
#include "filepath.h"
#include "interfaces/scsi_block_commands.h"
#include "mode_page_device.h"
#include <string>
#include <set>
#include <map>

using namespace std;

class Disk : public ModePageDevice, ScsiBlockCommands
{
private:
	enum access_mode { RW6, RW10, RW16 };

	// The supported configurable block sizes, empty if not configurable
	set<uint32_t> sector_sizes;
	uint32_t configured_sector_size;

	// The mapping of supported capacities to block sizes and block counts, empty if there is no capacity restriction
	map<uint64_t, Geometry> geometries;

	typedef struct {
		uint32_t size;							// Sector Size (8=256, 9=512, 10=1024, 11=2048, 12=4096)
		// TODO blocks should be a 64 bit value in order to support higher capacities
		uint32_t blocks;						// Total number of sectors
		DiskImageHandle *dcache;						// Disk cache
		off_t image_offset;						// Offset to actual data
		bool is_medium_changed;
	} disk_t;

	Dispatcher<Disk, SASIDEV> dispatcher;

public:
	Disk(const string&);
	virtual ~Disk();

	virtual bool Dispatch(SCSIDEV *) override;

	void MediumChanged();
	void ReserveFile(const string&);

	// Media Operations
	virtual void Open(const Filepath& path);
	void GetPath(Filepath& path) const;
	bool Eject(bool) override;

private:
	typedef ModePageDevice super;

	// Commands covered by the SCSI specification (see https://www.t10.org/drafts.htm)
	void StartStopUnit(SASIDEV *);
	void SendDiagnostic(SASIDEV *);
	void PreventAllowMediumRemoval(SASIDEV *);
	void SynchronizeCache10(SASIDEV *);
	void SynchronizeCache16(SASIDEV *);
	void ReadDefectData10(SASIDEV *);
	virtual void Read6(SASIDEV *);
	void Read10(SASIDEV *) override;
	void Read16(SASIDEV *) override;
	virtual void Write6(SASIDEV *);
	void Write10(SASIDEV *) override;
	void Write16(SASIDEV *) override;
	void ReadLong10(SASIDEV *);
	void ReadLong16(SASIDEV *);
	void WriteLong10(SASIDEV *);
	void WriteLong16(SASIDEV *);
	void Verify10(SASIDEV *);
	void Verify16(SASIDEV *);
	void Seek(SASIDEV *);
	void Seek10(SASIDEV *);
	void ReadCapacity10(SASIDEV *) override;
	void ReadCapacity16(SASIDEV *) override;
	void Reserve(SASIDEV *);
	void Release(SASIDEV *);

public:

	// Commands covered by the SCSI specification (see https://www.t10.org/drafts.htm)
	void Rezero(SASIDEV *);
	void FormatUnit(SASIDEV *) override;
	void ReassignBlocks(SASIDEV *);
	void Seek6(SASIDEV *);

	// Command helpers
	virtual int Inquiry(const DWORD *cdb, BYTE *buf) = 0;	// INQUIRY command
	virtual int WriteCheck(DWORD block);					// WRITE check
	virtual bool Write(const DWORD *cdb, const BYTE *buf, DWORD block);			// WRITE command
	bool StartStop(const DWORD *cdb);				// START STOP UNIT command
	bool SendDiag(const DWORD *cdb);				// SEND DIAGNOSTIC command

	virtual int Read(const DWORD *cdb, BYTE *buf, uint64_t block);
	int ReadDefectData10(const DWORD *, BYTE *, int);

	uint32_t GetSectorSizeInBytes() const;
	void SetSectorSizeInBytes(uint32_t, bool);
	uint32_t GetSectorSizeShiftCount() const;
	void SetSectorSizeShiftCount(uint32_t);
	bool IsSectorSizeConfigurable() const;
	set<uint32_t> GetSectorSizes() const;
	void SetSectorSizes(const set<uint32_t>&);
	uint32_t GetConfiguredSectorSize() const;
	bool SetConfiguredSectorSize(uint32_t);
	void SetGeometries(const map<uint64_t, Geometry>&);
	bool SetGeometryForCapacity(uint64_t);
	uint64_t GetBlockCount() const;
	void SetBlockCount(uint32_t);
	bool CheckBlockAddress(SASIDEV *, access_mode);
	bool GetStartAndCount(SASIDEV *, uint64_t&, uint32_t&, access_mode);
	void FlushCache();

protected:
	int ModeSense6(const DWORD *cdb, BYTE *buf);
	int ModeSense10(const DWORD *cdb, BYTE *buf, int);
	virtual void SetDeviceParameters(BYTE *);
	void AddModePages(map<int, vector<BYTE>>&, int, bool) const override;
	virtual void AddErrorPage(map<int, vector<BYTE>>&, bool) const;
	virtual void AddFormatPage(map<int, vector<BYTE>>&, bool) const;
	virtual void AddDrivePage(map<int, vector<BYTE>>&, bool) const;
	void AddCachePage(map<int, vector<BYTE>>&, bool) const;
	virtual void AddVendorPage(map<int, vector<BYTE>>&, int, bool) const;

	// Internal disk data
	disk_t disk;

private:
	void Read(SASIDEV *, uint64_t);
	void Write(SASIDEV *, uint64_t);
	void Verify(SASIDEV *, uint64_t);
	void ReadWriteLong10(SASIDEV *);
	void ReadWriteLong16(SASIDEV *);
	void ReadCapacity16_ReadLong16(SASIDEV *);
	bool Format(const DWORD *cdb);
};