From a8c300e36fbe9d53421d5c54ed0ca4da66c3dc25 Mon Sep 17 00:00:00 2001
From: Lionel Koenig <lionelk@webrtc.org>
Date: Wed, 7 Dec 2022 15:02:12 +0100
Subject: [PATCH] neteq: Add legend in test plot tools

Add a legend when on the python plots generated with neteq_rtpplay.


Bug: None
Change-Id: I4299858bb9e8e59564c824c99272e4fabc610162
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/286840
Reviewed-by: Jesus de Vicente Pena <devicentepena@webrtc.org>
Commit-Queue: Lionel Koenig <lionelk@webrtc.org>
Reviewed-by: Ivo Creusen <ivoc@webrtc.org>
Cr-Commit-Position: refs/heads/main@{#38838}
---
 modules/audio_coding/neteq/tools/neteq_delay_analyzer.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules/audio_coding/neteq/tools/neteq_delay_analyzer.cc b/modules/audio_coding/neteq/tools/neteq_delay_analyzer.cc
index 91c3a1d96b..020199e9ac 100644
--- a/modules/audio_coding/neteq/tools/neteq_delay_analyzer.cc
+++ b/modules/audio_coding/neteq/tools/neteq_delay_analyzer.cc
@@ -284,6 +284,9 @@ void NetEqDelayAnalyzer::CreatePythonScript(
   output << "  plt.ylabel('relative delay [ms]')" << std::endl;
   if (!ssrcs_.empty()) {
     auto ssrc_it = ssrcs_.cbegin();
+    output << "  plt.legend((\"arrival delay\", \"target delay\", \"playout "
+              "delay\"))"
+           << std::endl;
     output << "  plt.title('SSRC: 0x" << std::hex
            << static_cast<int64_t>(*ssrc_it++);
     while (ssrc_it != ssrcs_.end()) {