mirror of
https://github.com/jopohl/urh.git
synced 2026-03-10 18:26:48 +01:00
9f1c0341ed
* setup environment for native sdrplay support * prepare stream callback * add sdrplay class with error codes and pack complex method * add receive callback for sdrplay * add call in virtual device * simplify unpack complex method * improve initial gain setting * remove debug prints * sdrplay rx prototype * make protocol sniff settings collapsabile * move spectrum analyzer action to the top of signal actions * make device settings collapsable * improve collapse boxes * fix collapse icon * wrap live methods * fix layout * integrate methods * fix gil problems when live setting parameters * enable antenna selection for sdrplay * add optional dependency to readme * add winlibs * improve device list output * increase timeout for travis * consider changed set gr for rsp1
88 lines
3.4 KiB
Python
88 lines
3.4 KiB
Python
import unittest
|
|
import sys
|
|
import time
|
|
|
|
import numpy as np
|
|
|
|
from urh.util import util
|
|
|
|
util.set_windows_lib_path()
|
|
|
|
from urh.dev.native.RTLSDR import RTLSDR
|
|
try:
|
|
from urh.dev.native.lib import rtlsdr
|
|
except ImportError:
|
|
import urh.dev.native.lib.rtlsdr_fallback as rtlsdr
|
|
|
|
class TestRTLSDR(unittest.TestCase):
|
|
def test_cython_wrapper(self):
|
|
|
|
print("Device count:", rtlsdr.get_device_count())
|
|
print("Device name:", rtlsdr.get_device_name(0))
|
|
manufact, product, serial = rtlsdr.get_device_usb_strings(0)
|
|
print("Manufacturer:", manufact)
|
|
print("Product:", product)
|
|
print("Serial", serial)
|
|
print("Index by serial", rtlsdr.get_index_by_serial(serial))
|
|
print("Open:", rtlsdr.open(0))
|
|
print("Reset Buffer:", rtlsdr.reset_buffer()) # IMPORTANT
|
|
print("XTAL Freq:", rtlsdr.get_xtal_freq())
|
|
print("USB device strings", rtlsdr.get_usb_strings())
|
|
print("Center Freq:", rtlsdr.get_center_freq())
|
|
print("Set center freq to 433MHz", rtlsdr.set_center_freq(int(433e6)))
|
|
print("Center Freq:", rtlsdr.get_center_freq())
|
|
print("Freq Correction", rtlsdr.get_freq_correction())
|
|
print("Set Freq Correction to 10", rtlsdr.set_freq_correction(10))
|
|
print("Freq Correction", rtlsdr.get_freq_correction())
|
|
print("tuner type", rtlsdr.get_tuner_type())
|
|
print("tuner_gains", rtlsdr.get_tuner_gains())
|
|
print("set_manual_gain_mode", rtlsdr.set_tuner_gain_mode(1))
|
|
print("tuner gain", rtlsdr.get_tuner_gain())
|
|
print("set gain to 338", rtlsdr.set_tuner_gain(338))
|
|
print("tuner gain", rtlsdr.get_tuner_gain())
|
|
print("set tuner if gain", rtlsdr.set_tuner_if_gain(1, 10))
|
|
print("Sample Rate", rtlsdr.get_sample_rate())
|
|
print("Set Sample Rate to 300k", rtlsdr.set_sample_rate(300 * 10 ** 3))
|
|
print("Sample Rate", rtlsdr.get_sample_rate())
|
|
read_samples = rtlsdr.read_sync(1024)
|
|
print(read_samples)
|
|
rtlsdr.close()
|
|
|
|
def test_receive(self):
|
|
rtlsdr_class = RTLSDR(433.92e6, 20, 2e6, device_number=0)
|
|
self.assertEqual(rtlsdr_class.current_recv_index, 0)
|
|
rtlsdr_class.start_rx_mode()
|
|
time.sleep(2)
|
|
rtlsdr_class.stop_rx_mode("Finished")
|
|
index = rtlsdr_class.current_recv_index
|
|
print(rtlsdr_class.current_recv_index)
|
|
self.assertGreater(rtlsdr_class.current_recv_index, 0)
|
|
time.sleep(2)
|
|
self.assertEqual(index, rtlsdr_class.current_recv_index)
|
|
rtlsdr_class.start_rx_mode()
|
|
time.sleep(1)
|
|
self.assertGreater(rtlsdr_class.current_recv_index, index)
|
|
|
|
def test_pack_unpack_complex(self):
|
|
arr = np.array([0, 0, 127.5, 127.5, 255, 255], dtype=np.uint8)
|
|
self.assertEqual(arr[0], 0)
|
|
self.assertEqual(arr[1], 0)
|
|
self.assertEqual(arr[-1], 255)
|
|
|
|
received = arr.tostring()
|
|
self.assertEqual(len(received), len(arr))
|
|
self.assertEqual(np.int8(received[0]), 0)
|
|
self.assertEqual(np.int8(received[1]), 0)
|
|
unpacked = RTLSDR.unpack_complex(received, len(received) // 2)
|
|
|
|
self.assertEqual(unpacked[0], complex(-1, -1))
|
|
self.assertAlmostEqual(unpacked[1], complex(0, 0), places=1)
|
|
self.assertEqual(unpacked[2], complex(1, 1))
|
|
|
|
packed = RTLSDR.pack_complex(unpacked)
|
|
self.assertEqual(received, packed)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|