49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
# People's Video Editor: high quality, GPU accelerated mp4 editor
|
|
# Copyright (C) 2025 Roz K <roz@rozk.net>
|
|
#
|
|
# This file is part of People's Video Editor.
|
|
#
|
|
# People's Video Editor is free software: you can redistribute it and/or modify it under the terms of the
|
|
# GNU General Public License as published by the Free Software Foundation, either version 3 of the License,
|
|
# or (at your option) any later version.
|
|
#
|
|
# People's Video Editor is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
|
|
# without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
|
# See the GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along with People's Video Editor.
|
|
# If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import cProfile
|
|
|
|
from pyav.demuxer import Demuxer
|
|
from pyav.decoder import Decoder
|
|
|
|
demuxer = Demuxer('test.mp4')
|
|
|
|
print(f"nb_streams = {demuxer.nb_streams}")
|
|
print(f"video codec = {demuxer.video_stream.codec.description()}")
|
|
print(f"audio codec = {demuxer.audio_stream.codec.description()}")
|
|
|
|
video_decoder = Decoder(demuxer.video_stream)
|
|
audio_decoder = Decoder(demuxer.audio_stream)
|
|
|
|
num_frames = 0
|
|
|
|
with cProfile.Profile() as pr:
|
|
while True:
|
|
packet = demuxer.read_packet()
|
|
eof = (packet is None)
|
|
if eof or demuxer.video_stream.contains(packet):
|
|
video_frames = video_decoder.decode(packet)
|
|
num_frames += len(video_frames)
|
|
# print(f"decoded {len(video_frames)} video frames")
|
|
if eof or demuxer.audio_stream.contains(packet):
|
|
audio_frames = audio_decoder.decode(packet)
|
|
# print(f"decoded {len(audio_frames)} audio frames")
|
|
if eof:
|
|
break
|
|
|
|
print(f"num frames: {num_frames}")
|
|
pr.print_stats()
|