//
//  CPCDSK.cpp
//  Clock Signal
//
//  Created by Thomas Harte on 05/08/2017.
//  Copyright © 2017 Thomas Harte. All rights reserved.
//

#include "CPCDSK.hpp"

#include "../../Encodings/MFM/Encoder.hpp"

using namespace Storage::Disk;

CPCDSK::CPCDSK(const char *file_name) :
	Storage::FileHolder(file_name), is_extended_(false) {
	if(!check_signature("MV - CPC", 8)) {
		is_extended_ = true;
		fseek(file_, 0, SEEK_SET);
		if(!check_signature("EXTENDED", 8))
			throw ErrorNotCPCDSK;
	}

	// Don't really care about about the creator; skip.
	fseek(file_, 0x30, SEEK_SET);
	head_position_count_ = fgetc(file_);
	head_count_ = fgetc(file_);

	// Used only for non-extended disks.
	long size_of_a_track;

	// Used only for extended disks.
	std::vector<size_t> track_sizes;

	if(is_extended_) {
		// Skip two unused bytes and grab the track size table.
		fseek(file_, 2, SEEK_CUR);
		for(int c = 0; c < head_position_count_ * head_count_; c++) {
			track_sizes.push_back(static_cast<size_t>(fgetc(file_) << 8));
		}
	} else {
		size_of_a_track = fgetc16le();
	}

	long file_offset = 0x100;
	for(size_t c = 0; c < static_cast<size_t>(head_position_count_ * head_count_); c++) {
		if(!is_extended_ || (track_sizes[c] > 0)) {
			// Skip the introductory text, 'Track-Info\r\n' and its unused bytes.
			fseek(file_, file_offset + 16, SEEK_SET);

			tracks_.emplace_back(new Track);
			Track *track = tracks_.back().get();

			// Track and side are stored, being a byte each.
			track->track = static_cast<uint8_t>(fgetc(file_));
			track->side = static_cast<uint8_t>(fgetc(file_));
			
			// If this is an extended disk image then John Elliott's extension provides some greater
			// data rate and encoding context. Otherwise the next two bytes have no defined meaning.
			if(is_extended_) {
				switch(fgetc(file_)) {
					default: track->data_rate = Track::DataRate::Unknown;				break;
					case 1: track->data_rate = Track::DataRate::SingleOrDoubleDensity;	break;
					case 2: track->data_rate = Track::DataRate::HighDensity;			break;
					case 3: track->data_rate = Track::DataRate::ExtendedDensity;		break;
				}
				switch(fgetc(file_)) {
					default: track->data_encoding = Track::DataEncoding::Unknown;		break;
					case 1: track->data_encoding = Track::DataEncoding::FM;				break;
					case 2: track->data_encoding = Track::Track::DataEncoding::MFM;		break;
				}
			} else {
				track->data_rate = Track::DataRate::Unknown;
				track->data_encoding = Track::DataEncoding::Unknown;
				fseek(file_, 2, SEEK_CUR);
			}
			
			// Sector size, number of sectors, gap 3 length and the filler byte are then common
			// between both variants of DSK.
			track->sector_length = static_cast<uint8_t>(fgetc(file_));
			size_t number_of_sectors = static_cast<size_t>(fgetc(file_));
			track->gap3_length = static_cast<uint8_t>(fgetc(file_));
			track->filler_byte = static_cast<uint8_t>(fgetc(file_));

			// Sector information begins immediately after the track information table.
			while(number_of_sectors--) {
				track->sectors.emplace_back();
				Track::Sector &sector = track->sectors.back();

				// Track, side, sector, size and two FDC8272-esque status bytes are stored
				// per sector, in both regular and extended DSK files.
				sector.track = static_cast<uint8_t>(fgetc(file_));
				sector.side = static_cast<uint8_t>(fgetc(file_));
				sector.sector = static_cast<uint8_t>(fgetc(file_));
				sector.size = static_cast<uint8_t>(fgetc(file_));
				sector.fdc_status1 = static_cast<uint8_t>(fgetc(file_));
				sector.fdc_status2 = static_cast<uint8_t>(fgetc(file_));

				// Figuring out the actual data size is a little more work...
				size_t data_size = static_cast<size_t>(128 << sector.size);
				size_t stored_data_size = data_size;
				size_t number_of_samplings = 1;

				if(is_extended_) {
					// In an extended DSK, oblige two Simon Owen extensions:
					//
					// Allow a declared data size less than the sector's declared size to act as an abbreviation.
					// Extended DSK varies the 8kb -> 0x1800 bytes special case by this means.
					//
					// Use a declared data size greater than the sector's declared size as a record that this
					// sector was weak or fuzzy and that multiple samplings are provided. If the greater size
					// is not an exact multiple then my reading of the documentation is that this is an invalid
					// disk image.
					size_t declared_data_size = fgetc16le();
					if(declared_data_size != stored_data_size) {
						if(declared_data_size > data_size) {
							number_of_samplings = declared_data_size / data_size;
							if(declared_data_size % data_size)
								throw ErrorNotCPCDSK;
						} else {
							stored_data_size = declared_data_size;
						}
					}
				} else {
					// In a regular DSK, these two bytes are unused, and a special case is applied that ostensibly 8kb
					// sectors are abbreviated to only 0x1800 bytes.
					if(data_size == 0x2000) stored_data_size = 0x1800;
					fseek(file_, 2, SEEK_CUR);
				}

				// As per the weak/fuzzy sector extension, multiple samplings may be stored here.
				// Plan to tead as many as there were.
				sector.data.resize(number_of_samplings);
				while(number_of_samplings--) {
					sector.data[number_of_samplings].resize(stored_data_size);
				}
			}
			
			// Sector contents are at offset 0x100 into the track.
			fseek(file_, file_offset + 0x100, SEEK_SET);
			for(auto &sector: track->sectors) {
				for(auto &data : sector.data) {
					fread(data.data(), 1, data.size(), file_);
				}
			}
		} else {
			// An extended disk image, which declares that there is no data stored for this track.
			tracks_.emplace_back();
		}

		// Advance to the beginning of the next track.
		if(is_extended_)
			file_offset += static_cast<long>(track_sizes[c]);
		else
			file_offset += size_of_a_track;
	}
}

