added dac-scale.py: short script to calculate avrdd DAC+Vref values from level_calc.py ramp data

1 files changed, 89 insertions, 0 deletions

diff --git a/bin/dac-scale.py b/bin/dac-scale.py
new file mode 100755
index 0000000..d57f8c9
--- /dev/null
+++ b/bin/dac-scale.py
@@ -0,0 +1,89 @@
+#!/usr/bin/env python
+
+#
+# Calculates DAC values and Vref values from a list of raw intensity values
+# Usage: dac-scale.py 5,11,18,25,...,370474,384985,400000
+# Output: #defines suitable for use in hw/*/anduril.h
+# Assumptions:
+# - DAC data range is 0 to 1023
+# - DAC Vrefs are 1.024V and 2.5V
+# - HDR is available and has 2 steps
+# - HDR channel ratio is "highest_ramp_value / 2500"
+#
+# Output values thus have 4 "engine gears":
+# - HDR low , 1.024V Vref
+# - HDR low , 2.5  V Vref
+# - HDR high, 1.024V Vref
+# - HDR high, 2.5  V Vref
+#
+
+def main(args):
+    max_pwm = 1023
+    raw_pwm = [int(x) for x in args[0].split(',')]
+    ratio = raw_pwm[-1] / 2500.0
+    cooked = [[]]
+
+    def limit(p):
+        return min(max_pwm, int(p))
+
+    phase = 0
+    for raw in raw_pwm:
+        if 0 == phase:
+            if raw <= 1023:
+                cooked[-1].append(limit(raw))
+            else:
+                phase += 1
+                cooked.append([])
+        if 1 == phase:
+            if raw <= int(2500):
+                cooked[-1].append(limit(raw * 1.024 / 2.5))
+            else:
+                phase += 1
+                cooked.append([])
+        if 2 == phase:
+            if raw <= int(1024 * ratio):
+                cooked[-1].append(limit(raw / ratio))
+            else:
+                phase += 1
+                cooked.append([])
+        if 3 == phase:
+            cooked[-1].append(limit(raw * 1.024 / 2.5 / ratio))
+
+    # "gear change" boundaries
+    b1 =      len(cooked[0])
+    b2 = b1 + len(cooked[1])
+    b3 = b2 + len(cooked[2])
+    b4 = b3 + len(cooked[3])
+
+    #print(','.join(['%4i' % n for n in cooked]))
+
+    def fmt_pwms(l):
+        return ','.join(['%4i' % n for n in l])
+
+    def fmt_tops(v, l):
+        return ','.join([' %s' % v for n in l])
+
+    lines = []
+
+    lines.append('// top level for each "gear": %i %i %i %i' % (b1, b2, b3, b4))
+
+    lines.append('#define PWM1_LEVELS \\')
+    lines.append('        ' + fmt_pwms(cooked[0]) + ', \\')
+    lines.append('        ' + fmt_pwms(cooked[1]) + ', \\')
+    lines.append('        ' + fmt_pwms(cooked[2]) + ', \\')
+    lines.append('        ' + fmt_pwms(cooked[3]))
+    lines.append('#define PWM2_LEVELS \\')
+    lines.append('        ' + fmt_tops('V10', cooked[0]) + ', \\')
+    lines.append('        ' + fmt_tops('V25', cooked[1]) + ', \\')
+    lines.append('        ' + fmt_tops('V10', cooked[2]) + ', \\')
+    lines.append('        ' + fmt_tops('V25', cooked[3]))
+
+    lines.append('#define MAX_1x7135           %3i' % b2)
+    lines.append('#define HDR_ENABLE_LEVEL_MIN %3i' % (b2+1))
+
+    print('\n'.join(lines))
+
+if __name__ == "__main__":
+    import sys
+    main(sys.argv[1:])
+