乘波前行的问题

子正

已于 2023-11-27 16:30:38 修改

阅读量226

点赞数

分类专栏：问题建模 # 测量 # Python俱乐部文章标签： python 开发语言

于 2023-11-26 14:51:15 首次发布

本文链接：https://blog.csdn.net/twicave/article/details/134627656

版权

问题建模同时被 3 个专栏收录

88 篇文章 1 订阅

订阅专栏

测量

69 篇文章 0 订阅

订阅专栏

Python俱乐部

32 篇文章 0 订阅

订阅专栏

1.问题：

考虑两个信号叠加在一起，比如，一个是工频信号50Hz，一个是叠加的高频信号比如有3KHz，简单起见，两个信号都是幅值固定的标准的正弦波，现在我们期望得到那个高频信号，相对工频信号的时域波形，它会是什么样的？它的幅值会抖动吗？它会发生类似红移的效应吗？你怎么验算这个操作？

2.FirstTry: 带通滤波

接下来我们打算这样做，我们对两个信号之外的部分，完全滤掉低频和高频部分（带通滤波），然后还原会时域，我们想看看此时的时域信号会有什么特征。

2.1 代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# paramsIn => resultOut 
#
import json
import binascii
import numpy as np;
import sys
import struct
import numpy as np
import matplotlib.pyplot as plt
import signal_2048_two_sin

# 绘制信号和包络
def showTwoPairSignals(t, x, spectrum_before, amplitude_envelope, spectrum_after, freqScale, fs):
    freq = np.arange(0, 1024*(10000.0/2048), 10000.0/2048)
    fig, axs = plt.subplots(2, 2, figsize=(10, 8))
    axs[0, 0].plot(t, x)
    axs[0, 0].set_xlabel('Time (s)')
    axs[0, 0].set_ylabel('Amplitude')
    axs[0, 0].set_title('Original Signal')

    axs[1, 0].plot(freq, spectrum_before)
    axs[1, 0].set_xlabel('Frequency (Hz)')
    axs[1, 0].set_ylabel('Amplitude')
    axs[1, 0].set_title('Spectrum Before Transformation')
    axs[1, 0].set_xlim(0, fs/2)

    axs[0, 1].plot(t, amplitude_envelope)
    axs[0, 1].set_xlabel('Time (s)')
    axs[0, 1].set_ylabel('Amplitude')
    axs[0, 1].set_title('Amplitude Envelope')

    axs[1, 1].plot(freq, spectrum_after)
    axs[1, 1].set_xlabel('Frequency (Hz)')
    axs[1, 1].set_ylabel('Amplitude')
    axs[1, 1].set_title('Spectrum After Transformation')
    axs[1, 1].set_xlim(0, fs/2)

    plt.tight_layout()
    plt.show()

