From 14ec98c9bfc49f963e30159dc7db9f554088fb44 Mon Sep 17 00:00:00 2001
From: hannajankowski <hkj@yorku.ca>
Date: Thu, 9 Jun 2022 17:57:31 -0400
Subject: Testing testing 123

---
 R/WP_known.R | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 R/WP_known.R

(limited to 'R/WP_known.R')

diff --git a/R/WP_known.R b/R/WP_known.R
new file mode 100644
index 0000000..40f1ded
--- /dev/null
+++ b/R/WP_known.R
@@ -0,0 +1,27 @@
+#' WP method background function WP_known
+#'
+#' This is a background/internal function called by \code{WP}.  It computes the maximum likelihood estimator of R0 assuming that the serial distribution is known and finite discrete.
+#'
+#' @param NT vector of case counts
+#' @param p discretized version of the serial distribution
+#' @return The function returns \code{Rhat}, the maximum likelihood estimator of R0.
+#' @export
+# 
+
+WP_known		<-	function(NT, p){
+	
+	k		<-	length(p)
+	TT		<-	length(NT)-1
+	mu_t	<-	rep(0, TT)
+	for(i in 1:TT){
+		Nt		<-	NT[i:max(1,i-k+1)]
+#		print(Nt)
+#		print(p[1:min(k,i)])
+		mu_t[i]	<-	sum(p[1:min(k,i)]*Nt)	
+	}
+	Rhat	<-	sum(NT[-1])/sum(mu_t)
+	return(list(Rhat=Rhat))
+		
+}
+
+
-- 
cgit v1.2.3