cpython/Mac/Demo/textedit/ped.py

# A minimal text editor.
#
# To be done:
# - Update viewrect after resize
# - Handle horizontal scrollbar correctly
# - Functionality: find, etc.

from Carbon.Menu import DrawMenuBar
from FrameWork import *
from Carbon import Win
from Carbon import Qd
from Carbon import TE
from Carbon import Scrap
import os
import macfs

class TEWindow(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, 0)
        self.wid = w
        x0, y0, x1, y1 = self.wid.GetWindowPort().GetPortBounds()
        x0 = x0 + 4
        y0 = y0 + 4
        x1 = x1 - 20
        y1 = y1 - 20
        vr = dr = x0, y0, x1, y1
        ##vr = 4, 0, r[2]-r[0]-15, r[3]-r[1]-15
        ##dr = (0, 0, vr[2], 0)
        Qd.SetPort(w)
        Qd.TextFont(4)
        Qd.TextSize(9)
        self.ted = TE.TENew(dr, vr)
        self.ted.TEAutoView(1)
        self.ted.TESetText(data)
        w.DrawGrowIcon()
        self.scrollbars()
        self.changed = 0
        self.do_postopen()
        self.do_activate(1, None)

    def do_idle(self):
        self.ted.TEIdle()

    def getscrollbarvalues(self):
        dr = self.ted.destRect
        vr = self.ted.viewRect
        height = self.ted.nLines * self.ted.lineHeight
        vx = self.scalebarvalue(dr[0], dr[2]-dr[0], vr[0], vr[2])
        vy = self.scalebarvalue(dr[1], dr[1]+height, vr[1], vr[3])
        print dr, vr, height, vx, vy
        return None, vy

    def scrollbar_callback(self, which, what, value):
        if which == 'y':
            if what == 'set':
                height = self.ted.nLines * self.ted.lineHeight
                cur = self.getscrollbarvalues()[1]
                delta = (cur-value)*height/32767
            if what == '-':
                delta = self.ted.lineHeight
            elif what == '--':
                delta = (self.ted.viewRect[3]-self.ted.lineHeight)
                if delta <= 0:
                    delta = self.ted.lineHeight
            elif what == '+':
                delta = -self.ted.lineHeight
            elif what == '++':
                delta = -(self.ted.viewRect[3]-self.ted.lineHeight)
                if delta >= 0:
                    delta = -self.ted.lineHeight
            self.ted.TEPinScroll(0, delta)
            print 'SCROLL Y', delta
        else:
            pass # No horizontal scrolling

    def do_activate(self, onoff, evt):
        print "ACTIVATE", onoff
        ScrolledWindow.do_activate(self, onoff, evt)
        if onoff:
            self.ted.TEActivate()
            self.parent.active = self
            self.parent.updatemenubar()
        else:
            self.ted.TEDeactivate()

    def do_update(self, wid, event):
        Qd.EraseRect(wid.GetWindowPort().GetPortBounds())
        self.ted.TEUpdate(wid.GetWindowPort().GetPortBounds())
        self.updatescrollbars()

    def do_contentclick(self, local, modifiers, evt):
        shifted = (modifiers & 0x200)
        self.ted.TEClick(local, shifted)
        self.updatescrollbars()
        self.parent.updatemenubar()

    def do_char(self, ch, event):
        self.ted.TESelView()
        self.ted.TEKey(ord(ch))
        self.changed = 1
        self.updatescrollbars()
        self.parent.updatemenubar()

    def close(self):
        if self.changed:
            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
        print 'Saving to ', self.path
        dhandle = self.ted.TEGetText()
        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()
        self.changed = 0

    def menu_save_as(self):
        path = EasyDialogs.AskFileForSave(message='Save as:')
        if not path: return
        self.path = path
        self.name = os.path.split(self.path)[-1]
        self.wid.SetWTitle(self.name)
        self.menu_save()

    def menu_cut(self):
        self.ted.TESelView()
        self.ted.TECut()
        if hasattr(Scrap, 'ZeroScrap'):
            Scrap.ZeroScrap()
        else:
            Scrap.ClearCurrentScrap()
        TE.TEToScrap()
        self.updatescrollbars()
        self.parent.updatemenubar()
        self.changed = 1

    def menu_copy(self):
        self.ted.TECopy()
        if hasattr(Scrap, 'ZeroScrap'):
            Scrap.ZeroScrap()
        else:
            Scrap.ClearCurrentScrap()
        TE.TEToScrap()
        self.updatescrollbars()
        self.parent.updatemenubar()

    def menu_paste(self):
        TE.TEFromScrap()
        self.ted.TESelView()
        self.ted.TEPaste()
        self.updatescrollbars()
        self.parent.updatemenubar()
        self.changed = 1

    def menu_clear(self):
        self.ted.TESelView()
        self.ted.TEDelete()
        self.updatescrollbars()
        self.parent.updatemenubar()
        self.changed = 1

    def have_selection(self):
        return (self.ted.selStart < self.ted.selEnd)

class Ped(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)

        # Not yet implemented:
        self.undoitem.enable(0)

        # Groups of items enabled together:
        self.windowgroup = [self.closeitem, self.saveitem, self.saveasitem, self.editmenu]
        self.focusgroup = [self.cutitem, self.copyitem, self.clearitem]
        self.windowgroup_on = -1
        self.focusgroup_on = -1
        self.pastegroup_on = -1

    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
            if hasattr(Scrap, 'InfoScrap'):
                on = (Scrap.InfoScrap()[0] <> 0)
            else:
                flavors = Scrap.GetCurrentScrap().GetScrapFlavorInfoList()
                for tp, info in flavors:
                    if tp == 'TEXT':
                        on = 1
                        break
                else:
                    on = 0
            if on <> self.pastegroup_on:
                self.pasteitem.enable(on)
                self.pastegroup_on = on
                changed = 1
        if changed:
            DrawMenuBar()

    #
    # Apple menu
    #

    def do_about(self, id, item, window, event):
        EasyDialogs.Message("A simple single-font text editor")

    #
    # File menu
    #

    def open(self, *args):
        self._open(0)

    def openfile(self, *args):
        self._open(1)

    def _open(self, askfile):
        if askfile:
            path = EasyDialogs.AskFileForOpen(typeList=('TEXT',))
            if not path:
                return
            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: %r" % (arg,))
                return
        else:
            path = None
            name = "Untitled %d"%self.num
            data = ''
        w = TEWindow(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
        self._quit()

    #
    # Edit menu
    #

    def undo(self, *args):
        pass

    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, *args):
        if self.active:
            self.active.do_idle()
        else:
            Qd.SetCursor(Qd.GetQDGlobalsArrow())

def main():
    App = Ped()
    App.mainloop()

if __name__ == '__main__':
    main()