-
Notifications
You must be signed in to change notification settings - Fork 15
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge remote-tracking branch 'origin' into nsb_tuning_test
- Loading branch information
Showing
45 changed files
with
30,329 additions
and
121 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
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 |
|---|---|---|
| @@ -1,17 +1,15 @@ | ||
| name: lstmcpipe | ||
| channels: | ||
| - default | ||
| - cta-observatory | ||
| - conda-forge | ||
| dependencies: | ||
| - python>=3.7 | ||
| - python>=3.9 | ||
| - setuptools_scm<8.0 | ||
| - astropy>=4.2 | ||
| - pip | ||
| - pre-commit | ||
| - ctapipe~=0.12 | ||
| - ctapipe | ||
| - ruamel.yaml>=0.17 | ||
| - lstchain | ||
| - pyirf | ||
| - ctaplot | ||
| - scipy~=1.11.4 |
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,147 @@ | ||
| import subprocess | ||
| import json | ||
| import logging | ||
| from lstchain.io.config import get_mc_config | ||
| import argparse | ||
| from datetime import date | ||
|
|
||
| BASE_LSTCHAIN_MC_CONFIG = get_mc_config() | ||
|
|
||
| # Configure logging | ||
| logging.basicConfig(level=logging.INFO) # Set the default log level to INFO | ||
| logger = logging.getLogger(__name__) | ||
|
|
||
|
|
||
| def build_argparser(): | ||
| """ | ||
| Build the argument parser for the script. | ||
| Returns: | ||
| argparse.ArgumentParser: The argument parser object. | ||
| """ | ||
| parser = argparse.ArgumentParser(description="Generate a set of lstchain and lstmcpipe configuration files for different nsb tuning ratios.") | ||
| parser.add_argument("--nsb_ratios", "-nsb", | ||
| nargs="+", type=float, default=None, | ||
| help="List of nsb tuning ratios. If not provided, no NSB tuning is applied.") | ||
|
|
||
| parser.add_argument( | ||
| "--dec_list", | ||
| nargs="+", | ||
| help="List of declination values", | ||
| default=[ | ||
| "dec_2276", | ||
| "dec_3476", | ||
| "dec_4822", | ||
| "dec_6166", | ||
| # "dec_6166_high_density", | ||
| "dec_6676", | ||
| "dec_931", | ||
| "dec_min_1802", | ||
| "dec_min_2924", | ||
| "dec_min_413" | ||
| ], | ||
| ) | ||
|
|
||
| return parser | ||
|
|
||
|
|
||
| def lstchain_config_name(nsb_tuning_ratio): | ||
| """ | ||
| Generate the name of the lstchain configuration file based on the given nsb_tuning_ratio. | ||
| Parameters: | ||
| nsb_tuning_ratio (float): The nsb tuning ratio. | ||
| Returns: | ||
| str: The name of the lstchain configuration file. | ||
| """ | ||
| if nsb_tuning_ratio == 0 or nsb_tuning_ratio is None: | ||
| return "lstchain_config.json" | ||
| else: | ||
| return f"lstchain_config_nsb{nsb_tuning_ratio}.json" | ||
|
|
||
|
|
||
| def dump_lstchain_nsb_config(nsb_tuning_ratio): | ||
| """ | ||
| Dump the lstchain configuration file with the given nsb_tuning_ratio. | ||
| Parameters: | ||
| nsb_tuning_ratio (float): The nsb tuning ratio. | ||
| """ | ||
| new_config = BASE_LSTCHAIN_MC_CONFIG.copy() | ||
| if nsb_tuning_ratio == 0 or nsb_tuning_ratio is None: | ||
| new_config["waveform_nsb_tuning"]["nsb_tuning"] = False | ||
| else: | ||
| new_config["waveform_nsb_tuning"]["nsb_tuning"] = True | ||
| new_config["waveform_nsb_tuning"]["nsb_tuning_ratio"] = nsb_tuning_ratio | ||
| json_filename = lstchain_config_name(nsb_tuning_ratio) | ||
| with open(json_filename, 'w') as f: | ||
| json.dump(new_config, f, indent=4) | ||
| logger.info(f"Dumped lstchain configuration file: {json_filename}") | ||
|
|
||
|
|
||
| def prod_id(nsb_tuning_ratio): | ||
| """ | ||
| Generate the prod ID based on the given nsb_tuning_ratio. | ||
| Parameters: | ||
| nsb_tuning_ratio (float): The nsb tuning ratio. | ||
| Returns: | ||
| str: The product ID. | ||
| """ | ||
| return f"{date.today()}_allsky_nsb_tuning_{nsb_tuning_ratio}" | ||
|
|
||
|
|
||
| def lstmcpipe_config_filename(nsb_tuning_ratio): | ||
| """ | ||
| Generate the name of the lstmcpipe configuration file based on the given nsb_tuning_ratio. | ||
| Parameters: | ||
| nsb_tuning_ratio (float): The nsb tuning ratio. | ||
| Returns: | ||
| str: The name of the lstmcpipe configuration file. | ||
| """ | ||
| if nsb_tuning_ratio == 0 or nsb_tuning_ratio is None: | ||
| return "lstmcpipe_config.json" | ||
| else: | ||
| return f"lstmcpipe_config_nsb{nsb_tuning_ratio}.json" | ||
|
|
||
|
|
||
| def main(): | ||
| """ | ||
| Dump the lstchain and lstmcpipe configuration files for the given nsb_tuning_ratios. | ||
| """ | ||
| parser = build_argparser() | ||
| args = parser.parse_args() | ||
|
|
||
| dec_list = " ".join(args.dec_list) | ||
| nsb_tuning_ratios = args.nsb_ratios | ||
|
|
||
| if nsb_tuning_ratios is None: | ||
| nsb_tuning_ratios = [None] | ||
| for nsb_tuning_ratio in nsb_tuning_ratios: | ||
| logger.info(f"Working on ratio {nsb_tuning_ratio}") | ||
| dump_lstchain_nsb_config(nsb_tuning_ratio) | ||
| tmp_lstchain_config = "tmp_lstchain_config.json" | ||
| command = [ | ||
| "lstmcpipe_generate_config", | ||
| "PathConfigAllSkyFull", | ||
| "--prod_id", | ||
| prod_id(nsb_tuning_ratio), | ||
| "-o", | ||
| lstmcpipe_config_filename(nsb_tuning_ratio), | ||
| "--dec_list", | ||
| dec_list, | ||
| "--lstchain_conf", | ||
| tmp_lstchain_config | ||
| ] | ||
| subprocess.run(command, check=True) | ||
| # Delete tmp_lstchain_config (the lstchain configs with nsb tuning are already dumped) | ||
| subprocess.run(["rm", tmp_lstchain_config], check=True) | ||
| logger.info(f"Generated lstmcpipe configuration file: {lstmcpipe_config_filename(nsb_tuning_ratio)}") | ||
|
|
||
|
|
||
| if __name__ == "__main__": | ||
| main() |
Oops, something went wrong.