from serial   import Serial
from time     import sleep
from pykakasi import kakasi

import time
import sys

class text2speech(object):

	SERIALPORT = '/dev/ttyAMA0'
	DEVICE     = 'ATP3012'

	TIMEOUT   = 3
	BAUDRATE = 9600
	DICT_BEFORE = {'霊夢':'reimu','は、':'wa,','？':'?','！':'\'.','～':'kara',
		'（':',','）':',','「':',','」':',','、':',','':'','。':'.','．':'.',}
	DICT_AFTER = {'uu':'u-','oo':'o-','ou':'o-','ee':'e-','aa':'a-','ii':'i-','iu':'yu-','ha,':'wa,'}

	def __init__(self, model='ATP3012'):

		self.model=model

		self.ser = Serial(self.SERIALPORT, baudrate=self.BAUDRATE, timeout=self.TIMEOUT)

		self.kks = kakasi()
		self.kks.setMode('H', 'a')
		self.kks.setMode('K', 'a')
		self.kks.setMode('J', 'a')
		self.cnv = self.kks.getConverter()

		if model == 'ATP3011':
			self.ser.write(bytes('?','ascii'))
			rx = self.ser.read()

	def kanji2roman(self, msg):

		for target, dest in self.DICT_BEFORE.items():
			msg = msg.replace(target,dest)

		msg = self.cnv.do(msg)

		for target, dest in self.DICT_AFTER.items():
			msg = msg.replace(target,dest)

		return msg;

	def write(self, msg):

		#self.msg = self.kanji2roman(msg) + "\r"
		self.msg = msg

		while True:

			self.ser.write(bytes("\r",'ascii'))
			rx = self.ser.read()
			status = rx.decode('utf-8') # status is null

			if status=='*':    # busy
				sleep(1)
			elif status=='>':  # ready
				self.ser.write(bytes(self.msg,'ascii'))
				rx = self.ser.read()
				break

if __name__ == '__main__':

	import sys, os

	if len(sys.argv) == 2:
		model = sys.argv[1]
	else:
		model = text2speech.DEVICE

	if not os.path.exists(text2speech.SERIALPORT):
		sys.exit("ERROR: Serial port not found at: %s" % text2speech.SERIALPORT)

	print ("Testing AquesTalk Pico LSI: %s" % model)

	ts = text2speech(model=model)

	text='きょうの天気は、曇り時々晴れです'
	ts.write(text)