9 |
|
#suppres the EvalInstace conversion warning bug |
10 |
|
import warnings |
11 |
|
warnings.filterwarnings( action='ignore', category=RuntimeWarning, message='creating converter.*' ) |
12 |
+ |
from optparse import OptionParser |
13 |
|
from BetterConfigParser import BetterConfigParser |
14 |
|
from samplesclass import sample |
15 |
|
from mvainfos import mvainfo |
20 |
|
#CONFIGURE |
21 |
|
|
22 |
|
#load config |
23 |
+ |
#os.mkdir(path+'/sys') |
24 |
+ |
argv = sys.argv |
25 |
+ |
parser = OptionParser() |
26 |
+ |
parser.add_option("-U", "--update", dest="update", default=0, |
27 |
+ |
help="update infofile") |
28 |
+ |
parser.add_option("-D", "--discr", dest="discr", default="", |
29 |
+ |
help="discriminators to be added") |
30 |
+ |
parser.add_option("-P", "--path", dest="path", default="", |
31 |
+ |
help="path to samples") |
32 |
+ |
parser.add_option("-S", "--samples", dest="names", default="", |
33 |
+ |
help="samples you want to run on") |
34 |
+ |
parser.add_option("-C", "--config", dest="config", default=[], action="append", |
35 |
+ |
help="configuration file") |
36 |
+ |
(opts, args) = parser.parse_args(argv) |
37 |
+ |
if opts.config =="": |
38 |
+ |
opts.config = "config" |
39 |
|
config = BetterConfigParser() |
40 |
< |
config.read('./config') |
40 |
> |
config.read(opts.config) |
41 |
> |
anaTag = config.get("Analysis","tag") |
42 |
|
|
43 |
|
#get locations: |
44 |
|
Wdir=config.get('Directories','Wdir') |
27 |
– |
|
45 |
|
MVAdir=config.get('Directories','MVAdir') |
46 |
|
|
47 |
|
#systematics |
57 |
|
###################### |
58 |
|
#Evaluate multi: Must Have same treeVars!!! |
59 |
|
|
60 |
< |
Apath=sys.argv[1] |
61 |
< |
arglist=sys.argv[2] #RTight_blavla,bsbsb |
60 |
> |
Apath=opts.path |
61 |
> |
infofile = open(Apath+'/samples.info','r') |
62 |
> |
info = pickle.load(infofile) |
63 |
> |
infofile.close() |
64 |
> |
arglist=opts.discr #RTight_blavla,bsbsb |
65 |
|
|
66 |
< |
namelistIN=sys.argv[3] |
66 |
> |
namelistIN=opts.names |
67 |
|
namelist=namelistIN.split(',') |
68 |
|
|
69 |
< |
doinfo=bool(int(sys.argv[4])) |
69 |
> |
doinfo=bool(int(opts.update)) |
70 |
|
|
71 |
|
MVAlist=arglist.split(',') |
72 |
|
|
134 |
|
if eval(job.active): |
135 |
|
if job.name in namelist: |
136 |
|
#get trees: |
137 |
< |
input = TFile.Open(job.getpath(),'read') |
137 |
> |
input = TFile.Open(Apath+'/'+job.getpath(),'read') |
138 |
|
outfile = TFile.Open(job.path+'/'+MVAdir+job.prefix+job.identifier+'.root','recreate') |
139 |
|
input.cd() |
140 |
|
obj = ROOT.TObject |