import os
|
from sys import argv
|
import sys
|
|
# add python path of microwave to sys.path
|
microwave_path = os.path.join(__file__, *(['..'] * 3))
|
microwave_path = os.path.abspath(microwave_path)
|
sys.path.insert(0, microwave_path)
|
|
from microwave.servers import MicroWaveReceiver
|
|
|
if __name__ == '__main__':
|
if len(sys.argv) != 2:
|
print('Usage: python MicroWaveReceiverTest.py [folder]')
|
exit(1)
|
folder = sys.argv[1]
|
files = ["packet{}.udp".format(i) for i in range(100)]
|
receiver = MicroWaveReceiver()
|
for file in files:
|
filename = os.path.join(folder, file)
|
with open(filename, "rb") as f:
|
data = f.read()
|
receiver.AddBuf(data)
|
if receiver.DataFrameReady:
|
frame_number = receiver.Header.FrameNumber
|
bin_name = os.path.join(folder, "{}.bin".format(frame_number))
|
print("Frame {} ready".format(frame_number))
|
with open(bin_name, "wb") as bin_file:
|
bin_file.write(receiver.DataFrame)
|
bin_file.close()
|