Extension to wed that handles styles, fonts and sizes

1 files changed, 627 insertions, 0 deletions

diff --git a/Mac/Demo/waste/swed.py b/Mac/Demo/waste/swed.py
new file mode 100644
index 0000000..c1b730d
--- /dev/null
+++ b/Mac/Demo/waste/swed.py
@@ -0,0 +1,627 @@
+# A minimal text editor.
+#
+# To be done:
+# - Functionality: find, etc.
+
+from Menu import DrawMenuBar
+from FrameWork import *
+import Win
+import Qd
+import Res
+import Fm
+import waste
+import WASTEconst
+import Scrap
+import os
+import macfs
+
+UNDOLABELS = [ # Indexed by WEGetUndoInfo() value
+	None, "", "typing", "Cut", "Paste", "Clear", "Drag", "Style"]
+	
+# Style and size menu. Note that style order is important (tied to bit values)
+STYLES = [
+	("Bold", "B"), ("Italic", "I"), ("Underline", "U"), ("Outline", "O"),
+	("Shadow", ""), ("Condensed", ""), ("Extended", "")
+	]
+SIZES = [ 9, 10, 12, 14, 18, 24]
+	
+BIGREGION=Qd.NewRgn()
+Qd.SetRectRgn(BIGREGION, -16000, -16000, 16000, 16000)
+
+class WasteWindow(ScrolledWindow):
+	def open(self, path, name, data):
+		self.path = path
+		self.name = name
+		r = windowbounds(400, 400)
+		w = Win.NewWindow(r, name, 1, 0, -1, 1, 0x55555555)
+		self.wid = w
+		vr = 0, 0, r[2]-r[0]-15, r[3]-r[1]-15
+		dr = (0, 0, 10240, 0)
+		Qd.SetPort(w)
+		Qd.TextFont(4)
+		Qd.TextSize(9)
+		flags = WASTEconst.weDoAutoScroll | WASTEconst.weDoOutlineHilite | \
+			WASTEconst.weDoMonoStyled | WASTEconst.weDoUndo
+		self.ted = waste.WENew(dr, vr, flags)
+		style, soup = self.getstylesoup()
+		self.ted.WEInsert(data, style, soup)
+		self.ted.WESetSelection(0,0)
+		self.ted.WECalText()
+		self.ted.WEResetModCount()
+		w.DrawGrowIcon()
+		self.scrollbars()
+		self.do_postopen()
+		self.do_activate(1, None)
+		
+	def getstylesoup(self):
+		if not self.path:
+			return None, None
+		oldrf = Res.CurResFile()
+		try:
+			rf = Res.OpenResFile(self.path)
+		except Res.Error:
+			return None, None
+		try:
+			hstyle = Res.Get1Resource('styl', 128)
+			hstyle.DetachResource()
+		except Res.Error:
+			hstyle = None
+		try:
+			hsoup = Res.Get1Resource('SOUP', 128)
+			hsoup.DetachResource()
+		except Res.Error:
+			hsoup = None
+		Res.CloseResFile(rf)
+		Res.UseResFile(oldrf)
+		return hstyle, hsoup
+				
+	def do_idle(self, event):
+		(what, message, when, where, modifiers) = event
+		Qd.SetPort(self.wid)
+		self.ted.WEIdle()	
+		if self.ted.WEAdjustCursor(where, BIGREGION):
+			return
+		Qd.SetCursor(Qd.qd.arrow)
+		
+	def getscrollbarvalues(self):
+		dr = self.ted.WEGetDestRect()
+		vr = self.ted.WEGetViewRect()
+		vx = self.scalebarvalue(dr[0], dr[2], vr[0], vr[2])
+		vy = self.scalebarvalue(dr[1], dr[3], vr[1], vr[3])
+		return vx, vy
+		
+	def scrollbar_callback(self, which, what, value):
+		if which == 'y':
+			if what == 'set':
+				height = self.ted.WEGetHeight(0, 0x3fffffff)
+				cur = self.getscrollbarvalues()[1]
+				delta = (cur-value)*height/32767
+			if what == '-':
+				topline_off,dummy = self.ted.WEGetOffset((1,1))
+				topline_num = self.ted.WEOffsetToLine(topline_off)
+				delta = self.ted.WEGetHeight(topline_num, topline_num+1)
+			elif what == '--':
+				delta = (self.ted.WEGetViewRect()[3]-10)
+				if delta <= 0:
+					delta = 10 # Random value
+			elif what == '+':
+				# XXXX Wrong: should be bottom line size
+				topline_off,dummy = self.ted.WEGetOffset((1,1))
+				topline_num = self.ted.WEOffsetToLine(topline_off)
+				delta = -self.ted.WEGetHeight(topline_num, topline_num+1)
+			elif what == '++':
+				delta = -(self.ted.WEGetViewRect()[3]-10)
+				if delta >= 0:
+					delta = -10
+			self.ted.WEScroll(0, delta)
+		else:
+			if what == 'set':
+				return # XXXX
+			vr = self.ted.WEGetViewRect()
+			winwidth = vr[2]-vr[0]
+			if what == '-':
+				delta = winwidth/10
+			elif what == '--':
+				delta = winwidth/2
+			elif what == '+':
+				delta = -winwidth/10
+			elif what == '++':
+				delta = -winwidth/2
+			self.ted.WEScroll(delta, 0)
+		# Pin the scroll
+		l, t, r, b = self.ted.WEGetDestRect()
+		vl, vt, vr, vb = self.ted.WEGetViewRect()
+		if t > 0 or l > 0:
+			dx = dy = 0
+			if t > 0: dy = -t
+			if l > 0: dx = -l
+			self.ted.WEScroll(dx, dy)
+		elif b < vb:
+			self.ted.WEScroll(0, b-vb)
+
+		
+	def do_activate(self, onoff, evt):
+		Qd.SetPort(self.wid)
+		ScrolledWindow.do_activate(self, onoff, evt)
+		if onoff:
+			self.ted.WEActivate()
+			self.parent.active = self
+			self.parent.updatemenubar()
+		else:
+			self.ted.WEDeactivate()
+
+	def do_update(self, wid, event):
+		region = wid.GetWindowPort().visRgn
+		if Qd.EmptyRgn(region):
+			return
+		Qd.EraseRgn(region)
+		self.ted.WEUpdate(region)
+		self.updatescrollbars()
+		
+	def do_postresize(self, width, height, window):
+		l, t, r, b = self.ted.WEGetViewRect()
+		vr = (l, t, l+width-15, t+height-15)
+		self.ted.WESetViewRect(vr)
+		Win.InvalRect(vr)
+		ScrolledWindow.do_postresize(self, width, height, window)
+		
+	def do_contentclick(self, local, modifiers, evt):
+		(what, message, when, where, modifiers) = evt
+		self.ted.WEClick(local, modifiers, when)
+		self.updatescrollbars()
+		self.parent.updatemenubar()
+
+	def do_char(self, ch, event):
+		self.ted.WESelView()
+		(what, message, when, where, modifiers) = event
+		self.ted.WEKey(ord(ch), modifiers)
+		self.updatescrollbars()
+		self.parent.updatemenubar()
+		
+	def close(self):
+		if self.ted.WEGetModCount():
+			save = EasyDialogs.AskYesNoCancel('Save window "%s" before closing?'%self.name, 1)
+			if save > 0:
+				self.menu_save()
+			elif save < 0:
+				return
+		if self.parent.active == self:
+			self.parent.active = None
+		self.parent.updatemenubar()
+		del self.ted
+		self.do_postclose()
+		
+	def menu_save(self):
+		if not self.path:
+			self.menu_save_as()
+			return # Will call us recursively
+		#
+		# First save data
+		#
+		dhandle = self.ted.WEGetText()
+		data = dhandle.data
+		fp = open(self.path, 'wb')  # NOTE: wb, because data has CR for end-of-line
+		fp.write(data)
+		if data[-1] <> '\r': fp.write('\r')
+		fp.close()
+		#
+		# Now save style and soup
+		#
+		oldresfile = Res.CurResFile()
+		try:
+			rf = Res.OpenResFile(self.path)
+		except Res.Error:
+			Res.CreateResFile(self.path)
+			rf = Res.OpenResFile(self.path)
+		styles = Res.Resource('')
+		soup = Res.Resource('')
+		self.ted.WECopyRange(0, 0x3fffffff, None, styles, soup)
+		styles.AddResource('styl', 128, '')
+		soup.AddResource('SOUP', 128, '')
+		Res.CloseResFile(rf)
+		Res.UseResFile(oldresfile)
+		
+		self.ted.WEResetModCount()
+		
+	def menu_save_as(self):
+		fss, ok = macfs.StandardPutFile('Save as:')
+		if not ok: return
+		self.path = fss.as_pathname()
+		self.name = os.path.split(self.path)[-1]
+		self.wid.SetWTitle(self.name)
+		self.menu_save()
+		
+	def menu_cut(self):
+		self.ted.WESelView()
+		self.ted.WECut()
+		Scrap.ZeroScrap()
+		self.ted.WECut()
+		self.updatescrollbars()
+		self.parent.updatemenubar()
+		
+	def menu_copy(self):
+		Scrap.ZeroScrap()
+		self.ted.WECopy()
+		self.updatescrollbars()
+		self.parent.updatemenubar()
+		
+	def menu_paste(self):
+		self.ted.WESelView()
+		self.ted.WEPaste()
+		self.updatescrollbars()
+		self.parent.updatemenubar()
+		
+	def menu_clear(self):
+		self.ted.WESelView()
+		self.ted.WEDelete()
+		self.updatescrollbars()
+		self.parent.updatemenubar()
+
+	def menu_undo(self):
+		self.ted.WEUndo()
+		self.updatescrollbars()
+		self.parent.updatemenubar()
+		
+	def menu_setfont(self, font):
+		font = Fm.GetFNum(font)
+		self.mysetstyle(WASTEconst.weDoFont, (font, 0, 0, (0,0,0)))
+		self.parent.updatemenubar()
+				
+	def menu_modface(self, face):
+		self.mysetstyle(WASTEconst.weDoFace|WASTEconst.weDoToggleFace, 
+			(0, face, 0, (0,0,0)))
+
+	def menu_setface(self, face):
+		self.mysetstyle(WASTEconst.weDoFace|WASTEconst.weDoReplaceFace, 
+			(0, face, 0, (0,0,0)))
+
+	def menu_setsize(self, size):
+		self.mysetstyle(WASTEconst.weDoSize, (0, 0, size, (0,0,0)))
+								
+	def menu_incsize(self, size):
+		self.mysetstyle(WASTEconst.weDoAddSize, (0, 0, size, (0,0,0)))
+
+	def mysetstyle(self, which, how):
+		self.ted.WESelView()
+		self.ted.WESetStyle(which, how)
+		self.parent.updatemenubar()
+								
+	def have_selection(self):
+		start, stop = self.ted.WEGetSelection()
+		return start < stop
+		
+	def can_paste(self):
+		return self.ted.WECanPaste()
+		
+	def can_undo(self):
+		which, redo = self.ted.WEGetUndoInfo()
+		which = UNDOLABELS[which]
+		if which == None: return None
+		if redo:
+			return "Redo "+which
+		else:
+			return "Undo "+which
+			
+	def getruninfo(self):
+		all = (WASTEconst.weDoFont | WASTEconst.weDoFace | WASTEconst.weDoSize)
+		dummy, mode, (font, face, size, color) = self.ted.WEContinuousStyle(all)
+		if not (mode & WASTEconst.weDoFont):
+			font = None
+		else:
+			font = Fm.GetFontName(font)
+		if not (mode & WASTEconst.weDoFace): fact = None
+		if not (mode & WASTEconst.weDoSize): size = None
+		return font, face, size
+
+class Wed(Application):
+	def __init__(self):
+		Application.__init__(self)
+		self.num = 0
+		self.active = None
+		self.updatemenubar()
+		
+	def makeusermenus(self):
+		self.filemenu = m = Menu(self.menubar, "File")
+		self.newitem = MenuItem(m, "New window", "N", self.open)
+		self.openitem = MenuItem(m, "Open...", "O", self.openfile)
+		self.closeitem = MenuItem(m, "Close", "W", self.closewin)
+		m.addseparator()
+		self.saveitem = MenuItem(m, "Save", "S", self.save)
+		self.saveasitem = MenuItem(m, "Save as...", "", self.saveas)
+		m.addseparator()
+		self.quititem = MenuItem(m, "Quit", "Q", self.quit)
+		
+		self.editmenu = m = Menu(self.menubar, "Edit")
+		self.undoitem = MenuItem(m, "Undo", "Z", self.undo)
+		self.cutitem = MenuItem(m, "Cut", "X", self.cut)
+		self.copyitem = MenuItem(m, "Copy", "C", self.copy)
+		self.pasteitem = MenuItem(m, "Paste", "V", self.paste)
+		self.clearitem = MenuItem(m, "Clear", "", self.clear)
+		
+		self.makefontmenu()
+		
+		# Groups of items enabled together:
+		self.windowgroup = [self.closeitem, self.saveitem, self.saveasitem,
+			self.editmenu, self.fontmenu, self.facemenu, self.sizemenu]
+		self.focusgroup = [self.cutitem, self.copyitem, self.clearitem]
+		self.windowgroup_on = -1
+		self.focusgroup_on = -1
+		self.pastegroup_on = -1
+		self.undo_label = "never"
+		self.ffs_values = ()
+		
+	def makefontmenu(self):
+		self.fontmenu = Menu(self.menubar, "Font")
+		self.fontnames = getfontnames()
+		self.fontitems = []
+		for n in self.fontnames:
+			m = MenuItem(self.fontmenu, n, "", self.selfont)
+			self.fontitems.append(m)
+		self.facemenu = Menu(self.menubar, "Style")
+		self.faceitems = []
+		for n, shortcut in STYLES:
+			m = MenuItem(self.facemenu, n, shortcut, self.selface)
+			self.faceitems.append(m)
+		self.facemenu.addseparator()
+		self.faceitem_normal = MenuItem(self.facemenu, "Normal", "N", 
+			self.selfacenormal)
+		self.sizemenu = Menu(self.menubar, "Size")
+		self.sizeitems = []
+		for n in SIZES:
+			m = MenuItem(self.sizemenu, `n`, "", self.selsize)
+			self.sizeitems.append(m)
+		self.sizemenu.addseparator()
+		self.sizeitem_bigger = MenuItem(self.sizemenu, "Bigger", "+", 
+			self.selsizebigger)
+		self.sizeitem_smaller = MenuItem(self.sizemenu, "Smaller", "-", 
+			self.selsizesmaller)
+					
+	def selfont(self, id, item, *rest):
+		if self.active:
+			font = self.fontnames[item-1]
+			self.active.menu_setfont(font)
+		else:
+			EasyDialogs.Message("No active window?")
+
+	def selface(self, id, item, *rest):
+		if self.active:
+			face = (1<<(item-1))
+			self.active.menu_modface(face)
+		else:
+			EasyDialogs.Message("No active window?")
+
+	def selfacenormal(self, *rest):
+		if self.active:
+			self.active.menu_setface(0)
+		else:
+			EasyDialogs.Message("No active window?")
+
+	def selsize(self, id, item, *rest):
+		if self.active:
+			size = SIZES[item-1]
+			self.active.menu_setsize(size)
+		else:
+			EasyDialogs.Message("No active window?")
+
+	def selsizebigger(self, *rest):
+		if self.active:
+			self.active.menu_incsize(2)
+		else:
+			EasyDialogs.Message("No active window?")
+
+	def selsizesmaller(self, *rest):
+		if self.active:
+			self.active.menu_incsize(-2)
+		else:
+			EasyDialogs.Message("No active window?")
+
+	def updatemenubar(self):
+		changed = 0
+		on = (self.active <> None)
+		if on <> self.windowgroup_on:
+			for m in self.windowgroup:
+				m.enable(on)
+			self.windowgroup_on = on
+			changed = 1
+		if on:
+			# only if we have an edit menu
+			on = self.active.have_selection()
+			if on <> self.focusgroup_on:
+				for m in self.focusgroup:
+					m.enable(on)
+				self.focusgroup_on = on
+				changed = 1
+			on = self.active.can_paste()
+			if on <> self.pastegroup_on:
+				self.pasteitem.enable(on)
+				self.pastegroup_on = on
+				changed = 1
+			on = self.active.can_undo()
+			if on <> self.undo_label:
+				if on:
+					self.undoitem.enable(1)
+					self.undoitem.settext(on)
+					self.undo_label = on
+				else:
+					self.undoitem.settext("Nothing to undo")
+					self.undoitem.enable(0)
+				changed = 1
+			if self.updatefontmenus():
+				changed = 1
+		if changed:
+			DrawMenuBar()
+			
+	def updatefontmenus(self):
+		info = self.active.getruninfo()
+		if info == self.ffs_values:
+			return 0
+		# Remove old checkmarks
+		if self.ffs_values == ():
+			self.ffs_values = (None, None, None)
+		font, face, size = self.ffs_values
+		if font <> None:
+			fnum = self.fontnames.index(font)
+			self.fontitems[fnum].check(0)
+		if face <> None:
+			for i in range(len(self.faceitems)):
+				if face & (1<<i):
+					self.faceitems[i].check(0)
+		if size <> None:
+			for i in range(len(self.sizeitems)):
+				if SIZES[i] == size:
+					self.sizeitems[i].check(0)
+				
+		self.ffs_values = info
+		# Set new checkmarks
+		font, face, size = self.ffs_values
+		if font <> None:
+			fnum = self.fontnames.index(font)
+			self.fontitems[fnum].check(1)
+		if face <> None:
+			for i in range(len(self.faceitems)):
+				if face & (1<<i):
+					self.faceitems[i].check(1)
+		if size <> None:
+			for i in range(len(self.sizeitems)):
+				if SIZES[i] == size:
+					self.sizeitems[i].check(1)
+		# Set outline/normal for sizes
+		if font:
+			exists = getfontsizes(font, SIZES)
+			for i in range(len(self.sizeitems)):
+				if exists[i]:
+					self.sizeitems[i].setstyle(0)
+				else:
+					self.sizeitems[i].setstyle(8)
+
+	#
+	# Apple menu
+	#
+	
+	def do_about(self, id, item, window, event):
+		EasyDialogs.Message("A simple single-font text editor based on WASTE")
+			
+	#
+	# File menu
+	#
+
+	def open(self, *args):
+		self._open(0)
+		
+	def openfile(self, *args):
+		self._open(1)
+
+	def _open(self, askfile):
+		if askfile:
+			fss, ok = macfs.StandardGetFile('TEXT')
+			if not ok:
+				return
+			path = fss.as_pathname()
+			name = os.path.split(path)[-1]
+			try:
+				fp = open(path, 'rb') # NOTE binary, we need cr as end-of-line
+				data = fp.read()
+				fp.close()
+			except IOError, arg:
+				EasyDialogs.Message("IOERROR: "+`arg`)
+				return
+		else:
+			path = None
+			name = "Untitled %d"%self.num
+			data = ''
+		w = WasteWindow(self)
+		w.open(path, name, data)
+		self.num = self.num + 1
+		
+	def closewin(self, *args):
+		if self.active:
+			self.active.close()
+		else:
+			EasyDialogs.Message("No active window?")
+		
+	def save(self, *args):
+		if self.active:
+			self.active.menu_save()
+		else:
+			EasyDialogs.Message("No active window?")
+		
+	def saveas(self, *args):
+		if self.active:
+			self.active.menu_save_as()
+		else:
+			EasyDialogs.Message("No active window?")
+			
+		
+	def quit(self, *args):
+		for w in self._windows.values():
+			w.close()
+		if self._windows:
+			return
+		raise self
+		
+	#
+	# Edit menu
+	#
+	
+	def undo(self, *args):
+		if self.active:
+			self.active.menu_undo()
+		else:
+			EasyDialogs.Message("No active window?")
+		
+	def cut(self, *args):
+		if self.active:
+			self.active.menu_cut()
+		else:
+			EasyDialogs.Message("No active window?")
+		
+	def copy(self, *args):
+		if self.active:
+			self.active.menu_copy()
+		else:
+			EasyDialogs.Message("No active window?")
+		
+	def paste(self, *args):
+		if self.active:
+			self.active.menu_paste()
+		else:
+			EasyDialogs.Message("No active window?")
+
+	def clear(self, *args):
+		if self.active:
+			self.active.menu_clear()
+		else:
+			EasyDialogs.Message("No active window?")
+		
+	#
+	# Other stuff
+	#	
+
+	def idle(self, event):
+		if self.active:
+			self.active.do_idle(event)
+			
+def getfontnames():
+	names = []
+	for i in range(256):
+		n = Fm.GetFontName(i)
+		if n: names.append(n)
+	return names
+	
+def getfontsizes(name, sizes):
+	exist = []
+	num = Fm.GetFNum(name)
+	for sz in sizes:
+		if Fm.RealFont(num, sz):
+			exist.append(1)
+		else:
+			exist.append(0)
+	return exist
+
+def main():
+	App = Wed()
+	App.mainloop()
+	
+if __name__ == '__main__':
+	main()
+