from PTMCMCSampler.PTMCMCSampler import PTSampler as ptmcmc
from enterprise_extensions import models, model_utils, sampler
from enterprise_extensions.frequentist import optimal_statistic as OS
sys.path.insert(0,'/Users/hazboun/software_development/pta_sim/')
import pta_sim
import pta_sim.parse_sim as parse_sim
from pta_sim.sim_gw import Simulation, model_simple
from pta_sim.bayes import chain_length_bool, save_core, get_freqs, filter_psr_path
args = parse_sim.arguments()
#Is chain longer than niter?
longer = chain_length_bool(args.outdir, int(args.niter//10))
if longer and os.path.exists(args.corepath):
sys.exit()
elif longer:
save_core(args.corepath, args.outdir)
sys.exit() #Hmmmm what to do here?
else:
pass
#Get par and tim files.
parfiles = sorted(glob.glob(args.pardir+'*.par'))
timfiles = sorted(glob.glob(args.timdir+'*.tim'))
if args.psr_list is not None:
parfiles = filter_psr_path(parfiles,args.psr_list,rhs='_')
from shutil import copyfile, copy2
from PTMCMCSampler.PTMCMCSampler import PTSampler as ptmcmc
from enterprise_extensions import models, model_utils
from enterprise_extensions.frequentist import optimal_statistic as OS
sys.path.insert(0, '/Users/hazboun/software_development/pta_sim/')
import pta_sim
import pta_sim.parse_sim as parse_sim
from pta_sim.sim_gw import Simulation, model_simple
from pta_sim.bayes import chain_length_bool, save_core, get_freqs, filter_psr_path
args = parse_sim.arguments()
#Is chain longer than niter?
longer = chain_length_bool(args.outdir, args.niter)
if longer and os.path.exists(args.core_path):
sys.end()
elif longer:
save_core(args.corepath, args.outdir)
sys.end() #Hmmmm what to do here?
else:
pass
#Get par and tim files.
parfiles = sorted(glob.glob(args.pardir + '*.par'))
timfiles = sorted(glob.glob(args.timdir + '*.tim'))
if args.psr_list is not None:
parfiles = filter_psr_path(parfiles, args.psr_list, rhs='_')
pidx = args.process
psr = psrs[pidx]
else:
psr = psrs
if args.end_time is None:
Outdir = args.outdir+'/{0}/'.format(psr.name)
else:
start_time = psr.toas.min()/(24*3600)
if (args.end_time-start_time)/365.25 <= 3.0:
print('PSR {0} baseline too short for this slice.'.format(p.name))
sys.exit()
psr.filter_data(start_time=start_time, end_time=args.end_time)
Outdir = args.outdir+'{0}/{1}/'.format(args.nyears,psr.name)
longer = bys.chain_length_bool(Outdir, int(args.niter/10))
if longer and os.path.exists(args.corepath+f'{psr.name}.core'):
sys.exit()
elif longer:
c0 = Core(chaindir=Outdir)
c0.set_rn_freqs(freq_path=Outdir+'/achrom_freqs.txt')
c0.save(args.corepath+f'{psr.name}.core')
sys.exit() #Hmmmm what to do here?
else:
pass
emp_dist_path = args.emp_distr
if args.gwb_bf or args.gwb_ul:
if args.gwb_bf:
