fdbbc3a20d
This does introduce some not-so-nice things, like having to annotate each `__init__` function with `-> None`. However, the benefits of having static type checking in a complex bit of software like BAT outweigh the downsides.
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import typing
|
|
|
|
from blender_asset_tracer import cdefs
|
|
from . import BlendFileBlock
|
|
from .dna import FieldPath
|
|
|
|
|
|
def listbase(block: BlendFileBlock, next_path: FieldPath = b'next') \
|
|
-> typing.Iterator[BlendFileBlock]:
|
|
"""Generator, yields all blocks in the ListBase linked list."""
|
|
while block:
|
|
yield block
|
|
next_ptr = block[next_path]
|
|
if next_ptr == 0:
|
|
break
|
|
block = block.bfile.dereference_pointer(next_ptr)
|
|
|
|
|
|
def sequencer_strips(sequence_editor: BlendFileBlock) \
|
|
-> typing.Iterator[typing.Tuple[BlendFileBlock, int]]:
|
|
"""Generator, yield all sequencer strip blocks with their type number.
|
|
|
|
Recurses into meta strips, yielding both the meta strip itself and the
|
|
strips contained within it.
|
|
|
|
See blender_asset_tracer.cdefs.SEQ_TYPE_xxx for the type numbers.
|
|
"""
|
|
|
|
def iter_seqbase(seqbase) -> typing.Iterator[typing.Tuple[BlendFileBlock, int]]:
|
|
for seq in listbase(seqbase):
|
|
seq.refine_type(b'Sequence')
|
|
seq_type = seq[b'type']
|
|
yield seq, seq_type
|
|
|
|
if seq_type == cdefs.SEQ_TYPE_META:
|
|
# Recurse into this meta-sequence.
|
|
subseq = seq.get_pointer((b'seqbase', b'first'))
|
|
yield from iter_seqbase(subseq)
|
|
|
|
sbase = sequence_editor.get_pointer((b'seqbase', b'first'))
|
|
yield from iter_seqbase(sbase)
|