data = np.concatenate((data,mp), axis=1)
else:
x_list['mp_%s' %str(m)] = np.arange(len(mp[0]))
data = mp
x_list['pp_%s' %str(m)] = len(data[0]) + np.arange(len(pp[0]))
data = np.concatenate((data,pp), axis=1)
for s in ms:
mp, pp = decay.read_mres_bs(psql,decay_params,s)
x_list['mp_%s' %str(s)] = len(data[0]) + np.arange(len(mp[0]))
data = np.concatenate((data,mp), axis=1)
x_list['pp_%s' %str(s)] = len(data[0]) + np.arange(len(pp[0]))
data = np.concatenate((data,pp), axis=1)
# mesons
for m in ml:
decay_params['decay_ward_fit']['ml'] = m
ss, ps = decay.read_decay_bs(psql,decay_params,'pion')
x_list['ss_%s_%s' %(str(m),str(m))] = len(data[0]) + np.arange(len(ss[0]))
data = np.concatenate((data,ss), axis=1)
x_list['ps_%s_%s' %(str(m),str(m))] = len(data[0]) + np.arange(len(ps[0]))
data = np.concatenate((data,ps), axis=1)
for s in ms:
decay_params['decay_ward_fit']['ms'] = s
ss, ps = decay.read_decay_bs(psql,decay_params,'kaon')
x_list['ss_%s_%s' %(str(m),str(s))] = len(data[0]) + np.arange(len(ss[0]))
data = np.concatenate((data,ss), axis=1)
x_list['ps_%s_%s' %(str(m),str(s))] = len(data[0]) + np.arange(len(ps[0]))
data = np.concatenate((data,ps), axis=1)
# make gvars
data_gv = c51.make_gvars(data)
### fit ###
result = dict()
dparams = yaml.load(f)
dparams['decay_ward_fit']['ens'] = params['axial_fit']['ens']
dparams['decay_ward_fit']['ml'] = params['axial_fit']['ml']
dparams['decay_ward_fit']['ms'] = params['axial_fit']['ms']
f.close()
# yaml entires
fitmeta = params['axial_fit']
# log in sql
psqlpwd = pwd.passwd()
psql = sql.pysql('cchang5', 'cchang5', psqlpwd)
# read data
ll = read_axial(psql, params, 'axial_ll')
ls = read_axial(psql, params, 'axial_ls')
# read two-point
pionSS, pionPS = decay.read_decay_bs(psql, dparams, 'pion')
kaonSS, kaonPS = decay.read_decay_bs(psql, dparams, 'kaon')
# axial_ll
print np.shape(pionSS), np.shape(pionPS), np.shape(ll)
if params['flags']['fit_axial_ll']:
ll = np.concatenate((pionSS, pionPS, ll), axis=1)
llgv = c51.make_gvars(ll)
fpi = fit_axial(psql, params, dparams, 'axial_ll', llgv, flow=True)
print fphi(fpi['axial_fit'].p['F0'], fpi['axial_fit'].p['E0'])
# axial_ls
if params['flags']['fit_axial_ls']:
ls = np.concatenate((kaonSS, kaonPS, ls), axis=1)
lsgv = c51.make_gvars(ls)
fka = fit_axial(psql, params, dparams, 'axial_ls', lsgv, flow=True)
print fphi(fka['axial_fit'].p['F0'], fka['axial_fit'].p['E0'])
decay_params['decay_ward_fit']['ens'] = params['grand_ensemble']['ens']
decay_params['decay_ward_fit']['ml'] = params['grand_ensemble']['ml']
# gA parameter file
f = open('./fhprotonmaster.yml.%s' %(user_flag),'r')
gA_params = yaml.load(f)
f.close()
gA_params['gA_fit']['ens'] = params['grand_ensemble']['ens']
gA_params['gA_fit']['ml'] = params['grand_ensemble']['ml']
# log in sql #
psqlpwd = pwd.passwd()
psql = sql.pysql('cchang5','cchang5',psqlpwd)
# read data #
# mres for ml
mp, pp = decay.read_mres_bs(psql,decay_params,decay_params['decay_ward_fit']['ml'])
# pion correlator
ss_pion, ps_pion = decay.read_decay_bs(psql,decay_params,'pion')
# gA parameter file
gAboot0 = gA.read_gA_bs(psql,gA_params)
# concatenate and build grand covariance matrix #
# index
x_mp = np.arange(len(mp[0]))
x_pp = np.arange(len(pp[0]))+len(x_mp)
x_ss_pion = np.arange(len(ss_pion[0]))+len(x_mp)+len(x_pp)
x_ps_pion = np.arange(len(ps_pion[0]))+len(x_mp)+len(x_pp)+len(x_ss_pion)
x_gAboot0 = np.arange(len(gAboot0[0]))+len(x_mp)+len(x_pp)+len(x_ss_pion)+len(x_ps_pion)
print "len mp:", np.shape(mp)
print "len pp:", np.shape(pp)
print "len ss_pion:", np.shape(ss_pion)
print "len ps_pion:", np.shape(ps_pion)
print "len gAboot0:", np.shape(gAboot0)
# concatenate
f = open('./decay_flow.yml', 'r')
decay_params = yaml.load(f)
f.close()
ensemble = decay_params['decay_ward_fit']['ens']['tag']
# log in sql #
psqlpwd = pwd.passwd()
psql = sql.pysql('cchang5', 'cchang5', psqlpwd)
# read data #
# mres for ml
mpl, ppl = decay.read_mres_bs(psql, decay_params,
decay_params['decay_ward_fit']['ml'])
# ms
mps, pps = decay.read_mres_bs(psql, decay_params,
decay_params['decay_ward_fit']['ms'])
# pion correlator
ss_pion, ps_pion = decay.read_decay_bs(psql, decay_params, 'pion')
ss_kaon, ps_kaon = decay.read_decay_bs(psql, decay_params, 'kaon')
# index
x_mpl = np.arange(len(mpl[0]))
x_ppl = len(x_mpl) + np.arange(len(ppl[0]))
x_mps = len(x_ppl) + len(x_mpl) + np.arange(len(mps[0]))
x_pps = len(x_mps) + len(x_ppl) + len(x_mpl) + np.arange(len(pps[0]))
x_ss_pion = len(x_pps) + len(x_mps) + len(x_ppl) + len(x_mpl) + np.arange(
len(ss_pion[0]))
x_ps_pion = len(x_ss_pion) + len(x_pps) + len(x_mps) + len(x_ppl) + len(
x_mpl) + np.arange(len(ps_pion[0]))
x_ss_kaon = len(x_ps_pion) + len(x_ss_pion) + len(x_pps) + len(
x_mps) + len(x_ppl) + len(x_mpl) + np.arange(len(ss_kaon[0]))
x_ps_kaon = len(x_ss_kaon) + len(x_ps_pion) + len(x_ss_pion) + len(
x_pps) + len(x_mps) + len(x_ppl) + len(x_mpl) + np.arange(
len(ps_kaon[0]))
data = np.concatenate((data,mp), axis=1)
else:
x_list['mp_%s' %str(m)] = np.arange(len(mp[0]))
data = mp
x_list['pp_%s' %str(m)] = len(data[0]) + np.arange(len(pp[0]))
data = np.concatenate((data,pp), axis=1)
for s in ms:
mp, pp = decay.read_mres_bs(psql,decay_params,s)
x_list['mp_%s' %str(s)] = len(data[0]) + np.arange(len(mp[0]))
data = np.concatenate((data,mp), axis=1)
x_list['pp_%s' %str(s)] = len(data[0]) + np.arange(len(pp[0]))
data = np.concatenate((data,pp), axis=1)
# mesons
for m in ml:
decay_params['decay_ward_fit']['ml'] = m
ss, ps = decay.read_decay_bs(psql,decay_params,'pion')
x_list['ss_%s_%s' %(str(m),str(m))] = len(data[0]) + np.arange(len(ss[0]))
data = np.concatenate((data,ss), axis=1)
x_list['ps_%s_%s' %(str(m),str(m))] = len(data[0]) + np.arange(len(ps[0]))
data = np.concatenate((data,ps), axis=1)
for s in ms:
decay_params['decay_ward_fit']['ms'] = s
ss, ps = decay.read_decay_bs(psql,decay_params,'kaon')
x_list['ss_%s_%s' %(str(m),str(s))] = len(data[0]) + np.arange(len(ss[0]))
data = np.concatenate((data,ss), axis=1)
x_list['ps_%s_%s' %(str(m),str(s))] = len(data[0]) + np.arange(len(ps[0]))
data = np.concatenate((data,ps), axis=1)
# make gvars
data_gv = c51.make_gvars(data)
### fit ###
result = dict()
