dc35dcd4dc
Also makes the properties protected, as previously requested by Stefan. Review-Url: https://codereview.webrtc.org/2179223003 Cr-Commit-Position: refs/heads/master@{#13593}
106 lines
3.4 KiB
C++
106 lines
3.4 KiB
C++
/*
|
|
* 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 "webrtc/tools/event_log_visualizer/plot_python.h"
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <memory>
|
|
|
|
namespace webrtc {
|
|
namespace plotting {
|
|
|
|
PythonPlot::PythonPlot() {}
|
|
|
|
PythonPlot::~PythonPlot() {}
|
|
|
|
void PythonPlot::Draw() {
|
|
// Write python commands to stdout. Intended program usage is
|
|
// ./event_log_visualizer event_log160330.dump | python
|
|
|
|
if (!series_list_.empty()) {
|
|
printf("color_count = %zu\n", series_list_.size());
|
|
printf(
|
|
"hls_colors = [(i*1.0/color_count, 0.25+i*0.5/color_count, 0.8) for i "
|
|
"in range(color_count)]\n");
|
|
printf("rgb_colors = [colorsys.hls_to_rgb(*hls) for hls in hls_colors]\n");
|
|
|
|
for (size_t i = 0; i < series_list_.size(); i++) {
|
|
// List x coordinates
|
|
printf("x%zu = [", i);
|
|
if (series_list_[i].points.size() > 0)
|
|
printf("%G", series_list_[i].points[0].x);
|
|
for (size_t j = 1; j < series_list_[i].points.size(); j++)
|
|
printf(", %G", series_list_[i].points[j].x);
|
|
printf("]\n");
|
|
|
|
// List y coordinates
|
|
printf("y%zu = [", i);
|
|
if (series_list_[i].points.size() > 0)
|
|
printf("%G", series_list_[i].points[0].y);
|
|
for (size_t j = 1; j < series_list_[i].points.size(); j++)
|
|
printf(", %G", series_list_[i].points[j].y);
|
|
printf("]\n");
|
|
|
|
if (series_list_[i].style == BAR_GRAPH) {
|
|
// There is a plt.bar function that draws bar plots,
|
|
// but it is *way* too slow to be useful.
|
|
printf(
|
|
"plt.vlines(x%zu, map(lambda t: min(t,0), y%zu), map(lambda t: "
|
|
"max(t,0), y%zu), color=rgb_colors[%zu], "
|
|
"label=\'%s\')\n",
|
|
i, i, i, i, series_list_[i].label.c_str());
|
|
} else if (series_list_[i].style == LINE_GRAPH) {
|
|
printf("plt.plot(x%zu, y%zu, color=rgb_colors[%zu], label=\'%s\')\n", i,
|
|
i, i, series_list_[i].label.c_str());
|
|
} else if (series_list_[i].style == LINE_DOT_GRAPH) {
|
|
printf(
|
|
"plt.plot(x%zu, y%zu, color=rgb_colors[%zu], label=\'%s\', "
|
|
"marker='.')\n",
|
|
i, i, i, series_list_[i].label.c_str());
|
|
} else {
|
|
printf("raise Exception(\"Unknown graph type\")\n");
|
|
}
|
|
}
|
|
}
|
|
|
|
printf("plt.xlim(%f, %f)\n", xaxis_min_, xaxis_max_);
|
|
printf("plt.ylim(%f, %f)\n", yaxis_min_, yaxis_max_);
|
|
printf("plt.xlabel(\'%s\')\n", xaxis_label_.c_str());
|
|
printf("plt.ylabel(\'%s\')\n", yaxis_label_.c_str());
|
|
printf("plt.title(\'%s\')\n", title_.c_str());
|
|
if (!series_list_.empty()) {
|
|
printf("plt.legend(loc=\'best\', fontsize=\'small\')\n");
|
|
}
|
|
}
|
|
|
|
PythonPlotCollection::PythonPlotCollection() {}
|
|
|
|
PythonPlotCollection::~PythonPlotCollection() {}
|
|
|
|
void PythonPlotCollection::Draw() {
|
|
printf("import matplotlib.pyplot as plt\n");
|
|
printf("import colorsys\n");
|
|
for (size_t i = 0; i < plots_.size(); i++) {
|
|
printf("plt.figure(%zu)\n", i);
|
|
plots_[i]->Draw();
|
|
}
|
|
printf("plt.show()\n");
|
|
}
|
|
|
|
Plot* PythonPlotCollection::AppendNewPlot() {
|
|
Plot* plot = new PythonPlot();
|
|
plots_.push_back(std::unique_ptr<Plot>(plot));
|
|
return plot;
|
|
}
|
|
|
|
} // namespace plotting
|
|
} // namespace webrtc
|