OCCT

// 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(nullptr),
      myCodec(nullptr),
      myPtsStartBase(0.0),
      myPtsStartStream(0.0),
      myTimeBase(1.0),
      myStreamIndex(0),
      myPixelAspectRatio(1.0f)
{
#ifdef HAVE_FFMPEG
  myCodecCtx = avcodec_alloc_context3(nullptr);
#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 == nullptr)
  {
    Message::Send("FFmpeg: unable to find decoder", Message_Fail);
    Close();
    return false;
  }

  myCodecCtx->codec_id = aCodecId;
  AVDictionary* anOpts = nullptr;
  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 != nullptr)
  {
#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 != nullptr)
  {
#ifdef HAVE_FFMPEG
    avcodec_flush_buffers(myCodecCtx);
#endif
  }
}

//=================================================================================================

int Media_CodecContext::SizeX() const
{
#ifdef HAVE_FFMPEG
  return (myCodecCtx != nullptr) ? myCodecCtx->width : 0;
#else
  return 0;
#endif
}

//=================================================================================================

int Media_CodecContext::SizeY() const
{
#ifdef HAVE_FFMPEG
  return (myCodecCtx != nullptr) ? myCodecCtx->height : 0;
#else
  return 0;
#endif
}

//=================================================================================================

bool Media_CodecContext::CanProcessPacket(const occ::handle<Media_Packet>& thePacket) const
{
  return !thePacket.IsNull() && myStreamIndex == thePacket->StreamIndex();
}

//=================================================================================================

bool Media_CodecContext::SendPacket(const occ::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());
  return aRes >= 0 || aRes == AVERROR_EOF;
  #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 occ::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
}