81 lines
2.0 KiB
Python
81 lines
2.0 KiB
Python
|
import os,json,pickle,csv
|
||
|
|
from shutil import copyfile
|
||
|
|
|
||
|
|
def event_times( eventTimeFn ):
|
||
|
|
|
||
|
|
eventL = []
|
||
|
|
|
||
|
|
with open(eventTimeFn,"r") as f:
|
||
|
|
|
||
|
|
rdr = csv.reader(f)
|
||
|
|
|
||
|
|
for row in rdr:
|
||
|
|
if row[0] == 'start':
|
||
|
|
beginMs = int(row[1])
|
||
|
|
elif row[0] == 'key_down':
|
||
|
|
key_downMs = int(row[1]) - beginMs
|
||
|
|
elif row[0] == 'key_up':
|
||
|
|
key_upMs = row[1]
|
||
|
|
|
||
|
|
eventL.append( [ key_downMs, key_downMs+1000 ] )
|
||
|
|
|
||
|
|
return eventL
|
||
|
|
|
||
|
|
def pulse_lengths( pulseLenFn ):
|
||
|
|
|
||
|
|
with open(pulseLenFn,'rb') as f:
|
||
|
|
d = pickle.load(f)
|
||
|
|
msL = d['msL']
|
||
|
|
# note: first posn in table is a multiplier
|
||
|
|
return [ msL[i]*msL[0] for i in range(1,len(msL))]
|
||
|
|
|
||
|
|
|
||
|
|
def convert( inDir, outDir ):
|
||
|
|
|
||
|
|
if not os.path.isdir(outDir):
|
||
|
|
os.mkdir(outDir)
|
||
|
|
|
||
|
|
for dirStr in os.listdir(inDir):
|
||
|
|
|
||
|
|
idir = os.path.join( inDir, dirStr )
|
||
|
|
|
||
|
|
if os.path.isdir(idir):
|
||
|
|
|
||
|
|
eventTimeFn = os.path.join( idir, "labels_0.csv" )
|
||
|
|
|
||
|
|
eventTimeL = event_times(eventTimeFn)
|
||
|
|
|
||
|
|
pulseTimeFn = os.path.join( idir, "table_0.pickle")
|
||
|
|
|
||
|
|
pulseUsL = pulse_lengths( pulseTimeFn )
|
||
|
|
|
||
|
|
pitch = idir.split("/")[-1]
|
||
|
|
|
||
|
|
|
||
|
|
d = {
|
||
|
|
"pulseUsL":pulseUsL,
|
||
|
|
"pitchL":[ pitch ],
|
||
|
|
"noteDurMs":1000,
|
||
|
|
"pauseDurMs":0,
|
||
|
|
"holdDutyPct":50,
|
||
|
|
"eventTimeL":eventTimeL,
|
||
|
|
"beginMs":0
|
||
|
|
}
|
||
|
|
|
||
|
|
odir = os.path.join( outDir, pitch )
|
||
|
|
if not os.path.isdir(odir):
|
||
|
|
os.mkdir(odir)
|
||
|
|
|
||
|
|
with open(os.path.join( odir, "seq.json" ),"w") as f:
|
||
|
|
f.write(json.dumps( d ))
|
||
|
|
|
||
|
|
copyfile( os.path.join(idir,"audio_0.wav"), os.path.join(odir,"audio.wav"))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
inDir = "/home/kevin/temp/picadae_ac_2/full_map"
|
||
|
|
outDir = "/home/kevin/temp/p_ac_3_cvt"
|
||
|
|
|
||
|
|
convert( inDir, outDir )
|
||
|
|
|