channel.py

# a channel contains a corpus and associated settings

from Tkinter import *
import tkFont
from tkSimpleDialog import askstring
from tkFileDialog   import asksaveasfilename
import corpus
import operator

class Channel(Frame):
	def __init__(self, parent, textframe, corpus, num=0):
		Frame.__init__(self, parent)
		self.channel_name = corpus.name
		self.channel_num = num
<<<<<<< Updated upstream
		self.mode = 'letterlabel'
=======
		self.mode = 'shift'
>>>>>>> Stashed changes
		self.num_options = 15
		self.settings = {'color': 'black'}
		self.parent = parent
		self.textframe = textframe
		self.corpus = corpus
		self.font = parent.font
		self.pack(side = LEFT)
		self.current_options = self.get_options()
		self.keyboard = Frame()
		self.wt_scale = Scale(from_=-100, to=100, orient=HORIZONTAL)
		self.wt_scale.set(10)
		self.refresh_keyboard()
		self.keyboard.pack(anchor = W)

	# first member of ith tuple is the label. second member is the keystroke to input i
	def optionmap(self):
		if self.mode == 'alpha':
			return [('1', '1'), ('2', '2'), ('3', '3'), ('4', '4'), ('5', '5'), 
				('6', '6'), ('7', '7'), ('8', '8'), ('9', '9'), ('0', '0'),
				('a', 'a'), ('b', 'b'), ('c', 'c'), ('d', 'd'), ('e', 'e'), 
				('f', 'f'), ('g', 'g'), ('h', 'h'), ('i', 'i'), ('j', 'j')]
		elif self.mode == 'shift':
			return [('1', '1'), ('2', '2'), ('3', '3'), ('4', '4'), ('5', '5'), 
				('6', '6'), ('7', '7'), ('8', '8'), ('9', '9'), ('0', '0'),
				('1*', '!'), ('2*', '@'), ('3*', '#'), ('4*', '$'), ('5*', '%'), 
				('6*', '^'), ('7*', '&'), ('8*', '*'), ('9*', '('), ('0*', ')')]
		elif self.mode == 'letterlabel':
			return [('a', 'a'), ('b', 'b'), ('c', 'c'), ('d', 'd'), ('e', 'e'), 
				('f', 'f'), ('g', 'g'), ('h', 'h'), ('i', 'i'), ('j', 'j'),
				('k', 'k'), ('l', 'l'), ('m', 'm'), ('n', 'n'), ('o', 'o'), 
				('p', 'p'), ('q', 'q'), ('r', 'r'), ('s', 's'), ('t', 't')] 
		
	def make_keyboard(self, parent, words_and_scores, weight=100):
		keyboard = Frame(parent, padx = 10)
		header = Frame(keyboard)
		self.title = Label(header, text = self.channel_name, fg = self.settings['color'], font = self.font)
		self.title.pack(side = LEFT)
		Button(header, text = 'X', command = self.onDel).pack(side = RIGHT)
		self.wt_scale = Scale(header, from_=-100, to=100, orient=HORIZONTAL)
		self.wt_scale.set(weight)
		self.wt_scale.pack()
		header.pack()
		mainkeys = Frame(keyboard)
		for i in range(len(words_and_scores)):
			optkey = Frame(mainkeys)
			num = (i + 1) % 10
			keylabel = '%s.' % self.optionmap()[i][0]
			keystroke = self.optionmap()[i][1]
			option = words_and_scores[i][0]
			score = words_and_scores[i][1]
			color = self.score_to_color(score*2)
			#print score, color
			Label(optkey, text = keylabel, width = 4, anchor = W, font = self.font, bg = color, fg = 'white').pack(side = LEFT)
			label = option
			b = Label(optkey, text=label, font = self.font, width = 14, anchor = W, borderwidth = 0, 
			#command= lambda word=option: self.onAddWord(word),
			 pady = 0, padx = 10)
			b.pack(side = LEFT)
			self.textframe.bind(keystroke, lambda event, arg=option: self.onAddWord(arg))
			optkey.pack(side = TOP)
		mainkeys.pack()		
		return keyboard
		
	def score_to_color(self, score):
		return self.get_color(score,0,0)
	
	# convert list of three RGB values to a string representing a color
	def get_color(self, x, y, z):
		if x >= 256:
			x = 255
		if y >= 256:
			y = 255
		if z >= 256:
			z = 255
		#print x, y, z
		return '#%02x%02x%02x' % (x, y, z)
	
	def getWeight(self):
		return self.wt_scale.get()
	
	def onDel(self):
		self.master.removeChannel(self.channel_num)
		self.destroy()
	
	def onAddWord(self, word):
		t = self.textframe
		t.insert(INSERT, " "+str(word))
		t.see(END)
		self.refresh_keyboard()
		return 'break'
	
	def refresh_keyboard(self):
		self.current_options = self.get_options()
		words_and_scores = []
		#print '\n',self.channel_name
		for word, score in self.current_options[0:self.num_options]:
			weighted_score = score * self.wt_scale.get()
			#print word, weighted_score
			words_and_scores.append((word, weighted_score))
		weight = self.wt_scale.get()
		self.keyboard.destroy()
		self.keyboard = self.make_keyboard(self, words_and_scores, weight)
		self.keyboard.pack(anchor = W)
		
	def get_options(self):
		previous_words = self.parent.get_previous()
		next_words = self.parent.get_next()
		
		full_list = self.corpus.suggest(previous_words, next_words)
		short_list = full_list[0:100]
		suggestions = full_list[0:self.num_options]
		return short_list