render.py

#  Copyright (c) New Cloud Technologies, Ltd. 2013-2022.
#  Author: Vitaly Isaev <vitaly.isaev@myoffice.team>
#  License: https://github.com/newcloudtechnologies/memlimiter/blob/master/LICENSE
import os
from pathlib import Path
from typing import List

import humanize as humanize
import matplotlib.pyplot as plt
import matplotlib.ticker
import numpy as np

from report import Report


@matplotlib.ticker.FuncFormatter
def bytes_major_formatter(x, pos):
    return humanize.naturalsize(int(x), binary=True).replace(".0", "")


def single_report(report: Report):
    df = report.df

    fig, ax = plt.subplots(figsize=(6, 4.5))
    ax.set_xlim(0, 60)

    ax.set_xlabel('Time, seconds')

    # RSS plot
    color = 'tab:red'
    l0 = ax.plot(df['elapsed_time'], df['rss'], color=color, label='RSS')
    ax.set_ylabel('RSS, bytes')
    ax.set_ylim(0, 1024 * 1024 * 1024)
    ax.set_yticks([ml * 1024 * 1024 for ml in (256, 512, 512 + 256, 1024)])
    ax.yaxis.set_major_formatter(bytes_major_formatter)

    # GOGC plot
    color = 'tab:blue'
    twin1 = ax.twinx()
    l1 = twin1.plot(df['elapsed_time'], df['gogc'], color=color, label='GOGC')
    twin1.set_ylabel('GOGC')
    twin1.set_ylim(-5, 105)

    # Throttling plot
    color = 'tab:green'
    twin2 = ax.twinx()
    twin2.spines.right.set_position(("axes", 1.2))
    l2 = twin2.plot(df['elapsed_time'], df['throttling'], color=color, label='Throttling')
    twin2.set_ylabel('Throttling')
    twin2.set_ylim(-5, 105)

    # legend
    ls = l0 + l1 + l2
    labels = [l.get_label() for l in ls]
    fig.legend(ls, labels, loc=4)

    # title
    if report.session.params.unlimited:
        title = "MemLimiter disabled"
    else:
        coefficient = report.session.params.coefficient_str
        title = f'MemLimiter enabled, $K_{{p}} = {coefficient}$'
    ax.title.set_text(title)

    fig.tight_layout()
    fig.savefig(report.plot_file_path, transparent=False)


def multiple_reports(reports: List[Report], path: os.PathLike):
    fig, ax = plt.subplots(figsize=(8, 6))
    ax.set_xlim(0, 60)
    ax.set_xlabel('Time, seconds')
    ax.set_ylabel('RSS, bytes')
    ax.set_ylim(0, 1024 * 1024 * 1024)
    ax.set_yticks([ml * 1024 * 1024 for ml in (256, 512, 512 + 256, 1024)])
    ax.yaxis.set_major_formatter(bytes_major_formatter)

    n = len(reports)

    colors = plt.cm.turbo(np.linspace(0, 1, n))
    for i in range(n):
        report = reports[n-1-i]
        if report.session.params.unlimited:
            label = 'No limits'
        else:
            label = f'$K_{{p}} = {report.session.params.coefficient_str}$'

        ax.plot(report.df['elapsed_time'], report.df['rss'], color=colors[i], label=label)

    ax.legend()
    ax.title.set_text('RSS consumption dependence on $K_{{p}}$')
    fig.tight_layout()
    fig.savefig(path, transparent=False)