R/fitList.oSCR.R

In jaroyle/oSCR: Multi-Session Sex-Structured Spatial Capture-Recapture Models

fitList.oSCR <- function(x, rename=F, names = NULL, drop=NULL,drop.asu=FALSE){
  
  if(rename==FALSE & is.null(names(x)))
    fl.names <- 1:length(x)
  
  if(rename==FALSE & !is.null(names(x)))
    fl.names <- names(x)
  
  if(rename==TRUE){
    n1 <- paste(lapply(x,function(z) update.formula(formula(z$model[[1]]),NULL ~ .)))    
    n1 <- ifelse(n1 %in% "NULL",".",n1)
    n2 <- paste(lapply(x,function(z) update.formula(formula(z$model[[2]]),NULL ~ .)))    
    n2 <- ifelse(n2 %in% "NULL",".",n2)
    n3 <- paste(lapply(x,function(z) update.formula(formula(z$model[[3]]),NULL ~ .)))    
    n3 <- ifelse(n3 %in% "NULL",".",n3)

    count.mods <- lapply(x, function(z)length(z$model))
    if(any(count.mods>3)){
      extract.fn <- function(z){
        if(length(z$model)==4)
          tmp <- unlist(update.formula(formula(z$model[[4]]),NULL ~ .))
        if(length(z$model)<4)
          tmp <- "."
        return(tmp)
      }
     n4 <- lapply(x,extract.fn)    
    }else{
      n4 <- rep("~1",length(x))    
    }
    if(drop.asu){
      fl.names <- paste("D(",n1,") ","p(",n2,") ",
                        "sig(",n3,") ", sep="")
    }else{
      fl.names <- paste("D(",n1,") ","p(",n2,") ",
                        "sig(",n3,") ","asu(",n4,")", sep="")
    }
    
  }
  names(x) <- fl.names
  class(x) <- "oSCR.fitList"
  return(x)
}