From 97f7af2a5c689e48872b70844e2d45d662430108 Mon Sep 17 00:00:00 2001
From: Dominik Krzeminski <dkk33@cam.ac.uk>
Date: Sun, 6 Feb 2022 13:44:35 +0000
Subject: [PATCH] apply vaccine script added

---
 snippets/apply_vaccine.R | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 snippets/apply_vaccine.R

diff --git a/snippets/apply_vaccine.R b/snippets/apply_vaccine.R
new file mode 100644
index 0000000..81cf0f5
--- /dev/null
+++ b/snippets/apply_vaccine.R
@@ -0,0 +1,16 @@
+apply_vaccine_to_population <- function(population, ndoses, threshold_age) {
+  for (individual in population) {
+    if (ndoses > 0) {
+      if (indiv["age"] > threshold_age || indiv["isAtRisk"]) {
+        if (indiv["hasFirstJab"]) {
+          inject_second_jab(indiv)
+        } else {
+          inject_first_jab(indiv)
+        }
+      }
+      ndoses <- ndoses - 1
+    } else {
+      return("No more doses")
+    }
+  }
+}
\ No newline at end of file