D = OMPursuit.OMPursuitDictionary(path, dsfactor) print("Total time is %0.1f"%(time.clock()-p)) Ac = OMPursuit.OMPAnalysisConstraintSet(mpconstrPath) C = OMPursuit.OMPursuitCompoundConstraint(constraint_path + '/' + constraint) markers = OMPursuit.OMPursuitMarkers(markerPath) target = OMPursuit.OMPursuitTarget(targetPath, dsfactor) target.zeropadSignal(D, markers) #omdictionary, omcompoundconstraint, omtarget, ommarkers, maxtotal, mindistance, maxsimultaneous) #A = OMPursuit.OMPursuitAnalysis(D, target, markers, maxit, constraint=C, mpconstraint=Ac) A = OMPursuit.OMPursuitAnalysis(D, target, markers, maxit, constraint=None, mpconstraint=None) A.constrainedMP() audiolab.aiffwrite(A.ompModel.signal, os.path.expanduser('~/Research/OM-Pursuit/output/cry%d.aif'%ci), A.ompModel.samplerate) f = open(os.path.expanduser('~/Research/OM-Pursuit/output/cry%d.txt'%ci), 'w') f.write('Dictionary : %s\n'%path) f.write('Constraint : %s\n'%constraint) f.write('Markers : %s\n'%markerPath) f.write('Max. iterations : %d\n'%maxit) f.write('Total soundgrains in model : %d'%len(A.ompModel.parameterArray)) f.close() A.writeModelSdif('~/Research/OM-Pursuit/output/cry%d.sdif'%ci) #print(A.ompModel.parameterArray break
#for ci, constraint in enumerate(os.listdir(os.path.expanduser(constraintPaths[i]))): for ci, constraint in enumerate(constraintPaths): if os.path.splitext(constraint)[1].lower() == '.sdif': print(constraint) D = OMPursuit.OMPursuitDictionary(path, dsfactor) markers = OMPursuit.OMPursuitMarkers(markerPaths[i]) Ac = OMPursuit.OMPAnalysisConstraintSet(mpconstrPath) C = OMPursuit.OMPursuitCompoundConstraint(constraint) target = OMPursuit.OMPursuitTarget(targetPaths[i], dsfactor) target.zeropadSignal(D, markers) A = OMPursuit.OMPursuitAnalysis(D, target, markers, maxit, constraint=C, mpconstraint=Ac) A.constrainedMP() audiolab.aiffwrite(A.ompModel.signal, os.path.expanduser(outpath + '-%d.aif'%ci), A.ompModel.samplerate) #log the analysis data f = open(os.path.expanduser(outpath + '-%d.txt'%ci), 'w') f.write('Dictionary : %s\n'%path) f.write('Constraint : %s\n'%constraint) f.write('MPConstraint : %s\n'%mpconstrPath) f.write('Markers : %s\n'%markerPaths[i]) f.write('Max. iterations : %d\n'%maxit) f.write('Total soundgrains in model : %d'%len(A.ompModel.parameterArray)) f.close() A.writeModelSdif(outpath + '-%d.sdif'%ci)
Ac = OMPursuit.OMPAnalysisConstraintSet(args.mpconstraint_path) else: Ac = None if args.constraint_path: C = OMPursuit.OMPursuitCompoundConstraint(args.constraint_path) else: C = None markers = OMPursuit.OMPursuitMarkers(args.marker_path) target = OMPursuit.OMPursuitTarget(args.target_path, args.dsf) target.zeropadSignal(D, markers) A = OMPursuit.OMPursuitAnalysis(D, target, markers, args.max_iterations, constraint=C, mpconstraint=Ac) A.constrainedMP() audiolab.aiffwrite(A.ompModel.signal, os.path.expanduser(args.output_path), A.ompModel.samplerate) audiolab.aiffwrite(A.ompTarget.signal, os.path.expanduser(args.res_output_path), A.ompModel.samplerate) A.writeModelSdif(args.sdif_output_path) #duration of analysis procedure, target name, SRR if args.logfile != None: f = open(os.path.expanduser(args.logfile), 'w') f.write('Dictionary : %s\n'%args.dictionary_path) f.write('Constraint : %s\n'%args.constraint_path) f.write('MP Constraint : %s\n'%args.mpconstraint_path) f.write('Markers : %s\n'%args.marker_path) f.write('Max. iterations : %d\n'%args.max_iterations) f.write('Total soundgrains in model : %d'%len(A.ompModel.parameterArray)) f.close()
markers = OMPursuit.OMPursuitMarkers(markerPaths[i]) Ac = OMPursuit.OMPAnalysisConstraintSet(mpconstrPath) C = OMPursuit.OMPursuitCompoundConstraint(constraint) target = OMPursuit.OMPursuitTarget(targetPaths[i], dsfactor) target.zeropadSignal(D, markers) A = OMPursuit.OMPursuitAnalysis(D, target, markers, maxit, constraint=C, mpconstraint=Ac) A.constrainedMP() audiolab.aiffwrite(A.ompModel.signal, os.path.expanduser(outpath + '-%d.aif' % ci), A.ompModel.samplerate) #log the analysis data f = open(os.path.expanduser(outpath + '-%d.txt' % ci), 'w') f.write('Dictionary : %s\n' % path) f.write('Constraint : %s\n' % constraint) f.write('MPConstraint : %s\n' % mpconstrPath) f.write('Markers : %s\n' % markerPaths[i]) f.write('Max. iterations : %d\n' % maxit) f.write('Total soundgrains in model : %d' % len(A.ompModel.parameterArray)) f.close() A.writeModelSdif(outpath + '-%d.sdif' % ci)