forked from premiere/premiere-libtorrent
159 lines
5.0 KiB
C++
159 lines
5.0 KiB
C++
/*
|
|
|
|
Copyright (c) 2003-2017, Arvid Norberg
|
|
All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions
|
|
are met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in
|
|
the documentation and/or other materials provided with the distribution.
|
|
* Neither the name of the author nor the names of its
|
|
contributors may be used to endorse or promote products derived
|
|
from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
#include "session_view.hpp"
|
|
#include "print.hpp"
|
|
#include "libtorrent/session_stats.hpp"
|
|
#include "libtorrent/torrent_handle.hpp"
|
|
|
|
#include <algorithm> // for std::max
|
|
|
|
using libtorrent::span;
|
|
|
|
session_view::session_view()
|
|
: m_position(0)
|
|
{
|
|
using lt::find_metric_idx;
|
|
|
|
m_width = 128;
|
|
|
|
std::vector<lt::stats_metric> metrics = lt::session_stats_metrics();
|
|
m_cnt[0].resize(metrics.size(), 0);
|
|
m_cnt[1].resize(metrics.size(), 0);
|
|
}
|
|
|
|
void session_view::set_pos(int pos)
|
|
{
|
|
m_position = pos;
|
|
}
|
|
|
|
void session_view::set_width(int width)
|
|
{
|
|
m_width = width;
|
|
}
|
|
|
|
int session_view::pos() const { return m_position; }
|
|
|
|
int session_view::height() const
|
|
{
|
|
return 3;
|
|
}
|
|
|
|
void session_view::render()
|
|
{
|
|
char str[1024];
|
|
int pos = 0;
|
|
|
|
int y = m_position;
|
|
|
|
float const seconds = (m_timestamp[0] - m_timestamp[1]) / 1000000.f;
|
|
|
|
int const download_rate = int((m_cnt[0][m_recv_idx] - m_cnt[1][m_recv_idx])
|
|
/ seconds);
|
|
int const upload_rate = int((m_cnt[0][m_sent_idx] - m_cnt[1][m_sent_idx])
|
|
/ seconds);
|
|
|
|
pos += std::snprintf(str, sizeof(str), "%s%s fail: %s down: %s (%s) "
|
|
" bw queue: %s | %s conns: %3d unchoked: %2d / %2d%*s\x1b[K"
|
|
, esc("48;5;238")
|
|
, esc("1")
|
|
, add_suffix(m_cnt[0][m_failed_bytes_idx]).c_str()
|
|
, color(add_suffix(download_rate, "/s"), col_green).c_str()
|
|
, color(add_suffix(m_cnt[0][m_recv_idx]), col_green).c_str()
|
|
, color(to_string(int(m_cnt[0][m_limiter_up_queue_idx]), 3), col_red).c_str()
|
|
, color(to_string(int(m_cnt[0][m_limiter_down_queue_idx]), 3), col_green).c_str()
|
|
, int(m_cnt[0][m_num_peers_idx])
|
|
, int(m_cnt[0][m_unchoked_idx])
|
|
, int(m_cnt[0][m_unchoke_slots_idx])
|
|
, std::max(0, m_width - 86)
|
|
, esc("0"));
|
|
|
|
set_cursor_pos(0, y++);
|
|
print(str);
|
|
|
|
std::snprintf(str, sizeof(str), "%s%swaste: %s up: %s (%s) "
|
|
"disk queue: %s | %s cache w: %3d%% r: %3d%% "
|
|
"size: w: %s r: %s total: %s %*s\x1b[K"
|
|
#ifdef _WIN32
|
|
, esc("40")
|
|
#else
|
|
, esc("48;5;238")
|
|
#endif
|
|
, esc("1")
|
|
, add_suffix(m_cnt[0][m_wasted_bytes_idx]).c_str()
|
|
, color(add_suffix(upload_rate, "/s"), col_red).c_str()
|
|
, color(add_suffix(m_cnt[0][m_sent_idx]), col_red).c_str()
|
|
, color(to_string(int(m_cnt[0][m_queued_reads_idx]), 3), col_red).c_str()
|
|
, color(to_string(int(m_cnt[0][m_queued_writes_idx]), 3), col_green).c_str()
|
|
, int((m_cnt[0][m_blocks_written_idx] - m_cnt[0][m_write_ops_idx]) * 100
|
|
/ std::max(std::int64_t(1), m_cnt[0][m_blocks_written_idx]))
|
|
, int(m_cnt[0][m_cache_hit_idx] * 100
|
|
/ std::max(std::int64_t(1), m_cnt[0][m_num_blocks_read_idx]))
|
|
, add_suffix(m_cnt[0][m_writes_cache_idx] * 16 * 1024).c_str()
|
|
, add_suffix(m_cnt[0][m_reads_cache_idx] * 16 * 1024).c_str()
|
|
, add_suffix(m_cnt[0][m_blocks_in_use_idx] * 16 * 1024).c_str()
|
|
, std::max(0, m_width - 119)
|
|
, esc("0"));
|
|
set_cursor_pos(0, y++);
|
|
print(str);
|
|
|
|
std::snprintf(str, sizeof(str), "%s%suTP idle: %d syn: %d est: %d fin: %d wait: %d%*s\x1b[K"
|
|
, esc("48;5;238")
|
|
, esc("1")
|
|
, int(m_cnt[0][m_utp_idle])
|
|
, int(m_cnt[0][m_utp_syn_sent])
|
|
, int(m_cnt[0][m_utp_connected])
|
|
, int(m_cnt[0][m_utp_fin_sent])
|
|
, int(m_cnt[0][m_utp_close_wait])
|
|
, int(m_width - 37)
|
|
, esc("0"));
|
|
set_cursor_pos(0, y++);
|
|
print(str);
|
|
}
|
|
|
|
void session_view::update_counters(span<std::int64_t const> stats_counters
|
|
, std::uint64_t const t)
|
|
{
|
|
// only update the previous counters if there's been enough
|
|
// time since it was last updated
|
|
if (t - m_timestamp[1] > 2000000)
|
|
{
|
|
m_cnt[1].swap(m_cnt[0]);
|
|
m_timestamp[1] = m_timestamp[0];
|
|
}
|
|
|
|
m_cnt[0].assign(stats_counters.begin(), stats_counters.end());
|
|
m_timestamp[0] = t;
|
|
render();
|
|
}
|
|
|