class Channel(wx.Panel):
def __init__(self, parent, writer, doc, log, color=(0, 0, 0)):
wx.Panel.__init__(self, parent, style=wx.SUNKEN_BORDER)
self.writer = writer
self.color = color
self.sizer = wx.BoxSizer(wx.VERTICAL)
self.SetSizer(self.sizer)
self.doc = doc
self.corpus = Corpus([doc])
self.log = log
self.active = False
context = self.log.GetValue().split()[-2:]
suggestions = self.corpus.suggest(context, 20)
self.keyboard = Keyboard(self, self.doc.name, suggestions, self.log)
self.sizer.Add(self.keyboard)
#self.sizer.Add(wx.StaticLine(self, -1, wx.Point(10, 30), wx.Size(200, 30)))
#self.inspector = Inspector(self, doc)
#self.inspector.SetBackgroundColour((150,150,150))
#self.sizer.Add(self.inspector)
def suggest(self):
context = self.log.GetValue().split()[-2:]
return self.corpus.suggest(context, 20)
def weighted_choice(self, choices):
total = sum(w for (c, w) in choices)
r = random.uniform(0, total)
upto = 0
for c, w in choices:
if upto + w >= r:
return c
upto += w
assert False, "Shouldn't get here"
def refresh(self):
context = self.log.before().split()[-2:]
suggestions = self.corpus.suggest(context, 20)
#print self.weighted_choice(suggestions)
self.keyboard.Hide()
self.keyboard = Keyboard(self, self.doc.name, suggestions, self.log)
if self.active:
self.keyboard.SetForegroundColour(self.color)
self.keyboard.header.SetForegroundColour(self.color)
#if sum(self.color) > 500:
# self.keyboard.header.SetForegroundColour("Black")
#else:
# self.keyboard.header.SetForegroundColour("White")
self.keyboard.Layout()
else:
#self.keyboard.SetForegroundColour((80,80,80))
self.keyboard.header.SetBackgroundColour((0, 0, 0))
self.sizer.Prepend(self.keyboard)
self.Layout()
