MB07192Hack/hack_read.py at master · ulens-hack/MB07192Hack · 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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
from astropy.coordinates import SkyCoord
from astropy import units as u


def read_files_from_config(config):
    """
    Read names of photometry files. For each file read also format:
    'mag' or 'flux'.
    """
    section = 'photometry files'
    ids = [t for t in config[section]]
    files = [config.get(section, id_).split() for id_ in ids]
    return files

def read_model_settings(config):
    """
    Read basic parameters of the microlensing models: methods used for
    calculating magnification and coordinates.
    """
    settings = {}
    section = 'model'
    if config.has_option(section, 'methods'):
        methods = config.get(section, 'methods').split()
        for i in range(0, len(methods), 2):
            methods[i] = float(methods[i])
        settings['methods'] = methods
    if config.has_option(section, 'default_method'):
        settings['default_method'] = config.get(section, 'default_method')
    try:
        ra = config.getfloat(section, 'RA')
        dec = config.getfloat(section, 'Dec')
        settings['coords'] = SkyCoord(ra, dec, unit=u.deg)
    except:
        settings['coords'] = None
    return settings

def read_parameters_start(config):
    """
    Read info on starting parameter values. The distributions can be
    'gauss', 'uniform', or 'log-uniform'.
    """
    section = 'EMCEE starting'
    parameters_to_fit = [var for var in config[section]]
    starting = {}
    for param in parameters_to_fit:
        words = config.get(section, param).split()
        starting[param] = [words[0]] + [float(word) for word in words[1:]]
    return (parameters_to_fit, starting)

def read_fix_parameters(config):
    """
    Read parameters that will be kept fixed during fitting process.
    """
    section = 'fixed parameters'
    fixed = {}
    if section not in config:
        return fixed
    for var in config[section]:
        fixed[var] = config.getfloat(section, var)
    return fixed

def read_min_max(config):
    """
    Read minimum and maximum values of parameters used for prior.
    """
    min_values = {}
    section = 'EMCEE min values'
    if section in config.sections():
        for var in config[section]:
            min_values[var] = config.getfloat(section, var)

    max_values = {}
    section = 'EMCEE max values'
    if section in config.sections():
        for var in config[section]:
            max_values[var] = config.getfloat(section, var)
    return (min_values, max_values)

def read_emcee_settings(config):
    """
    Read EMCEE parameters, i.e., number of walkers, steps, and burn-in steps.
    """
    emcee_settings = {}
    section = 'EMCEE'
    for name in ['n_walkers', 'n_steps', 'n_burn']:
        emcee_settings[name] = config.getint(section, name)
    return emcee_settings

def read_other(config):
    """
    Read other options.
    """
    other = {}
    section = 'other'
    if 'print models' in config[section]:
        other['print_models'] = config.getboolean(section, 'print models')
    plot = config.get(section, 'plot time').split()
    other['plot_time'] = [float(t)-2450000. for t in plot]
    return other

def read_grid_parameters(config):
    """
    KK: New function used for reading in the min_q, max_q and delta_q (step size)
    for grid search.
    Read parameters that will be used for grid search during fitting process.
    """
    section = 'grid parameters'
    grid = {}
    if section not in config:
        return grid
    for var in config[section]:
        grid[var] = config.getfloat(section, var)
    return grid