blob: 4c8f1260a5fac65b4bca04e9de81e66e8430e9e6 [file] [log] [blame]
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "media/capture/video/blob_utils.h"
#include "media/base/video_frame.h"
#include "media/capture/video_capture_types.h"
#include "third_party/libyuv/include/libyuv.h"
#include "third_party/skia/include/core/SkImage.h"
#include "ui/gfx/codec/png_codec.h"
namespace media {
mojom::BlobPtr Blobify(const uint8_t* buffer,
const uint32_t bytesused,
const VideoCaptureFormat& capture_format) {
DCHECK(buffer);
DCHECK(bytesused);
DCHECK(capture_format.IsValid());
const VideoPixelFormat pixel_format = capture_format.pixel_format;
if (pixel_format == VideoPixelFormat::PIXEL_FORMAT_MJPEG) {
mojom::BlobPtr blob = mojom::Blob::New();
blob->data.resize(bytesused);
memcpy(blob->data.data(), buffer, bytesused);
blob->mime_type = "image/jpeg";
return blob;
}
uint32_t src_format;
if (pixel_format == VideoPixelFormat::PIXEL_FORMAT_UYVY)
src_format = libyuv::FOURCC_UYVY;
else if (pixel_format == VideoPixelFormat::PIXEL_FORMAT_YUY2)
src_format = libyuv::FOURCC_YUY2;
else if (pixel_format == VideoPixelFormat::PIXEL_FORMAT_I420)
src_format = libyuv::FOURCC_I420;
else if (pixel_format == VideoPixelFormat::PIXEL_FORMAT_RGB24)
src_format = libyuv::FOURCC_24BG;
else
return nullptr;
const gfx::Size frame_size = capture_format.frame_size;
// PNGCodec does not support YUV formats, convert to a temporary ARGB buffer.
std::unique_ptr<uint8_t[]> tmp_argb(
new uint8_t[VideoFrame::AllocationSize(PIXEL_FORMAT_ARGB, frame_size)]);
if (ConvertToARGB(buffer, bytesused, tmp_argb.get(), frame_size.width() * 4,
0 /* crop_x_pos */, 0 /* crop_y_pos */, frame_size.width(),
frame_size.height(), frame_size.width(),
frame_size.height(), libyuv::RotationMode::kRotate0,
src_format) != 0) {
return nullptr;
}
mojom::BlobPtr blob = mojom::Blob::New();
const gfx::PNGCodec::ColorFormat codec_color_format =
(kN32_SkColorType == kRGBA_8888_SkColorType) ? gfx::PNGCodec::FORMAT_RGBA
: gfx::PNGCodec::FORMAT_BGRA;
const bool result = gfx::PNGCodec::Encode(
tmp_argb.get(), codec_color_format, frame_size, frame_size.width() * 4,
true /* discard_transparency */, std::vector<gfx::PNGCodec::Comment>(),
&blob->data);
DCHECK(result);
blob->mime_type = "image/png";
return blob;
}
} // namespace media