mirror of
https://git.dev.opencascade.org/repos/occt.git
synced 2025-07-15 12:35:51 +03:00
06f6a5afec
- Wrap deprecated FFmpeg calls in MSVC warning push/pop pragmas in two source files - Introduce `.github/workflows/master-validation.yml` and remove several old workflows - Broadened `if` conditions in `daily-ir-vcpkg-configure.yml` and updated macOS brew deps
333 lines
8.9 KiB
C++
333 lines
8.9 KiB
C++
// Created by: Kirill GAVRILOV
|
|
// Copyright (c) 2019 OPEN CASCADE SAS
|
|
//
|
|
// This file is part of Open CASCADE Technology software library.
|
|
//
|
|
// This library is free software; you can redistribute it and/or modify it under
|
|
// the terms of the GNU Lesser General Public License version 2.1 as published
|
|
// by the Free Software Foundation, with special exception defined in the file
|
|
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
|
|
// distribution for complete text of the license and disclaimer of any warranty.
|
|
//
|
|
// Alternatively, this file may be used under the terms of Open CASCADE
|
|
// commercial license or contractual agreement.
|
|
|
|
// activate some C99 macros like UINT64_C in "stdint.h" which used by FFmpeg
|
|
#ifndef __STDC_CONSTANT_MACROS
|
|
#define __STDC_CONSTANT_MACROS
|
|
#endif
|
|
|
|
#include <Media_CodecContext.hxx>
|
|
#include "../Media/Media_FFmpegCompatibility.pxx"
|
|
|
|
#include <Media_Frame.hxx>
|
|
#include <Media_FormatContext.hxx>
|
|
|
|
#include <Message.hxx>
|
|
#include <Message_Messenger.hxx>
|
|
#include <OSD_Parallel.hxx>
|
|
|
|
IMPLEMENT_STANDARD_RTTIEXT(Media_CodecContext, Standard_Transient)
|
|
|
|
//=================================================================================================
|
|
|
|
Media_CodecContext::Media_CodecContext()
|
|
: myCodecCtx(NULL),
|
|
myCodec(NULL),
|
|
myPtsStartBase(0.0),
|
|
myPtsStartStream(0.0),
|
|
myTimeBase(1.0),
|
|
myStreamIndex(0),
|
|
myPixelAspectRatio(1.0f)
|
|
{
|
|
#ifdef HAVE_FFMPEG
|
|
myCodecCtx = avcodec_alloc_context3(NULL);
|
|
#endif
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
Media_CodecContext::~Media_CodecContext()
|
|
{
|
|
Close();
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
bool Media_CodecContext::Init(const AVStream& theStream, double thePtsStartBase, int theNbThreads)
|
|
{
|
|
#ifdef HAVE_FFMPEG
|
|
return Init(theStream, thePtsStartBase, theNbThreads, AV_CODEC_ID_NONE);
|
|
#else
|
|
return Init(theStream, thePtsStartBase, theNbThreads, 0);
|
|
#endif
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
bool Media_CodecContext::Init(const AVStream& theStream,
|
|
double thePtsStartBase,
|
|
int theNbThreads,
|
|
int theCodecId)
|
|
{
|
|
#ifdef HAVE_FFMPEG
|
|
myStreamIndex = theStream.index;
|
|
#if FFMPEG_HAVE_AVCODEC_PARAMETERS
|
|
if (avcodec_parameters_to_context(myCodecCtx, theStream.codecpar) < 0)
|
|
{
|
|
Message::SendFail("Internal error: unable to copy codec parameters");
|
|
Close();
|
|
return false;
|
|
}
|
|
#else
|
|
// For older FFmpeg, copy from stream's codec context
|
|
#ifdef _MSC_VER
|
|
#pragma warning(push)
|
|
#pragma warning(disable : 4996) // deprecated declaration
|
|
#endif
|
|
if (avcodec_copy_context(myCodecCtx, theStream.codec) < 0)
|
|
#ifdef _MSC_VER
|
|
#pragma warning(pop)
|
|
#endif
|
|
{
|
|
Message::SendFail("Internal error: unable to copy codec context");
|
|
Close();
|
|
return false;
|
|
}
|
|
#endif
|
|
|
|
myTimeBase = av_q2d(theStream.time_base);
|
|
myPtsStartBase = thePtsStartBase;
|
|
myPtsStartStream = Media_FormatContext::StreamUnitsToSeconds(theStream, theStream.start_time);
|
|
|
|
#if FFMPEG_HAVE_AVCODEC_PARAMETERS
|
|
const AVCodecID aCodecId =
|
|
theCodecId != AV_CODEC_ID_NONE ? (AVCodecID)theCodecId : theStream.codecpar->codec_id;
|
|
#else
|
|
#ifdef _MSC_VER
|
|
#pragma warning(push)
|
|
#pragma warning(disable : 4996) // deprecated declaration
|
|
#endif
|
|
const AVCodecID aCodecId = theCodecId != 0 ? (AVCodecID)theCodecId : theStream.codec->codec_id;
|
|
#ifdef _MSC_VER
|
|
#pragma warning(pop)
|
|
#endif
|
|
#endif
|
|
|
|
myCodec = ffmpeg_find_decoder(aCodecId);
|
|
if (myCodec == NULL)
|
|
{
|
|
Message::Send("FFmpeg: unable to find decoder", Message_Fail);
|
|
Close();
|
|
return false;
|
|
}
|
|
|
|
myCodecCtx->codec_id = aCodecId;
|
|
AVDictionary* anOpts = NULL;
|
|
av_dict_set(&anOpts, "refcounted_frames", "1", 0);
|
|
|
|
#if FFMPEG_HAVE_AVCODEC_PARAMETERS
|
|
if (theStream.codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
|
|
#else
|
|
#ifdef _MSC_VER
|
|
#pragma warning(push)
|
|
#pragma warning(disable : 4996) // deprecated declaration
|
|
#endif
|
|
if (theStream.codec->codec_type == AVMEDIA_TYPE_VIDEO)
|
|
#ifdef _MSC_VER
|
|
#pragma warning(pop)
|
|
#endif
|
|
#endif
|
|
{
|
|
myCodecCtx->thread_count =
|
|
theNbThreads <= -1 ? OSD_Parallel::NbLogicalProcessors() : theNbThreads;
|
|
}
|
|
|
|
if (avcodec_open2(myCodecCtx, myCodec, &anOpts) < 0)
|
|
{
|
|
Message::SendFail("FFmpeg: unable to open decoder");
|
|
Close();
|
|
return false;
|
|
}
|
|
|
|
myPixelAspectRatio = 1.0f;
|
|
if (theStream.sample_aspect_ratio.num
|
|
&& av_cmp_q(theStream.sample_aspect_ratio, myCodecCtx->sample_aspect_ratio) != 0)
|
|
{
|
|
myPixelAspectRatio =
|
|
float(theStream.sample_aspect_ratio.num) / float(theStream.sample_aspect_ratio.den);
|
|
}
|
|
else
|
|
{
|
|
if (myCodecCtx->sample_aspect_ratio.num == 0 || myCodecCtx->sample_aspect_ratio.den == 0)
|
|
{
|
|
myPixelAspectRatio = 1.0f;
|
|
}
|
|
else
|
|
{
|
|
myPixelAspectRatio =
|
|
float(myCodecCtx->sample_aspect_ratio.num) / float(myCodecCtx->sample_aspect_ratio.den);
|
|
}
|
|
}
|
|
|
|
#if FFMPEG_HAVE_AVCODEC_PARAMETERS
|
|
if (theStream.codecpar->codec_type == AVMEDIA_TYPE_VIDEO
|
|
&& (myCodecCtx->width <= 0 || myCodecCtx->height <= 0))
|
|
#else
|
|
#ifdef _MSC_VER
|
|
#pragma warning(push)
|
|
#pragma warning(disable : 4996) // deprecated declaration
|
|
#endif
|
|
if (theStream.codec->codec_type == AVMEDIA_TYPE_VIDEO
|
|
&& (myCodecCtx->width <= 0 || myCodecCtx->height <= 0))
|
|
#ifdef _MSC_VER
|
|
#pragma warning(pop)
|
|
#endif
|
|
#endif
|
|
{
|
|
Message::SendFail("FFmpeg: video stream has invalid dimensions");
|
|
Close();
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#else
|
|
(void)&theStream;
|
|
(void)thePtsStartBase;
|
|
(void)theNbThreads;
|
|
(void)theCodecId;
|
|
return false;
|
|
#endif
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
void Media_CodecContext::Close()
|
|
{
|
|
if (myCodecCtx != NULL)
|
|
{
|
|
#ifdef HAVE_FFMPEG
|
|
#if FFMPEG_NEW_API
|
|
avcodec_free_context(&myCodecCtx);
|
|
#else
|
|
avcodec_close(myCodecCtx);
|
|
av_free(myCodecCtx);
|
|
myCodecCtx = NULL;
|
|
#endif
|
|
#endif
|
|
}
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
void Media_CodecContext::Flush()
|
|
{
|
|
if (myCodecCtx != NULL)
|
|
{
|
|
#ifdef HAVE_FFMPEG
|
|
avcodec_flush_buffers(myCodecCtx);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
int Media_CodecContext::SizeX() const
|
|
{
|
|
#ifdef HAVE_FFMPEG
|
|
return (myCodecCtx != NULL) ? myCodecCtx->width : 0;
|
|
#else
|
|
return 0;
|
|
#endif
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
int Media_CodecContext::SizeY() const
|
|
{
|
|
#ifdef HAVE_FFMPEG
|
|
return (myCodecCtx != NULL) ? myCodecCtx->height : 0;
|
|
#else
|
|
return 0;
|
|
#endif
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
bool Media_CodecContext::CanProcessPacket(const Handle(Media_Packet)& thePacket) const
|
|
{
|
|
return !thePacket.IsNull() && myStreamIndex == thePacket->StreamIndex();
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
bool Media_CodecContext::SendPacket(const Handle(Media_Packet)& thePacket)
|
|
{
|
|
if (!CanProcessPacket(thePacket))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
#ifdef HAVE_FFMPEG
|
|
#if FFMPEG_HAVE_NEW_DECODE_API
|
|
const int aRes = avcodec_send_packet(myCodecCtx, thePacket->Packet());
|
|
if (aRes < 0 && aRes != AVERROR_EOF)
|
|
{
|
|
return false;
|
|
}
|
|
return true;
|
|
#else
|
|
// For older FFmpeg versions, fallback to older decode API if needed
|
|
const int aRes = avcodec_send_packet(myCodecCtx, thePacket->Packet());
|
|
if (aRes < 0 && aRes != AVERROR_EOF)
|
|
{
|
|
return false;
|
|
}
|
|
return true;
|
|
#endif
|
|
#else
|
|
return false;
|
|
#endif
|
|
}
|
|
|
|
//=================================================================================================
|
|
|
|
bool Media_CodecContext::ReceiveFrame(const Handle(Media_Frame)& theFrame)
|
|
{
|
|
if (theFrame.IsNull())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
#ifdef HAVE_FFMPEG
|
|
#if FFMPEG_HAVE_NEW_DECODE_API
|
|
const int aRes2 = avcodec_receive_frame(myCodecCtx, theFrame->ChangeFrame());
|
|
if (aRes2 < 0)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
const int64_t aPacketPts =
|
|
theFrame->BestEffortTimestamp() != AV_NOPTS_VALUE ? theFrame->BestEffortTimestamp() : 0;
|
|
const double aFramePts = double(aPacketPts) * myTimeBase - myPtsStartBase;
|
|
theFrame->SetPts(aFramePts);
|
|
return true;
|
|
#else
|
|
// For older FFmpeg, use the older decoding API
|
|
const int aRes2 = avcodec_receive_frame(myCodecCtx, theFrame->ChangeFrame());
|
|
if (aRes2 < 0)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
const int64_t aPacketPts =
|
|
theFrame->BestEffortTimestamp() != AV_NOPTS_VALUE ? theFrame->BestEffortTimestamp() : 0;
|
|
const double aFramePts = double(aPacketPts) * myTimeBase - myPtsStartBase;
|
|
theFrame->SetPts(aFramePts);
|
|
return true;
|
|
#endif
|
|
#else
|
|
return false;
|
|
#endif
|
|
}
|