mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 13:50:40 +01:00
1a778a24ba
Bug: None Change-Id: I4c0ab1ba7730bdcdd826aa41b67b80a96d92c8f3 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/221204 Reviewed-by: Björn Terelius <terelius@webrtc.org> Commit-Queue: Danil Chapovalov <danilchap@webrtc.org> Cr-Commit-Position: refs/heads/master@{#34231}
109 lines
3.3 KiB
C++
109 lines
3.3 KiB
C++
/*
|
|
* Copyright (c) 2014 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 "modules/remote_bitrate_estimator/tools/bwe_rtp.h"
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <set>
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
#include "absl/flags/flag.h"
|
|
#include "absl/flags/parse.h"
|
|
#include "modules/rtp_rtcp/include/rtp_header_extension_map.h"
|
|
#include "test/rtp_file_reader.h"
|
|
|
|
ABSL_FLAG(std::string,
|
|
extension_type,
|
|
"abs",
|
|
"Extension type, either abs for absolute send time or tsoffset "
|
|
"for timestamp offset.");
|
|
std::string ExtensionType() {
|
|
return absl::GetFlag(FLAGS_extension_type);
|
|
}
|
|
|
|
ABSL_FLAG(int, extension_id, 3, "Extension id.");
|
|
int ExtensionId() {
|
|
return absl::GetFlag(FLAGS_extension_id);
|
|
}
|
|
|
|
ABSL_FLAG(std::string, input_file, "", "Input file.");
|
|
std::string InputFile() {
|
|
return absl::GetFlag(FLAGS_input_file);
|
|
}
|
|
|
|
ABSL_FLAG(std::string,
|
|
ssrc_filter,
|
|
"",
|
|
"Comma-separated list of SSRCs in hexadecimal which are to be "
|
|
"used as input to the BWE (only applicable to pcap files).");
|
|
std::set<uint32_t> SsrcFilter() {
|
|
std::string ssrc_filter_string = absl::GetFlag(FLAGS_ssrc_filter);
|
|
if (ssrc_filter_string.empty())
|
|
return std::set<uint32_t>();
|
|
std::stringstream ss;
|
|
std::string ssrc_filter = ssrc_filter_string;
|
|
std::set<uint32_t> ssrcs;
|
|
|
|
// Parse the ssrcs in hexadecimal format.
|
|
ss << std::hex << ssrc_filter;
|
|
uint32_t ssrc;
|
|
while (ss >> ssrc) {
|
|
ssrcs.insert(ssrc);
|
|
ss.ignore(1, ',');
|
|
}
|
|
return ssrcs;
|
|
}
|
|
|
|
bool ParseArgsAndSetupRtpReader(
|
|
int argc,
|
|
char** argv,
|
|
std::unique_ptr<webrtc::test::RtpFileReader>& rtp_reader,
|
|
webrtc::RtpHeaderExtensionMap& rtp_header_extensions) {
|
|
absl::ParseCommandLine(argc, argv);
|
|
std::string filename = InputFile();
|
|
|
|
std::set<uint32_t> ssrc_filter = SsrcFilter();
|
|
fprintf(stderr, "Filter on SSRC: ");
|
|
for (auto& s : ssrc_filter) {
|
|
fprintf(stderr, "0x%08x, ", s);
|
|
}
|
|
fprintf(stderr, "\n");
|
|
if (filename.substr(filename.find_last_of('.')) == ".pcap") {
|
|
fprintf(stderr, "Opening as pcap\n");
|
|
rtp_reader.reset(webrtc::test::RtpFileReader::Create(
|
|
webrtc::test::RtpFileReader::kPcap, filename.c_str(), SsrcFilter()));
|
|
} else {
|
|
fprintf(stderr, "Opening as rtp\n");
|
|
rtp_reader.reset(webrtc::test::RtpFileReader::Create(
|
|
webrtc::test::RtpFileReader::kRtpDump, filename.c_str()));
|
|
}
|
|
if (!rtp_reader) {
|
|
fprintf(stderr, "Cannot open input file %s\n", filename.c_str());
|
|
return false;
|
|
}
|
|
fprintf(stderr, "Input file: %s\n\n", filename.c_str());
|
|
|
|
webrtc::RTPExtensionType extension = webrtc::kRtpExtensionAbsoluteSendTime;
|
|
if (ExtensionType() == "tsoffset") {
|
|
extension = webrtc::kRtpExtensionTransmissionTimeOffset;
|
|
fprintf(stderr, "Extension: toffset\n");
|
|
} else if (ExtensionType() == "abs") {
|
|
fprintf(stderr, "Extension: abs\n");
|
|
} else {
|
|
fprintf(stderr, "Unknown extension type\n");
|
|
return false;
|
|
}
|
|
|
|
rtp_header_extensions.RegisterByType(ExtensionId(), extension);
|
|
|
|
return true;
|
|
}
|