diff options
Diffstat (limited to 'client.py')
| -rw-r--r-- | client.py | 132 |
1 files changed, 130 insertions, 2 deletions
@@ -11,6 +11,8 @@ import socket import optparse import array import random +import threading +import thread from packet import Packet, CMD, stoi @@ -21,6 +23,12 @@ parser.add_option('--generators', dest='generators', action='store_true', help=' parser.add_option('-u', '--uid', dest='uid', default='', help='Set the UID (identifier) of this client in the network') parser.add_option('-p', '--port', dest='port', type='int', default=13676, help='Set the port to listen on') parser.add_option('-r', '--rate', dest='rate', type='int', default=44100, help='Set the sample rate of the audio device') +parser.add_option('-V', '--volume', dest='volume', type='float', default=1.0, help='Set the volume factor (>1 distorts, <1 attenuates)') +parser.add_option('-G', '--gui', dest='gui', default='', help='set a GUI to use') +parser.add_option('--pg-fullscreen', dest='fullscreen', action='store_true', help='Use a full-screen video mode') +parser.add_option('--pg-samp-width', dest='samp_width', type='int', help='Set the width of the sample pane (by default display width / 2)') +parser.add_option('--pg-bgr-width', dest='bgr_width', type='int', help='Set the width of the bargraph pane (by default display width / 2)') +parser.add_option('--pg-height', dest='height', type='int', help='Set the height of the window or full-screen video mode') options, args = parser.parse_args() @@ -30,6 +38,7 @@ IDENT = 'TONE' UID = options.uid LAST_SAMP = 0 +LAST_SAMPLES = [] FREQ = 0 PHASE = 0 RATE = options.rate @@ -43,6 +52,102 @@ MIN = -0x80000000 def lin_interp(frm, to, p): return p*to + (1-p)*frm +# GUIs + +GUIs = {} + +def GUI(f): + GUIs[f.__name__] = f + return f + +@GUI +def pygame_notes(): + import pygame + import pygame.gfxdraw + pygame.init() + + dispinfo = pygame.display.Info() + DISP_WIDTH = 640 + DISP_HEIGHT = 480 + if dispinfo.current_h > 0 and dispinfo.current_w > 0: + DISP_WIDTH = dispinfo.current_w + DISP_HEIGHT = dispinfo.current_h + + SAMP_WIDTH = DISP_WIDTH / 2 + if options.samp_width > 0: + SAMP_WIDTH = options.samp_width + BGR_WIDTH = DISP_WIDTH / 2 + if options.bgr_width > 0: + BGR_WIDTH = options.bgr_width + HEIGHT = DISP_HEIGHT + if options.height > 0: + HEIGHT = options.height + + flags = 0 + if options.fullscreen: + flags |= pygame.FULLSCREEN + + disp = pygame.display.set_mode((SAMP_WIDTH + BGR_WIDTH, HEIGHT), flags) + + WIDTH, HEIGHT = disp.get_size() + SAMP_WIDTH = WIDTH / 2 + BGR_WIDTH = WIDTH - SAMP_WIDTH + PFAC = HEIGHT / 128.0 + + sampwin = pygame.Surface((SAMP_WIDTH, HEIGHT)) + lastsy = HEIGHT / 2 + + clock = pygame.time.Clock() + + while True: + if FREQ > 0: + try: + pitch = 12 * math.log(FREQ / 440.0, 2) + 69 + except ValueError: + pitch = 0 + else: + pitch = 0 + col = [int((AMP / MAX) * 255)] * 3 + + disp.fill((0, 0, 0), (BGR_WIDTH, 0, SAMP_WIDTH, HEIGHT)) + disp.scroll(-1, 0) + disp.fill(col, (BGR_WIDTH - 1, HEIGHT - pitch * PFAC - PFAC, 1, PFAC)) + + sampwin.scroll(-len(LAST_SAMPLES), 0) + x = max(0, SAMP_WIDTH - len(LAST_SAMPLES)) + sampwin.fill((0, 0, 0), (x, 0, SAMP_WIDTH - x, HEIGHT)) + for i in LAST_SAMPLES: + sy = int((float(i) / MAX) * (HEIGHT / 2) + (HEIGHT / 2)) + pygame.gfxdraw.line(sampwin, x - 1, lastsy, x, sy, (0, 255, 0)) + x += 1 + lastsy = sy + del LAST_SAMPLES[:] + #w, h = SAMP_WIDTH, HEIGHT + #pts = [(BGR_WIDTH, HEIGHT / 2), (w + BGR_WIDTH, HEIGHT / 2)] + #x = w + BGR_WIDTH + #for i in reversed(LAST_SAMPLES): + # pts.insert(1, (x, int((h / 2) + (float(i) / MAX) * (h / 2)))) + # x -= 1 + # if x < BGR_WIDTH: + # break + #if len(pts) > 2: + # pygame.gfxdraw.aapolygon(disp, pts, [0, 255, 0]) + disp.blit(sampwin, (BGR_WIDTH, 0)) + pygame.display.flip() + + for ev in pygame.event.get(): + if ev.type == pygame.KEYDOWN: + if ev.key == pygame.K_ESCAPE: + thread.interrupt_main() + pygame.quit() + exit() + elif ev.type == pygame.QUIT: + thread.interrupt_main() + pygame.quit() + exit() + + clock.tick(60) + # Generator functions--should be cyclic within [0, 2*math.pi) and return [-1, 1] GENERATORS = [{'name': 'math.sin', 'args': None, 'desc': 'Sine function'}, @@ -65,6 +170,10 @@ def tri_wave(theta): else: return lin_interp(-1, 0, (theta-3*math.pi/2)/(math.pi/2)) +@generator('Saw wave (line from (0, 1) to (2pi, -1))') +def saw_wave(theta): + return lin_interp(1, -1, theta/(math.pi * 2)) + @generator('Simple square wave (piecewise 1 at x<pi, 0 else)') def square_wave(theta): if theta < math.pi: @@ -113,6 +222,14 @@ class harmonic(object): def __call__(self, theta): return max(-1, min(1, sum([amp*self.gen((i+1)*theta % (2*math.pi)) for i, amp in enumerate(self.spectrum)]))) +@generator('General harmonics generator (adds arbitrary overtones)', '(<generator>, <factor of f>, <amplitude>, <factor>, <amplitude>, ...)') +class genharmonic(object): + def __init__(self, gen, *harmonics): + self.gen = gen + self.harmonics = zip(harmonics[::2], harmonics[1::2]) + def __call__(self, theta): + return max(-1, min(1, sum([amp * self.gen(i * theta % (2*math.pi)) for i, amp in self.harmonics]))) + @generator('Mix generator', '(<generator>[, <amp>], [<generator>[, <amp>], [...]])') class mixer(object): def __init__(self, *specs): @@ -171,28 +288,39 @@ def samps(freq, phase, cnt): global RATE, AMP samps = [0]*cnt for i in xrange(cnt): - samps[i] = int(AMP * generator((phase + 2 * math.pi * freq * i / RATE) % (2*math.pi))) + samps[i] = int(AMP * max(-1, min(1, options.volume*generator((phase + 2 * math.pi * freq * i / RATE) % (2*math.pi))))) return samps, (phase + 2 * math.pi * freq * cnt / RATE) % (2*math.pi) def to_data(samps): return struct.pack('i'*len(samps), *samps) def gen_data(data, frames, time, status): - global FREQ, PHASE, Z_SAMP, LAST_SAMP + global FREQ, PHASE, Z_SAMP, LAST_SAMP, LAST_SAMPLES if FREQ == 0: PHASE = 0 if LAST_SAMP == 0: + if options.gui: + LAST_SAMPLES.extend([0]*frames) return (Z_SAMP*frames, pyaudio.paContinue) fdata = lin_seq(LAST_SAMP, 0, frames) + if options.gui: + LAST_SAMPLES.extend(fdata) LAST_SAMP = fdata[-1] return (to_data(fdata), pyaudio.paContinue) fdata, PHASE = samps(FREQ, PHASE, frames) + if options.gui: + LAST_SAMPLES.extend(fdata) LAST_SAMP = fdata[-1] return (to_data(fdata), pyaudio.paContinue) pa = pyaudio.PyAudio() stream = pa.open(rate=RATE, channels=1, format=pyaudio.paInt32, output=True, frames_per_buffer=FPB, stream_callback=gen_data) +if options.gui: + guithread = threading.Thread(target=GUIs[options.gui]) + guithread.setDaemon(True) + guithread.start() + if options.test: FREQ = 440 time.sleep(1) |