#### Apollo standard script #####

library(apollo) # Load apollo package 

database <- database_full %>% filter(!is.na(Treatment_A)) %>% 
  mutate(Dummy_Treated = case_when(Treatment_A == "Treated" ~ 1, TRUE ~ 0),
         Dummy_Vol_Treated = case_when(Treatment_A == "Vol_Treated" ~ 1, TRUE ~ 0))

  #initialize model 
  
  apollo_initialise()
  
  
  ### Set core controls
  apollo_control = list(
    modelName  = "Clogit case A pref rent",
    modelDescr = "Conditonal logit model case A pref rent",
    indivID    ="id",
    mixing     = FALSE,
    HB= FALSE,
    nCores     = 1, 
    outputDirectory = "Estimation_results/clogit"
  )
  
  ##### Define model parameters depending on your attributes and model specification! ####
  # set values to 0 for conditional logit model
  
  apollo_beta=c(mu_natural = 0,
                mu_walking = 0,
                mu_rent = 0,
                ASC_sq = 0,
                mu_rent_T = 0,
                mu_nat_T = 0,
                mu_wd_T= 0,
                mu_asc_T = 0,
                mu_rent_VT = 0,
                mu_nat_VT = 0,
                mu_wd_VT= 0,
                mu_asc_VT = 0)
  
  ### specify parameters that should be kept fixed, here = none
  apollo_fixed = c()
  
  
  ### validate 
  apollo_inputs = apollo_validateInputs()
  apollo_probabilities=function(apollo_beta, apollo_inputs, functionality="estimate"){
    
    ### Function initialisation: do not change the following three commands
    ### Attach inputs and detach after function exit
    apollo_attach(apollo_beta, apollo_inputs)
    on.exit(apollo_detach(apollo_beta, apollo_inputs))
    
    ### Create list of probabilities P
    P = list()
    
    #### List of utilities (later integrated in mnl_settings below)  ####
    # Define utility functions here:
     
    V = list()
    V[['alt1']] = mu_rent*Rent_1 + mu_rent_T*Dummy_Treated*Rent_1 + mu_rent_VT*Dummy_Vol_Treated*Rent_1 + mu_natural * Naturalness_1 + mu_walking * WalkingDistance_1 + 
                  mu_nat_T * Naturalness_1 * Dummy_Treated + mu_wd_T * WalkingDistance_1 * Dummy_Treated+
                  mu_nat_VT * Naturalness_1 * Dummy_Vol_Treated + mu_wd_VT * WalkingDistance_1 * Dummy_Vol_Treated
    
    
    V[['alt2']] = mu_rent*Rent_2 + mu_rent_T*Dummy_Treated*Rent_2 + mu_rent_VT*Dummy_Vol_Treated*Rent_2 + mu_natural * Naturalness_2 + mu_walking * WalkingDistance_2 + 
                              mu_nat_T * Naturalness_2 * Dummy_Treated + mu_wd_T * WalkingDistance_2 * Dummy_Treated+
                              mu_nat_VT * Naturalness_2 * Dummy_Vol_Treated + mu_wd_VT * WalkingDistance_2 * Dummy_Vol_Treated
                  
    
    V[['alt3']] = mu_rent*Rent_3 + mu_rent_T*Dummy_Treated*Rent_3 + mu_rent_VT*Dummy_Vol_Treated*Rent_3 + mu_natural * Naturalness_3 + mu_walking * WalkingDistance_3 + ASC_sq +  mu_asc_T *  Dummy_Treated +
                              mu_asc_VT *  Dummy_Vol_Treated +
                              mu_nat_T * Naturalness_3 * Dummy_Treated + mu_wd_T * WalkingDistance_3 * Dummy_Treated+
                              mu_nat_VT * Naturalness_3 * Dummy_Vol_Treated + mu_wd_VT * WalkingDistance_3 * Dummy_Vol_Treated
    
    
    ### Define settings for MNL model component
    mnl_settings = list(
      alternatives  = c(alt1=1, alt2=2, alt3=3),
      avail         = 1, # all alternatives are available in every choice
      choiceVar     = choice,
      V             = V#,  # tell function to use list vector defined above
      
    )
    
    ### Compute probabilities using MNL model
    P[['model']] = apollo_mnl(mnl_settings, functionality)
    
    ### Take product across observation for same individual
    P = apollo_panelProd(P, apollo_inputs, functionality)
    
    ### Average across inter-individual draws - nur bei Mixed Logit!
    #P = apollo_avgInterDraws(P, apollo_inputs, functionality)
    
    ### Prepare and return outputs of function
    P = apollo_prepareProb(P, apollo_inputs, functionality)
    return(P)
  }
  
  
  
  # ################################################################# #
  #### MODEL ESTIMATION                                            ##
  # ################################################################# #
  # estimate model with bfgs algorithm
  
  clogit_caseA_pref_rent = apollo_estimate(apollo_beta, apollo_fixed,
                        apollo_probabilities, apollo_inputs, 
                        estimate_settings=list(maxIterations=12000,
                                               estimationRoutine="bfgs",
                                               hessianRoutine="analytic"))
  
  
  
  # ################################################################# #
  #### MODEL OUTPUTS                                               ##
  # ################################################################# #
  apollo_saveOutput(clogit_caseA_pref_rent)