From 2f7924e58089493d6a46aebd25c966fbf5fb8a68 Mon Sep 17 00:00:00 2001
From: Joseph Timothy Foley <foley@ru.is>
Date: Fri, 27 Feb 2026 17:55:12 +0100
Subject: [PATCH] Option to have stats on graph or not

---
 infocalc.py | 26 ++++++++++++++------------
 tests.sh    |  2 +-
 2 files changed, 15 insertions(+), 13 deletions(-)

diff --git a/infocalc.py b/infocalc.py
index 419dfcc..64ad3fc 100755
--- a/infocalc.py
+++ b/infocalc.py
@@ -44,7 +44,7 @@ parser.add_argument('--normalizey', action="store_true",
                     help='Set y-axis to normalized probability density')
 parser.add_argument('--log', default="INFO",
                     help='Console log level:  Number or DEBUG, INFO, WARNING, ERROR')
-parser.add_argument('--graphinfo',
+parser.add_argument('--graphinfo', action="store_true",
                     help='Put information on the PDF graph')
 
 args = parser.parse_args()
@@ -103,18 +103,20 @@ info = -np.emath.log2(prob)
 #print("information content: %f bits", info)
 ## place text on plot:  https://matplotlib.org/3.3.4/gallery/recipes/placing_text_boxes.html
 fig, ax = plt.subplots()
-textstr = '\n'.join((
-    r'$n=%d$' % (samplesize),
-    r'$\mu=%.2f$' % (mean, ),
-    r'$\sigma=%.2f$' % (stddev, ),
-    r'$P=%.2f$' % (prob, ),
-    r'$I=%.2f$ bits' % (info, )))
-# these are matplotlib.patch.Patch properties
-props = dict(boxstyle='round', facecolor='wheat', alpha=0.5)
 
-# place a text box in upper left in axes coords
-ax.text(0.05, 0.95, textstr, transform=ax.transAxes, fontsize=14,
-        verticalalignment='top', bbox=props)
+if args.graphinfo:#put info on corner of graph
+    textstr = '\n'.join((
+        r'$n=%d$' % (samplesize),
+        r'$\mu=%.2f$' % (mean, ),
+        r'$\sigma=%.2f$' % (stddev, ),
+        r'$P=%.2f$' % (prob, ),
+        r'$I=%.2f$ bits' % (info, )))
+    # these are matplotlib.patch.Patch properties
+    props = dict(boxstyle='round', facecolor='wheat', alpha=0.5)
+    
+    # place a text box in upper left in axes coords
+    ax.text(0.05, 0.95, textstr, transform=ax.transAxes, fontsize=14,
+            verticalalignment='top', bbox=props)
 
 x = np.linspace(mean-3*stddev, mean+3*stddev, 500)
 y = norm.pdf(x, loc=mean, scale=stddev)
diff --git a/tests.sh b/tests.sh
index d60ea96..1e224d9 100755
--- a/tests.sh
+++ b/tests.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
 echo "Loading data from file"
-./infocalc.py DATA testdata.csv data1 0.9 1.1
+./infocalc.py DATA testdata.csv data1 0.9 1.1 --graphinfo
 echo "Creating simulated curve from parameters"
 ./infocalc.py SIM 8 1.0 0.5 0.9 1.1