Add 'cpp' segment

ethteck · Aug 16, 2023 · bcb24f6 · bcb24f6
1 parent 6ec2b39
commit bcb24f6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/segtypes/common/c.py b/segtypes/common/c.py
@@ -19,6 +19,8 @@ class CommonSegC(CommonSegCodeSubsegment):
     global_asm_funcs: Set[str] = set()
     global_asm_rodata_syms: Set[str] = set()
 
+    file_extension = "c"
+
     STRIP_C_COMMENTS_RE = re.compile(
         r'//.*?$|/\*.*?\*/|\'(?:\\.|[^\\\'])*\'|"(?:\\.|[^\\"])*"',
         re.DOTALL | re.MULTILINE,
@@ -124,7 +126,7 @@ def is_text() -> bool:
         return True
 
     def out_path(self) -> Optional[Path]:
-        return options.opts.src_path / self.dir / f"{self.name}.c"
+        return options.opts.src_path / self.dir / f"{self.name}.{self.file_extension}"
 
     def scan(self, rom_bytes: bytes):
         if (

diff --git a/segtypes/common/cpp.py b/segtypes/common/cpp.py
@@ -0,0 +1,5 @@
+from segtypes.common.c import CommonSegC
+
+
+class CommonSegCpp(CommonSegC):
+    file_extension = "cpp"