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: