PhotometryPy/main.py at main · drkh-n/PhotometryPy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import argparse
import pandas as pd


from src.config import load_yaml_config
from src.flux_snr5 import process_all_magnetars
from src.ensemble_photometry import ensemble_photometry
from src.logging_setup import configure_logging
import logging


# ========================================================================
# Main Execution
# =======================================================================

def main(config_file='configs/default.yml', verbose=False):

    configure_logging()
    logger = logging.getLogger(__name__)
    logger.info("Starting PhotometryPy run")
    logger.debug(f"Loading config from {config_file}")

    cfg = load_yaml_config(filename=config_file)
    logger.debug(f"Loaded configuration: {cfg.as_dict()}")
    mag_path = cfg.magnetars_list_path_file

    logger.debug(f"Reading magnetar list from {mag_path}")
    mag_list = pd.read_csv(mag_path, comment='#', sep='\s+',
                           names=['name','ra','dec'])
    logger.info(f"Found {len(mag_list['name'])} targets to process")

    for i in range(len(mag_list['name'])):
        logger.info(f"Processing target {i+1}/{len(mag_list['name'])}: {mag_list['name'][i]}")
        ra = mag_list['ra'][i]
        dec = mag_list['dec'][i]
        logger.info(f"Reading coordinates for {mag_list['name'][i]}: RA={ra}, DEC={dec}")
        mag_config = {
            'psf_file_path': cfg.prf_path,
            'name': mag_list['name'][i],
            'image_file_path': cfg.data_path_folder+'/'+mag_list['name'][i],
            'channels': cfg.channels,
            'ap_radius': cfg.rap_cam_pix,
            'inner_ann_radius': cfg.rbackin_cam_pix,
            'outer_ann_radius': cfg.rbackout_cam_pix,
            'x_coord': ra,
            'y_coord': dec,
            'spacing': cfg.spacing,
            'grid': cfg.grid,
            'intermediate_path_file' : cfg.intermediate_path_file
        }

        logger.debug(f"Target config: {mag_config}")
        ensemble_photometry(configs=mag_config)
        # from scripts.visualize_fake_sources import ensemble_photometry
        # ensemble_photometry(configs=mag_config)

    # =======================================================================
    # Calculate SNR=5 fluxes from the photometry results
    logger.info("Combining results and computing SNR=5 fluxes")
    process_all_magnetars(cfg.intermediate_path_file, cfg.result_path_file)

    logger.info("PhotometryPy run completed successfully")


if __name__ == "__main__":
    argparse = argparse.ArgumentParser(description="Simulate PSF placement and perform aperture photometry.")
    argparse.add_argument('-i', '--config', type=str, default="configs/default.yml", help="YAML config file with circapphot parameters")
    args = argparse.parse_args()

    main(args.config)