21: Added option to watermakr in #17

pymule/__init__.py

@@ -16,6 +16,6 @@ from loader import importreg, pattern, setup, sigma,                          \
                    mergeset, mergeseeds, mergefks,                            \
                    addsets, scaleset
 
-from plot import errorband, kplot
+from plot import errorband, kplot, watermark
 
 from constants import pi, alpha, GF, conv, Mmu, Mel, Mtau

pymule/plot.py

@@ -14,6 +14,20 @@ rc('text.latex', preamble="\n".join([
 ]))
 
 
+def watermark(fig, txt='PRELIMINARY', fontsize=60, rotation=20):
+    axs = fig.axes
+    corners = matplotlib.transforms.Bbox.union(
+        [i.get_position() for i in axs]
+    ).corners()
+    center = np.average(corners, axis=0)
+    fig.text(
+        center[0], center[1],
+        txt,
+        fontsize=fontsize, rotation=rotation,
+        ha='center', va='center', alpha=0.1
+    )
+
+
 def errorband(p, ax=None, col='default', underflow=False, overflow=False):
     if ax is None:
         ax = plt.gca()