library(OpenImageR)
library(png)
library(imager)
library(ggplot2)
library(grid)
library(gridExtra)
library(magick)


data_sets = c("usstates_PB-P_NM_NB-NC_NM_NG33_F!GR-N_NT33_F!T-4", "usstates_PB-N_NM_NB-P_NM_NG33_F!GR-E_NT33_F!T-7", "usstates_PB-NC_NM_NB-N_NM_NG33_F!GR-S_NT33_F!T-5", "usstates_PB-P_NM_NB-N_NM_NG0_NT33_F!T-3", "usstates_PB-N_NM_NB-NC_NM_NG0_NT33_F!T-6", "usstates_PB-NC_NM_NB-P_NM_NG0_NT33_F!T-4", "usstates_PB-P_MI_NB-N_MI_NG33_F!GL-MN_NT33_F!TR-[2, 5]", "usstates_PB-N_NM_NB-NC_NM_NG33_F!GL-MI_NT33_F!TR-[5, 8]", "usstates_PB-NC_MI_NB-P_MI_NG33_F!GL-ID_NT33_F!TR-[4, 7]", "usstates_PB-P_NM_NB-NC_NM_NG33_F!GR-W_NT33_F!TR-[3, 6]", "usstates_PB-N_MD_NB-NC_MD_NG33_F!GR-N_NT33_F!TR-[2, 5]", "usstates_PB-NC_NM_NB-P_NM_NG33_F!GR-E_NT33_F!TR-[5, 8]", "usstates_PB-P_MD_NB-N_MD_NG0_NT33_F!TR-[4, 7]", "usstates_PB-N_NM_NB-NC_MI_NG0_NT33_F!TR-[3, 6]", "usstates_PB-NC_NM_NB-P_NM_NG0_NT33_F!TR-[2, 5]", "usstates_PB-P_NM_NB-NC_NM_NG33_F!GL-CA_NT0", "usstates_PB-N_MI_NB-P_MI_NG33_F!GL-TX_NT0", "usstates_PB-NC_NM_NB-N_NM_NG33_F!GL-NC_NT0", "usstates_PB-P_MI_NB-N_MI_NG33_F!GR-S_NT0", "usstates_PB-N_NM_NB-NC_NM_NG33_F!GR-W_NT0", "usstates_PB-NC_NM_NB-P_NM_NG33_F!GR-N_NT0", "usstates_PB-P_NM_NB-NC_NM_NG0_NT0", "usstates_PB-N_MD_NB-NC_MD_NG0_NT0", "usstates_PB-NC_NM_NB-P_NM_NG0_NT0", "usstates_PB-P_NM_NB-NC_NM_NG33_F!GR-N_NT33_F!T-5", "usstates_PB-N_NM_NB-P_NM_NG33_F!GR-E_NT33_F!T-3", "usstates_PB-NC_NM_NB-N_NM_NG33_F!GR-S_NT33_F!T-6", "usstates_PB-P_NM_NB-N_NM_NG0_NT33_F!T-4", "usstates_PB-N_NM_NB-NC_NM_NG0_NT33_F!T-7", "usstates_PB-NC_NM_NB-P_NM_NG0_NT33_F!T-5", "usstates_PB-P_MD_NB-N_MD_NG33_F!GL-FL_NT33_F!TR-[2, 5]", "usstates_PB-N_NM_NB-NC_NM_NG33_F!GL-ME_NT33_F!TR-[5, 8]", "usstates_PB-NC_MI_NB-P_MI_NG33_F!GL-SC_NT33_F!TR-[4, 7]", "usstates_PB-P_NM_NB-NC_NM_NG33_F!GR-W_NT33_F!TR-[3, 6]_v2", "usstates_PB-N_MI_NB-NC_MI_NG33_F!GR-N_NT33_F!TR-[2, 5]", "usstates_PB-NC_NM_NB-P_NM_NG33_F!GR-E_NT33_F!TR-[5, 8]_v2", "usstates_PB-P_MI_NB-N_MI_NG0_NT33_F!TR-[4, 7]", "usstates_PB-N_NM_NB-NC_MI_NG0_NT33_F!TR-[3, 6]_v2", "usstates_PB-NC_NM_NB-P_NM_NG0_NT33_F!TR-[2, 5]_v2", "usstates_PB-P_NM_NB-NC_NM_NG33_F!GL-ND_NT0", "usstates_PB-N_MD_NB-P_MD_NG33_F!GL-NV_NT0", "usstates_PB-NC_NM_NB-N_NM_NG33_F!GL-LA_NT0", "usstates_PB-P_MD_NB-N_MD_NG33_F!GR-S_NT0", "usstates_PB-N_NM_NB-NC_NM_NG33_F!GR-W_NT0_v2", "usstates_PB-NC_NM_NB-P_NM_NG33_F!GR-N_NT0_v2", "usstates_PB-P_NM_NB-NC_NM_NG0_NT0_v2", "usstates_PB-N_MI_NB-NC_MI_NG0_NT0", "usstates_PB-NC_NM_NB-P_NM_NG0_NT0_v2", "usstates_PB-P_NM_NB-NC_NM_NG33_F!GR-N_NT33_F!T-6", "usstates_PB-N_NM_NB-P_NM_NG33_F!GR-E_NT33_F!T-4", "usstates_PB-NC_NM_NB-N_NM_NG33_F!GR-S_NT33_F!T-3", "usstates_PB-P_NM_NB-N_NM_NG0_NT33_F!T-5", "usstates_PB-N_NM_NB-NC_NM_NG0_NT33_F!T-3", "usstates_PB-NC_NM_NB-P_NM_NG0_NT33_F!T-6", "usstates_PB-P_MI_NB-N_MI_NG33_F!GL-NM_NT33_F!TR-[2, 5]", "usstates_PB-N_NM_NB-NC_NM_NG33_F!GL-AL_NT33_F!TR-[5, 8]", "usstates_PB-NC_MI_NB-P_MI_NG33_F!GL-GA_NT33_F!TR-[4, 7]", "usstates_PB-P_NM_NB-NC_NM_NG33_F!GR-W_NT33_F!TR-[3, 6]_v3", "usstates_PB-N_MD_NB-NC_MD_NG33_F!GR-N_NT33_F!TR-[2, 5]_v2", "usstates_PB-NC_NM_NB-P_NM_NG33_F!GR-E_NT33_F!TR-[5, 8]_v3", "usstates_PB-P_MD_NB-N_MD_NG0_NT33_F!TR-[4, 7]_v2", "usstates_PB-N_NM_NB-NC_MI_NG0_NT33_F!TR-[3, 6]_v3", "usstates_PB-NC_NM_NB-P_NM_NG0_NT33_F!TR-[2, 5]_v3", "usstates_PB-P_NM_NB-NC_NM_NG33_F!GL-VA_NT0", "usstates_PB-N_MI_NB-P_MI_NG33_F!GL-WY_NT0", "usstates_PB-NC_NM_NB-N_NM_NG33_F!GL-OR_NT0", "usstates_PB-P_MI_NB-N_MI_NG33_F!GR-S_NT0_v2", "usstates_PB-N_NM_NB-NC_NM_NG33_F!GR-W_NT0_v3", "usstates_PB-NC_NM_NB-P_NM_NG33_F!GR-N_NT0_v3", "usstates_PB-P_NM_NB-NC_NM_NG0_NT0_v3", "usstates_PB-N_MD_NB-NC_MD_NG0_NT0_v2", "usstates_PB-NC_NM_NB-P_NM_NG0_NT0_v3", "usstates_PB-N_NM_NB-NC_NM_NG33_F!GR-E_NT33_F!T-5", "usstates_PB-P_NM_NB-N_NM_NG0_NT33_F!T-7", "usstates_PB-NC_NM_NB-N_NM_NG33_F!GL-NY_NT33_F!TR-[3, 6]", "usstates_PB-P_MI_NB-NC_MI_NG33_F!GR-N_NT33_F!TR-[5, 8]", "usstates_PB-N_NM_NB-NC_NM_NG0_NT33_F!TR-[2, 5]", "usstates_PB-NC_NM_NB-N_NM_NG33_F!GL-WA_NT0", "usstates_PB-N_MD_NB-NC_MD_NG33_F!GR-S_NT0", "usstates_PB-P_NM_NB-NC_NM_NG0_NT0_v4")

    
path <- "."

