R/get_dictionaries.R

In ekmaloney/inteRact: Implements Affect Control Theory Equations in R

#' Get Dictionary Identities
#'
#' @param dict_key of dictionary, options = china1999, china2000, egypt2014, gaysex1980,
#' germany1989, germany2007, household1994, indiana2003, internationaldomesticrelations1981,
#' internet1998, japan1984, japan19892002, morocco2015, nc1978, nireland1977, ontario1980,
#' ontario2001, politics2003, prisonersdilemma, texas1998, uga2015, uga2015bayesactsubset,
#' us2010, usfullsurveyor2015, usmturk2015, usstudent2015
#'
#' @param type behaviors, identities, mods
#'
#' @param gender average, male, female
#'
#' @return the ACT dictionary from actdata
#'
#' @importFrom actdata epa_subset
#' @export
#'
#' @examples
#' get_dictionary("morocco2015", "average")
#'
get_dictionary <- function(dict_key, g){

    d <- actdata::epa_subset(dataset = dict_key)
    d <- d %>% dplyr::filter(group == g)

  return(d)
}


#' Get Equation
#'
#' @param name name of equation. options are all equation data set keys available in the actdata package (call actdata::eqn_info() for more information)
#' @param type type of equation. options: emotionid, impressionabo, selfdir, traitid
#' @param gender gender of equation. options: f, m, av
#'
#' @return equation dataframe
#' @export
#'
#' @examples
#'
#' get_equation("nc1978", "impressionabo", "male")
get_equation <- function(name = NULL,
                         type,
                         g = NULL,
                         eq_df = NULL){

  if(is.null(name)){
    if(typeof(eq_df[[1]]) == "character"){
      eq_clean <- eq_df
    }else{
      eq_clean <- eq_df[[1]]
    }
  }else{
    # eq_df <- inteRact::equations_dataframe %>%
    #           dplyr::filter(gender == g &
    #                    key == name &
    #                    equation_type == type) %>%
    #           dplyr::pull(df)
    #
    # eq_df <- eq_df[[1]]

    eq_df <- actdata::get_eqn(key = name, equation_type = type, group = g)

    if(type == "impressionabo"){
      eq_clean <- reshape_new_equation(eq_df)
    }else if (type == "traitid" | type == "emotionid"){
      eq_clean <- reshape_emotion_equation(eq_df)
    }

  }

  return(eq_clean)

}