File size: 2,340 Bytes
a4ba82a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | import bisect
from decimal import Decimal, ROUND_HALF_EVEN
def microseconds(value):
return int((Decimal(str(value)) * 1000000).to_integral_value(rounding=ROUND_HALF_EVEN))
def frame_span(pts_s, start_us, end_us):
if type(start_us) is not int or type(end_us) is not int or start_us < 0 or (end_us <= start_us):
raise ValueError('invalid integer-microsecond source interval')
times = [microseconds(t) for t in pts_s]
if not times or any((a >= b for a, b in zip(times, times[1:]))):
raise ValueError('PTS not strictly increasing after microsecond serialization')
return (bisect.bisect_left(times, start_us), bisect.bisect_left(times, end_us))
def native_blocks(pts_s, start_us, end_us, duration_us, rate, contract):
if end_us > duration_us:
raise ValueError('target beyond source')
windows = [('BEFORE', max(0, start_us - 2000000), start_us), ('TARGET', start_us, end_us), ('AFTER', end_us, min(duration_us, end_us + 2000000))]
blocks = []
for role, a, b in windows:
if a == b:
continue
lo, hi = frame_span(pts_s, a, b)
target = role == 'TARGET'
factor = contract.TARGET_FRAME_FACTOR if target else contract.CONTEXT_FRAME_FACTOR
if hi - lo < factor:
if target:
raise ValueError('target_has_fewer_than_native_frame_factor')
continue
ids, receipt = contract.duration_adaptive_sample(lo, hi, source_fps_num=rate.numerator, source_fps_den=rate.denominator, sample_fps=contract.TARGET_SAMPLE_FPS if target else contract.CONTEXT_SAMPLE_FPS, min_frames=contract.TARGET_MIN_FRAMES if target else contract.CONTEXT_MIN_FRAMES, max_frames=contract.TARGET_MAX_FRAMES if target else contract.CONTEXT_MAX_FRAMES, frame_factor=factor)
assert len(set(ids)) == len(ids) and all((lo <= i < hi for i in ids))
times = [microseconds(pts_s[i]) for i in ids]
assert all((a <= t < b for t in times))
blocks.append({'role': role, 'interval_us': [a, b], 'start_frame': lo, 'end_frame_exclusive': hi, 'frame_indices': ids, 'actual_pts_us': times, 'sampling_receipt': receipt, 'model_timestamp_clock': 'native ordinal/average FPS; exact source PTS retained separately'})
assert sum((len(b['frame_indices']) for b in blocks)) <= contract.MAX_TOTAL_FRAMES
return blocks
|