def __setstate__(self, dict): """Unpickle.""" self.__dict__.update(dict) self.agefunc, self.redshiftfunc, e_f, e_t = \ cd.quick_age_function(zmax=1.1 * self.z_max, zmin=self.z_min-0.05, zstep=0.01, logspacing=True, return_inverse=True, **self.cosmo)
def __setstate__(self, dict): """Unpickle.""" self.__dict__.update(dict) self.agefunc, self.redshiftfunc, e_f, e_t = cd.quick_age_function( zmax=1.1 * self.z_max, zmin=self.z_min - 0.05, zstep=0.01, logspacing=True, return_inverse=True, **self.cosmo )
def __init__(self, z_min, z_max, dt_yr=2e6, **cosmo): self.z_min = z_min self.z_max = z_max self.dt_yr = dt_yr self.cosmo = cosmo self.agefunc, self.redshiftfunc, e_f, e_t = cd.quick_age_function( zmax=1.1 * z_max, zmin=z_min - 0.05, zstep=0.01, logspacing=True, return_inverse=True, **cosmo ) self.tmax = self.agefunc(z_min) self.tmin = self.agefunc(z_max) self.dt = self.dt_yr * cc.yr_s self.t = numpy.arange(self.tmin, self.tmax + 1.01 * self.dt, self.dt) self.t_yr = self.t / cc.yr_s self.z = self.redshiftfunc(self.t) print " Using %i points in t, dt = %.3g yr." % (len(self.t_yr), self.dt_yr)
def aout(zmin, zmax, noutput, zstep=0.001, **cosmo): ''' Function which returns list of expansion factors to output at, evenly spaced in proper time ''' import numpy as np import cosmolopy.distance as cd from seren3.array import SimArray age_func = cd.quick_age_function(1000, 0, zstep, False, **cosmo) z_func = cd.quick_redshift_age_function(1000, 0, zstep, **cosmo) age_start = age_func(zmin) age_end = age_func(zmax) age_out = np.linspace(age_start, age_end, noutput) z_out = z_func(age_out) a_out = 1. / (1. + z_out) return a_out[::-1]
def __init__(self, z_min, z_max, dt_yr=2e6, **cosmo): self.z_min = z_min self.z_max = z_max self.dt_yr = dt_yr self.cosmo = cosmo self.agefunc, self.redshiftfunc, e_f, e_t = \ cd.quick_age_function(zmax=1.1 * z_max, zmin=z_min-0.05, zstep=0.01, logspacing=True, return_inverse=True, **cosmo) self.tmax = self.agefunc(z_min) self.tmin = self.agefunc(z_max) self.dt = self.dt_yr * cc.yr_s self.t = numpy.arange(self.tmin, self.tmax + 1.01 * self.dt, self.dt) self.t_yr = self.t / cc.yr_s self.z = self.redshiftfunc(self.t) print(" Using %i points in t, dt = %.3g yr." % (len(self.t_yr), self.dt_yr))
# This program reads in master.txt which includes all gas particle ID's, scale factors, and radii for all snapshots and plots the particle trajectories for each import numpy as np import matplotlib.pyplot as plt import matplotlib import read_snap import cosmolopy.distance as cd import cosmolopy.constants as cc cosmo = {'omega_M_0': 0.3, 'omega_lambda_0': 0.7, 'omega_k_0': 0.0, 'h': 0.674} cosmo = cd.set_omega_k_0(cosmo) agefunc = cd.quick_age_function(**cosmo) # Specify galaxy gal = 858 # Specify path path = '/Volumes/Happy/Choi16_Fiducial/BH_trace/m0' + str(gal).zfill(3) + '/' # Read in files gasparticleID = np.loadtxt('/Users/sandyspicer02/Research/m0' + str(gal).zfill(3) + '/r_vs_t/master.txt', dtype='uint', usecols=(0, )) scalef = np.loadtxt('/Users/sandyspicer02/Research/m0' + str(gal).zfill(3) + '/r_vs_t/master.txt', usecols=(1, )) radius = np.loadtxt('/Users/sandyspicer02/Research/m0' + str(gal).zfill(3) + '/r_vs_t/master.txt', usecols=(2, )) cosmictime = np.loadtxt('/Users/sandyspicer02/Research/m0' +
def z_to_age(zmax=20., zmin=0., return_inverse=False, **cosmo): """ Return functions to compute age/z from each other respectivly """ import cosmolopy.distance as cd return cd.quick_age_function(zmax=zmax, zmin=zmin, return_inverse=return_inverse, **cosmo)