RTSP视频流编解码过程中计算PSNR指标，C代码

在RTSP（Real-Time Streaming Protocol）视频流处理中，PSNR（Peak Signal-to-Noise Ratio）通常用于评估源图像质量和解码后的重建图像的质量。PSNR是一个衡量两个信号之间相似度的指标，数值越高，表示图像质量越好。

以下是一个简单的C代码片段，用于计算单通道（灰度）图像的PSNR：

#include <stdio.h>
#include <math.h>

double calculate_PSNR(const uint8_t *source, const uint8_t *reference, int width, int height) {
    double mse = 0.0;
    int pixel_count = width * height;

    for (int y = 0; y < height; y++) {
        for (int x = 0; x < width; x++) {
            int bgr_error = source[y * width + x] - reference[y * width + x];
            mse += (bgr_error * bgr_error);
        }
    }

    mse /= pixel_count;
    double psnr = 20 * log10((double)(MAX_VALUE - 1) / sqrt(mse));
    return psnr;
}

// MAX_VALUE常用于计算最大可能误差，这里假设为255（对于8位无损编码）
#define MAX_VALUE 255

int main() {
    // 将解码后的像素数据和原始像素数据传递给calculate_PSNR函数
    uint8_t decoded_image[width*height], original_image[width*height];
    // ...填充解码后的数据...
    // ...获取原始数据...

    double psnr = calculate_PSNR(decoded_image, original_image, width, height);
    printf("PSNR: %.2f dB\n", psnr);

    return 0;
}

这个例子假设你已经有了解码后的图像（decoded_image）和原始图像（original_image）。实际应用中，你需要从RTSP流中获取原始图像，然后与解码后的图像进行比较。