/*
DingusPPC - The Experimental PowerPC Macintosh emulator
Copyright (C) 2018-22 divingkatae and maximum
                      (theweirdo)     spatium

(Contact divingkatae#1017 or powermax#2286 on Discord for more info)

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

/** @file Sander-Wozniak Machine 3 (SWIM3) emulation. */

#include <core/timermanager.h>
#include <devices/deviceregistry.h>
#include <devices/common/hwinterrupt.h>
#include <devices/floppy/superdrive.h>
#include <devices/floppy/swim3.h>
#include <loguru.hpp>
#include <machines/machinebase.h>
#include <machines/machineproperties.h>

#include <cinttypes>
#include <memory>
#include <string>

using namespace Swim3;

Swim3Ctrl::Swim3Ctrl()
{
    this->name = "SWIM3";
    this->supported_types = HWCompType::FLOPPY_CTRL;

    this->setup_reg = 0;
    this->mode_reg  = 0;
    this->int_reg   = 0;
    this->int_flags = 0;
    this->int_mask  = 0;
    this->error     = 0;
    this->xfer_cnt  = 0;
    this->first_sec = 0xFF;

    // Attach virtual Superdrive to the internal drive connector
    // TODO: make SWIM3/drive wiring user selectable
    this->int_drive = std::unique_ptr<MacSuperdrive::MacSuperDrive>
        (new MacSuperdrive::MacSuperDrive());
}

int Swim3Ctrl::device_postinit()
{
    this->int_ctrl = dynamic_cast<InterruptCtrl*>(
        gMachineObj->get_comp_by_type(HWCompType::INT_CTRL));
    this->irq_id = this->int_ctrl->register_dev_int(IntSrc::SWIM3);

    // if a floppy image was given "insert" it into the virtual superdrive
    std::string fd_image_path = GET_STR_PROP("fdd_img");
    std::string fd_write_prot = GET_STR_PROP("fdd_wr_prot");
    if (!fd_image_path.empty()) {
        bool write_flag = false;

        if (!fd_write_prot.empty()) {
            if ((fd_write_prot.compare("ON") == 0) || (fd_write_prot.compare("on") == 0)) {
                write_flag = true;
            }
        }

        this->int_drive->insert_disk(fd_image_path, write_flag);
    }

    return 0;
};

uint8_t Swim3Ctrl::read(uint8_t reg_offset)
{
    uint8_t status_addr, status_val, old_int_flags, old_error;

    switch(reg_offset) {
    case Swim3Reg::Error:
        old_error = this->error;
        this->error = 0;
        return old_error;
    case Swim3Reg::Phase:
        return this->phase_lines;
    case Swim3Reg::Setup:
        return this->setup_reg;
    case Swim3Reg::Handshake_Mode1:
        if (this->mode_reg & 2) { // internal drive?
            status_addr = ((this->mode_reg & 0x20) >> 2) | (this->phase_lines & 7);
            status_val = this->int_drive->status(status_addr) & 1;

            // transfer status_val to both bit 2 (RDDATA) and bit 3 (SENSE)
            // because those signals seem to be historically wired together
            return ((status_val << 2) | (status_val << 3));
        }
        return 4;
    case Swim3Reg::Interrupt_Flags:
        old_int_flags = this->int_flags;
        this->int_flags = 0; // read from this register clears all flags
        update_irq();
        return old_int_flags;
    case Swim3Reg::Current_Track:
        return this->cur_track;
    case Swim3Reg::Current_Sector:
        return this->cur_sector;
    case Swim3Reg::Gap_Format:
        return this->format;
    case Swim3Reg::First_Sector:
        return this->first_sec;
    case Swim3Reg::Sectors_To_Xfer:
        return this->xfer_cnt;
    case Swim3Reg::Interrupt_Mask:
        return this->int_mask;
    default:
        LOG_F(INFO, "SWIM3: reading from 0x%X register", reg_offset);
    }
    return 0;
}

