metawards.utils.get_functions

metawards.utils.get_functions(stage: str, network: Union[metawards._network.Network, metawards._networks.Networks], population: metawards._population.Population, infections: metawards._infections.Infections, output_dir: metawards._outputfiles.OutputFiles, workspace: metawards._workspace.Workspace, iterator: Callable[[...], None], extractor: Callable[[...], None], mixer: Callable[[...], None], mover: Callable[[...], None], rngs, nthreads, profiler: metawards.utils._profiler.Profiler, trajectory: metawards._population.Populations = None, results=None) → List[Callable[[...], None]][source]

Return the functions that must be called for the specified stage of the day;

  • “initialise”: model initialisation. Called once before the
    whole model run is performed
  • “setup”: day setup. Called once at the start of the day.
    Should be used to import new seeds, move populations between demographics, move infected individuals through disease stages etc. There is no calculation performed at this stage.
  • “foi”: foi calculation. Called to recalculate the force of infection
    (foi) for each ward in the network (and subnetworks). This is calculated based on the populations in each state in each ward in each demographic
  • “infect”: Called to advance the outbreak by calculating
    the number of new infections
  • “analyse”: Called at the end of the day to analyse
    the data and extract the results
  • “finalise”: Called at the end of the model run to finalise
    any outputs or produce overall summary files
  • “summary”: Called at the end of lots of model runs, to write
    overview summary files. Only the extractor has a summary stage
Parameters:
  • stage (str) – The stage of the day/model for which to get the functions
  • network (Network or Networks) – The network(s) to be modelled
  • population (Population) – The population experiencing the outbreak
  • infections (Infections) – Space to record the infections through the day
  • iterator (function) – Iterator used to obtain the function used to advance the outbreak
  • extractor (function) – Extractor used to analyse the data and output results
  • mixer (function) – Mixer used to mix and merge data between different demographics
  • mover (function) – Mover used to move populations between demographics
  • rngs (list[random number generate pointers]) – Pointers to the random number generators to use for each thread
  • nthreads (int) – The number of threads to use to run the model
  • profiler (Profiler) – The profiler used to profile the calculation
Returns:

functions – The list of all functions that should be called for this stage of the day

Return type:

List[MetaFunction]