Source code for pm4py.objects.heuristics_net.edge

'''
    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/>.
'''
[docs]class Edge: def __init__(self, start_node, end_node, dependency_value, dfg_value, repr_value, label="", repr_color="#000000", edge_type="frequency", net_name=""): """ Constructor Parameters ------------ start_node Start node of the edge end_node End node of the edge dependency_value Dependency value of the edge dfg_value DFG value of the edge repr_value Value used in the representation label (If provided) label of the edge repr_color (If provided) Color that shall represent the edge edge_type Type of the edge (frequency or performance) net_name (If provided) name of the Heuristics Net """ self.start_node = start_node self.end_node = end_node self.dependency_value = dependency_value self.dfg_value = dfg_value self.repr_value = repr_value self.label = label self.repr_color = repr_color self.edge_type = edge_type self.net_name = net_name self.repr_font_color = None self.pen_width = None
[docs] def get_color(self): """ Gets the color to use for the representation """ return self.repr_color
[docs] def get_font_color(self): """ Gets the font color to use for the representation """ if self.repr_font_color is not None: return self.repr_font_color return self.repr_color
[docs] def get_penwidth(self, default): """ Gets the pen width to use for the representation Parameters -------------- default Default value """ if self.pen_width is not None: return self.pen_width return default