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

add mapper info

parent aa88ab99

astrort/simulator/base_cleaner.py

+6 −2
Original line number Diff line number Diff line
@@ -8,6 +8,7 @@ 


import argparse

import pandas as pd

import numpy as np

from time import time

from os import makedirs

from os.path import join
@@ -51,6 +52,9 @@ def base_cleaner(configuration_file, seeds=None): 
        log.debug(f"IRF: {mapper.irf}")

        mapper.caldb = configuration['simulator']['prod']

        mapper.e = adjust_tev_range_to_irf(get_instrument_tev_range(configuration['simulator']['array']), mapper.irf)

        if configuration['mapper']['exposure'] == 'random': 

            mapper.t = [0, np.random.randint(10, configuration['simulator']['duration'])]

        else:

            mapper.t = [0, configuration['mapper']['exposure']]

        mapper.roi = get_instrument_fov(configuration['simulator']['array'])

        # make noisy map
@@ -78,7 +82,7 @@ def base_cleaner(configuration_file, seeds=None): 
        # timing simulation

        clock_map = time() - clock_map

        # save simulation data

        write_mapping_info(configuration, datfile, clock_map)

        write_mapping_info(configuration, mapper, datfile, clock_map)

        configuration['simulator']['seed'] += 1

    # end simulations

    log.info(f"\n {'-'*15} \n| STOP MAPPER | \n {'-'*15} \n")

astrort/utils/wrap.py

+2 −2
Original line number Diff line number Diff line
@@ -128,10 +128,10 @@ def merge_simulation_info(configuration, log): 
    # write merger file

    table.to_csv(merger, index=False, header=True, sep=' ', na_rep=np.nan)



def write_mapping_info(configuration, datfile, clock):

def write_mapping_info(configuration, mapper, datfile, clock):

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

    seed = configuration['simulator']['seed']

    exposure = configuration['mapper']['exposure']

    exposure = mapper.t[1] - mapper.t[0]

    center_type = configuration['mapper']['center']  

    pixelsize = configuration['mapper']['pixelsize']

    smooth = configuration['mapper']['smooth']