def __arguments():
parser = argparse.ArgumentParser(prog="rtlsdr_scan.py",
description='''
Scan a range of frequencies and
save the results to a file''')
parser.add_argument("-s", "--start", help="Start frequency (MHz)",
type=int)
parser.add_argument("-e", "--end", help="End frequency (MHz)", type=int)
parser.add_argument("-w", "--sweeps", help="Number of sweeps", type=int,
default=1)
parser.add_argument("-p", "--delay", help="Delay between sweeps (s)",
type=int, default=0)
parser.add_argument("-g", "--gain", help="Gain (dB)", type=float, default=0)
parser.add_argument("-d", "--dwell", help="Dwell time (seconds)",
type=float, default=0.1)
parser.add_argument("-f", "--fft", help="FFT bins", type=int, default=1024)
parser.add_argument("-l", "--lo", help="Local oscillator offset",
type=int, default=0)
parser.add_argument("-c", "--conf", help="Load a config file",
default=None)
group = parser.add_mutually_exclusive_group()
group.add_argument("-i", "--index", help="Device index (from 0)", type=int,
default=0)
group.add_argument("-r", "--remote", help="Server IP and port", type=str)
types = File.get_type_pretty(File.Types.SAVE)
types += File.get_type_pretty(File.Types.PLOT)
help = 'Output file (' + types + ')'
parser.add_argument("file", help=help, nargs='?')
args = parser.parse_args()
error = None
isGui = True
if args.start is not None or args.end is not None:
if args.start is not None:
if args.end is not None:
if args.file is not None:
isGui = False
else:
error = "No filename specified"
else:
error = "No end frequency specified"
else:
error = "No start frequency specified"
elif args.file is not None:
args.dirname, args.filename = os.path.split(args.file)
if error is not None:
print("Error: {}".format(error))
parser.exit(1)
return isGui, (args)
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")
