用100行左右Python进行OQAM_OFDM通感一体化波形设计

在Python中设计OQAM (Offset Quadrature Amplitude Modulation) OFDM (Orthogonal Frequency Division Multiplexing) 的一体化学术通信波形通常需要一些专门的库，如numpy、scipy以及matplotlib等。下面是一个简化版的100行左右的示例，这个例子并不包含完整的信号处理流程，仅展示了基本的OQAM变换和OFDM传输部分。实际应用中可能需要更多的代码来设置符号率、载波数、循环前缀等。

import numpy as np
import scipy.signal as sig
import matplotlib.pyplot as plt

# 设定参数
symbol_rate = 1e6  # 符号速率
num_subcarriers = 512  # 子载波数目
ofdm_symbols = 100  # OFDM符号数
cp_len = num_subcarriers // 4  # 循环前缀长度

def create_carrier(carrier_freq):
    return np.arange(-num_subcarriers//2, num_subcarriers//2) * carrier_freq / symbol_rate

def apply_ofdm(carrier, data, cp=True):
    ofdm = []
    for i in range(ofdm_symbols):
        if cp:
            ofdm.append(np.concatenate([carrier, np.zeros(cp_len), data[i:i+num_subcarriers-cp_len]]))
        else:
            ofdm.append(carrier + data[i])
    return ofdm

# 创建数据和随机序列
data = np.random.randint(2, size=(ofdm_symbols, num_subcarriers//2)) * 2 - 1  # I/Q数据对

# OQAM变换
iq_data = np.zeros((ofdm_symbols, num_subcarriers), dtype=complex)
iq_data[:, ::2] = data
iq_data[:, 1::2] *= 1j

# 应用OFDM
ofdm_waveform = apply_ofdm(create_carrier(0), iq_data)

# 将每个OFDM符号显示出来
fig, axs = plt.subplots(nrows=ofdm_symbols, ncols=2, figsize=(10, 10*ofdm_symbols/8))
for ax, subcarrier in zip(axs.flat, ofdm_waveform):
    ax.plot(subcarrier)
    ax.set_xlabel('Samples')
    ax.set_ylabel(f'Subcarrier {ax.get_position().y0:.1f}')
plt.tight_layout()
plt.show()