print ('Configuring weighting...')

v.add_weighting ('weight', 'train_sig',
        color='cyan',
        )
v.add_weighting ('livetime', 'train_data',
        line=False, markers=True, marker='.', color='.5',
        errorbars=True,
        )

v.add_weighting ('weight', 'test_sig',
        color='blue',
        add_to_mc=True,
        )
v.add_weighting ('livetime', 'test_data',
        line=False, markers=True, marker='.', color='black',
        errorbars=True,
        use_as_data=True,
        )

v.setup_total_mc (
        color='green',
        )

v.add_weighting ('livetime', 'bg',
        color='purple',
        add_to_mc=True,
        )

save (v, 'output/fiveparam_final_hes.validator')
### Make Training Samples ###
train_sig_sim = make_sig_sim ()
test_sig_sim = make_sig_sim ()
bg_sim = make_bgsim ()

### Datamerize ###
train_data = make_data ()
test_data = make_data ()

### Make them Sets ###
train_sig = DataSet (train_sig_sim)
train_data = DataSet (train_data)
test_sig = DataSet (test_sig_sim)
test_data = DataSet (test_data)

bg = DataSet (bg_sim)

# create a data directory, if one does not already exist
mkdir ('hesdata')

print ('Saving data in hesdata/ ...')

# save these files
save (train_sig, 'hesdata/train_sig.ds')
save (train_data, 'hesdata/train_data.ds')
save (test_sig, 'hesdata/test_sig.ds')
save (test_data, 'hesdata/test_data.ds')
save (bg, 'hesdata/bg.ds')

bg_sim = make_bgsim ()

### Datamerize ###
train_data = make_data ()
test_data = make_data ()

### Make them Sets ###
train_sig = DataSet (train_sig_sim)
train_data = DataSet (train_data)
test_sig = DataSet (test_sig_sim)
test_data = DataSet (test_data)

bg = DataSet (bg_sim)

# create a data directory, if one does not already exist
mkdir ('data')

print ('Saving data in data/ ...')

# save these files
save (train_sig, 'data/train_sig_thirds.ds')
save (train_data, 'data/train_data_thirds.ds')
save (test_sig, 'data/test_sig_thirds.ds')
save (test_data, 'data/test_data_thirds.ds')
save (bg, 'data/bg_thirds.ds')





bg_sim = make_bgsim ()

### Datamerize ###
train_data = make_data ()
test_data = make_data ()

### Make them Sets ###
train_sig = DataSet (train_sig_sim)
train_data = DataSet (train_data)
test_sig = DataSet (test_sig_sim)
test_data = DataSet (test_data)

bg = DataSet (bg_sim)

# create a data directory, if one does not already exist
mkdir ('data')

print ('Saving data in data/ ...')

# save these files
save (train_sig, 'data/train_sig_woreco.ds')
save (train_data, 'data/train_data_woreco.ds')
save (test_sig, 'data/test_sig_woreco.ds')
save (test_data, 'data/test_data_woreco.ds')
save (bg, 'data/bg_woreco.ds')





print ('Configuring weighting...')

v.add_weighting ('weight', 'train_sig',
        color='cyan',
        )
v.add_weighting ('livetime', 'train_data',
        line=False, markers=True, marker='.', color='.5',
        errorbars=True,
        )

v.add_weighting ('weight', 'test_sig',
        color='blue',
        add_to_mc=True,
        )
v.add_weighting ('livetime', 'test_data',
        line=False, markers=True, marker='.', color='black',
        errorbars=True,
        use_as_data=True,
        )

v.setup_total_mc (
        color='green',
        )

v.add_weighting ('livetime', 'bg',
        color='purple',
        add_to_mc=True,
        )

save (v, 'output/sample.validator')