pcie_interface.h

浏览该文件的文档.

//
// Created by 56585 on 2023/8/15.
//
 
#ifndef NSUKIT_PCIE_INTERFACE_H
#define NSUKIT_PCIE_INTERFACE_H
 
#include "base_itf.h"
#include "xdma_api.h"
 
 
namespace nsukit {
    class XDMAOperation_t {
    protected:
        std::map<int, int> openedBoard = {};
        std::mutex OpLock{};
 
    public:
        nsukitStatus_t open_xdma_board(int board);
 
        nsukitStatus_t close_xdma_board(int board);
    };
 
 
    class NSU_DLLEXPORT PCIECmdUItf : public I_BaseCmdUItf {
    protected:
        static unsigned int irqNum;
 
        nsuBoardNum_t pciBoard = 0;
        nsuRegAddr_t sentBase = 0;
        nsuRegAddr_t recvBase = 0;
        nsuRegAddr_t irqBase = 0;
        nsuRegAddr_t sentDownBase = 0;
        int pciTimeout = 0;
 
        double onceTimeout = 0;
        std::mutex OpLock{};
 
        nsukitStatus_t reset_irq();
 
        nsukitStatus_t per_recv();
 
    public:
        PCIECmdUItf() { mixin_ = new Mixin_NativeRegCmd(this); };
 
        ~PCIECmdUItf() override { delete mixin_; };
 
        nsukitStatus_t accept(nsuInitParam_t *param) override;
 
        nsukitStatus_t close() override;
 
        nsukitStatus_t set_timeout(float s) override;
 
        nsukitStatus_t write(nsuRegAddr_t addr, nsuRegValue_t value) override;
 
        nsukitStatus_t read(nsuRegAddr_t addr, nsuRegValue_t *buf) override;
 
        nsukitStatus_t send_bytes(nsuBytes_t &bytes) override;
 
        nsukitStatus_t send_bytes(nsuCharBuf_p bytes, nsuSize_t length) override;
 
        nsukitStatus_t recv_bytes(nsuSize_t size, nsuCharBuf_p buf) override;
    };
 
 
    class NSU_DLLEXPORT PCIEStreamUItf : public I_BaseStreamUItf {
    private:
        const uint8_t byteWidth = 4;
        const uint8_t maxChnl = 4;
    protected:
        struct StreamProcess {
            nsuStreamLen_t current = 0;
            nsuStreamLen_t total = 0;
        };
 
        nsuBoardNum_t pciBoard = 0;
 
        std::map<nsuMemory_p, StreamProcess> uploadProcess{};
 
        std::map<nsuMemory_p, StreamProcess> downloadProcess{};
    public:
        PCIEStreamUItf();
 
        ~PCIEStreamUItf() override = default;
 
        nsukitStatus_t accept(nsuInitParam_t *param) override;
 
        nsukitStatus_t close() override;
 
        nsukitStatus_t set_timeout(float s) override;
 
        nsuMemory_p alloc_buffer(nsuStreamLen_t length, nsuVoidBuf_p buf = nullptr) override;
 
        nsukitStatus_t free_buffer(nsuMemory_p fd) override;
 
        nsuVoidBuf_p get_buffer(nsuMemory_p fd, nsuStreamLen_t length) override;
 
        nsukitStatus_t
        open_send(nsuChnlNum_t chnl, nsuMemory_p fd, nsuStreamLen_t length, nsuStreamLen_t offset = 0) override;
 
        nsukitStatus_t
        open_recv(nsuChnlNum_t chnl, nsuMemory_p fd, nsuStreamLen_t length, nsuStreamLen_t offset = 0) override;
 
        nsukitStatus_t wait_stream(nsuMemory_p fd, float timeout = 1.) override;
 
        nsukitStatus_t break_stream(nsuMemory_p fd) override;
 
        nsukitStatus_t
        stream_recv(nsuChnlNum_t chnl, nsuMemory_p fd, nsuStreamLen_t length, nsuStreamLen_t offset = 0,
                    bool(*stop_event) () = nullptr, float timeout = 5., int flag = 1) override;
 
        nsukitStatus_t
        stream_send(nsuChnlNum_t chnl, nsuMemory_p fd, nsuStreamLen_t length, nsuStreamLen_t offset = 0,
                    bool(*stop_event) () = nullptr, float timeout = 5., int flag = 1) override;
    };
 
 
    class NSU_DLLEXPORT PCIERingUItf : public PCIEStreamUItf {
    private:
        const uint8_t byteWidth = 4;
        const uint8_t maxChnl = 4;
        std::map<nsuChnlNum_t, nsuSize_t> ring_chnl_size= {
                {0, 512 * 1024 * 1024},
                {1, 512 * 1024 * 1024},
                {2, 512 * 1024 * 1024},
                {3, 512 * 1024 * 1024},
        };
    protected:
        struct Memory {
            nsuMemory_p memory;
            std::string error_msg{};
            nsuStreamLen_t mem_size;
            unsigned int idx;
            nsuChnlNum_t chnl;
            HANDLE ring = nullptr;
            ThreadSafeEvent finish_event;
            std::mutex mtx;
        };
        std::map<unsigned int, Memory *> memory_dict{};
        unsigned int memory_index = 0;
 
        std::map<uint32_t, HANDLE> ringCache{
                {0, nullptr},
                {1, nullptr},
                {2, nullptr},
                {3, nullptr}
        };
    public:
        PCIERingUItf();
 
        ~PCIERingUItf() override = default;
 
        nsukitStatus_t accept(nsuInitParam_t *param) override;
 
        nsukitStatus_t close() override;
 
        nsuMemory_p alloc_buffer(nsuStreamLen_t length, nsuVoidBuf_p buf = nullptr) override;
 
        nsukitStatus_t free_buffer(nsuMemory_p fd) override;
 
        nsuVoidBuf_p get_buffer(nsuMemory_p fd, nsuStreamLen_t length) override;
 
        nsukitStatus_t
        open_recv(nsuChnlNum_t chnl, nsuMemory_p fd, nsuStreamLen_t length, nsuStreamLen_t offset = 0) override;
 
        nsukitStatus_t wait_stream(nsuMemory_p fd, float timeout = 1.) override;
 
        nsukitStatus_t break_stream(nsuMemory_p fd) override;
    };
}
#endif //NSUKIT_PCIE_INTERFACE_H