int CPCDSK::get_head_position_count() {
	return head_position_count_;
}

int CPCDSK::get_head_count() {
	return head_count_;
}

std::shared_ptr<Track> CPCDSK::get_track_at_position(::Storage::Disk::Track::Address address) {
	// Given that thesea are interleaved images, determine which track, chronologically, is being requested.
	size_t chronological_track = static_cast<size_t>((address.position * head_count_) + address.head);

	// Return a nullptr if out of range or not provided.
	if(chronological_track >= tracks_.size()) return nullptr;
	
	Track *track = tracks_[chronological_track].get();
	if(!track) return nullptr;

	// Transcribe sectors and return.
	// TODO: is this transcription really necessary?
	std::vector<Storage::Encodings::MFM::Sector> sectors;
	for(auto &sector : track->sectors) {
		Storage::Encodings::MFM::Sector new_sector;
		new_sector.address.track = sector.track;
		new_sector.address.side = sector.side;
		new_sector.address.sector = sector.sector;
		new_sector.size = sector.size;

		// TODO: deal with weak/fuzzy sectors.
		new_sector.data.insert(new_sector.data.begin(), sector.data[0].begin(), sector.data[0].end());

		if(sector.fdc_status2 & 0x20) {
			// The CRC failed in the data field.
			new_sector.has_data_crc_error = true;
		} else {
			if(sector.fdc_status1 & 0x20) {
				// The CRC failed in the ID field.
				new_sector.has_header_crc_error = true;
			}
		}

		if(sector.fdc_status2 & 0x40) {
			// This sector is marked as deleted.
			new_sector.is_deleted = true;
		}

		if(sector.fdc_status2 & 0x01) {
			// Data field wasn't found.
			new_sector.data.clear();
		}

		sectors.push_back(std::move(new_sector));
	}

	// TODO: FM encoding, data rate?
	return Storage::Encodings::MFM::GetMFMTrackWithSectors(sectors, track->gap3_length, track->filler_byte);
}