prez/utils.py

# __all__ = ("str_to_bytes", "make_string", "format_rect", "format_point", "format_size")

# decorators
def export_enum(cls):
	global __all__

	members = cls.__members__
	globals().update(members)
	if __all__ != None: __all__.extend(list(members))
	return cls	


# helper functions
def str_to_bytes(text):
	if isinstance(text, str): return text.encode("macroman")
	if isinstance(text, bytes): return text
	if isinstance(text, bytearray): return bytes(text)
	raise TypeError("Bad text type: {}".format(type(text)))



def format_rect(x):
	return "{{ {:d}, {:d}, {:d}, {:d} }}".format(*x)

def format_point(x):
	return "{{ {:d}, {:d} }}".format(*x)

def format_size(x):
	return "{{ {:d}, {:d} }}".format(*x)


def _generate_map():
	map = { x: "\\${:02x}".format(x) for x in range(0, 256) if x < 32 or x > 126 }
	map[0x0d] = "\\r" # intentionally backwards. -- no more
	map[0x0a] = "\\n" # intentionally backwards. -- no more
	map[0x09] = "\\t"
	# \b \f \v \? also supported. 
	map[ord('"')] = '\\"'
	map[ord("'")] = "\\'"
	map[ord("\\")] = "\\\\"
	# map[0x7f] = "\\?" # rubout

	return map

_map = _generate_map()

def format_string(bstring, quote=True):
	s = "".join([_map[x] if x in _map else chr(x) for x in bstring])
	if quote: return '"' + s + '"'
	return s

# rez "char" is a string of length 1.
# 'x' is character literal which is a number.
def format_char(x):
	if x == 0: return '""'
	c = _map.get(x, chr(x))
	return '"' + c + '"'

def multi_format_string(bstring, indent=""):
	q = '"'

	rv = []
	tmp = [indent + q]
	for x in bstring:
		if x in _map: tmp.append(_map[x])
		else: tmp.append(chr(x))
		if len(tmp) > 32 or x == 0x0a:
			tmp.append(q)
			rv.append("".join(tmp))
			tmp = [indent + q]
	if len(tmp)>1:
		tmp.append(q)
		rv.append("".join(tmp))
	return ("\n").join(rv)