From d3ef79883035c6db3c24002b958c5beed1aee257 Mon Sep 17 00:00:00 2001 From: Justin Bedo Date: Tue, 30 Aug 2016 10:48:41 +1000 Subject: check for 0 counts --- R/pmt.R | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/R/pmt.R b/R/pmt.R index 090ceca..a1994c0 100644 --- a/R/pmt.R +++ b/R/pmt.R @@ -12,7 +12,10 @@ # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF # OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. -pmt <- function(a, b, two.sided=F) { +pmt <- function(a, b) { + if(any(a == 0) | any(b == 0)) + stop("cannot handle 0 counts") + ce <- function(mu){ pt1 <- mapply(function(xa, mu, rho) logsumexp(cumsum(log(((1 + rho) * (xa - (0:xa-1))/(2 * mu))))), a, mu, rho) pt2 <- mapply(function(xb, mu, rho) logsumexp(cumsum(log((2*mu*rho) / ((1 + rho)*(xb + (1:100)))))), b, mu, rho) -- cgit v1.2.3