source("linear_scaling.R")
library(ggplot2)
library(RColorBrewer)

fittrans.id <- function(popul) {
    return(popul)
}

fittrans.linscale <- function(a, b) {
    return(function(popul) {
        popul[,2] = a * popul[,2] + b

        return(popul)
    })
}

select.fps <- function(popul) {
    relfit <- linscale.relfitness(popul[,2])
    
    filtr <- sample(popul[,1], length(popul[,1]), prob=relfit, replace=TRUE)
    return(popul[filtr,])
}

ev.run <- function(popul, gen, selection, fittrans=fittrans.id) {
    npopul <- length(popul)

    namedpopul <- matrix(c(1:npopul, popul), ncol=2)

    df = data.frame( gen = 0
                    ,fitness = namedpopul[,2]
                    ,ancestor = namedpopul[,1]
                    ,relfit = linscale.relfitness(namedpopul[,2]))

    for(g in 1:gen) {
        namedpopul <- selection(fittrans(namedpopul))

        df = rbind(df, data.frame( gen = g
                                  ,fitness = namedpopul[,2]
                                  ,ancestor = namedpopul[,1]
                                  ,relfit = linscale.relfitness(namedpopul[,2])))
    }

    return(df)
}

ev.plot <- function(df, filename) {
    pdf(file=filename, onefile=TRUE)
    

    for(g in unique(df$gen)) {
        p <- ggplot(data=df[df$gen == g,], aes(x="", y=relfit, fill=factor(ancestor))) +
            geom_bar(stat="identity", width=1) +
            coord_polar("y", start=0) +
            labs(x=NULL, y=NULL, fill="ancestor", title=sprintf("generation: %i", g))

        print(p)
    }

    dev.off()
}

ev.animate <- function(df, filename) {

    anim <- ggplot(data=df, aes(x="", y=relfit, fill=factor(ancestor))) +
        geom_bar(stat="identity", width=1) +
        coord_polar("y", start=0) +
        labs(x=NULL, y=NULL, fill="ancestor", title="generation: {closest_state}") +
        transition_states(gen)

    anim_save(filename, animation=anim)
}