bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 1 | /* |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 2 | * Copyright 2011 Google Inc. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 6 | */ |
| 7 | |
bungeman@google.com | 8530296 | 2013-10-11 20:04:08 +0000 | [diff] [blame] | 8 | #include "SkTypes.h" |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 9 | |
Mike Klein | 8f11d4d | 2018-01-24 12:42:55 -0500 | [diff] [blame] | 10 | #if defined(SK_BUILD_FOR_WIN) |
mtklein | 1ee7651 | 2015-11-02 10:20:27 -0800 | [diff] [blame] | 11 | |
bungeman@google.com | 8530296 | 2013-10-11 20:04:08 +0000 | [diff] [blame] | 12 | // Workaround for: |
| 13 | // http://connect.microsoft.com/VisualStudio/feedback/details/621653/ |
| 14 | // http://crbug.com/225822 |
| 15 | // In VS2010 both intsafe.h and stdint.h define the following without guards. |
| 16 | // SkTypes brought in windows.h and stdint.h and the following defines are |
| 17 | // not used by this file. However, they may be re-introduced by wincodec.h. |
| 18 | #undef INT8_MIN |
| 19 | #undef INT16_MIN |
| 20 | #undef INT32_MIN |
| 21 | #undef INT64_MIN |
| 22 | #undef INT8_MAX |
| 23 | #undef UINT8_MAX |
| 24 | #undef INT16_MAX |
| 25 | #undef UINT16_MAX |
| 26 | #undef INT32_MAX |
| 27 | #undef UINT32_MAX |
| 28 | #undef INT64_MAX |
| 29 | #undef UINT64_MAX |
| 30 | |
bungeman@google.com | 9df621d | 2011-06-23 21:43:52 +0000 | [diff] [blame] | 31 | #include "SkAutoCoInitialize.h" |
Hal Canary | 95e3c05 | 2017-01-11 12:44:43 -0500 | [diff] [blame] | 32 | #include "SkAutoMalloc.h" |
msarett | d442d32 | 2016-03-25 11:08:14 -0700 | [diff] [blame] | 33 | #include "SkBitmap.h" |
Hal Canary | db68301 | 2016-11-23 08:55:18 -0700 | [diff] [blame] | 34 | #include "SkImageEncoderPriv.h" |
bungeman@google.com | 9df621d | 2011-06-23 21:43:52 +0000 | [diff] [blame] | 35 | #include "SkIStream.h" |
Hal Canary | 95e3c05 | 2017-01-11 12:44:43 -0500 | [diff] [blame] | 36 | #include "SkImageEncoder.h" |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 37 | #include "SkStream.h" |
bungeman@google.com | 9df621d | 2011-06-23 21:43:52 +0000 | [diff] [blame] | 38 | #include "SkTScopedComPtr.h" |
Ben Wagner | 4d1955c | 2017-03-10 13:08:15 -0500 | [diff] [blame] | 39 | #include "SkTemplates.h" |
robertphillips@google.com | 59bfb12 | 2012-11-08 15:06:52 +0000 | [diff] [blame] | 40 | #include "SkUnPreMultiply.h" |
Hal Canary | 95e3c05 | 2017-01-11 12:44:43 -0500 | [diff] [blame] | 41 | #include <wincodec.h> |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 42 | |
bungeman@google.com | c18143e | 2013-01-11 20:02:32 +0000 | [diff] [blame] | 43 | //All Windows SDKs back to XPSP2 export the CLSID_WICImagingFactory symbol. |
| 44 | //In the Windows8 SDK the CLSID_WICImagingFactory symbol is still exported |
| 45 | //but CLSID_WICImagingFactory is then #defined to CLSID_WICImagingFactory2. |
| 46 | //Undo this #define if it has been done so that we link against the symbols |
| 47 | //we intended to link against on all SDKs. |
| 48 | #if defined(CLSID_WICImagingFactory) |
| 49 | #undef CLSID_WICImagingFactory |
| 50 | #endif |
| 51 | |
Hal Canary | 1fcc404 | 2016-11-30 17:07:59 -0500 | [diff] [blame] | 52 | bool SkEncodeImageWithWIC(SkWStream* stream, const SkPixmap& pixmap, |
| 53 | SkEncodedImageFormat format, int quality) { |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 54 | GUID type; |
Hal Canary | 1fcc404 | 2016-11-30 17:07:59 -0500 | [diff] [blame] | 55 | switch (format) { |
Hal Canary | db68301 | 2016-11-23 08:55:18 -0700 | [diff] [blame] | 56 | case SkEncodedImageFormat::kJPEG: |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 57 | type = GUID_ContainerFormatJpeg; |
| 58 | break; |
Hal Canary | db68301 | 2016-11-23 08:55:18 -0700 | [diff] [blame] | 59 | case SkEncodedImageFormat::kPNG: |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 60 | type = GUID_ContainerFormatPng; |
| 61 | break; |
| 62 | default: |
| 63 | return false; |
| 64 | } |
Hal Canary | 1fcc404 | 2016-11-30 17:07:59 -0500 | [diff] [blame] | 65 | SkBitmap bitmapOrig; |
| 66 | if (!bitmapOrig.installPixels(pixmap)) { |
| 67 | return false; |
| 68 | } |
| 69 | bitmapOrig.setImmutable(); |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 70 | |
msarett | 36c38cb | 2016-08-15 18:52:17 -0700 | [diff] [blame] | 71 | // First convert to BGRA if necessary. |
| 72 | SkBitmap bitmap; |
Matt Sarett | 68b8e3d | 2017-04-28 11:15:22 -0400 | [diff] [blame] | 73 | if (!bitmap.tryAllocPixels(bitmapOrig.info().makeColorType(kBGRA_8888_SkColorType)) || |
| 74 | !bitmapOrig.readPixels(bitmap.info(), bitmap.getPixels(), bitmap.rowBytes(), 0, 0)) |
| 75 | { |
msarett | 36c38cb | 2016-08-15 18:52:17 -0700 | [diff] [blame] | 76 | return false; |
bungeman@google.com | 22b4950 | 2011-08-01 19:37:43 +0000 | [diff] [blame] | 77 | } |
| 78 | |
msarett | 36c38cb | 2016-08-15 18:52:17 -0700 | [diff] [blame] | 79 | // WIC expects unpremultiplied pixels. Unpremultiply if necessary. |
| 80 | if (kPremul_SkAlphaType == bitmap.alphaType()) { |
| 81 | uint8_t* pixels = reinterpret_cast<uint8_t*>(bitmap.getPixels()); |
| 82 | for (int y = 0; y < bitmap.height(); ++y) { |
| 83 | for (int x = 0; x < bitmap.width(); ++x) { |
| 84 | uint8_t* bytes = pixels + y * bitmap.rowBytes() + x * bitmap.bytesPerPixel(); |
robertphillips@google.com | 59bfb12 | 2012-11-08 15:06:52 +0000 | [diff] [blame] | 85 | SkPMColor* src = reinterpret_cast<SkPMColor*>(bytes); |
| 86 | SkColor* dst = reinterpret_cast<SkColor*>(bytes); |
robertphillips@google.com | 59bfb12 | 2012-11-08 15:06:52 +0000 | [diff] [blame] | 87 | *dst = SkUnPreMultiply::PMColorToColor(*src); |
| 88 | } |
| 89 | } |
| 90 | } |
| 91 | |
msarett | 36c38cb | 2016-08-15 18:52:17 -0700 | [diff] [blame] | 92 | // Finally, if we are performing a jpeg encode, we must convert to BGR. |
| 93 | void* pixels = bitmap.getPixels(); |
| 94 | size_t rowBytes = bitmap.rowBytes(); |
| 95 | SkAutoMalloc pixelStorage; |
| 96 | WICPixelFormatGUID formatDesired = GUID_WICPixelFormat32bppBGRA; |
Hal Canary | 1fcc404 | 2016-11-30 17:07:59 -0500 | [diff] [blame] | 97 | if (SkEncodedImageFormat::kJPEG == format) { |
msarett | 36c38cb | 2016-08-15 18:52:17 -0700 | [diff] [blame] | 98 | formatDesired = GUID_WICPixelFormat24bppBGR; |
| 99 | rowBytes = SkAlign4(bitmap.width() * 3); |
| 100 | pixelStorage.reset(rowBytes * bitmap.height()); |
| 101 | for (int y = 0; y < bitmap.height(); y++) { |
| 102 | uint8_t* dstRow = SkTAddOffset<uint8_t>(pixelStorage.get(), y * rowBytes); |
| 103 | for (int x = 0; x < bitmap.width(); x++) { |
| 104 | uint32_t bgra = *bitmap.getAddr32(x, y); |
| 105 | dstRow[0] = (uint8_t) (bgra >> 0); |
| 106 | dstRow[1] = (uint8_t) (bgra >> 8); |
| 107 | dstRow[2] = (uint8_t) (bgra >> 16); |
| 108 | dstRow += 3; |
| 109 | } |
| 110 | } |
| 111 | |
| 112 | pixels = pixelStorage.get(); |
| 113 | } |
| 114 | |
| 115 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 116 | //Initialize COM. |
bungeman@google.com | 2e2f3f5 | 2011-09-16 15:37:20 +0000 | [diff] [blame] | 117 | SkAutoCoInitialize scopedCo; |
| 118 | if (!scopedCo.succeeded()) { |
| 119 | return false; |
| 120 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 121 | |
bungeman@google.com | 2e2f3f5 | 2011-09-16 15:37:20 +0000 | [diff] [blame] | 122 | HRESULT hr = S_OK; |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 123 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 124 | //Create Windows Imaging Component ImagingFactory. |
bungeman@google.com | 9df621d | 2011-06-23 21:43:52 +0000 | [diff] [blame] | 125 | SkTScopedComPtr<IWICImagingFactory> piImagingFactory; |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 126 | if (SUCCEEDED(hr)) { |
| 127 | hr = CoCreateInstance( |
| 128 | CLSID_WICImagingFactory |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 129 | , nullptr |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 130 | , CLSCTX_INPROC_SERVER |
| 131 | , IID_PPV_ARGS(&piImagingFactory) |
| 132 | ); |
| 133 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 134 | |
bungeman@google.com | 9df621d | 2011-06-23 21:43:52 +0000 | [diff] [blame] | 135 | //Convert the SkWStream to an IStream. |
| 136 | SkTScopedComPtr<IStream> piStream; |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 137 | if (SUCCEEDED(hr)) { |
bungeman@google.com | 9df621d | 2011-06-23 21:43:52 +0000 | [diff] [blame] | 138 | hr = SkWIStream::CreateFromSkWStream(stream, &piStream); |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 139 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 140 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 141 | //Create an encode of the appropriate type. |
bungeman@google.com | 9df621d | 2011-06-23 21:43:52 +0000 | [diff] [blame] | 142 | SkTScopedComPtr<IWICBitmapEncoder> piEncoder; |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 143 | if (SUCCEEDED(hr)) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 144 | hr = piImagingFactory->CreateEncoder(type, nullptr, &piEncoder); |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 145 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 146 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 147 | if (SUCCEEDED(hr)) { |
| 148 | hr = piEncoder->Initialize(piStream.get(), WICBitmapEncoderNoCache); |
| 149 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 150 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 151 | //Create a the frame. |
bungeman@google.com | 9df621d | 2011-06-23 21:43:52 +0000 | [diff] [blame] | 152 | SkTScopedComPtr<IWICBitmapFrameEncode> piBitmapFrameEncode; |
| 153 | SkTScopedComPtr<IPropertyBag2> piPropertybag; |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 154 | if (SUCCEEDED(hr)) { |
| 155 | hr = piEncoder->CreateNewFrame(&piBitmapFrameEncode, &piPropertybag); |
| 156 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 157 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 158 | if (SUCCEEDED(hr)) { |
Chris Dalton | 1ef8094 | 2017-12-04 12:01:30 -0700 | [diff] [blame] | 159 | PROPBAG2 name; |
| 160 | memset(&name, 0, sizeof(name)); |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 161 | name.dwType = PROPBAG2_TYPE_DATA; |
| 162 | name.vt = VT_R4; |
Mike Klein | c722f79 | 2017-07-31 11:57:21 -0400 | [diff] [blame] | 163 | name.pstrName = const_cast<LPOLESTR>(L"ImageQuality"); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 164 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 165 | VARIANT value; |
| 166 | VariantInit(&value); |
| 167 | value.vt = VT_R4; |
| 168 | value.fltVal = (FLOAT)(quality / 100.0); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 169 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 170 | //Ignore result code. |
| 171 | // This returns E_FAIL if the named property is not in the bag. |
| 172 | //TODO(bungeman) enumerate the properties, |
| 173 | // write and set hr iff property exists. |
| 174 | piPropertybag->Write(1, &name, &value); |
| 175 | } |
| 176 | if (SUCCEEDED(hr)) { |
| 177 | hr = piBitmapFrameEncode->Initialize(piPropertybag.get()); |
| 178 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 179 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 180 | //Set the size of the frame. |
msarett | 36c38cb | 2016-08-15 18:52:17 -0700 | [diff] [blame] | 181 | const UINT width = bitmap.width(); |
| 182 | const UINT height = bitmap.height(); |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 183 | if (SUCCEEDED(hr)) { |
| 184 | hr = piBitmapFrameEncode->SetSize(width, height); |
| 185 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 186 | |
msarett | e820dfe | 2016-03-18 12:13:47 -0700 | [diff] [blame] | 187 | //Set the pixel format of the frame. If native encoded format cannot match BGRA, |
| 188 | //it will choose the closest pixel format that it supports. |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 189 | WICPixelFormatGUID formatGUID = formatDesired; |
| 190 | if (SUCCEEDED(hr)) { |
| 191 | hr = piBitmapFrameEncode->SetPixelFormat(&formatGUID); |
| 192 | } |
msarett | 60316a9 | 2016-03-21 13:35:43 -0700 | [diff] [blame] | 193 | if (SUCCEEDED(hr)) { |
| 194 | //Be sure the image format is the one requested. |
| 195 | hr = IsEqualGUID(formatGUID, formatDesired) ? S_OK : E_FAIL; |
| 196 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 197 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 198 | //Write the pixels into the frame. |
| 199 | if (SUCCEEDED(hr)) { |
msarett | 36c38cb | 2016-08-15 18:52:17 -0700 | [diff] [blame] | 200 | hr = piBitmapFrameEncode->WritePixels(height, |
| 201 | (UINT) rowBytes, |
| 202 | (UINT) rowBytes * height, |
| 203 | reinterpret_cast<BYTE*>(pixels)); |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 204 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 205 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 206 | if (SUCCEEDED(hr)) { |
| 207 | hr = piBitmapFrameEncode->Commit(); |
| 208 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 209 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 210 | if (SUCCEEDED(hr)) { |
| 211 | hr = piEncoder->Commit(); |
| 212 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 213 | |
bungeman@google.com | 242bb89 | 2011-06-22 20:42:34 +0000 | [diff] [blame] | 214 | return SUCCEEDED(hr); |
| 215 | } |
| 216 | |
Mike Klein | 8f11d4d | 2018-01-24 12:42:55 -0500 | [diff] [blame] | 217 | #endif // defined(SK_BUILD_FOR_WIN) |