Beispiel #1
0
 def on_save(self, _event):
     dlg = wx.FileDialog(
         self, "Save a scan", self.settings.dirScans, self.filename, File.RFS, wx.SAVE | wx.OVERWRITE_PROMPT
     )
     if dlg.ShowModal() == wx.ID_OK:
         self.status.set_general("Saving")
         self.filename = os.path.splitext(dlg.GetFilename())[0]
         dirname = dlg.GetDirectory()
         self.settings.dirScans = dirname
         save_plot(dirname, dlg.GetFilename(), self.scanInfo, self.spectrum)
         self.saved(True)
         self.status.set_general("Finished")
         self.settings.fileHistory.AddFileToHistory(os.path.join(dirname, dlg.GetFilename()))
     dlg.Destroy()
 def __on_save(self, _event):
     dlg = wx.FileDialog(self, "Save a scan", self.settings.dirScans,
                         self.filename,
                         File.get_type_filters(File.Types.SAVE),
                         wx.SAVE | wx.OVERWRITE_PROMPT)
     if dlg.ShowModal() == wx.ID_OK:
         self.status.set_general("Saving...")
         fileName = dlg.GetFilename()
         dirName = dlg.GetDirectory()
         self.filename = os.path.splitext(fileName)[0]
         self.settings.dirScans = dirName
         fileName = extension_add(fileName, dlg.GetFilterIndex(),
                                  File.Types.SAVE)
         fullName = os.path.join(dirName, fileName)
         save_plot(fullName, self.scanInfo, self.spectrum, self.locations)
         self.__saved(True)
         self.status.set_general("Finished")
         self.settings.fileHistory.AddFileToHistory(fullName)
     dlg.Destroy()
 def __on_save(self, _event):
     dlg = wx.FileDialog(self, "Save a scan", self.settings.dirScans,
                         self.filename,
                         File.get_type_filters(File.Types.SAVE),
                         wx.SAVE | wx.OVERWRITE_PROMPT)
     if dlg.ShowModal() == wx.ID_OK:
         self.status.set_general("Saving...")
         fileName = dlg.GetFilename()
         dirName = dlg.GetDirectory()
         self.filename = os.path.splitext(fileName)[0]
         self.settings.dirScans = dirName
         fileName = extension_add(fileName, dlg.GetFilterIndex(),
                                  File.Types.SAVE)
         fullName = os.path.join(dirName, fileName)
         save_plot(fullName, self.scanInfo, self.spectrum, self.locations)
         self.__saved(True)
         self.status.set_general("Finished")
         self.settings.fileHistory.AddFileToHistory(fullName)
     dlg.Destroy()
Beispiel #4
0
    def __init__(self, pool, 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 = {}
        self.settings = Settings(load=False)

        self.queue = Queue.Queue()

        error = None

        if end <= start:
            error = "Start should be lower than end"
        elif dwell <= 0:
            error = "Dwell should be positive"
        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 error is not None:
            print "Error: {}".format(error)
            exit(1)

        if end - 1 < start:
            end = start + 1
        if remote is None:
            gain = nearest(gain, self.settings.devicesRtl[index].gains)

        self.settings.start = start
        self.settings.stop = end
        self.settings.dwell = calc_real_dwell(dwell)
        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

        self.__scan(sweeps, self.settings, index, pool)

        fullName = os.path.join(directory, filename)
        if ext == ".rfs":
            scanInfo = ScanInfo()
            scanInfo.set_from_settings(self.settings)

            save_plot(fullName, scanInfo, self.spectrum, {})
        else:
            exportType = File.get_type_index(ext)
            export_plot(fullName, exportType, self.spectrum)

        print "Done"
Beispiel #5
0
    def __init__(self, pool, 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.Queue()
        self.queueScan = Queue.Queue()
        self.queueLocation = Queue.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 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)
            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):
            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, pool)

        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"