mirror of
https://github.com/ksherlock/prez.git
synced 2024-11-02 23:06:54 +00:00
295 lines
6.7 KiB
Python
295 lines
6.7 KiB
Python
|
|
from base import *
|
|
from utils import *
|
|
import struct
|
|
|
|
__all__ = [
|
|
'rMenuBar',
|
|
'rMenu',
|
|
'rMenuItem',
|
|
|
|
'UndoMenuItem',
|
|
'CutMenuItem',
|
|
'CopyMenuItem',
|
|
'PasteMenuItem',
|
|
'ClearMenuItem',
|
|
'CloseMenuItem',
|
|
'DividerMenuItem'
|
|
]
|
|
|
|
# see:
|
|
# Programmer's Reference for System 6 (Ch 13 Menu Manager Update, pg 103+)
|
|
# TB Vol 3 Chapter 37
|
|
# TB Vol 1 Chapter 13
|
|
|
|
# TODO - menu item icon support (system 6) rItemStruct
|
|
|
|
# A menu ID must be unique for each menu; that is, no two menus
|
|
# can have the same ID or the system will fall. Similarly, no two
|
|
# items can have the same Item ID.
|
|
|
|
_menu_ids = {}
|
|
_menu_item_ids = {}
|
|
|
|
|
|
|
|
def _to_char_string(x):
|
|
if not x: return '""'
|
|
if x == 0x0d: return "\\n" # intentionally backwards
|
|
if x == 0x0a: return "\\r" #
|
|
if chr(x) in "\\\\'" : return "\\" + x
|
|
if x >= 32 and x < 0x7e: return '"' + chr(x) + '"'
|
|
return "\\${:02x}".format(x)
|
|
|
|
class rMenuBar(rObject):
|
|
rName = "rMenuBar"
|
|
rType = 0x8008
|
|
|
|
def __init__(self, *children, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.children = children[:]
|
|
for x in children:
|
|
if not isinstance(x, rMenu):
|
|
raise TypeError("bad type: {}".format(type(x)))
|
|
|
|
def __bytes__(self):
|
|
bb = struct.pack("<HH", 0, 0x8000) # version, resIDs
|
|
for x in self.children:
|
|
bb += struct.pack("<I", x.get_id())
|
|
bb += b"\x00\x00\x00\x00" # 0-terminate
|
|
return bb
|
|
|
|
def _rez_string(self):
|
|
|
|
rv = "\t{\n"
|
|
kiddos = [x.get_id() for x in self.children]
|
|
rv += ",\n".join(["\t\t0x{:08x}".format(x) for x in kiddos])
|
|
if kiddos: rv += "\n"
|
|
rv += "\t}"
|
|
return rv
|
|
|
|
|
|
# valid menu ids: $0001-$fffe
|
|
class rMenu(rObject):
|
|
rName = "rMenu"
|
|
rType = 0x8009
|
|
# rChildType = rMenuItem
|
|
rRange = range(0x0001,0xffff)
|
|
|
|
# /*-------------------------------------------------------*/
|
|
# /* Equates for Menu Flags
|
|
# /*-------------------------------------------------------*/
|
|
# #Define rmAllowCache $0008
|
|
# #Define rmCustom $0010
|
|
# #Define rmNo_Xor $0020
|
|
# #Define rmDisabled $0080
|
|
|
|
#flags = all off = a080 (disabled)
|
|
|
|
def __init__(self, title, *children,
|
|
flags=0x0000, menuID=None,
|
|
**kwargs
|
|
):
|
|
super().__init__(**kwargs)
|
|
self.title = rPString.make_string(title)
|
|
self.children = children[:]
|
|
self.menuID = menuID
|
|
|
|
|
|
flags |= 0xa000 # title ref is resource, menus are resources
|
|
if kwargs.get("allowCache"): flags |= 0x0008
|
|
if kwargs.get("custom"): flags |= 0x0010
|
|
if kwargs.get("xor"): flags |= 0x0020
|
|
if kwargs.get("disabled"): flags |= 0x0080
|
|
if kwargs.get("mChoose"): flags |= 0x0100
|
|
|
|
self.flags = flags
|
|
|
|
for x in children:
|
|
if not isinstance(x, rMenuItem):
|
|
raise TypeError("bad type: {}".format(type(x)))
|
|
|
|
def __bytes__(self):
|
|
menuID = self.menuID
|
|
if menuID == None: menuID = self.get_id()
|
|
bb = struct.pack("<HHHI", 0,
|
|
menuID, self.flags, self.title.get_id()
|
|
)
|
|
for x in self.children:
|
|
bb += struct.pack("<I", x.get_id())
|
|
bb += b"\x00\x00\x00\x00" # 0-terminate
|
|
return bb
|
|
|
|
def _rez_string(self):
|
|
|
|
menuID = self.menuID
|
|
if menuID == None: menuID = self.get_id()
|
|
rv = (
|
|
"\t0x{:04x}, /* menu ID */\n"
|
|
"\t0x{:04x}, /* flags */\n"
|
|
"\t0x{:08x}, /* title ref (rPString) */\n"
|
|
).format(menuID, self.flags, self.title.get_id())
|
|
rv += "\t{\n"
|
|
kiddos = [x.get_id() for x in self.children]
|
|
rv += ",\n".join(["\t\t0x{:08x}".format(x) for x in kiddos])
|
|
if kiddos: rv += "\n"
|
|
rv += "\t}"
|
|
return rv
|
|
|
|
|
|
miUndo = 0xfa
|
|
miCut = 0xfb
|
|
miCopy = 0xfc
|
|
miPaste = 0xfd
|
|
miClear = 0xfe
|
|
miClose = 0xff
|
|
|
|
|
|
# valid menu item ids: $0100-$fffe
|
|
|
|
class rMenuItem(rObject):
|
|
rName = "rMenuItem"
|
|
rType = 0x800a
|
|
rRange = range(0x100,0xffff)
|
|
|
|
# /* --------------------------------------------------*/
|
|
# /* flag word for menu item
|
|
# /* --------------------------------------------------*/
|
|
# #Define fBold $0001
|
|
# #Define fItalic $0002
|
|
# #Define fUnderline $0004
|
|
# #Define fXOR $0020
|
|
# #Define fDivider $0040
|
|
# #Define fDisabled $0080
|
|
# #Define fItemStruct $0400
|
|
# #Define fOutline $0800
|
|
# #Define fShadow $1000
|
|
# #define ItemStructRefShift $0100
|
|
# #Define ItemTitleRefShift $4000
|
|
|
|
|
|
def __init__(self, title, keys="", *,
|
|
checkMark=None, itemID=None, flags=0x0000,
|
|
**kwargs):
|
|
super().__init__(**kwargs)
|
|
|
|
self.title = rPString.make_string(title)
|
|
|
|
flags |= 0x8000 # title ref is resource
|
|
if kwargs.get("bold"): flags |= 0x0001
|
|
if kwargs.get("italic"): flags |= 0x0002
|
|
if kwargs.get("underline"): flags |= 0x0004
|
|
if kwargs.get("xor"): flags |= 0x0020
|
|
if kwargs.get("divider"): flags |= 0x0040
|
|
if kwargs.get("disabled"): flags |= 0x0080
|
|
if kwargs.get("outline"): flags |= 0x0800
|
|
if kwargs.get("shadow"): flags |= 0x1000
|
|
|
|
# bit 15 of flags set if title ref is actually an
|
|
# item struct ref.
|
|
|
|
self.flags = flags
|
|
|
|
self.itemChar = 0
|
|
self.altItemChar = 0
|
|
self.itemID = itemID
|
|
self.checkMark = 0
|
|
|
|
|
|
if checkMark:
|
|
if checkMark == True: checkMark = 0x12
|
|
else: checkMark = ord(str_to_bytes(checkMark))
|
|
self.checkMark = checkMark
|
|
|
|
if keys:
|
|
bb = str_to_bytes(keys)
|
|
if len(bb) >= 1:
|
|
self.itemChar = bb[0]
|
|
self.altItemChar = bb[0] # upper to lower?
|
|
if len(bb) >= 2:
|
|
self.altItemChar = bb[1]
|
|
if len(bb) > 2:
|
|
raise ValueError("keys too long: {}".format(keys))
|
|
|
|
|
|
def __bytes__(self):
|
|
itemID = self.itemID
|
|
if itemID == None: itemID = self.get_id()
|
|
bb = struct.pack("<HHBBHHI",
|
|
0, itemID,
|
|
self.itemChar, self.altItemChar,
|
|
self.checkMark, self.flags,
|
|
self.title.get_id()
|
|
)
|
|
|
|
return bb
|
|
|
|
def _rez_string(self):
|
|
|
|
itemID = self.itemID
|
|
if itemID == None: itemID = self.get_id()
|
|
|
|
return (
|
|
"\t0x{:04x}, /* id */\n"
|
|
"\t{}, {}, /* chars */\n"
|
|
"\t0x{:04x}, /* check */\n"
|
|
"\t0x{:04x}, /* flags */\n"
|
|
"\t0x{:04x} /* title ref (rPString) */"
|
|
).format(
|
|
itemID,
|
|
_to_char_string(self.itemChar),
|
|
_to_char_string(self.altItemChar),
|
|
self.checkMark,
|
|
self.flags,
|
|
self.title.get_id()
|
|
)
|
|
|
|
|
|
def _singleton(func):
|
|
value = None
|
|
def inner():
|
|
nonlocal value
|
|
if not value: value = func()
|
|
return value
|
|
return inner
|
|
|
|
|
|
def DividerMenuItem():
|
|
return rMenuItem("-", flags = 0x0080) # disabled
|
|
|
|
def EditMenu(*children, **kwargs):
|
|
return rMenu(" Edit ",
|
|
UndoMenuItem(),
|
|
CutMenuItem(),
|
|
CopyMenuItem(),
|
|
PasteMenuItem(),
|
|
ClearMenuItem(),
|
|
*children,
|
|
**kwargs
|
|
)
|
|
|
|
@ _singleton
|
|
def UndoMenuItem():
|
|
# normally underlined....
|
|
return rMenuItem("Undo", "Zz", itemID = miUndo)
|
|
|
|
@ _singleton
|
|
def CutMenuItem():
|
|
return rMenuItem("Cut", "Xx", itemID = miCut)
|
|
|
|
@ _singleton
|
|
def CopyMenuItem():
|
|
return rMenuItem("Copy", "Cc", itemID = miCopy)
|
|
|
|
@ _singleton
|
|
def PasteMenuItem():
|
|
return rMenuItem("Paste", "Vv", itemID = miPaste)
|
|
|
|
@ _singleton
|
|
def ClearMenuItem():
|
|
return rMenuItem("Clear", "", itemID = miClear)
|
|
|
|
@ _singleton
|
|
def CloseMenuItem():
|
|
return rMenuItem("Close", "Ww", itemID = miClose)
|