Skip to content

overlay_node

OverlayNode

Bases: Node

Node that only requires an ID. Use to overlay a Node.

Source code in emod_api/demographics/overlay_node.py 
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class OverlayNode(Node):
    """
    Node that only requires an ID. Use to overlay a Node.
    """

    def __init__(self,
                 node_id: int,
                 latitude: float = None,
                 longitude: float = None,
                 initial_population: int = None,
                 **kwargs
                 ):
        super().__init__(lat=latitude, lon=longitude, pop=initial_population, forced_id=node_id, **kwargs)

    def to_dict(self) -> dict:
        """
        Translate node structure to a dictionary for EMOD
        """
        d = {"NodeID": self.id}

        if self.node_attributes:
            na_dict = self.node_attributes.to_dict()
            if na_dict:
                d["NodeAttributes"] = na_dict

        if self.individual_attributes:
            ia_dict = self.individual_attributes.to_dict()
            if ia_dict:
                d["IndividualAttributes"] = ia_dict

        if self.individual_properties:
            ip_dict = {"IndividualProperties": []}
            for ip in self.individual_properties:
                ip_dict["IndividualProperties"].append(ip.to_dict())
            d.update(ip_dict)

        d.update(self.meta)

        return d

to_dict()

Translate node structure to a dictionary for EMOD

Source code in emod_api/demographics/overlay_node.py 
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
def to_dict(self) -> dict:
    """
    Translate node structure to a dictionary for EMOD
    """
    d = {"NodeID": self.id}

    if self.node_attributes:
        na_dict = self.node_attributes.to_dict()
        if na_dict:
            d["NodeAttributes"] = na_dict

    if self.individual_attributes:
        ia_dict = self.individual_attributes.to_dict()
        if ia_dict:
            d["IndividualAttributes"] = ia_dict

    if self.individual_properties:
        ip_dict = {"IndividualProperties": []}
        for ip in self.individual_properties:
            ip_dict["IndividualProperties"].append(ip.to_dict())
        d.update(ip_dict)

    d.update(self.meta)

    return d