def calc_envelope(fft_complex_result, scale_fft, center_fft, zoomOfFreq):
    print(len(fft_complex_result))
    data_array = fft_complex_result;
    center_freq = center_fft/scale_fft;
    zoomOfFreq = zoomOfFreq/scale_fft;


    #频谱进行数字带通滤波
    min_freq = center_freq - zoomOfFreq;
    max_freq = center_freq + zoomOfFreq;
    if(min_freq <0): min_freq = 0;
    if(max_freq>=(len(data_array)//2)): max_freq =len(data_array)//2;
    for i in range(len(data_array)//2): #注意data_array包含对称的频谱。
        if(i<min_freq): 
            data_array[i] = 0+0j;
            data_array[-1*i] = 0+0j;
        else:
            if(i>max_freq): 
                data_array[i] = 0+0j;
                data_array[-1*i] = 0+0j;
            #no else

    # 幅角变换
    X = data_array;
    phase_shift = np.zeros(len(X))
    for idx in np.arange(0, len(X)//2):
        phase_shift[idx] = -np.angle(X[-idx])
    X_corrected = X * np.exp( 1j * phase_shift)
    
    # 转回时域
    sampleDataFilted  = np.real(np.fft.ifft(X_corrected)); #我不知道为什么不取abs，但是既有的电机时域数据去零点的结果是正确的。
    amplitude_spectrum = [(abs(x)/len(sampleDataFilted)) for x in sampleDataFilted]
    amplitude_spectrum = np.clip(amplitude_spectrum, 0, 65535).astype(np.uint16)

    # 傅里叶变换
    X_of_Envelope = np.fft.fft(amplitude_spectrum)  # 计算原始信号的傅里叶变换
    X_of_Envelope = X_of_Envelope[:len(X_of_Envelope)//2]
    absFFTOfEnvelope = [abs(item)/len(X_of_Envelope) for item in X_of_Envelope];
        
    # to hex again 
    # 将浮点数组转换为 2 字节uint16编码的二进制数据
    ba = amplitude_spectrum.tobytes()
    hex_data = binascii.hexlify(ba).decode('utf-8')
    root = {
        'binData':hex_data,
        'pt':len(hex_data)/2/2, #hex=>byte /2 byte=>uint16 /2
        'byte_len': len(hex_data),
        'scale': (scale_fft*(len(hex_data)/2))
    }

    # out put  json obj: root is ready 
    # it should turn into str => python variable: resultOut
    #######################################################################################

    # json => str
    json_string = json.dumps(root)
    print(json_string);
    resultOut = json_string;
    return (json_string, amplitude_spectrum, absFFTOfEnvelope);

(signal, FFT_result, FFT_abs) = signal_2048_two_sin.genSignal(2048, 2000, 1800)
(envelopeHex, envelope_result, absFFTOfEnvelope) = calc_envelope(FFT_result, 10000/2048., 2000, 250);
print(envelope_result)


showTwoPairSignals(np.arange(0, 2048.0/10000, 1/10000), signal, FFT_abs[:len(FFT_abs)//2], envelope_result, absFFTOfEnvelope, 10000.0/2048, 10000.0)



#其中：signal_2048_two_sin.genSignal(2048, 2000, 1800)的代码参见下文

import math
import wave
import struct
import numpy as np
import matplotlib.pyplot as plt


def genSignal(pt, freq1, freq2):
    # 采样率和采样点数
    sample_rate = 10000
    num_samples = 2048

    # 信号频率和振幅
    frequency_1 = freq1
    frequency_2 = freq2
    amplitude = 0.5

    # 计算每个采样点的时间间隔
    time_interval = 1 / sample_rate

    # 初始化采样点列表
    samples = []
    signal = []
    # 生成采样点
    for i in range(num_samples):
        t = i * time_interval
        value = amplitude * math.sin(2 * math.pi * frequency_1 * t) + amplitude * math.sin(2 * math.pi * frequency_2 * t)
        scaled_value = round(value * 32767)  # 缩放到S16范围
        if(scaled_value>=32767):
            scaled_value = 32767
        if(scaled_value<=-32767):
            scaled_value = 32767
        signal.append(scaled_value)
        samples.append(struct.pack('<h', scaled_value))  # 打包为16位整数，并添加到采样点列表

    # 保存到WAV文件
    wave_file = wave.open('output.wav', 'wb')
    wave_file.setnchannels(1)
    wave_file.setsampwidth(2)
    wave_file.setframerate(sample_rate)
    wave_file.writeframes(b''.join(samples))
    wave_file.close()

    t = np.arange(0, len(signal));
    fig = plt.figure(1, figsize=(8, 10));
    plt1 = fig.add_subplot(2, 1, 1)
    plt1.plot(t,signal)
    plt1.set_xlabel('Time (s)')
    plt1.set_ylabel('Amplitude')
    plt1.set_title('Original Signal')


    # 傅里叶变换
    X = np.fft.fft(signal)  # 计算原始信号的傅里叶变换
    freq = np.arange(0,len(X))
    absFFT = [abs(item) for item in X]; #*2/len(X)
    plt2 = fig.add_subplot(2, 1, 2)
    plt2.plot(freq, absFFT)
    plt2.set_xlabel('Freq (?Hz)')
    plt2.set_ylabel('Amplitude')
    plt2.set_title('FFT')
    fig.show()
    return (signal, X, absFFT);