115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
import os
|
|
import time
|
|
from smbus import SMBus
|
|
from StartUpTest import StartUPTest
|
|
from TotOnline import TotOnline
|
|
import binascii
|
|
|
|
addr = 0x8
|
|
bus = SMBus(1)
|
|
|
|
# define the number of numbers and bits per number to generate
|
|
num_numbers = 6
|
|
bits_per_number = 1000000
|
|
|
|
# calculate the total file size in bits
|
|
filesize = num_numbers * bits_per_number
|
|
|
|
# define the filename
|
|
filename = f'{num_numbers}numbers_{bits_per_number}bits'
|
|
|
|
start_time_ges = start_time = time.time()
|
|
start_time = time.time() # start time of read
|
|
|
|
with open(filename + '.bin', 'wb') as file:
|
|
for i in range(num_numbers):
|
|
# write the separator between numbers except for the last one
|
|
#if i != num_numbers - 1:
|
|
#file.write(b'' + b'')
|
|
for j in range(bits_per_number // 8):
|
|
# read one byte from the serial port
|
|
data = bus.read_byte(addr)
|
|
file.write(bytes([data]))
|
|
time.sleep(0.0000001)
|
|
|
|
end_time = time.time() # end time of read
|
|
elapsed_time = end_time - start_time
|
|
|
|
seconds = int(elapsed_time)
|
|
milliseconds = int((elapsed_time % 1) * 1000)
|
|
|
|
new_filename = f"{filename}_TimeInSeconds_{seconds}_{milliseconds}.bin" # filename in format filename_seconds_milliseconds as txt with needed time to finish read
|
|
|
|
os.rename(filename + '.bin', new_filename) # change filename to new filename
|
|
|
|
'''with open(new_filename, 'rb') as f:
|
|
# Read the contents of the file as bytes
|
|
content = f.read()
|
|
result = StartUPTest.monobit_test(content)
|
|
print(result[0])
|
|
print(result[1])
|
|
result = StartUPTest.chi_square(content)
|
|
print(result[0])
|
|
print(result[1])'''
|
|
|
|
with open(new_filename, 'rb') as f:
|
|
# Read the contents of the file as bytes
|
|
content = f.read()
|
|
|
|
# Convert the bytes to a string of hexadecimal digits with zero padding
|
|
hex_str = ''.join(format(byte, '02x') for byte in content).zfill(num_numbers * (bits_per_number // 4))
|
|
|
|
# Write the hex string to the file with separators and newlines
|
|
with open(filename + '.txt', 'w') as f:
|
|
for i in range(num_numbers):
|
|
hex_number = hex_str[i * (bits_per_number // 4) : (i+1) * (bits_per_number // 4)]
|
|
# write the separator between numbers except for the last one
|
|
if i != num_numbers - 1:
|
|
hex_number += '//'
|
|
f.write(hex_number + '\n')
|
|
|
|
print(f"time needed in seconds: {elapsed_time:.2f}. New filename: {new_filename}.") # console write
|
|
|
|
# Open the file in binary mode
|
|
with open(new_filename, 'rb') as f:
|
|
# Read the contents of the file as bytes
|
|
content = f.read()
|
|
|
|
# Convert the bytes to a string of binary digits
|
|
binary_str = ''.join(format(byte, '08b') for byte in content)
|
|
|
|
# Write the binary string back to the file with separators
|
|
with open(filename + '_binary.txt', 'w') as f:
|
|
for i in range(0, len(binary_str), bits_per_number):
|
|
binary_number = binary_str[i:i+bits_per_number]
|
|
f.write(binary_number + '')
|
|
|
|
# Open the file in binary mode
|
|
with open(filename + '_binary.txt', 'rb') as f:
|
|
# Read the contents of the file as bytes
|
|
content = f.read()
|
|
#print(content)
|
|
|
|
result = TotOnline.total_failure_test(content)
|
|
print(result[0])
|
|
print(result[1])
|
|
result = TotOnline.block_frequency_test(content)
|
|
print(result[0])
|
|
print(result[1])
|
|
result = TotOnline.longest_one_block_test(content)
|
|
print(result[0])
|
|
print(result[1])
|
|
result = TotOnline.monobit_test(content)
|
|
print(result[0])
|
|
print(result[1])
|
|
result = TotOnline.run_test(content)
|
|
print(result[0])
|
|
print(result[1])
|
|
|
|
end_time_ges = time.time()
|
|
elapsed_time_ges = end_time_ges - start_time_ges
|
|
|
|
|
|
seconds_ges = int(elapsed_time_ges)
|
|
milliseconds_ges = int((elapsed_time_ges % 1) * 1000)
|
|
print(elapsed_time_ges) |