水声多径信道下LMS仿真(Python代码)

1. 模块的导入

做仿真，numpy应该都知道
arlpy是水声通信工具箱，不仅可以产生信道的冲激响应，还有一些通信相关的函数
bokeh是用来画图的，配合jupyter notebook使用很爽！！（Matplotlib是常用的绘图工具包，个人感觉bokeh更舒服…）

# 数组运算工具包
import numpy as np
# 水声通信工具包
import arlpy.uwa as au
import arlpy.uwapm as aup
import arlpy.comms as ac
# 绘图工具包
from bokeh.plotting import show, figure
from bokeh.layouts import gridplot
from bokeh.io import output_notebook
output_notebook()
import warnings
warnings.filterwarnings('ignore') 

2. 准备工作

2.1 水声信道的冲激响应

如何设置参数，得到冲激响应、衰减、时延等等在arlpy的官方文档中，都有介绍，可自行查阅
https://arlpy.readthedocs.io/en/latest/

# 设置环境参数
env = aup.create_env2d()
env['bottom_soundspeed'] = 1800
env['bottom_absorption'] = 0
env['depth'] = 100
env['max_angle'] = 30
env['min_angle'] = -30
env['frequency'] = 100
env['nbeams'] = 10
env['tx_depth'] = 20
env['rx_depth'] = 20
env['rx_range'] = 1000
# 假设声速在浅海环境下是递增的
sound_speed = np.zeros(shape=[6, 2])
for i in range(6):
    sound_speed[i, 0] = 20 * i
    sound_speed[i, 1] = au.soundspeed(depth=20 * i)
env['soundspeed'] = sound_speed

arrivals_sd = aup.compute_arrivals(env) # 声线到达的时间
delay_sd = min(arrivals_sd.time_of_arrival) # 端到端的传输时延
multidelay_sd = max(arrivals_sd.time_of_arrival) - \
    min(arrivals_sd.time_of_arrival) # 多径时延
fs = 100  # 信道的采样频率
cir_sd = aup.arrivals_to_impulse_response(arrivals_sd, fs=fs)
cir_sd = np.real(cir_sd)
cir_sd[(cir_sd < 0)] *= -1 # 信道的冲激响应

2.2 归一化及均衡代码

考虑到信道衰减比较大，在对信号进行处理的时候，需要考虑归一化的问题

def nor(x):
    '''
    param x: 带归一化信号
    return: 已归一化信号
    '''
    alpha = np.sum(np.square(x))/x.size
    return x/np