Age structured IPM devlogs

.vscode/settings.json

@@ -11,12 +11,15 @@
   },
   "files.insertFinalNewline": true,
   "files.trimFinalNewlines": true,
-
+
+  "editor.formatOnSave": true,
   "editor.codeActionsOnSave": {
     "source.organizeImports": "explicit"
   },
-  "editor.defaultFormatter": null,
-  "editor.formatOnSave": true,
+  "notebook.formatOnSave.enabled": true,
+  "notebook.codeActionsOnSave": {
+    "source.organizeImports": "explicit"
+  },
 
   "autopep8.importStrategy": "fromEnvironment",
   "isort.importStrategy": "fromEnvironment",
@@ -36,6 +39,7 @@
     "*_test.py"
   ],
 
+  "editor.defaultFormatter": null,
   "[python]": {
     "editor.detectIndentation": false,
     "editor.insertSpaces": true,

doc/devlog/README.md

@@ -35,11 +35,14 @@ This folder is a handy place to put Jupyter notebooks or other documents which h
 | 2023-10-26.ipynb | Tyler | | Describes a major Geo system refactor and introduces new systems. |
 | 2023-11-03-seirs-example.ipynb | Ajay | | Demonstrates the building and running of an SEIRS model. |
 | 2023-11-08.ipynb | Ajay | | Demonstration of using proxy geo to access data in parameter functions. |
+| 2023-11-08-age-ipm.ipynb | Jarom | | Initial prototyping of age-class IPMs. |
 | 2023-11-15.ipynb | Ajay | | Detailed description of parameter functions functionality. |
 | 2023-11-20-adrio-phase-2-demo.ipynb | Trevor | | Demonstrates the refactor work on DynamicGeos and the ADRIO system, and geo cache handling. |
 | 2023-11-22-ipm-probs.ipynb | Tyler | | Analyzing statistical correctness of our IPM processing algorithms. |
 | 2023-12-05.ipynb | Tyler | | A brief tour of changes to epymorph due to the refactor effort. |
 | 2024-01-08.ipynb | Tyler | | Another functional parameters demonstration, revisiting the Bonus Example from 2023-10-10. |
+| 2024-02-06.ipynb | Tyler | | Revisiting age-class IPMs, and thinking about modularity of approach. |
+| 2024-02-12.ipynb | Tyler | | Continued age-class IPM work, this time in more than one geo node. |
 
 ## Contributing
 

epymorph/data/mm/flat.movement

@@ -0,0 +1,21 @@
+# This model evenly weights the probability of movement to all other nodes.
+# It uses parameter 'commuter_proportion' to determine how many people should
+# be moving, based on the total normal population of each node.
+
+[move-steps: per-day=2; duration=[1/3, 2/3]]
+
+[predef: function=
+def flat_predef():
+    ones = np.ones((dim.nodes, dim.nodes))
+    np.fill_diagonal(ones, 0)
+    dispersal_kernel = row_normalize(ones)
+    return { 'dispersal_kernel': dispersal_kernel }
+]
+
+# Assume a percentage of the population move around,
+# evenly weighted to all other locations.
+[mtype: days=all; leave=1; duration=0d; return=2; function=
+def flat_movement(t):
+    n_commuters = np.floor(geo['population'] * params['commuter_proportion']).astype(SimDType)
+    return np.multinomial(n_commuters, predef['dispersal_kernel'])
+]

epymorph/movement/compile.py

@@ -134,6 +134,18 @@ def visit_Subscript(self, node: ast.Subscript) -> Any:
                 return node
         return self.generic_visit(node)
 
+    def visit_Attribute(self, node: ast.Attribute) -> Any:
+        """Modify references to objects that should be in context."""
+        if isinstance(node.value, ast.Name):
+            if node.value.id in ['dim']:
+                node.value = ast.Attribute(
+                    value=ast.Name(id='ctx', ctx=ast.Load()),
+                    attr=node.value.id,
+                    ctx=ast.Load(),
+                )
+                return node
+        return self.generic_visit(node)
+
     def visit_FunctionDef(self, node: ast.FunctionDef) -> Any:
         """Modify function parameters."""
         new_node = self.generic_visit(node)

epymorph/simulation.py

@@ -232,6 +232,7 @@ def epymorph_namespace() -> dict[str, Any]:
             'concatenate': partial(np.concatenate, dtype=SimDType),
             'sum': partial(np.sum, dtype=SimDType),
             'newaxis': np.newaxis,
+            'fill_diagonal': np.fill_diagonal,
             # numpy math functions
             'radians': np.radians,
             'degrees': np.degrees,

epymorph/util.py

@@ -33,6 +33,13 @@ def call_all(*fs: Callable[[], Any]) -> None:
         f()
 
 
+def or_raise(value: T | None, message: str) -> T:
+    """Enforce that the given value is not None, or else raise an exception."""
+    if value is None:
+        raise Exception(message)
+    return value
+
+
 # collection utilities