Commit d26c59b9 authored by Ambra Di Piano's avatar Ambra Di Piano
Browse files

move pointing to its own function conf

parent a7904902

astrort/testing/test_utils/test_wrap.py

+3 −1
Original line number Diff line number Diff line
@@ -41,3 +41,5 @@ def test_get_point_source_info(test_conf_file): 
    assert type(pointing['offset']) == type(np.float64(1))

    assert type(pointing['source_ra']) == type(np.float64(1))

    assert type(pointing['source_dec']) == type(np.float64(1))



    
 No newline at end of file

astrort/utils/wrap.py

+13 −4
Original line number Diff line number Diff line
@@ -31,11 +31,20 @@ def configure_simulator_no_visibility(simulator, configuration): 
    simulator.fov = get_instrument_fov(configuration['array'])

    simulator.t = [0, configuration['duration']]

    simulator.seed = configuration['seed']

    simulator.ra = configuration['pointing']['ra']

    simulator.dec = configuration['pointing']['dec']

    simulator.set_log = False

    return simulator



def set_pointing(simulator, configuration, log):

    if configuration['pointing'] == 'random':

        log.info(f"Randomising pointing coordinates")

        point = randomise_pointing_sim(configuration)

    else:

        log.info(f"Using fixed pointing coordinates")

        point = get_point_source_info(configuration)

    simulator.ra = point['point_ra']

    simulator.dec = point['point_dec']

    return simulator, point



def randomise_pointing_sim(simulator):

    if '$TEMPLATES$' in simulator['model']:

        simulator['model'] = join(dirname(abspath(__file__)).replace('utils', 'templates'), basename(simulator['model']))
@@ -63,10 +72,10 @@ def get_point_source_info(simulator): 
    return {'point_ra': pointing.ra.deg, 'point_dec': pointing.dec.deg, 'offset': separation.value, 'source_ra': source.ra.deg, 'source_dec': source.dec.deg}



def write_simulation_info(simulator, configuration, pointing, datfile):

    name = configuration['simulator']['name']

    name = seeds_to_string_formatter(configuration['samples'], configuration['output'], configuration['name'], configuration['seed'], '')

    seed = simulator.seed

    tstart, tstop = simulator.t

    duration = configuration['simulator']['duration']

    duration = configuration['duration']

    point_ra, point_dec, offset, source_ra, source_dec = pointing['point_ra'], pointing['point_dec'], pointing['offset'], pointing['source_ra'], pointing['source_dec']

    if not isfile(datfile):

        with open(datfile, 'w+') as f: