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
|
#! /usr/bin/env python
from __future__ import absolute_import, division, print_function
import os
import numpy as np
source_ratios = [
(1, 0, 0),
(0, 1, 0)
]
textures = [
'OET', 'OUT'
]
datadir = '/data/user/smandalia/flavour_ratio/data/mc_texture'
prefix = ''
# prefix = '_noprior'
golemfitsourcepath = os.environ['GOLEMSOURCEPATH'] + '/GolemFit'
condor_script = golemfitsourcepath + '/scripts/flavour_ratio/submitter/mc_texture_submit.sub'
GLOBAL_PARAMS = {}
GLOBAL_PARAMS.update(dict(
threads = 1,
seed = 26
))
# Emcee
GLOBAL_PARAMS.update(dict(
run_mcmc = 'True',
burnin = 200,
nsteps = 1000,
nwalkers = 60,
mcmc_seed_type = 'uniform'
))
# FR
GLOBAL_PARAMS.update(dict(
binning = '6e4 1e7 20',
dimension = 6,
no_bsm = 'False'
))
# Plot
GLOBAL_PARAMS.update(dict(
plot_angles = 'False',
plot_elements = 'False',
))
dagfile = 'dagman_mc_texture'
dagfile += prefix + '.submit'
with open(dagfile, 'w') as f:
job_number = 1
for src in source_ratios:
print('src', src)
for tex in textures:
print('texture', tex)
f.write('JOB\tjob{0}\t{1}\n'.format(job_number, condor_script))
f.write('VARS\tjob{0}\tsr0="{1}"\n'.format(job_number, src[0]))
f.write('VARS\tjob{0}\tsr1="{1}"\n'.format(job_number, src[1]))
f.write('VARS\tjob{0}\tsr2="{1}"\n'.format(job_number, src[2]))
f.write('VARS\tjob{0}\ttexture="{1}"\n'.format(job_number, tex))
for key in GLOBAL_PARAMS.iterkeys():
f.write('VARS\tjob{0}\t{1}="{2}"\n'.format(job_number, key, GLOBAL_PARAMS[key]))
f.write('VARS\tjob{0}\tdatadir="{1}"\n'.format(job_number, datadir))
job_number += 1
print('total jobs = {0}'.format(job_number - 1))
print('dag file = {0}'.format(dagfile))
|
