gitting project in git

python/data_structures.py

@@ -0,0 +1,173 @@
+import ctypes
+from ctypes import Structure, c_uint64, c_uint32, c_uint16, c_uint8, c_float
+
+AXI_WRITE_REG = 1
+AXI_READ_REG = 2
+AXI_READ_RESP = 3
+ACK_MSG = 4
+NACK_MSG = 5
+AXI_WRITE_REG_BURST = 6
+RF_SPI_WRITE = 7
+SET_AD9081_DAC_NCO = 128
+SET_AD9081_ADC_NCO = 129
+
+ACK_FLAG_VALID_PACKET = 0x01
+ACK_FLAG_VALID_EXECUTION = 0x02
+ACK_FLAG_GOT_FRAME_END = 0x04
+
+MAX_BURST_LENGTH = 512
+
+HDR_FLAG_REQ_ACK = 0x01
+
+class CpiHeader(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("sync", c_uint32),
+        ("num_pulses", c_uint32),
+        ("num_samples", c_uint32),
+        ("start_sample", c_uint32),
+        ("tx_num_samples", c_uint32),
+        ("tx_start_sample", c_uint32),
+        ("pri", c_uint32),
+        ("inter_cpi", c_uint32),
+
+        # ("spare", c_uint32),  # Populated by FPGA
+        ("pps_sec", c_uint64),  # Populated by FPGA
+        ("pps_frac_sec", c_uint64),  # Populated by FPGA
+        ("system_time", c_uint64),  # Populated by FPGA
+
+        ("tx_lo_offset", c_float),
+        ("rx_lo_offset", c_float),
+        ("data1", c_uint32 * 240)  # Populated by user
+
+    ]
+
+
+class Header(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("fsync", c_uint32),
+        ("type", c_uint16),
+        ("flags", c_uint16),
+        ("length", c_uint16)
+    ]
+
+    # Have to put this here to get pycharm autocomplete to work, don't like this, but autocomplete is too convenient
+    def __init__(self):
+        self.fsync = 0xAABBCCDD
+        self.type = 0
+        self.flags = 0
+        self.length = 0
+
+
+def init_header(self, msg_id):
+    self.header = Header()
+    self.header.type = msg_id
+    self.header.length = ctypes.sizeof(self)
+
+
+class WriteRegType(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("header", Header),
+        ("address", c_uint32),
+        ("data", c_uint32)
+    ]
+
+    def __init__(self):
+        init_header(self, AXI_WRITE_REG)
+        self.address = 0
+        self.data = 0
+
+
+class WriteRegBurstType(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("header", Header),
+        ("address", c_uint32),
+        ("length", c_uint32),
+        ("data", c_uint32 * MAX_BURST_LENGTH)
+    ]
+
+    def __init__(self):
+        init_header(self, AXI_WRITE_REG_BURST)
+        self.address = 0
+        self.length = 0
+
+
+class AckType(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("header", Header),
+        ("flags", c_uint32)
+    ]
+
+    def __init__(self):
+        init_header(self, ACK_MSG)
+        self.flags = 0
+
+
+class ReadRequestType(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("header", Header),
+        ("address", c_uint32)
+    ]
+
+    def __init__(self):
+        init_header(self, AXI_READ_REG)
+        self.address = 0
+
+
+class ReadResponseType(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("header", Header),
+        ("data", c_uint32)
+    ]
+
+    def __init__(self):
+        init_header(self, AXI_READ_RESP)
+        self.data = 0
+
+
+class DacNcoConfigType(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("header", Header),
+        ("channel", c_uint32),
+        ("frequency", c_float)
+    ]
+
+    def __init__(self):
+        init_header(self, SET_AD9081_DAC_NCO)
+        self.channel = 0
+        self.frequency = 0
+
+class AdcNcoConfigType(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("header", Header),
+        ("channel", c_uint32),
+        ("frequency", c_float)
+    ]
+
+    def __init__(self):
+        init_header(self, SET_AD9081_ADC_NCO)
+        self.channel = 0
+        self.frequency = 0
+
+class RfSpiWriteType(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("header", Header),
+        ("dev_sel", c_uint32),
+        ("num_bits", c_uint32),
+        ("data", c_uint32)
+    ]
+
+    def __init__(self):
+        init_header(self, RF_SPI_WRITE)
+        self.dev_sel = 0
+        self.num_bits = 0
+        self.data = 0