void Swim3Ctrl::write(uint8_t reg_offset, uint8_t value)
{
    uint8_t old_mode_reg;

    switch(reg_offset) {
    case Swim3Reg::Param_Data:
        this->pram = value;
        break;
    case Swim3Reg::Phase:
        this->phase_lines = value & 0xF;
        if (value & 8) {
            if (this->mode_reg & 2) { // internal drive?
                this->int_drive->command(
                    ((this->mode_reg & 0x20) >> 3) | (this->phase_lines & 3),
                    (value >> 2) & 1
                );
            }
        }
        break;
    case Swim3Reg::Setup:
        this->setup_reg = value;
        break;
    case Swim3Reg::Status_Mode0:
        // ones in value clear the corresponding bits in the mode register
        if ((this->mode_reg & value) & (SWIM3_GO | SWIM3_GO_STEP)) {
            if (value & SWIM3_GO_STEP) {
                stop_stepping();
            } else {
                stop_disk_access();
            }
        }
        this->mode_reg &= ~value;
        break;
    case Swim3Reg::Handshake_Mode1:
        // ones in value set the corresponding bits in the mode register
        if ((this->mode_reg ^ value) & (SWIM3_GO | SWIM3_GO_STEP)) {
            if (value & SWIM3_GO_STEP) {
                start_stepping();
            } else {
                start_disk_access();
            }
        }
        this->mode_reg |= value;
        break;
    case Swim3Reg::Step:
        this->step_count = value;
        break;
    case Swim3Reg::Gap_Format:
        this->gap_size = value;
        break;
    case Swim3Reg::First_Sector:
        this->first_sec = value;
        break;
    case Swim3Reg::Sectors_To_Xfer:
        this->xfer_cnt = value;
        break;
    case Swim3Reg::Interrupt_Mask:
        this->int_mask = value;
        break;
    default:
        LOG_F(INFO, "SWIM3: writing 0x%X to register 0x%X", value, reg_offset);
    }
}

void Swim3Ctrl::update_irq()
{
    if (this->mode_reg & SWIM3_INT_ENA) {
        uint8_t new_irq = !!(this->int_flags & this->int_mask);
        if (new_irq != this->irq) {
            this->irq = new_irq;
            this->int_ctrl->ack_int(this->irq_id, new_irq);
        }
    }
}

void Swim3Ctrl::do_step()
{
    if (this->mode_reg & SWIM3_GO_STEP && this->step_count) { // are we still stepping?
        // instruct the drive to perform single step in current direction
        this->int_drive->command(MacSuperdrive::CommandAddr::Do_Step, 0);
        if (--this->step_count == 0) {
            if (this->step_timer_id) {
                this->stop_stepping();
            }
            this->int_flags |= INT_STEP_DONE;
            update_irq();
        }
    }
}

void Swim3Ctrl::start_stepping()
{
    if (!this->step_count) {
        LOG_F(WARNING, "SWIM3: step_count is zero while go_step is active!");
        return;
    }

    if (this->mode_reg & SWIM3_GO_STEP || this->step_timer_id) {
        LOG_F(ERROR, "SWIM3: another stepping action is running!");
        return;
    }

    if (this->mode_reg & SWIM3_GO || this->access_timer_id) {
        LOG_F(ERROR, "SWIM3: stepping attempt while disk access is in progress!");
        return;
    }

    if ((((this->mode_reg & 0x20) >> 3) | (this->phase_lines & 3))
        != MacSuperdrive::CommandAddr::Do_Step) {
        LOG_F(WARNING, "SWIM3: invalid command address on the phase lines!");
        return;
    }

    this->mode_reg |= SWIM3_GO_STEP;

    // step count > 1 requires periodic task
    if (this->step_count > 1) {
        this->step_timer_id = TimerManager::get_instance()->add_cyclic_timer(
            USECS_TO_NSECS(80),
            [this]() {
                this->do_step();
            }
        );
    }

    // perform the first step immediately
    do_step();
}

