R/get_stay_indices.R

In twig: For Streamlining Decision and Economic Evaluation Models using Grammar of Modeling

get_stay_indices <- function(state_layer, n_expanded_states, arg_values, core_non_event_args, 
            size_core_non_event_args, expanded_states, is_cycle_dep, dim_P, dimnames_P,
            total_size_core_non_event_args){

    size_p_stay <- total_size_core_non_event_args * n_expanded_states
    origin_state_name <- state_layer$tunneled_states
    stay_dest_names <- get_stay_dest_names(state_layer)

    P_df <- expand.grid(dimnames_P)
    p_stay <- logical(size_p_stay) 

    for (i in 1:length(stay_dest_names)){
        p_stay <- p_stay | (P_df$dest == stay_dest_names[i] & 
                            P_df$state == origin_state_name[i])
    }

    dim(p_stay) <- dim_P
    dimnames(p_stay) <- dimnames_P

    if (is_cycle_dep){
        p_stay <- aperm(p_stay, c(1,4, 2, 3))
    } else {
        p_stay <- aperm(p_stay, c(1,3, 2))
    }
return(p_stay)
}