From 4fbe8d2d27dd8873c1f309e15b116f8800a5e605 Mon Sep 17 00:00:00 2001 From: Jan Tuomi Date: Fri, 3 Nov 2023 20:02:03 +0200 Subject: Improve string I/O helper macros --- asm/ext_string_io.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'asm/ext_string_io.py') diff --git a/asm/ext_string_io.py b/asm/ext_string_io.py index cf7d598..7b026ba 100644 --- a/asm/ext_string_io.py +++ b/asm/ext_string_io.py @@ -13,6 +13,7 @@ def get_unique_name(prefix: str): return ret def expand_put_char(cursor_reg: str, scratch_reg: str, c: str) -> ExpandResult: + c = c.strip("\"") return [ ["ldi", str(to_char_code(c)), scratch_reg], ["str", scratch_reg, cursor_reg], @@ -21,6 +22,7 @@ def expand_put_char(cursor_reg: str, scratch_reg: str, c: str) -> ExpandResult: def expand_put_string(cursor_reg: str, scratch_reg: str, *rest: str) -> ExpandResult: string = " ".join(rest) + string = string.strip("\"") result: ExpandResult = [] for c in string: result += expand_put_char(cursor_reg, scratch_reg, c) -- cgit v1.3