first commit

cpp/data_recorder.h

@@ -0,0 +1,79 @@
+#pragma once
+
+#include <thread>
+#include <semaphore.h>
+#include <atomic>
+#include <stdint.h>
+#include <complex>
+#include <string>
+#include <queue>
+
+#define UTIL_REG_BASE 0x100000
+#define TIMING_REG_BASE 0x110000
+#define DATA_GEN_REG_BASE 0x120000
+
+#define WSRDMA_DMA_INIT             0
+#define WSRDMA_DMA_CLEAR            1
+#define WSRDMA_DMA_START            2
+#define WSRDMA_DMA_STOP             3
+#define WSRDMA_SET_NUM_BUFS         4
+#define WSRDMA_SET_NUM_BYTES        5
+#define WSRDMA_GET_NUM_BUFS         6
+#define WSRDMA_GET_NUM_BYTES        7
+#define WSRDMA_GET_FREE_BUFS        8
+
+typedef struct {
+   unsigned int cmd;
+   unsigned int offset;
+   unsigned int value;
+} wsrpcie_ioctl_t;
+
+#define NUM_DMA_CH      2
+#define BUFFER_SIZE     (4 * 1024 * 1024)
+// #define BUFFER_SIZE     (8192)
+#define NUM_BUFFERS     128
+#define OVERALL_BUFFER_SIZE (BUFFER_SIZE * NUM_BUFFERS)
+
+#define CLK_RATE 250e6
+
+class DataRecorder {
+    public:
+        DataRecorder();
+        ~DataRecorder();
+        
+        int start_recording(const char* filename, int save_to_disk);
+        int stop_recording();
+        std::vector<float> get_recording_rate();
+        std::vector<uint64_t> get_filesize();
+        void set_validate_cnt_data(bool enable, uint32_t pri, uint32_t inter, uint32_t count);
+        void write_reg(uint32_t addr, uint32_t data);
+        uint32_t read_reg(uint32_t addr);        
+
+        bool recording_active;
+
+    private:
+        void allocate_memory();
+        void get_data(int ch_ind, int save_to_disk);
+        void write_data(int ch_ind);
+
+        int plfd;
+        volatile uint32_t * plmmap;
+
+        std::thread recorder[NUM_DMA_CH];
+        std::thread writer[NUM_DMA_CH];
+        std::queue<int> buffer_queue[NUM_DMA_CH];
+        sem_t buffer_ready_sem[NUM_DMA_CH];
+        int num_bytes_to_write[NUM_DMA_CH][NUM_BUFFERS];
+        int out_fd[NUM_DMA_CH];
+        char * data_buffer[NUM_DMA_CH];
+        std::atomic<bool> exit_thread;
+
+        float recording_rate[NUM_DMA_CH];
+        uint64_t total_bytes[NUM_DMA_CH];
+
+        bool validate_cnt_data;
+        uint32_t cnt_pri;
+        uint32_t cnt_inter_pri;
+        uint32_t cnt_num_pulse;
+
+};