forked from M-Labs/artiq-zynq
sim: add frame generator & image viewer
This commit is contained in:
parent
1456010942
commit
6834f136c7
146
sim_frame.py
146
sim_frame.py
|
|
@ -3,9 +3,14 @@ from misoc.interconnect import stream
|
|||
|
||||
from sim_pipeline import *
|
||||
from sim_generator import CXPCRC32Inserter
|
||||
from sim_frame_gen import get_frame_packet
|
||||
|
||||
from src.gateware.cxp_pipeline import *
|
||||
|
||||
import numpy as np
|
||||
from PIL import Image
|
||||
|
||||
|
||||
class Frame(Module):
|
||||
def __init__(self):
|
||||
# to construct correct crc and ack/stb signal
|
||||
|
|
@ -23,16 +28,17 @@ class Frame(Module):
|
|||
|
||||
# no backpressure for sim
|
||||
self.sync += self.source.ack.eq(1)
|
||||
|
||||
|
||||
|
||||
dut = Frame()
|
||||
|
||||
|
||||
def check_case(packet=[]):
|
||||
print("=================TEST========================")
|
||||
|
||||
sink = dut.sink
|
||||
stream_pipe = dut.stream_pipe
|
||||
|
||||
|
||||
for i, p in enumerate(packet):
|
||||
yield sink.data.eq(p["data"])
|
||||
yield sink.k.eq(p["k"])
|
||||
|
|
@ -45,23 +51,16 @@ def check_case(packet=[]):
|
|||
# check cycle result
|
||||
yield
|
||||
# source = dut.dchar_decoder.source
|
||||
source = dut.stream_pipe.frame_extractor.sink
|
||||
# source = dut.stream_pipe.frame_extractor.sink
|
||||
source = dut.sink
|
||||
print(
|
||||
f"\nCYCLE#{i} : source char = {yield source.data:#X} k = {yield source.k:#X} stb = {yield source.stb} ack = {yield source.ack} eop = {yield source.eop}"
|
||||
# f" source dchar = {yield source.dchar:#X} dchar_k = {yield source.dchar_k:#X}"
|
||||
# f"\nCYCLE#{i} : stream id = {yield decoder.stream_id:#X} pak_tag = {yield decoder.pak_tag:#X}"
|
||||
# f" stream_pak_size = {yield decoder.stream_pak_size:#X}"
|
||||
)
|
||||
# crc = downconns[1].generator.crc_inserter.crc
|
||||
# crc = dut.double_buffer.crc
|
||||
# print(
|
||||
# f"CYCLE#{i} : crc error = {yield crc.error:#X} crc value = {yield crc.value:#X}"
|
||||
# f" crc data = {yield crc.data:#X} engine next = {yield crc.engine.next:#X} ce = {yield crc.ce}"
|
||||
# )
|
||||
|
||||
|
||||
# extra clk cycles
|
||||
cyc = i + 1
|
||||
img = []
|
||||
line = -1
|
||||
for i in range(cyc, cyc + 50):
|
||||
yield sink.data.eq(0)
|
||||
yield sink.k.eq(0)
|
||||
|
|
@ -70,86 +69,61 @@ def check_case(packet=[]):
|
|||
yield
|
||||
print(
|
||||
f"\nCYCLE#{i} : source char = {yield source.data:#X} k = {yield source.k:#X} stb = {yield source.stb} ack = {yield source.ack} eop = {yield source.eop}"
|
||||
# f" source dchar = {yield source.dchar:#X} dchar_k = {yield source.dchar_k:#X}"
|
||||
# f"\nCYCLE#{i} : stream id = {yield decoder.stream_id:#X} pak_tag = {yield decoder.pak_tag:#X}"
|
||||
# f" stream_pak_size = {yield decoder.stream_pak_size:#X}"
|
||||
)
|
||||
metadata = dut.stream_pipe.frame_extractor.metadata
|
||||
img_header_layout = [
|
||||
"stream_id",
|
||||
"source_tag",
|
||||
"x_size",
|
||||
"x_offset",
|
||||
"y_size",
|
||||
"y_offset",
|
||||
"l_size", # number of data words per image line
|
||||
"pixel_format",
|
||||
"tap_geo",
|
||||
"flag",
|
||||
]
|
||||
for name in img_header_layout:
|
||||
print(f"{name} = {yield getattr(metadata, name):#04X} ", end="")
|
||||
print()
|
||||
|
||||
frame_extractoer = dut.stream_pipe.frame_extractor
|
||||
new_line = yield frame_extractoer.new_line
|
||||
if new_line:
|
||||
img.append([])
|
||||
line += 1
|
||||
|
||||
stb = yield frame_extractoer.source.stb
|
||||
data = yield frame_extractoer.source.data
|
||||
if stb:
|
||||
# CXP use MSB
|
||||
img[line].append(np.uint16(data & 0xFFFF))
|
||||
img[line].append(np.uint16(data >> 16))
|
||||
print(img)
|
||||
|
||||
# metadata = dut.stream_pipe.frame_extractor.metadata
|
||||
# img_header_layout = [
|
||||
# "stream_id",
|
||||
# "source_tag",
|
||||
# "x_size",
|
||||
# "x_offset",
|
||||
# "y_size",
|
||||
# "y_offset",
|
||||
# "l_size", # number of data words per image line
|
||||
# "pixel_format",
|
||||
# "tap_geo",
|
||||
# "flag",
|
||||
# ]
|
||||
# for name in img_header_layout:
|
||||
# print(f"{name} = {yield getattr(metadata, name):#04X} ", end="")
|
||||
# print()
|
||||
Image.fromarray(np.array(img, dtype=np.uint16)).show()
|
||||
assert True
|
||||
|
||||
|
||||
def testbench():
|
||||
# stream_id = 0x01
|
||||
streams = [
|
||||
[
|
||||
{"data": 0x11111111, "k": Replicate(0, 4)},
|
||||
{"data": 0xB105F00D, "k": Replicate(0, 4)},
|
||||
],
|
||||
[
|
||||
{"data": 0x22222222, "k": Replicate(0, 4)},
|
||||
{"data": 0xC001BEA0, "k": Replicate(0, 4)},
|
||||
],
|
||||
[
|
||||
{"data": Replicate(KCode["stream_marker"], 4), "k": Replicate(1, 4)},
|
||||
{"data": Replicate(C(0x01,char_width), 4), "k": Replicate(0, 4)},
|
||||
|
||||
{"data": Replicate(C(0x02,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x03,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x04,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x05,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x06,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x07,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x08,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x09,char_width), 4), "k": Replicate(0, 4)},
|
||||
|
||||
{"data": Replicate(C(0x10,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x11,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x12,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x13,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x14,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x15,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x16,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x17,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x18,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x19,char_width), 4), "k": Replicate(0, 4)},
|
||||
|
||||
{"data": Replicate(C(0x20,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x21,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x22,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x23,char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(0x24,char_width), 4), "k": Replicate(0, 4)},
|
||||
],
|
||||
stream_id = 0x69
|
||||
packet_tag = 0
|
||||
frame_packet = get_frame_packet(stream_id)
|
||||
packet = [
|
||||
{"data": Replicate(C(stream_id, char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(packet_tag, char_width), 4), "k": Replicate(0, 4)},
|
||||
{
|
||||
"data": Replicate(C(len(frame_packet), char_width)[8:], 4),
|
||||
"k": Replicate(0, 4),
|
||||
},
|
||||
{
|
||||
"data": Replicate(C(len(frame_packet), char_width)[:8], 4),
|
||||
"k": Replicate(0, 4),
|
||||
},
|
||||
]
|
||||
packet += frame_packet
|
||||
|
||||
packet = []
|
||||
for i, s in enumerate(streams):
|
||||
s[-1]["eop"] = 0
|
||||
packet += [
|
||||
{"data": Replicate(C(i % 2, char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(C(i, char_width), 4), "k": Replicate(0, 4)},
|
||||
{
|
||||
"data": Replicate(C(len(s) >> 8 & 0xFF, char_width), 4),
|
||||
"k": Replicate(0, 4),
|
||||
},
|
||||
{"data": Replicate(C(len(s) & 0xFF, char_width), 4), "k": Replicate(0, 4)},
|
||||
*s,
|
||||
]
|
||||
|
||||
packet[-1]["eop"] = 0
|
||||
|
||||
yield from check_case(packet)
|
||||
|
||||
|
|
|
|||
|
|
@ -0,0 +1,120 @@
|
|||
from migen import *
|
||||
from misoc.interconnect import stream
|
||||
|
||||
from src.gateware.cxp_pipeline import *
|
||||
from sim_pipeline import *
|
||||
|
||||
from PIL import Image
|
||||
import numpy as np
|
||||
|
||||
|
||||
def get_image_header(
|
||||
stream_id, source_tag, xsize, xoffset, ysize, yoffset, dsize, pixelF, tag_geo, flag
|
||||
):
|
||||
stream_id = C(stream_id, char_width)
|
||||
source_tag = C(source_tag, 2 * char_width)
|
||||
xsize = C(xsize, 3 * char_width)
|
||||
xoffset = C(xoffset, 3 * char_width)
|
||||
ysize = C(ysize, 3 * char_width)
|
||||
yoffset = C(yoffset, 3 * char_width)
|
||||
dsize = C(dsize, 3 * char_width)
|
||||
pixelF = C(pixelF, 2 * char_width)
|
||||
tag_geo = C(tag_geo, 2 * char_width)
|
||||
flag = C(flag, char_width)
|
||||
|
||||
assert len(stream_id) == len(flag) == char_width
|
||||
assert len(source_tag) == len(pixelF) == len(tag_geo) == 2 * char_width
|
||||
assert len(xsize) == len(xoffset) == len(ysize) == len(yoffset) == 3 * char_width
|
||||
return [
|
||||
{"data": Replicate(KCode["stream_marker"], 4), "k": Replicate(1, 4)},
|
||||
{"data": Replicate(C(0x01, char_width), 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(stream_id, 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(source_tag[8:], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(source_tag[:8], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(xsize[16:], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(xsize[8:16], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(xsize[:8], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(xoffset[16:], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(xoffset[8:16], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(xoffset[:8], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(ysize[16:], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(ysize[8:16], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(ysize[:8], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(yoffset[16:], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(yoffset[8:16], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(yoffset[:8], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(dsize[16:], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(dsize[8:16], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(dsize[:8], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(pixelF[8:], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(pixelF[:8], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(tag_geo[8:], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(tag_geo[:8], 4), "k": Replicate(0, 4)},
|
||||
{"data": Replicate(flag, 4), "k": Replicate(0, 4)},
|
||||
]
|
||||
|
||||
|
||||
def get_line_marker():
|
||||
return [
|
||||
{"data": Replicate(KCode["stream_marker"], 4), "k": Replicate(1, 4)},
|
||||
{"data": Replicate(C(0x02, char_width), 4), "k": Replicate(0, 4)},
|
||||
]
|
||||
|
||||
|
||||
def get_frame_packet(stream_id, pixel_format="mono16"):
|
||||
assert pixel_format in ["mono16"]
|
||||
|
||||
arr = [[4156, 31078], [65063, 24837]]
|
||||
source_tag = 0
|
||||
xsize, ysize = len(arr[0]), len(arr)
|
||||
|
||||
xoffset, yoffset = 0, 0
|
||||
if pixel_format == "mono16":
|
||||
dsize = xsize // 2
|
||||
pixelF = 0x0105
|
||||
tag_geo = 0
|
||||
flag = 0
|
||||
|
||||
packet = []
|
||||
# Image header
|
||||
packet += get_image_header(
|
||||
stream_id,
|
||||
source_tag,
|
||||
xsize,
|
||||
xoffset,
|
||||
ysize,
|
||||
yoffset,
|
||||
dsize,
|
||||
pixelF,
|
||||
tag_geo,
|
||||
flag,
|
||||
)
|
||||
|
||||
for line in arr:
|
||||
packet += get_line_marker()
|
||||
if pixel_format == "mono16":
|
||||
for i in range(len(line)):
|
||||
if (i % 2) == 0:
|
||||
|
||||
if i == len(line) - 1:
|
||||
print(C(line[i]))
|
||||
packet += [
|
||||
{
|
||||
"data": C(line[i], 4 * char_width),
|
||||
"k": Replicate(0, 4),
|
||||
},
|
||||
]
|
||||
else:
|
||||
print(C(line[i], 2 * char_width), C(line[i + 1]))
|
||||
# CXP use MSB
|
||||
packet += [
|
||||
{
|
||||
"data": Cat(
|
||||
C(line[i], 2 * char_width),
|
||||
C(line[i + 1], 2 * char_width),
|
||||
),
|
||||
"k": Replicate(0, 4),
|
||||
},
|
||||
]
|
||||
|
||||
return packet
|
||||
Loading…
Reference in New Issue