1835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org/* 2835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. 3835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org * 4835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org * Use of this source code is governed by a BSD-style license 5835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org * that can be found in the LICENSE file in the root of the source 6835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org * tree. An additional intellectual property rights grant can be found 7835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org * in the file PATENTS. All contributing project authors may 8835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org * be found in the AUTHORS file in the root of the source tree. 9835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org */ 10835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 11b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org#include <math.h> 12835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org#include <stdio.h> 13e5abc854f3dc47de16067c2a41476c39b7626722henrik.lundin@webrtc.org#include "webrtc/modules/audio_coding/neteq/tools/neteq_quality_test.h" 14835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 15835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgnamespace webrtc { 16835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgnamespace test { 17835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 18835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgconst uint8_t kPayloadType = 95; 19835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgconst int kOutputSizeMs = 10; 20b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgconst int kInitSeed = 0x12345678; 21b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgconst int kPacketLossTimeUnitMs = 10; 22b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 23b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// Define switch for packet loss rate. 24b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgstatic bool ValidatePacketLossRate(const char* /* flag_name */, int32_t value) { 25b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (value >= 0 && value <= 100) 26b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return true; 27b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org printf("Invalid packet loss percentile, should be between 0 and 100."); 28b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return false; 29b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 30b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 31b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgDEFINE_int32(packet_loss_rate, 10, "Percentile of packet loss."); 32b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 33b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgstatic const bool packet_loss_rate_dummy = 34b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org RegisterFlagValidator(&FLAGS_packet_loss_rate, &ValidatePacketLossRate); 35b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 36b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// Define switch for random loss mode. 37b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgstatic bool ValidateRandomLossMode(const char* /* flag_name */, int32_t value) { 38b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (value >= 0 && value <= 2) 39b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return true; 40b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org printf("Invalid random packet loss mode, should be between 0 and 2."); 41b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return false; 42b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 43b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 44b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgDEFINE_int32(random_loss_mode, 1, 45b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org "Random loss mode: 0--no loss, 1--uniform loss, 2--Gilbert Elliot loss."); 46b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgstatic const bool random_loss_mode_dummy = 47b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org RegisterFlagValidator(&FLAGS_random_loss_mode, &ValidateRandomLossMode); 48b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 49b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// Define switch for burst length. 50b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgstatic bool ValidateBurstLength(const char* /* flag_name */, int32_t value) { 51b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (value >= kPacketLossTimeUnitMs) 52b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return true; 53b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org printf("Invalid burst length, should be greater than %d ms.", 54b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org kPacketLossTimeUnitMs); 55b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return false; 56b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 57b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 58b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgDEFINE_int32(burst_length, 30, 59b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org "Burst length in milliseconds, only valid for Gilbert Elliot loss."); 60b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 61b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgstatic const bool burst_length_dummy = 62b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org RegisterFlagValidator(&FLAGS_burst_length, &ValidateBurstLength); 63b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 64b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// Define switch for drift factor. 65b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgstatic bool ValidateDriftFactor(const char* /* flag_name */, double value) { 66b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (value > -0.1) 67b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return true; 68b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org printf("Invalid drift factor, should be greater than -0.1."); 69b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return false; 70b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 71b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 72b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgDEFINE_double(drift_factor, 0.0, "Time drift factor."); 73b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 74b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgstatic const bool drift_factor_dummy = 75b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org RegisterFlagValidator(&FLAGS_drift_factor, &ValidateDriftFactor); 76b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 77b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// ProbTrans00Solver() is to calculate the transition probability from no-loss 78b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// state to itself in a modified Gilbert Elliot packet loss model. The result is 79b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// to achieve the target packet loss rate |loss_rate|, when a packet is not 80b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// lost only if all |units| drawings within the duration of the packet result in 81b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// no-loss. 82b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgstatic double ProbTrans00Solver(int units, double loss_rate, 83b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org double prob_trans_10) { 84b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (units == 1) 85b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return prob_trans_10 / (1.0f - loss_rate) - prob_trans_10; 86b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// 0 == prob_trans_00 ^ (units - 1) + (1 - loss_rate) / prob_trans_10 * 87b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// prob_trans_00 - (1 - loss_rate) * (1 + 1 / prob_trans_10). 88b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// There is a unique solution between 0.0 and 1.0, due to the monotonicity and 89b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// an opposite sign at 0.0 and 1.0. 90b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// For simplicity, we reformulate the equation as 91b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// f(x) = x ^ (units - 1) + a x + b. 92b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// Its derivative is 93b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// f'(x) = (units - 1) x ^ (units - 2) + a. 94b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// The derivative is strictly greater than 0 when x is between 0 and 1. 95b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// We use Newton's method to solve the equation, iteration is 96b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org// x(k+1) = x(k) - f(x) / f'(x); 97b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org const double kPrecision = 0.001f; 98b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org const int kIterations = 100; 99b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org const double a = (1.0f - loss_rate) / prob_trans_10; 100b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org const double b = (loss_rate - 1.0f) * (1.0f + 1.0f / prob_trans_10); 101b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org double x = 0.0f; // Starting point; 102b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org double f = b; 103b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org double f_p; 104b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org int iter = 0; 105b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org while ((f >= kPrecision || f <= -kPrecision) && iter < kIterations) { 106b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org f_p = (units - 1.0f) * pow(x, units - 2) + a; 107b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org x -= f / f_p; 108b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (x > 1.0f) { 109b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org x = 1.0f; 110b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } else if (x < 0.0f) { 111b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org x = 0.0f; 112b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 113b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org f = pow(x, units - 1) + a * x + b; 114b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org iter ++; 115b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 116b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return x; 117b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 118835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 119835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgNetEqQualityTest::NetEqQualityTest(int block_duration_ms, 120835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org int in_sampling_khz, 121835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org int out_sampling_khz, 122835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org enum NetEqDecoder decoder_type, 123835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org int channels, 124835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org std::string in_filename, 125835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org std::string out_filename) 126835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org : decoded_time_ms_(0), 127835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org decodable_time_ms_(0), 128b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org drift_factor_(FLAGS_drift_factor), 129b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org packet_loss_rate_(FLAGS_packet_loss_rate), 130835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org block_duration_ms_(block_duration_ms), 131835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org in_sampling_khz_(in_sampling_khz), 132835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org out_sampling_khz_(out_sampling_khz), 133835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org decoder_type_(decoder_type), 134835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org channels_(channels), 135835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org in_filename_(in_filename), 136835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org out_filename_(out_filename), 137b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org log_filename_(out_filename + ".log"), 138835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org in_size_samples_(in_sampling_khz_ * block_duration_ms_), 139835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org out_size_samples_(out_sampling_khz_ * kOutputSizeMs), 140835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org payload_size_bytes_(0), 141835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org max_payload_bytes_(0), 142835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org in_file_(new InputAudioFile(in_filename_)), 143835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org out_file_(NULL), 144b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org log_file_(NULL), 145835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org rtp_generator_(new RtpGenerator(in_sampling_khz_, 0, 0, 146b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org decodable_time_ms_)), 147b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org total_payload_size_bytes_(0) { 1484087215f83d5a77b86fde9ebf8598dfe7ecd70a1henrik.lundin@webrtc.org NetEq::Config config; 1494087215f83d5a77b86fde9ebf8598dfe7ecd70a1henrik.lundin@webrtc.org config.sample_rate_hz = out_sampling_khz_ * 1000; 1504087215f83d5a77b86fde9ebf8598dfe7ecd70a1henrik.lundin@webrtc.org neteq_.reset(NetEq::Create(config)); 151835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org max_payload_bytes_ = in_size_samples_ * channels_ * sizeof(int16_t); 152835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org in_data_.reset(new int16_t[in_size_samples_ * channels_]); 153835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org payload_.reset(new uint8_t[max_payload_bytes_]); 154835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org out_data_.reset(new int16_t[out_size_samples_ * channels_]); 155835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org} 156835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 157b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgbool NoLoss::Lost() { 158b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return false; 159b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 160b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 161e291f572fe08a38b2501d490969ddf42235962deminyue@webrtc.orgUniformLoss::UniformLoss(double loss_rate) 162b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org : loss_rate_(loss_rate) { 163b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 164b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 165b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgbool UniformLoss::Lost() { 166b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org int drop_this = rand(); 167b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return (drop_this < loss_rate_ * RAND_MAX); 168b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 169b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 170b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgGilbertElliotLoss::GilbertElliotLoss(double prob_trans_11, double prob_trans_01) 171b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org : prob_trans_11_(prob_trans_11), 172b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org prob_trans_01_(prob_trans_01), 173b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org lost_last_(false), 174b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org uniform_loss_model_(new UniformLoss(0)) { 175b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 176b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 177b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgbool GilbertElliotLoss::Lost() { 178b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // Simulate bursty channel (Gilbert model). 179b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // (1st order) Markov chain model with memory of the previous/last 180b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // packet state (lost or received). 181b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (lost_last_) { 182b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // Previous packet was not received. 183b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org uniform_loss_model_->set_loss_rate(prob_trans_11_); 184b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return lost_last_ = uniform_loss_model_->Lost(); 185b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } else { 186b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org uniform_loss_model_->set_loss_rate(prob_trans_01_); 187b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return lost_last_ = uniform_loss_model_->Lost(); 188b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 189b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 190b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 191835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgvoid NetEqQualityTest::SetUp() { 192835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org out_file_ = fopen(out_filename_.c_str(), "wb"); 193b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org log_file_ = fopen(log_filename_.c_str(), "wt"); 194835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org ASSERT_TRUE(out_file_ != NULL); 195835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org ASSERT_EQ(0, neteq_->RegisterPayloadType(decoder_type_, kPayloadType)); 196835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org rtp_generator_->set_drift_factor(drift_factor_); 197b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 198b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org int units = block_duration_ms_ / kPacketLossTimeUnitMs; 199b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org switch (FLAGS_random_loss_mode) { 200b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org case 1: { 201b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // |unit_loss_rate| is the packet loss rate for each unit time interval 202b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // (kPacketLossTimeUnitMs). Since a packet loss event is generated if any 203b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // of |block_duration_ms_ / kPacketLossTimeUnitMs| unit time intervals of 204b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // a full packet duration is drawn with a loss, |unit_loss_rate| fulfills 205b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // (1 - unit_loss_rate) ^ (block_duration_ms_ / kPacketLossTimeUnitMs) == 206b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // 1 - packet_loss_rate. 207b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org double unit_loss_rate = (1.0f - pow(1.0f - 0.01f * packet_loss_rate_, 208b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 1.0f / units)); 209b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org loss_model_.reset(new UniformLoss(unit_loss_rate)); 210b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org break; 211b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 212b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org case 2: { 213b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // |FLAGS_burst_length| should be integer times of kPacketLossTimeUnitMs. 214b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org ASSERT_EQ(0, FLAGS_burst_length % kPacketLossTimeUnitMs); 215b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 216b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // We do not allow 100 percent packet loss in Gilbert Elliot model, which 217b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // makes no sense. 218b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org ASSERT_GT(100, packet_loss_rate_); 219b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 220b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // To guarantee the overall packet loss rate, transition probabilities 221b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // need to satisfy: 222b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // pi_0 * (1 - prob_trans_01_) ^ units + 223b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // pi_1 * prob_trans_10_ ^ (units - 1) == 1 - loss_rate 224b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // pi_0 = prob_trans_10 / (prob_trans_10 + prob_trans_01_) 225b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // is the stationary state probability of no-loss 226b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // pi_1 = prob_trans_01_ / (prob_trans_10 + prob_trans_01_) 227b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // is the stationary state probability of loss 228b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // After a derivation prob_trans_00 should satisfy: 229b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // prob_trans_00 ^ (units - 1) = (loss_rate - 1) / prob_trans_10 * 230b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // prob_trans_00 + (1 - loss_rate) * (1 + 1 / prob_trans_10). 231b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org double loss_rate = 0.01f * packet_loss_rate_; 232b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org double prob_trans_10 = 1.0f * kPacketLossTimeUnitMs / FLAGS_burst_length; 233b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org double prob_trans_00 = ProbTrans00Solver(units, loss_rate, prob_trans_10); 234b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org loss_model_.reset(new GilbertElliotLoss(1.0f - prob_trans_10, 235b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 1.0f - prob_trans_00)); 236b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org break; 237b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 238b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org default: { 239b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org loss_model_.reset(new NoLoss); 240b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org break; 241b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 242b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 243b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 244b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // Make sure that the packet loss profile is same for all derived tests. 245b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org srand(kInitSeed); 246835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org} 247835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 248835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgvoid NetEqQualityTest::TearDown() { 249835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org fclose(out_file_); 250835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org} 251835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 252b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.orgbool NetEqQualityTest::PacketLost() { 253b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org int cycles = block_duration_ms_ / kPacketLossTimeUnitMs; 254b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 255b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // The loop is to make sure that codecs with different block lengths share the 256b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // same packet loss profile. 257b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org bool lost = false; 258b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org for (int idx = 0; idx < cycles; idx ++) { 259b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (loss_model_->Lost()) { 260b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // The packet will be lost if any of the drawings indicates a loss, but 261b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // the loop has to go on to make sure that codecs with different block 262b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // lengths keep the same pace. 263b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org lost = true; 264b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 265b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 266b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return lost; 267b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org} 268b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org 269835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgint NetEqQualityTest::Transmit() { 270835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org int packet_input_time_ms = 271835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org rtp_generator_->GetRtpHeader(kPayloadType, in_size_samples_, 272835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org &rtp_header_); 273b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (payload_size_bytes_ > 0) { 274b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org fprintf(log_file_, "Packet at %d ms", packet_input_time_ms); 275b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (!PacketLost()) { 276b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org int ret = neteq_->InsertPacket(rtp_header_, &payload_[0], 277b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org payload_size_bytes_, 278b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org packet_input_time_ms * in_sampling_khz_); 279b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org if (ret != NetEq::kOK) 280b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org return -1; 281b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org fprintf(log_file_, " OK.\n"); 282b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } else { 283b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org fprintf(log_file_, " Lost.\n"); 284b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org } 285835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org } 286835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org return packet_input_time_ms; 287835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org} 288835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 289835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgint NetEqQualityTest::DecodeBlock() { 290835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org int channels; 291835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org int samples; 292835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org int ret = neteq_->GetAudio(out_size_samples_ * channels_, &out_data_[0], 293835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org &samples, &channels, NULL); 294835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 295835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org if (ret != NetEq::kOK) { 296835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org return -1; 297835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org } else { 298835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org assert(channels == channels_); 299835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org assert(samples == kOutputSizeMs * out_sampling_khz_); 300835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org fwrite(&out_data_[0], sizeof(int16_t), samples * channels, out_file_); 301835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org return samples; 302835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org } 303835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org} 304835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 305835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.orgvoid NetEqQualityTest::Simulate(int end_time_ms) { 306835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org int audio_size_samples; 307835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 308835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org while (decoded_time_ms_ < end_time_ms) { 309b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org // Assume 10 packets in packets buffer. 310b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org while (decodable_time_ms_ - 10 * block_duration_ms_ < decoded_time_ms_) { 311835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org ASSERT_TRUE(in_file_->Read(in_size_samples_ * channels_, &in_data_[0])); 312835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org payload_size_bytes_ = EncodeBlock(&in_data_[0], 313835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org in_size_samples_, &payload_[0], 314835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org max_payload_bytes_); 315b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org total_payload_size_bytes_ += payload_size_bytes_; 316835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org decodable_time_ms_ = Transmit() + block_duration_ms_; 317835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org } 318835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org audio_size_samples = DecodeBlock(); 319835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org if (audio_size_samples > 0) { 320835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org decoded_time_ms_ += audio_size_samples / out_sampling_khz_; 321835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org } 322835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org } 323b5272dfc7e7599580263bcde04b0484a6379f2b9minyue@webrtc.org fprintf(log_file_, "%f", 8.0f * total_payload_size_bytes_ / end_time_ms); 324835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org} 325835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org 326835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org} // namespace test 327835b016d936aeb6c48b42c211ef257d6d3057c4fminyue@webrtc.org} // namespace webrtc 328