'''
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 graphviz import Source
import tempfile
from pm4py.util import exec_utils
from enum import Enum
from typing import Optional, Dict, Any, Union, Tuple
[docs]class Parameters(Enum):
FORMAT = "format"
XOR_SYMBOL = "#"
PREV_SYMBOL = "<"
SEQUENCE_SYMBOL = ">"
PARALLEL_SYMBOL = "||"
[docs]def apply(fp1: Dict[str, Any], fp2: Dict[str, Any], parameters: Optional[Dict[Union[str, Parameters], Any]] = None) -> Source:
"""
Visualize a comparison between two footprint tables
Parameters
---------------
fp1
Footprints associated to the log (NOT a list)
fp2
Footprints associated to the model
parameters
Parameters of the algorithm, including:
- Parameters.FORMAT => Format of the visualization
Returns
---------------
gviz
Graphviz object
"""
if parameters is None:
parameters = {}
if type(fp1) is list or type(fp2) is list:
raise Exception("footprints visualizer does not work on list of footprints!")
activities = sorted(list(set(x[0] for x in fp1["sequence"]).union(set(x[1] for x in fp1["sequence"])).union(
set(x[0] for x in fp1["parallel"])).union(set(x[1] for x in fp1["parallel"]))))
fp_table = {}
for a1 in activities:
fp_table[a1] = {}
for a2 in activities:
fp_table[a1][a2] = ("black", XOR_SYMBOL)
for x in fp1["sequence"]:
if x not in fp1["parallel"]:
if x in fp2["sequence"]:
fp_table[x[0]][x[1]] = ("black", SEQUENCE_SYMBOL)
fp_table[x[1]][x[0]] = ("black", PREV_SYMBOL)
else:
fp_table[x[0]][x[1]] = ("red", SEQUENCE_SYMBOL)
fp_table[x[1]][x[0]] = ("red", PREV_SYMBOL)
for x in fp1["parallel"]:
if x in fp2["parallel"]:
fp_table[x[0]][x[1]] = ("black", PARALLEL_SYMBOL)
else:
fp_table[x[0]][x[1]] = ("red", PARALLEL_SYMBOL)
image_format = exec_utils.get_param_value(Parameters.FORMAT, parameters, "png")
filename = tempfile.NamedTemporaryFile(suffix='.gv')
footprints_table = ["digraph {\n", "tbl [\n", "shape=plaintext\n", "label=<\n"]
footprints_table.append("<table border='0' cellborder='1' color='blue' cellspacing='0'>\n")
footprints_table.append("<tr><td></td>")
for act in activities:
footprints_table.append("<td><b>"+act+"</b></td>")
footprints_table.append("</tr>\n")
for a1 in activities:
footprints_table.append("<tr><td><b>"+a1+"</b></td>")
for a2 in activities:
footprints_table.append("<td><font color=\""+fp_table[a1][a2][0]+"\">"+fp_table[a1][a2][1]+"</font></td>")
footprints_table.append("</tr>\n")
footprints_table.append("</table>\n")
footprints_table.append(">];\n")
footprints_table.append("}\n")
footprints_table = "".join(footprints_table)
gviz = Source(footprints_table, filename=filename.name)
gviz.format = image_format
return gviz