def __init__(self, args=None):
settings = Settings()
self.directory = settings.dirScans
wx.Frame.__init__(self, None, -1, 'SHARPMAP Viewer', size=(800, 600))
Figure = app.GetFigureClass()
fig = Figure(self)
panel = wx.Panel(self)
button = wx.Button(panel, wx.ID_ANY, 'Open')
button.Bind(wx.EVT_BUTTON, self.__on_open)
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(button, 0, wx.ALIGN_CENTRE | wx.ALL, border=5)
panel.SetSizer(sizer)
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(fig._widget, 1, wx.EXPAND | wx.ALL, border=5)
sizer.Add(panel, 0, wx.EXPAND | wx.ALL, border=5)
self.SetSizer(sizer)
self.SetAutoLayout(True)
self.Layout()
self.Show()
if args.file is not None:
if os.path.exists(args.file):
self.__open(*os.path.split(args.file))
else:
wx.MessageBox('File not found', 'Error', wx.OK | wx.ICON_ERROR)
def __init__(self, args):
start = args.start
end = args.end
sweeps = args.sweeps
gain = args.gain
dwell = args.dwell
nfft = args.fft
lo = args.lo
index = args.index
remote = args.remote
directory, filename = os.path.split(args.file)
_null, ext = os.path.splitext(args.file)
self.lock = threading.Lock()
self.stepsTotal = 0
self.steps = 0
self.spectrum = OrderedDict()
self.locations = OrderedDict()
self.settings = Settings(load=False)
self.queueNotify = Queue()
self.queueScan = Queue()
self.queueLocation = Queue()
self.threadLocation = None
error = None
if end <= start:
error = "Start should be lower than end"
elif dwell <= 0:
error = "Dwell should be positive"
elif dwell > max(get_dwells()[1::2]):
error = "Dwell should equal lower than {}s".format(max(get_dwells()[1::2]))
elif nfft <= 0:
error = "FFT bins should be positive"
elif ext != ".rfs" and File.get_type_index(ext) == -1:
error = "File extension should be "
error += File.get_type_pretty(File.Types.SAVE)
error += File.get_type_pretty(File.Types.PLOT)
else:
device = DeviceRTL()
if remote is None:
self.settings.devicesRtl = get_devices_rtl()
count = len(self.settings.devicesRtl)
if index > count - 1:
error = "Device not found ({} devices in total):\n".format(count)
for device in self.settings.devicesRtl:
error += "\t{}: {}\n".format(device.indexRtl,
device.name)
else:
device.isDevice = False
url = urlparse('//' + remote)
if url.hostname is not None:
device.server = url.hostname
else:
error = "Invalid hostname"
if url.port is not None:
device.port = url.port
else:
device.port = 1234
self.settings.devicesRtl.append(device)
index = len(self.settings.devicesRtl) - 1
if args.conf is not None:
if os.path.exists(args.conf):
error = self.settings.load_conf(args.conf)
if error is not None:
error = 'GPS configuration - ' + error
else:
error = 'Cannot find {}'.format(args.conf)
if end - 1 < start:
end = start + 1
if remote is None:
if len(self.settings.devicesRtl):
gain = nearest(gain, self.settings.devicesRtl[index].gains)
else:
error = 'No devices found'
if error is not None:
print("Error: {}".format(error))
exit(1)
self.settings.start = start
self.settings.stop = end
self.settings.dwell = calc_real_dwell(dwell)
self.settings.scanDelay = args.delay
self.settings.nfft = nfft
self.settings.devicesRtl[index].gain = gain
self.settings.devicesRtl[index].lo = lo
print("{} - {}MHz".format(start, end))
print("{} Sweeps".format(sweeps))
print("{}dB Gain".format(gain))
print("{}s Dwell".format(self.settings.dwell))
print("{} FFT points".format(nfft))
print("{}MHz LO".format(lo))
if remote is not None:
print(remote)
else:
print(self.settings.devicesRtl[index].name)
if len(self.settings.devicesGps):
print('Using GPS configuration \'{}\''.format(self.settings.devicesGps[0].name))
self.threadLocation = ThreadLocation(self.queueLocation,
self.settings.devicesGps[0])
if not self.__gps_wait():
self.__gps_stop()
exit(1)
self.__scan(sweeps, self.settings, index)
fullName = os.path.join(directory, filename)
if ext == ".rfs":
scanInfo = ScanInfo()
scanInfo.set_from_settings(self.settings)
save_plot(fullName, scanInfo, self.spectrum, self.locations)
else:
exportType = File.get_type_index(ext)
export_plot(fullName, exportType, self.spectrum)
self.__gps_stop()
print("Done")
class Cli:
def __init__(self, args):
start = args.start
end = args.end
sweeps = args.sweeps
gain = args.gain
dwell = args.dwell
nfft = args.fft
lo = args.lo
index = args.index
remote = args.remote
directory, filename = os.path.split(args.file)
_null, ext = os.path.splitext(args.file)
self.lock = threading.Lock()
self.stepsTotal = 0
self.steps = 0
self.spectrum = OrderedDict()
self.locations = OrderedDict()
self.settings = Settings(load=False)
self.queueNotify = Queue()
self.queueScan = Queue()
self.queueLocation = Queue()
self.threadLocation = None
error = None
if end <= start:
error = "Start should be lower than end"
elif dwell <= 0:
error = "Dwell should be positive"
elif dwell > max(get_dwells()[1::2]):
error = "Dwell should equal lower than {}s".format(max(get_dwells()[1::2]))
elif nfft <= 0:
error = "FFT bins should be positive"
elif ext != ".rfs" and File.get_type_index(ext) == -1:
error = "File extension should be "
error += File.get_type_pretty(File.Types.SAVE)
error += File.get_type_pretty(File.Types.PLOT)
else:
device = DeviceRTL()
if remote is None:
self.settings.devicesRtl = get_devices_rtl()
count = len(self.settings.devicesRtl)
if index > count - 1:
error = "Device not found ({} devices in total):\n".format(count)
for device in self.settings.devicesRtl:
error += "\t{}: {}\n".format(device.indexRtl,
device.name)
else:
device.isDevice = False
url = urlparse('//' + remote)
if url.hostname is not None:
device.server = url.hostname
else:
error = "Invalid hostname"
if url.port is not None:
device.port = url.port
else:
device.port = 1234
self.settings.devicesRtl.append(device)
index = len(self.settings.devicesRtl) - 1
if args.conf is not None:
if os.path.exists(args.conf):
error = self.settings.load_conf(args.conf)
if error is not None:
error = 'GPS configuration - ' + error
else:
error = 'Cannot find {}'.format(args.conf)
if end - 1 < start:
end = start + 1
if remote is None:
if len(self.settings.devicesRtl):
gain = nearest(gain, self.settings.devicesRtl[index].gains)
else:
error = 'No devices found'
if error is not None:
print("Error: {}".format(error))
exit(1)
self.settings.start = start
self.settings.stop = end
self.settings.dwell = calc_real_dwell(dwell)
self.settings.scanDelay = args.delay
self.settings.nfft = nfft
self.settings.devicesRtl[index].gain = gain
self.settings.devicesRtl[index].lo = lo
print("{} - {}MHz".format(start, end))
print("{} Sweeps".format(sweeps))
print("{}dB Gain".format(gain))
print("{}s Dwell".format(self.settings.dwell))
print("{} FFT points".format(nfft))
print("{}MHz LO".format(lo))
if remote is not None:
print(remote)
else:
print(self.settings.devicesRtl[index].name)
if len(self.settings.devicesGps):
print('Using GPS configuration \'{}\''.format(self.settings.devicesGps[0].name))
self.threadLocation = ThreadLocation(self.queueLocation,
self.settings.devicesGps[0])
if not self.__gps_wait():
self.__gps_stop()
exit(1)
self.__scan(sweeps, self.settings, index)
fullName = os.path.join(directory, filename)
if ext == ".rfs":
scanInfo = ScanInfo()
scanInfo.set_from_settings(self.settings)
save_plot(fullName, scanInfo, self.spectrum, self.locations)
else:
exportType = File.get_type_index(ext)
export_plot(fullName, exportType, self.spectrum)
self.__gps_stop()
print("Done")
def __gps_wait(self):
print('\nWaiting for GPS fix: {}'.format(self.settings.devicesGps[0].get_desc()))
while True:
if not self.queueLocation.empty():
status = self.__process_event(self.queueLocation)
if status == Event.LOC:
return True
elif status == Event.LOC_ERR:
return False
def __gps_stop(self):
if self.threadLocation and self.threadLocation.is_alive():
self.threadLocation.stop()
def __scan(self, sweeps, settings, index):
samples = settings.dwell * SAMPLE_RATE
samples = next_2_to_pow(int(samples))
for sweep in range(0, sweeps):
print('\nSweep {}:'.format(sweep + 1))
threadScan = ThreadScan(self.queueNotify, self.queueScan, None,
settings, index, samples, False)
while threadScan.is_alive() or self.steps > 0:
if not self.queueNotify.empty():
self.__process_event(self.queueNotify)
if not self.queueLocation.empty():
self.__process_event(self.queueLocation)
if self.settings.scanDelay > 0 and sweep < sweeps - 1:
print('\nDelaying {}s'.format(self.settings.scanDelay))
time.sleep(self.settings.scanDelay)
threadScan.rtl_close()
print("")
print("")
def __process_event(self, queue):
event = queue.get()
status = event.data.get_status()
arg1 = event.data.get_arg1()
arg2 = event.data.get_arg2()
if status == Event.STARTING:
print("Starting")
elif status == Event.STEPS:
self.stepsTotal = (arg1 + 1) * 2
self.steps = self.stepsTotal
elif status == Event.INFO:
if arg2 != -1:
self.settings.devicesRtl[self.settings.indexRtl].tuner = arg2
elif status == Event.DATA:
cal = self.settings.devicesRtl[self.settings.indexRtl].calibration
levelOff = self.settings.devicesRtl[self.settings.indexRtl].levelOff
freq, scan = self.queueScan.get()
process = ThreadProcess(self.queueNotify,
freq, scan, cal, levelOff,
self.settings.nfft,
self.settings.overlap,
self.settings.winFunc)
process.start()
self.__progress()
elif status == Event.ERROR:
print("Error: {}".format(arg2))
exit(1)
elif status == Event.PROCESSED:
offset = self.settings.devicesRtl[self.settings.indexRtl].offset
Thread(target=update_spectrum, name='Update',
args=(self.queueNotify, self.lock,
self.settings.start,
self.settings.stop,
arg1,
offset,
self.spectrum,
not self.settings.retainScans,
False)).start()
elif status == Event.UPDATED:
self.__progress()
elif status == Event.LOC:
if len(self.spectrum) > 0:
self.locations[max(self.spectrum)] = (arg2[0],
arg2[1],
arg2[2])
elif status == Event.LOC_ERR:
print('Error: {}'.format(arg2))
exit(1)
return status
def __progress(self):
self.steps -= 1
comp = (self.stepsTotal - self.steps) * 100 / self.stepsTotal
sys.stdout.write("\r{0:.1f}%".format(comp))