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