Skip to content

Commit

Permalink
Merge branch 'main' of github.com:des-science/cosmosis-standard-library
Browse files Browse the repository at this point in the history
  • Loading branch information
aferte committed Apr 21, 2024
2 parents 9440823 + c1e7750 commit bd097c6
Show file tree
Hide file tree
Showing 4 changed files with 96 additions and 9 deletions.
12 changes: 6 additions & 6 deletions modules_desy6/desi_likelihood/desi_y1_bao_ani.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# zeff DM/rd error DH/rd error corr
0.51 13.62 0.25 20.98 0.61 -0.445
0.71 16.84 0.32 20.08 0.60 -0.420
0.93 21.73 0.28 17.87 0.35 -0.389
1.32 27.80 0.69 13.82 0.42 -0.444
2.33 39.71 0.94 8.52 0.17 -0.477
# zeff DM/rd error DH/rd error corr
0.510 13.62003080 0.25192504 20.98334647 0.61071168 -0.44544776
0.706 16.84645313 0.31933636 20.07872919 0.59535633 -0.42047527
0.930 21.70841761 0.28207716 17.87612922 0.34631732 -0.38910542
1.317 27.78720817 0.69034039 13.82372285 0.42222091 -0.44396689
2.330 39.70838281 0.94326716 8.52256583 0.17083924 -0.4775

6 changes: 3 additions & 3 deletions modules_desy6/desi_likelihood/desi_y1_bao_iso.txt
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
# zeff DV/rd error
0.30 7.92 0.15
1.49 26.09 0.67
# zeff DV/rd error
0.295 7.92512927 0.15074178
1.491 26.07217182 0.66868153
51 changes: 51 additions & 0 deletions modules_desy6/photoz_uz/module.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
#This is a template for module description files
name: photoz_uz
version: 1
purpose: Modify a set of loaded n(z) distributions by a set of eignmodes and amplitudes.
url: ''
interface: photoz_uz.py
attribution: Troxel & Gary Bernstein
rules: ''
cite: []
assumptions:
- files containing information on n(z) modes and priors on amplitudes

explanation: |

params:
sample:
meaning: name of sample
type: str
default: ""
basis_file:
meaning: file containing n(z) mode array
type: str
default: ""
n_modes:
meaning: number of modes to use
type: int
default: 0
inputs:
wl_number_density:
nbin:
meaning: Number of redshift bins
type: int
default:
z:
meaning: Redshift sample points of n(z) estimates
type: real 1d
default:
bin_i:
meaning: n(z)for i=1..nbin. n(z) estimates
type: real 1d
default:
{sample}_photoz_u:
bias_i:
meaning: For i=1..nmode. Amplitude for mode i.
type: real
default:
outputs:
wl_number_density:
bin_i:
meaning: n(z) for i=1..nbin. Modified n(z) estimates replaced old value
type: real 1d
36 changes: 36 additions & 0 deletions modules_desy6/photoz_uz/photoz_uz.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
from builtins import range
from cosmosis.datablock import option_section, names
import numpy as np

def setup(options):
sample = options.get_string(option_section, "sample", "")
bias_section = sample+"_photoz_u"
basis_file = options.get_string(option_section, "basis_file", "")
n_modes = options.get_int(option_section, "n_modes", 0)
U = np.loadtxt(basis_file)[:,:n_modes]
return {"sample": sample,
"bias_section": bias_section,
"basis":U,
"n_modes":n_modes}

def execute(block, config):
pz = 'nz_'+config['sample']
uvals = config['bias_section']
U = config['basis']
n_modes = config['n_modes']
nbin = block[pz, "nbin"]
if nbin>4:
nbin=4
z = block[pz, "z"]
dz = U@np.array([block[uvals, "u_%d" % j] for j in range(n_modes)])
dz = dz.reshape((nbin,len(dz)//nbin))
for i in range(1,nbin+1):
bin_name = "bin_%d" % i
nz = block[pz, bin_name]
nz[1:]+=dz[i-1,:]
nz /= np.trapz(nz, z)
block[pz, bin_name] = nz
return 0

def cleanup(config):
pass

0 comments on commit bd097c6

Please sign in to comment.