params_sim.py

import argparse
import numpy as np
import os
import sys
from scipy.io import readsav
import pandas as pd
from tqdm import tqdm
from numerize import numerize as nrz

import simulator as sm


# ============================== Define Parameters ===============================

# (u,v)-point coordinates
uv_coordinates = readsav(os.path.join(".", "stx_data", "uv.sav"))
u = uv_coordinates["u"]
v = uv_coordinates["v"]

# Phase correction factors
phase_corr = readsav(os.path.join(".", "stx_data", "phase_corr.sav"))
phase_corr = phase_corr["phase_corr"]

# Pitch and slit values
pitch_slit_data = readsav(os.path.join(".", "stx_data", "pitch_slit.sav"))

pf = pitch_slit_data["pf"]  # Pitch front grid
pr = pitch_slit_data["pr"]  # Pitch rear grid
sf = pitch_slit_data["sf"]  # Slit front grid
sr = pitch_slit_data["sr"]  # Slit rear grid

# Pixel area
pixel_area = 0.22 * 0.92
pixel_phase_factor = 45.0

# ================================ Configuration =================================

n_pix = 128  # output_size
pix_size = 2.  # pixel size in arcsec

FWHM_max_min      = 10
FWHM_max_max_1    = 85
FWHM_max_max_2    = 35
fov               = n_pix*pix_size
flux_min          = 1000.
flux_max          = 12000.
alpha_flux        = 2.
ecc_min           = 0.3
dynamic_range     = 0.3
dist_min          = 25.
dist_max          = 100.
add_noise         = True

# ================================ Fourier Matrix ================================

# Fourier remains the same for all the samples
F = sm.Fourier_matrix_STIX(u, v, n_pix, pix_size)
inv_F = np.transpose(F)