From 9b2816477b3101c4a6a749d094b690145c50f24b Mon Sep 17 00:00:00 2001
From: Kevin O'Connor <kevin@koconnor.net>
Date: Wed, 9 Sep 2020 17:14:09 -0400
Subject: [PATCH] graph_temp_sensor: Add support for graphing sensor resistance

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
---
 scripts/graph_temp_sensor.py | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/scripts/graph_temp_sensor.py b/scripts/graph_temp_sensor.py
index 8d090ca1..c1d18526 100755
--- a/scripts/graph_temp_sensor.py
+++ b/scripts/graph_temp_sensor.py
@@ -77,6 +77,27 @@ def plot_adc_resolution(config, sensors):
     fig.tight_layout()
     return fig
 
+def plot_resistance(config, sensors):
+    # Temperature list
+    all_temps = [float(i) for i in range(1, 351)]
+    # Build plot
+    fig, ax = matplotlib.pyplot.subplots()
+    pullup = config.getfloat('pullup_resistor', 0.)
+    ax.set_title("Temperature Sensor (pullup=%.0f)" % (pullup,))
+    ax.set_ylabel('Resistance (Ohms)')
+    for sensor in sensors:
+        sc = config.do_create_sensor(sensor)
+        adcs = [sc.calc_adc(t) for t in all_temps]
+        rs = [pullup * adc / (1.0 - adc) for adc in adcs]
+        ax.plot(all_temps, rs, label=sensor, alpha=0.6)
+    fontP = matplotlib.font_manager.FontProperties()
+    fontP.set_size('x-small')
+    ax.legend(loc='best', prop=fontP)
+    ax.set_xlabel('Temperature (C)')
+    ax.grid(True)
+    fig.tight_layout()
+    return fig
+
 
 ######################################################################
 # Startup
@@ -111,6 +132,8 @@ def main():
                     default=5., help="pullup resistor")
     opts.add_option("-s", "--sensors", type="string", dest="sensors",
                     default="", help="list of sensors (comma separated)")
+    opts.add_option("-r", "--resistance", action="store_true",
+                    help="graph sensor resistance")
     options, args = opts.parse_args()
     if len(args) != 0:
         opts.error("Incorrect number of arguments")
@@ -129,7 +152,10 @@ def main():
 
     # Draw graph
     setup_matplotlib(options.output is not None)
-    fig = plot_adc_resolution(config, sensors)
+    if options.resistance:
+        fig = plot_resistance(config, sensors)
+    else:
+        fig = plot_adc_resolution(config, sensors)
 
     # Show graph
     if options.output is None: