Method copyCurImageToBuffer

YUViewLib/src/decoder/decoderFFmpeg.cpp:175–268

}

void decoderFFmpeg::copyCurImageToBuffer()
{
  if (!frame)
    return;

  //// get metadata
  // AVDictionaryWrapper dict = this->ff.get_metadata(frame);
  // QStringPairList values = this->ff.getDictionary_entries(dict, "", 0);

  if (this->rawFormat == video::RawFormat::YUV)
  {
    // At first get how many bytes we are going to write
    const auto pixFmt           = this->getPixelFormatYUV();
    const auto nrBytesPerSample = pixFmt.getBitsPerSample() <= 8 ? 1 : 2;
    const auto nrBytesY         = this->frameSize.width * this->frameSize.height * nrBytesPerSample;
    const auto nrBytesC         = this->frameSize.width / pixFmt.getSubsamplingHor() *
                          this->frameSize.height / pixFmt.getSubsamplingVer() * nrBytesPerSample;
    const auto nrBytes = nrBytesY + 2 * nrBytesC;

    // Is the output big enough?
    if (auto c = functions::clipToUnsigned(this->currentOutputBuffer.capacity()); c < nrBytes)
      this->currentOutputBuffer.resize(nrBytes);

    // Copy line by line. The linesize of the source may be larger than the width of the frame.
    // This may be because the frame buffer is (8) byte aligned. Also the internal decoded
    // resolution may be larger than the output frame size.
    for (unsigned plane = 0; plane < pixFmt.getNrPlanes(); plane++)
    {
      const auto component =
          (plane == 0) ? video::yuv::Component::Luma : video::yuv::Component::Chroma;
      auto       src         = frame.getData(plane);
      const auto srcLinesize = frame.getLineSize(plane);
      auto       dst         = this->currentOutputBuffer.data();
      if (plane > 0)
        dst += (nrBytesY + (plane - 1) * nrBytesC);
      const auto dstLinesize =
          this->frameSize.width / pixFmt.getSubsamplingHor(component) * nrBytesPerSample;
      const auto height = this->frameSize.height / pixFmt.getSubsamplingVer(component);
      for (unsigned y = 0; y < height; y++)
      {
        memcpy(dst, src, dstLinesize);
        dst += dstLinesize;
        src += srcLinesize;
      }
    }
  }
  else if (this->rawFormat == video::RawFormat::RGB)
  {
    const auto pixFmt           = this->getRGBPixelFormat();
    const auto nrBytesPerSample = pixFmt.getBitsPerSample() <= 8 ? 1 : 2;
    const auto nrBytesPerComponent =
        this->frameSize.width * this->frameSize.height * nrBytesPerSample;
    const auto nrBytes = nrBytesPerComponent * pixFmt.nrChannels();

    // Is the output big enough?
    if (auto c = functions::clipToUnsigned(this->currentOutputBuffer.capacity()); c < nrBytes)
      this->currentOutputBuffer.resize(nrBytes);