comparison pyspark/sonic-annotator-notimeside/sonic_annotator_vamp.py @ 0:e34cf1b6fe09 tip

commit

+# Part of DML (Digital Music Laboratory)
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+"""
+Created on Fri Oct 11 13:22:37 2013
+
+@author: thomas
+"""
+
+from __future__ import division
+
+import sys
+import shutil
+import os
+import errno
+import subprocess
+import time 
+import random
+import hashlib
+
+# uses a separate console process to achieve the file conversion
+def vamp_host_process(argslist):
+    #"""Call sonic annotator"""
+
+    vamp_host = 'sonic-annotator'
+    command = [vamp_host]
+    command.extend(argslist)
+
+    # which sa version?
+    #p = subprocess.Popen([vamp_host, '-v'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    #print(p.stdout.read())
+
+
+    #stdout = subprocess.check_output(command, stderr=subprocess.STDOUT)
+    time.sleep(random.random()*1.0)
+    p = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    text = p.stdout.read()
+    retcode = p.wait()
+    if (retcode==0):
+        print "Finished."
+        return 1
+    else:
+        print "Error " + text
+        return 0
+    
+    #time.sleep(random.random()*1.0)
+    #print command
+    #subprocess.call(command,shell=True)
+    return 1
+
+
+# processes the given file using a vamp plugin and sonic annotator
+# @param string wav_file file to be processed
+def transform(wav_file = 'sweep.mp3',
+              transform_file = 'bbc_speechmusic.n3'):
+
+    # get transform hash
+    BLOCKSIZE = 65536
+    hasher = hashlib.sha1()
+    with open(transform_file, 'rb') as afile:
+        buf = afile.read(BLOCKSIZE)
+        while len(buf) > 0:
+            hasher.update(buf)
+            buf = afile.read(BLOCKSIZE)
+    tpath = os.path.split(transform_file)
+    
+    # prepare output directory:
+    # get audio file subpath
+    # create directory _Features for output if doesnt exist
+    spath = os.path.split(wav_file)
+    hash = str(hasher.hexdigest())
+    featpath = spath[0] + '/_Analysis/' + tpath[1] + "_" + hash[:5]
+    
+    #if not os.path.exists(featpath):
+    #    os.makedirs(featpath)
+    print 'Creating directory' + featpath
+    try:
+        os.makedirs(featpath)
+    except OSError as exception:
+        if exception.errno!= errno.EEXIST:
+            raise   
+    
+    # copy transform file into directory
+    shutil.copy(transform_file, featpath + '/' +  tpath[1][:-3] + '_' + hash[:5] + '.n3')
+
+    #./sonic-annotator -t silvet_settings.n3 input.wav -w csv
+    # prepare arguments
+    args = ['-t', transform_file, wav_file, '-w', 'csv', '-w', 'rdf',
+            '--rdf-basedir',featpath,'--csv-basedir',featpath, '--rdf-many-files', '--rdf-append']
+    #args = ['-t', transform_file, wav_file, '-w', 'csv', '--csv-force','--csv-basedir',featpath]
+
+
+    print "Analysing " + wav_file
+    
+    result = vamp_host_process(args)
+    # execute vamp host
+    return [wav_file, result]
+
+
+# entry function only for testing
+if __name__ == "__main__":
+    if len(sys.argv) >= 2:
+        transform(sys.argv[1])
+    else:
+        transform()
+