Add ASSERT macro + fix label substractions banks + add . for current address

1 files changed, 25 insertions, 0 deletions

diff --git a/macros.go b/macros.go
index 2324a14..44b72a2 100644
--- a/macros.go
+++ b/macros.go
@@ -43,6 +43,31 @@ func NewInstructionSetMacros() InstructionSet {
 		},
 	}
 
+	result[".ASSERT"] = []InstructionParams{
+		{
+			Types: []ParamType{Raw8, Raw8},
+			Assembler: func(currentAddress uint32, args []uint32) ([]byte, error) {
+				if args[0] != args[1] {
+					return nil, fmt.Errorf("ASSERTION FAILED, %v != %v", args[0], args[1])
+				}
+				return []byte{}, nil
+			},
+			MacroForbidden: false,
+			SkipFirstPass:  true,
+		},
+		{
+			Types: []ParamType{Raw16, Raw16},
+			Assembler: func(currentAddress uint32, args []uint32) ([]byte, error) {
+				if args[0] != args[1] {
+					return nil, fmt.Errorf("ASSERTION FAILED, %v != %v", args[0], args[1])
+				}
+				return []byte{}, nil
+			},
+			MacroForbidden: false,
+			SkipFirstPass:  true,
+		},
+	}
+
 	result[".DB"] = []InstructionParams{
 		{
 			Types: []ParamType{Raw8},