-
Notifications
You must be signed in to change notification settings - Fork 0
/
processChannels.py
executable file
·145 lines (119 loc) · 7.01 KB
/
processChannels.py
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
#!/usr/bin/env python
import sys
import os
import re
from functools import partial
import workspaceCombiner as wsc
import aux_utils as utils
import click
import git
input_dir_prepath = ("../input" if len(sys.argv) < 2 else sys.argv[1]) + '/20210309/'
output_dir_prepath = ("../output" if len(sys.argv) < 3 else sys.argv[2]) + '/v140invfb_20210309'
new_poiname = "xsec_br"
exp_or_obs = "obs"
doBetterBands = "true"
dataName = "combData"
asimovDataName = "asimovData_0"
CL = "0.95"
scaling_release = 'r02'
blind = True
git_stamp_path = os.path.join(output_dir_prepath, "git.stamp")
def create_task_arg(type, channel):
input_dir_path = os.path.join(input_dir_prepath, channel, type)
job_batch_start = [s for s in sys.argv if 'job_batch_start=' in s]
job_batch_start = int(job_batch_start[0].split('=')[-1]) if job_batch_start else None
job_batch_stop = [s for s in sys.argv if 'job_batch_stop=' in s]
job_batch_stop = int(job_batch_stop[0].split('=')[-1]) if job_batch_stop else None
return (input_dir_path, output_dir_prepath, type, channel, scaling_release, new_poiname,
exp_or_obs, doBetterBands, dataName, asimovDataName, CL, blind, job_batch_start, job_batch_stop)
channel = [s for s in sys.argv if 'channel=' in s]
if channel:
channel = channel[0].split('=')[-1]
signal = [s for s in sys.argv if 'signal=' in s]
if signal:
signal = signal[0].split('=')[-1]
task_list = []
if signal:
print('zhangr signal', signal, channel)
task_list.append( create_task_arg(signal, channel) )
nProc = 4
else:
# task_list.append( create_task_arg('nonres', 'bbbb') )
# task_list.append( create_task_arg('nonres', 'bbtautau') )
# task_list.append( create_task_arg('nonres', 'bbyy') )
# task_list.append( create_task_arg('nonres', 'WWWW') )
#task_list.append( create_task_arg('nonres', 'bbWW') )
#task_list.append( create_task_arg('nonres', 'WWWW') )
# task_list.append( create_task_arg('spin0', 'bbbb') )
# task_list.append( create_task_arg('spin0', 'bbtautau') )
# task_list.append( create_task_arg('spin0', 'bbyy') )
#task_list.append( create_task_arg('spin0', 'WWyy') )
#task_list.append( create_task_arg('spin0', 'bbWW') )
#task_list.append( create_task_arg('spin0', 'WWWW') )
#task_list.append( create_task_arg('spin0_boosted_statOnly_fromChannel', 'bbbb') )
#task_list.append( create_task_arg('spin0_boosted', 'bbtautau') )
#task_list.append( create_task_arg('spin2_c_1.0', 'bbbb') )
#task_list.append( create_task_arg('spin2_c_1.0', 'bbtautau') )
#task_list.append( create_task_arg('spin2_c_1.0', 'bbWW') )
#task_list.append( create_task_arg('spin2_c_2.0', 'bbbb') )
#task_list.append( create_task_arg('spin2_c_2.0', 'bbtautau') )
#task_list.append( create_task_arg('spin2_c_2.0', 'bbWW') )
# task_list.append( create_task_arg('lambda', 'bbbb') )
# task_list.append( create_task_arg('lambda', 'bbtautau') )
# task_list.append( create_task_arg('lambda', 'bbyy') )
#task_list.append( create_task_arg('nonres_statOnly', 'bbbb') )
#task_list.append( create_task_arg('nonres_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('nonres_statOnly', 'bbyy') )
#task_list.append( create_task_arg('nonres_statOnly', 'WWyy') )
#task_list.append( create_task_arg('nonres_statOnly', 'bbWW') )
#task_list.append( create_task_arg('nonres_statOnly', 'WWWW') )
#task_list.append( create_task_arg('spin0_statOnly', 'bbbb') )
#task_list.append( create_task_arg('spin0_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('spin0_statOnly', 'bbyy') )
#task_list.append( create_task_arg('spin0_statOnly', 'WWyy') )
#task_list.append( create_task_arg('spin0_statOnly', 'bbWW') )
#task_list.append( create_task_arg('spin0_statOnly', 'WWWW') )
#task_list.append( create_task_arg('spin2_c_1.0_statOnly', 'bbbb') )
#task_list.append( create_task_arg('spin2_c_1.0_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('spin2_c_1.0_statOnly', 'bbWW') )
#task_list.append( create_task_arg('spin2_c_2.0_statOnly', 'bbbb') )
#task_list.append( create_task_arg('spin2_c_2.0_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('spin2_c_2.0_statOnly', 'bbWW') )
#task_list.append( create_task_arg('lambda_statOnly', 'bbbb') )
#task_list.append( create_task_arg('lambda_statOnly', 'bbtautau') )
# task_list.append( create_task_arg('lambda_statOnly', 'bbyy') )
#task_list.append( create_task_arg('nonres_140invfb', 'bbtautau') )
#task_list.append( create_task_arg('nonres_300invfb', 'bbtautau') )
#task_list.append( create_task_arg('nonres_440invfb', 'bbtautau') )
#task_list.append( create_task_arg('nonres_140invfb', 'bbyy') )
#task_list.append( create_task_arg('nonres_440invfb', 'bbyy') )
#task_list.append( create_task_arg('lambda_140invfb', 'bbtautau') )
#task_list.append( create_task_arg('lambda_300invfb', 'bbtautau') )
#task_list.append( create_task_arg('lambda_440invfb', 'bbtautau') )
#task_list.append( create_task_arg('lambda_140invfb', 'bbyy') )
#task_list.append( create_task_arg('lambda_440invfb', 'bbyy') )
#task_list.append( create_task_arg('nonres_140invfb_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('nonres_300invfb_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('nonres_440invfb_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('lambda_140invfb_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('lambda_300invfb_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('lambda_440invfb_statOnly', 'bbtautau') )
#task_list.append( create_task_arg('spin0_interp275', 'bbtautau') )
#task_list.append( create_task_arg('spin0_interp300', 'bbtautau') )
#task_list.append( create_task_arg('spin2_c_1.0_interp260', 'bbtautau') )
#task_list.append( create_task_arg('spin2_c_1.0_interp300', 'bbtautau') )
#task_list.append( create_task_arg('lambda-test', 'bbbb') )
#task_list.append( create_task_arg('lambda-test', 'bbtautau') )
#task_list.append( create_task_arg('lambda-test', 'bbyy') )
#task_list.append( create_task_arg('nonres_withXsecSyst', 'bbtautau') )
#task_list.append( create_task_arg('nonres-test', 'bbtautau') )
#task_list.append( create_task_arg('spin0-interpolated', 'bbbb') )
#task_list.append( create_task_arg('nonres_testLibIssue', 'bbbb') )
#task_list.append( create_task_arg('nonres_testLibIssue', 'bbtautau') )
#task_list.append( create_task_arg('nonres_testLibIssue', 'bbyy') )
nProc = 16
# - Create a job manager
manager = utils.job_manager(func=wsc.task_pipeline_ws, nProc=nProc)
manager.set_task_args(task_list)
manager.submit()
git.save_hash_to_file(git_stamp_path)