forked from Firmware/Firmwares
47 lines
1.3 KiB
Python
Executable File
47 lines
1.3 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
import re
|
|
|
|
import sys
|
|
|
|
if (len(sys.argv) < 2):
|
|
print('No argument given')
|
|
exit()
|
|
|
|
|
|
nips = sys.argv[1]
|
|
|
|
input_file = open(nips + '_inputdata.txt', 'r')
|
|
in_data = input_file.read();
|
|
|
|
# convert double to uint8_t
|
|
in_data = re.sub(r'(\d)\.0+e\+00', r'\1,', in_data)
|
|
in_data = re.sub(r'(\d)\.(\d)0+e\+01', r'\1\2,', in_data)
|
|
in_data = re.sub(r'(\d)\.(\d)(\d)0+e\+02', r'\1\2\3,', in_data)
|
|
in_data = in_data.replace(";", "")
|
|
|
|
# remove last comma
|
|
in_data = in_data[:-2]
|
|
# count samples
|
|
input_sample_cnt = len(in_data.split(","))
|
|
#####################################################################
|
|
|
|
ref_file = open(nips + '_outputdata.txt', 'r')
|
|
ref_data = ref_file.read()
|
|
ref_data = re.sub(r'\n', r' / ln2,\n', ref_data)
|
|
ref_data = ref_data[:-2]
|
|
ref_sample_cnt = len(ref_data.split(","))
|
|
|
|
# create cpp file
|
|
f = open("../xspn_data.cpp", "w")
|
|
cpp_file = "#include <array>\n"
|
|
cpp_file += "#include <cmath>\n"
|
|
cpp_file += "#include <bits/stdint-uintn.h>\n\n"
|
|
# The results in the outputdata.txt file are not directly what comes out of the PE but the natural logarithm of it.
|
|
cpp_file += "constexpr auto ln2 = std::log(2);\n"
|
|
cpp_file += "std::array<uint8_t, " + str(input_sample_cnt) + "> input_data = {\n" + str(in_data) + "}; \n\n"
|
|
cpp_file += "std::array<double, " + str(ref_sample_cnt) + "> ref_data = {\n" + str(ref_data)+ "}; \n"
|
|
|
|
f.write(cpp_file)
|
|
f.close()
|