waveforms 1.6.1__tar.gz → 1.6.2__tar.gz

{waveforms-1.6.1/waveforms.egg-info → waveforms-1.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: waveforms
-Version: 1.6.1
+Version: 1.6.2
 Summary: Edit waveforms used in experiment
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>

{waveforms-1.6.1 → waveforms-1.6.2}/waveforms/version.py

@@ -1,2 +1,2 @@
 """Define version number here and read it from setup.py automatically"""
-__version__ = "1.6.1"
+__version__ = "1.6.2"

{waveforms-1.6.1 → waveforms-1.6.2}/waveforms/waveform.py

@@ -687,6 +687,50 @@ class Waveform:
             ])
         return "$$\n{}\n$$".format(expr)
 
+    def _play(self, time_unit):
+        RATE = 48000
+        y = self.sample(sample_rate=RATE / time_unit)
+        play(y, RATE)
+
+    def play(self, time_unit=1):
+        import multiprocessing as mp
+        p = mp.Process(target=self._play, args=(time_unit, ), daemon=True)
+        p.start()
+
+
+def play(data, rate=48000):
+    import io
+    import pyaudio
+
+    CHUNK = 1024
+
+    max_amp = np.max(np.abs(data))
+
+    if max_amp > 1:
+        data /= max_amp
+
+    data = np.array(2**15 * 0.999 * data, dtype=np.int16)
+    buff = io.BytesIO(data.data)
+    p = pyaudio.PyAudio()
+
+    try:
+        stream = p.open(format=pyaudio.paInt16,
+                        channels=1,
+                        rate=rate,
+                        output=True)
+        try:
+            while True:
+                data = buff.read(CHUNK)
+                if data:
+                    stream.write(data)
+                else:
+                    break
+        finally:
+            stream.stop_stream()
+            stream.close()
+    finally:
+        p.terminate()
+
 
 _zero_waveform = Waveform()
 _one_waveform = Waveform(seq=(_one, ))

{waveforms-1.6.1 → waveforms-1.6.2/waveforms.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: waveforms
-Version: 1.6.1
+Version: 1.6.2
 Summary: Edit waveforms used in experiment
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>