Merge pull request #319 from hathach/add-esp32s2

port ESP32 S2

2 files changed, 119 insertions, 11 deletions

diff --git a/tools/build_all.py b/tools/build_all.py
index 0437bcaac..25cdc9778 100644
--- a/tools/build_all.py
+++ b/tools/build_all.py
@@ -11,6 +11,9 @@ exit_status = 0
 
 total_time = time.monotonic()
 
+build_format = '| {:23} | {:30} | {:9} | {:7} | {:6} | {:6} |'
+build_separator = '-' * 100
+
 # 1st Argument is Example, build all examples if not existed
 all_examples = []
 if len(sys.argv) > 1:
@@ -39,7 +42,8 @@ def build_example(example, board):
                           stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
 
 def build_size(example, board):
-    elf_file = 'examples/device/{}/_build/build-{}/{}-firmware.elf'.format(example, board, board)
+    #elf_file = 'examples/device/{}/_build/build-{}/{}-firmware.elf'.format(example, board, board)
+    elf_file = 'examples/device/{}/_build/build-{}/*.elf'.format(example, board)
     size_output = subprocess.run('size {}'.format(elf_file), shell=True, stdout=subprocess.PIPE).stdout.decode("utf-8")
     size_list = size_output.split('\n')[1].split('\t')
     flash_size = int(size_list[0])
@@ -49,16 +53,22 @@ def build_size(example, board):
 def skip_example(example, board):
     ex_dir = 'examples/device/' + example
     board_mk = 'hw/bsp/{}/board.mk'.format(board)
-    for skip_file in glob.iglob(ex_dir + '/.skip.MCU_*'):
-        mcu_cflag = '-DCFG_TUSB_MCU=OPT_' + os.path.basename(skip_file).split('.')[2]
-        with open(board_mk) as mk:
-            if mcu_cflag in mk.read():
+
+    with open(board_mk) as mk:
+        mk_contents = mk.read()
+
+        # Skip all ESP32-S2 board for CI
+        if 'CROSS_COMPILE = xtensa-esp32s2-elf-' in mk_contents:
+            return 1
+
+        # Skip if CFG_TUSB_MCU in board.mk to match skip file
+        for skip_file in glob.iglob(ex_dir + '/.skip.MCU_*'):
+            mcu_cflag = '-DCFG_TUSB_MCU=OPT_' + os.path.basename(skip_file).split('.')[2]
+            if mcu_cflag in mk_contents:
                 return 1
 
     return 0
 
-build_format = '| {:20} | {:30} | {:9} | {:5} | {:6} | {:6} |'
-build_separator = '-' * 95
 print(build_separator)
 print(build_format.format('Example', 'Board', 'Result', 'Time', 'Flash', 'SRAM'))
 
@@ -93,10 +103,7 @@ for example in all_examples:
             if build_result.returncode != 0:
                 print(build_result.stdout.decode("utf-8"))
 
-# FreeRTOS example
-# example = 'cdc_msc_hid_freertos'
-# board = 'pca10056'
-# build_example(example, board)
+
 
 total_time = time.monotonic() - total_time
 print(build_separator)
diff --git a/tools/build_esp32s.py b/tools/build_esp32s.py
new file mode 100644
index 000000000..0708ced80
--- /dev/null
+++ b/tools/build_esp32s.py
@@ -0,0 +1,101 @@
+import os
+import glob
+import sys
+import subprocess
+import time
+
+success_count = 0
+fail_count = 0
+skip_count = 0
+exit_status = 0
+
+total_time = time.monotonic()
+
+build_format = '| {:23} | {:30} | {:9} | {:7} | {:6} | {:6} |'
+build_separator = '-' * 100
+
+# 1st Argument is Example, build all examples if not existed
+all_examples = []
+if len(sys.argv) > 1:
+    all_examples.append(sys.argv[1])
+else:
+    for entry in os.scandir("examples/device"):
+        # Only includes example with CMakeLists.txt for esp32s
+        if entry.is_dir() and os.path.exists(entry.path + "/CMakeLists.txt"):
+            all_examples.append(entry.name)
+all_examples.sort()
+
+# 2nd Argument is Board, build all boards if not existed
+all_boards = []
+if len(sys.argv) > 2:
+    all_boards.append(sys.argv[2])
+else:
+    for entry in os.scandir("hw/bsp"):
+        if entry.is_dir():
+            with open(entry.path + '/board.mk') as mk:
+                # Only includes ESP32-S2 board
+                if 'CROSS_COMPILE = xtensa-esp32s2-elf-' in mk.read():
+                    all_boards.append(entry.name)
+                                
+all_boards.sort()
+
+def build_example(example, board):
+    subprocess.run("make -C examples/device/{} BOARD={} clean".format(example, board), shell=True,
+                   stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    return subprocess.run("make -j 4 -C examples/device/{} BOARD={} all".format(example, board), shell=True,
+                          stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+
+def build_size(example, board):
+    #elf_file = 'examples/device/{}/_build/build-{}/{}-firmware.elf'.format(example, board, board)
+    elf_file = 'examples/device/{}/_build/build-{}/*.elf'.format(example, board)
+    size_output = subprocess.run('size {}'.format(elf_file), shell=True, stdout=subprocess.PIPE).stdout.decode("utf-8")
+    size_list = size_output.split('\n')[1].split('\t')
+    flash_size = int(size_list[0])
+    sram_size = int(size_list[1]) + int(size_list[2])
+    return (flash_size, sram_size)
+
+def skip_example(example, board):
+    return 0
+
+print(build_separator)
+print(build_format.format('Example', 'Board', 'Result', 'Time', 'Flash', 'SRAM'))
+print(build_separator)
+
+for example in all_examples:
+    for board in all_boards:
+        start_time = time.monotonic()
+
+        flash_size = "-"
+        sram_size = "-"
+
+        # Check if board is skipped
+        if skip_example(example, board):
+            success = "\033[33mskipped\033[0m  "
+            skip_count += 1
+            print(build_format.format(example, board, success, '-', flash_size, sram_size))
+        else:
+            build_result = build_example(example, board)
+
+            if build_result.returncode == 0:
+                success = "\033[32msucceeded\033[0m"
+                success_count += 1
+                (flash_size, sram_size) = build_size(example, board)
+            else:
+                exit_status = build_result.returncode
+                success = "\033[31mfailed\033[0m   "
+                fail_count += 1
+
+            build_duration = time.monotonic() - start_time
+            print(build_format.format(example, board, success, "{:.2f}s".format(build_duration), flash_size, sram_size))
+
+            if build_result.returncode != 0:
+                print(build_result.stdout.decode("utf-8"))
+
+
+
+total_time = time.monotonic() - total_time
+print(build_separator)
+print("Build Sumamary: {} \033[32msucceeded\033[0m, {} \033[31mfailed\033[0m, {} \033[33mskipped\033[0m and took {:.2f}s".format(success_count, fail_count, skip_count, total_time))
+print(build_separator)
+
+sys.exit(exit_status)