title = 'Color spectrum demonstration'

# Import Pmw from this directory tree.
import sys
sys.path[:0] = ['../../..']

import string
import Tkinter
import Pmw

class Demo:
    def __init__(self, parent):
	parent = Tkinter.Frame(parent)
	parent.pack(padx=10, pady=10, fill='both', expand=1)
	self.width = 350
	self.height = 250
	self.canvas = Tkinter.Canvas(parent,
		width = self.width, height = self.height)
	self.canvas.grid(row = 0, column = 0, columnspan = 2, sticky = 'news')

	self.numColors = Pmw.EntryField(parent,
		labelpos = 'w',
		label_text = 'Number of colors:',
		entry_width = 5,
		validate = 'numeric',
		command = Pmw.busycallback(self.execute))
	self.numColors.grid(row = 1, column = 0, sticky = 'ew')

	self.correction = Pmw.EntryField(parent,
		labelpos = 'w',
		label_text = 'Correction:',
		validate = 'real',
		entry_width = 5,
		command = Pmw.busycallback(self.execute))
	self.correction.grid(row = 1, column = 1, sticky = 'ew')

	self.saturation = Pmw.EntryField(parent,
		labelpos = 'w',
		label_text = 'Saturation:',
		validate = 'real',
		entry_width = 5,
		command = Pmw.busycallback(self.execute))
	self.saturation.grid(row = 2, column = 0, sticky = 'ew')

	self.intensity = Pmw.EntryField(parent,
		labelpos = 'w',
		label_text = 'Intensity:',
		validate = 'real',
		entry_width = 5,
		command = Pmw.busycallback(self.execute))
	self.intensity.grid(row = 2, column = 1, sticky = 'ew')

	self.extraOrange = Pmw.EntryField(parent,
		labelpos = 'w',
		label_text = 'Emphasize orange (0 or 1):',
		validate = {'validator' : 'numeric', 'min' : 0, 'max' : 1},
		entry_width = 5,
		command = Pmw.busycallback(self.execute))
	self.extraOrange.grid(row = 3, column = 0, sticky = 'ew')

	self.text = Pmw.EntryField(parent,
		labelpos = 'w',
		label_text = 'Text:',
		entry_width = 20,
		command = Pmw.busycallback(self.execute))
	self.text.grid(row = 4, column = 0, sticky = 'ew')

	self.brightness = Pmw.EntryField(parent,
		labelpos = 'w',
		label_text = 'Brightness:',
		validate = 'real',
		entry_width = 5,
		command = Pmw.busycallback(self.execute))
	self.brightness.grid(row = 3, column = 1, sticky = 'ew')

	self.radiobuttons = Pmw.RadioSelect(parent,
		command = Pmw.busycallback(self.radio_cb),
	)
        self.radiobuttons.grid(row = 4, column = 1)
        self.radiobuttons.add('Use saturation\nand intensity')
        self.radiobuttons.add('Use\nbrightness')

	parent.grid_columnconfigure(0, weight = 1)
	parent.grid_columnconfigure(1, weight = 1)
	parent.grid_rowconfigure(0, weight = 1)

	Pmw.alignlabels((self.numColors, self.saturation, self.extraOrange))
	Pmw.alignlabels((self.correction, self.intensity, self.brightness))

	# Set initial values for all entries.
	self.numColors.setentry('64')
	self.correction.setentry('1.0')
	self.saturation.setentry('1.0')
	self.intensity.setentry('1.0')
	self.extraOrange.setentry('1')
	self.brightness.setentry('0.7')
	self.text.setentry('This is a test')
	self.radiobuttons.invoke('Use saturation\nand intensity')

	self.execute()

    def radio_cb(self, value):
	self.execute()

    def execute(self):
	try:
	    numColors = string.atoi(self.numColors.get())
	    correction = string.atof(self.correction.get())
	    saturation = string.atof(self.saturation.get())
	    intensity = string.atof(self.intensity.get())
	    extraOrange = string.atof(self.extraOrange.get())
	    brightness = string.atof(self.brightness.get())
	except ValueError:
	    self.numColors.bell()
	    return

	if numColors <= 0:
	    self.numColors.bell()
	    return

        self.canvas.delete('all')

	colorList = Pmw.Color.spectrum(
		numColors, correction, saturation, intensity, extraOrange)
	extent = 360.0 / numColors

        useBrightness = \
                (self.radiobuttons.getcurselection() == 'Use\nbrightness')

	if numColors == 1:
	    # Special case circle, since create_arc does not work when
	    # extent is 360.
	    background = colorList[0]
            if useBrightness:
                background = Pmw.Color.changebrightness(
                        self.canvas, background, brightness)
	    self.canvas.create_oval(10, 10, self.width - 10, self.height - 10,
		fill = background, outline = background)

	for index in range(numColors):
	    start = index * extent - extent / 2
	    background = colorList[index]
            if useBrightness:
                background = Pmw.Color.changebrightness(
                        self.canvas, background, brightness)
	    self.canvas.create_arc(10, 10, self.width - 10, self.height - 10,
		start = start, extent = extent,
		fill = background, outline = background)

        text = self.text.get()
        self.canvas.create_text(self.width / 2, self.height / 3, text = text)
        self.canvas.create_text(self.width / 2, self.height / 2, text = text)
        self.canvas.create_text(self.width / 2, 2 * self.height / 3, text = text)

######################################################################

# Create demo in root window for testing.
if __name__ == '__main__':
    root = Tkinter.Tk()
    Pmw.initialise(root)
    root.title(title)

    exitButton = Tkinter.Button(root, text = 'Exit', command = root.destroy)
    exitButton.pack(side = 'bottom')
    widget = Demo(root)
    root.mainloop()
