#!/usr/bin/python
# -*- coding: utf-8 -*-

import pyaudio
import struct
import math
import wave
import numpy as np
import matplotlib.pyplot as plt
from scipy.io import wavfile
#import urllib
#import urllib2
import os
import smtplib, getpass
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

class Microphone:

    def rms(self,frame):
        count = len(frame)/2
        format = "%dh"%(count)
        shorts = struct.unpack( format, frame )
        sum_squares = 0.0
        for sample in shorts:
            n = sample * (1.0/32768.0)
            sum_squares += n*n
        rms = math.pow(sum_squares/count,0.5);
        return rms * 1000


    def passiveListen(self,persona):

        CHUNK = 1024; RATE = 8000; THRESHOLD = 200; LISTEN_TIME = 5

        didDetect = False

        # prepare recording stream
        p = pyaudio.PyAudio()
        stream = p.open(format=pyaudio.paInt16, channels=1, rate=RATE, input=True, frames_per_buffer=CHUNK)

        # stores the audio data
        all =[]

        # starts passive listening for disturbances
        Var_float = RATE / CHUNK * LISTEN_TIME
        Var_int = RATE // CHUNK * LISTEN_TIME
        for i in range(0,int(RATE / CHUNK * LISTEN_TIME)):
            input = stream.read(CHUNK)
            rms_value = self.rms(input)
            print (rms_value)
            if (rms_value > THRESHOLD):
                didDetect = True
                print ("Listening...\n")
                break

        if not didDetect:
            stream.stop_stream()
            stream.close()
            return False

        while didDetect==True:
            print ("**** Enviar email con Gmail ****")
            user = "LCDT.ytb@gmail.com"
            password = "joricombiomontado"

            #Para las cabeceras del email
            remitente = "LCDT.ytb@gmail.com"
            destinatario = "Korbatos@gmail.com"
            asunto = "Detectada actividad sospechosa"
            mensaje = "Se ha detectado un movimiento o sonido extraño en su habitacion. Por favor, revise la aplicacion para cerciorarse"

            #Host y puerto SMTP de Gmail
            gmail = smtplib.SMTP('smtp.gmail.com', 587)

            #Protocolo de cifrado de datos utilizado por Gmail
            gmail.starttls()

            #Credenciales
            gmail.login(user, password)

            #Muestra la depuración de la operación de envío l=true
            gmail.set_debuglevel(1)

            header = MIMEMultipart()
            header['Subject'] = asunto
            header['From'] = remitente
            header['To'] = destinatario

            mensaje = MIMEText(mensaje, 'html') #Content-type:text/html
            header.attach(mensaje)

            #Enviar email
            gmail.sendmail(remitente, destinatario, header.as_string())

            #Cerrar la conexion SMTP
            gmail.quit()
            break

        # append all the chunks
        all.append(input)
        for i in range(0, 7):
            data = stream.read(CHUNK)
            all.append(data)

        # save the audio data
        data = ''.join(str(e) for e in all)
        stream.stop_stream()
        stream.close()
        wf = wave.open('audio.wav', 'wb')
        wf.setnchannels(1)
        wf.setsampwidth(p.get_sample_size(pyaudio.paInt16))
        wf.setframerate(RATE)
        wf.writeframes(data)
        wf.close()

        return True

if __name__ == '__main__':
    mic = Microphone()
    while True:
        if  mic.passiveListen('ok Google'):
            fs, data = wavfile.read('audio.wav')
            L = len(data)
            c = np.fft.fft(data) # create a list of complex number
            freq = np.fft.fftfreq(L)
            #freq = np.linspace(0, 1/(2L), L/2)
            print (freq)
            freq_in_hertz = abs(freq * fs)
            plt.plot(freq_in_hertz, abs(c))
            plt.show()