X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=hacks%2Fexamine.py;h=f498dab2222793106ffa6f4c3a95224188d38e61;hb=refs%2Fheads%2F2226-add-video-waveform-to-player;hp=34544c7f6399e418d3145115dceb9bacb88cc1cb;hpb=a6ca3acafbfb6a868f10a9e9e1e7c61008643e72;p=dcpomatic.git

diff --git a/hacks/examine.py b/hacks/examine.py
index 34544c7f6..f498dab22 100644
--- a/hacks/examine.py
+++ b/hacks/examine.py
@@ -5,7 +5,8 @@ import shlex
 import sys
 
 AUDIO_STREAM = "1"
-types = ['audio']
+types = ['video']
+VIDEO_RATE = 29.97
 
 last_video = None
 last_video_pts = None
@@ -13,53 +14,68 @@ last_video_pts = None
 last_audio_pts = {}
 last_channels = {}
 
+video_frame_rates = []
+
+video_frame_count = 0
+
 def check_pts_dts(frame):
-    diff = frame['pkt_pts_time'] - frame['pkt_dts_time']
-    if abs(diff) > 1e-8:
-        print("\tPTS != DTS: %f" % diff)
+    if 'pkt_dts_time' in frame:
+        diff = frame['pkt_pts_time'] - frame['pkt_dts_time']
+        if abs(diff) > 1e-8:
+            print("\tPTS != DTS: %f" % diff)
 
 def handle(frame):
     global last_video
     global last_video_pts
     global last_audio_pts
+    global video_frame_count
+    global video_frame_rates
     if frame['media_type'] == 'video' and 'video' in types:
         if last_video_pts is not None and frame['pkt_pts_time'] <= last_video_pts:
-            print 'Out of order video frame %f (%d) is same as or behind %f (%d)' % (frame['pkt_pts_time'], frame['pkt_pts'], last_video_pts, last_video)
+            print('Out of order video frame %f (%d) is same as or behind %f (%d)' % (frame['pkt_pts_time'], frame['pkt_pts'], last_video_pts, last_video))
         elif last_video_pts is not None:
-            print 'OK V    frame %f %f %f %f %d' % (frame['pkt_pts_time'], frame['pkt_dts_time'], frame['pkt_pts_time'] - last_video_pts, 1 / (frame['pkt_pts_time'] - last_video_pts), frame['pkt_size'])
+            pkt_dts_time = frame['pkt_dts_time'] if 'pkt_dts_time' in frame else 0.0
+            rate = 1 / (frame['pkt_pts_time'] - last_video_pts)
+            print(f'OK V    frame %f %f %f %f %d indices %d/%d' % (frame['pkt_pts_time'], pkt_dts_time, frame['pkt_pts_time'] - last_video_pts, rate, frame['pkt_size'], video_frame_count, frame['pkt_pts_time'] * VIDEO_RATE))
             check_pts_dts(frame)
-	else:
-            print 'OK V    frame %f' % (frame['pkt_pts_time'])
+            video_frame_rates.append(rate)
+        else:
+            print('OK V    frame %f counted %d' % (frame['pkt_pts_time'], video_frame_count))
         last_video = frame['pkt_pts']
         last_video_pts = frame['pkt_pts_time']
+        video_frame_count += 1
     elif frame['media_type'] == 'audio' and 'audio' in types:
-	stream_index = frame['stream_index']
-	if stream_index in last_audio_pts and (stream_index == AUDIO_STREAM or AUDIO_STREAM is None):
-            print 'OK A[%s] frame %4.8f %4.8f %4.8f %4.8f %d' % (stream_index, frame['pkt_pts_time'], frame['pkt_dts_time'], frame['pkt_pts_time'] - last_audio_pts[stream_index], 1 / (frame['pkt_pts_time'] - last_audio_pts[stream_index]), frame['pkt_size'])
+        stream_index = frame['stream_index']
+        if stream_index in last_audio_pts and (stream_index == AUDIO_STREAM or AUDIO_STREAM is None):
+            print('OK A[%s] frame %4.8f %4.8f %4.8f %4.8f %d' % (stream_index, frame['pkt_pts_time'], frame['pkt_dts_time'], frame['pkt_pts_time'] - last_audio_pts[stream_index], 1 / (frame['pkt_pts_time'] - last_audio_pts[stream_index]), frame['pkt_size']))
             check_pts_dts(frame)
         if stream_index in last_channels and frame['channels'] != last_channels[stream_index]:
-            print "\t*** unusual channel count"
+            print("\t*** unusual channel count")
         last_audio_pts[stream_index] = frame['pkt_pts_time']
-	last_channels[stream_index] = frame['channels']
+        last_channels[stream_index] = frame['channels']
 
 
 p = subprocess.Popen(shlex.split('ffprobe -show_frames "%s"' % sys.argv[1]), stdin=None, stdout=subprocess.PIPE)
 frame = dict()
 while True:
     l = p.stdout.readline()
-    if l == '':
+    if l == b'':
         break
 
-    l = l.strip()
+    l = l.strip().decode('UTF-8')
 
     if l == '[/FRAME]':
         handle(frame)
         frame = dict()
     elif l != '[FRAME]' and l != '[SIDE_DATA]' and l != '[/SIDE_DATA]':
         s = l.split('=')
-	if s[0] in ['pkt_pts_time', 'pkt_dts_time', 'pkt_pts', 'pkt_dts']:
-            frame[s[0]] = float(s[1])
-	elif s[0] in ['channels', 'pkt_size', 'nb_samples']:
-	    frame[s[0]] = int(s[1])
+        if s[0] in ['pkt_pts_time', 'pkt_dts_time', 'pkt_pts', 'pkt_dts']:
+            if s[1] != 'N/A':
+                frame[s[0]] = float(s[1])
+        elif s[0] in ['channels', 'pkt_size', 'nb_samples']:
+            frame[s[0]] = int(s[1])
         elif len(s) > 1:
             frame[s[0]] = s[1]
+
+
+print(f'Average frame rate: {sum(video_frame_rates) / float(len(video_frame_rates))}')