import numpy
from PIL import Image, ImageDraw
from PyQt4 import uic, QtGui
from PyQt4.QtGui import QColor
import os, random
from . import __base__
import time
from copy import copy


class Component(__base__.Component):
    '''Original Audio Visualization'''
    def widget(self, parent):
        self.parent = parent
        self.visColor = (255,255,255)

        page = uic.loadUi(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'original.ui'))
        page.comboBox_visLayout.addItem("Classic")
        page.comboBox_visLayout.addItem("Split")
        page.comboBox_visLayout.addItem("Bottom")
        #visLayoutValue = int(self.settings.value('visLayout'))
        page.comboBox_visLayout.setCurrentIndex(0)
        page.comboBox_visLayout.currentIndexChanged.connect(self.update)
        page.lineEdit_visColor.setText('%s,%s,%s' % self.visColor)
        page.pushButton_visColor.clicked.connect(lambda: self.pickColor())
        btnStyle = "QPushButton { background-color : %s; outline: none; }" % QColor(*self.visColor).name()
        page.pushButton_visColor.setStyleSheet(btnStyle)
        page.lineEdit_visColor.textChanged.connect(self.update)
        self.page = page
        self.canceled = False
        return page
    
    def update(self):
        self.layout = self.page.comboBox_visLayout.currentIndex()
        self.visColor = self.RGBFromString(self.page.lineEdit_visColor.text())
        self.parent.drawPreview()

    def loadPreset(self, pr):
        self.page.lineEdit_visColor.setText('%s,%s,%s' % pr['visColor'])
        btnStyle = "QPushButton { background-color : %s; outline: none; }" % QColor(*pr['visColor']).name()
        self.page.pushButton_visColor.setStyleSheet(btnStyle)
        self.page.comboBox_visLayout.setCurrentIndex(pr['layout'])
        
    def savePreset(self):
        return { 'layout' : self.layout,
                  'visColor' : self.visColor,
                }

    def previewRender(self, previewWorker):
        spectrum = numpy.fromfunction(lambda x: 0.008*(x-128)**2, (255,), dtype="int16")
        width = int(previewWorker.core.settings.value('outputWidth'))
        height = int(previewWorker.core.settings.value('outputHeight'))
        return self.drawBars(width, height, spectrum, self.visColor, self.layout)
    
    def preFrameRender(self, **kwargs):
        super().preFrameRender(**kwargs)
        self.smoothConstantDown = 0.08
        self.smoothConstantUp = 0.8
        self.lastSpectrum = None
        self.spectrumArray = {}
        self.width = int(self.worker.core.settings.value('outputWidth'))
        self.height = int(self.worker.core.settings.value('outputHeight'))

        for i in range(0, len(self.completeAudioArray), self.sampleSize):
            if self.canceled:
                break
            self.lastSpectrum = self.transformData(i, self.completeAudioArray, self.sampleSize,
                self.smoothConstantDown, self.smoothConstantUp, self.lastSpectrum)
            self.spectrumArray[i] = copy(self.lastSpectrum)

            progress = int(100*(i/len(self.completeAudioArray)))
            if progress >= 100:
                progress = 100
            pStr = "Analyzing audio: "+ str(progress) +'%'
            self.progressBarSetText.emit(pStr)
            self.progressBarUpdate.emit(int(progress))
        
    def frameRender(self, moduleNo, arrayNo, frameNo):
        return self.drawBars(self.width, self.height, self.spectrumArray[arrayNo], self.visColor, self.layout)

    def pickColor(self):
        RGBstring, btnStyle = super().pickColor()
        if not RGBstring:
            return
        self.page.lineEdit_visColor.setText(RGBstring)
        self.page.pushButton_visColor.setStyleSheet(btnStyle)

    def transformData(self, i, completeAudioArray, sampleSize, smoothConstantDown, smoothConstantUp, lastSpectrum):
        if len(completeAudioArray) < (i + sampleSize):
            sampleSize = len(completeAudioArray) - i

        window = numpy.hanning(sampleSize)
        data = completeAudioArray[i:i+sampleSize][::1] * window
        paddedSampleSize = 2048
        paddedData = numpy.pad(data, (0, paddedSampleSize - sampleSize), 'constant')
        spectrum = numpy.fft.fft(paddedData)
        sample_rate = 44100
        frequencies = numpy.fft.fftfreq(len(spectrum), 1./sample_rate)

        y = abs(spectrum[0:int(paddedSampleSize/2) - 1])

        # filter the noise away
        # y[y<80] = 0

        y = 20 * numpy.log10(y)
        y[numpy.isinf(y)] = 0

        if lastSpectrum is not None:
            lastSpectrum[y < lastSpectrum] = y[y < lastSpectrum] * smoothConstantDown + lastSpectrum[y < lastSpectrum] * (1 - smoothConstantDown)
            lastSpectrum[y >= lastSpectrum] = y[y >= lastSpectrum] * smoothConstantUp + lastSpectrum[y >= lastSpectrum] * (1 - smoothConstantUp)
        else:
            lastSpectrum = y

        x = frequencies[0:int(paddedSampleSize/2) - 1]

        return lastSpectrum

    def drawBars(self, width, height, spectrum, color, layout):
        vH = height-height/8
        bF = width / 64
        bH = bF / 2
        bQ = bF / 4
        imTop = Image.new("RGBA", (width, height),(0,0,0,0))
        draw = ImageDraw.Draw(imTop)
        r, g, b = color
        color2 = (r, g, b, 125)

        bP = height / 1200

        for j in range(0, 63):
            draw.rectangle((bH + j * bF, vH+bQ, bH + j * bF + bF, vH + bQ - spectrum[j * 4] * bP - bH), fill=color2)
            draw.rectangle((bH + bQ + j * bF, vH , bH + bQ + j * bF + bH, vH - spectrum[j * 4] * bP), fill=color)

        imBottom = imTop.transpose(Image.FLIP_TOP_BOTTOM)

        im = Image.new("RGBA", (width, height),(0,0,0,0))

        if layout == 0:
            y = 0 - int(height/100*43)
            im.paste(imTop, (0, y), mask=imTop)
            y = 0 + int(height/100*43)
            im.paste(imBottom, (0, y), mask=imBottom)

        if layout == 1:
            y = 0 + int(height/100*10)
            im.paste(imTop, (0, y), mask=imTop)
            y = 0 - int(height/100*10)
            im.paste(imBottom, (0, y), mask=imBottom)

        if layout == 2:
            y = 0 + int(height/100*10)
            im.paste(imTop, (0, y), mask=imTop)

        return im