R/stamp.group.summary.r

In colinr23/stampr: Spatial Temporal Analysis of Moving Polygons

# ---- roxygen documentation TEMPLATE----
#
#' @title Compile stamp summary statistics by group
#'
#' @description
#'  The function \code{stamp.group.summary} compiles summary statistics for each STAMP grouping.
#'  Specifically, it computes the area of each STAMP event type (e.g., generation, expansion, etc.)
#'  within each grouping. It also computes the number of events belonging to each event type.
#'
#' @details
#'  \code{stamp.group.summary} computes area and count summary statistics of STAMP output. Note that if
#'  both \code{area} and \code{count} are set to \code{FALSE}, \code{stamp.group.summary} returns a
#'  \code{data.frame} with just the group IDs as the only column.
#'
#' @param stmp a \code{SpatialPolygonsDataFrame} generated from the \code{stamp} function.
#' @param area logical, whether or not to compute the STAMP event areas.
#' @param count logical, whether or not to compute the count of STAMP evets within each group.
#'
#' @return
#'  A \code{data.frame} where rows are stamp groups and columns correspond to the STAMP event types (ID, areas, and counts).
#'
#' @keywords stamp
#' @export

#
# ---- End of roxygen documentation ----

stamp.group.summary <- function(stmp,area=TRUE,count=TRUE){
  grps <- unique(stmp$GROUP)
  outdf <- data.frame(GROUP=grps,nEVENTS=0,AREA=0)
  evnts <- c("CONV","CONC","CONT","DISP1","DISA","STBL","EXPN","FRAG","DIVR","DISP2","GENR")
  for (i in grps){
      ind1 <- which(stmp$GROUP == i)
      outdf$nEVENTS[i] <- length(ind1)
      outdf$AREA[i] <- sum(stmp$AREA[ind1])
      }
  if (count==TRUE){
    nevnts <- paste("n",evnts,sep="")
    outdf[,nevnts] <- 0
    for (i in grps){
      ind1 <- which(stmp$GROUP == i)
      for (j in 1:length(evnts)){
        ind <- which(stmp$GROUP == i & stmp$LEV3 == evnts[j])
        if (length(ind) > 0){ outdf[i,nevnts[j]] <- length(ind) }
        }
      }
    }
  if (area==TRUE){
    aevnts <- paste("a",evnts,sep="")
    outdf[,aevnts] <- 0
    for (i in grps){
      ind1 <- which(stmp$GROUP == i)
      for (j in 1:length(evnts)){
        ind <- which(stmp$GROUP == i & stmp$LEV3 == evnts[j])
        if (length(ind) > 0){ outdf[i,aevnts[j]] <- sum(stmp$AREA[ind]) }
        }
      }
    }
  #return a dataframe with group summaries
  return(outdf)
  }