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

simulate test

parent 4a9e2ae5

astrort/configure/test.yml

+1 −1
Original line number Diff line number Diff line
@@ -2,7 +2,7 @@ simulator: 
  name: test

  array: lst

  irf: North_z60_0.5h_LST

  prod: prod5

  prod: prod5-v0.1

  pointing: random

  duration: 300

  samples: 5

astrort/simulator/base_simulator.py

+12 −4
Original line number Diff line number Diff line
@@ -7,15 +7,23 @@ 
# *****************************************************************************



import argparse

from rtasci.rtasci.lib.RTACtoolsSimulation import RTACtoolsSimulation

from astrort.utils.wrap import load_yaml_conf

from os import makedirs

from rtasci.lib.RTACtoolsSimulation import RTACtoolsSimulation

from astrort.utils.wrap import load_yaml_conf, configure_simulator_no_visibility



def main():

    parser = argparse.ArgumentParser(description='')

    parser.add_argument('-cf', '--conf', type=str, required=True, help="Path of yaml configuration file")

    parser.add_argument('-f', '--configuration', type=str, required=True, help="Path of yaml configuration file")

    args = parser.parse_args()



    conf = load_yaml_conf(args.conf)

    configutation = load_yaml_conf(args.configuration)

    makedirs(configutation['simulator']['output'], exist_ok=True)

    

    simulator = RTACtoolsSimulation()

    for i in range(configutation['simulator']['samples']):

        configutation['simulator']['seed'] += i

        simulator = configure_simulator_no_visibility(simulator, configutation['simulator'])

        simulator.run_simulation()



if __name__ == '__main__':

    main()

astrort/utils/wrap.py

+5 −0
Original line number Diff line number Diff line
@@ -7,6 +7,8 @@ 
# *****************************************************************************



import yaml

from os import makedirs

from os.path import dirname, abspath, join, basename

from astrort.utils.utils import seeds_to_string_formatter, get_instrument_fov

from astrort.configure.check_configuration import CheckConfiguration
@@ -17,6 +19,8 @@ def load_yaml_conf(yamlfile): 
    return configuration



def configure_simulator_no_visibility(simulator, configuration):

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

        configuration['model'] = join(dirname(abspath(__file__)).replace('utils', 'templates'), basename(configuration['model']))

    simulator.model = configuration['model']

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

    simulator.caldb = configuration['prod']
@@ -24,3 +28,4 @@ def configure_simulator_no_visibility(simulator, configuration): 
    simulator.fov = get_instrument_fov(configuration['array'])

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

    simulator.seed = configuration['seed']

    return simulator