def lnprob2(x):
"""logL stub function for two-parameter search, also implements alpha prior."""
global resid_f,alphaab,gmat,meta,cpn
if alpha_min < x[1] < alpha_max:
return logL2(resid_f,alphaab,times_f,gmat,meta,cpn,A=x[0],alpha=x[1]) - math.log(alpha_max - alpha_min)
else:
return -N.inf
def lnprob4(x):
"""logL stub function for four-parameter search (including global red noise),
also implements alpha/alphared priors."""
global resid_f,alphaab,gmat,meta,cpn
if x[0] > 0 and (alpha_min < x[1] < alpha_max) and x[2] > 0 and (alphared_min < x[3] < alphared_max):
return (logL2(resid_f,alphaab,times_f,gmat,meta,cpn,A=x[0],alpha=x[1],Ared=x[2],alphared=x[3])
- math.log(alpha_max - alpha_min) - math.log(alphared_max - alphared_min))
else:
return -N.inf
def lnprob22N(x):
"""logL stub function for (2+2N)-parameter search (including individual red noises),
also implements alpha/alphared priors."""
global resid_f,alphaab,gmat,meta,cpn
Ared,alphared = x[2::2],x[3::2]
if x[0] > 0 and (alpha_min < x[1] < alpha_max) and N.all(Ared > 0) and N.all((alphared > alphared_min) & (alphared < alphared_max)):
return (logL2(resid_f,alphaab,times_f,gmat,meta,cpn,A=x[0],alpha=x[1],Ared=Ared,alphared=alphared)
- math.log(alpha_max - alpha_min) - len(alphared) * math.log(alphared_max - alphared_min))
else:
return -N.inf
def lnprob23N(x):
"""logL stub function for (2+3N)-parameter search (including individual red noises and efacs),
also implements alpha/alphared priors."""
global resid_f,alphaab,gmat,meta,cpn
Ared,alphared,log10_efac = x[2::3],x[3::3],x[4::3]
if (x[0] > 0 and (alpha_min < x[1] < alpha_max)
and N.all(Ared > 0) and N.all((alphared > alphared_min) & (alphared < alphared_max))
and N.all((log10_efac > log10_efac_min) & (log10_efac < log10_efac_max))):
return (logL2(resid_f,alphaab,times_f,gmat,meta,cpn,A=x[0],alpha=x[1],Ared=Ared,alphared=alphared,efac=10.0**log10_efac)
- math.log(alpha_max - alpha_min)
- len(alphared) * math.log(alphared_max - alphared_min)
- len(alphared) * math.log(log10_efac_max - log10_efac_min))
else:
return -N.inf
