#pragma once
/*#
    # fdd.h

    Floppy-disc drive helper functions.

    Do this:
    ~~~C
    #define CHIPS_IMPL
    ~~~
    before you include this file in *one* C or C++ file to create the 
    implementation.

    Optionally provide the following macros with your own implementation
    
    ~~~C
    CHIPS_ASSERT(c)
    ~~~
        your own assert macro (default: assert(c))

    FIXME: DOCS

    ## Modifications for bsnes-plus SFX-DOS emulation
        - Modified values for maximum sector count and size

    ## zlib/libpng license

    Copyright (c) 2018 Andre Weissflog
    This software is provided 'as-is', without any express or implied warranty.
    In no event will the authors be held liable for any damages arising from the
    use of this software.
    Permission is granted to anyone to use this software for any purpose,
    including commercial applications, and to alter it and redistribute it
    freely, subject to the following restrictions:
        1. The origin of this software must not be misrepresented; you must not
        claim that you wrote the original software. If you use this software in a
        product, an acknowledgment in the product documentation would be
        appreciated but is not required.
        2. Altered source versions must be plainly marked as such, and must not
        be misrepresented as being the original software.
        3. This notice may not be removed or altered from any source
        distribution. 
#*/
#include <stdint.h>
#include <stdbool.h>

#ifdef __cplusplus
extern "C" {
#endif

#define FDD_MAX_SIDES (2)           /* max number of disc sides */
#define FDD_MAX_TRACKS (80)         /* max number of tracks per side */
#define FDD_MAX_SECTORS (9)         /* max sectors per track */
#define FDD_MAX_SECTOR_SIZE (1024)  /* max size of a sector in bytes */
#define FDD_MAX_TRACK_SIZE (FDD_MAX_SECTORS*FDD_MAX_SECTOR_SIZE)
#define FDD_MAX_DISC_SIZE (FDD_MAX_SIDES*FDD_MAX_TRACKS*FDD_MAX_TRACK_SIZE)

/* result bits (compatible with UPD765_RESULT_*) */
#define FDD_RESULT_SUCCESS (0)
#define FDD_RESULT_NOT_READY (1<<0)
#define FDD_RESULT_NOT_FOUND (1<<1)
#define FDD_RESULT_END_OF_SECTOR (1<<2)

/* UPD765 disc controller overlay of the sector info bytes */
typedef struct fdd_upd765_sectorinfo_t {
    uint8_t c;      /* cylinder number (track number) */
    uint8_t h;      /* head address (side) */
    uint8_t r;      /* record (sector number) */
    uint8_t n;      /* number (sector size) */
    uint8_t st1;    /* ST1 status register result */
    uint8_t st2;    /* ST2 status register result */
} fdd_upd765_sectorinfo_t;

/* a sector description */
typedef struct fdd_sector_t {
    union {
        fdd_upd765_sectorinfo_t upd765;
        uint8_t raw[8];
    } info;
    int data_offset;    /* start of sector data in disc data blob */
    int data_size;      /* size in bytes of sector data drive data buffer */
} fdd_sector_t;

/* a track description */
typedef struct fdd_track_t {
    int data_offset;    /* offset of track data in disc data blob */
    int data_size;      /* track data size in bytes */
    int num_sectors;    /* number of sectors in track */
    fdd_sector_t sectors[FDD_MAX_SECTORS];  /* the sector descriptions */
} fdd_track_t;

/* a disc description */
typedef struct fdd_disc_t {
    bool formatted;         /* disc is formatted */
    bool write_protected;   /* disc is write protected */
    int num_sides;
    int num_tracks;
    fdd_track_t tracks[FDD_MAX_SIDES][FDD_MAX_TRACKS];
} fdd_disc_t;

/* a floppy disc drive description */
typedef struct fdd_t {
    int cur_side;
    int cur_track_index;
    int cur_sector_index;
    int cur_sector_pos;
    bool has_disc;
    bool motor_on;
    fdd_disc_t disc;
    int data_size;
    uint8_t data[FDD_MAX_DISC_SIZE];
} fdd_t;

/* initialize a floppy disc drive */
void fdd_init(fdd_t* fdd);
/* drive motor on/off */
void fdd_motor(fdd_t* fdd, bool on);
/* insert a disc, the disc structure and data will be copied */
bool fdd_insert_disc(fdd_t* fdd, const fdd_disc_t* disc, const uint8_t* data, int data_size);
/* eject current disc */
void fdd_eject_disc(fdd_t* fdd);
/* seek to physical track (happens instantly), returns FDD_RESULT_* */
int fdd_seek_track(fdd_t* fdd, int track);
/* seek to sector on current physical track (happens instantly), returns FDD_RESULT_* */
int fdd_seek_sector(fdd_t* fdd, uint8_t c, uint8_t h, uint8_t r, uint8_t n);
/* read the next byte from the seeked-to sector, return FDD_RESULT_* */
int fdd_read(fdd_t* fdd, uint8_t h, uint8_t* out_data);
/* write the next byte from the seeked-to sector, return FDD_RESULT_* */
int fdd_write(fdd_t* fdd, uint8_t h, uint8_t in_data);

#ifdef __cplusplus
} /* extern "C" */
#endif

