Source code for viswaternet.utils.save_fig

import matplotlib.pyplot as plt
import os


def save_fig(self, save_name=None, style=None):
    model = self.model
    if style is None:
        style = self.default_style
    args = style.args
    dpi = args['dpi']
    save_format = args['save_format']
    if model['inp_file'] is not None:
        networkName = model["inp_file"]
    else:
        networkName = 'Viswaternet'
    if networkName.endswith(".inp"):
        try:
            prefixRemove = networkName.rfind("/")
            if prefixRemove == -1:
                prefixRemove = networkName.rfind("\\")
            networkName = networkName[prefixRemove + 1:]
        except Exception:
            pass
        networkName = networkName[:-4]
        networkName = networkName+"."+save_format
    if save_name is not None:
        file_name = str(save_name) + networkName
    else:
        file_name = networkName
    image_path_full = os.path.join(os.getcwd(), file_name)
    plt.savefig(image_path_full, dpi=dpi,
                format=save_format, bbox_inches="tight")