lut support

3 files changed, 51 insertions, 0 deletions

diff --git a/bin/dfatool b/bin/dfatool
index cdaae5a..0c57b7c 100755
--- a/bin/dfatool
+++ b/bin/dfatool
@@ -29,6 +29,7 @@ GetOptions(
 	  trigger-pin=i
 	  trigger-port=s
 	  voltage=f
+	  with-lut!
 	  offset=i
 	  zomg-fasta-nao
 	  }
@@ -61,6 +62,7 @@ my $driver = Kratos::DFADriver->new(
 	mimosa_voltage => $opt{voltage}         // 3.60,
 	trace_filter   => $opt{'trace-filter'}  // [],
 	trace_revisit  => $opt{'trace-revisit'} // 2,
+	with_lut       => $opt{'with-lut'},
 	xml_file       => $xml_file,
 );
 
diff --git a/bin/merge.py b/bin/merge.py
index f7dbf90..6091ba1 100755
--- a/bin/merge.py
+++ b/bin/merge.py
@@ -288,6 +288,14 @@ def param_values(parameters, by_param):
 
     return paramvalues
 
+def param_hash(values):
+    ret = {}
+
+    for i, param in enumerate(parameters):
+        ret[param] = values[i]
+
+    return ret
+
 # Returns the values used for each function argument in the measurement, e.g.
 # { 'data': [], 'length' : [16, 31, 32] }
 # non-numeric values such as '' or 'long_test_string' are skipped
@@ -547,11 +555,24 @@ def param_measures(name, paramdata, key, fun):
 def arg_measures(name, argdata, key, fun):
     return param_measures(name, argdata, key, fun)
 
+def lookup_table(name, paramdata, key, fun, keyfun):
+    lut = []
+
+    for pkey, pval in paramdata.items():
+        if pkey[0] == name:
+            lut.append({
+                'key': keyfun(pkey[1]),
+                'value': fun(pval[key]),
+            })
+
+    return lut
+
 def keydata(name, val, argdata, paramdata, tracedata, key):
     ret = {
         'count' : len(val[key]),
         'median' : np.median(val[key]),
         'mean'   : np.mean(val[key]),
+        'median_by_param' : lookup_table(name, paramdata, key, np.median, param_hash),
         'mean_goodness' : aggregate_measures(np.mean(val[key]), val[key]),
         'median_goodness' : aggregate_measures(np.median(val[key]), val[key]),
         'param_mean_goodness' : param_measures(name, paramdata, key, np.mean),
@@ -567,6 +588,7 @@ def keydata(name, val, argdata, paramdata, tracedata, key):
     if val['isa'] == 'transition':
         ret['arg_mean_goodness'] = arg_measures(name, argdata, key, np.mean)
         ret['arg_median_goodness'] = arg_measures(name, argdata, key, np.median)
+        ret['median_by_arg'] = lookup_table(name, argdata, key, np.median, list)
         ret['std_arg'] = np.mean([np.std(argdata[x][key]) for x in argdata.keys() if x[0] == name])
         ret['std_by_arg'] = {}
         ret['arg_fit_guess'] = {}
diff --git a/bin/mimplot b/bin/mimplot
new file mode 100755
index 0000000..1768da8
--- /dev/null
+++ b/bin/mimplot
@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+
+import csv
+import numpy as np
+import os
+import struct
+import sys
+import tarfile
+import matplotlib.pyplot as plt
+from dfatool import running_mean, MIMOSA
+
+voltage = float(sys.argv[1])
+shunt = float(sys.argv[2])
+mimfile = sys.argv[3]
+
+mim = MIMOSA(voltage, shunt)
+
+charges, triggers = mim.load_data(mimfile)
+charges = charges[:3000000]
+
+currents = running_mean(mim.charge_to_current_nocal(charges), 10) * 1e-6
+xr = np.arange(len(currents)) * 1e-5
+plt.plot( xr, currents, "r-")
+plt.xlabel('Zeit [s]')
+plt.ylabel('Strom [A]')
+plt.grid(True)
+plt.show()