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")