aud_audio.py

'''
SUMMARY:  importing file
AUTHOR:   adityac8
Created:  2018.03.28
Modified: 2018.03.28
--------------------------------------
'''

from __future__ import print_function
from __future__ import division
import os
import cPickle
import modules as M
import feature_description as F

def call_ftr(feature_name,featx,wav_fd,fe_fd,print_arr):
    flag1 = True if 'names' in print_arr else False
    flag2 = True if 'shape' in print_arr else False
    try:
        M.CreateFolder(fe_fd)
        M.rem_all_files(fe_fd)
    except Exception as e:
        print("Error.",e)
        return    
    
    names = [ na for na in os.listdir(wav_fd) if na.endswith('.wav') ]
    names = sorted(names)
    if names==[]:
        print('Empty folder found!! Try Again.')
        return
    for na in names:
        path = wav_fd + '/' + na
        # Introduce features here
        X=M.call_ftr_one(feature_name,featx,path)
        if flag1:
            print(na)
        if flag2:
            print(X.shape)
        out_path = fe_fd + '/' + na[0:-4] + '.f'
        cPickle.dump( X, open(out_path, 'wb'), protocol=cPickle.HIGHEST_PROTOCOL )
    print("extraction complete!")



def extract(feature_name,wav_fd=None,fe_fd=None,yaml_file='',print_arr=[]):
    """
    This function extracts features from audio.

    Parameters
    ----------
        feature_name : str
            Name of feature to be extracted
        wav_fd : str
            Path to audio files
        fe_fd : str
            Path to feature files
        yaml_file : str
            Path to yaml file
        print_arr : str array, optional
            Description of feature that should be printed 
    """
    # Introduce features here
    if feature_name in M.get_list():
        yaml_load=M.read_yaml(yaml_file)
        try:            
            featx=yaml_load[feature_name]
        except Exception as e:
            print("Make sure you add the {} to the YAML file".format(e))
            raise SystemExit
        x=call_ftr(feature_name,featx,wav_fd,fe_fd,print_arr)
        print("Something wrong happened" if x==1000 else "Feature found")
    else:
        print("Invalid Feature Name")