OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2025-01-27 12:48:13 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat(libvideo2x): allow processing videos without PTS information
Some checks failed
Build / ubuntu (push) Has been cancelled
Details
Build / windows (push) Has been cancelled
Details
Build / container (push) Has been cancelled
Details
Build / appimage (push) Has been cancelled
Details

Browse Source 
Signed-off-by: k4yt3x <i@k4yt3x.com>
This commit is contained in:
k4yt3x 2025-01-18 00:00:00 +00:00
parent 43ecf9e825
commit eae89cea4b
No known key found for this signature in database
4 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CHANGELOG.md
View File

@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
### Added
- Multi-versioning to critical functions to enhance performance in generic architecture builds.
- Support for processing videos without PTS information (#1278).
- The feature to copy input streams' metadata to the output streams (#1282).
### Changed

2
include/libvideo2x/encoder.h
View File

@ -65,7 +65,7 @@ class Encoder {
int in_vstream_idx
);
int write_frame(AVFrame* frame, int64_t frame_idx);
int write_frame(AVFrame* frame);
int flush();
AVCodecContext* get_encoder_context() const;

5
src/encoder.cpp
View File

@ -269,16 +269,13 @@ int Encoder::init(
}
[[gnu::target_clones("arch=x86-64-v4", "arch=x86-64-v3", "default")]]
int Encoder::write_frame(AVFrame* frame, int64_t frame_idx) {
int Encoder::write_frame(AVFrame* frame) {
AVFrame* converted_frame = nullptr;
int ret;
// Let the encoder decide the frame type
frame->pict_type = AV_PICTURE_TYPE_NONE;
// Calculate this frame's presentation timestamp (PTS)
frame->pts = av_rescale_q(frame_idx, av_inv_q(enc_ctx_->framerate), enc_ctx_->time_base);
// Convert the frame to the encoder's pixel format if needed
if (frame->format != enc_ctx_->pix_fmt) {
converted_frame = conversions::convert_avframe_pix_fmt(frame, enc_ctx_->pix_fmt);

8
src/libvideo2x.cpp
View File

@ -1,4 +1,5 @@
#include "libvideo2x.h"
#include <libavcodec/avcodec.h>
extern "C" {
#include <libavutil/avutil.h>
@ -152,6 +153,7 @@ int VideoProcessor::process_frames(
AVCodecContext* dec_ctx = decoder.get_codec_context();
int in_vstream_idx = decoder.get_video_stream_index();
AVFormatContext* ofmt_ctx = encoder.get_format_context();
AVCodecContext* enc_ctx = encoder.get_encoder_context();
int* stream_map = encoder.get_stream_map();
// Reference to the previous frame does not require allocation
@ -235,6 +237,10 @@ int VideoProcessor::process_frames(
return ret;
}
// Calculate this frame's presentation timestamp (PTS)
frame->pts =
av_rescale_q(frame_idx_, av_inv_q(enc_ctx->framerate), enc_ctx->time_base);
// Process the frame based on the selected processing mode
AVFrame* proc_frame = nullptr;
switch (processor->get_processing_mode()) {
@ -308,7 +314,7 @@ int VideoProcessor::write_frame(AVFrame* frame, encoder::Encoder& encoder) {
int ret = 0;
if (!benchmark_) {
ret = encoder.write_frame(frame, frame_idx_);
ret = encoder.write_frame(frame);
if (ret < 0) {
av_strerror(ret, errbuf, sizeof(errbuf));
logger()->critical("Error encoding/writing frame: {}", errbuf);