49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
|
|
from uuid import uuid4
|
|
from path import Path
|
|
import logging
|
|
import matplotlib.pyplot as plt
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def includegraphics(fig_ax, document_path="./", fig_path="fig/",
|
|
prefix="", scale=1):
|
|
""" Jinja2 filter which save the figure
|
|
and return latex includegraphic to display it.
|
|
|
|
:param fig_ax: TODO
|
|
:param path: TODO
|
|
:param prefix: TODO
|
|
:param scale: TODO
|
|
:returns: TODO
|
|
|
|
"""
|
|
try:
|
|
fig, ax = fig_ax
|
|
except (TypeError, ValueError):
|
|
ax = fig_ax
|
|
fig = ax.figure
|
|
|
|
if prefix:
|
|
filename = "{}_{}.pdf".format(prefix, str(uuid4())[:8])
|
|
else:
|
|
filename = "{}.pdf".format(str(uuid4())[:8])
|
|
|
|
path_to_file = Path(document_path/fig_path)
|
|
path_to_file.mkdir_p()
|
|
|
|
fig.tight_layout()
|
|
fig.savefig(path_to_file/filename)
|
|
logger.info(f"Graphique {filename} sauvé à {path_to_file}")
|
|
plt.close(fig)
|
|
return "\includegraphics[scale={sc}]{{{f}}}".format(sc = scale,
|
|
f=Path(fig_path)/filename)
|
|
|
|
|
|
# -----------------------------
|
|
# Reglages pour 'vim'
|
|
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
|
|
# cursor: 16 del
|