class Spacer(Frame): def __init__(self, parent, relief='raised', *args, **kw): apply(Frame.__init__, (self, parent) + args, kw) #self.grid_rowconfigure(0, weight=1) #self.grid_columnconfigure(1, weight=1) self.bg = self.cget('bg') self.relief = relief self.grid_columnconfigure(0, weight=1) self.canvas = Canvas(self, background=self.bg, width=1, height=6) self.canvas.grid(row=0, column=0, sticky=Tkinter.NSEW) #self.canvas.bind('<Configure>', self.resizeCallback) self.grid(sticky=Tkinter.EW) self.canvasL1 = [] self.canvasL2 = [] self.canvasL3 = [] self.refresh = 0 self.event = None self.bind('<Configure>', self.resizeAfter) def resizeAfter(self, event): self.event = event if self.refresh: return else: self.refresh = 1 self.after_idle(lambda: self.resize(self.event)) def resize(self, event): m = 6 w = event.width h = event.height N = int(h / m) self.canvas.config(width=w, height=h) if self.relief == 'sunken': fill1 = 'grey65' fill2 = 'grey95' else: fill1 = 'grey95' fill2 = 'grey65' for i in range(N): y = 1 + (i / float(N)) * h if i >= len(self.canvasL1): rect1 = self.canvas.create_rectangle(4, y + 0, w - 5, y + 2, width=0, fill=fill1) rect2 = self.canvas.create_rectangle(5, y + 1, w - 4, y + 3, width=0, fill=fill2) rect3 = self.canvas.create_rectangle(5, y + 1, w - 5, y + 2, width=0, fill=self.bg) self.canvasL1.append(rect1) self.canvasL2.append(rect2) self.canvasL3.append(rect3) else: self.canvas.coords(self.canvasL1[i], 4, y + 0, w - 5, y + 2) self.canvas.coords(self.canvasL2[i], 5, y + 1, w - 4, y + 3) self.canvas.coords(self.canvasL3[i], 5, y + 1, w - 5, y + 2) if N > len(self.canvasL1): for i in range(N, len(self.canvasL1)): self.canvas.delete(self.canvasL1[i]) self.canvas.delete(self.canvasL2[i]) self.canvas.delete(self.canvasL3[i]) self.refresh = 0
class PulldownMenu(Frame): # if using indentation then should use list as stack only # in other words only delete or insert at end # entries entries must have attribute text_attr or if this is # None then assume that entries entries themselves provide name def __init__(self, parent, callback=None, entries=None, text_attr=None, colors=None, outline='#602000', fill='#B05848', label_color='#501000', selected_index=-1, indent='', extra='', sticky='w', do_initial_callback=True, force_callback=False, *args, **kw): if (entries is None): entries = [] Frame.__init__(self, parent, sticky=sticky, *args, **kw) self.text_attr = text_attr #indentation and extra string for popup information self.indent = indent self.extra = extra self.callback = callback self.do_initial_callback = do_initial_callback self.force_callback = force_callback frame = Frame(self) #frame = parent self.frame = frame self.bg = frame.cget('bg') self.outline = outline self.fill = fill self.label_color = label_color self.menu = Menu(parent, tearoff=0, borderwidth=1, activeborderwidth=1) self.menu.images = [] self.label = Label(self, foreground=label_color) self.first_pass = True #self.entries = entries self.entries = [] self.colors = [] self.setup(entries, selected_index, colors=colors) s = 10 self.canvas = Canvas(self, width=s, height=s, background=self.bg) self.label.bind("<Button-1>", self.labelPopup) # below does not work for some reason #self.bind("<Button-1>", self.labelPopup) # below does not work any more since can have submenus and # no way to reach those with below since involves leaving menu #self.menu.bind("<Leave>", self.popdown) self.menu.bind("<Leave>", self.leave) self.menu.bind("<Button-1>", self.buttonPress) self.poppedUpSubmenu = False self.canvas.bind("<Button-1>", self.canvasPopup) self.canvas.bind("<Configure>", self.resizeCallback) self.label.grid(row=0, column=0, sticky=Tkinter.W) self.canvas.grid(row=0, column=1, sticky=Tkinter.E, padx=2) def leave(self, event): if (not self.poppedUpSubmenu): x = event.x y = event.y x1 = self.menu.winfo_width() y1 = self.menu.winfo_height() if (x < 0) or (y < 0) or (x >= x1) or (y >= y1): self.popdown() def buttonPress(self, event): if (event.widget != self.menu): return for n in range(len(self.entries) - 1, -1, -1): if event.y >= self.menu.yposition(n): break else: return item = self.entries[n] if (isinstance(item, {}.__class__) and item['kind'] == 'cascade'): self.poppedUpSubmenu = True else: self.poppedUpSubmenu = False def resizeCallback(self, *event): w = self.canvas.winfo_width() / 2 h = self.canvas.winfo_height() / 2 self.canvas.delete('all') self.canvas.create_polygon(1, 1, w, 2 * h - 1, 2 * w - 1, 1, fill=self.fill, outline=self.outline) def getEntry(self, entry_index, entries=None): if not entries: entries = self.entries if type(entry_index) == types.TupleType: e = entry_index[0] if len(entry_index) > 1: items = entries[e]['submenu'] entry = self.getEntry(entry_index[1:], items) else: entry = entries[e]['label'] elif (entry_index >= 0): entry = entries[entry_index] else: entry = None return entry def getEntryText(self, entry_index, entries=None): if not entries: entries = self.entries if type(entry_index) in (types.TupleType, types.ListType): e = entry_index[0] if len(entry_index) > 1: items = entries[e]['submenu'] text = self.getEntryText(entry_index[1:], items) else: text = entries[e]['label'] elif (entry_index != -1): #print 'getEntryText', entry_index, entries entry = entries[entry_index] if (self.text_attr): text = getattr(entry, self.text_attr) else: text = entry else: text = '' # TBD: for now text returned should not be unicode text = str(text) # below did not work for one person (Chinese characters??) but above did ###text = text.encode('utf-8') return text def setup(self, entries, selected_index, first_pass=True, colors=None): #print 'setup', entries, selected_index if first_pass is not None: self.first_pass = first_pass self.clearMenuItems() if (entries): e = list(entries) else: e = [No_entries_label] self.entries = e self.colors = colors or [None] * len(e) if type(selected_index) in (types.TupleType, types.ListType): if (not selected_index or selected_index[0] >= len(entries)): selected_index = -1 else: if (selected_index >= len(entries)): selected_index = -1 if (selected_index == -1): selected_index = 0 self.selected_index = -1 # changed in setSelectedIndex() self.setMenuItems() self.setSelectedIndex(selected_index) def substituteCascadeCallbacks(self, item, selected_index): if (item['kind'] == 'cascade'): n = 0 for subitem in item['submenu']: self.substituteCascadeCallbacks(subitem, selected_index + [n]) n = n + 1 else: callback = item.get('command') if (callback): label = item['label'] item['command'] = lambda: self.cascadeCallback( callback, label, tuple(selected_index)) def cascadeCallback(self, callback, label, selected_index): self.selected_index = selected_index self.setLabel(selected_index) callback(selected_index, label) def setMenuItems(self): self.menu.images = [] for n in range(len(self.entries)): columnbreak = 0 if n and n % 20 == 0: columnbreak = 1 item = self.entries[n] color = None if n < len(self.colors): color = self.colors[n] if isinstance(item, {}.__class__): if (item['kind'] == 'cascade'): self.substituteCascadeCallbacks(item, [n]) else: t = (self.indent * n) + self.extra + str(self.getEntryText(n)) command = lambda n=n: self.setSelectedIndex(n) if color: image = self.makeColorTile(color) item = { 'kind': 'command', 'accelerator': t, 'command': command, 'image': image, 'columnbreak': columnbreak } else: item = { 'kind': 'command', 'label': t, 'command': command, 'columnbreak': columnbreak } self.menu.addMenuItem(item) def clearMenuItems(self): if self.entries: self.menu.delete(0, len(self.entries)) # below assumes entry_index is integer not tuple def insert(self, entry_index, entry, make_selected=False, color=None): if (self.haveNoEntries()): if (entry_index != 0): return elif ((entry_index < 0) or (entry_index >= len(self.entries))): return self.clearMenuItems() self.checkForNone() self.entries.insert(entry_index, entry) self.colors.insert(entry_index, color) self.setMenuItems() if (make_selected or (self.selected_index == -1)): self.setSelectedIndex(entry_index, force_callback=True) else: self.setSelectedIndex(self.selected_index) # below assumes entry is integer not tuple def append(self, entry, make_selected=False, color=None): self.clearMenuItems() self.checkForNone() self.entries.append(entry) self.colors.append(color) self.setMenuItems() if (make_selected or (self.selected_index == -1)): self.setSelectedIndex(len(self.entries) - 1, force_callback=True) else: self.setSelectedIndex(self.selected_index) def checkForNone(self): if (self.haveNoEntries()): self.entries = [] self.colors = [] self.menu.delete(0) def haveNoEntries(self): if (len(self.entries) == 1 and self.entries[0] == No_entries_label): return True else: return False def baseIndex(self, ind): if type(ind) == types.TupleType: ind = ind[0] return ind # below assumes entry_index is integer not tuple def delete(self, entry_index, n=1): if (self.haveNoEntries()): return if ((entry_index < 0) or (entry_index >= len(self.entries))): return self.clearMenuItems() n1 = entry_index n2 = n1 + n m = self.baseIndex(self.selected_index) del self.entries[n1:n2] del self.colors[n1:n2] if (not self.entries): self.entries = [No_entries_label] self.colors = [None] self.setMenuItems() if ((m >= n1) and (m < n2)): if (n1 == 0): selected_index = 0 else: selected_index = n1 - 1 self.setSelectedIndex(selected_index, force_callback=True) def replace(self, entries, selected_index=-1, colors=None): self.menu.delete(0, len(self.entries)) self.setup(entries, selected_index, first_pass=False, colors=colors) def popdown(self, *event): self.menu.unpost() def setLabel(self, selected_index): text = self.getEntryText(selected_index) if (not text): text = self.extra # arbitrary self.label.set(text=text) def labelPopup(self, event): self.poppedUpSubmenu = False s = self.baseIndex(self.selected_index) x = event.x_root - event.x + 2 y = event.y_root - event.y - \ max(0, s) * (self.label.winfo_height() + 1) self.menu.post(x, y) def canvasPopup(self, event): s = self.baseIndex(self.selected_index) x = event.x_root - event.x - self.label.winfo_width() + 2 y = event.y_root - event.y - \ max(0, s) * (self.label.winfo_height() + 1) + \ self.canvas.winfo_height() - self.label.winfo_height() self.menu.post(x, y) def getSelectedIndex(self): return self.selected_index def setSelectedIndex(self, selected_index, force_callback=False): if (force_callback or self.force_callback or \ (selected_index != self.selected_index)): #print 'setSelectedIndex', self.entries, selected_index, self.selected_index, self.first_pass self.selected_index = selected_index self.setLabel(selected_index) if (self.callback): if (self.haveNoEntries()): if (not self.first_pass): self.callback(-1, None) elif (selected_index != -1): if (not self.first_pass or self.do_initial_callback): self.callback(selected_index, self.entries[selected_index]) self.first_pass = False def get(self): return self.getSelected() def getSelected(self): ind = self.getSelectedIndex() if (ind == -1): return None else: return self.getEntry(ind) def set(self, selected): self.setSelected(selected) def findEntryIndex(self, entry, entries=None): #print 'findEntryIndex1', entry, entries if (entries): noEntries = True else: noEntries = False entries = self.entries try: ind = entries.index(entry) return ind except: pass #print 'findEntryIndex2', entry, entries for n in range(len(entries)): e = entries[n] #print 'findEntryIndex3', entry, e if e == entry: if noEntries: return [n] else: return n elif type(e) == types.DictType: if (len(entries) > 1): try: ind = self.findEntryIndex(entry, e['submenu']) return [n] + ind except: pass else: if (entry == e['label']): return [n] raise 'unknown entry "%s"' % entry # selects first item found in entries which matches def setSelected(self, selected): #print 'setSelected1', selected, type(selected), self.entries try: selected_index = self.findEntryIndex(selected) except: #selected_index = -1 return #print 'setSelected2', selected_index self.setSelectedIndex(selected_index) def makeColorTile(self, color): image = Tkinter.PhotoImage() self.menu.images.append(image) if type(color) == type([]): colors = [scaleColor(self.menu, c, 1.0) for c in color] else: colors = [ scaleColor(self.menu, color, 1.0), ] cols = max(8, len(colors)) for x in range(cols): i = x % len(colors) c = colors[i] for y in range(16): image.put('{%s %s}' % (c, c), to=(2 * x, y)) return image
class ProgressBar(BasePopup): def __init__(self, parent, text='', progress=0, total=100, title=None, width=200, height=15, transient=True, *args, **kw): self.progress = float(progress) self.cWidth = width self.cHeight = height self.total = float(total) or 1.0 self.text = text BasePopup.__init__(self, parent=parent, title=title or 'Progress Bar', transient=transient, *args, **kw) def body(self, parent): self.bw = 2 # borderwidth width = self.cWidth + self.bw height = self.cHeight + self.bw self.label = Label(parent, text=self.text) self.label.grid(row=0, column=0, sticky=Tkinter.W) self.percent = Label(parent, text=' %') self.percent.grid(row=0, column=2, sticky=Tkinter.W) self.canvas = Canvas(parent, background='grey70', width=width, height=height) self.canvas.grid(row=0, column=1, sticky=Tkinter.W) self.canvas.create_rectangle(0, 0, width, height, outline='black', width=self.bw) self.bar = self.canvas.create_rectangle(self.bw, self.bw, self.bw, self.cHeight, outline='#B05848', fill='#B05848', width=self.bw) self.update() def update(self): p = self.progress / self.total width = int(self.cWidth * p) self.canvas.coords(self.bar, self.bw, self.bw, width, self.cHeight) self.percent.set(' %3.1d' % int(100 * p) + "%") self.label.set(self.text) self.update_idletasks() if self.progress == self.total: time.sleep(0.1) self.close() def increment(self, n=1): p = self.progress + n self.progress = max(min(self.total, p), 0) self.update() def get(self): return self.progress / self.total def set(self, value): if value <= 1: self.progress = float(value) * self.total elif value <= self.total: self.progress = float(value) else: raise 'Cannot set to a value exceeding the total' self.update() def setText(self, text=''): self.text = text self.label.set(self.text)
class ScrolledCanvas(Frame): def __init__(self, parent, resizeCallback=None, width=600, height=600, *args, **kw): self.bbox = None self.busy = 0 self.initialX = None self.initialY = None self.resizeCallback = resizeCallback apply(Frame.__init__, (self, parent) + args, kw) self.grid_rowconfigure(0, weight=1) self.grid_columnconfigure(0, weight=1) self.menu = Menu(self, tearoff=0, include_event=True) self.configMenu() self.canvas = Canvas(self, relief='flat', borderwidth=0, width=width, height=height) self.canvas.configure(xscrollincrement=2, yscrollincrement=2) self.canvas.grid(row=0, column=0, sticky=Tkinter.NSEW) self.horizScrollbar = Tkinter.Scrollbar(self, bg=self.cget('bg'), command=self.canvas.xview, orient=Tkinter.HORIZONTAL, borderwidth=1) self.vertScrollbar = Tkinter.Scrollbar(self, bg=self.cget('bg'), command=self.canvas.yview, orient=Tkinter.VERTICAL, borderwidth=1) self.canvas.configure(xscrollcommand=self.horizScrollbar.set, yscrollcommand=self.vertScrollbar.set) self.canvas.bind('<Configure>', self.resizeAfter) self.canvas.bind('<Button-1>', self.mouseButton1) self.canvas.bind('<Button-2>', self.mouseButton2) self.canvas.bind('<Button-3>', self.mouseButton3) self.canvas.bind('<ButtonRelease-1>', self.mouseButtonRelease1) self.canvas.bind('<ButtonRelease-2>', self.mouseButtonRelease2) self.canvas.bind('<B2-Motion>', self.mouseScroll) self.canvas.bind('<B3-Motion>', self.doNothing) self.canvas.bind('<Motion>', self.mouseEnter) self.canvas.bind('<Enter>', self.mouseEnter) def doNothing(self, event): pass def mouseEnter(self, event): if self.menu.winfo_ismapped(): self.removeMenu() def refresh(self): self.bbox = self.canvas.bbox('all') bbox = self.bbox if not bbox: self.busy = 0 return cWidth = int(self.canvas.cget('width')) cHeight = int(self.canvas.cget('height')) if cHeight > bbox[3] - bbox[1]: self.vertScrollbar.grid_forget() else: self.vertScrollbar.grid(row=0, column=1, sticky=Tkinter.NS) if cWidth > bbox[2] - bbox[0]: self.horizScrollbar.grid_forget() else: self.horizScrollbar.grid(row=1, column=0, sticky=Tkinter.EW) x1 = bbox[0] y1 = bbox[1] x2 = max(bbox[0] + cWidth, bbox[2]) y2 = max(bbox[1] + cHeight, bbox[3]) self.canvas.configure(scrollregion=(x1, y1, x2, y2)) self.update_idletasks() self.busy = 0 def printCanvas(self, *event): fileTypes = [FileType('PostScript', ['*.ps']), FileType('All', ['*'])] fileSelectPopup = FileSelectPopup(self, file_types=fileTypes, title='Print canvas to file', dismiss_text='Cancel', selected_file_must_exist=False) fileName = fileSelectPopup.getFile() self.bbox = bbox = self.canvas.bbox('all') w = bbox[2] - bbox[0] h = bbox[3] - bbox[1] self.canvas.postscript(colormode='color', file=fileName, x=bbox[0], y=bbox[1], width=w + 2, pagewidth='21.c', height=h + 2) def configMenu(self): items = [ { 'kind': 'command', 'label': 'Print to file', 'command': self.printCanvas }, ] self.menu.setMenuItems(items) def removeMenu(self, *event): self.menu.unpost() def mouseButton3(self, event): self.menu.popupMenu(event) def mouseButton1(self, event): self.removeMenu() if not self.initialX: self.initialX = event.x self.initialY = event.y def mouseButton2(self, event): self.removeMenu() if not self.initialX: self.initialX = event.x self.initialY = event.y self.initialPX = self.horizScrollbar.get()[0] self.initialPY = self.vertScrollbar.get()[0] def mouseButtonRelease1(self, event): self.initialX = None self.initialY = None def mouseButtonRelease2(self, event): self.initialX = None self.initialY = None self.initialPX = None self.initialPY = None def mouseScroll(self, event): self.menu.unpost() bbox = self.bbox if not bbox: return bW = float(bbox[2] - bbox[0]) bH = float(bbox[3] - bbox[1]) cWidth = int(self.canvas.cget('width')) cHeight = int(self.canvas.cget('height')) if cWidth < bW: dx = self.initialX - event.x prop = self.initialPX + (dx / bW) self.canvas.xview('moveto', prop) if cHeight < bH: dy = self.initialY - event.y prop = self.initialPY + (dy / bH) self.canvas.yview('moveto', prop) def resizeAfter(self, event): if self.busy: return else: self.busy = 1 self.removeMenu() self.after_idle(lambda: self.resize(event)) def resize(self, event): if self.resizeCallback: self.after_idle( lambda: self.resizeCallback(event.width, event.height)) self.canvas.configure(width=event.width, height=event.height) self.after_idle(self.refresh)
class ColorChooser(Frame): def __init__(self, parent, *args, **kw): apply(Frame.__init__, (self, parent) + args, kw) self.grid_rowconfigure(0, weight=1) self.grid_rowconfigure(1, weight=0) self.grid_rowconfigure(2, weight=0) self.grid_rowconfigure(3, weight=0) self.canvas = Canvas(self, relief='sunken', borderwidth=2, width=150) self.canvas.grid(row=0, column=1, columnspan=1, sticky=Tkinter.NSEW, pady=2, padx=2) frame = Frame(self) frame.grid_columnconfigure(1, weight=1) self.boxSize = 100 self.numPixels = 10 self.pixelWidth = round(self.boxSize / self.numPixels) self.colorBox = Canvas(frame, relief='sunken', borderwidth=2, width=self.boxSize + 2, height=self.boxSize + 2) self.colorBox.bind('<Button-1>', self.pickInColorBox) self.colorBox.bind('<Button-2>', self.pickInColorBox) self.colorBox.bind('<Button-3>', self.pickInColorBox) self.colorBox.grid(row=0, column=2, rowspan=3, sticky=Tkinter.NSEW, padx=4, pady=4) self.pixel = [] self.colors = [] self.setupColorBox() self.scale = Tkinter.Scale(frame, orient=Tkinter.VERTICAL, length=self.boxSize, from_=0, to=99, label='', showvalue=0, command=self.refreshColorBox) self.scale.grid(row=0, column=3, rowspan=3, sticky=Tkinter.NS, padx=4, pady=4) frame.grid(row=1, column=0, columnspan=2, sticky=Tkinter.NSEW) labels = ('Red', 'Green', 'Blue') self.labeled_scale = 3 * [None] for n in range(3): label = Label(frame, text=labels[n] + ':', anchor=Tkinter.W) label.grid(row=n, column=0, sticky=Tkinter.EW) self.labeled_scale[n] = LabeledScale( frame, values=range(101), label_format='%3d', set_callback=self.scaleCallback) self.labeled_scale[n].grid(row=n, column=1, sticky=Tkinter.EW) def setupColorBox(self): pw = self.pixelWidth np = self.numPixels for x in range(np): self.pixel.append([]) self.colors.append([]) for y in range(np): rgb = hsbToRgb(0, 1 - x / float(np), 1 - y / float(np)) c = '#%02x%02x%02x' % (rgb[0] * 255, rgb[1] * 255, rgb[2] * 255) p = self.colorBox.create_rectangle(2 + pw * x, 2 + pw * y, 2 + (pw * x) + pw, (2 + pw * y) + pw, outline=c, fill=c) self.pixel[x].append(p) self.colors[x].append(rgb) def pickInColorBox(self, event): x = int(event.x / self.pixelWidth) y = int(event.y / self.pixelWidth) X = min(x, self.numPixels - 1) Y = min(y, self.numPixels - 1) x = max(X, 0) y = max(Y, 0) rgb = self.colors[x][y] self.setColor(rgb[0], rgb[1], rgb[2]) def refreshColorBox(self, H): np = self.numPixels H = int(H) / 100. for x in range(np): for y in range(np): rgb = hsbToRgb(H, 1 - x / float(np), 1 - y / float(np)) c = '#%02x%02x%02x' % (rgb[0] * 255, rgb[1] * 255, rgb[2] * 255) self.colorBox.itemconfigure(self.pixel[x][y], outline=c, fill=c) self.colors[x][y] = rgb def scaleCallback(self, *index): c = 3 * [0] for n in range(3): c[n] = int(2.559 * float(self.labeled_scale[n].getValue())) color = '#%02x%02x%02x' % (c[0], c[1], c[2]) self.canvas.config(bg=color) def setColor(self, r, g, b): c = (r, g, b) for n in range(3): s = round(100.0 * c[n]) s = min(100, s) self.labeled_scale[n].set(s) def getColor(self): c = 3 * [0] for n in range(3): c[n] = float(self.labeled_scale[n].getValue()) / 100.0 return tuple(c)
class ToggleArrow(Frame): def __init__(self, parent, arrowSize=11, outline='#5050b0', fill='#a0a0ff', isArrowClosed=True, callback=None, *args, **kw): self.arrowSize = arrowSize self.isArrowClosed = isArrowClosed self.callback = callback apply(Frame.__init__, (self, parent) + args, kw) bg = kw.get('bg') if (not bg): bg = kw.get('background') if (not bg): bg = parent.cget('bg') # below does not work for some reason, instead explicitly create rectangle below #self.canvas = Canvas(self, width=arrowSize, height=arrowSize, bg=bg) self.canvas = Canvas(self, width=arrowSize + 3, height=arrowSize + 1) self.canvas.grid(row=0, column=0) # need +/-1 otherwise get line along borders self.canvas.create_rectangle(-1, -1, arrowSize + 3, arrowSize + 1, fill=bg) # arbitrary coords self.arrow = self.canvas.create_polygon(0, 0, 1, 0, 1, 1, fill=fill, outline=outline) self.drawArrow() self.canvas.bind('<Button-1>', self.flipState) def drawArrow(self): s = self.arrowSize if (self.isArrowClosed): self.canvas.coords(self.arrow, 2, 0, s + 2, s / 2, 2, s) else: self.canvas.coords(self.arrow, 2, 0, s + 2, 0, (s / 2) + 2, s) def flipState(self, *event): self.isArrowClosed = not self.isArrowClosed if (self.callback): self.callback(self.isArrowClosed) self.drawArrow()
class PulldownList(Frame): # if using indentation then should use list as stack only # in other words only delete or insert at end # There is a 1:1 correspondance between the ordered lists of texts and # list of objects # categories is the category name for each text/object # objects with the same category will go under a submenu # of that name def __init__(self, parent, callback=None, texts=None, objects=None, categories=None, colors=None, index=0, prefix='', indent='', initCallback=False, forceCallback=False, numbering=False, arrowLine='#602000', arrowFill='#B05848', labelColor='#501000', menuBg='#F0F0FF', sticky='w', *args, **kw): Frame.__init__(self, parent, sticky=sticky, *args, **kw) self.callback = callback self.texts = texts or [] self.objects = objects or [] self.categories = categories or [] self.colors = colors or [] self.prefix = prefix self.indent = indent self.initCallback = initCallback self.numbering = numbering self.arrowLine = arrowLine self.arrowFill = arrowFill self.labelColor = labelColor # Current selection self.index = None self.object = NullText self.rows = [] self.bg = self.cget('bg') self.label = Label(self, foreground=labelColor) self.canvas = Canvas(self, width=12, height=12, background=self.bg) self.menu = Menu(self.canvas, tearoff=False, bg=menuBg, relief='solid', borderwidth=1, activeborderwidth=1) self.menu.images = [] # Photoimage has to remain referenced self.setup(self.texts, self.objects, index, self.colors, self.categories) self.label.bind("<Button-1>", self._labelClick) self.menu.bind("<Leave>", self._leave) self.canvas.bind("<Button-1>", self._canvasClick) self.canvas.bind("<Configure>", self._resizeCallback) self.grid_columnconfigure(0, weight=1) self.label.grid(row=0, column=0, sticky='w') self.canvas.grid(row=0, column=1, sticky='w', padx=2) # # Retrieval # def get(self): return (self.getText(), self.getObject()) def getSelected(self): return self.get() def getObject(self): return self._fetch(self.index, self.objects) def getText(self): return self._fetch(self.index, self.texts) def getSelectedIndex(self): return self.index # # Setting selected # def set(self, item): # Works with an object or a text index = None if item in self.texts: index = list(self.texts).index(item) elif item in self.objects: index = list(self.objects).index(item) if index is not None: self.setIndex(index) def setSelected(self, item): self.set(item) def setIndex(self, index, doCallback=False): self.index = index if self.objects: obj = self.objects[index] if obj is not self.object: self.object = obj if (doCallback or self.initCallback) and self.texts and self.callback: self.callback(obj) self._updateLabel() # # Bulk configuration # def clear(self): self.setup([], [], 0) def setup(self, texts, objects, index, colors=None, categories=None): self.texts = texts nTexts = len(texts) if not objects: objects = texts while len(objects) < nTexts: objects.append(None) self.objects = objects if colors is None: self.colors = [None] * nTexts else: while len(colors) < nTexts: colors.append(None) self.colors = colors if categories is None: self.categories = [None] * nTexts else: while len(categories) < nTexts: categories.append(None) self.categories = categories self._setMenuItems() self.setIndex(index or 0) # # In-place/minor configuration # def insert(self, index, text, object=None, color=None, category=None, select=False): index = max(0, min(len(self.texts), index)) self.texts.insert(index, text) self.objects.insert(index, object) self.colors.insert(index, color) self.categories.insert(index, category) self._setMenuItems() if select: self.setIndex(index) def append(self, text, object=None, color=None, category=None, select=False): self.insert(len(self.texts), text, object=None, color=None, category=None, select=False) def delete(self, index, howMany=1): if index < 0: return elif index >= len(self.texts): return end = min(index + howMany, len(self.texts)) self._clearMenu() del self.texts[index:end] del self.objects[index:end] del self.colors[index:end] del self.categories[index:end] self._setMenuItems() index = min(len(self.texts) - 1, self.index) self.setIndex(index) # # Internal methods # def _leave(self, event): x = event.x y = event.y x1 = self.menu.winfo_width() y1 = self.menu.winfo_height() if (x < 0) or (y < 0) or (x >= x1) or (y >= y1): self._popdown() def _resizeCallback(self, *event): c = self.canvas w = c.winfo_width() - 1 h = c.winfo_height() - 1 c.delete('all') c.create_rectangle(0, 0, 10, 2, fill=self.arrowFill, outline=self.arrowLine) c.create_polygon(0, 4, 0, 6, 5, 11, 10, 6, 10, 4, fill=self.arrowFill, outline=self.arrowLine) def _fetch(self, index, array): if index is None: return None if index < 0: index += len(array) if index < 0: return None elif index >= len(array): return None else: return array[index] def _setMenuItems(self): self._clearMenu() self.menu.images = [] # Clear photoimages if not self.texts and not self.menu.entrycget(1, 'label'): item = {'kind': 'command', 'label': NullText, 'command': None} self.menu.addMenuItem(item) self.rows = [0] return topList = [] categoryDict = {} for i in range(len(self.texts)): text = self.texts[i] color = self.colors[i] category = self.categories[i] if category: if categoryDict.get(category) is None: categoryDict[category] = [] topList.append((None, category, None, category)) categoryDict[category].append((i, text, color)) else: topList.append((i, text, color, None)) row = 0 for index, text, color, cat in topList: columnbreak = 0 if row and row % 20 == 0: columnbreak = 1 if cat: string = (self.indent * row) + self.prefix + text items = [] rowB = 0 for index2, text2, color2 in categoryDict.get(cat, []): columnbreakB = 0 if rowB and rowB % 20 == 0: columnbreakB = 1 if self.numbering: number = '%d%. ' % (index2 + 1) else: number = '' string2 = number + self.prefix + text2 command = lambda n=index2: self.setIndex(n, True) if color2: image = self._makeColorTile(color2) item2 = { 'kind': 'command', 'accelerator': string2, 'command': command, 'image': image, 'columnbreak': columnbreakB } else: item2 = { 'kind': 'command', 'label': string2, 'command': command, 'columnbreak': columnbreakB } items.append(item2) self.rows.append(row) rowB += 1 item = { 'kind': 'cascade', 'label': string, 'submenu': items, 'columnbreak': columnbreak } else: if self.numbering: number = '%d%. ' % (index + 1) else: number = '' string = (self.indent * row) + number + self.prefix + text command = lambda n=index: self.setIndex(n, True) if color: image = self._makeColorTile(color) item = { 'kind': 'command', 'accelerator': string, 'command': command, 'image': image, 'columnbreak': columnbreak } else: item = { 'kind': 'command', 'label': string, 'command': command, 'columnbreak': columnbreak } self.menu.addMenuItem(item) self.rows.append(row) row += 1 def _clearMenu(self): self.menu.delete(0, 'end') self.index = 0 self.rows = [] def _popdown(self, *event): self.menu.unpost() def _labelClick(self, event): s = self.rows[self.index] #x = event.x_root - event.x + 2 + self.label.winfo_width() x = event.x_root - 2 y = event.y_root - event.y - max(0, s) * (self.label.winfo_height() + 1) self.menu.post(x, y) def _canvasClick(self, event): s = self.rows[self.index] x = event.x_root - event.x + 2 y = event.y_root - event.y - max(0, s) * (self.label.winfo_height() + 1) self.menu.post(x, y) def _updateLabel(self): if self.texts: text = self.texts[self.index] or NullText else: text = NullText self.label.set(text=text) def _makeColorTile(self, color): image = Tkinter.PhotoImage() self.menu.images.append(image) if type(color) == type([]): colors = [scaleColor(self.menu, c, 1.0) for c in color] else: colors = [ scaleColor(self.menu, color, 1.0), ] cols = max(8, len(colors)) for x in range(cols): i = x % len(colors) c = colors[i] for y in range(16): image.put('{%s %s}' % (c, c), to=(2 * x, y)) return image