FFmpeg/Bitrate_Histogram.py at master · antatura/FFmpeg · GitHub

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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# Requirements: FFprobe, Python, matplotlib, numpy
# References: https://github.com/zeroepoch/plotbitrate

import subprocess
import argparse
import matplotlib.pyplot as plt
import numpy as np
from datetime import datetime


parser = argparse.ArgumentParser()
parser.add_argument('input')
args = parser.parse_args()

print('\nRunning...')
startime = datetime.now()

csv = subprocess.Popen(
    ['ffprobe','-v','error','-select_streams','v:0','-show_entries',\
     'frame=pkt_pts_time,pkt_size','-of','csv=p=0',args.input],
    stdout=subprocess.PIPE,shell=True)

duration_b = subprocess.Popen(
    ['ffprobe','-v','error','-select_streams','v:0','-show_entries',\
     'stream=duration','-of','csv=p=0',args.input],
    stdout=subprocess.PIPE,shell=True)

csv_list = list(csv.stdout)
# [b'0.000000,182002\r\n', b'0.016667,117\r\n', b'0.033333,189\r\n'......]
duration_list = list(duration_b.stdout)


s_size = 0
list_size = []
frames = 0

for a in range(len(csv_list)):
    if csv_list[a] != b'\r\n':
        frames += 1
        aa = str(csv_list[a],'ascii')      # '0.016667,117\r\n'
        a_ = aa.index(',')
        ax = float(aa[:a_])
        ay = int(aa[a_+1:-2])          # ax=0.016667, ay=117
        if a == 0:
            ax0 = ax
            ax0_int = int(ax0)
            sec = ax0_int + 1
        if ax < sec:
            s_size += ay*8/1000       # kbit
        else:
            list_size.append(s_size)
            s_size = ay*8/1000
            sec += 1
list_size.append(s_size)
ls_0 = [0 for x in range(ax0_int)]
list_size = ls_0 + list_size

try:
    duration = float(str(duration_list[0],'ascii'))
except:
    duration = round((ax-ax0)/(frames-1)+(ax-ax0),6)

end = ax0 + duration
maxi = int(np.max(list_size))
mean = int(np.sum(list_size)/duration)

endtime = datetime.now()
eal = str(endtime-startime)
w_title = '['+eal+']'+' '*3+" Bitrate Histogram"
xlabel = 'Start: '+str(ax0)+'    Duration: '+str(duration)+'    End: '+str(end)\
         +'    Frames: '+str(frames)+'    fps: '+str(round(frames/duration,2))

##print(list_size[:14])


plt.get_current_fig_manager().set_window_title(w_title)
plt.title(args.input,fontsize=16,fontweight='bold')
plt.xlabel(xlabel,fontsize=12,fontweight='bold')
plt.ylabel("kbit per sec",fontsize=12,fontweight='bold')
plt.grid(False)    #网格线
plt.bar(range(len(list_size)),list_size,width=1,ec='#195f90',align='edge')
plt.axhline(y=maxi,color='r',lw=2,ls='--',label='Max: '+str(maxi))
plt.axhline(y=mean,color='g',lw=2,ls='--',label='mean: '+str(mean))
plt.axvline(x=ax0,color='y',lw=2.5,ls=':')
plt.axvline(x=end,color='y',lw=2.5,ls=':')
plt.legend(prop={'size': 16})
plt.show()