File size: 3,176 Bytes
9ae74ae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 | #!/usr/bin/env python
import distutils.spawn
import os
import sys
sys.path.append(os.path.dirname(distutils.spawn.find_executable("silent_tools.py")))
import silent_tools
from silent_tools import eprint
from collections import defaultdict
# Don't throw an error when someone uses head
from signal import signal, SIGPIPE, SIG_DFL
signal(SIGPIPE, SIG_DFL)
if (len(sys.argv) == 1):
eprint("")
eprint('silentinjectscores by bcov - a tool rewrite the scores in a silentfile')
eprint(' Uses mv-like syntax')
eprint("Usage:")
eprint(" silentrename myscores.sc myfile.silent > new.silent")
eprint("")
eprint("Important! The tags must match exactly. silentrename can rename scorefiles if needed")
eprint("")
eprint("This also works with two silent files")
eprint("")
sys.exit(1)
score_file = sys.argv[1]
silent_file = sys.argv[2]
assert(os.path.exists(silent_file))
assert(os.path.exists(score_file))
# First we need to make sure that there's only one kind of score inside the scorefile
# This doesn't technically need to be true, but I don't want to deal with it
tmpfile = silent_tools.cmd("head /dev/urandom | tr -dc A-Za-z0-9 | head -c 13").strip()
assert(len(tmpfile) > 0)
silent_tools.cmd("cp %s %s"%(score_file, tmpfile))
stdout,stderr,code = silent_tools.cmd2("silentscorefile %s"%tmpfile)
silent_tools.cmd("rm %s"%(tmpfile))
if ( code != 0 ):
silent_tools.cmd("rm %s.sc"%(tmpfile))
eprint(stderr)
sys.exit(1)
if ( "Multiple scorefiles" in stderr ):
silent_tools.cmd("rm %s.sc"%(tmpfile))
eprint("silentinjectscores: Scorefile contains multiple different score headers. Aborting")
sys.exit(1)
score_index = silent_tools.get_silent_index( tmpfile + ".sc" )
silent_index = silent_tools.get_silent_index( silent_file )
tag_exists = defaultdict(lambda : False, {})
bad = ""
for tag in score_index['tags']:
if ( tag in silent_index['tags'] ):
tag_exists[tag] = True
else:
bad = tag
break
for tag in silent_index['tags']:
if ( not tag_exists[tag] ):
bad = tag
break
if ( bad != "" ):
if ( len(score_index['tags']) != len(silent_index['tags'])):
eprint("silentinjectscores: Files don't have same number of structures")
else:
eprint("silentinjectscores: Mismatch between files. Example: %s"%bad)
silent_tools.cmd("rm %s.sc"%(tmpfile))
silent_tools.cmd("rm %s.sc.idx"%(tmpfile))
sys.exit(1)
sys.stdout.write( silent_tools.silent_header( score_index ) )
sys.stdout.flush()
with open(silent_file, errors='ignore') as sf:
with open(tmpfile + ".sc") as scoref:
for tag in silent_index['tags']:
structure = silent_tools.get_silent_structure_file_open( sf, silent_index, tag )
for i in range(len(structure)):
if ( structure[i].startswith("SCORE:") ):
structure[i] = silent_tools.get_silent_structure_file_open( scoref, score_index, tag )[0]
break
sys.stdout.write("".join(structure))
sys.stdout.flush()
silent_tools.cmd("rm %s.sc"%(tmpfile))
silent_tools.cmd("rm %s.sc.idx"%(tmpfile))
|