from shutil import *
import cPickle

def name2class(name,B):
	if len(B[where(name==B[:,0]),1][0])!=0:
		return B[where(name==B[:,0]),1][0]
	else:
		return asarray(-10)


#CREATE THE LINK NAME CLASSES
a=open('name_class.pkl','r')
b=cPickle.load(a)
a.close()
for i in range(len(b)):
	b[i,0]=b[i,0][:-11]

from os import *
import glob
from numpy import *

#CREATE THE SETS
chdir('/NAS3/SABIOD/public_data/SCATT_BIRD_50mostE_perfile_april2014_RB_HG/DELTA_ACT')


chdir('./TXT')
txtdeltalist=[[],[],[]]
txtdeltalist[0]=glob.glob('L*.txt')
print len(txtdeltalist[0])
#for i in range(len(txtdeltalist[0])):
#        txtdeltalist[2].append(loadtxt(txtdeltalist[0][i]))
#	print txtdeltalist[0][i].split('wav')[0]
#        txtdeltalist[0][i]=txtdeltalist[0][i].split('wav')[0]+'wav.txt'
#	print txtdeltalist[0][i][:-4]
#        txtdeltalist[1].append(name2class(txtdeltalist[0][i][:-4],b))
#	savetxt('/NAS3/SABIOD/public_data/BIRD_DELTAEE_FFT/TXT/DELTAEE/'+str(int(txtdeltalist[1][i].astype('int')))+'/'+txtdeltalist[0][i],txtdeltalist[2][i])
#	print i


#txtdeltalist=asarray(txtdeltalist)

txtfftlist=[[],[],[]]
txtfftlist[0]=glob.glob('FFT*.txt')
for i in range(len(txtfftlist[0])):
        txtfftlist[2].append(loadtxt(txtfftlist[0][i]))
        txtfftlist[0][i]=txtfftlist[0][i][4:]
	txtfftlist[0][i]=txtfftlist[0][i].split('wav')[0]+'wav.txt'
        txtfftlist[1].append(name2class(txtfftlist[0][i][:-4],b))
	savetxt('/NAS3/SABIOD/public_data/BIRD_DELTAEE_FFT/TXT/FFT/'+str(int(txtfftlist[1][i].astype('int')))+'/'+txtfftlist[0][i],txtfftlist[2][i])
        print i