happy.py

import scipy.io.wavfile as wavfile
import numpy as np
import pylab as pl
import time
import os
import sys
import subprocess
from scipy import mean
from random import randint

# music wav file
FILE = "karaoke.wav"
rate, data = wavfile.read(FILE)
t_total = len(data[:,0])/rate
display_rate = 1500 #number of frames processed in one iteration
sample_size = 120
max_display = 90
data_length  = len(data) #total number of frames
_min = min([abs(x) for x in data[:,0]]) #max amplitude in the wav
_max = max([abs(x) for x in data[:,0]]) #min amplitude in the wav

# IMPORTANT: correction factor. Change this value to match the song with equalizer
correction = 0.645

# cake settings
cols = int(subprocess.Popen("tput cols",shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE).stdout.readlines()[0]) #columns in terminal
display_char = "8"
cake_size = 50

# flame control
flame_flutter_rate = 50
FLAMES = [ " . ", ".  ", "  ." ]
current_flame = ""

os.system("tput civis") #hide cursor

# TODO open file with some player. If you are on mac, uncomment following line
#os.system("open "+FILE)

for _f in range(data_length/display_rate):

    # fluttering effect to candle flames
    if _f%flame_flutter_rate == 0:
        current_flame = (" "*(cols/2 - cake_size/2))+((" "+FLAMES[randint(0,2)]+" ")*(cake_size/5))
    print current_flame

    # candles
    print (" "*(cols/2 - cake_size/2))+("  |  "*(cake_size/5))
    # cake top layer
    print (" "*(cols/2 - cake_size/2))+("-"*cake_size)

    bucket = []
    mug = []
    # mug contains the current frame samples (absolute values) of given sample_size
    # average of mugs are put into bucket
    for value in data[:,0][_f*display_rate+1:(_f+1)*display_rate]:
        mug.append(abs(value))
        if len(mug) == sample_size:
            bucket.append(mean(mug))
            mug = []
    bucket = [ (float)((x - _min) * max_display)/(_max - _min) for x in bucket ]

    # print the equalizer from the bucket
    for value in bucket:
        print (" "*(cols/2 - cake_size/2))+"| "+("8"*(value%(cake_size-2)))+(" "*(cake_size-value-2))+"|"

    # bottom crust of the cake
    print (" "*(cols/2 - cake_size/2))+("-"*cake_size)

    # print happy birthday message
    os.system("figlet -c -f small Happy Birthday qwerty!")

    # sleep to match with the audio
    """
        NOTE: correction has to be multiplied to sleep time
        This is because of several factors like time taken to wake from sleep, type of terminal used..etc
        CHANGE THE VALUE OF correction TO FIT YOUR NEED
    """
    time.sleep(((float)(display_rate * t_total) / data_length)*correction)

    # clear screen
    if _f != data_length/display_rate-1:
        os.system("clear")

raw_input()