83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
|
import socket
|
||
|
|
from litescope.host.driver.reg import *
|
||
|
|
|
||
|
|
from liteeth.test.model.etherbone import *
|
||
|
|
|
||
|
|
class LiteScopeEtherboneDriver:
|
||
|
|
def __init__(self, ip_address, udp_port=20000, addrmap=None, busword=8, debug=False):
|
||
|
|
self.ip_address = ip_address
|
||
|
|
self.udp_port = udp_port
|
||
|
|
self.debug = debug
|
||
|
|
|
||
|
|
self.tx_sock = None
|
||
|
|
self.rx_sock = None
|
||
|
|
self.regs = build_map(addrmap, busword, self.read, self.write)
|
||
|
|
|
||
|
|
def open(self):
|
||
|
|
self.tx_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
|
self.rx_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
|
self.rx_sock.bind(("", self.udp_port))
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def read(self, addr, burst_length=1):
|
||
|
|
reads_addrs = [addr+4*j for j in range(burst_length)]
|
||
|
|
reads = EtherboneReads(base_ret_addr=0x1000, addrs=reads_addrs)
|
||
|
|
record = EtherboneRecord()
|
||
|
|
record.writes = None
|
||
|
|
record.reads = reads
|
||
|
|
record.bca = 0
|
||
|
|
record.rca = 0
|
||
|
|
record.rff = 0
|
||
|
|
record.cyc = 0
|
||
|
|
record.wca = 0
|
||
|
|
record.wff = 0
|
||
|
|
record.byte_enable = 0xf
|
||
|
|
record.wcount = 0
|
||
|
|
record.rcount = len(reads_addrs)
|
||
|
|
|
||
|
|
packet = EtherbonePacket()
|
||
|
|
packet.records = [record]
|
||
|
|
packet.encode()
|
||
|
|
self.tx_sock.sendto(bytes(packet), (self.ip_address, self.udp_port))
|
||
|
|
|
||
|
|
datas, addrs = self.rx_sock.recvfrom(8192)
|
||
|
|
packet = EtherbonePacket(datas)
|
||
|
|
packet.decode()
|
||
|
|
values = packet.records.pop().writes.get_datas()
|
||
|
|
if self.debug:
|
||
|
|
for i, val in enumerate(values):
|
||
|
|
print("RD %08X @ %08X" %(val, addr + 4*i))
|
||
|
|
if burst_length == 1:
|
||
|
|
return values[0]
|
||
|
|
else:
|
||
|
|
return values
|
||
|
|
|
||
|
|
def write(self, addr, datas):
|
||
|
|
if not isinstance(datas, list):
|
||
|
|
datas = [datas]
|
||
|
|
writes_datas = [d for d in datas]
|
||
|
|
writes = EtherboneWrites(base_addr=addr, datas=writes_datas)
|
||
|
|
record = EtherboneRecord()
|
||
|
|
record.writes = writes
|
||
|
|
record.reads = None
|
||
|
|
record.bca = 0
|
||
|
|
record.rca = 0
|
||
|
|
record.rff = 0
|
||
|
|
record.cyc = 0
|
||
|
|
record.wca = 0
|
||
|
|
record.wff = 0
|
||
|
|
record.byte_enable = 0xf
|
||
|
|
record.wcount = len(writes_datas)
|
||
|
|
record.rcount = 0
|
||
|
|
|
||
|
|
packet = EtherbonePacket()
|
||
|
|
packet.records = [record]
|
||
|
|
packet.encode()
|
||
|
|
self.tx_sock.sendto(bytes(packet), (self.ip_address, self.udp_port))
|
||
|
|
|
||
|
|
if self.debug:
|
||
|
|
for i, data in enumerate(datas):
|
||
|
|
print("WR %08X @ %08X" %(data, addr + 4*i))
|