# -*- coding: utf-8 -*-
# @Author: Elena Graverini
# @Date:   2017-03-16 15:22:00
# @Last Modified by:   Elena Graverini
# @Last Modified time: 2017-07-04 09:23:44
import os
# import subprocess

chunk_size = 100000
chunks = 100
leptons = ['mu', 'tau']
commands = []
seed = 12345
form_factors = "BBGIOvD2017"
signal_pdf = "Lambda_b->Lambda_c(2625)lnu::dGamma"
model = "SM"
prerun_min = 500000
prerun_update = 2000
kin_limits = {
    'mu': (0.011163612964000001, 8.947815864099999),
    'tau': (3.1570893124, 8.947815864099999),
}

os.system('mkdir -p HQT')

commands = []
for l in leptons:
    rho = {
        # -0.35: 'HQT/mcmc_samples-HQT-rho-0.35-%s.hdf5' % l,
        # -0.15: 'HQT/mcmc_samples-HQT-rho-0.15-%s.hdf5' % l,
        -0.05: 'MSBAR/mcmc_samples-MSBAR-rho-0.05-%s.hdf5' % l,
        0.00: 'MSBAR/mcmc_samples-MSBAR-rho0.00-%s.hdf5' % l,
        0.05: 'MSBAR/mcmc_samples-MSBAR-rho0.05-%s.hdf5' % l,
    }
    for r in rho.keys():
        rho3b = {
            -0.05: 'MSBAR/mcmc_samples-MSBAR-rho%s-rho3b-0.05-%s.hdf5' % (r, l),
            0.00: 'MSBAR/mcmc_samples-MSBAR-rho%s-rho3b0.00-%s.hdf5' % (r, l),
            0.05: 'MSBAR/mcmc_samples-MSBAR-rho%s-rho3b0.05-%s.hdf5' % (r, l),
        }
        for r3 in rho3b.keys():
            commands.append(['eos-sample-events-mcmc',
                             '  --fix \"Lambda_b->Lambda_c(2625)::rho@BBGIOvD2017-HQT\" %s' % r,
                             '  --fix \"Lambda_b->Lambda_c(2625)::rho3b@BBGIOvD2017-HQT\" %s' % r3,
                             '  --global-option \"form-factors\" \"%s\"' % form_factors,
                             '  --signal-pdf \"%s;l=%s,model=%s\"' % (signal_pdf, l, model),
                             '  --seed %s' % seed,
                             '  --output %s' % rho3b[r3],
                             '  --prerun-min %s' % prerun_min,
                             '  --prerun-update %s' % prerun_update,
                             '  --chunk-size %s' % chunk_size,
                             '  --chunks %s' % chunks,
                             ])

for c in commands:
    print(' '.join(c))
    # subprocess.call(c)
    os.system(' '.join(c))