.include "tgi-kernel.inc"
- .export _tgi_setcolor
+.proc _tgi_setcolor
-_tgi_setcolor = tgi_setcolor ; Call the driver
+ cmp _tgi_colorcount ; Compare to available colors
+ bcs @L1
+ sta _tgi_color ; Remember the drawing color
+ jmp tgi_setcolor ; Call the driver
+@L1: jmp tgi_inv_arg ; Invalid argument
+.endproc