diff options
author | Arturs Artamonovs <dos21h@gmail.com> | 2023-03-06 22:56:47 +0000 |
---|---|---|
committer | Arturs Artamonovs <dos21h@gmail.com> | 2023-03-06 22:56:47 +0000 |
commit | 669926aa70eb78ea8f0ada46355a617abae61832 (patch) | |
tree | 66c026b1b03f5836847a84923334b87ffcc37aed | |
parent | cbecc12b023526e99bf6310fe26db17278f314a1 (diff) | |
download | pyairspyhf-669926aa70eb78ea8f0ada46355a617abae61832.tar.gz pyairspyhf-669926aa70eb78ea8f0ada46355a617abae61832.zip |
Move test code to airspyhf_rx.py
-rw-r--r-- | airspyhf_rx.py | 128 |
1 files changed, 128 insertions, 0 deletions
diff --git a/airspyhf_rx.py b/airspyhf_rx.py new file mode 100644 index 0000000..9770054 --- /dev/null +++ b/airspyhf_rx.py @@ -0,0 +1,128 @@ +import os +from airspyhf import * +from ctypes import * +import time +import sys +import wave +import struct +import argparse + +parser = argparse.ArgumentParser() +parser.add_argument("--frequency") +parser.add_argument("--samplerate") +parser.add_argument("--outputfile") +parser.add_argument("--serial") +args = parser.parse_args() + + + +print("Check airspyHF version") + +p = airspyhf_lib_version_t() +libairspyhf.airspyhf_lib_version(byref(p)) +print(f"libairspyhf version {p.major_version}.{p.minor_version}.{p.revision}") + +print("Get list of devices if there is any") +ndev = libairspyhf.airspyhf_list_devices(None,0) +print("Found %d devices"%(ndev)) + +for devi in range(0,ndev): + serial = c_uint64(0) + libairspyhf.airspyhf_list_devices(byref(serial),devi+1) + print("Device %d: Serial number %s"%(int(devi),hex(serial.value) )) + +print("try to open device") +dev_p = airspyhf_device_t_p(None) +ret = libairspyhf.airspyhf_open_sn(dev_p,0x3b52ab5dada12535) +print("open_sn: Returned %d"%(ret)) +if (ret != 0): + print("airspyhf_open_sn returned != 0, error") + sys.exit() + +print("List sample rates") +nsrates = c_uint32(0) + +ret = libairspyhf.airspyhf_get_samplerates(dev_p,byref(nsrates),c_uint32(0)) +print("ret %d"%ret) +print("sample rates %d"% nsrates.value) + +supportet_samplerates = (c_uint32*4)(0) +ret = libairspyhf.airspyhf_get_samplerates(dev_p,supportet_samplerates,nsrates) +print("ret %d"%ret) +for i in range(0,nsrates.value): + print("Sample rates %d"% supportet_samplerates[i]) + +#try to get some samples +ret = libairspyhf.airspyhf_set_samplerate(dev_p, supportet_samplerates[3]) +print(f"airspyhf_set_samplerate ret={ret}") + +ret = libairspyhf.airspyhf_set_hf_agc(dev_p, 1) +print(f"airspyhf_set_hf_agc ret={ret}") + +ret = libairspyhf.airspyhf_set_hf_agc_threshold(dev_p, 0) +print(f"airspyhf_set_hf_agc_threshold ret={ret}") + +sample_count = 0 +wave_file = wave.open("record.wav","w") +wave_file.setnchannels(2) +wave_file.setsampwidth(4) +wave_file.setframerate(supportet_samplerates[1]) +def read_samples(transfer): + global sample_count + global wave_file + #print("Python call back") + t = transfer.contents + bytes_to_write = t.sample_count * 4 * 2 + rx_buffer = t.samples + #print(f"{bytes_to_write} bytes receieved") + sample_count += t.sample_count + for i in range(0,t.sample_count): + d_re = t.samples[i].re + d_im = t.samples[i].im + data = struct.pack("<f",d_re) # FIX ?! + wave_file.writeframesraw(data) + data = struct.pack("<f", d_im) # FIX ?! + wave_file.writeframesraw(data) + #print("End call back") + return 0 + + +read_samples_cb = airspyhf_sample_block_cb_fn(read_samples) + + +ret = libairspyhf.airspyhf_start(dev_p, airspyhf_sample_block_cb_fn(read_samples), None) +print(f"airspyhf_start ret={ret}") + +#ret = airspyhf.libairspyhf.py_cb_wrapper(dev_p) +#print(f"airspyhf_start ret={ret}") + + +ret = libairspyhf.airspyhf_set_freq(dev_p, 3865000) +print(f"airspyhf_set_freq ret={ret}") + +count = 0 +try: + while (libairspyhf.airspyhf_is_streaming(dev_p)) and (count < 3): + print("Main loop") + time.sleep(1) + count += 1 +except: + print("Error in main loop") + +ret = libairspyhf.airspyhf_stop(dev_p) +print(f"airspyhf_stop ret={ret}") + +#Not close for now +ret = libairspyhf.close(dev_p) +print("closed: Returned %d"%(ret)) + +print(f"Total samples received {sample_count}") + +libairspyhf.py_test() + +libairspyhf.py_test_cb(read_samples_cb) + +wave_file.close() + +print("All is ok") + |