import numpy as np
import cv2

# 读取视频文件
cap = cv2.VideoCapture('bike.mp4')

# 循环播放视频文件，同时显示原视频及其对应的灰度图
while cap.isOpened():
    # 逐帧读取视频，ret 为布尔值，表示是否成功读取帧，frame 为当前帧的图像数据
    ret, frame = cap.read()

    if not ret:
        print("没有内容，退出啦 :) ")
        break
    # 使用 cv2.cvtColor() 将当前帧的彩色图像转换为灰度图

    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # 平移前
    fcoef1 = np.fft.fft2(gray)
    amp_im_fft1 = np.log( np.abs(fcoef1))
    amp_im_fft1 = cv2.normalize(amp_im_fft1, None, 0, 255, cv2.NORM_MINMAX, dtype=cv2.CV_8U)
    # 平移后
    fCoef2 = np.fft.fftshift(np.fft.fft2(gray))
    amp_im_fft2 = np.log(np.abs(fCoef2))
    amp_im_fft2 = cv2.normalize(amp_im_fft2, None, 0, 255, cv2.NORM_MINMAX, dtype=cv2.CV_8U)
    # 在名为 "frame" 的窗口中显示灰度图
    cv2.imshow('frame', gray)

    # 在名为 "mg_spc" 的窗口中显示原振幅谱图
    cv2.imshow('mg_spc', amp_im_fft1)

    # 在名为 "mg_spc_shift" 的窗口中显示平移之后的振幅谱图
    cv2.imshow('mg_spc_shift', amp_im_fft2)

    # 每隔 1ms 检查一次用户输入，如果按下 'q' 键，退出循环
    if cv2.waitKey(1) == ord('q'):
        break

# 释放视频捕捉资源
cap.release()
# 关闭所有的 GUI 窗口
cv2.destroyAllWindows()