tlo.core module

Core framework classes.

This contains things that didn’t obviously go in their own file, such as specification for parameters and properties, and the base Module class for disease modules.

class Types(value)

Bases: Enum

Possible types for parameters and properties.

This lets us hide the details of numpy & Pandas dtype strings and give users an easy list to reference instead.

Most of these should be intuitive. The CATEGORICAL type is useful for things like sex where there are a fixed number of options to choose from. The LIST type is used for properties where the value is a collection, e.g. the set of children of a person.

DATE = 1

BOOL = 2

INT = 3

REAL = 4

CATEGORICAL = 5

LIST = 6

SERIES = 7

DATA_FRAME = 8

STRING = 9

DICT = 10

class Specifiable(type_, description, categories=None)

Bases: object

Base class for Parameter and Property.

PANDAS_TYPE_MAP = {<Types.DATE: 1>: 'datetime64[ns]', <Types.BOOL: 2>: <class 'bool'>, <Types.INT: 3>: 'int64', <Types.REAL: 4>: <class 'float'>, <Types.CATEGORICAL: 5>: 'category', <Types.LIST: 6>: <class 'object'>, <Types.SERIES: 7>: <class 'object'>, <Types.DATA_FRAME: 8>: <class 'object'>, <Types.STRING: 9>: <class 'object'>, <Types.DICT: 10>: <class 'object'>}

PYTHON_TYPE_MAP = {<Types.DATE: 1>: <class 'pandas._libs.tslibs.timestamps.Timestamp'>, <Types.BOOL: 2>: <class 'bool'>, <Types.INT: 3>: <class 'int'>, <Types.REAL: 4>: <class 'float'>, <Types.CATEGORICAL: 5>: <class 'pandas.core.arrays.categorical.Categorical'>, <Types.LIST: 6>: <class 'list'>, <Types.SERIES: 7>: <class 'pandas.core.series.Series'>, <Types.DATA_FRAME: 8>: <class 'pandas.core.frame.DataFrame'>, <Types.STRING: 9>: <class 'object'>, <Types.DICT: 10>: <class 'dict'>}

property python_type

Return the Python type corresponding to this Specifiable.

property pandas_type

Return the Pandas type corresponding to this Specifiable.

class Parameter(type_, description, categories=None)

Bases: Specifiable

Used to specify parameters for disease modules etc.

class Property(type_, description, categories=None, *, ordered=False)

Bases: Specifiable

Used to specify properties of individuals.

PANDAS_TYPE_DEFAULT_VALUE_MAP = {'datetime64[ns]': NaT, <class 'bool'>: False, 'int64': 0, <class 'float'>: nan, 'category': nan, <class 'object'>: nan}

create_series(name, size)

Create a Pandas Series for this property.

The values will be left uninitialised.

Parameters:

• name – The name for the series.

• size – The length of the series.

class Module(name=None)

Bases: object

The base class for disease modules.

This declares the methods which individual modules must implement, and contains the core functionality linking modules into a simulation. Useful properties available on instances are:

name

The unique name of this module within the simulation.

parameters

A dictionary of module parameters, derived from specifications in the PARAMETERS class attribute on a subclass.

rng

A random number generator specific to this module, with its own internal state. It’s an instance of numpy.random.RandomState

sim

The simulation this module is part of, once registered.

INIT_DEPENDENCIES = frozenset({})

OPTIONAL_INIT_DEPENDENCIES = frozenset({})

ADDITIONAL_DEPENDENCIES = frozenset({})

ALTERNATIVE_TO = frozenset({})

METADATA = {}

CAUSES_OF_DEATH = {}

CAUSES_OF_DISABILITY = {}

PARAMETERS = {}

PROPERTIES = {}

parameters

load_parameters_from_dataframe(resource: DataFrame)

Automatically load parameters from resource dataframe, updating the class parameter dictionary

Goes through parameters dict self.PARAMETERS and updates the self.parameters with values Automatically updates the values of data types: - Integers - Real numbers - Lists - Categorical - Strings - Dates (Any numbers will be converted into dated without warnings) - Booleans (An value in the csv ‘0’, 0, ‘False’, ‘false’, ‘FALSE’, or None will be interpreted as False;

everything else as True)

Will also make the parameter_name the index of the resource DataFrame.

Parameters:

resource (DataFrame) – DataFrame with a column of the parameter_name and a column of value

read_parameters(data_folder)

Read parameter values from file, if required.

Must be implemented by subclasses.

Parameters:

data_folder – path of a folder supplied to the Simulation containing data files. Typically, modules would read a particular file within here.

initialise_population(population)

Set our property values for the initial population.

Must be implemented by subclasses.

This method is called by the simulation when creating the initial population, and is responsible for assigning initial values, for every individual, of those properties ‘owned’ by this module, i.e. those declared in its PROPERTIES dictionary.

TODO: We probably need to declare somehow which properties we ‘read’ here, so the simulation knows what order to initialise modules in!

Parameters:

population – the population of individuals

initialise_simulation(sim)

Get ready for simulation start.

Must be implemented by subclasses.

This method is called just before the main simulation loop begins, and after all modules have read their parameters and the initial population has been created. It is a good place to add initial events to the event queue.

pre_initialise_population()

Carry out any work before any populations have been initialised

This optional method allows access to all other registered modules, before any of the modules have initialised a population. This is expected to be useful for when a module’s properties rely upon information from other modules.

on_birth(mother_id, child_id)

Initialise our properties for a newborn individual.

Must be implemented by subclasses.

This is called by the simulation whenever a new person is born.

Parameters:

• mother_id – the person id for the mother of this child (can be -1 if the mother is not identified).

• child_id – the person id of new child

on_simulation_end()

This is called after the simulation has ended. Modules do not need to declare this.