From 45b17988c605bacb6872b91c0f440aad94d102fd Mon Sep 17 00:00:00 2001 From: Petr Baudis Date: Sun, 19 Aug 2012 23:42:27 +0200 Subject: [PATCH] equalizer.py: Parse commandline parameters (-n PIXELS, -h) --- host_python/equalizer.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) diff --git a/host_python/equalizer.py b/host_python/equalizer.py index 5fb738d..37b7108 100755 --- a/host_python/equalizer.py +++ b/host_python/equalizer.py @@ -6,6 +6,8 @@ import struct import math import numpy as np import time +import sys +import getopt from datetime import datetime import ledbar @@ -24,6 +26,35 @@ MAX_FREQ = 12000 FREQ_STEP = float(RATE) / (CHUNK_SIZE * HISTORY_SIZE) PIXEL_FREQ_RANGE = math.pow(float(MAX_FREQ) / MIN_FREQ, 1.0/PIXELS) + +def print_usage(): + print '''\ +USAGE: + %s [-n number] [-h] +OPTIONS: + -n number number of controlled boxes + -h --help show this help +''' % sys.argv[0] + +try: + opts, args = getopt.getopt(sys.argv[1:], 'n:h', ['help']) +except getopt.GetOptError: + print_usage() + sys.exit(1) +if len(args): + print_usage() + sys.exit(1) +for k, v in opts: + if k == '-n': + if not v.isdigit(): + print_usage() + sys.exit(1) + PIXELS = int(v) + elif k == '-h' or k == '--help': + print_usage() + sys.exit(0) + + p = pyaudio.PyAudio() stream = p.open(format = FORMAT,