Add support for the assembly instructions in @asm_pio functions

[DeflateAwning]

Here's an example of what the type checking looks like currently (observe all the errors) for @asm_pio-decorated Programmable IO assembly code blocks:

Doing a bit of digging around, I found that the PIOASMEmit class seems to define these functions, but there's no way to move those member functions to imports.

I think the best solution would be to make them global functions, probably. Or maybe there's a way to highjack the @asm_pio decorator so that it considers the assembly functions only in that function.

[Josverl]

Can you share the code,
which type-stubs you had installed ?

[DeflateAwning]

The code is here: https://gist.github.com/DeflateAwning/cfc26095d25390fcd3c619176c7bf23e

I was using these stubs: https://pypi.org/project/micropython-rp2-pico-stubs/

[Josverl]

I have already built it , but not (yet) included it as part of the rp2 stubs.

I plan to add this in the upcoming release , but it needed some significant restructuring

if you want it today - you need to add it manually to the stubs

• rename or delete ..../rp2.pyi
• create a rp2 folder
• copy in the new rp2 & PIO stubs

See for more details :
https://github.com/Josverl/PIO_ASM_typing

[DeflateAwning]

That's very neat, thank you very much! Looking forward to giving it a try.

In the final release, will we need to import anything? Or it'll work automatically?

[Josverl]

You'll need the somewhat ugly activation blurb

# -----------------------------------------------
# add type hints for the rp2.PIO Instructions
try: 
    from typing_extensions import TYPE_CHECKING # type: ignore
except ImportError:
    TYPE_CHECKING = False
if TYPE_CHECKING:
    from rp2.asm_pio import *
# -----------------------------------------------

before the first @asm_pio .
as there are a bunch of side effects that don't apply to pure python , I would not want to enable it by default,
and I have not found a way to have the type checkers enable it by default.

there is a cpython decorator @no_type_check , but that is not (yet) part of micropython ,
but can be loaded.
That suppresses errors, but offers no type-hints.
https://micropython-stubs.readthedocs.io/en/main/typing_mpy.html#using-the-no-type-check-decorator

[DeflateAwning]

This is perfect as-is! Thanks a lot!