Method ReadNextFrameWithoutIndex

dali/operators/video/frames_decoder_gpu.cc:648–758

}

bool FramesDecoderGpu::ReadNextFrameWithoutIndex(uint8_t *data) {
  current_copy_to_output_ = data != nullptr;
  current_frame_output_ = data;

  int64_t frame_to_return_index = -1;

  // Handle the case, when packet has more frames that we have empty spots
  // in the buffer.
  // If so, we need to return frame from the buffer before sending last packet.
  if (frame_index_if_no_pts_ != 0) {
    if (NumEmptySpots() < (piped_pts_.size())) {
      for (size_t i = 0; i < frame_buffer_.size(); ++i) {
        if (frame_buffer_[i].pts_ == NextFrameIdx()) {
          frame_to_return_index = i;
          break;
        }
      }
    }
  }

  // Initial fill of the buffer
  frame_returned_ = false;
  while (
    // as we may enlarge the buffer make sure to not decode more than num_decode_surfaces_ frames
    NumEmptySpots() > frame_buffer_.size() - num_decode_surfaces_ &&
    more_frames_to_decode_ &&
    !frame_returned_ &&
    frame_to_return_index == -1) {
    int ret = av_read_frame(ctx_, packet_);
    auto packet = AVPacketScope(packet_, av_packet_unref);
    if (ret == 0) {
      if (!SendFrameToParser()) {
        continue;
      }
    } else {
      // Handle the case, when last packet has more frames that we have empty spots
      // in the buffer.
      // If so, we need to return frame from the buffer before sending last packet.
      if (frame_index_if_no_pts_ != 0) {
        if (NumEmptySpots() < (piped_pts_.size())) {
          for (size_t i = 0; i < frame_buffer_.size(); ++i) {
            if (frame_buffer_[i].pts_ == NextFrameIdx()) {
              frame_to_return_index = i;
              break;
            }
          }
        }
      }

      if (frame_to_return_index == -1) {
        SendLastPacket();
        more_frames_to_decode_ = false;
      } else {
        break;
      }
    }
  }