Delete beamsteer.py
Jon Kraft
GitHub
parent
f6a8208079
commit
42ffe10434
248
beamsteer.py
248
beamsteer.py
|
|
@ -1,248 +0,0 @@
|
|||
#!/usr/bin/env python3
|
||||
# Must use Python 3
|
||||
# Copyright (C) 2022 Analog Devices, Inc.
|
||||
#
|
||||
# All rights reserved.
|
||||
#
|
||||
# Redistribution and use in source and binary forms, with or without modification,
|
||||
# are permitted provided that the following conditions are met:
|
||||
# - Redistributions of source code must retain the above copyright
|
||||
# notice, this list of conditions and the following disclaimer.
|
||||
# - Redistributions in binary form must reproduce the above copyright
|
||||
# notice, this list of conditions and the following disclaimer in
|
||||
# the documentation and/or other materials provided with the
|
||||
# distribution.
|
||||
# - Neither the name of Analog Devices, Inc. nor the names of its
|
||||
# contributors may be used to endorse or promote products derived
|
||||
# from this software without specific prior written permission.
|
||||
# - The use of this software may or may not infringe the patent rights
|
||||
# of one or more patent holders. This license does not release you
|
||||
# from the requirement that you obtain separate licenses from these
|
||||
# patent holders to use this software.
|
||||
# - Use of the software either in source or binary form, must be run
|
||||
# on or directly connected to an Analog Devices Inc. component.
|
||||
#
|
||||
# THIS SOFTWARE IS PROVIDED BY ANALOG DEVICES "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
|
||||
# INCLUDING, BUT NOT LIMITED TO, NON-INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A
|
||||
# PARTICULAR PURPOSE ARE DISCLAIMED.
|
||||
#
|
||||
# IN NO EVENT SHALL ANALOG DEVICES BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, INTELLECTUAL PROPERTY
|
||||
# RIGHTS, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
|
||||
# BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
|
||||
# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
|
||||
# THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
''' Simple Beamforming Example Using Phaser and Python'''
|
||||
|
||||
# =============================================================================
|
||||
# Import statements
|
||||
# =============================================================================
|
||||
import adi
|
||||
import ADAR_pyadi_functions as ADAR # import the ADAR1000 functions
|
||||
import SDR_functions as SDR # import the Pluto SDR functions
|
||||
|
||||
import sys
|
||||
import pickle
|
||||
import matplotlib.pyplot as plt
|
||||
import numpy as np
|
||||
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# User parameters
|
||||
# =============================================================================
|
||||
|
||||
rpi_ip = "ip:phaser.local" # default IP address of Phaser's Raspberry Pi
|
||||
sdr_ip = "ip:192.168.2.1" # default Pluto IP address
|
||||
|
||||
# select which signal source to use
|
||||
# HB100 (external source)
|
||||
# OUT1 (transmit freq is set in config.py)
|
||||
# OUT2 (transmit freq is set in config.py)
|
||||
SignalSource = 'HB100' # 'HB100', 'OUT1', or 'OUT2'
|
||||
|
||||
# config.py has all the key parameters that you might want to modify
|
||||
try:
|
||||
import config as config
|
||||
except:
|
||||
print("Make sure config.py is in this directory")
|
||||
sys.exit(0)
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# Variables setup
|
||||
# =============================================================================
|
||||
|
||||
# if using HB100, load the signal frequency from "phaser_find_hb100.py" output file
|
||||
if SignalSource == 'HB100':
|
||||
try:
|
||||
with open("hb100_freq_val.pkl", "rb") as file1:
|
||||
config.SignalFreq = pickle.load(file1)
|
||||
print("Found signal freq file, ", config.SignalFreq/1e9, " GHz")
|
||||
except:
|
||||
print("No signal freq found, keeping at ", config.SignalFreq/1e9, " GHz")
|
||||
|
||||
"""SET DEFAULT VALUES"""
|
||||
sdr_address = sdr_ip
|
||||
SignalFreq = config.SignalFreq
|
||||
Tx_freq = config.Tx_freq # Pluto's Tx LO freq.
|
||||
Rx_freq = config.Rx_freq # Pluto's Rx LO freq
|
||||
LO_freq = SignalFreq + Rx_freq # freq of the LTC5548 mixer LO
|
||||
SampleRate = config.SampleRate
|
||||
Rx_gain = config.Rx_gain
|
||||
Tx_gain = config.Tx_gain
|
||||
RxGain1 = 100
|
||||
RxGain2 = 100
|
||||
RxGain3 = 100
|
||||
RxGain4 = 100
|
||||
RxGain5 = 100
|
||||
RxGain6 = 100
|
||||
RxGain7 = 100
|
||||
RxGain8 = 100
|
||||
RxPhase1 = config.Rx1_cal
|
||||
RxPhase2 = config.Rx2_cal
|
||||
RxPhase3 = config.Rx3_cal
|
||||
RxPhase4 = config.Rx4_cal
|
||||
RxPhase5 = config.Rx5_cal
|
||||
RxPhase6 = config.Rx6_cal
|
||||
RxPhase7 = config.Rx7_cal
|
||||
RxPhase8 = config.Rx8_cal
|
||||
phase_step_size = 2.8125
|
||||
c = 299792458 # speed of light in m/s
|
||||
d = config.d # antenna spacing for phaser is 14mm
|
||||
gainList = [RxGain1, RxGain2, RxGain3, RxGain4,
|
||||
RxGain5, RxGain6, RxGain7, RxGain8]
|
||||
phaseList = [RxPhase1, RxPhase2, RxPhase3, RxPhase4,
|
||||
RxPhase5, RxPhase6, RxPhase7, RxPhase8]
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# Hardware setup
|
||||
# =============================================================================
|
||||
|
||||
# Use the onboard VCO to generate the LO? Or apply source to EXT_LO?
|
||||
gpios = adi.one_bit_adc_dac(rpi_ip)
|
||||
gpios.gpio_vctrl_1 = 1 # 1=Use onboard PLL/LO source (0=use external LO input)
|
||||
gpios.gpio_vctrl_2 = 1 # 1=Send LO to transmit circuitry (0=disable Tx path and send LO to LO_OUT)
|
||||
|
||||
# setup GPIOs to control if Tx is output on OUT1 or OUT2
|
||||
gpios.gpio_div_mr = 1
|
||||
gpios.gpio_div_s0 = 0
|
||||
gpios.gpio_div_s1 = 0
|
||||
gpios.gpio_div_s2 = 0
|
||||
|
||||
# Initialize Pluto
|
||||
sdr = SDR.SDR_init(
|
||||
sdr_address,
|
||||
SampleRate,
|
||||
Tx_freq,
|
||||
Rx_freq,
|
||||
Rx_gain,
|
||||
Tx_gain,
|
||||
config.buffer_size,
|
||||
)
|
||||
SDR.SDR_LO_init(rpi_ip, LO_freq) # Set Phaser's ADF4159 to the LO_freq
|
||||
|
||||
# Intialize the ADAR1000 receive array
|
||||
rx_array = adi.adar1000_array(
|
||||
uri=rpi_ip,
|
||||
chip_ids=["BEAM0", "BEAM1"], # these are the ADAR1000s' labels in the device tree
|
||||
device_map=[[1], [2]],
|
||||
element_map=[[1, 2, 3, 4, 5, 6, 7, 8]],
|
||||
device_element_map={
|
||||
1: [7, 8, 5, 6], # i.e. channel2 of device1 (BEAM0), maps to element 8
|
||||
2: [3, 4, 1, 2],
|
||||
},
|
||||
)
|
||||
for device in rx_array.devices.values():
|
||||
ADAR.ADAR_init(device) # resets the ADAR1000
|
||||
ADAR.ADAR_set_mode(device, "rx") # ADAR1000s on Phaser are receive only, so mode is always "rx"
|
||||
ADAR.ADAR_set_Taper(
|
||||
rx_array,
|
||||
gainList
|
||||
)
|
||||
|
||||
# Set transmitter to either OUT1 or OUT2 SMA port. Or disable if using HB100
|
||||
if SignalSource == 'OUT1': # use Phaser's OUT1 SMA port as the transmitter
|
||||
gpios.gpio_tx_sw = 1 # 0=OUT2, 1=OUT1
|
||||
gpios.gpio_vctrl_2 = 1 # 1=Send LO to transmit circuitry
|
||||
elif SignalSource == 'OUT2': # use OUT2 as the transmitter
|
||||
gpios.gpio_tx_sw = 0 # 0=OUT2, 1=OUT1
|
||||
gpios.gpio_vctrl_2 = 1 # 1=Send LO to transmit circuitry
|
||||
else: # use HB100 as the transmit signal source
|
||||
gpios.gpio_tx_sw = 0
|
||||
SDR.SDR_setTx(sdr, -80) # disable tx output by attenuating it
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# Define Common Functions
|
||||
# =============================================================================
|
||||
def ConvertPhaseToSteerAngle(PhDelta):
|
||||
# steering angle theta = arcsin(c*deltaphase/(2*pi*f*d)
|
||||
value1 = (c * np.radians(np.abs(PhDelta))) / (
|
||||
2 * 3.14159 * (SignalFreq) * d)
|
||||
clamped_value1 = max(min(1, value1), -1) # arcsin argument must be between 1 and -1
|
||||
theta = np.degrees(np.arcsin(clamped_value1))
|
||||
if PhDelta >= 0:
|
||||
SteerAngle = theta # positive PhaseDelta covers 0deg to 90 deg
|
||||
else:
|
||||
SteerAngle = -theta # negative phase delta covers 0 deg to -90 deg
|
||||
return SteerAngle
|
||||
|
||||
def dbfs(raw_data):
|
||||
# function to convert IQ samples to FFT plot, scaled in dBFS
|
||||
NumSamples = len(raw_data)
|
||||
win = np.hamming(NumSamples)
|
||||
y = raw_data * win
|
||||
s_fft = np.fft.fft(y) / np.sum(win)
|
||||
s_shift = np.fft.fftshift(s_fft)
|
||||
s_dbfs = 20*np.log10(np.abs(s_shift)/(2**11)) # Pluto is a signed 12 bit ADC, so use 2^11 to convert to dBFS
|
||||
return s_dbfs
|
||||
|
||||
|
||||
# =============================================================================================
|
||||
# Loop through all the steering angles and record the peak FFT amplitude at each steering angle
|
||||
# =============================================================================================
|
||||
angles = [] # stores the list of steering angles
|
||||
peak_gains = [] # stores the peak FFT gain received for each steering angle
|
||||
|
||||
steering_step = 1 # steering angle step size (in degrees)
|
||||
SteerValues = np.arange(-90, 90 + steering_step, steering_step)
|
||||
# Phase delta = 2*Pi*d*sin(theta)/lambda = 2*Pi*d*sin(theta)*f/c
|
||||
PhaseValues = np.degrees(
|
||||
2*np.pi*d* np.sin(np.radians(SteerValues))
|
||||
* SignalFreq / c
|
||||
)
|
||||
|
||||
for PhDelta in PhaseValues:
|
||||
ADAR.ADAR_set_Phase(
|
||||
rx_array,
|
||||
PhDelta,
|
||||
phase_step_size,
|
||||
phaseList
|
||||
)
|
||||
|
||||
data = sdr.rx()
|
||||
data_sum = data[0]+data[1]
|
||||
sum_dbfs = dbfs(data_sum)
|
||||
peak_dbfs = max(sum_dbfs)
|
||||
angles.append(ConvertPhaseToSteerAngle(PhDelta))
|
||||
peak_gains.append(peak_dbfs)
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# Plotting results
|
||||
# =============================================================================
|
||||
|
||||
plt.figure(1)
|
||||
plt.subplot(2, 1, 1)
|
||||
plt.title("Beam sweep plot")
|
||||
plt.plot(angles, peak_gains, marker="o", ms=2)
|
||||
plt.xlabel("Steering angle (deg)")
|
||||
plt.ylabel("Peak Amplitude (dBFS)")
|
||||
plt.tight_layout()
|
||||
plt.show()
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue