mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-02-05 11:40:04 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use a profile pathname in worker.py

Browse Source
This commit is contained in:
Steven Robertson 2012-01-21 00:03:45 -05:00
parent a803216551
commit 0aa0106f51
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
worker.py
View File

@ -113,7 +113,7 @@ def work(server):
push_frame(r, riter.send(((sid, sidx, ftag), float(ftime)))) push_frame(r, riter.send(((sid, sidx, ftag), float(ftime))))
def iter_genomes(prof, gpaths): def iter_genomes(prof, gpaths, pname='540p'):
""" """
Walk a list of genome paths, yielding them in an order suitable for Walk a list of genome paths, yielding them in an order suitable for
the `genomes` argument of `create_jobs()`. the `genomes` argument of `create_jobs()`.
@ -125,7 +125,7 @@ def iter_genomes(prof, gpaths):
gsrc = fp.read() gsrc = fp.read()
gnm = genome.Genome(json.loads(gsrc)) gnm = genome.Genome(json.loads(gsrc))
err, times = gnm.set_profile(prof) err, times = gnm.set_profile(prof)
odir = 'out/540p/%s/untracked' % gname odir = 'out/%s/%s/untracked' % (pname, gname)
gtimes = [] gtimes = []
for i, t in enumerate(times): for i, t in enumerate(times):
opath = os.path.join(odir, '%05d.jpg' % (i+1)) opath = os.path.join(odir, '%05d.jpg' % (i+1))
@ -230,8 +230,9 @@ def client(ppath, gpaths):
with open(ppath) as fp: with open(ppath) as fp:
psrc = fp.read() psrc = fp.read()
prof = json.loads(psrc) prof = json.loads(psrc)
pname = os.path.basename(ppath).rsplit('.', 1)[0]
jobiter = create_jobs(r, psrc, iter_genomes(prof, gpaths)) jobiter = create_jobs(r, psrc, iter_genomes(prof, gpaths, pname))
for sidx, gpu_time, ftag, jpg in run_jobs(r, rev, jobiter): for sidx, gpu_time, ftag, jpg in run_jobs(r, rev, jobiter):
with open(ftag, 'w') as fp: with open(ftag, 'w') as fp:
fp.write(jpg) fp.write(jpg)