for(dataset_name in data_sets){
    vis <- 'symbol'
    folder_url <- paste("/figures/", vis, "/", sep = "")

    filename = paste(dataset_name, '.csv', sep = "")
    filename_len <- nchar(filename)
    filename_image <- substr(filename, 0, filename_len-4)
    n_times = 0
    filename_image <- paste(vis, '_', filename_image, '.png', sep = '')

    n_times <- 9
    plot_list <- list()
    for (time in 1:n_times)
        local({
        time <- time        
        path <- getwd()

        # load figure
        filename <- paste(path, folder_url, vis, "_", dataset_name, "_", time, ".png", sep = "")
        print(filename)
        im <- readImage(filename) 
        plot_list[[time]] <<- rasterGrob(im)
    })
    for (time in 1:1)
        local({
    
        extra_filename <- paste(path, folder_url, vis, "_extra.png", sep = "")
        extra_image <- readImage(extra_filename)
        plot_list[[10]] <<- rasterGrob(extra_image)

        legend_filename <- paste(path, folder_url, "legend_", vis, "_", dataset_name, ".png", sep = "")
        print(legend_filename)
        legend_image <- readImage(legend_filename)
        plot_list[[11]] <<- rasterGrob(legend_image)

        plot_list[[12]] <<- rasterGrob(extra_image)
    })

    filename <- paste("figures/", filename_image, sep = "")
    g <- grid.arrange(grobs = plot_list, ncol=3)
    ggsave(filename, plot=g, height=40, width=45, dpi=60) # 12 ntimes height=2400, width=2700)
    figure <- image_read(filename)
    figure <- image_crop(figure, "2700x2000")
    image_write(figure, path = filename, format = "png")
    
}