constants.py

import os

USE_GPU = False
REF_DIR = os.path.expanduser("~/crds_cache/references/jwst/miri/")
RIGHT_MARGIN_BKD = 5
HIGH_ERROR = 1e10
OPT_EXTRACT_WINDOW = 5
INSTRUMENT = "MIRI_IMAGE" #Change this
SUBARRAY = "SUB128" ### Change this

if INSTRUMENT == "NIRSPEC":    
    SUBARRAY = "SUB512" #PRISM
    FILTER = "CLEAR" 
    
    NONLINEAR_FILE = REF_DIR + "jwst_nirspec_linearity_0024.fits" 
    DARK_FILE = REF_DIR + "jwst_nirspec_dark_0374.fits"  
    GAIN_FILE = REF_DIR + "jwst_nirspec_gain_0025.fits" 
    MASK_FILE = REF_DIR + "jwst_nirspec_mask_0052.fits"
    RNOISE_FILE = REF_DIR + "jwst_nirspec_readnoise_0043.fits"

    FLAT_FILE = None
    SAT_FILE = REF_DIR + "jwst_nirspec_saturation_0028.fits"
    SUPERBIAS_FILE = REF_DIR + "jwst_nirspec_superbias_0299.fits"
    SKIP_SUPERBIAS = False
    SKIP_REF = True #must be True for SUB512
    SKIP_FLAT = True
    TOP_MARGIN = 0
    BAD_GRPS = 0
    TOP = 958
    BOT = 990
    LEFT = 1024
    RIGHT = 1536
    ROTATE = 0
    N_REF = 0
    Y_CENTER = 15
    BKD_REG_TOP = [0, 5]
    BKD_REG_BOT = [-6, -1]
    X_MIN = 31
    X_MAX = 463
    SUM_EXTRACT_WINDOW = 4
    WCS_FILE = None
    SAT_LEVEL = 5.9e4    


if INSTRUMENT == "MIRI_IMAGE":
    FILTER = "F1500W"
    BAD_GRPS = 0 #11 before, 5 before that
    ROTATE = 0
    SKIP_SUPERBIAS = True
    SKIP_FLAT = True
    SKIP_REF = True
    SKIP_EMICORR = False
    
    TOP_MARGIN = 0
    N_REF = 4
    
    GAIN_FILE = REF_DIR + "jwst_miri_gain_0048.fits"
    NONLINEAR_FILE = REF_DIR + "jwst_miri_linearity_0039.fits"
    DARK_FILE = REF_DIR + "jwst_miri_dark_0105.fits"
    FLAT_FILE = REF_DIR + "jwst_miri_flat_0834.fits"
    RNOISE_FILE = REF_DIR + "jwst_miri_readnoise_0085.fits"
    MASK_FILE = REF_DIR + "jwst_miri_mask_0092.fits"
    WCS_FILE = REF_DIR + "jwst_miri_specwcs_0006.fits"
    EMICORR_FILE = REF_DIR + "jwst_miri_emicorr_0003.asdf"
    SUPERBIAS_FILE = None


    ###emicorr parameters
    EMICORR = True
    if  SUBARRAY.strip() == "FULL":
        ROW_CLOCK = 271
        FRAME_CLOCK = 277504
        XSTART = 1
        XSIZE = 1032
        LEFT = 0
        RIGHT = 1032
        TOP = 0
        BOT = 1024
        SKIP_REF = True
        DARK_FILE = REF_DIR + "jwst_miri_dark_0113.fits"
        NONLINEAR_FILE = REF_DIR + "jwst_miri_linearity_0039.fits"
    if SUBARRAY.strip() == "SUB256":
        ROW_CLOCK = 96
        FRAME_CLOCK = 29952
        XSTART = 50
        XSIZE = 256
        LEFT = 412
        RIGHT = 668
        TOP = 50
        BOT = 306
        DARK_FILE = REF_DIR + "jwst_miri_dark_0104.fits"
    if SUBARRAY.strip() == "SUB128":
        ROW_CLOCK = 44
        FRAME_CLOCK = 11904
        XSTART = 0
        XSIZE = 128
        LEFT = 0
        RIGHT = 136
        TOP = 890
        BOT = 1018
        DARK_FILE = REF_DIR + "jwst_miri_dark_0105.fits"
    if SUBARRAY.strip() == "BRIGHTSKY":
        ROW_CLOCK = 162
        FRAME_CLOCK = 86528
        XSTART = 50
        XSIZE = 512
        LEFT = 456
        RIGHT = 968
        TOP = 50
        BOT = 562
        DARK_FILE = REF_DIR + "jwst_miri_dark_0101.fits"
    if SUBARRAY.strip() == "SUB64":
        ROW_CLOCK = 28
        FRAME_CLOCK = 8512
        XSTART = 50
        XSIZE = 512
        LEFT = 0
        RIGHT = 72
        TOP = 778
        BOT = 842
        DARK_FILE = REF_DIR + "jwst_miri_dark_0106.fits"     
        
        

