import os
import shutil

dirname='/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/BD50CLASSES/'

for i in range(50):
    if not os.path.exists(dirname+'FOLDERS/SUBTRAIN_SP'+str(i+1)):
        os.makedirs(dirname+'FOLDERS/SUBTRAIN_SP'+str(i+1))
    if not os.path.exists(dirname+'FOLDERS/SUBVALID_SP'+str(i+1)):
        os.makedirs(dirname+'FOLDERS/SUBVALID_SP'+str(i+1))


ls_wav=os.listdir(dirname+'SUBTRAIN')
ls_label=[]
for wav in ls_wav:
    label=wav[-10:-4]
    if label not in ls_label:
        ls_label=ls_label+[label]

ls_label.sort()

for wav in ls_wav:
    label=wav[-10:-4]
    fileID=wav[31:-11]
    ind=str(int(ls_label.index(label))+1)
    shutil.copy2(dirname+'SUBTRAIN/'+wav, dirname+'FOLDERS/SUBTRAIN_SP'+str(ind)+'/SABIOD50bird2015_SP'+str(ind)+'_'+label+'_'+fileID+'.wav')

ls_wav_bis=os.listdir(dirname+'SUBVALID')
for wav in ls_wav:
    label=wav[-10:-4]
    fileID=wav[31:-11]
    ind=str(int(ls_label.index(label))+1)
    shutil.copy2(dirname+'SUBVALID/'+wav, dirname+'FOLDERS/SUBVALID_SP'+str(ind)+'/SABIOD50bird2015_SP'+str(ind)+'_'+label+'_'+fileID+'.wav')