mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-11-26 01:39:05 -05:00
Web API: NFM scanner Python script example (3)
This commit is contained in:
f4exb
parent
efcbf9b7cb
commit
0e669cf56a
@ -7,6 +7,7 @@ import numpy as np
|
||||
|
||||
base_url = "http://127.0.0.1:8091/sdrangel"
|
||||
deviceset_url = ""
|
||||
verbosity = 2
|
||||
|
||||
requests_methods = {
|
||||
"GET": requests.get,
|
||||
@ -47,6 +48,11 @@ def getInputOptions():
|
||||
parser.add_option("-r", "--rf-bw", dest="rf_bw", help="RF babdwidth (Hz). Sets to nearest available", metavar="FREQUENCY", type="int", default=10000)
|
||||
parser.add_option("-c", "--create", dest="create", help="create a new device set", metavar="BOOLEAN", action="store_true", default=False)
|
||||
parser.add_option("-m", "--mock", dest="mock", help="just print calculated values and exit", metavar="BOOLEAN", action="store_true", default=False)
|
||||
parser.add_option("--ppm", dest="lo_ppm", help="LO correction in PPM", metavar="PPM", type="float", default=0.0)
|
||||
parser.add_option("-t", "--settling-time", dest="settling_time", help="Scan step settling time in seconds", metavar="SECONDS", type="float", default=1.0)
|
||||
parser.add_option("--sq", dest="squelch_db", help="Squelsch threshold in dB", metavar="DECIBEL", type="float", default=-50.0)
|
||||
parser.add_option("--sq-gate", dest="squelch_gate", help="Squelsch gate in ms", metavar="MILLISECONDS", type="int", default=50)
|
||||
parser.add_option("--re-run", dest="rerun", help="re run with given parameters without setting up device and channels", metavar="BOOLEAN", action="store_true", default=False)
|
||||
|
||||
(options, args) = parser.parse_args()
|
||||
|
||||
@ -79,11 +85,15 @@ def setupDevice(scan_control, options):
|
||||
settings['rtlSdrSettings']['gain'] = 496
|
||||
settings['rtlSdrSettings']['log2Decim'] = options.log2_decim
|
||||
settings['rtlSdrSettings']['dcBlock'] = 1
|
||||
settings['rtlSdrSettings']['iqImbalance'] = 1
|
||||
settings['rtlSdrSettings']['agc'] = 1
|
||||
settings['rtlSdrSettings']['loPpmCorrection'] = int(options.lo_ppm)
|
||||
settings['rtlSdrSettings']['rfBandwidth'] = scan_control.device_step_freq + 100000
|
||||
elif options.device_hwid == "HackRF":
|
||||
settings['hackRFInputSettings']['LOppmTenths'] = -51
|
||||
settings['hackRFInputSettings']['LOppmTenths'] = options.lo_ppm * 10 # in tenths of PPM
|
||||
settings['hackRFInputSettings']['centerFrequency'] = scan_control.device_start_freq
|
||||
settings['hackRFInputSettings']['dcBlock'] = 1
|
||||
settings['hackRFInputSettings']['iqImbalance'] = 1
|
||||
settings['hackRFInputSettings']['devSampleRate'] = scan_control.device_sample_rate
|
||||
settings['hackRFInputSettings']['lnaExt'] = 1
|
||||
settings['hackRFInputSettings']['lnaGain'] = 32
|
||||
@ -94,28 +104,60 @@ def setupDevice(scan_control, options):
|
||||
if r is None:
|
||||
exit(-1)
|
||||
|
||||
def changeDeviceFrequency(fc, options):
|
||||
settings = callAPI(deviceset_url + "/device/settings", "GET", None, None, "Get device settings")
|
||||
if settings is None:
|
||||
exit(-1)
|
||||
|
||||
if options.device_hwid == "LimeSDR":
|
||||
settings["limeSdrInputSettings"]["centerFrequency"] = fc + 500000
|
||||
elif options.device_hwid == "RTLSDR":
|
||||
settings['rtlSdrSettings']['centerFrequency'] = fc
|
||||
elif options.device_hwid == "HackRF":
|
||||
settings['hackRFInputSettings']['centerFrequency'] = fc
|
||||
|
||||
r = callAPI(deviceset_url + "/device/settings", "PATCH", None, settings, "Patch device center frequncy")
|
||||
if r is None:
|
||||
exit(-1)
|
||||
|
||||
# ======================================================================
|
||||
def setupChannels(scan_control, options):
|
||||
i = 0
|
||||
for shift in scan_control.channel_shifts:
|
||||
settings = callAPI("/deviceset/0/channel", "POST", None, {"channelType": "NFMDemod", "tx": 0}, "Create NFM demod")
|
||||
settings = callAPI(deviceset_url + "/channel", "POST", None, {"channelType": "NFMDemod", "tx": 0}, "Create NFM demod")
|
||||
if settings is None:
|
||||
exit(-1)
|
||||
|
||||
settings = callAPI("/deviceset/0/channel/%d/settings" % i, "GET", None, None, "Get NFM demod settings")
|
||||
settings = callAPI(deviceset_url + "/channel/%d/settings" % i, "GET", None, None, "Get NFM demod settings")
|
||||
if settings is None:
|
||||
exit(-1)
|
||||
|
||||
settings["NFMDemodSettings"]["inputFrequencyOffset"] = int(shift)
|
||||
settings["NFMDemodSettings"]["afBandwidth"] = options.af_bw * 1000
|
||||
settings["NFMDemodSettings"]["rfBandwidth"] = options.rf_bw
|
||||
settings["NFMDemodSettings"]["squelch"] = options.squelch_db * 10 # centi-Bels
|
||||
settings["NFMDemodSettings"]["squelchGate"] = options.squelch_gate / 10 # 10's of ms
|
||||
settings["NFMDemodSettings"]["title"] = "Channel %d" % i
|
||||
|
||||
r = callAPI("/deviceset/0/channel/%d/settings" % i, "PATCH", None, settings, "Change NFM demod")
|
||||
r = callAPI(deviceset_url + "/channel/%d/settings" % i, "PATCH", None, settings, "Change NFM demod")
|
||||
if r is None:
|
||||
exit(-1)
|
||||
|
||||
i += 1
|
||||
|
||||
# ======================================================================
|
||||
def checkScanning():
|
||||
reports = callAPI(deviceset_url + "/channels/report", "GET", None, None, "Get channels report")
|
||||
if reports is None:
|
||||
exit(-1)
|
||||
for i in range(reports["channelcount"]):
|
||||
channel = reports["channels"][i]
|
||||
if "report" in channel:
|
||||
if "NFMDemodReport" in channel["report"]:
|
||||
if channel["report"]["NFMDemodReport"]["squelch"] == 1:
|
||||
return False # stop scanning
|
||||
return True # continue scanning
|
||||
|
||||
# ======================================================================
|
||||
def printResponse(response):
|
||||
content_type = response.headers.get("Content-Type", None)
|
||||
@ -131,12 +173,16 @@ def callAPI(url, method, params, json, text):
|
||||
if request_method is not None:
|
||||
r = request_method(url=base_url+url, params=params, json=json)
|
||||
if r.status_code / 100 == 2:
|
||||
print(text + " succeeded")
|
||||
printResponse(r)
|
||||
if verbosity >= 1:
|
||||
print(text + " succeeded")
|
||||
if verbosity >= 2:
|
||||
printResponse(r)
|
||||
return r.json() # all 200 yield application/json response
|
||||
else:
|
||||
print(text + " failed")
|
||||
printResponse(r)
|
||||
if verbosity >= 1:
|
||||
print(text + " failed")
|
||||
if verbosity >= 2:
|
||||
printResponse(r)
|
||||
return None
|
||||
|
||||
# ======================================================================
|
||||
@ -145,6 +191,8 @@ def main():
|
||||
options = getInputOptions()
|
||||
scan_control = ScanControl(options.num_channels, options.freq_step, options.freq_start, options.freq_stop, options.log2_decim)
|
||||
|
||||
# Print calculated scan parameters
|
||||
|
||||
print("Channel shifts: %s" % scan_control.channel_shifts)
|
||||
print("Sample rate: %d" % scan_control.device_sample_rate)
|
||||
print("Start: %d" % scan_control.device_start_freq)
|
||||
@ -155,13 +203,17 @@ def main():
|
||||
print("Frequency error")
|
||||
exit(1)
|
||||
|
||||
if options.mock:
|
||||
freqs = []
|
||||
fc = scan_control.device_start_freq
|
||||
while fc <= scan_control.device_stop_freq:
|
||||
freqs += [x+fc for x in scan_control.channel_shifts]
|
||||
fc += scan_control.device_step_freq
|
||||
print("Scanned frequencies: %s" % freqs)
|
||||
freqs = []
|
||||
nb_steps = 1
|
||||
fc = scan_control.device_start_freq
|
||||
while fc <= scan_control.device_stop_freq:
|
||||
freqs += [x+fc for x in scan_control.channel_shifts]
|
||||
fc += scan_control.device_step_freq
|
||||
nb_steps += 1
|
||||
print("Scanned frequencies: %s" % freqs)
|
||||
print("In %d steps" % nb_steps)
|
||||
|
||||
if options.mock: # Stop there if we are just mocking (no API access)
|
||||
exit(0)
|
||||
|
||||
global base_url
|
||||
@ -169,37 +221,58 @@ def main():
|
||||
|
||||
# Set Rx
|
||||
|
||||
if options.create:
|
||||
r = callAPI("/deviceset", "POST", {"tx": 0}, None, "Add Rx device set")
|
||||
if r is None:
|
||||
exit(-1)
|
||||
|
||||
global deviceset_url
|
||||
deviceset_url = "/deviceset/%d" % options.device_index
|
||||
|
||||
r = callAPI(deviceset_url + "/device", "PUT", None, {"hwType": "%s" % options.device_hwid, "tx": 0}, "setup device on Rx device set")
|
||||
if not options.rerun: # Skip device and channels settings in re-run mode
|
||||
if options.create:
|
||||
r = callAPI("/deviceset", "POST", {"tx": 0}, None, "Add Rx device set")
|
||||
if r is None:
|
||||
exit(-1)
|
||||
|
||||
r = callAPI(deviceset_url + "/device", "PUT", None, {"hwType": "%s" % options.device_hwid, "tx": 0}, "setup device on Rx device set")
|
||||
if r is None:
|
||||
exit(-1)
|
||||
|
||||
# Set device and channels
|
||||
|
||||
setupDevice(scan_control, options)
|
||||
setupChannels(scan_control, options)
|
||||
|
||||
# Start running and scanning
|
||||
|
||||
r = callAPI(deviceset_url + "/device/run", "POST", None, None, "Start running device")
|
||||
if r is None:
|
||||
exit(-1)
|
||||
fc = scan_control.device_start_freq
|
||||
|
||||
setupDevice(scan_control, options)
|
||||
setupChannels(scan_control, options)
|
||||
global verbosity
|
||||
verbosity = 0
|
||||
|
||||
exit(0)
|
||||
|
||||
|
||||
|
||||
settings = callAPI("/deviceset/0/channel", "POST", None, {"channelType": "NFMDemod", "tx": 0}, "Create NFM demod")
|
||||
if settings is None:
|
||||
exit(-1)
|
||||
|
||||
settings["NFMDemodSettings"]["inputFrequencyOffset"] = 12500
|
||||
settings["NFMDemodSettings"]["afBandwidth"] = 5000
|
||||
|
||||
r = callAPI("/deviceset/0/channel/0/settings", "PATCH", None, settings, "Change NFM demod")
|
||||
if r is None:
|
||||
exit(-1)
|
||||
print("Move center to %d Hz" % fc)
|
||||
changeDeviceFrequency(fc, options)
|
||||
|
||||
try:
|
||||
while True:
|
||||
time.sleep(options.settling_time)
|
||||
if checkScanning(): # shall we move on ?
|
||||
fc += scan_control.device_step_freq
|
||||
if fc > scan_control.device_stop_freq:
|
||||
fc = scan_control.device_start_freq
|
||||
print("New pass")
|
||||
print("Move center to %d Hz" % fc)
|
||||
changeDeviceFrequency(fc, options)
|
||||
except KeyboardInterrupt:
|
||||
print("Terminated by user")
|
||||
pass
|
||||
finally:
|
||||
verbosity = 2
|
||||
r = callAPI(deviceset_url + "/device/run", "DELETE", None, None, "Stop running device")
|
||||
if r is None:
|
||||
exit(-1)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
except Exception, msg:
|
||||
tb = traceback.format_exc()
|
||||
print >> sys.stderr, tb
|
||||
|
||||
Loading…
Reference in New Issue
Block a user