| Index: webrtc/tools/frame_analyzer/reference_less_video_analysis_lib.cc
|
| diff --git a/webrtc/tools/frame_analyzer/reference_less_video_analysis_lib.cc b/webrtc/tools/frame_analyzer/reference_less_video_analysis_lib.cc
|
| deleted file mode 100644
|
| index ab484e9bb52b96eeac84d7d9723ccf99615649df..0000000000000000000000000000000000000000
|
| --- a/webrtc/tools/frame_analyzer/reference_less_video_analysis_lib.cc
|
| +++ /dev/null
|
| @@ -1,202 +0,0 @@
|
| -/*
|
| - * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
|
| - *
|
| - * Use of this source code is governed by a BSD-style license
|
| - * that can be found in the LICENSE file in the root of the source
|
| - * tree. An additional intellectual property rights grant can be found
|
| - * in the file PATENTS. All contributing project authors may
|
| - * be found in the AUTHORS file in the root of the source tree.
|
| - */
|
| -#include <stdio.h>
|
| -#include <stdlib.h>
|
| -#include <string.h>
|
| -#include <iostream>
|
| -#include <numeric>
|
| -#include <vector>
|
| -
|
| -#include "webrtc/tools/frame_analyzer/reference_less_video_analysis_lib.h"
|
| -#include "webrtc/tools/frame_analyzer/video_quality_analysis.h"
|
| -
|
| -#define STATS_LINE_LENGTH 28
|
| -#define PSNR_FREEZE_THRESHOLD 47
|
| -#define SSIM_FREEZE_THRESHOLD .999
|
| -
|
| -#if defined(_WIN32) || defined(_WIN64)
|
| -#define strtok_r strtok_s
|
| -#endif
|
| -
|
| -void get_height_width_fps(int *height, int *width, int *fps,
|
| - const std::string& video_file) {
|
| - // File header looks like :
|
| - // YUV4MPEG2 W1280 H720 F25:1 Ip A0:0 C420mpeg2 XYSCSS=420MPEG2.
|
| - char frame_header[STATS_LINE_LENGTH];
|
| - FILE* input_file = fopen(video_file.c_str(), "rb");
|
| -
|
| - size_t bytes_read = fread(frame_header, 1, STATS_LINE_LENGTH - 1, input_file);
|
| -
|
| - frame_header[bytes_read] = '\0';
|
| - std::string file_header_stats[5];
|
| - int no_of_stats = 0;
|
| - char *save_ptr;
|
| - char *token = strtok_r(frame_header, " ", &save_ptr);
|
| -
|
| - while (token != NULL) {
|
| - file_header_stats[no_of_stats++] = token;
|
| - token = strtok_r(NULL, " ", &save_ptr);
|
| - }
|
| -
|
| - *width = std::stoi(file_header_stats[1].erase(0, 1));
|
| - *height = std::stoi(file_header_stats[2].erase(0, 1));
|
| - *fps = std::stoi(file_header_stats[3].erase(0, 1));
|
| -
|
| - printf("Height: %d Width: %d fps:%d \n", *height, *width, *fps);
|
| - fclose(input_file);
|
| -}
|
| -
|
| -bool frozen_frame(std::vector<double> psnr_per_frame,
|
| - std::vector<double> ssim_per_frame, size_t frame) {
|
| - if (psnr_per_frame[frame] >= PSNR_FREEZE_THRESHOLD ||
|
| - ssim_per_frame[frame] >= SSIM_FREEZE_THRESHOLD)
|
| - return true;
|
| - return false;
|
| -}
|
| -
|
| -std::vector<int> find_frame_clusters(const std::vector<double>& psnr_per_frame,
|
| - const std::vector<double>& ssim_per_frame) {
|
| - std::vector<int> identical_frame_clusters;
|
| - int num_frozen = 0;
|
| - size_t total_no_of_frames = psnr_per_frame.size();
|
| -
|
| - for (size_t each_frame = 0; each_frame < total_no_of_frames; each_frame++) {
|
| - if (frozen_frame(psnr_per_frame, ssim_per_frame, each_frame)) {
|
| - num_frozen++;
|
| - } else if (num_frozen > 0) {
|
| - // Not frozen anymore.
|
| - identical_frame_clusters.push_back(num_frozen);
|
| - num_frozen = 0;
|
| - }
|
| - }
|
| - return identical_frame_clusters;
|
| -}
|
| -
|
| -void print_freezing_metrics(const std::vector<double>& psnr_per_frame,
|
| - const std::vector<double>& ssim_per_frame) {
|
| - /*
|
| - * Prints the different metrics mainly:
|
| - * 1) Identical frame number, PSNR and SSIM values.
|
| - * 2) Length of continuous frozen frames.
|
| - * 3) Max length of continuous freezed frames.
|
| - * 4) No of unique frames found.
|
| - * 5) Total different identical frames found.
|
| - *
|
| - * Sample output:
|
| - * Printing metrics for file: /src/webrtc/tools/test_3.y4m
|
| - =============================
|
| - Total number of frames received: 74
|
| - Total identical frames: 5
|
| - Number of unique frames: 69
|
| - Printing Identical Frames:
|
| - Frame Number: 29 PSNR: 48.000000 SSIM: 0.999618
|
| - Frame Number: 30 PSNR: 48.000000 SSIM: 0.999898
|
| - Frame Number: 60 PSNR: 48.000000 SSIM: 0.999564
|
| - Frame Number: 64 PSNR: 48.000000 SSIM: 0.999651
|
| - Frame Number: 69 PSNR: 48.000000 SSIM: 0.999684
|
| - Print identical frame which appears in clusters :
|
| - 2 1 1 1
|
| - *
|
| - */
|
| - size_t total_no_of_frames = psnr_per_frame.size();
|
| - std::vector<int> identical_frame_clusters = find_frame_clusters(
|
| - psnr_per_frame, ssim_per_frame);
|
| - int total_identical_frames = std::accumulate(
|
| - identical_frame_clusters.begin(), identical_frame_clusters.end(), 0);
|
| - size_t unique_frames = total_no_of_frames - total_identical_frames;
|
| -
|
| - printf("Total number of frames received: %zu\n", total_no_of_frames);
|
| - printf("Total identical frames: %d\n", total_identical_frames);
|
| - printf("Number of unique frames: %zu\n", unique_frames);
|
| -
|
| - printf("Printing Identical Frames: \n");
|
| - for (size_t frame = 0; frame < total_no_of_frames; frame++) {
|
| - if (frozen_frame(psnr_per_frame, ssim_per_frame, frame)) {
|
| - printf(" Frame Number: %zu PSNR: %f SSIM: %f \n", frame,
|
| - psnr_per_frame[frame], ssim_per_frame[frame]);
|
| - }
|
| - }
|
| -
|
| - printf("Print identical frame which appears in clusters : \n");
|
| - for (int cluster = 0;
|
| - cluster < static_cast<int>(identical_frame_clusters.size()); cluster++)
|
| - printf("%d ", identical_frame_clusters[cluster]);
|
| - printf("\n");
|
| -}
|
| -
|
| -void compute_metrics(const std::string& video_file_name,
|
| - std::vector<double>* psnr_per_frame,
|
| - std::vector<double>* ssim_per_frame) {
|
| - int height = 0, width = 0, fps = 0;
|
| - get_height_width_fps(&height, &width, &fps, video_file_name);
|
| -
|
| - int no_of_frames = 0;
|
| - int size = webrtc::test::GetI420FrameSize(width, height);
|
| -
|
| - // Allocate buffers for test and reference frames.
|
| - uint8_t* current_frame = new uint8_t[size];
|
| - uint8_t* next_frame = new uint8_t[size];
|
| -
|
| - while (true) {
|
| - if (!(webrtc::test::ExtractFrameFromY4mFile (video_file_name.c_str(),
|
| - width, height,
|
| - no_of_frames,
|
| - current_frame))) {
|
| - break;
|
| - }
|
| -
|
| - if (!(webrtc::test::ExtractFrameFromY4mFile (video_file_name.c_str(),
|
| - width, height,
|
| - no_of_frames + 1,
|
| - next_frame))) {
|
| - break;
|
| - }
|
| -
|
| - double result_psnr = webrtc::test::CalculateMetrics(webrtc::test::kPSNR,
|
| - current_frame,
|
| - next_frame,
|
| - width, height);
|
| - double result_ssim = webrtc::test::CalculateMetrics(webrtc::test::kSSIM,
|
| - current_frame,
|
| - next_frame,
|
| - width, height);
|
| -
|
| - psnr_per_frame->push_back(result_psnr);
|
| - ssim_per_frame->push_back(result_ssim);
|
| - no_of_frames++;
|
| - }
|
| - // Cleanup.
|
| - delete[] current_frame;
|
| - delete[] next_frame;
|
| -}
|
| -
|
| -bool check_file_extension(const std::string& video_file_name) {
|
| - if (video_file_name.substr(video_file_name.length()-3, 3) != "y4m") {
|
| - printf("Only y4m video file format is supported. Given: %s\n",
|
| - video_file_name.c_str());
|
| - return false;
|
| - }
|
| - return true;
|
| -}
|
| -
|
| -int run_analysis(const std::string& video_file) {
|
| - std::vector<double> psnr_per_frame;
|
| - std::vector<double> ssim_per_frame;
|
| - if (check_file_extension(video_file)) {
|
| - compute_metrics(video_file, &psnr_per_frame, &ssim_per_frame);
|
| - } else {
|
| - return -1;
|
| - }
|
| - printf("=============================\n");
|
| - printf("Printing metrics for file: %s\n", video_file.c_str());
|
| - printf("=============================\n");
|
| - print_freezing_metrics(psnr_per_frame, ssim_per_frame);
|
| - return 0;
|
| -}
|
|
|