#include "registerdevice.h"

#include "max2871.h"
#include "stw81200.h"
#include "ad9913.h"
#include "Device/device.h"

RegisterDevice *RegisterDevice::create(Device *dev, int number, QString partnumber, QString name)
{
    RegisterDevice *regdev = nullptr;
    if(partnumber == "MAX2871") {
        regdev = new MAX2871();
    } else if(partnumber == "STW81200") {
        regdev = new STW81200();
    } else if(partnumber == "AD9913") {
        regdev = new AD9913();
    }
    if(regdev) {
        regdev->dev = dev;
        regdev->number = number;
        regdev->partnumber = partnumber;
        regdev->name = name;

        // read initial register content
        Protocol::PacketInfo p;
        p.type = Protocol::PacketType::DirectRegisterRead;
        p.directRegRead.device = number;
        for(unsigned int i=0;i<regdev->regs.size();i++) {
            p.directRegRead.address = regdev->regs[i]->getAddress();
            dev->SendPacket(p);
        }
    }
    return regdev;
}

RegisterDevice::~RegisterDevice()
{
    delete widget;
}

void RegisterDevice::setRegister(int address, unsigned long value)
{
    for(auto reg : regs) {
        if(reg->getAddress() == address) {
            reg->setValue(value);
        }
    }
}

RegisterDevice::RegisterDevice()
{
    widget = new QWidget;
}

QString RegisterDevice::getName() const
{
    return name;
}

void RegisterDevice::addPossibleInputs(RegisterDevice *inputDevice)
{
    for(auto o : inputDevice->outputs) {
        possibleInputs[inputDevice->name+":"+o.first] = o.second;
    }
}

QString RegisterDevice::getPartnumber() const
{
    return partnumber;
}

QWidget *RegisterDevice::getWidget() const
{
    return widget;
}

void RegisterDevice::addRegister(Register *reg)
{
    regs.push_back(reg);
    QObject::connect(reg, &Register::valueChanged, [=]() {
        // transfer register content to device
        Protocol::PacketInfo p;
        p.type = Protocol::PacketType::DirectRegisterWrite;
        p.directRegWrite.device = number;
        p.directRegWrite.address = reg->getAddress();
        p.directRegWrite.data = reg->getValue();
        dev->SendPacket(p);
    });
}

nlohmann::json RegisterDevice::registersToJSON()
{
    nlohmann::json j;
    for(auto r : regs) {
        j[r->getAddress()] = r->getValue();
    }
    return j;
}

void RegisterDevice::registersFromJSON(nlohmann::json j)
{
    for(auto val : j.items()) {
        auto address = QString::fromStdString(val.key()).toInt();
        for(auto r : regs) {
            if(r->getAddress() == address) {
                r->setValue(val.value());
                break;
            }
        }
    }
}