criticon.py

#coding:latin-1
#!/usr/bin/python

## taken from https://gist.github.com/loleg/5b581d774fc8500325f7
import os
import sys
import time
import random
import json
from picamera.array import PiRGBArray
from picamera import PiCamera
import cv2
#import zbar
import zbar
import festival
from PIL import Image
from generador_frases import GeneradorFrases

# Debug mode
DEBUG = False
if len(sys.argv) > 1:
    DEBUG = sys.argv[-1] == '-d'

# Configuration options

if not DEBUG:
    RESOLUTION = (640, 480)
else:
    RESOLUTION = (480, 270)

generador = GeneradorFrases(5)
codes_filename = "data/codes.txt"

with open(codes_filename) as f:
    codes = f.readlines()
    codes = [x.strip() for x in codes] 


# Initialise Raspberry Pi camera
camera = PiCamera()
camera.resolution = RESOLUTION
#camera.framerate = 10
#camera.vflip = True
#camera.hflip = True
#camera.color_effects = (128, 128)
# set up stream buffer
rawCapture = PiRGBArray(camera, size=RESOLUTION)
# allow camera to warm up
time.sleep(0.1)
print("PiCamera ready")

#seleccionar voz en espaõl
festival.execCommand("(voice_JuntaDeAndalucia_es_sf_diphone)")
# Initialise OpenCV window
if DEBUG:
    #cv2.namedWindow("#iothack15", cv2.WND_PROP_FULLSCREEN)
    #cv2.setWindowProperty("#iothack15", cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN)
    cv2.namedWindow("#criticon")


print("OpenCV version: %s" % (cv2.__version__))
print("Press q to exit ...")

#scanner = zbar.ImageScanner()
scanner = zbar.Scanner()

#scanner.parse_config('enable')


#festival.sayText("Criticon ha despertado, buscando codigo de barras")
saludo = u"criticón ha despertado, buscando código de barras."

os.system( "echo "+ saludo + " | iconv -f utf-8 -t iso-8859-1 | festival --tts")
#os.system( "echo "+ saludo + " | festival --tts")

# Capture frames from the camera
for frame in camera.capture_continuous(rawCapture, format="bgr", use_video_port=True):
    # as raw NumPy array
    output = frame.array.copy()

    # raw detection code
    gray = cv2.cvtColor(output, cv2.COLOR_BGR2GRAY, dstCn=0)
    pil = Image.fromarray(gray)
    width, height = pil.size
    raw = pil.tobytes()

    # create a reader
    #image = zbar.Image(width, height, 'Y800', raw)
    symbols = scanner.scan(gray)

    # extract results
    for symbol in symbols:
        # do something useful with results
        streeng = "decoded " + str(symbol.type) + " symbol " + str(symbol.data)
        code = str(symbol.data, 'utf-8')
        #code = _code.encode('utf-8')
        print(streeng)
        print(code)
        if code in codes:
            text = generador.autogenerar()
            #text = _text.encode('utf-8')
            print(text)
            cmd = "echo " + text + " | iconv -f utf-8 -t iso-8859-1 | festival --tts"
            os.system(cmd)
            #festival.sayText(text.encode('latin-1'))

        else:
            os.system( "echo criticón no entiende esta obra. Nada que decir | iconv -f utf-8 -t iso-8859-1 | festival --tts")



    # show the frame
    if DEBUG:
        cv2.imshow("#criticon", output)

    # clear stream for next frame
    rawCapture.truncate(0)

    # Wait for the magic key
    keypress = cv2.waitKey(1) & 0xFF
    if keypress == ord('q'):
        break

# When everything is done, release the capture
camera.close()
if DEBUG:
    cv2.destroyAllWindows()