if INSTRUMENT == "MIRI":

    SUBARRAY = "SLITLESSPRISM"
    FILTER = "P750L"
    BAD_GRPS = 5 #11 before, 5 before that
    ROTATE = -1
    SKIP_SUPERBIAS = True
    SKIP_FLAT = False
    LEFT = 80
    RIGHT = 496
    TOP = 0
    BOT = 72
    TOP_MARGIN = 10
    N_REF = 4
    Y_CENTER = 36
    
    GAIN_FILE = REF_DIR + "jwst_miri_gain_0019.fits"
    NONLINEAR_FILE = REF_DIR + "jwst_miri_linearity_0032.fits"
    DARK_FILE = REF_DIR + "jwst_miri_dark_0096.fits"
    FLAT_FILE = REF_DIR + "jwst_miri_flat_0789.fits"
    RNOISE_FILE = REF_DIR + "jwst_miri_readnoise_0085.fits"
    MASK_FILE = REF_DIR + "jwst_miri_mask_0036.fits"
    WCS_FILE = REF_DIR + "jwst_miri_specwcs_0006.fits"
    SUPERBIAS_FILE = None
    SKIP_REF = True

    X_MIN = 30
    X_MAX = 275
    #BKD_REG_TOP = [10, 25]
    #BKD_REG_BOT = [-25, -10]
    BKD_REG_TOP = [10,21]
    BKD_REG_BOT = [-21,-10]
    SUM_EXTRACT_WINDOW = 3
    SAT_LEVEL = 5e4

elif INSTRUMENT == "NIRCAM":
    BAD_GRPS = 0
    ROTATE = 0
    SUBARRAY = "SUBGRISM64" #Change this
    FILTER = "F444W" #Change this"
    
    SKIP_SUPERBIAS = False
    SKIP_FLAT = True
    SKIP_REF = False

    N_REF = 4
    TOP_MARGIN = N_REF
    
    Y_CENTER = 33    
    BKD_REG_TOP = [N_REF, N_REF + 7]
    SUM_EXTRACT_WINDOW = 6

    FLAT_FILE = None
    GAIN_FILE = REF_DIR + "jwst_nircam_gain_0097.fits"
    MASK_FILE = REF_DIR + "jwst_nircam_mask_0063.fits"
    NONLINEAR_FILE = REF_DIR + "jwst_nircam_linearity_0052.fits"
    WCS_FILE = None

    if FILTER == "F444W":
        ONE_OVER_F_WINDOW_LEFT = 4
        ONE_OVER_F_WINDOW_RIGHT = 600
        X_MIN = 821
        X_MAX = 1900
    if FILTER == "F322W2":
        ONE_OVER_F_WINDOW_LEFT = 1894
        ONE_OVER_F_WINDOW_RIGHT = 2044
        X_MIN = 50
        X_MAX = 1645

    if SUBARRAY == "SUBGRISM256":
        SUBARR_SIZE = 256
        RNOISE_FILE = REF_DIR + "jwst_nircam_readnoise_0157.fits"
        DARK_FILE = REF_DIR + "jwst_nircam_dark_0355.fits"
        SUPERBIAS_FILE = REF_DIR + "jwst_nircam_superbias_0141.fits"
        BKD_REG_BOT = [57, 253]        
    if SUBARRAY == "SUBGRISM64":
        SUBARR_SIZE = 64
        RNOISE_FILE = REF_DIR + "jwst_nircam_readnoise_0180.fits"
        DARK_FILE = REF_DIR + "jwst_nircam_dark_0364.fits"
        SUPERBIAS_FILE = REF_DIR + "jwst_nircam_superbias_0145.fits"
        BKD_REG_BOT = [57, 64]
        
    LEFT = 0
    RIGHT = 2048 
    TOP = 0
    BOT = SUBARR_SIZE
    SAT_LEVEL = 5.9e4