add vs code launch.json auto generation in waf configure

Tools/ardupilotwaf/vscode_helper.py

@@ -0,0 +1,105 @@
+import os
+import json
+import shutil
+
+H7_DUAL_BANK_LIST = [
+   "STM32H7A3xx",
+   "STM32H7A3xxq",
+   "STM32H7B3xx",
+   "STM32H7B3xxq",
+   "STM32H742xx",
+   "STM32H743xx",
+   "STM32H745xg",
+   "STM32H745xx",
+   "STM32H747xg",
+   "STM32H747xx",
+   "STM32H753xx",
+   "STM32H755xx",
+   "STM32H755xx",
+   "STM32H757xx",
+] # List of H7 boards with dual bank
+
+ELF_NAME_MAPPING = {
+    'copter': 'arducopter',
+    'plane': 'arduplane',
+    'rover': 'ardurover',
+    'sub': 'ardusub',
+    'blimp': 'blimp',
+    'antennatracker': 'antennatracker',
+    'bootloader': 'AP_Bootloader',
+    'AP_Periph': 'AP_Periph',
+}
+
+def update_settings(bld):
+    if not bld.cmd in ELF_NAME_MAPPING:
+        return
+    board_name = bld.env.BOARD
+    elf_file_name = ELF_NAME_MAPPING.get(bld.cmd, bld.cmd)
+    if elf_file_name == 'ap_bootloader':
+        elf_file_path = os.path.join("${workspaceFolder}", "build", board_name, "bootloader", elf_file_name)
+    else:
+        elf_file_path = os.path.join("${workspaceFolder}", "build", board_name, "bin", elf_file_name)
+    vscode_setting_json_path = os.path.join(bld.srcnode.abspath(), '.vscode', 'settings.json')
+
+    if not os.path.exists(vscode_setting_json_path):
+        with open(vscode_setting_json_path, 'w') as f:
+            json.dump({}, f, indent=4)
+
+    try:
+        with open(vscode_setting_json_path, 'r') as f:
+            content = f.read().strip()
+            if content:
+                settings_json = json.loads(content)
+            else:
+                settings_json = {}
+    except json.JSONDecodeError:
+        print(f"VS-LAUNCH: \033[91m Error: invalid JSON in .vscode/settings.json, please fix it and try again.\033[0m")
+        return
+
+    settings_json['wscript.elf_file_path'] = elf_file_path
+    settings_json['wscript.board'] = board_name
+    if board_name == 'sitl':
+        if os.uname().sysname == 'Darwin':
+            settings_json['wscript.MIMode'] =  "lldb" 
+        else:
+            settings_json['wscript.MIMode'] =  "gdb" 
+
+    with open(vscode_setting_json_path, 'w') as f:
+        json.dump(settings_json, f, indent=4)
+
+def update_openocd_cfg(cfg):
+    if cfg.options.board != 'sitl':
+        openocd_cfg_path = os.path.join(cfg.srcnode.abspath(), 'build', cfg.options.board, 'openocd.cfg')
+        mcu_type = cfg.env.get_flat('APJ_BOARD_TYPE')
+        openocd_target = ''
+        if mcu_type.startswith("STM32H7"):
+            if mcu_type in H7_DUAL_BANK_LIST:
+                openocd_target = 'stm32h7_dual_bank.cfg'
+            else:
+                openocd_target = 'stm32h7x.cfg'
+        elif mcu_type.startswith("STM32F7"):
+            openocd_target = 'stm32f7x.cfg'
+        elif mcu_type.startswith("STM32F4"):
+            openocd_target = 'stm32f4x.cfg'
+        elif mcu_type.startswith("STM32F3"):
+            openocd_target = 'stm32f3x.cfg'
+        elif mcu_type.startswith("STM32L4"):
+            openocd_target = 'stm32l4x.cfg'
+        elif mcu_type.startswith("STM32G4"):
+            openocd_target = 'stm32g4x.cfg'
+
+        if openocd_target:
+            with open(openocd_cfg_path, 'w+') as f:
+                f.write("source [find interface/stlink.cfg]\n")
+                f.write(f"source [find target/{openocd_target}]\n")
+                f.write("init\n")
+                f.write("$_TARGETNAME configure -rtos auto\n")
+
+def init_launch_json_if_not_exist(cfg):
+    launch_json_path = os.path.join(cfg.srcnode.abspath(), '.vscode', 'launch.json')
+    launch_default_json_path = os.path.join(cfg.srcnode.abspath(),'.vscode', 'launch.default.json')
+
+    if not os.path.exists(launch_json_path):
+        if os.path.exists(launch_default_json_path):
+            print(f"Copying {launch_default_json_path} to {launch_json_path}")
+            shutil.copy(launch_default_json_path, launch_json_path)

wscript

@@ -155,6 +155,11 @@ def options(opt):
         action='store_true',
         default=False,
         help='Add debug symbolds to build.')
+
+    g.add_option('--vs-launch',
+        action='store_true',
+        default=False,
+        help='Generate launch.json template for Visual Studio Code.')
 
     g.add_option('--disable-watchdog',
         action='store_true',
@@ -502,6 +507,7 @@ def configure(cfg):
 
     cfg.env.BOARD = cfg.options.board
     cfg.env.DEBUG = cfg.options.debug
+    cfg.env.VS_LAUNCH = cfg.options.vs_launch
     cfg.env.DEBUG_SYMBOLS = cfg.options.debug_symbols
     cfg.env.COVERAGE = cfg.options.coverage
     cfg.env.FORCE32BIT = cfg.options.force_32bit
@@ -607,6 +613,13 @@ def configure(cfg):
     else:
         cfg.end_msg('disabled', color='YELLOW')
 
+    if cfg.env.DEBUG:
+        cfg.start_msg('VS Code launch')
+        if cfg.env.VS_LAUNCH:
+            cfg.end_msg('enabled')
+        else:
+            cfg.end_msg('disabled', color='YELLOW')
+
     cfg.start_msg('Coverage build')
     if cfg.env.COVERAGE:
         cfg.end_msg('enabled')
@@ -652,6 +665,11 @@ def configure(cfg):
     cfg.remove_target_list()
     _collect_autoconfig_files(cfg)
 
+    if cfg.env.VS_LAUNCH:
+        import vscode_helper
+        vscode_helper.init_launch_json_if_not_exist(cfg)
+        vscode_helper.update_openocd_cfg(cfg)
+
 def collect_dirs_to_recurse(bld, globs, **kw):
     dirs = []
     globs = Utils.to_list(globs)
@@ -912,6 +930,10 @@ def build(bld):
 
     _build_post_funs(bld)
 
+    if bld.env.VS_LAUNCH:
+        import vscode_helper
+        vscode_helper.update_settings(bld)
+
 ardupilotwaf.build_command('check',
     program_group_list='all',
     doc='builds all programs and run tests',