forked from sugarlabs/speak
/
fft_mouth.py
128 lines (99 loc) · 4.09 KB
/
fft_mouth.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
# Speak.activity
# A simple front end to the espeak text-to-speech engine on the XO laptop
# http://wiki.laptop.org/go/Speak
#
# Copyright (C) 2008 Joshua Minor
# This file is part of Speak.activity
#
# Parts of Speak.activity are based on code from Measure.activity
# Copyright (C) 2007 Arjun Sarwal - arjun@laptop.org
#
# Speak.activity is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Speak.activity is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Speak.activity. If not, see <http://www.gnu.org/licenses/>.
# This code is a stripped down version of the fft view from Measure
from mouth import *
# Newer OLPC builds (and Fedora) have numpy instead of numeric:
try:
from numpy.oldnumeric import ceil
from numpy.fft import *
except:
from Numeric import ceil
from FFT import *
class FFTMouth(Mouth):
def __init__(self, audioSource, fill_color):
Mouth.__init__(self, audioSource, fill_color)
self.peaks = []
self.y_mag = 1.7
self.freq_range=70
self.draw_interval = 1
self.num_of_points = 105
self.stop=False
self.y_mag_bias_multiplier = 1 #constant to multiply with self.param2 while scaling values
self.fftx = []
self.scaleX = "10"
self.scaleY = "10"
def processBuffer(self, bounds):
self.param1 = bounds.height/65536.0
self.param2 = bounds.height/2.0
if(self.stop==False):
Fs = 48000
nfft= 65536
self.newest_buffer=self.newest_buffer[0:256]
self.fftx = fft(self.newest_buffer, 256,-1)
self.fftx=self.fftx[0:self.freq_range*2]
self.draw_interval=bounds.width/(self.freq_range*2.)
NumUniquePts = ceil((nfft+1)/2)
self.buffers=abs(self.fftx)*0.02
self.y_mag_bias_multiplier=0.1
self.scaleX = "hz"
self.scaleY = ""
if(len(self.buffers)==0):
return False
# Scaling the values
val = []
for i in self.buffers:
temp_val_float = float(self.param1*i*self.y_mag) + self.y_mag_bias_multiplier * self.param2
if(temp_val_float >= bounds.height):
temp_val_float = bounds.height-25
if(temp_val_float <= 0):
temp_val_float = 25
val.append( temp_val_float )
self.peaks = val
def expose(self, widget, event):
"""This function is the "expose" event handler and does all the drawing."""
bounds = self.get_allocation()
self.processBuffer(bounds)
#Create context, disable antialiasing
self.context = widget.window.cairo_create()
self.context.set_antialias(cairo.ANTIALIAS_NONE)
#set a clip region for the expose event. This reduces redrawing work (and time)
self.context.rectangle(event.area.x, event.area.y,event.area.width, event.area.height)
self.context.clip()
# background
self.context.set_source_rgba(*self.fill_color.get_rgba())
self.context.rectangle(0,0, bounds.width,bounds.height)
self.context.fill()
# Draw the waveform
self.context.set_line_width(min(bounds.height/10.0, 10))
self.context.set_source_rgb(0,0,0)
count = 0
for peak in self.peaks:
self.context.line_to(bounds.width/2 + count,bounds.height/2 - peak)
count += self.draw_interval
self.context.stroke()
count = 0
for peak in self.peaks:
self.context.line_to(bounds.width/2 - count,bounds.height/2 - peak)
count += self.draw_interval
self.context.stroke()
return True