view Code/time_freq_representation/utils.py @ 47:b0186d4a4496 tip

Move 7Digital dataset to Downloads

"""
Source code:
https://github.com/sidsig/ICASSP-MLP-Code/blob/master/utils.py
"""

import os

def getFiles(root_dir,ext='.mp3',verbose=True) :
    """
    Returns a list of files
    """
    fileList=[]
    if verbose:
        print "Populating %s files..."%ext
    for (root,dirs,files) in os.walk(root_dir):
        for f in files:
            if f.endswith(ext):
                filePath=os.path.join(root,f)
                fileList.append(filePath)
    if verbose:
        print "%i files found."%len(fileList)
    return fileList

def parseFile(filePath):
    """
    Parses the file path and returns (root,fileName,ext)
    """
    root,file=os.path.split(filePath)
    fileName,fileExt=os.path.splitext(file)
    return (root,fileName,fileExt)

def read_file(filename):
    """
    Loads a file into a list
    """
    file_list=[l.strip() for l in open(filename,'r').readlines()]
    return file_list

def writeFile(dataList,filename):
    with open(filename,'w') as f:
        f.writelines(dataList)