Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools.git synced 2024-12-22 21:48:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add matching to no selection

Browse source
This commit is contained in:
rstiskalek 2023-12-13 20:42:58 +00:00
parent 506d7a7c70
commit 2be20f763b
1 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
scripts/field_sample.py
View file

@ -140,6 +140,22 @@ def evaluate_field(field, pos, nrand, smooth_scales=None, seed=42,
return val, rand_val, smooth_scales return val, rand_val, smooth_scales
def match_to_no_selection(val, rand_val, parser_args):
if parser_args.survey == "SDSSxALFALFA":
survey = csiborgtools.SDSSxALFALFA()(apply_selection=False)
else:
raise NotImplementedError(
f"Survey `{parser_args.survey}` not implemented for matching to no selection.") # noqa
if val is not None:
val = csiborgtools.read.match_array_to_no_masking(val, survey)
if rand_val is not None:
rand_val = csiborgtools.read.match_array_to_no_masking(rand_val,
survey)
return val, rand_val
def main(nsim, parser_args, pos, indxs, paths, verbose): def main(nsim, parser_args, pos, indxs, paths, verbose):
"""Load the field, interpolate it and save it to disk.""" """Load the field, interpolate it and save it to disk."""
fpath_field = paths.field(parser_args.kind, parser_args.MAS, fpath_field = paths.field(parser_args.kind, parser_args.MAS,
@ -161,6 +177,10 @@ def main(nsim, parser_args, pos, indxs, paths, verbose):
fout = paths.field_interpolated(parser_args.survey, parser_args.kind, fout = paths.field_interpolated(parser_args.survey, parser_args.kind,
parser_args.MAS, parser_args.grid, parser_args.MAS, parser_args.grid,
nsim, parser_args.in_rsp) nsim, parser_args.in_rsp)
# The survey above had some cuts, however for compatibility we want
# the same shape as the `uncut` survey
val, rand_val = match_to_no_selection(val, rand_val, parser_args)
if verbose: if verbose:
print(f"Saving to ... `{fout}`.") print(f"Saving to ... `{fout}`.")
numpy.savez(fout, val=val, rand_val=rand_val, indxs=indxs, numpy.savez(fout, val=val, rand_val=rand_val, indxs=indxs,