2 * Copyright 2009 Vincent Povirk for CodeWeavers
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 #include "wine/port.h"
35 #include "wincodecs_private.h"
37 #include "wine/debug.h"
38 #include "wine/library.h"
40 WINE_DEFAULT_DEBUG_CHANNEL(wincodecs
);
44 static void *libpng_handle
;
45 #define MAKE_FUNCPTR(f) static typeof(f) * p##f
46 MAKE_FUNCPTR(png_create_read_struct
);
47 MAKE_FUNCPTR(png_create_info_struct
);
48 MAKE_FUNCPTR(png_create_write_struct
);
49 MAKE_FUNCPTR(png_destroy_read_struct
);
50 MAKE_FUNCPTR(png_destroy_write_struct
);
51 MAKE_FUNCPTR(png_error
);
52 MAKE_FUNCPTR(png_get_bit_depth
);
53 MAKE_FUNCPTR(png_get_color_type
);
54 MAKE_FUNCPTR(png_get_error_ptr
);
55 MAKE_FUNCPTR(png_get_image_height
);
56 MAKE_FUNCPTR(png_get_image_width
);
57 MAKE_FUNCPTR(png_get_io_ptr
);
58 MAKE_FUNCPTR(png_get_pHYs
);
59 MAKE_FUNCPTR(png_get_progressive_ptr
);
60 MAKE_FUNCPTR(png_get_PLTE
);
61 MAKE_FUNCPTR(png_get_tRNS
);
62 MAKE_FUNCPTR(png_process_data
);
63 MAKE_FUNCPTR(png_progressive_combine_row
);
64 MAKE_FUNCPTR(png_set_bgr
);
65 MAKE_FUNCPTR(png_set_error_fn
);
66 #if HAVE_PNG_SET_EXPAND_GRAY_1_2_4_TO_8
67 MAKE_FUNCPTR(png_set_expand_gray_1_2_4_to_8
);
69 MAKE_FUNCPTR(png_set_gray_1_2_4_to_8
);
71 MAKE_FUNCPTR(png_set_filler
);
72 MAKE_FUNCPTR(png_set_gray_to_rgb
);
73 MAKE_FUNCPTR(png_set_IHDR
);
74 MAKE_FUNCPTR(png_set_pHYs
);
75 MAKE_FUNCPTR(png_set_progressive_read_fn
);
76 MAKE_FUNCPTR(png_set_read_fn
);
77 MAKE_FUNCPTR(png_set_strip_16
);
78 MAKE_FUNCPTR(png_set_tRNS_to_alpha
);
79 MAKE_FUNCPTR(png_set_write_fn
);
80 MAKE_FUNCPTR(png_read_end
);
81 MAKE_FUNCPTR(png_read_image
);
82 MAKE_FUNCPTR(png_read_info
);
83 MAKE_FUNCPTR(png_read_update_info
);
84 MAKE_FUNCPTR(png_write_end
);
85 MAKE_FUNCPTR(png_write_info
);
86 MAKE_FUNCPTR(png_write_rows
);
89 static void *load_libpng(void)
91 if((libpng_handle
= wine_dlopen(SONAME_LIBPNG
, RTLD_NOW
, NULL
, 0)) != NULL
) {
93 #define LOAD_FUNCPTR(f) \
94 if((p##f = wine_dlsym(libpng_handle, #f, NULL, 0)) == NULL) { \
95 libpng_handle = NULL; \
98 LOAD_FUNCPTR(png_create_read_struct
);
99 LOAD_FUNCPTR(png_create_info_struct
);
100 LOAD_FUNCPTR(png_create_write_struct
);
101 LOAD_FUNCPTR(png_destroy_read_struct
);
102 LOAD_FUNCPTR(png_destroy_write_struct
);
103 LOAD_FUNCPTR(png_error
);
104 LOAD_FUNCPTR(png_get_bit_depth
);
105 LOAD_FUNCPTR(png_get_color_type
);
106 LOAD_FUNCPTR(png_get_error_ptr
);
107 LOAD_FUNCPTR(png_get_image_height
);
108 LOAD_FUNCPTR(png_get_image_width
);
109 LOAD_FUNCPTR(png_get_io_ptr
);
110 LOAD_FUNCPTR(png_get_pHYs
);
111 LOAD_FUNCPTR(png_get_progressive_ptr
);
112 LOAD_FUNCPTR(png_get_PLTE
);
113 LOAD_FUNCPTR(png_get_tRNS
);
114 LOAD_FUNCPTR(png_process_data
);
115 LOAD_FUNCPTR(png_progressive_combine_row
);
116 LOAD_FUNCPTR(png_set_bgr
);
117 LOAD_FUNCPTR(png_set_error_fn
);
118 #if HAVE_PNG_SET_EXPAND_GRAY_1_2_4_TO_8
119 LOAD_FUNCPTR(png_set_expand_gray_1_2_4_to_8
);
121 LOAD_FUNCPTR(png_set_gray_1_2_4_to_8
);
123 LOAD_FUNCPTR(png_set_filler
);
124 LOAD_FUNCPTR(png_set_gray_to_rgb
);
125 LOAD_FUNCPTR(png_set_IHDR
);
126 LOAD_FUNCPTR(png_set_pHYs
);
127 LOAD_FUNCPTR(png_set_progressive_read_fn
);
128 LOAD_FUNCPTR(png_set_read_fn
);
129 LOAD_FUNCPTR(png_set_strip_16
);
130 LOAD_FUNCPTR(png_set_tRNS_to_alpha
);
131 LOAD_FUNCPTR(png_set_write_fn
);
132 LOAD_FUNCPTR(png_read_end
);
133 LOAD_FUNCPTR(png_read_image
);
134 LOAD_FUNCPTR(png_read_info
);
135 LOAD_FUNCPTR(png_read_update_info
);
136 LOAD_FUNCPTR(png_write_end
);
137 LOAD_FUNCPTR(png_write_info
);
138 LOAD_FUNCPTR(png_write_rows
);
142 return libpng_handle
;
145 static void user_error_fn(png_structp png_ptr
, png_const_charp error_message
)
149 /* This uses setjmp/longjmp just like the default. We can't use the
150 * default because there's no way to access the jmp buffer in the png_struct
151 * that works in 1.2 and 1.4 and allows us to dynamically load libpng. */
152 WARN("PNG error: %s\n", debugstr_a(error_message
));
153 pjmpbuf
= ppng_get_error_ptr(png_ptr
);
154 longjmp(*pjmpbuf
, 1);
157 static void user_warning_fn(png_structp png_ptr
, png_const_charp warning_message
)
159 WARN("PNG warning: %s\n", debugstr_a(warning_message
));
163 IWICBitmapDecoder IWICBitmapDecoder_iface
;
164 IWICBitmapFrameDecode IWICBitmapFrameDecode_iface
;
172 const WICPixelFormatGUID
*format
;
174 CRITICAL_SECTION lock
; /* must be held when png structures are accessed or initialized is set */
177 static inline PngDecoder
*impl_from_IWICBitmapDecoder(IWICBitmapDecoder
*iface
)
179 return CONTAINING_RECORD(iface
, PngDecoder
, IWICBitmapDecoder_iface
);
182 static inline PngDecoder
*impl_from_IWICBitmapFrameDecode(IWICBitmapFrameDecode
*iface
)
184 return CONTAINING_RECORD(iface
, PngDecoder
, IWICBitmapFrameDecode_iface
);
187 static const IWICBitmapFrameDecodeVtbl PngDecoder_FrameVtbl
;
189 static HRESULT WINAPI
PngDecoder_QueryInterface(IWICBitmapDecoder
*iface
, REFIID iid
,
192 PngDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
193 TRACE("(%p,%s,%p)\n", iface
, debugstr_guid(iid
), ppv
);
195 if (!ppv
) return E_INVALIDARG
;
197 if (IsEqualIID(&IID_IUnknown
, iid
) || IsEqualIID(&IID_IWICBitmapDecoder
, iid
))
204 return E_NOINTERFACE
;
207 IUnknown_AddRef((IUnknown
*)*ppv
);
211 static ULONG WINAPI
PngDecoder_AddRef(IWICBitmapDecoder
*iface
)
213 PngDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
214 ULONG ref
= InterlockedIncrement(&This
->ref
);
216 TRACE("(%p) refcount=%u\n", iface
, ref
);
221 static ULONG WINAPI
PngDecoder_Release(IWICBitmapDecoder
*iface
)
223 PngDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
224 ULONG ref
= InterlockedDecrement(&This
->ref
);
226 TRACE("(%p) refcount=%u\n", iface
, ref
);
231 ppng_destroy_read_struct(&This
->png_ptr
, &This
->info_ptr
, NULL
);
232 This
->lock
.DebugInfo
->Spare
[0] = 0;
233 DeleteCriticalSection(&This
->lock
);
234 HeapFree(GetProcessHeap(), 0, This
->image_bits
);
235 HeapFree(GetProcessHeap(), 0, This
);
241 static HRESULT WINAPI
PngDecoder_QueryCapability(IWICBitmapDecoder
*iface
, IStream
*pIStream
,
242 DWORD
*pdwCapability
)
244 FIXME("(%p,%p,%p): stub\n", iface
, pIStream
, pdwCapability
);
248 static void info_callback(png_structp png_ptr
, png_infop info
)
250 int color_type
, bit_depth
;
253 png_uint_32 transparency
;
254 png_color_16p trans_values
;
259 This
= (PngDecoder
*) ppng_get_progressive_ptr(png_ptr
);
261 /* choose a pixel format */
262 color_type
= ppng_get_color_type(This
->png_ptr
, This
->info_ptr
);
263 bit_depth
= ppng_get_bit_depth(This
->png_ptr
, This
->info_ptr
);
265 /* check for color-keyed alpha */
266 transparency
= ppng_get_tRNS(This
->png_ptr
, This
->info_ptr
, &trans
, &num_trans
, &trans_values
);
268 if (transparency
&& color_type
!= PNG_COLOR_TYPE_PALETTE
)
271 if (color_type
== PNG_COLOR_TYPE_GRAY
)
275 #if HAVE_PNG_SET_EXPAND_GRAY_1_2_4_TO_8
276 ppng_set_expand_gray_1_2_4_to_8(This
->png_ptr
);
278 ppng_set_gray_1_2_4_to_8(This
->png_ptr
);
282 ppng_set_gray_to_rgb(This
->png_ptr
);
284 ppng_set_tRNS_to_alpha(This
->png_ptr
);
285 color_type
= PNG_COLOR_TYPE_RGB_ALPHA
;
290 case PNG_COLOR_TYPE_GRAY
:
291 This
->bpp
= bit_depth
;
294 case 1: This
->format
= &GUID_WICPixelFormatBlackWhite
; break;
295 case 2: This
->format
= &GUID_WICPixelFormat2bppGray
; break;
296 case 4: This
->format
= &GUID_WICPixelFormat4bppGray
; break;
297 case 8: This
->format
= &GUID_WICPixelFormat8bppGray
; break;
298 case 16: This
->format
= &GUID_WICPixelFormat16bppGray
; break;
300 ERR("invalid grayscale bit depth: %i\n", bit_depth
);
305 case PNG_COLOR_TYPE_GRAY_ALPHA
:
306 /* WIC does not support grayscale alpha formats so use RGBA */
307 ppng_set_gray_to_rgb(This
->png_ptr
);
308 case PNG_COLOR_TYPE_RGB_ALPHA
:
309 This
->bpp
= bit_depth
* 4;
313 ppng_set_bgr(This
->png_ptr
);
314 This
->format
= &GUID_WICPixelFormat32bppBGRA
;
316 case 16: This
->format
= &GUID_WICPixelFormat64bppRGBA
; break;
318 ERR("invalid RGBA bit depth: %i\n", bit_depth
);
323 case PNG_COLOR_TYPE_PALETTE
:
324 This
->bpp
= bit_depth
;
327 case 1: This
->format
= &GUID_WICPixelFormat1bppIndexed
; break;
328 case 2: This
->format
= &GUID_WICPixelFormat2bppIndexed
; break;
329 case 4: This
->format
= &GUID_WICPixelFormat4bppIndexed
; break;
330 case 8: This
->format
= &GUID_WICPixelFormat8bppIndexed
; break;
332 ERR("invalid indexed color bit depth: %i\n", bit_depth
);
337 case PNG_COLOR_TYPE_RGB
:
338 This
->bpp
= bit_depth
* 3;
342 ppng_set_bgr(This
->png_ptr
);
343 This
->format
= &GUID_WICPixelFormat24bppBGR
;
345 case 16: This
->format
= &GUID_WICPixelFormat48bppRGB
; break;
347 ERR("invalid RGB color bit depth: %i\n", bit_depth
);
353 ERR("invalid color type %i\n", color_type
);
358 This
->width
= ppng_get_image_width(This
->png_ptr
, This
->info_ptr
);
359 This
->height
= ppng_get_image_height(This
->png_ptr
, This
->info_ptr
);
360 This
->stride
= This
->width
* This
->bpp
;
361 image_size
= This
->stride
* This
->height
;
363 This
->image_bits
= HeapAlloc(GetProcessHeap(), 0, image_size
);
364 if (!This
->image_bits
)
370 ppng_read_update_info(This
->png_ptr
, This
->info_ptr
);
371 This
->initialized
= TRUE
;
375 ERR("PNG info parsing failed, hr=0x%08X\n", hr
);
378 static void row_callback(png_structp png_ptr
, png_bytep new_row
,
379 png_uint_32 row_num
, int pass
)
384 This
= (PngDecoder
*) ppng_get_progressive_ptr(png_ptr
);
385 if (!This
->initialized
)
388 old_row
= ((png_bytep
)This
->image_bits
) + row_num
*This
->stride
;
389 ppng_progressive_combine_row(png_ptr
, old_row
, new_row
);
392 static void end_callback(png_structp png_ptr
, png_infop info
)
396 static HRESULT WINAPI
PngDecoder_Initialize(IWICBitmapDecoder
*iface
, IStream
*pIStream
,
397 WICDecodeOptions cacheOptions
)
399 PngDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
402 png_bytep
*row_pointers
=NULL
;
407 TRACE("(%p,%p,%x)\n", iface
, pIStream
, cacheOptions
);
409 EnterCriticalSection(&This
->lock
);
411 /* initialize libpng */
412 This
->png_ptr
= ppng_create_read_struct(PNG_LIBPNG_VER_STRING
, NULL
, NULL
, NULL
);
419 This
->info_ptr
= ppng_create_info_struct(This
->png_ptr
);
422 ppng_destroy_read_struct(&This
->png_ptr
, NULL
, NULL
);
423 This
->png_ptr
= NULL
;
428 /* set up setjmp/longjmp error handling */
431 ppng_destroy_read_struct(&This
->png_ptr
, &This
->info_ptr
, NULL
);
432 HeapFree(GetProcessHeap(), 0, row_pointers
);
433 This
->png_ptr
= NULL
;
437 ppng_set_error_fn(This
->png_ptr
, &jmpbuf
, user_error_fn
, user_warning_fn
);
439 /* seek to the start of the stream */
441 hr
= IStream_Seek(pIStream
, seek
, STREAM_SEEK_SET
, NULL
);
442 if (FAILED(hr
)) goto end
;
444 /* set up progressive loading */
445 ppng_set_progressive_read_fn(This
->png_ptr
, (void*)This
,
446 info_callback
, row_callback
, end_callback
);
450 hr
= IStream_Read(pIStream
, buffer
, sizeof(buffer
), &bytesRead
);
452 ppng_process_data(This
->png_ptr
, This
->info_ptr
, buffer
, bytesRead
);
453 } while (bytesRead
== sizeof(buffer
));
457 ERR("reading PNG file failed, error 0x%08X\n", hr
);
458 if (!This
->initialized
)
463 LeaveCriticalSection(&This
->lock
);
468 static HRESULT WINAPI
PngDecoder_GetContainerFormat(IWICBitmapDecoder
*iface
,
469 GUID
*pguidContainerFormat
)
471 memcpy(pguidContainerFormat
, &GUID_ContainerFormatPng
, sizeof(GUID
));
475 static HRESULT WINAPI
PngDecoder_GetDecoderInfo(IWICBitmapDecoder
*iface
,
476 IWICBitmapDecoderInfo
**ppIDecoderInfo
)
479 IWICComponentInfo
*compinfo
;
481 TRACE("(%p,%p)\n", iface
, ppIDecoderInfo
);
483 hr
= CreateComponentInfo(&CLSID_WICPngDecoder
, &compinfo
);
484 if (FAILED(hr
)) return hr
;
486 hr
= IWICComponentInfo_QueryInterface(compinfo
, &IID_IWICBitmapDecoderInfo
,
487 (void**)ppIDecoderInfo
);
489 IWICComponentInfo_Release(compinfo
);
494 static HRESULT WINAPI
PngDecoder_CopyPalette(IWICBitmapDecoder
*iface
,
495 IWICPalette
*pIPalette
)
497 FIXME("(%p,%p): stub\n", iface
, pIPalette
);
501 static HRESULT WINAPI
PngDecoder_GetMetadataQueryReader(IWICBitmapDecoder
*iface
,
502 IWICMetadataQueryReader
**ppIMetadataQueryReader
)
504 FIXME("(%p,%p): stub\n", iface
, ppIMetadataQueryReader
);
508 static HRESULT WINAPI
PngDecoder_GetPreview(IWICBitmapDecoder
*iface
,
509 IWICBitmapSource
**ppIBitmapSource
)
511 FIXME("(%p,%p): stub\n", iface
, ppIBitmapSource
);
515 static HRESULT WINAPI
PngDecoder_GetColorContexts(IWICBitmapDecoder
*iface
,
516 UINT cCount
, IWICColorContext
**ppIColorContexts
, UINT
*pcActualCount
)
518 FIXME("(%p,%u,%p,%p)\n", iface
, cCount
, ppIColorContexts
, pcActualCount
);
522 static HRESULT WINAPI
PngDecoder_GetThumbnail(IWICBitmapDecoder
*iface
,
523 IWICBitmapSource
**ppIThumbnail
)
525 TRACE("(%p,%p)\n", iface
, ppIThumbnail
);
526 return WINCODEC_ERR_CODECNOTHUMBNAIL
;
529 static HRESULT WINAPI
PngDecoder_GetFrameCount(IWICBitmapDecoder
*iface
,
536 static HRESULT WINAPI
PngDecoder_GetFrame(IWICBitmapDecoder
*iface
,
537 UINT index
, IWICBitmapFrameDecode
**ppIBitmapFrame
)
539 PngDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
540 TRACE("(%p,%u,%p)\n", iface
, index
, ppIBitmapFrame
);
542 if (!This
->initialized
) return WINCODEC_ERR_NOTINITIALIZED
;
544 if (index
!= 0) return E_INVALIDARG
;
546 IWICBitmapDecoder_AddRef(iface
);
548 *ppIBitmapFrame
= &This
->IWICBitmapFrameDecode_iface
;
553 static const IWICBitmapDecoderVtbl PngDecoder_Vtbl
= {
554 PngDecoder_QueryInterface
,
557 PngDecoder_QueryCapability
,
558 PngDecoder_Initialize
,
559 PngDecoder_GetContainerFormat
,
560 PngDecoder_GetDecoderInfo
,
561 PngDecoder_CopyPalette
,
562 PngDecoder_GetMetadataQueryReader
,
563 PngDecoder_GetPreview
,
564 PngDecoder_GetColorContexts
,
565 PngDecoder_GetThumbnail
,
566 PngDecoder_GetFrameCount
,
570 static HRESULT WINAPI
PngDecoder_Frame_QueryInterface(IWICBitmapFrameDecode
*iface
, REFIID iid
,
573 if (!ppv
) return E_INVALIDARG
;
575 if (IsEqualIID(&IID_IUnknown
, iid
) ||
576 IsEqualIID(&IID_IWICBitmapSource
, iid
) ||
577 IsEqualIID(&IID_IWICBitmapFrameDecode
, iid
))
584 return E_NOINTERFACE
;
587 IUnknown_AddRef((IUnknown
*)*ppv
);
591 static ULONG WINAPI
PngDecoder_Frame_AddRef(IWICBitmapFrameDecode
*iface
)
593 PngDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
594 return IUnknown_AddRef((IUnknown
*)This
);
597 static ULONG WINAPI
PngDecoder_Frame_Release(IWICBitmapFrameDecode
*iface
)
599 PngDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
600 return IUnknown_Release((IUnknown
*)This
);
603 static HRESULT WINAPI
PngDecoder_Frame_GetSize(IWICBitmapFrameDecode
*iface
,
604 UINT
*puiWidth
, UINT
*puiHeight
)
606 PngDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
607 *puiWidth
= This
->width
;
608 *puiHeight
= This
->height
;
609 TRACE("(%p)->(%u,%u)\n", iface
, *puiWidth
, *puiHeight
);
613 static HRESULT WINAPI
PngDecoder_Frame_GetPixelFormat(IWICBitmapFrameDecode
*iface
,
614 WICPixelFormatGUID
*pPixelFormat
)
616 PngDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
617 TRACE("(%p,%p)\n", iface
, pPixelFormat
);
619 memcpy(pPixelFormat
, This
->format
, sizeof(GUID
));
624 static HRESULT WINAPI
PngDecoder_Frame_GetResolution(IWICBitmapFrameDecode
*iface
,
625 double *pDpiX
, double *pDpiY
)
627 PngDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
628 png_uint_32 ret
, xres
, yres
;
631 EnterCriticalSection(&This
->lock
);
633 ret
= ppng_get_pHYs(This
->png_ptr
, This
->info_ptr
, &xres
, &yres
, &unit_type
);
635 if (ret
&& unit_type
== PNG_RESOLUTION_METER
)
637 *pDpiX
= xres
* 0.0254;
638 *pDpiY
= yres
* 0.0254;
642 WARN("no pHYs block present\n");
643 *pDpiX
= *pDpiY
= 96.0;
646 LeaveCriticalSection(&This
->lock
);
648 TRACE("(%p)->(%0.2f,%0.2f)\n", iface
, *pDpiX
, *pDpiY
);
653 static HRESULT WINAPI
PngDecoder_Frame_CopyPalette(IWICBitmapFrameDecode
*iface
,
654 IWICPalette
*pIPalette
)
656 PngDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
658 png_colorp png_palette
;
660 WICColor palette
[256];
663 png_color_16p trans_values
;
667 TRACE("(%p,%p)\n", iface
, pIPalette
);
669 EnterCriticalSection(&This
->lock
);
671 ret
= ppng_get_PLTE(This
->png_ptr
, This
->info_ptr
, &png_palette
, &num_palette
);
674 hr
= WINCODEC_ERR_PALETTEUNAVAILABLE
;
678 if (num_palette
> 256)
680 ERR("palette has %i colors?!\n", num_palette
);
685 for (i
=0; i
<num_palette
; i
++)
687 palette
[i
] = (0xff000000|
688 png_palette
[i
].red
<< 16|
689 png_palette
[i
].green
<< 8|
690 png_palette
[i
].blue
);
693 ret
= ppng_get_tRNS(This
->png_ptr
, This
->info_ptr
, &trans
, &num_trans
, &trans_values
);
696 for (i
=0; i
<num_trans
; i
++)
698 palette
[trans
[i
]] = 0x00000000;
704 LeaveCriticalSection(&This
->lock
);
707 hr
= IWICPalette_InitializeCustom(pIPalette
, palette
, num_palette
);
712 static HRESULT WINAPI
PngDecoder_Frame_CopyPixels(IWICBitmapFrameDecode
*iface
,
713 const WICRect
*prc
, UINT cbStride
, UINT cbBufferSize
, BYTE
*pbBuffer
)
715 PngDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
716 TRACE("(%p,%p,%u,%u,%p)\n", iface
, prc
, cbStride
, cbBufferSize
, pbBuffer
);
718 return copy_pixels(This
->bpp
, This
->image_bits
,
719 This
->width
, This
->height
, This
->stride
,
720 prc
, cbStride
, cbBufferSize
, pbBuffer
);
723 static HRESULT WINAPI
PngDecoder_Frame_GetMetadataQueryReader(IWICBitmapFrameDecode
*iface
,
724 IWICMetadataQueryReader
**ppIMetadataQueryReader
)
726 FIXME("(%p,%p): stub\n", iface
, ppIMetadataQueryReader
);
730 static HRESULT WINAPI
PngDecoder_Frame_GetColorContexts(IWICBitmapFrameDecode
*iface
,
731 UINT cCount
, IWICColorContext
**ppIColorContexts
, UINT
*pcActualCount
)
733 FIXME("(%p,%u,%p,%p): stub\n", iface
, cCount
, ppIColorContexts
, pcActualCount
);
737 static HRESULT WINAPI
PngDecoder_Frame_GetThumbnail(IWICBitmapFrameDecode
*iface
,
738 IWICBitmapSource
**ppIThumbnail
)
740 FIXME("(%p,%p): stub\n", iface
, ppIThumbnail
);
744 static const IWICBitmapFrameDecodeVtbl PngDecoder_FrameVtbl
= {
745 PngDecoder_Frame_QueryInterface
,
746 PngDecoder_Frame_AddRef
,
747 PngDecoder_Frame_Release
,
748 PngDecoder_Frame_GetSize
,
749 PngDecoder_Frame_GetPixelFormat
,
750 PngDecoder_Frame_GetResolution
,
751 PngDecoder_Frame_CopyPalette
,
752 PngDecoder_Frame_CopyPixels
,
753 PngDecoder_Frame_GetMetadataQueryReader
,
754 PngDecoder_Frame_GetColorContexts
,
755 PngDecoder_Frame_GetThumbnail
758 HRESULT
PngDecoder_CreateInstance(IUnknown
*pUnkOuter
, REFIID iid
, void** ppv
)
763 TRACE("(%p,%s,%p)\n", pUnkOuter
, debugstr_guid(iid
), ppv
);
767 if (pUnkOuter
) return CLASS_E_NOAGGREGATION
;
769 if (!libpng_handle
&& !load_libpng())
771 ERR("Failed reading PNG because unable to find %s\n",SONAME_LIBPNG
);
775 This
= HeapAlloc(GetProcessHeap(), 0, sizeof(PngDecoder
));
776 if (!This
) return E_OUTOFMEMORY
;
778 This
->IWICBitmapDecoder_iface
.lpVtbl
= &PngDecoder_Vtbl
;
779 This
->IWICBitmapFrameDecode_iface
.lpVtbl
= &PngDecoder_FrameVtbl
;
781 This
->png_ptr
= NULL
;
782 This
->info_ptr
= NULL
;
783 This
->initialized
= FALSE
;
784 This
->image_bits
= NULL
;
785 InitializeCriticalSection(&This
->lock
);
786 This
->lock
.DebugInfo
->Spare
[0] = (DWORD_PTR
)(__FILE__
": PngDecoder.lock");
788 ret
= IUnknown_QueryInterface((IUnknown
*)This
, iid
, ppv
);
789 IUnknown_Release((IUnknown
*)This
);
794 struct png_pixelformat
{
795 const WICPixelFormatGUID
*guid
;
803 static const struct png_pixelformat formats
[] = {
804 {&GUID_WICPixelFormat24bppBGR
, 24, 8, PNG_COLOR_TYPE_RGB
, 0, 1},
805 {&GUID_WICPixelFormatBlackWhite
, 1, 1, PNG_COLOR_TYPE_GRAY
, 0, 0},
806 {&GUID_WICPixelFormat2bppGray
, 2, 2, PNG_COLOR_TYPE_GRAY
, 0, 0},
807 {&GUID_WICPixelFormat4bppGray
, 4, 4, PNG_COLOR_TYPE_GRAY
, 0, 0},
808 {&GUID_WICPixelFormat8bppGray
, 8, 8, PNG_COLOR_TYPE_GRAY
, 0, 0},
809 {&GUID_WICPixelFormat16bppGray
, 16, 16, PNG_COLOR_TYPE_GRAY
, 0, 0},
810 {&GUID_WICPixelFormat32bppBGR
, 32, 8, PNG_COLOR_TYPE_RGB
, 1, 1},
811 {&GUID_WICPixelFormat32bppBGRA
, 32, 8, PNG_COLOR_TYPE_RGB_ALPHA
, 0, 1},
812 {&GUID_WICPixelFormat48bppRGB
, 48, 16, PNG_COLOR_TYPE_RGB
, 0, 0},
813 {&GUID_WICPixelFormat64bppRGBA
, 64, 16, PNG_COLOR_TYPE_RGB_ALPHA
, 0, 0},
817 typedef struct PngEncoder
{
818 IWICBitmapEncoder IWICBitmapEncoder_iface
;
819 IWICBitmapFrameEncode IWICBitmapFrameEncode_iface
;
825 BOOL frame_initialized
;
826 const struct png_pixelformat
*format
;
831 BOOL frame_committed
;
833 CRITICAL_SECTION lock
;
836 static inline PngEncoder
*impl_from_IWICBitmapEncoder(IWICBitmapEncoder
*iface
)
838 return CONTAINING_RECORD(iface
, PngEncoder
, IWICBitmapEncoder_iface
);
841 static inline PngEncoder
*impl_from_IWICBitmapFrameEncode(IWICBitmapFrameEncode
*iface
)
843 return CONTAINING_RECORD(iface
, PngEncoder
, IWICBitmapFrameEncode_iface
);
846 static HRESULT WINAPI
PngFrameEncode_QueryInterface(IWICBitmapFrameEncode
*iface
, REFIID iid
,
849 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
850 TRACE("(%p,%s,%p)\n", iface
, debugstr_guid(iid
), ppv
);
852 if (!ppv
) return E_INVALIDARG
;
854 if (IsEqualIID(&IID_IUnknown
, iid
) ||
855 IsEqualIID(&IID_IWICBitmapFrameEncode
, iid
))
857 *ppv
= &This
->IWICBitmapFrameEncode_iface
;
862 return E_NOINTERFACE
;
865 IUnknown_AddRef((IUnknown
*)*ppv
);
869 static ULONG WINAPI
PngFrameEncode_AddRef(IWICBitmapFrameEncode
*iface
)
871 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
872 return IUnknown_AddRef((IUnknown
*)This
);
875 static ULONG WINAPI
PngFrameEncode_Release(IWICBitmapFrameEncode
*iface
)
877 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
878 return IUnknown_Release((IUnknown
*)This
);
881 static HRESULT WINAPI
PngFrameEncode_Initialize(IWICBitmapFrameEncode
*iface
,
882 IPropertyBag2
*pIEncoderOptions
)
884 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
885 TRACE("(%p,%p)\n", iface
, pIEncoderOptions
);
887 EnterCriticalSection(&This
->lock
);
889 if (This
->frame_initialized
)
891 LeaveCriticalSection(&This
->lock
);
892 return WINCODEC_ERR_WRONGSTATE
;
895 This
->frame_initialized
= TRUE
;
897 LeaveCriticalSection(&This
->lock
);
902 static HRESULT WINAPI
PngFrameEncode_SetSize(IWICBitmapFrameEncode
*iface
,
903 UINT uiWidth
, UINT uiHeight
)
905 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
906 TRACE("(%p,%u,%u)\n", iface
, uiWidth
, uiHeight
);
908 EnterCriticalSection(&This
->lock
);
910 if (!This
->frame_initialized
|| This
->info_written
)
912 LeaveCriticalSection(&This
->lock
);
913 return WINCODEC_ERR_WRONGSTATE
;
916 This
->width
= uiWidth
;
917 This
->height
= uiHeight
;
919 LeaveCriticalSection(&This
->lock
);
924 static HRESULT WINAPI
PngFrameEncode_SetResolution(IWICBitmapFrameEncode
*iface
,
925 double dpiX
, double dpiY
)
927 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
928 TRACE("(%p,%0.2f,%0.2f)\n", iface
, dpiX
, dpiY
);
930 EnterCriticalSection(&This
->lock
);
932 if (!This
->frame_initialized
|| This
->info_written
)
934 LeaveCriticalSection(&This
->lock
);
935 return WINCODEC_ERR_WRONGSTATE
;
941 LeaveCriticalSection(&This
->lock
);
946 static HRESULT WINAPI
PngFrameEncode_SetPixelFormat(IWICBitmapFrameEncode
*iface
,
947 WICPixelFormatGUID
*pPixelFormat
)
949 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
951 TRACE("(%p,%s)\n", iface
, debugstr_guid(pPixelFormat
));
953 EnterCriticalSection(&This
->lock
);
955 if (!This
->frame_initialized
|| This
->info_written
)
957 LeaveCriticalSection(&This
->lock
);
958 return WINCODEC_ERR_WRONGSTATE
;
961 for (i
=0; formats
[i
].guid
; i
++)
963 if (memcmp(formats
[i
].guid
, pPixelFormat
, sizeof(GUID
)) == 0)
967 if (!formats
[i
].guid
) i
= 0;
969 This
->format
= &formats
[i
];
970 memcpy(pPixelFormat
, This
->format
->guid
, sizeof(GUID
));
972 LeaveCriticalSection(&This
->lock
);
977 static HRESULT WINAPI
PngFrameEncode_SetColorContexts(IWICBitmapFrameEncode
*iface
,
978 UINT cCount
, IWICColorContext
**ppIColorContext
)
980 FIXME("(%p,%u,%p): stub\n", iface
, cCount
, ppIColorContext
);
984 static HRESULT WINAPI
PngFrameEncode_SetPalette(IWICBitmapFrameEncode
*iface
,
985 IWICPalette
*pIPalette
)
987 FIXME("(%p,%p): stub\n", iface
, pIPalette
);
988 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
991 static HRESULT WINAPI
PngFrameEncode_SetThumbnail(IWICBitmapFrameEncode
*iface
,
992 IWICBitmapSource
*pIThumbnail
)
994 FIXME("(%p,%p): stub\n", iface
, pIThumbnail
);
995 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
998 static HRESULT WINAPI
PngFrameEncode_WritePixels(IWICBitmapFrameEncode
*iface
,
999 UINT lineCount
, UINT cbStride
, UINT cbBufferSize
, BYTE
*pbPixels
)
1001 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
1002 png_byte
**row_pointers
=NULL
;
1005 TRACE("(%p,%u,%u,%u,%p)\n", iface
, lineCount
, cbStride
, cbBufferSize
, pbPixels
);
1007 EnterCriticalSection(&This
->lock
);
1009 if (!This
->frame_initialized
|| !This
->width
|| !This
->height
|| !This
->format
)
1011 LeaveCriticalSection(&This
->lock
);
1012 return WINCODEC_ERR_WRONGSTATE
;
1015 if (lineCount
== 0 || lineCount
+ This
->lines_written
> This
->height
)
1017 LeaveCriticalSection(&This
->lock
);
1018 return E_INVALIDARG
;
1021 /* set up setjmp/longjmp error handling */
1024 LeaveCriticalSection(&This
->lock
);
1025 HeapFree(GetProcessHeap(), 0, row_pointers
);
1028 ppng_set_error_fn(This
->png_ptr
, &jmpbuf
, user_error_fn
, user_warning_fn
);
1030 if (!This
->info_written
)
1032 ppng_set_IHDR(This
->png_ptr
, This
->info_ptr
, This
->width
, This
->height
,
1033 This
->format
->bit_depth
, This
->format
->color_type
, PNG_INTERLACE_NONE
,
1034 PNG_COMPRESSION_TYPE_DEFAULT
, PNG_FILTER_TYPE_DEFAULT
);
1036 if (This
->xres
!= 0.0 && This
->yres
!= 0.0)
1038 ppng_set_pHYs(This
->png_ptr
, This
->info_ptr
, (This
->xres
+0.0127) / 0.0254,
1039 (This
->yres
+0.0127) / 0.0254, PNG_RESOLUTION_METER
);
1042 ppng_write_info(This
->png_ptr
, This
->info_ptr
);
1044 if (This
->format
->remove_filler
)
1045 ppng_set_filler(This
->png_ptr
, 0, PNG_FILLER_AFTER
);
1047 if (This
->format
->swap_rgb
)
1048 ppng_set_bgr(This
->png_ptr
);
1050 This
->info_written
= TRUE
;
1053 row_pointers
= HeapAlloc(GetProcessHeap(), 0, lineCount
* sizeof(png_byte
*));
1056 LeaveCriticalSection(&This
->lock
);
1057 return E_OUTOFMEMORY
;
1060 for (i
=0; i
<lineCount
; i
++)
1061 row_pointers
[i
] = pbPixels
+ cbStride
* i
;
1063 ppng_write_rows(This
->png_ptr
, row_pointers
, lineCount
);
1064 This
->lines_written
+= lineCount
;
1066 LeaveCriticalSection(&This
->lock
);
1068 HeapFree(GetProcessHeap(), 0, row_pointers
);
1073 static HRESULT WINAPI
PngFrameEncode_WriteSource(IWICBitmapFrameEncode
*iface
,
1074 IWICBitmapSource
*pIBitmapSource
, WICRect
*prc
)
1076 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
1079 WICPixelFormatGUID guid
;
1082 TRACE("(%p,%p,%p)\n", iface
, pIBitmapSource
, prc
);
1084 if (!This
->frame_initialized
|| !This
->width
|| !This
->height
)
1085 return WINCODEC_ERR_WRONGSTATE
;
1089 hr
= IWICBitmapSource_GetPixelFormat(pIBitmapSource
, &guid
);
1090 if (FAILED(hr
)) return hr
;
1091 hr
= IWICBitmapFrameEncode_SetPixelFormat(iface
, &guid
);
1092 if (FAILED(hr
)) return hr
;
1095 hr
= IWICBitmapSource_GetPixelFormat(pIBitmapSource
, &guid
);
1096 if (FAILED(hr
)) return hr
;
1097 if (memcmp(&guid
, This
->format
->guid
, sizeof(GUID
)) != 0)
1099 /* FIXME: should use WICConvertBitmapSource to convert */
1100 ERR("format %s unsupported\n", debugstr_guid(&guid
));
1104 if (This
->xres
== 0.0 || This
->yres
== 0.0)
1107 hr
= IWICBitmapSource_GetResolution(pIBitmapSource
, &xres
, &yres
);
1108 if (FAILED(hr
)) return hr
;
1109 hr
= IWICBitmapFrameEncode_SetResolution(iface
, xres
, yres
);
1110 if (FAILED(hr
)) return hr
;
1116 hr
= IWICBitmapSource_GetSize(pIBitmapSource
, &width
, &height
);
1117 if (FAILED(hr
)) return hr
;
1125 if (prc
->Width
!= This
->width
) return E_INVALIDARG
;
1127 stride
= (This
->format
->bpp
* This
->width
+ 7)/8;
1129 pixeldata
= HeapAlloc(GetProcessHeap(), 0, stride
* prc
->Height
);
1130 if (!pixeldata
) return E_OUTOFMEMORY
;
1132 hr
= IWICBitmapSource_CopyPixels(pIBitmapSource
, prc
, stride
,
1133 stride
*prc
->Height
, pixeldata
);
1137 hr
= IWICBitmapFrameEncode_WritePixels(iface
, prc
->Height
, stride
,
1138 stride
*prc
->Height
, pixeldata
);
1141 HeapFree(GetProcessHeap(), 0, pixeldata
);
1146 static HRESULT WINAPI
PngFrameEncode_Commit(IWICBitmapFrameEncode
*iface
)
1148 PngEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
1150 TRACE("(%p)\n", iface
);
1152 EnterCriticalSection(&This
->lock
);
1154 if (!This
->info_written
|| This
->lines_written
!= This
->height
|| This
->frame_committed
)
1156 LeaveCriticalSection(&This
->lock
);
1157 return WINCODEC_ERR_WRONGSTATE
;
1160 /* set up setjmp/longjmp error handling */
1163 LeaveCriticalSection(&This
->lock
);
1166 ppng_set_error_fn(This
->png_ptr
, &jmpbuf
, user_error_fn
, user_warning_fn
);
1168 ppng_write_end(This
->png_ptr
, This
->info_ptr
);
1170 This
->frame_committed
= TRUE
;
1172 LeaveCriticalSection(&This
->lock
);
1177 static HRESULT WINAPI
PngFrameEncode_GetMetadataQueryWriter(IWICBitmapFrameEncode
*iface
,
1178 IWICMetadataQueryWriter
**ppIMetadataQueryWriter
)
1180 FIXME("(%p, %p): stub\n", iface
, ppIMetadataQueryWriter
);
1184 static const IWICBitmapFrameEncodeVtbl PngEncoder_FrameVtbl
= {
1185 PngFrameEncode_QueryInterface
,
1186 PngFrameEncode_AddRef
,
1187 PngFrameEncode_Release
,
1188 PngFrameEncode_Initialize
,
1189 PngFrameEncode_SetSize
,
1190 PngFrameEncode_SetResolution
,
1191 PngFrameEncode_SetPixelFormat
,
1192 PngFrameEncode_SetColorContexts
,
1193 PngFrameEncode_SetPalette
,
1194 PngFrameEncode_SetThumbnail
,
1195 PngFrameEncode_WritePixels
,
1196 PngFrameEncode_WriteSource
,
1197 PngFrameEncode_Commit
,
1198 PngFrameEncode_GetMetadataQueryWriter
1201 static HRESULT WINAPI
PngEncoder_QueryInterface(IWICBitmapEncoder
*iface
, REFIID iid
,
1204 PngEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1205 TRACE("(%p,%s,%p)\n", iface
, debugstr_guid(iid
), ppv
);
1207 if (!ppv
) return E_INVALIDARG
;
1209 if (IsEqualIID(&IID_IUnknown
, iid
) ||
1210 IsEqualIID(&IID_IWICBitmapEncoder
, iid
))
1217 return E_NOINTERFACE
;
1220 IUnknown_AddRef((IUnknown
*)*ppv
);
1224 static ULONG WINAPI
PngEncoder_AddRef(IWICBitmapEncoder
*iface
)
1226 PngEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1227 ULONG ref
= InterlockedIncrement(&This
->ref
);
1229 TRACE("(%p) refcount=%u\n", iface
, ref
);
1234 static ULONG WINAPI
PngEncoder_Release(IWICBitmapEncoder
*iface
)
1236 PngEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1237 ULONG ref
= InterlockedDecrement(&This
->ref
);
1239 TRACE("(%p) refcount=%u\n", iface
, ref
);
1243 This
->lock
.DebugInfo
->Spare
[0] = 0;
1244 DeleteCriticalSection(&This
->lock
);
1246 ppng_destroy_write_struct(&This
->png_ptr
, &This
->info_ptr
);
1248 IStream_Release(This
->stream
);
1249 HeapFree(GetProcessHeap(), 0, This
);
1255 static void user_write_data(png_structp png_ptr
, png_bytep data
, png_size_t length
)
1257 PngEncoder
*This
= ppng_get_io_ptr(png_ptr
);
1261 hr
= IStream_Write(This
->stream
, data
, length
, &byteswritten
);
1262 if (FAILED(hr
) || byteswritten
!= length
)
1264 ppng_error(png_ptr
, "failed writing data");
1268 static void user_flush(png_structp png_ptr
)
1272 static HRESULT WINAPI
PngEncoder_Initialize(IWICBitmapEncoder
*iface
,
1273 IStream
*pIStream
, WICBitmapEncoderCacheOption cacheOption
)
1275 PngEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1278 TRACE("(%p,%p,%u)\n", iface
, pIStream
, cacheOption
);
1280 EnterCriticalSection(&This
->lock
);
1284 LeaveCriticalSection(&This
->lock
);
1285 return WINCODEC_ERR_WRONGSTATE
;
1288 /* initialize libpng */
1289 This
->png_ptr
= ppng_create_write_struct(PNG_LIBPNG_VER_STRING
, NULL
, NULL
, NULL
);
1292 LeaveCriticalSection(&This
->lock
);
1296 This
->info_ptr
= ppng_create_info_struct(This
->png_ptr
);
1297 if (!This
->info_ptr
)
1299 ppng_destroy_write_struct(&This
->png_ptr
, NULL
);
1300 This
->png_ptr
= NULL
;
1301 LeaveCriticalSection(&This
->lock
);
1305 IStream_AddRef(pIStream
);
1306 This
->stream
= pIStream
;
1308 /* set up setjmp/longjmp error handling */
1311 ppng_destroy_write_struct(&This
->png_ptr
, &This
->info_ptr
);
1312 This
->png_ptr
= NULL
;
1313 IStream_Release(This
->stream
);
1314 This
->stream
= NULL
;
1315 LeaveCriticalSection(&This
->lock
);
1318 ppng_set_error_fn(This
->png_ptr
, &jmpbuf
, user_error_fn
, user_warning_fn
);
1320 /* set up custom i/o handling */
1321 ppng_set_write_fn(This
->png_ptr
, This
, user_write_data
, user_flush
);
1323 LeaveCriticalSection(&This
->lock
);
1328 static HRESULT WINAPI
PngEncoder_GetContainerFormat(IWICBitmapEncoder
*iface
,
1329 GUID
*pguidContainerFormat
)
1331 FIXME("(%p,%s): stub\n", iface
, debugstr_guid(pguidContainerFormat
));
1335 static HRESULT WINAPI
PngEncoder_GetEncoderInfo(IWICBitmapEncoder
*iface
,
1336 IWICBitmapEncoderInfo
**ppIEncoderInfo
)
1338 FIXME("(%p,%p): stub\n", iface
, ppIEncoderInfo
);
1342 static HRESULT WINAPI
PngEncoder_SetColorContexts(IWICBitmapEncoder
*iface
,
1343 UINT cCount
, IWICColorContext
**ppIColorContext
)
1345 FIXME("(%p,%u,%p): stub\n", iface
, cCount
, ppIColorContext
);
1349 static HRESULT WINAPI
PngEncoder_SetPalette(IWICBitmapEncoder
*iface
, IWICPalette
*pIPalette
)
1351 TRACE("(%p,%p)\n", iface
, pIPalette
);
1352 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
1355 static HRESULT WINAPI
PngEncoder_SetThumbnail(IWICBitmapEncoder
*iface
, IWICBitmapSource
*pIThumbnail
)
1357 TRACE("(%p,%p)\n", iface
, pIThumbnail
);
1358 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
1361 static HRESULT WINAPI
PngEncoder_SetPreview(IWICBitmapEncoder
*iface
, IWICBitmapSource
*pIPreview
)
1363 TRACE("(%p,%p)\n", iface
, pIPreview
);
1364 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
1367 static HRESULT WINAPI
PngEncoder_CreateNewFrame(IWICBitmapEncoder
*iface
,
1368 IWICBitmapFrameEncode
**ppIFrameEncode
, IPropertyBag2
**ppIEncoderOptions
)
1370 PngEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1372 TRACE("(%p,%p,%p)\n", iface
, ppIFrameEncode
, ppIEncoderOptions
);
1374 EnterCriticalSection(&This
->lock
);
1376 if (This
->frame_count
!= 0)
1378 LeaveCriticalSection(&This
->lock
);
1379 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
1384 LeaveCriticalSection(&This
->lock
);
1385 return WINCODEC_ERR_NOTINITIALIZED
;
1388 hr
= CreatePropertyBag2(ppIEncoderOptions
);
1391 LeaveCriticalSection(&This
->lock
);
1395 This
->frame_count
= 1;
1397 LeaveCriticalSection(&This
->lock
);
1399 IWICBitmapEncoder_AddRef(iface
);
1400 *ppIFrameEncode
= &This
->IWICBitmapFrameEncode_iface
;
1405 static HRESULT WINAPI
PngEncoder_Commit(IWICBitmapEncoder
*iface
)
1407 PngEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1408 TRACE("(%p)\n", iface
);
1410 EnterCriticalSection(&This
->lock
);
1412 if (!This
->frame_committed
|| This
->committed
)
1414 LeaveCriticalSection(&This
->lock
);
1415 return WINCODEC_ERR_WRONGSTATE
;
1418 This
->committed
= TRUE
;
1420 LeaveCriticalSection(&This
->lock
);
1425 static HRESULT WINAPI
PngEncoder_GetMetadataQueryWriter(IWICBitmapEncoder
*iface
,
1426 IWICMetadataQueryWriter
**ppIMetadataQueryWriter
)
1428 FIXME("(%p,%p): stub\n", iface
, ppIMetadataQueryWriter
);
1432 static const IWICBitmapEncoderVtbl PngEncoder_Vtbl
= {
1433 PngEncoder_QueryInterface
,
1436 PngEncoder_Initialize
,
1437 PngEncoder_GetContainerFormat
,
1438 PngEncoder_GetEncoderInfo
,
1439 PngEncoder_SetColorContexts
,
1440 PngEncoder_SetPalette
,
1441 PngEncoder_SetThumbnail
,
1442 PngEncoder_SetPreview
,
1443 PngEncoder_CreateNewFrame
,
1445 PngEncoder_GetMetadataQueryWriter
1448 HRESULT
PngEncoder_CreateInstance(IUnknown
*pUnkOuter
, REFIID iid
, void** ppv
)
1453 TRACE("(%p,%s,%p)\n", pUnkOuter
, debugstr_guid(iid
), ppv
);
1457 if (pUnkOuter
) return CLASS_E_NOAGGREGATION
;
1459 if (!libpng_handle
&& !load_libpng())
1461 ERR("Failed writing PNG because unable to find %s\n",SONAME_LIBPNG
);
1465 This
= HeapAlloc(GetProcessHeap(), 0, sizeof(PngEncoder
));
1466 if (!This
) return E_OUTOFMEMORY
;
1468 This
->IWICBitmapEncoder_iface
.lpVtbl
= &PngEncoder_Vtbl
;
1469 This
->IWICBitmapFrameEncode_iface
.lpVtbl
= &PngEncoder_FrameVtbl
;
1471 This
->png_ptr
= NULL
;
1472 This
->info_ptr
= NULL
;
1473 This
->stream
= NULL
;
1474 This
->frame_count
= 0;
1475 This
->frame_initialized
= FALSE
;
1476 This
->format
= NULL
;
1477 This
->info_written
= FALSE
;
1482 This
->lines_written
= 0;
1483 This
->frame_committed
= FALSE
;
1484 This
->committed
= FALSE
;
1485 InitializeCriticalSection(&This
->lock
);
1486 This
->lock
.DebugInfo
->Spare
[0] = (DWORD_PTR
)(__FILE__
": PngEncoder.lock");
1488 ret
= IUnknown_QueryInterface((IUnknown
*)This
, iid
, ppv
);
1489 IUnknown_Release((IUnknown
*)This
);
1494 #else /* !HAVE_PNG_H */
1496 HRESULT
PngDecoder_CreateInstance(IUnknown
*pUnkOuter
, REFIID iid
, void** ppv
)
1498 ERR("Trying to load PNG picture, but PNG support is not compiled in.\n");
1502 HRESULT
PngEncoder_CreateInstance(IUnknown
*pUnkOuter
, REFIID iid
, void** ppv
)
1504 ERR("Trying to save PNG picture, but PNG support is not compiled in.\n");