/*--- IMPLEMENTATION ---------------------------------------------------------*/
#ifdef CHIPS_IMPL
#include <string.h>
#ifndef CHIPS_ASSERT
    #include <assert.h>
    #define CHIPS_ASSERT(c) assert(c)
#endif

void fdd_init(fdd_t* fdd) {
    CHIPS_ASSERT(fdd);
    memset(fdd, 0, sizeof(fdd_t));
}

void fdd_motor(fdd_t* fdd, bool on) {
    CHIPS_ASSERT(fdd);
    fdd->motor_on = on;
}

void fdd_eject_disc(fdd_t* fdd) {
    CHIPS_ASSERT(fdd);
    fdd->cur_side = 0;
    fdd->cur_track_index = 0;
    fdd->cur_sector_index = 0;
    fdd->cur_sector_pos = 0;
    fdd->has_disc = false;
    fdd->motor_on = false;
    memset(&fdd->disc, 0, sizeof(fdd->disc));
    memset(&fdd->data, 0, sizeof(fdd->data));
}

bool _fdd_validate_disc(const fdd_disc_t* disc) {
    CHIPS_ASSERT(disc);
    if ((disc->num_sides < 0) || (disc->num_sides > FDD_MAX_SIDES)) {
        return false;
    }
    if ((disc->num_tracks < 0) || (disc->num_tracks > FDD_MAX_TRACKS)) {
        return false;
    }
    for (int side_index = 0; side_index < disc->num_sides; side_index++) {
        for (int track_index = 0; track_index < disc->num_tracks; track_index++) {
            const fdd_track_t* track = &(disc->tracks[side_index][track_index]);
            if (track->data_offset < 0) {
                return false;
            }
            if ((track->data_size < 0) || (track->data_size > FDD_MAX_TRACK_SIZE)) {
                return false;
            }
            if ((track->data_offset + track->data_size) > FDD_MAX_DISC_SIZE) {
                return false;
            }
            if ((track->num_sectors < 0) || (track->num_sectors > FDD_MAX_SECTORS)) {
                return false;
            }
            for (int sector_index = 0; sector_index < track->num_sectors; sector_index++) {
                const fdd_sector_t* sector = &(track->sectors[sector_index]);
                if (sector->data_offset < 0) {
                    return false;
                }
                if ((sector->data_size < 0) || (sector->data_size > FDD_MAX_SECTOR_SIZE)) {
                    return false;
                }
                if ((sector->data_offset + sector->data_size) > FDD_MAX_DISC_SIZE) {
                    return false;
                }
            }
        }
    }
    return true;
}

