diff options
Diffstat (limited to 'tools')
| -rwxr-xr-x | tools/seq_compiler.py | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/tools/seq_compiler.py b/tools/seq_compiler.py index 361aa63..f3a7f16 100755 --- a/tools/seq_compiler.py +++ b/tools/seq_compiler.py @@ -363,13 +363,13 @@ def detect_ping_pong(seq: SequenceDecl, sorted_effects: List[EffectDecl]) -> Dic return aliases def generate_cpp(seq: SequenceDecl, sorted_effects: List[EffectDecl], - aliases: Dict[str, str], flatten: bool = False) -> str: + aliases: Dict[str, str], seq_index: int, flatten: bool = False) -> str: """Generate C++ Sequence subclass.""" class_name = seq.name.replace(' ', '_').replace('-', '_') if not class_name[0].isalpha(): class_name = 'Seq_' + class_name - class_name += 'Sequence' + class_name += f'_{seq_index}_Sequence' # Generate includes includes = set() @@ -465,7 +465,7 @@ def main(): ''' - for seq in sequences: + for seq_idx, seq in enumerate(sequences): # Validate DAG validate_dag(seq) @@ -479,7 +479,7 @@ def main(): aliases = detect_ping_pong(seq, sorted_effects) # Generate C++ - cpp = generate_cpp(seq, sorted_effects, aliases, args.flatten) + cpp = generate_cpp(seq, sorted_effects, aliases, seq_idx, args.flatten) all_cpp += cpp + '\n' # Generate sequence registry and accessors @@ -504,8 +504,11 @@ void InitializeSequences(const GpuContext& ctx, int width, int height) { ''' # Instantiate each sequence - for seq in sequences: - class_name = f"{seq.name}Sequence" + for seq_idx, seq in enumerate(sequences): + seq_name = seq.name.replace(' ', '_').replace('-', '_') + if not seq_name[0].isalpha(): + seq_name = 'Seq_' + seq_name + class_name = f"{seq_name}_{seq_idx}_Sequence" all_cpp += f' g_sequences.push_back({{{seq.start_time}f, {seq.priority}, std::make_unique<{class_name}>(ctx, width, height)}});\n' all_cpp += ''' |