From 47290e583a50fd90bc13ad824f5f0fa2791afdef Mon Sep 17 00:00:00 2001
From: Leo Singer <leo.singer@ligo.org>
Date: Tue, 5 Dec 2017 09:38:34 -0500
Subject: [PATCH] Add metadata to image output
---
.../python/lalinference/bayestar/command.py | 19 ++++++++++++++++---
1 file changed, 16 insertions(+), 3 deletions(-)
diff --git a/lalinference/python/lalinference/bayestar/command.py b/lalinference/python/lalinference/bayestar/command.py
index 2448ed453b..70019cdbe5 100644
--- a/lalinference/python/lalinference/bayestar/command.py
+++ b/lalinference/python/lalinference/bayestar/command.py
@@ -54,6 +54,11 @@ import distutils.version
mpl_version = distutils.version.LooseVersion(matplotlib.__version__)
+def get_version():
+ from .. import InferenceVCSInfo as vcs_info
+ return vcs_info.name + ' ' + vcs_info.version
+
+
@contextlib.contextmanager
def TemporaryDirectory(suffix='', prefix='tmp', dir=None, delete=True):
try:
@@ -169,7 +174,16 @@ class MatplotlibFigureType(argparse.FileType):
def __save(self):
from matplotlib import pyplot as plt
- return plt.savefig(self.string)
+ _, ext = os.path.splitext(self.string)
+ ext = ext.lower()
+ program, _ = os.path.splitext(os.path.basename(sys.argv[0]))
+ cmdline = ' '.join([program] + sys.argv[1:])
+ metadata = {'Title': cmdline}
+ if ext == '.png':
+ metadata['Software'] = get_version()
+ elif ext in {'.pdf', '.ps', '.eps'}:
+ metadata['Creator'] = get_version()
+ return plt.savefig(self.string, metadata=metadata)
def __call__(self, string):
from matplotlib import pyplot as plt
@@ -284,8 +298,7 @@ del group
# Defer loading SWIG bindings until version string is needed.
class VersionAction(argparse._VersionAction):
def __call__(self, parser, namespace, values, option_string=None):
- from .. import InferenceVCSInfo
- self.version = 'LALInference ' + InferenceVCSInfo.version
+ self.version = get_version()
super(VersionAction, self).__call__(
parser, namespace, values, option_string)
--
GitLab