Merge pull request #6 from nzkarit/logging

Logging and some other little tweaks
This commit is contained in:
nzkarit 2017-09-18 23:31:29 +12:00 committed by GitHub
commit ae6ee74cd2
2 changed files with 14 additions and 7 deletions

View File

@ -84,17 +84,17 @@ def singlePlane(arguments):
return samples return samples
def manyPlanes(arguments): def manyPlanes(arguments):
logger.info('Processing CSV file') logger.info('Processing CSV file')
logger.info('Repeating the message %s times' % (arguments.repeats))
samples = bytearray() samples = bytearray()
print(arguments.repeats) logger.info('Repeating the message %s times' % (arguments.repeats))
for i in range(0, arguments.repeats): for i in range(0, arguments.repeats):
with open(arguments.csvfile, newline='') as csvfile: with open(arguments.csvfile, newline='') as csvfile:
reader = csv.DictReader(csvfile, delimiter=',') reader = csv.DictReader(csvfile, delimiter=',')
for row in reader: for row in reader:
if not 'icao' in row.keys(): if not 'icao' in row.keys():
row['icao'] = arguments.icao row['icao'] = arguments.icao
row['icao'] = int(row['icao'], 0) else:
row['icao'] = int(row['icao'], 0)
if not 'latitude' in row.keys(): if not 'latitude' in row.keys():
row['latitude'] = arguments.latitude row['latitude'] = arguments.latitude
if not 'longitude' in row.keys(): if not 'longitude' in row.keys():
@ -114,7 +114,6 @@ def manyPlanes(arguments):
if not 'surface' in row.keys(): if not 'surface' in row.keys():
row['surface'] = arguments.surface row['surface'] = arguments.surface
logger.debug('Row from CSV: %s' % (row)) logger.debug('Row from CSV: %s' % (row))
print(row)
modes = ModeS() modes = ModeS()
(df17_even, df17_odd) = modes.df17_pos_rep_encode(row['capability'], row['icao'], row['typecode'], row['surveillancestatus'], row['nicsupplementb'], row['altitude'], row['time'], row['latitude'], row['longitude'], row['surface']) (df17_even, df17_odd) = modes.df17_pos_rep_encode(row['capability'], row['icao'], row['typecode'], row['surveillancestatus'], row['nicsupplementb'], row['altitude'], row['time'], row['latitude'], row['longitude'], row['surface'])
@ -127,12 +126,14 @@ def manyPlanes(arguments):
return samples return samples
def writeOutputFile(filename, data): def writeOutputFile(filename, data):
logger.info('Writing tmp.iq8s file')
SamplesFile = open('tmp.iq8s', 'wb') SamplesFile = open('tmp.iq8s', 'wb')
SamplesFile.write(data) SamplesFile.write(data)
SamplesFile.close() SamplesFile.close()
os.system('sync') os.system('sync')
os.system('rm %s' % (filename)) os.system('rm %s' % (filename))
os.system("dd if=tmp.iq8s of=%s bs=4k seek=63" % (filename)) # TODO redirect output to /dev/null logger.info('dd for file: %s' % (filename))
os.system("dd if=tmp.iq8s of=%s bs=4k seek=63 > /dev/null 2>&1" % (filename))
os.system('sync') os.system('sync')
os.system('rm tmp.iq8s') os.system('rm tmp.iq8s')
@ -154,6 +155,7 @@ if __name__ == "__main__":
else: else:
data = manyPlanes(arguments) data = manyPlanes(arguments)
writeOutputFile(arguments.outputfilename, data) writeOutputFile(arguments.outputfilename, data)
logger.info('Complete')

View File

@ -1,4 +1,5 @@
import numpy import numpy
import logging
############################################################### ###############################################################
# Further work on fork # Further work on fork
# Copyright (C) 2017 David Robinson # Copyright (C) 2017 David Robinson
@ -6,6 +7,10 @@ class HackRF:
"""The HackRF class has functions from converting data into a format into which the hackrf can process """The HackRF class has functions from converting data into a format into which the hackrf can process
""" """
logger = None
def __init__(self):
self.logger = logging.getLogger(__name__)
def hackrf_raw_IQ_format(self, ppm): def hackrf_raw_IQ_format(self, ppm):
""" """
@ -15,7 +20,7 @@ class HackRF:
Returns: Returns:
bytearray: containing the IQ data bytearray: containing the IQ data
""" """
self.logger.debug('Creating hackRF bytearray from the ppm stuff')
signal = [] signal = []
bits = numpy.unpackbits(numpy.asarray(ppm, dtype=numpy.uint8)) bits = numpy.unpackbits(numpy.asarray(ppm, dtype=numpy.uint8))
for bit in bits: for bit in bits: