xref: /aosp_15_r20/external/libaom/tools/gop_bitrate/analyze_data.py (revision 77c1e3ccc04c968bd2bc212e87364f250e820521)
1*77c1e3ccSAndroid Build Coastguard Workerwith open('experiment.txt', 'r') as file:
2*77c1e3ccSAndroid Build Coastguard Worker    lines = file.readlines()
3*77c1e3ccSAndroid Build Coastguard Worker    curr_filename = ''
4*77c1e3ccSAndroid Build Coastguard Worker    keyframe = 0
5*77c1e3ccSAndroid Build Coastguard Worker    actual_value = 0
6*77c1e3ccSAndroid Build Coastguard Worker    estimate_value = 0
7*77c1e3ccSAndroid Build Coastguard Worker    print('filename, estimated value (b), actual value (b)')
8*77c1e3ccSAndroid Build Coastguard Worker    for line in lines:
9*77c1e3ccSAndroid Build Coastguard Worker        if line.startswith('input:'):
10*77c1e3ccSAndroid Build Coastguard Worker            curr_filename = line[13:].strip()
11*77c1e3ccSAndroid Build Coastguard Worker        if line.startswith('estimated'):
12*77c1e3ccSAndroid Build Coastguard Worker            estimate_value = float(line[19:].strip())
13*77c1e3ccSAndroid Build Coastguard Worker        if line.startswith('frame:'):
14*77c1e3ccSAndroid Build Coastguard Worker            actual_value += float(line[line.find('size')+6:line.find('total')-2])
15*77c1e3ccSAndroid Build Coastguard Worker        if line.startswith('****'):
16*77c1e3ccSAndroid Build Coastguard Worker            print(f'{curr_filename}, {estimate_value}, {actual_value}')
17*77c1e3ccSAndroid Build Coastguard Worker            estimate_value = 0
18*77c1e3ccSAndroid Build Coastguard Worker            actual_value = 0
19