// @file:Repository("*mavenLocal")
@file:DependsOn("com.github.holgerbrandl:krangl:0.17")

%use lets-plot

sleepData

sleepData.schema()

var sleepDataExt = sleepData.addColumn("rem_proportion"){it["sleep_rem"]/it["sleep_total"]}

sleepDataExt.select{startsWith("sleep") AND listOf("rem_proportion")}

sleepDataExt.letsPlot { }

sleepDataExt.letsPlot {x="sleep_total"; y="rem_proportion" } + geomPoint()

val myPlot = sleepDataExt.letsPlot {x="sleep_total"; y="rem_proportion" } + geomPoint()
myPlot

ggsave(myPlot, "testplot.svg")

sleepDataExt.letsPlot {x="sleep_total"; y="rem_proportion" }     +  
        geom_point(size=4, alpha=0.3)

sleepDataExt
    .letsPlot {x="sleep_total"; y="rem_proportion"; color="vore" } +  
    geomPoint(size=4, alpha=.7)

sleepDataExt
    .letsPlot {x="sleep_total"; y="rem_proportion"; color="vore"; size="brainwt" } +  
    geomPoint(size=4, alpha=.7) 


sleepDataExt.letsPlot { x="sleep_total"} + geomHistogram(binWidth = 2)

sleepDataExt.letsPlot { x="vore"; y="sleep_total"} + geomBoxplot()

sleepDataExt.letsPlot { x="vore"; y="sleep_total"} +
    geomBoxplot() + 
    geomPoint()

sleepDataExt.letsPlot { x="vore"; y="sleep_total"} +
    geomBoxplot() + 
    geomPoint(position=positionJitter(0.3))

sleepDataExt.letsPlot { x="conservation"} + geomBar()

sleepDataExt.letsPlot { x="conservation"; fill="vore"} +
    geomBar()

sleepDataExt.letsPlot { x="vore"; fill="conservation"} +
    geomBar()

sleepDataExt.letsPlot { x="vore"; fill="conservation"} + 
    geomBar(position=Pos.fill)

val corPlot = sleepData.letsPlot { x="bodywt"; y="sleep_total"} + 
    geomPoint()

corPlot + xlab("Body Weight")

sleepData
    .filter{it["bodywt"] lt 2000}
    .letsPlot { x="bodywt"; y="sleep_total"} + 
    geomPoint()

corPlot + xlab("Body Weight") + scaleXLog10()

corPlot + 
    xlab("Body Weight") + 
    scaleXLog10() +
    geomSmooth() + 
    ggtitle("Correlation between sleep time and body weight")

sleepData.letsPlot { x="bodywt"; y="sleep_total"} +
    geomPoint() +
//     scaleXLog10("Body Weight") + 
//     scaleYLog10("Sleep Time") + 
    facetWrap("vore") + // no scales argument?
    ggtitle("Correlation Total Sleep Time and Body Weight per Species")

@file:DependsOn("com.github.holgerbrandl:kravis:0.8.1")

sleepData.plot( x="bodywt", y="sleep_total")
    .geomPoint()
    .facetWrap("vore", scales=FacetScales.free)
    .title("Correlation Total Sleep Time and Body Weight per Species")

sleepDataExt
    .plot(x="sleep_total", y="rem_proportion", color="vore")
    .geomPoint(size=4.0, alpha=.7)
    .themeMinimal()
    .theme(axisTitle = ElementText(size = 20, color = RColor.red))
    .show()

sleepDataExt
    .plot(x="sleep_total", y="rem_proportion", color="vore", size="brainwt")
    .geomPoint(alpha=.7)
    .show()

sleepData
    .plot( x="bodywt", y="sleep_total") 
    .xLabel("Body Weight")
    .scaleXLog10()
    .geomPoint()
    .geomSmooth()