diff options
| -rwxr-xr-x | tools/seq_compiler.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/tools/seq_compiler.py b/tools/seq_compiler.py index 2e51574..e3f93af 100755 --- a/tools/seq_compiler.py +++ b/tools/seq_compiler.py @@ -391,12 +391,19 @@ def generate_cpp(seq: SequenceDecl, sorted_effects: List[EffectDecl], class_name += f'_{seq_index}_Sequence' # Generate includes + # Map class names that share a header file + CLASS_TO_HEADER = { + 'NtscYiq': 'ntsc', + } includes = set() for effect in seq.effects: - # Convert ClassName to snake_case header - header = re.sub('([A-Z])', r'_\1', effect.class_name).lower().lstrip('_') - if header.endswith('_effect'): - header = header[:-7] # Remove _effect suffix + if effect.class_name in CLASS_TO_HEADER: + header = CLASS_TO_HEADER[effect.class_name] + else: + # Convert ClassName to snake_case header + header = re.sub('([A-Z])', r'_\1', effect.class_name).lower().lstrip('_') + if header.endswith('_effect'): + header = header[:-7] # Remove _effect suffix includes.add(f'#include "effects/{header}_effect.h"') cpp = f'''// Generated by seq_compiler.py |