// Copyright (2019-2023) Paul Scherrer Institute

#ifndef JUNGFRAUJOCH_PCIEXPRESSDEVICE_H
#define JUNGFRAUJOCH_PCIEXPRESSDEVICE_H

#include "FPGAAcquisitionDevice.h"
#include "../fpga/host_library/JungfraujochDevice.h"

class PCIExpressDevice : public FPGAAcquisitionDevice {
    JungfraujochDevice dev;

    bool HW_ReadMailbox(uint32_t values[16]) override;
    bool HW_SendWorkRequest(uint32_t handle) override;
    void FPGA_StartAction(const DiffractionExperiment &experiment) override;
    bool HW_IsIdle() const final;
    void HW_WriteActionRegister(const DataCollectionConfig *job) override;
    void HW_ReadActionRegister(DataCollectionConfig *job) override;
    void HW_LoadCalibration(uint32_t modules, uint32_t storage_cells) override;
    void HW_LoadIntegrationMap(uint32_t modules) override;
    void HW_LoadInternalGeneratorFrame(uint32_t modules) override;
    void HW_SetSpotFinderParameters(const SpotFinderParameters &params) override;
    void FPGA_EndAction() override;
public:
    explicit PCIExpressDevice(uint16_t data_stream);
    PCIExpressDevice(uint16_t data_stream, uint16_t pci_slot);
    PCIExpressDevice(const std::string &device_name, uint16_t data_stream);

    void Cancel() override;
    void HW_GetStatus(DataCollectionStatus *status) const override;
    void HW_GetEnvParams(DeviceStatus *status) const override;
    uint32_t GetNumKernelBuffers() const;

    int32_t GetNUMANode() const override;

    void SetMACAddress(uint64_t mac_addr_network_order);
    void SetDefaultMAC();
    void SetIPv4Address(uint32_t ipv4_addr_network_order);

    std::string GetMACAddress() const override;
    std::string GetIPv4Address() const override;
    uint32_t GetCompletedDescriptors() const override;
};

#endif //JUNGFRAUJOCH_PCIEXPRESSDEVICE_H