audio_visualizer.py

"""We are creating an Interactive Music Visualizer users input an MP3 file into
the program which generates a movie with data extracted from the music file.
This program will run prior to playing the movie and music at the same time, and
will preprocess the audio data to generate the Computational Art equations that
will be visualized later on. Then, during the movie (the visualization of these
equations), the user is able to manipulate the movie behavior throughout the
song by interacting with several keys on the keyboard. These keystrokes will
essentially swap colors in the equations but will not involve regenerating
equations."""

from __future__ import print_function
import shutil
import ntpath
import wave
import os
import glob
import eyed3
import numpy
from pydub import AudioSegment
import pygame
import matplotlib.pyplot as plt
from pyAudioAnalysis import audioFeatureExtraction
from pyAudioAnalysis import audioBasicIO
from pyAudioAnalysis import audioAnalysis

__author__ = 'Sampei and Sara'
__version__ = '0.0.1'

def get_wav():
    """Controller function - asks for wav file from user and ensures that it is
    a wav file
    """
    got_wav = False
    while got_wav == False:
        wav_file = input('What audio file would you like to visualize?: ')
        if wav_file[-4:] == '.wav':
            return got_wav
        else:
            print('You must enter an WAV file.')

def process_key(key):
    """ Controller function - identifies key input
    """
    print(chr(key))

def bpm_to_frame(bpm):
    '''Gets the required number of frames to loop given the average beats per minute from a song'''
    bps = 60 / bpm
    frames = bps * 24
    return frames

def key_q(key, frames, x, y):
    '''Plays the frames with the regular color scheme'''
    for frame in range(1,frames):
        to_display = pygame.image.load('/home/sampeiomichi/music_visualizer/ComputationalArt-sampeiomichi/movie1_'+str(frame)+'.png') #This hardcoded line would have to be changed
        gameDisplay.blit(to_display, (x,y))
        pygame.time.delay(24)
        pygame.display.update()

def key_w(key, frames, x, y):
    '''Plays the frames with the inverted color scheme'''
    for frame in range(1,frames):
        to_display = pygame.image.load('/home/sampeiomichi/music_visualizer/ComputationalArt-sampeiomichi/movie2_'+str(frame)+'.png') #This hardcoded line would have to be changed
        gameDisplay.blit(to_display, (x,y))
        pygame.time.delay(24)
        pygame.display.update()

def key_e(key, frames, x, y):
    '''Plays the frames with the greyscaled color scheme'''
    for frame in range(1,frames):
        to_display = pygame.image.load('/home/sampeiomichi/music_visualizer/ComputationalArt-sampeiomichi/movie3_'+str(frame)+'.png') #This hardcoded line would have to be changed
        gameDisplay.blit(to_display, (x,y))
        pygame.time.delay(24)
        pygame.display.update()

if __name__ == "__main__": #Main
    file = '/home/sampeiomichi/music_visualizer/pyAudioAnalysis/data/beat/insideout.mp3' #This hardcoded line would have to be changed
    frames = int(bpm_to_frame(60)) #Gets the number of frames that need to be looped given 60 BPM
    pygame.init()
    pygame.mixer.init()
    pygame.mixer.music.load(file)
    pygame.mixer.music.play()
    display_width = 350
    display_height = 350

    gameDisplay = pygame.display.set_mode((display_width,display_height))
    end = False
    state = 1
    x = 0
    y = 0
    click = False
    while not end:
        input = ''
        key = ''
        for event in pygame.event.get(): #Takes a keystroke as an input
            if event.type == pygame.KEYDOWN:
                key = event.key
        if key == ord('z'): #Quits the program when 'z' is pressed
            end = True
        elif key == ord('q'): #Changes the state to the regular color scheme
            print('Regular')
            state = 1
        elif key == ord('w'): #Changes the state to the inverted color scheme
            print('Colors Inverted!')
            state = 2
        elif key == ord('e'): #Changes the state to the greyscaled color scheme
            print('Image Greyscaled!')
            state = 3
        elif key == 273: #Moves the screen upwards when up key is pressed
            if y >= -650 and y < 0:
                print('Up')
                y = y + 50
        elif key == 274: #Moves the screen downwards when down key is pressed
            if y > -650 and y <= 0:
                print('Down')
                y = y - 50
        elif key == 276: #Moves the screen left when left key is pressed
            if x >= -650 and x < 0:
                print('Left')
                x = x + 50
        elif key == 275: #Moves the screen right when right key is pressed
            if x > -650 and x <= 0:
                print('Right')
                x = x - 50
        if state == 1: #Loops over the regular color scheme
            key_q(key, frames, x, y)
        elif state == 2: #Loops over the inverted color scheme
            key_w(key, frames, x, y)
        elif state == 3: #Loops over the greyscaled color scheme
            key_e(key, frames, x, y)