From dd07501c5cf70bab4f4430a299bc6f5880490678 Mon Sep 17 00:00:00 2001
From: Manuel Cortez <manuel@manuelcortez.net>
Date: Mon, 24 Dec 2018 17:53:59 -0600
Subject: [PATCH] Added audio recorder controller.

---
 src/controller/audioRecorder.py | 131 ++++++++++++++++++++++++++++++++
 1 file changed, 131 insertions(+)
 create mode 100644 src/controller/audioRecorder.py

diff --git a/src/controller/audioRecorder.py b/src/controller/audioRecorder.py
new file mode 100644
index 0000000..b6a1900
--- /dev/null
+++ b/src/controller/audioRecorder.py
@@ -0,0 +1,131 @@
+# -*- coding: utf-8 -*-
+import time
+import os
+import tempfile
+import sound_lib
+import widgetUtils
+import sound
+import output
+from wxUI.dialogs import audioRecorder as gui
+from mysc.thread_utils import call_threaded
+
+class audioRecorder(object):
+	def __init__(self):
+		self.dialog = gui.audioRecorderDialog()
+		self.recorded = False
+		self.recording = None
+		self.duration = 0
+		self.playing = None
+		widgetUtils.connect_event(self.dialog.play, widgetUtils.BUTTON_PRESSED, self.on_play)
+		widgetUtils.connect_event(self.dialog.pause, widgetUtils.BUTTON_PRESSED, self.on_pause)
+		widgetUtils.connect_event(self.dialog.record, widgetUtils.BUTTON_PRESSED, self.on_record)
+		widgetUtils.connect_event(self.dialog.discard, widgetUtils.BUTTON_PRESSED, self.on_discard)
+		if self.dialog.get_response() == widgetUtils.OK:
+			self.postprocess()
+
+	def on_pause(self, *args, **kwargs):
+		if self.dialog.get("pause") == _(u"Pause"):
+			self.recording.pause()
+			self.dialog.set("pause", _(u"&Resume"))
+		elif self.dialog.get("pause") == _(u"Resume"):
+			self.recording.play()
+			self.dialog.set("pause", _(U"&Pause"))
+
+	def on_record(self, *args, **kwargs):
+		if self.recording != None:
+			self.stop_recording()
+			self.dialog.disable_control("pause")
+		else:
+			self.start_recording()
+			self.dialog.enable_control("pause")
+
+	def start_recording(self):
+		self.dialog.disable_control("attach_exists")
+		self.file = tempfile.mktemp(suffix='.wav')
+		self.recording = sound.get_recording(self.file)
+		self.duration = time.time()
+		self.recording.play()
+		self.dialog.set("record", _(u"&Stop"))
+		output.speak(_(u"Recording"))
+
+	def stop_recording(self):
+		self.recording.stop()
+		self.duration = int(time.time()-self.duration)
+		self.recording.free()
+		output.speak(_(u"Stopped"))
+		self.recorded = True
+		self.dialog.set("record", _(u"&Record"))
+		self.file_attached()
+
+	def file_attached(self):
+		self.dialog.set("pause", _(u"&Pause"))
+		self.dialog.disable_control("record")
+		self.dialog.enable_control("play")
+		self.dialog.enable_control("discard")
+		self.dialog.disable_control("attach_exists")
+		self.dialog.enable_control("attach")
+		self.dialog.play.SetFocus()
+
+	def on_discard(self, *args, **kwargs):
+		if self.playing:
+			self._stop()
+		if self.recording != None:
+			self.cleanup()
+		self.dialog.disable_control("attach")
+		self.dialog.disable_control("play")
+		self.file = None
+		self.dialog.enable_control("record")
+		self.dialog.enable_control("attach_exists")
+		self.dialog.record.SetFocus()
+		self.dialog.disable_control("discard")
+		self.recording = None
+		output.speak(_(u"Discarded"))
+
+	def on_play(self, *args, **kwargs):
+		if not self.playing:
+			call_threaded(self._play)
+		else:
+			self._stop()
+
+	def _play(self):
+		output.speak(_(u"Playing..."))
+#  try:
+		self.playing = sound_lib.stream.FileStream(file=unicode(self.file), flags=sound_lib.stream.BASS_UNICODE)
+		self.playing.play()
+		self.dialog.set("play", _(u"&Stop"))
+		try:
+			while self.playing.is_playing:
+				pass
+			self.dialog.set("play", _(u"&Play"))
+			self.playing.free()
+			self.playing = None
+		except:
+			pass
+
+	def _stop(self):
+		output.speak(_(u"Stopped"))
+		self.playing.stop()
+		self.playing.free()
+		self.dialog.set("play", _(u"&Play"))
+		self.playing = None
+
+	def postprocess(self):
+		if self.file.lower().endswith('.wav'):
+			output.speak(_(u"Recoding audio..."))
+			sound.recode_audio(self.file)
+			self.wav_file = self.file
+			self.file = '%s.ogg' % self.file[:-4]
+
+	def cleanup(self):
+		if self.playing and self.playing.is_playing:
+			self.playing.stop()
+		if self.recording != None:
+			if self.recording.is_playing:
+				self.recording.stop()
+			try:
+				self.recording.free()
+			except:
+				pass
+			os.remove(self.file)
+			if hasattr(self, 'wav_file'):
+				os.remove(self.wav_file)