void Swim3Ctrl::stop_stepping()
{
    // cancel stepping task
    if (this->step_timer_id) {
        TimerManager::get_instance()->cancel_timer(this->step_timer_id);
    }
    this->step_timer_id = 0;
    this->step_count = 0; // not sure this one is required
}

void Swim3Ctrl::start_disk_access()
{
    if (this->mode_reg & SWIM3_GO || this->access_timer_id) {
        LOG_F(ERROR, "SWIM3: another disk access is running!");
        return;
    }

    if (this->mode_reg & SWIM3_GO_STEP || this->step_timer_id) {
        LOG_F(ERROR, "SWIM3: disk access attempt while stepping is in progress!");
        return;
    }

    if (this->mode_reg & SWIM3_WR_MODE) {
        LOG_F(ERROR, "SWIM3: writing not implemented yet");
        return;
    }

    this->mode_reg |= SWIM3_GO;
    LOG_F(9, "SWIM3: disk access started!");

    if (this->first_sec == 0xFF) {
        // $FF means no sector to match ->
        // generate ID_read interrups as long as the GO bit is set
        this->int_drive->init_track_search(-1); // start at random sector
    } else {
        this->cur_sector = this->first_sec;
    }

    // HACK: figure out from bits in int_mask register which kind of disk access is requested
    if (this->int_mask & INT_ID_READ) { // read address header
        this->access_timer_id = TimerManager::get_instance()->add_cyclic_timer(
            static_cast<uint64_t>(this->int_drive->get_sector_delay() * NS_PER_SEC + 0.5f),
            [this]() {
                // get next sector's address field
                MacSuperdrive::SectorHdr addr = this->int_drive->next_sector_header();
                // set up the corresponding SWIM3 registers
                this->cur_track  = ((addr.side & 1) << 7) | (addr.track & 0x7F);
                this->cur_sector = 0x80 /* CRC/checksum valid */ | (addr.sector & 0x7F);
                this->format = addr.format;
                // generate ID_read interrupt
                this->int_flags |= INT_ID_READ;
                update_irq();
            }
        );
    } else { // otherwise, read sector data
        this->access_timer_id = TimerManager::get_instance()->add_cyclic_timer(
            static_cast<uint64_t>(this->int_drive->get_sector_delay() * NS_PER_SEC + 0.5f),
            [this]() {
                // transfer sector data over DMA
                this->dma_ch->push_data(this->int_drive->get_sector_data_ptr(this->cur_sector), 512);

                // get next address field
                MacSuperdrive::SectorHdr addr = this->int_drive->next_sector_header();
                // set up the corresponding SWIM3 registers
                this->cur_track  = ((addr.side & 1) << 7) | (addr.track & 0x7F);
                this->cur_sector = 0x80 /* CRC/checksum valid */ | (addr.sector & 0x7F);
                this->format = addr.format;

                if (--this->xfer_cnt == 0) {
                    this->stop_disk_access();
                    // generate sector_done interrupt
                    this->int_flags |= INT_SECT_DONE;
                    update_irq();
                }
            }
        );
    }
}

void Swim3Ctrl::stop_disk_access()
{
    // cancel disk access timer
    if (this->access_timer_id) {
        TimerManager::get_instance()->cancel_timer(this->access_timer_id);
    }
    this->access_timer_id = 0;
}

static const vector<string> WriteToggle = {"ON", "on", "OFF", "off"};

static const PropMap Swim3_Properties = {
    {"fdd_img",
        new StrProperty("")},
    {"fdd_wr_prot",
        new StrProperty("OFF", WriteToggle)}
};

static const DeviceDescription Swim3_Descriptor = {
    Swim3Ctrl::create, {}, Swim3_Properties
};

REGISTER_DEVICE(Swim3, Swim3_Descriptor);