#include "globals.h"
#include "flags.h"
#include "fluxmap.h"
#include "decoders/decoders.h"
#include "encoders/encoders.h"
//#include "arch/aeslanier/aeslanier.h"
#include "arch/amiga/amiga.h"
#include "arch/apple2/apple2.h"
#include "arch/brother/brother.h"
#include "arch/c64/c64.h"
#include "arch/f85/f85.h"
#include "arch/fb100/fb100.h"
#include "arch/ibm/ibm.h"
#include "arch/macintosh/macintosh.h"
#include "arch/micropolis/micropolis.h"
#include "arch/mx/mx.h"
#include "arch/northstar/northstar.h"
#include "arch/tids990/tids990.h"
#include "arch/victor9k/victor9k.h"
#include "arch/zilogmcz/zilogmcz.h"
#include "decoders/fluxmapreader.h"
#include "flux.h"
#include "protocol.h"
#include "decoders/rawbits.h"
#include "sector.h"
#include "image.h"
#include "lib/decoders/decoders.pb.h"
#include "fmt/format.h"
#include <numeric>

std::unique_ptr<AbstractDecoder> AbstractDecoder::create(const DecoderProto& config)
{
	static const std::map<int,
		std::function<std::unique_ptr<AbstractDecoder>(const DecoderProto&)>> decoders =
	{
		//{ DecoderProto::kAeslanier,  createAesLanierDecoder },
		{ DecoderProto::kAmiga,      createAmigaDecoder },
		{ DecoderProto::kApple2,     createApple2Decoder },
		{ DecoderProto::kBrother,    createBrotherDecoder },
		{ DecoderProto::kC64,        createCommodore64Decoder },
		{ DecoderProto::kF85,        createDurangoF85Decoder },
		{ DecoderProto::kFb100,      createFb100Decoder },
		{ DecoderProto::kIbm,        createIbmDecoder },
		{ DecoderProto::kMacintosh,  createMacintoshDecoder },
		{ DecoderProto::kMicropolis, createMicropolisDecoder },
		{ DecoderProto::kMx,         createMxDecoder },
		{ DecoderProto::kNorthstar,  createNorthstarDecoder },
		{ DecoderProto::kTids990,    createTids990Decoder },
		{ DecoderProto::kVictor9K,   createVictor9kDecoder },
		{ DecoderProto::kZilogmcz,   createZilogMczDecoder },
	};

	auto decoder = decoders.find(config.format_case());
	if (decoder == decoders.end())
		Error() << "no decoder specified";

	return (decoder->second)(config);
}

std::unique_ptr<TrackDataFlux> AbstractDecoder::decodeToSectors(
		std::shared_ptr<const Fluxmap> fluxmap, unsigned physicalCylinder, unsigned physicalHead)
{
	_trackdata = std::make_unique<TrackDataFlux>();
	_trackdata->fluxmap = fluxmap;
	_trackdata->physicalCylinder = physicalCylinder;
	_trackdata->physicalHead = physicalHead;
	
    FluxmapReader fmr(*fluxmap);
    _fmr = &fmr;

	auto newSector = [&] {
		_sector = std::make_shared<Sector>();
		_sector->status = Sector::MISSING;
		_sector->physicalCylinder = physicalCylinder;
		_sector->physicalHead = physicalHead;
	};

	newSector();
    beginTrack();
    for (;;)
    {
		newSector();

        Fluxmap::Position recordStart = fmr.tell();
        _sector->clock = advanceToNextRecord();
		if (fmr.eof() || !_sector->clock)
            return std::move(_trackdata);

        /* Read the sector record. */

		Fluxmap::Position before = fmr.tell();
        decodeSectorRecord();
		pushRecord(before, fmr.tell());

        if (_sector->status == Sector::DATA_MISSING)
        {
            /* The data is in a separate record. */

			for (;;)
			{
				_sector->clock = advanceToNextRecord();
				if (fmr.eof() || !_sector->clock)
					break;

				before = fmr.tell();
				decodeDataRecord();

				if (_sector->status != Sector::DATA_MISSING)
				{
					pushRecord(before, fmr.tell());
					break;
				}

				fmr.skipToEvent(F_BIT_PULSE);
				resetFluxDecoder();
			}
        }

        if (_sector->status != Sector::MISSING)
			_trackdata->sectors.push_back(_sector);
    }
}

void AbstractDecoder::pushRecord(const Fluxmap::Position& start, const Fluxmap::Position& end)
{
    Fluxmap::Position here = _fmr->tell();

	auto record = std::make_shared<Record>();
	_trackdata->records.push_back(record);
	_sector->records.push_back(record);
	
	record->startTime = start.ns();
	record->endTime = end.ns();
    record->clock = _sector->clock;

	record->rawData = toBytes(_recordBits);
	_recordBits.clear();
}

void AbstractDecoder::resetFluxDecoder()
{
	_decoder.reset(new FluxDecoder(_fmr, _sector->clock, _config));
}

nanoseconds_t AbstractDecoder::seekToPattern(const FluxMatcher& pattern)
{
	_fmr->skipToEvent(F_BIT_PULSE);
	nanoseconds_t clock = _fmr->seekToPattern(pattern);
	_decoder.reset(new FluxDecoder(_fmr, clock, _config));
	return clock;
}

void AbstractDecoder::seekToIndexMark()
{
	_fmr->skipToEvent(F_BIT_PULSE);
	_fmr->seekToIndexMark();
}

std::vector<bool> AbstractDecoder::readRawBits(unsigned count)
{
	auto bits = _decoder->readBits(count);
	_recordBits.insert(_recordBits.end(), bits.begin(), bits.end());
	return bits;
}

uint8_t AbstractDecoder::readRaw8()
{
	return toBytes(readRawBits(8)).reader().read_8();
}

uint16_t AbstractDecoder::readRaw16()
{
	return toBytes(readRawBits(16)).reader().read_be16();
}

uint32_t AbstractDecoder::readRaw20()
{
	std::vector<bool> bits(4);
	for (bool b : readRawBits(20))
		bits.push_back(b);

	return toBytes(bits).reader().read_be24();
}

uint32_t AbstractDecoder::readRaw24()
{
	return toBytes(readRawBits(24)).reader().read_be24();
}

uint32_t AbstractDecoder::readRaw32()
{
	return toBytes(readRawBits(32)).reader().read_be32();
}

uint64_t AbstractDecoder::readRaw48()
{
	return toBytes(readRawBits(48)).reader().read_be48();
}

uint64_t AbstractDecoder::readRaw64()
{
	return toBytes(readRawBits(64)).reader().read_be64();
}


std::set<unsigned> AbstractDecoder::requiredSectors(unsigned cylinder, unsigned head) const
{
	static std::set<unsigned> set;
	return set;
}