77 lines
3.1 KiB
Python
77 lines
3.1 KiB
Python
#!/usr/bin/python
|
|
|
|
from xml.etree import ElementTree
|
|
from collections import namedtuple
|
|
import optparse
|
|
import base64
|
|
import md5
|
|
import os
|
|
import sys
|
|
|
|
# Parse options
|
|
parser = optparse.OptionParser()
|
|
parser.add_option("--device", type="string", dest="device_type", help="Device Type. (Has to match the LVFPGA target plugin)", default=None)
|
|
parser.add_option("--input-bin", type="string", dest="input_bin", help="Path to bin file that needs to be merged with the LVBITX before exporting", default=None)
|
|
parser.add_option("--output-bin", type="string", dest="output_bin", help="Create a binary configuration bitstream", default=None)
|
|
parser.add_option("--output-lvbitx", type="string", dest="output_lvbitx_path", help="Output path for autogenerated LVBITX file", default=None)
|
|
parser.add_option("--output-src-path", type="string", dest="output_src_path", help="Output path for autogenerated src file", default=None)
|
|
(options, args) = parser.parse_args()
|
|
|
|
# Args
|
|
if (len(args) < 1):
|
|
print 'ERROR: Please specify the input LVBITX file name'
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
|
|
lvbitx_filename = args[0]
|
|
input_filename = os.path.abspath(lvbitx_filename)
|
|
|
|
if (not os.path.isfile(input_filename)):
|
|
print 'ERROR: LVBITX File ' + input_filename + ' could not be accessed or is not a file.'
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
if (options.input_bin is not None and not os.path.isfile(os.path.abspath(options.input_bin))):
|
|
print 'ERROR: FPGA Bin File ' + options.input_bin + ' could not be accessed or is not a file.'
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
if (options.output_lvbitx_path is not None and input_filename == options.output_lvbitx_path):
|
|
print 'ERROR: Input and output LVBITX files were the same. Choose a difference input or output file.'
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
|
|
# Get XML Tree Node
|
|
tree = ElementTree.parse(input_filename)
|
|
root = tree.getroot()
|
|
|
|
# Update device type
|
|
if (options.device_type is not None):
|
|
root.find('Project').find('TargetClass').text += '; ' + options.device_type
|
|
|
|
# Merge bitstream into LVBITX
|
|
if (options.input_bin is not None):
|
|
with open(os.path.abspath(options.input_bin), 'rb') as bin_file:
|
|
bitstream = bin_file.read()
|
|
bitstream_md5 = md5.new(bitstream).hexdigest()
|
|
bitstream_b64 = base64.b64encode(bitstream)
|
|
bitstream_b64_lb = ''
|
|
for i in range(0, len(bitstream_b64), 76):
|
|
bitstream_b64_lb += bitstream_b64[i:i+76] + '\n'
|
|
|
|
root.find('Bitstream').text = bitstream_b64_lb
|
|
root.find('BitstreamMD5').text = bitstream_md5
|
|
|
|
# Write BIN file
|
|
bitstream = base64.b64decode(root.find('Bitstream').text)
|
|
if (options.output_lvbitx_path is not None and md5.new(bitstream).hexdigest() != root.find('BitstreamMD5').text):
|
|
print 'ERROR: The MD5 sum for the output LVBITX was incorrect. Make sure that the bitstream in the input LVBITX or BIN file is valid.'
|
|
sys.exit(1)
|
|
if (options.output_bin is not None):
|
|
fpga_bin_file = open(options.output_bin, 'w')
|
|
fpga_bin_file.write(bitstream)
|
|
fpga_bin_file.close()
|
|
|
|
# Save LVBITX
|
|
if (options.output_lvbitx_path is not None):
|
|
tree.write(options.output_lvbitx_path, encoding="utf-8", xml_declaration=True, default_namespace=None, method="xml")
|
|
|