R/InjectParams.R

In James-Thorson/spatial_DFA: Spatial dynamic factor analysis

#' @export
InjectParams = function( previous, skeleton, inject_lambda=TRUE ){
  # Start from skeleton
  new = skeleton
  # Identical to previous MLE
  for(i in 1:4 ){
    parname = c("delta_i", "gamma_k", "log_sigma_p", "zinfl_pz")[i]
    new[[parname]] = previous[[parname]]
  }
  # Needs new vector slots
  for(i in 1:4){
    parname = c("alpha_j", "phi_j", "rho_j", "L_val")[i]
    new[[parname]][1:length(previous[[parname]])] = previous[[parname]]
  }
  if(inject_lambda==TRUE) new[["loglambda_j"]][1:length(previous[["loglambda_j"]])] = previous[["loglambda_j"]]  
  # Needs one new row (identical to the row above because of how "map" input works)
  new[["logkappa_jz"]][1:nrow(previous[["logkappa_jz"]]),][] = as.vector(previous[["logkappa_jz"]])
  new[["logkappa_jz"]][nrow(new[["logkappa_jz"]]),] = previous[["logkappa_jz"]][nrow(previous[["logkappa_jz"]]),]
  # Needs new column
  new[["Omega_input"]][,1:ncol(previous[["Omega_input"]])][1:length(as.vector(previous[["Omega_input"]]))] = as.vector(previous[["Omega_input"]])
  # Needs new column
  new[["Epsilon_input"]][,1:dim(previous[["Epsilon_input"]])[2],][1:length(as.vector(previous[["Epsilon_input"]]))] = as.vector(previous[["Epsilon_input"]])
  
  # Version specific additions
  if( "eta_mb" %in% names(previous) ){
    # Identical to previously
    new[["eta_mb"]] = previous[["eta_mb"]]
    if( length(new[["L2_val"]])==length(previous[["L2_val"]])  )new[["L2_val"]] = previous[["L2_val"]]
  }
  if( "gamma_lp" %in% names(previous) ){
    new[["gamma_lp"]] = previous[["gamma_lp"]]
  }
  if( "gamma_ptl" %in% names(previous) ){
    new[["gamma_ptl"]] = previous[["gamma_ptl"]]
  }
  
  # Return new parameter list
  return(new)
}