From 5d37cf72cc4632355cd973430e047c9fdff6684d Mon Sep 17 00:00:00 2001
From: Selene ToyKeeper
Date: Sun, 30 Sep 2018 01:25:43 -0600
Subject: Made level_calc.py's ramp shape configurable from the command line,
 instead of having to edit the source.

---
 bin/level_calc.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'bin/level_calc.py')

diff --git a/bin/level_calc.py b/bin/level_calc.py
index 2aa8727..c903800 100755
--- a/bin/level_calc.py
+++ b/bin/level_calc.py
@@ -6,7 +6,7 @@ import math
 
 interactive = False
 # supported shapes: ninth, fifth, cube, square, log_e, log_2
-ramp_shape = 'cube'
+#ramp_shape = 'cube'
 
 
 def main(args):
@@ -14,6 +14,7 @@ def main(args):
     """
     # Get parameters from the user
     questions_main = [
+            (str, 'ramp_shape', 'cube', 'Ramp shape? [cube, square, fifth, ninth, log_e, log_2]'),
             (int, 'num_channels', 1, 'How many power channels?'),
             (int, 'num_levels', 4, 'How many total levels do you want?'),
             ]
@@ -36,6 +37,10 @@ def main(args):
 
     answers = Empty()
     ask(questions_main, answers)
+
+    global ramp_shape 
+    ramp_shape = answers.ramp_shape
+
     channels = []
     if not args:
         print('Describe the channels in order of lowest to highest power.')
-- 
cgit v1.2.3