# Program to provide a PyChart interface to clroses.py.

# This program was written to provide small rose images for a web page.
# The default output file name is computed from 'rose<style>x<petal>.png'.
import clroses
from pychart import *
import sys, getopt

filename = None
style = 1
petal = 2
vectors = 99
width = 150

try:
    opts, args = getopt.getopt(sys.argv[1:], "f:p:s:w:v:", \
                               ['file=', 'petal=', 'style=', 'width=', 'vectors='])
except getopt.error, err:
    print err
    print '''Usage: python pcroses.py
    [-f name | --file name]
    [-p val | --petal val]
    [-s val | --style val]
    [-w val | --width val]
    [-v val | --vectors val]'''
    sys.exit(1)

for opt in opts:
    if opt[0] == '-f' or opt[0] == '--file':
        filename = int(opt[1])
    elif opt[0] == '-p' or opt[0] == '--petal':
        petal = int(opt[1])
    elif opt[0] == '-s' or opt[0] == '--style':
        style = int(opt[1])
    elif opt[0] == '-w' or opt[0] == '--width':
        width = int(opt[1])
    elif opt[0] == '-v' or opt[0] == '--vectors':
        vectors = int(opt[1])

if not filename:
    filename = 'rose%dx%d.png' % (style, petal)
can = canvas.init(filename)
thin_line = line_style.T(width = .1)
white_line = line_style.T(color = color.white)
 
# Many images don't reach (x,y) = (-1,0), and PyChart or GhostView seems to
# want to "optimize" the image area.  The following seems to force things to
# display the whole domain.
can.rectangle(white_line, None, 0, 0, width, width, shadow=None)
                         
# Complains about pcroses options: theme.get_options()

# There are many links between clroses and the application, but we
# need none of them, so we can use clroses directly.
rose = clroses.rose()
upoints = rose.rose(style, petal, vectors)
spoints = rose.rescale(upoints, (0, 0), width/2)

ar = area.T(x_range=(-width/2, width/2), size = (width, width),
            y_range=(-width/2, width/2), y_grid_style = None, legend = None)

plot = line_plot.T(data=spoints, ycol=1, line_style=thin_line)
ar.add_plot(plot)

# Draw everything
ar.draw()