Source code for pm4py.visualization.graphs.variants.barplot

'''
    This file is part of PM4Py (More Info: https://pm4py.fit.fraunhofer.de).

    PM4Py is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    PM4Py is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with PM4Py.  If not, see <https://www.gnu.org/licenses/>.
'''
from copy import copy
from enum import Enum

import matplotlib

from pm4py.util import exec_utils
from pm4py.visualization.graphs.util import common
from typing import Optional, Dict, Any, Union, Tuple, List


[docs]class Parameters(Enum): TITLE = "title" FORMAT = "format" X_AXIS = "x_axis" Y_AXIS = "y_axis" PYPLOT_FIGURE_KWARGS = "pylot_figure_kwargs"
[docs]def apply_plot(x: List[float], y: List[float], parameters: Optional[Dict[Union[str, Parameters], Any]] = None) -> str: """ Visualizes a barchar provided its x-axis and y-axis points Parameters ----------------- x X-axis points y Y-axis points parameters Parameters Returns ----------------- tmp_file_name Temporary file name """ if parameters is None: parameters = {} title = exec_utils.get_param_value(Parameters.TITLE, parameters, "") format = exec_utils.get_param_value(Parameters.FORMAT, parameters, "png") x_axis = exec_utils.get_param_value(Parameters.X_AXIS, parameters, "") y_axis = exec_utils.get_param_value(Parameters.Y_AXIS, parameters, "") pyplot_figure_kwargs = exec_utils.get_param_value(Parameters.PYPLOT_FIGURE_KWARGS, parameters, {}) filename = common.get_temp_file_name(format) current_backend = copy(matplotlib.get_backend()) matplotlib.use('Agg') from matplotlib import pyplot pyplot.clf() fig = pyplot.figure(**pyplot_figure_kwargs) ax = fig.add_axes([0, 0, 1, 1]) ax.bar(x, y) pyplot.xlabel(x_axis) pyplot.ylabel(y_axis) pyplot.title(title) pyplot.savefig(filename, bbox_inches="tight", transparent=True) pyplot.clf() matplotlib.use(current_backend) return filename