-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #48 from desihub/scripts_n_doc
Scripts n doc
- Loading branch information
Showing
9 changed files
with
212 additions
and
83 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,16 @@ | ||
| #!/usr/bin/env python | ||
| # | ||
| # See top-level LICENSE.rst file for Copyright information | ||
| # | ||
|
|
||
| """ | ||
| Write static html files from list of targets, based on cmx coadds | ||
| """ | ||
|
|
||
| import prospect.scripts.specview_cmx_targets as specview_cmx_targets | ||
|
|
||
| if __name__ == '__main__': | ||
| args = specview_cmx_targets.parse() | ||
| specview_cmx_targets.main(args) | ||
|
|
||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,88 @@ | ||
| """ | ||
| prospect.scripts.specview_cmx_targets | ||
| =================================== | ||
| Write static html files from a set of targets, | ||
| using tile-based coadds in CMX data | ||
| """ | ||
|
|
||
| import os | ||
| import argparse | ||
| import numpy as np | ||
|
|
||
| import desispec.io | ||
| from desiutil.log import get_logger | ||
|
|
||
| from prospect import plotframes | ||
| from prospect import utils_specviewer, myspecselect | ||
|
|
||
| def parse() : | ||
|
|
||
| parser = argparse.ArgumentParser(description='Create static html pages from a set of targets, using CMX tile-based coadds') | ||
| parser.add_argument('--specprod_dir', help='Location of directory tree (data in specprod_dir/tiles/)', type=str) | ||
| parser.add_argument('--target_list', help='ASCII file providing the list of targetids', type=str) | ||
| parser.add_argument('--tiles', help='Name of tile[s] to be processed (avoids to scan all tiles)', nargs='+', type=str, default=None) | ||
| parser.add_argument('--nights', help='Name of night[s] to be processed (avoids to scan all nights)', nargs='+', type=str, default=None) | ||
| parser.add_argument('--nspecperfile', help='Number of spectra in each html page', type=int, default=50) | ||
| parser.add_argument('--webdir', help='Base directory for webpages', type=str) | ||
| parser.add_argument('--titlepage_prefix', help='Prefix for webpage title', type=str, default='targetlist') | ||
| parser.add_argument('--with_multiple_models', help='Display several models (requires full redrock outputs)', action='store_true') | ||
| parser.add_argument('--template_dir', help='Redrock template directory', type=str, default=None) | ||
|
|
||
| args = parser.parse_args() | ||
| return args | ||
|
|
||
|
|
||
| def main(args) : | ||
|
|
||
| log = get_logger() | ||
|
|
||
| tile_dir = os.path.join(args.specprod_dir,'tiles') | ||
| if args.tiles : | ||
| tile_list = args.tiles | ||
| else : | ||
| tile_list = os.listdir(tile_dir) | ||
| obs_db = utils_specviewer.make_targetdict(tile_dir, tiles=tile_list, nights=args.nights) | ||
|
|
||
| targetids = np.loadtxt(args.target_list, dtype='int64', comments='#') | ||
| log.info(str(len(targetids))+" targets provided.") | ||
|
|
||
| if args.with_multiple_models : | ||
| spectra, zcat, rrtable = utils_specviewer.load_spectra_zcat_from_targets(targetids, tile_dir, obs_db, with_redrock=True) | ||
| else : | ||
| spectra, zcat = utils_specviewer.load_spectra_zcat_from_targets(targetids, tile_dir, obs_db, with_redrock=False) | ||
|
|
||
| # TODO? this may be put in a standalone fct, avoid code duplicate with other script | ||
| # Create several html pages : sort by targetid | ||
| nspec = spectra.num_spectra() | ||
| log.info(str(nspec)+" spectra obtained.") | ||
| sort_indices = np.argsort(spectra.fibermap["TARGETID"]) | ||
| nbpages = int(np.ceil((nspec/args.nspecperfile))) | ||
| for i_page in range(1,1+nbpages) : | ||
|
|
||
| log.info(" * Page "+str(i_page)+" / "+str(nbpages)) | ||
| the_indices = sort_indices[(i_page-1)*args.nspecperfile:i_page*args.nspecperfile] | ||
| thespec, kept_ind = myspecselect.myspecselect(spectra, indices=the_indices, remove_scores=True, output_indices=True) | ||
| the_zcat = zcat[kept_ind] | ||
| if not np.array_equal(the_zcat['TARGETID'], thespec.fibermap['TARGETID']) : | ||
| raise RuntimeError("targetids do not match between spec and zcat") | ||
| #the_zcat, kk = utils_specviewer.match_zcat_to_spectra(zcat, thespec) | ||
| if args.with_multiple_models : | ||
| the_rrtable = rrtable[kept_ind] | ||
| #the_rrtable, kk = utils_specviewer.match_zcat_to_spectra(rrtable, thespec) | ||
| num_approx_fits = 4 # TODO settle option | ||
| with_full_2ndfit = True # TODO settle option | ||
| else : | ||
| the_rrtable = None | ||
| num_approx_fits = None | ||
| with_full_2ndfit = False | ||
|
|
||
| titlepage = args.titlepage_prefix+"_"+str(i_page) | ||
| plotframes.plotspectra(thespec, with_noise=True, is_coadded=True, zcatalog=the_zcat, | ||
| title=titlepage, html_dir=args.webdir, mask_type='CMX_TARGET', with_thumb_only_page=True, | ||
| template_dir=args.template_dir, redrock_cat=the_rrtable, num_approx_fits=num_approx_fits, | ||
| with_full_2ndfit=with_full_2ndfit) | ||
|
|
||
| log.info("End of specview_cmx_targets script.") | ||
| return 0 | ||
|
|
Oops, something went wrong.