Merge pull request espressif#114 from adafruit/rp2040-uf2

Allow .generate for rp2040 targets

Author: ladyada

Diff for: .github/workflows/githubci.yml

@@ -16,5 +16,5 @@ jobs:
       run: bash ./actions_install.sh
     - name: test platforms
       run: |
-        python3 build_platform.py uno leonardo mega2560 zero esp8266 esp32 pico_rp2040
+        python3 build_platform.py uno leonardo mega2560 zero esp8266 esp32 pico_rp2040 feather_m4_express
 

Diff for: build_platform.py

@@ -3,9 +3,9 @@
 import time
 import os
 import shutil
-from distutils.dir_util import copy_tree
 import subprocess
 import collections
+from contextlib import contextmanager
 
 # optional wall option cause build failed if has warnings
 BUILD_WALL = False
@@ -248,6 +248,12 @@ def run_or_die(cmd, error):
 
 ################################ UF2 Utils.
 
+def glob01(pattern):
+    result = glob.glob(pattern)
+    if len(result) > 1:
+        raise RuntimeError(f"Required pattern {pattern} to match at most 1 file, got {result}")
+    return result[0] if result else None
+
 def glob1(pattern):
     result = glob.glob(pattern)
     if len(result) != 1:
@@ -276,11 +282,23 @@ def generate_uf2(example_path):
     """
     if not download_uf2_utils():
         return None
-    cli_build_path = "build/*.*." + fqbn.split(':')[2] + "/*.hex"
-    input_file = glob1(os.path.join(example_path, cli_build_path))
-    output_file = os.path.splitext(input_file)[0] + ".uf2"
+
+    cli_build_uf2_path = "build/*.*." + fqbn.split(':')[2] + "/*.uf2"
+    uf2_input_file = glob01(os.path.join(example_path, cli_build_uf2_path))
+
+    # Some platforms, like rp2040, directly generate a uf2 file, so no need to do it ourselves
+    if uf2_input_file is not None:
+        output_file = os.path.splitext(uf2_input_file)[0] + ".uf2"
+        ColorPrint.print_pass(CHECK)
+        ColorPrint.print_info("Used uf2 generated by arduino-cli")
+        return output_file
+
+    # For other uf2-supporting platforms, we can generate it from a hex file
+    cli_build_hex_path = "build/*.*." + fqbn.split(':')[2] + "/*.hex"
+    hex_input_file = glob1(os.path.join(example_path, cli_build_hex_path))
+    output_file = os.path.splitext(hex_input_file)[0] + ".uf2"
     family_id = ALL_PLATFORMS[platform][1]
-    cmd = ['python3', 'uf2conv.py', input_file, '-c', '-f', family_id, '-o', output_file]
+    cmd = ['python3', 'uf2conv.py', hex_input_file, '-c', '-f', family_id, '-o', output_file]
     proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     r = proc.wait(timeout=60)
     out = proc.stdout.read()
@@ -311,6 +329,20 @@ def generate_uf2(example_path):
         print("Unknown platform: ", arg)
         exit(-1)
 
+@contextmanager
+def group_output(title):
+    sys.stdout.flush()
+    sys.stderr.flush()
+    print(f"::group::{title}")
+    try:
+        yield
+    finally:
+        sys.stdout.flush()
+        sys.stderr.flush()
+        print(f"::endgroup::")
+        sys.stdout.flush()
+
+
 def test_examples_in_folder(folderpath):
     global success
     for example in sorted(os.listdir(folderpath)):
@@ -366,7 +398,8 @@ def test_examples_in_folder(folderpath):
             ColorPrint.print_pass(CHECK)
             if err:
                 # also print out warning message
-                ColorPrint.print_fail(err.decode("utf-8"))
+                with group_output(f"{example} {fqbn} build output"):
+                    ColorPrint.print_fail(err.decode("utf-8"))
             if os.path.exists(gen_file_name):
                 if ALL_PLATFORMS[platform][1] == None:
                     ColorPrint.print_info("Platform does not support UF2 files, skipping...")
@@ -383,8 +416,9 @@ def test_examples_in_folder(folderpath):
                         os.system("ls -lR "+BUILD_DIR+"/build")
         else:
             ColorPrint.print_fail(CROSS)
-            ColorPrint.print_fail(out.decode("utf-8"))
-            ColorPrint.print_fail(err.decode("utf-8"))
+            with group_output(f"{example} {fqbn} built output"):
+                ColorPrint.print_fail(out.decode("utf-8"))
+                ColorPrint.print_fail(err.decode("utf-8"))
             success = 1
 
 def test_examples_in_learningrepo(folderpath):