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

devs = importr('grDevices')

main = "Distribution of Orbit Types"
path = "tmp/bahntypen.png"

rep = reporter("galaxy", "comets", ["ecc"])
num = ro.r.nrow(rep.df)[0]
vals = ro.r.table(ro.r.cut(rep.df.rx2(1), [0, 0.9999, 1.0001, 2]))

label = lambda label, i: "%s %s %%" %(label, round((vals[i]*100)/num))
labels = [label("Ellipse", 0), label("Parabola", 1), label("Hyperbola", 2)]
col = ro.r.rainbow(len(labels))

devs.png(file=path, width=512, height=512)
ro.r.pie(vals, labels=labels, col=col, main=main)
devs.dev_off()

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