import os

import subprocess

import platform

import matplotlib.pyplot as plt

from matplotlib.backends.backend_pdf import PdfPages

import time

import warnings

def save_figs(filename=None, open=True, folder='.', save_latest=True):

"""

:param filename: (optional) path to pdf file for saving.

:param open: boolean flag for opening pdf file after saving

:param folder: folder to save pdf files (default=current dir)

:param save_latest: boolean flag for creating a "latest.pdf" in folder directory, symlinked to latest plots.

:return:

"""

if filename is None:

filename = time.strftime("%Y%m%d-%H%M%S.pdf")

if folder is not None:

if not os.path.exists(folder):

os.makedirs(folder, exist_ok=True)

filename = os.path.join(folder, filename)

fn = os.path.join(os.getcwd(), filename)

pp = PdfPages(fn)

for i in plt.get_fignums():

plt.figure(i).tight_layout()

pp.savefig(plt.figure(i))

plt.close(plt.figure(i))

pp.close()

if save_latest:

try:

latest_path = os.path.join(folder, 'latest.pdf')

if os.path.exists(latest_path):

os.remove(latest_path)

os.symlink(filename, latest_path)

if open:

_open_figs(latest_path)

return

except OSError:

warnings.warn('Cannot create symbolic link in Windows without administrator privileges. Skipping.')

if open:

_open_figs(filename)

def _open_figs(filename):

pdf_path = os.path.abspath(filename)

if os.path.exists(pdf_path):

if platform.system() == 'Darwin': # macOS

subprocess.call(('open', pdf_path))

elif platform.system() == 'Windows': # Windows

os.startfile(pdf_path)

else: # linux variants

subprocess.call(('xdg-open', pdf_path))

def make_fig():

"""

Returns figure and axis for plotting. Can easily override to adjust defaults.

:return:

"""

fig, ax = plt.subplots()

fig.tight_layout()

return fig, ax