1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#!/usr/bin/env python3
#
# This file is part of HackRF.
#
# Copyright (c) 2025 Great Scott Gadgets <info@greatscottgadgets.com>
# SPDX-License-Identifier: BSD-3-Clause
import lz4.block
import struct
import os
from board import PralinePlatform
from top.standard import Top as standard
from top.half_precision import Top as half_precision
from top.ext_precision_rx import Top as ext_precision_rx
from top.ext_precision_tx import Top as ext_precision_tx
BLOCK_SIZE = 4096 # 4 KiB blocks
OUTPUT_FILE = "praline_fpga.bin"
def compress_blockwise(input_stream, output_stream):
# For every block...
while block := f_in.read(BLOCK_SIZE):
# Compress the block using raw LZ4 block compression.
compressed_block = lz4.block.compress(block, store_size=False,
mode="high_compression", compression=12)
# Write the compressed block length and its contents to the output file.
f_out.write(struct.pack("<H", len(compressed_block)))
f_out.write(compressed_block)
# Write end marker (block of size 0)
f_out.write(struct.pack("<H", 0))
if __name__ == "__main__":
fpga_images = {
"0_standard": standard(),
"1_halfprec": half_precision(),
"2_extprec_rx": ext_precision_rx(),
"3_extprec_tx": ext_precision_tx(),
}
# Build bitstreams first.
for name, image in fpga_images.items():
PralinePlatform().build(image, name=name)
# Pack all the bitstreams.
with open(f"build/{OUTPUT_FILE}", "wb") as f_out:
f_out.write(struct.pack("<I", len(fpga_images))) # number of bitstreams
f_out.seek(4 * len(fpga_images), os.SEEK_CUR) # reserve 4-byte slots for offsets
offsets = []
# Write compressed bitstreams in our custom raw LZ4 block format.
for name, image in fpga_images.items():
img_path = f"build/{name}.bin"
offsets.append(f_out.tell())
with open(img_path, 'rb') as f_in:
compress_blockwise(f_in, f_out)
# Write offsets table, right after the number of bitstreams.
f_out.seek(4, os.SEEK_SET)
for offset in offsets:
f_out.write(struct.pack("<I", offset))
|
