Implemented arm_gcc.py
This commit is contained in:
parent
202e8700c2
commit
d1c6dbd0b1
|
|
@ -0,0 +1,202 @@
|
|||
#!/usr/bin/env python3
|
||||
"""
|
||||
ARM GCC compiler implementation for embedded targets.
|
||||
Used primarily for RP2040 and other ARM Cortex-M microcontrollers.
|
||||
"""
|
||||
|
||||
from pathlib import Path
|
||||
from typing import List, Optional
|
||||
from .base import Compiler, CompileResult
|
||||
from build_system.utils import run
|
||||
|
||||
|
||||
class ARMGCCCompiler(Compiler):
|
||||
"""ARM GCC compiler for embedded targets."""
|
||||
|
||||
def __init__(self, executable: str = "arm-none-eabi-gcc", cpu: str = "cortex-m0plus"):
|
||||
"""
|
||||
Initialize ARM GCC compiler.
|
||||
|
||||
Args:
|
||||
executable: Compiler executable name
|
||||
cpu: Target CPU (e.g., "cortex-m0plus", "cortex-m4")
|
||||
"""
|
||||
super().__init__("ARM GCC", executable)
|
||||
self.cpu = cpu
|
||||
self.thumb = True # Use Thumb instruction set by default
|
||||
|
||||
def compile(self,
|
||||
source: Path,
|
||||
output: Path,
|
||||
flags: Optional[List[str]] = None,
|
||||
defines: Optional[List[str]] = None,
|
||||
include_dirs: Optional[List[Path]] = None,
|
||||
dependency_file: Optional[Path] = None) -> CompileResult:
|
||||
"""
|
||||
Compile a source file using ARM GCC.
|
||||
|
||||
Args:
|
||||
source: Source file path
|
||||
output: Output object file path
|
||||
flags: Additional compiler flags
|
||||
defines: Preprocessor defines
|
||||
include_dirs: Additional include directories
|
||||
dependency_file: Path for dependency file (.d file)
|
||||
|
||||
Returns:
|
||||
CompileResult with object file and dependency file paths
|
||||
|
||||
Raises:
|
||||
subprocess.CalledProcessError: If compilation fails
|
||||
"""
|
||||
# Ensure output directory exists
|
||||
output.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Build command
|
||||
cmd = [self.executable]
|
||||
|
||||
# Add CPU-specific flags
|
||||
cmd.append(f"-mcpu={self.cpu}")
|
||||
if self.thumb:
|
||||
cmd.append("-mthumb")
|
||||
|
||||
# Add flags
|
||||
if flags:
|
||||
cmd.extend(flags)
|
||||
|
||||
# Add compile-only flag
|
||||
cmd.append(self.get_compile_only_flag())
|
||||
|
||||
# Add defines
|
||||
if defines:
|
||||
cmd.extend([self.get_define_flag(d) for d in defines])
|
||||
|
||||
# Add include directories
|
||||
if include_dirs:
|
||||
cmd.extend([self.get_include_flag(d) for d in include_dirs])
|
||||
|
||||
# Add dependency generation if requested
|
||||
dep_file = None
|
||||
if dependency_file:
|
||||
cmd.extend(["-MMD", "-MP"])
|
||||
dep_file = dependency_file
|
||||
|
||||
# Add source file
|
||||
cmd.append(str(source))
|
||||
|
||||
# Add output file
|
||||
cmd.extend(self.get_output_flag(output))
|
||||
|
||||
# Run compilation
|
||||
run(cmd) # type: ignore
|
||||
|
||||
return CompileResult(
|
||||
object_file=output,
|
||||
dependency_file=dep_file,
|
||||
success=True
|
||||
)
|
||||
|
||||
def link(self,
|
||||
objects: List[Path],
|
||||
output: Path,
|
||||
flags: Optional[List[str]] = None,
|
||||
libraries: Optional[List[str]] = None,
|
||||
library_dirs: Optional[List[Path]] = None) -> bool:
|
||||
"""
|
||||
Link object files into an executable using ARM GCC.
|
||||
|
||||
Args:
|
||||
objects: List of object file paths
|
||||
output: Output executable path
|
||||
flags: Additional linker flags
|
||||
libraries: Libraries to link against
|
||||
library_dirs: Additional library search directories
|
||||
|
||||
Returns:
|
||||
True if linking succeeded
|
||||
|
||||
Raises:
|
||||
subprocess.CalledProcessError: If linking fails
|
||||
"""
|
||||
# Ensure output directory exists
|
||||
output.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Build command
|
||||
cmd = [self.executable]
|
||||
|
||||
# Add CPU-specific flags
|
||||
cmd.append(f"-mcpu={self.cpu}")
|
||||
if self.thumb:
|
||||
cmd.append("-mthumb")
|
||||
|
||||
# Add object files
|
||||
cmd.extend([str(obj) for obj in objects])
|
||||
|
||||
# Add linker flags
|
||||
if flags:
|
||||
cmd.extend(flags)
|
||||
|
||||
# Add library directories
|
||||
if library_dirs:
|
||||
cmd.extend([f"-L{lib_dir}" for lib_dir in library_dirs])
|
||||
|
||||
# Add libraries
|
||||
if libraries:
|
||||
cmd.extend([f"-l{lib}" for lib in libraries])
|
||||
|
||||
# Add output file
|
||||
cmd.extend(["-o", str(output)])
|
||||
|
||||
# Run linker
|
||||
run(cmd) # type: ignore
|
||||
|
||||
return True
|
||||
|
||||
def get_object_extension(self) -> str:
|
||||
"""Get object file extension for ARM GCC."""
|
||||
return ".o"
|
||||
|
||||
def get_dependency_extension(self) -> str:
|
||||
"""Get dependency file extension for ARM GCC."""
|
||||
return ".d"
|
||||
|
||||
def supports_dependency_generation(self) -> bool:
|
||||
"""ARM GCC supports automatic dependency generation."""
|
||||
return True
|
||||
|
||||
def set_cpu(self, cpu: str) -> None:
|
||||
"""
|
||||
Set the target CPU.
|
||||
|
||||
Args:
|
||||
cpu: Target CPU name (e.g., "cortex-m0plus", "cortex-m4")
|
||||
"""
|
||||
self.cpu = cpu
|
||||
|
||||
def set_thumb(self, enabled: bool) -> None:
|
||||
"""
|
||||
Enable or disable Thumb instruction set.
|
||||
|
||||
Args:
|
||||
enabled: True to enable Thumb, False to disable
|
||||
"""
|
||||
self.thumb = enabled
|
||||
|
||||
|
||||
def check_arm_toolchain() -> bool:
|
||||
"""
|
||||
Check if ARM toolchain is available.
|
||||
|
||||
Returns:
|
||||
True if ARM toolchain is found, False otherwise
|
||||
"""
|
||||
from build_system.utils import check_tool
|
||||
|
||||
if check_tool("arm-none-eabi-gcc"):
|
||||
return True
|
||||
else:
|
||||
print(
|
||||
"ARM toolchain not found. Install arm-none-eabi-gcc:\n"
|
||||
" Ubuntu/Debian: sudo apt install gcc-arm-none-eabi\n"
|
||||
" macOS: brew install arm-none-eabi-gcc"
|
||||
)
|
||||
Loading…
Reference in New Issue