Method copyImgToByteArray

YUViewLib/src/decoder/decoderLibde265.cpp:470–528

void decoderLibde265::copyImgToByteArray(const de265_image *src, byteArrayAligned &dst)
#else
void decoderLibde265::copyImgToByteArray(const de265_image *src, QByteArray &dst)
#endif
{
  // How many image planes are there?
  auto cMode    = this->lib.de265_get_chroma_format(src);
  int  nrPlanes = (cMode == de265_chroma_mono) ? 1 : 3;

  // At first get how many bytes we are going to write
  int nrBytes = 0;
  int stride;
  for (int c = 0; c < nrPlanes; c++)
  {
    int width            = this->lib.de265_get_image_width(src, c);
    int height           = this->lib.de265_get_image_height(src, c);
    int nrBytesPerSample = (this->lib.de265_get_bits_per_pixel(src, c) > 8) ? 2 : 1;

    nrBytes += width * height * nrBytesPerSample;
  }

  DEBUG_LIBDE265("decoderLibde265::copyImgToByteArray nrBytes %d", nrBytes);

  // Is the output big enough?
  if (dst.capacity() < nrBytes)
    dst.resize(nrBytes);

  uint8_t *dst_c = (uint8_t *)dst.data();

  // We can now copy from src to dst
  for (int c = 0; c < nrPlanes; c++)
  {
    const int    width            = this->lib.de265_get_image_width(src, c);
    const int    height           = this->lib.de265_get_image_height(src, c);
    const int    nrBytesPerSample = (this->lib.de265_get_bits_per_pixel(src, c) > 8) ? 2 : 1;
    const size_t widthInBytes     = width * nrBytesPerSample;

    const uint8_t *img_c = nullptr;
    if (decodeSignal == 0)
      img_c = this->lib.de265_get_image_plane(src, c, &stride);
    else if (decodeSignal == 1)
      img_c = this->lib.de265_internals_get_image_plane(
          src, DE265_INTERNALS_DECODER_PARAM_SAVE_PREDICTION, c, &stride);
    else if (decodeSignal == 2)
      img_c = this->lib.de265_internals_get_image_plane(
          src, DE265_INTERNALS_DECODER_PARAM_SAVE_RESIDUAL, c, &stride);
    else if (decodeSignal == 3)
      img_c = this->lib.de265_internals_get_image_plane(
          src, DE265_INTERNALS_DECODER_PARAM_SAVE_TR_COEFF, c, &stride);

    if (img_c == nullptr)
      return;

    for (int y = 0; y < height; y++)
    {
      memcpy(dst_c, img_c, widthInBytes);
      img_c += stride;
      dst_c += widthInBytes;
    }
  }