106 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			106 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| import os
 | |
| import sys
 | |
| import time
 | |
| import subprocess
 | |
| import pathlib
 | |
| from multiprocessing import Pool
 | |
| 
 | |
| import build_utils
 | |
| 
 | |
| SUCCEEDED = "\033[32msucceeded\033[0m"
 | |
| FAILED = "\033[31mfailed\033[0m"
 | |
| SKIPPED = "\033[33mskipped\033[0m"
 | |
| 
 | |
| build_separator = '-' * 106
 | |
| 
 | |
| def filter_with_input(mylist):
 | |
|     if len(sys.argv) > 1:
 | |
|         input_args = list(set(mylist).intersection(sys.argv))
 | |
|         if len(input_args) > 0:
 | |
|             mylist[:] = input_args
 | |
| 
 | |
| 
 | |
| def build_family(family, cmake_option):
 | |
|     all_boards = []
 | |
|     for entry in os.scandir("hw/bsp/{}/boards".format(family)):
 | |
|         if entry.is_dir() and entry.name != 'pico_sdk':
 | |
|             all_boards.append(entry.name)
 | |
|     all_boards.sort()
 | |
| 
 | |
|     # success, failed, skipped
 | |
|     ret = [0, 0, 0]
 | |
|     for board in all_boards:
 | |
|         start_time = time.monotonic()
 | |
| 
 | |
|         build_dir = f"cmake-build/cmake-build-{board}"
 | |
| 
 | |
|         # Generate build
 | |
|         r = subprocess.run(f"cmake examples -B {build_dir} -G \"Ninja\" -DFAMILY={family} -DBOARD"
 | |
|                            f"={board} {cmake_option}", shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
 | |
| 
 | |
|         # Build
 | |
|         if r.returncode == 0:
 | |
|             r = subprocess.run(f"cmake --build {build_dir}", shell=True, stdout=subprocess.PIPE,
 | |
|                                stderr=subprocess.STDOUT)
 | |
| 
 | |
|         duration = time.monotonic() - start_time
 | |
| 
 | |
|         if r.returncode == 0:
 | |
|             status = SUCCEEDED
 | |
|             ret[0] += 1
 | |
|         else:
 | |
|             status = FAILED
 | |
|             ret[1] += 1
 | |
| 
 | |
|         flash_size = "-"
 | |
|         sram_size = "-"
 | |
|         example = 'all'
 | |
|         title = build_utils.build_format.format(example, board, status, "{:.2f}s".format(duration), flash_size, sram_size)
 | |
| 
 | |
|         if os.getenv('CI'):
 | |
|             # always print build output if in CI
 | |
|             print(f"::group::{title}")
 | |
|             print(r.stdout.decode("utf-8"))
 | |
|             print(f"::endgroup::")
 | |
|         else:
 | |
|             # print build output if failed
 | |
|             print(title)
 | |
|             if r.returncode != 0:
 | |
|                 print(r.stdout.decode("utf-8"))
 | |
| 
 | |
|     return ret
 | |
| 
 | |
| 
 | |
| if __name__ == '__main__':
 | |
|     cmake_options = ''
 | |
|     for a in sys.argv[1:]:
 | |
|         if a.startswith('-'):
 | |
|             cmake_options += ' ' + a
 | |
| 
 | |
|     # If family are not specified in arguments, build all supported
 | |
|     all_families = []
 | |
|     for entry in os.scandir("hw/bsp"):
 | |
|         if entry.is_dir() and entry.name != 'espressif' and os.path.isfile(entry.path + "/family.cmake"):
 | |
|             all_families.append(entry.name)
 | |
|     filter_with_input(all_families)
 | |
|     all_families.sort()
 | |
| 
 | |
|     print(build_separator)
 | |
|     print(build_utils.build_format.format('Example', 'Board', '\033[39mResult\033[0m', 'Time', 'Flash', 'SRAM'))
 | |
|     total_time = time.monotonic()
 | |
| 
 | |
|     # succeeded, failed, skipped
 | |
|     total_result = [0, 0, 0]
 | |
|     for family in all_families:
 | |
|         fret = build_family(family, cmake_options)
 | |
|         if len(fret) == len(total_result):
 | |
|             total_result = [total_result[i] + fret[i] for i in range(len(fret))]
 | |
| 
 | |
|     total_time = time.monotonic() - total_time
 | |
|     print(build_separator)
 | |
|     print("Build Summary: {} {}, {} {}, {} {} and took {:.2f}s".format(total_result[0], SUCCEEDED, total_result[1],
 | |
|                                                                        FAILED, total_result[2], SKIPPED, total_time))
 | |
|     print(build_separator)
 | |
| 
 | |
|     sys.exit(total_result[1])
 | 