bool fdd_insert_disc(fdd_t* fdd, const fdd_disc_t* disc, const uint8_t* data, int data_size) {
    CHIPS_ASSERT(fdd);
    if (fdd->has_disc) {
        fdd_eject_disc(fdd);
    }
    if (_fdd_validate_disc(disc)) {
        fdd->disc = *disc;
    }
    else {
        /* invalid disc structure */
        return false;
    }
    if (data) {
        if ((data_size > 0) && (data_size <= FDD_MAX_DISC_SIZE)) {
            fdd->data_size = data_size;
            memcpy(&fdd->data, data, data_size);
            fdd->disc.formatted = true;
        }
        else {
            /* invalid data size */
            return false;
        }
    }
    else {
        fdd->disc.formatted = false;
    }
    fdd->has_disc = true;
    return true;
}

int fdd_seek_track(fdd_t* fdd, int track) {
    CHIPS_ASSERT(fdd);
    if (fdd->has_disc && fdd->motor_on && (track < fdd->disc.num_tracks)) {
        fdd->cur_track_index = track;
        return FDD_RESULT_SUCCESS;
    }
    else {
        return FDD_RESULT_NOT_READY;
    }
}

int fdd_seek_sector(fdd_t* fdd, uint8_t c, uint8_t h, uint8_t r, uint8_t n) {
    CHIPS_ASSERT(fdd);
    CHIPS_ASSERT(h < FDD_MAX_SIDES);
    (void)c; // FIXME (?)
    (void)n; // FIXME (?)
    if (fdd->has_disc && fdd->motor_on) {
        fdd->cur_side = h;
        const fdd_track_t* track = &fdd->disc.tracks[h][fdd->cur_track_index];
        for (int si = 0; si < track->num_sectors; si++) {
            const fdd_sector_t* sector = &track->sectors[si];
            if (sector->info.upd765.r == r) {
                fdd->cur_sector_index = si;
                fdd->cur_sector_pos = 0;
                return FDD_RESULT_SUCCESS;
            }
        }
        return FDD_RESULT_NOT_FOUND;
    }
    else {
        return FDD_RESULT_NOT_READY;
    }
}

int fdd_read(fdd_t* fdd, uint8_t h, uint8_t* out_data) {
    CHIPS_ASSERT(fdd && (h < FDD_MAX_SIDES) && out_data);
    if (fdd->has_disc & fdd->motor_on) {
        fdd->cur_side = h;
        const fdd_sector_t* sector = &fdd->disc.tracks[h][fdd->cur_track_index].sectors[fdd->cur_sector_index];
        if (fdd->cur_sector_pos < sector->data_size) {
            const int data_offset = sector->data_offset + fdd->cur_sector_pos;
            *out_data = fdd->data[data_offset];
            fdd->cur_sector_pos++;
            if (fdd->cur_sector_pos < sector->data_size) {
                return FDD_RESULT_SUCCESS;
            }
            else {
                return FDD_RESULT_END_OF_SECTOR;
            }
        }
        return FDD_RESULT_NOT_FOUND;
    }
    *out_data = 0xFF;
    return FDD_RESULT_NOT_READY;
}

int fdd_write(fdd_t* fdd, uint8_t h, uint8_t in_data) {
    CHIPS_ASSERT(fdd && (h < FDD_MAX_SIDES) && in_data);
    if (fdd->has_disc & fdd->motor_on) {
        fdd->cur_side = h;
        const fdd_sector_t* sector = &fdd->disc.tracks[h][fdd->cur_track_index].sectors[fdd->cur_sector_index];
        if (fdd->cur_sector_pos < sector->data_size) {
            const int data_offset = sector->data_offset + fdd->cur_sector_pos;
            fdd->data[data_offset] = in_data;
            fdd->cur_sector_pos++;
            if (fdd->cur_sector_pos < sector->data_size) {
                return FDD_RESULT_SUCCESS;
            }
            else {
                return FDD_RESULT_END_OF_SECTOR;
            }
        }
        return FDD_RESULT_NOT_FOUND;
    }
    return FDD_RESULT_NOT_READY;
}

#endif /* CHIPS_IMPL */