/*
DingusPPC - The Experimental PowerPC Macintosh emulator
Copyright (C) 2018-23 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/>.
*/

/** IDE Channel (aka IDE port) emulation.

    One IDE channel is capable of controlling up to two IDE devices.
    This class handles device registration and passing of messages
    from and to the host.
 */

#include <devices/common/ata/atabasedevice.h>
#include <devices/common/ata/atadefs.h>
#include <devices/common/ata/idechannel.h>
#include <devices/common/hwcomponent.h>
#include <devices/deviceregistry.h>
#include <machines/machinebase.h>
#include <loguru.hpp>

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

using namespace ata_interface;

IdeChannel::IdeChannel(const std::string name)
{
    this->set_name(name);
    this->supports_types(HWCompType::IDE_BUS);

    this->device_stub = std::unique_ptr<AtaNullDevice>(new AtaNullDevice());

    this->devices[0] = this->device_stub.get();
    this->devices[1] = this->device_stub.get();
}

int IdeChannel::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(
        this->name == "IDE0" ? IntSrc::IDE0 : IntSrc::IDE1);

    return 0;
}

void IdeChannel::register_device(int id, AtaInterface* dev_obj) {
    if (id < 0 || id >= 2)
        ABORT_F("%s: invalid device ID", this->name.c_str());

    this->devices[id] = dev_obj;

    ((AtaBaseDevice*)dev_obj)->set_host(this);
}

uint32_t IdeChannel::read(const uint8_t reg_addr, const int size)
{
    if (reg_addr == TIME_CONFIG) {
        if (size != 4) {
            LOG_F(WARNING, "%s: non-DWORD read from the channel config", this->name.c_str());
        }
        return this->ch_config;
    } else {
        return this->devices[this->cur_dev]->read(reg_addr);
    }
}

void IdeChannel::write(const uint8_t reg_addr, const uint32_t val, const int size)
{
    if (reg_addr == TIME_CONFIG) {
        if (size != 4) {
            LOG_F(WARNING, "%s: non-DWORD write to the channel config", this->name.c_str());
        }
        this->ch_config = val;
    } else {
        // keep track of the currently selected device
        if (reg_addr == DEVICE_HEAD) {
            this->cur_dev = (val >> 4) & 1;
        }

        // redirect register writes to both devices
        for (auto& dev : this->devices) {
            dev->write(reg_addr, val);
        }
    }
}

static const DeviceDescription Ide0_Descriptor = {
    IdeChannel::create_first, {}, {}
};

static const DeviceDescription Ide1_Descriptor = {
    IdeChannel::create_second, {}, {}
};

REGISTER_DEVICE(Ide0, Ide0_Descriptor);
REGISTER_DEVICE(Ide1, Ide1_Descriptor);