blob: 2d8c73dc4e629dc783abc6fc406e95a9d30b1d6c [file] [log] [blame]
// Copyright 2017 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 "services/shape_detection/detection_utils_win.h"
#include <utility>
#include "base/logging.h"
#include "base/numerics/checked_math.h"
#include "base/win/winrt_storage_util.h"
#include "third_party/skia/include/core/SkBitmap.h"
namespace shape_detection {
using Microsoft::WRL::ComPtr;
ComPtr<ISoftwareBitmap> CreateWinBitmapFromSkBitmap(
const SkBitmap& bitmap,
ISoftwareBitmapStatics* bitmap_factory) {
DCHECK(bitmap_factory);
DCHECK_EQ(bitmap.colorType(), kN32_SkColorType);
if (!base::CheckedNumeric<uint32_t>(bitmap.computeByteSize()).IsValid()) {
DLOG(ERROR) << "Data overflow.";
return nullptr;
}
// Create IBuffer from bitmap data.
ComPtr<ABI::Windows::Storage::Streams::IBuffer> buffer;
HRESULT hr = base::win::CreateIBufferFromData(
static_cast<uint8_t*>(bitmap.getPixels()),
static_cast<UINT32>(bitmap.computeByteSize()), &buffer);
if (FAILED(hr)) {
DLOG(ERROR) << "Create IBuffer from bitmap data failed: "
<< logging::SystemErrorCodeToString(hr);
return nullptr;
}
ComPtr<ISoftwareBitmap> win_bitmap;
#if SK_PMCOLOR_BYTE_ORDER(R, G, B, A)
const BitmapPixelFormat pixel_format =
ABI::Windows::Graphics::Imaging::BitmapPixelFormat_Rgba8;
#else
const BitmapPixelFormat pixel_format =
ABI::Windows::Graphics::Imaging::BitmapPixelFormat_Bgra8;
#endif
// Create ISoftwareBitmap from SKBitmap that is kN32_SkColorType and copy the
// IBuffer into it.
hr = bitmap_factory->CreateCopyFromBuffer(
buffer.Get(), pixel_format, bitmap.width(), bitmap.height(), &win_bitmap);
if (FAILED(hr)) {
DLOG(ERROR) << "Create ISoftwareBitmap from buffer failed: "
<< logging::SystemErrorCodeToString(hr);
return nullptr;
}
return win_bitmap;
}
ComPtr<ISoftwareBitmap> CreateWinBitmapWithPixelFormat(
const SkBitmap& bitmap,
ISoftwareBitmapStatics* bitmap_factory,
BitmapPixelFormat pixel_format) {
ComPtr<ISoftwareBitmap> win_bitmap =
CreateWinBitmapFromSkBitmap(bitmap, bitmap_factory);
// Convert Rgba8/Bgra8 to Gray8/Nv12 SoftwareBitmap.
ComPtr<ISoftwareBitmap> converted_bitmap;
const HRESULT hr = bitmap_factory->Convert(win_bitmap.Get(), pixel_format,
&converted_bitmap);
if (FAILED(hr)) {
DLOG(ERROR) << "Convert Rgba8/Bgra8 to Gray8/Nv12 failed: "
<< logging::SystemErrorCodeToString(hr);
return nullptr;
}
return converted_bitmap;
}
} // namespace shape_detection