def run_tests(random_seed,
outfile,
config=None,
gsparams=None,
wmult=None,
logger=None,
fail_value=-666.):
"""Run a full set of tests, writing pickled tuple output to outfile.
"""
import sys
import cPickle
import numpy as np
import galsim
import galaxy_sample
# Load up the comparison_utilities module from the parent directory
sys.path.append('..')
import comparison_utilities
if config is None:
use_config = False
if gsparams is None:
import warnings
warnings.warn("No gsparams provided to run_tests?")
if wmult is None:
raise ValueError("wmult must be set if config=None.")
else:
use_config = True
if gsparams is not None:
import warnings
warnings.warn(
"gsparams is provided as a kwarg but the config['image']['gsparams'] will take "
+ "precedence.")
if wmult is not None:
import warnings
warnings.warn(
"wmult is provided as a kwarg but the config['image']['wmult'] will take "
+ "precedence.")
# Get galaxy sample
n_cosmos, hlr_cosmos, gabs_cosmos = galaxy_sample.get()
# Only take the first NOBS objects
n_cosmos = n_cosmos[0:NOBS]
hlr_cosmos = hlr_cosmos[0:NOBS]
gabs_cosmos = gabs_cosmos[0:NOBS]
ntest = len(SERSIC_N_TEST)
# Setup a UniformDeviate
ud = galsim.UniformDeviate(random_seed)
# Open the output file and write a header:
fout = open(outfile, 'wb')
fout.write(
'# g1obs_draw g2obs_draw sigma_draw delta_g1obs delta_g2obs delta_sigma '
+ 'err_g1obs err_g2obs err_sigma\n')
# Start looping through the sample objects and collect the results
for i, hlr, gabs in zip(range(NOBS), hlr_cosmos, gabs_cosmos):
print "Testing galaxy #"+str(i+1)+"/"+str(NOBS)+\
" with (hlr, |g|) = "+str(hlr)+", "+str(gabs)
random_theta = 2. * np.pi * ud()
g1 = gabs * np.cos(2. * random_theta)
g2 = gabs * np.sin(2. * random_theta)
for j, sersic_n in zip(range(ntest), SERSIC_N_TEST):
print "Exploring Sersic n = " + str(sersic_n)
if use_config:
# Increment the random seed so that each test gets a unique one
config['image'][
'random_seed'] = random_seed + i * NOBS * ntest + j * ntest + 1
config['gal'] = {
"type": "Sersic",
"n": sersic_n,
"half_light_radius": hlr,
"ellip": {
"type": "G1G2",
"g1": g1,
"g2": g2
}
}
config['psf'] = {
"type": "Airy",
"lam_over_diam": PSF_LAM_OVER_DIAM
}
try:
results = comparison_utilities.compare_dft_vs_photon_config(
config,
abs_tol_ellip=TOL_ELLIP,
abs_tol_size=TOL_SIZE,
logger=logger)
test_ran = True
except RuntimeError as err:
test_ran = False
pass
# Uncomment lines below to ouput a check image
#import copy
#checkimage = galsim.config.BuildImage(copy.deepcopy(config))[0] #im = first element
#checkimage.write('junk_'+str(i + 1)+'_'+str(j + 1)+'.fits')
else:
test_gsparams = galsim.GSParams(maximum_fft_size=MAX_FFT_SIZE)
galaxy = galsim.Sersic(sersic_n,
half_light_radius=hlr,
gsparams=test_gsparams)
galaxy.applyShear(g1=g1, g2=g2)
psf = galsim.Airy(lam_over_diam=PSF_LAM_OVER_DIAM,
gsparams=test_gsparams)
try:
results = comparison_utilities.compare_dft_vs_photon_object(
galaxy,
psf_object=psf,
rng=ud,
pixel_scale=PIXEL_SCALE,
size=IMAGE_SIZE,
abs_tol_ellip=TOL_ELLIP,
abs_tol_size=TOL_SIZE,
n_photons_per_trial=NPHOTONS,
wmult=wmult)
test_ran = True
except RuntimeError, err:
test_ran = False
pass
if not test_ran:
import warnings
warnings.warn('RuntimeError encountered for galaxy ' +
str(i + 1) + '/' + str(NOBS) + ' with ' +
'Sersic n = ' + str(sersic_n) + ': ' + str(err))
fout.write('%e %e %e %e %e %e %e %e %e %e %e %e %e\n' %
(fail_value, fail_value, fail_value, fail_value,
fail_value, fail_value, fail_value, fail_value,
fail_value, fail_value, fail_value, fail_value,
fail_value))
fout.flush()
else:
fout.write('%e %e %e %e %e %e %e %e %e %e %e %e %e\n' %
(results.g1obs_draw, results.g2obs_draw,
results.sigma_draw, results.delta_g1obs,
results.delta_g2obs, results.delta_sigma,
results.err_g1obs, results.err_g2obs,
results.err_sigma, sersic_n, hlr, g1, g2))
fout.flush()
def run_tests(random_seed, outfile, config=None, gsparams=None, wmult=None, logger=None,
fail_value=-666.):
"""Run a full set of tests, writing pickled tuple output to outfile.
"""
import sys
import cPickle
import numpy as np
import galsim
import galaxy_sample
# Load up the comparison_utilities module from the parent directory
sys.path.append('..')
import comparison_utilities
if config is None:
use_config = False
if gsparams is None:
import warnings
warnings.warn("No gsparams provided to run_tests?")
if wmult is None:
raise ValueError("wmult must be set if config=None.")
else:
use_config = True
if gsparams is not None:
import warnings
warnings.warn(
"gsparams is provided as a kwarg but the config['image']['gsparams'] will take "+
"precedence.")
if wmult is not None:
import warnings
warnings.warn(
"wmult is provided as a kwarg but the config['image']['wmult'] will take "+
"precedence.")
# Get galaxy sample
n_cosmos, hlr_cosmos, gabs_cosmos = galaxy_sample.get()
# Only take the first NOBS objects
n_cosmos = n_cosmos[0: NOBS]
hlr_cosmos = hlr_cosmos[0: NOBS]
gabs_cosmos = gabs_cosmos[0: NOBS]
ntest = len(SERSIC_N_TEST)
# Setup a UniformDeviate
ud = galsim.UniformDeviate(random_seed)
# Open the output file and write a header:
fout = open(outfile, 'wb')
fout.write(
'# g1obs_draw g2obs_draw sigma_draw delta_g1obs delta_g2obs delta_sigma '+
'err_g1obs err_g2obs err_sigma\n')
# Start looping through the sample objects and collect the results
for i, hlr, gabs in zip(range(NOBS), hlr_cosmos, gabs_cosmos):
print "Testing galaxy #"+str(i+1)+"/"+str(NOBS)+\
" with (hlr, |g|) = "+str(hlr)+", "+str(gabs)
random_theta = 2. * np.pi * ud()
g1 = gabs * np.cos(2. * random_theta)
g2 = gabs * np.sin(2. * random_theta)
for j, sersic_n in zip(range(ntest), SERSIC_N_TEST):
print "Exploring Sersic n = "+str(sersic_n)
if use_config:
# Increment the random seed so that each test gets a unique one
config['image']['random_seed'] = random_seed + i * NOBS * ntest + j * ntest + 1
config['gal'] = {
"type" : "Sersic" , "n" : sersic_n , "half_light_radius" : hlr ,
"ellip" : {
"type" : "G1G2" , "g1" : g1 , "g2" : g2
}
}
config['psf'] = {"type" : "Airy" , "lam_over_diam" : PSF_LAM_OVER_DIAM }
try:
results = comparison_utilities.compare_dft_vs_photon_config(
config, abs_tol_ellip=TOL_ELLIP, abs_tol_size=TOL_SIZE, logger=logger)
test_ran = True
except RuntimeError as err:
test_ran = False
pass
# Uncomment lines below to ouput a check image
#import copy
#checkimage = galsim.config.BuildImage(copy.deepcopy(config))[0] #im = first element
#checkimage.write('junk_'+str(i + 1)+'_'+str(j + 1)+'.fits')
else:
test_gsparams = galsim.GSParams(maximum_fft_size=MAX_FFT_SIZE)
galaxy = galsim.Sersic(sersic_n, half_light_radius=hlr, gsparams=test_gsparams)
galaxy.applyShear(g1=g1, g2=g2)
psf = galsim.Airy(lam_over_diam=PSF_LAM_OVER_DIAM, gsparams=test_gsparams)
try:
results = comparison_utilities.compare_dft_vs_photon_object(
galaxy, psf_object=psf, rng=ud, pixel_scale=PIXEL_SCALE, size=IMAGE_SIZE,
abs_tol_ellip=TOL_ELLIP, abs_tol_size=TOL_SIZE,
n_photons_per_trial=NPHOTONS, wmult=wmult)
test_ran = True
except RuntimeError, err:
test_ran = False
pass
if not test_ran:
import warnings
warnings.warn(
'RuntimeError encountered for galaxy '+str(i + 1)+'/'+str(NOBS)+' with '+
'Sersic n = '+str(sersic_n)+': '+str(err))
fout.write(
'%e %e %e %e %e %e %e %e %e %e %e %e %e\n' % (
fail_value, fail_value, fail_value, fail_value, fail_value, fail_value,
fail_value, fail_value, fail_value, fail_value, fail_value, fail_value,
fail_value)
)
fout.flush()
else:
fout.write(
'%e %e %e %e %e %e %e %e %e %e %e %e %e\n' % (
results.g1obs_draw, results.g2obs_draw, results.sigma_draw,
results.delta_g1obs, results.delta_g2obs, results.delta_sigma,
results.err_g1obs, results.err_g2obs, results.err_sigma, sersic_n, hlr, g1, g2
)
)
fout.flush()