R/project.leslie.R

In milokmilo/Stranded: Calculate Biological Parameters from Stranded Cetaceans

#' Project Leslie matrix
#'
#' Project Leslie matrix. Original from demoR_0.4.2
#' @param A 
#' @param no
#' @param tmax
#' @param pop.sum Default = FALSE
#' @keywords Leslie matrix projection
#' @export
#' @examples
#' project.leslie()

project.leslie <- function(A,no,tmax,pop.sum=FALSE){
  if(length(no) != dim(A)[1])
    stop("Projection matrix and population vector have different number of states!")
  N <- matrix(0, nr=length(no),nc=tmax+1)
  N[,1] <- no
  pop <- no
  for(t in 1:tmax){
    pop <- A%*%pop
    N[,t+1] <- pop
  }
  if(pop.sum){
    N <- apply(N,2,sum)
  }
  return(N)
}