#!/usr/bin/env python
from rpy2.robjects.packages import importr
from rreport import reporter
import rpy2.robjects as ro

devs = importr('grDevices')

main = "3. Kepler's Law"
path = "tmp/Kepler3.png"

rep = reporter("galaxy", "comets", ["semaj_axs", "period"], {"semaj_axs":{"$lt":100}})
x_lab = "Semi-major axis / [AU]"
x_vals = ro.FloatVector([x**3 for x in rep.df.rx2(1)])

y_lab = "Orbital Period / [Years]"
y_vals = ro.FloatVector([x**2 for x in rep.df.rx2(2)])

devs.png(file=path, width=512, height=512)
ro.r.plot(x_vals, y_vals, xlab=x_lab, ylab=y_lab, main=main)
devs.dev_off()

rep.respond({"fileref":path})