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"
31 /* This is a hack, so jpeglib.h does not redefine INT32 and the like*/
33 #define UINT8 JPEG_UINT8
34 #define UINT16 JPEG_UINT16
35 #define boolean jpeg_boolean
39 #define HAVE_STDLIB_H 1
52 #include "wincodecs_private.h"
54 #include "wine/debug.h"
55 #include "wine/library.h"
57 WINE_DEFAULT_DEBUG_CHANNEL(wincodecs
);
60 WINE_DECLARE_DEBUG_CHANNEL(jpeg
);
62 static void *libjpeg_handle
;
64 #define MAKE_FUNCPTR(f) static typeof(f) * p##f
65 MAKE_FUNCPTR(jpeg_CreateCompress
);
66 MAKE_FUNCPTR(jpeg_CreateDecompress
);
67 MAKE_FUNCPTR(jpeg_destroy_compress
);
68 MAKE_FUNCPTR(jpeg_destroy_decompress
);
69 MAKE_FUNCPTR(jpeg_finish_compress
);
70 MAKE_FUNCPTR(jpeg_read_header
);
71 MAKE_FUNCPTR(jpeg_read_scanlines
);
72 MAKE_FUNCPTR(jpeg_resync_to_restart
);
73 MAKE_FUNCPTR(jpeg_set_defaults
);
74 MAKE_FUNCPTR(jpeg_start_compress
);
75 MAKE_FUNCPTR(jpeg_start_decompress
);
76 MAKE_FUNCPTR(jpeg_std_error
);
77 MAKE_FUNCPTR(jpeg_write_scanlines
);
80 static void *load_libjpeg(void)
82 if((libjpeg_handle
= wine_dlopen(SONAME_LIBJPEG
, RTLD_NOW
, NULL
, 0)) != NULL
) {
84 #define LOAD_FUNCPTR(f) \
85 if((p##f = wine_dlsym(libjpeg_handle, #f, NULL, 0)) == NULL) { \
86 libjpeg_handle = NULL; \
90 LOAD_FUNCPTR(jpeg_CreateCompress
);
91 LOAD_FUNCPTR(jpeg_CreateDecompress
);
92 LOAD_FUNCPTR(jpeg_destroy_compress
);
93 LOAD_FUNCPTR(jpeg_destroy_decompress
);
94 LOAD_FUNCPTR(jpeg_finish_compress
);
95 LOAD_FUNCPTR(jpeg_read_header
);
96 LOAD_FUNCPTR(jpeg_read_scanlines
);
97 LOAD_FUNCPTR(jpeg_resync_to_restart
);
98 LOAD_FUNCPTR(jpeg_set_defaults
);
99 LOAD_FUNCPTR(jpeg_start_compress
);
100 LOAD_FUNCPTR(jpeg_start_decompress
);
101 LOAD_FUNCPTR(jpeg_std_error
);
102 LOAD_FUNCPTR(jpeg_write_scanlines
);
105 return libjpeg_handle
;
108 static void error_exit_fn(j_common_ptr cinfo
)
110 char message
[JMSG_LENGTH_MAX
];
113 cinfo
->err
->format_message(cinfo
, message
);
114 ERR_(jpeg
)("%s\n", message
);
116 longjmp(*(jmp_buf*)cinfo
->client_data
, 1);
119 static void emit_message_fn(j_common_ptr cinfo
, int msg_level
)
121 char message
[JMSG_LENGTH_MAX
];
123 if (msg_level
< 0 && ERR_ON(jpeg
))
125 cinfo
->err
->format_message(cinfo
, message
);
126 ERR_(jpeg
)("%s\n", message
);
128 else if (msg_level
== 0 && WARN_ON(jpeg
))
130 cinfo
->err
->format_message(cinfo
, message
);
131 WARN_(jpeg
)("%s\n", message
);
133 else if (msg_level
> 0 && TRACE_ON(jpeg
))
135 cinfo
->err
->format_message(cinfo
, message
);
136 TRACE_(jpeg
)("%s\n", message
);
141 IWICBitmapDecoder IWICBitmapDecoder_iface
;
142 IWICBitmapFrameDecode IWICBitmapFrameDecode_iface
;
145 BOOL cinfo_initialized
;
147 struct jpeg_decompress_struct cinfo
;
148 struct jpeg_error_mgr jerr
;
149 struct jpeg_source_mgr source_mgr
;
150 BYTE source_buffer
[1024];
152 CRITICAL_SECTION lock
;
155 static inline JpegDecoder
*impl_from_IWICBitmapDecoder(IWICBitmapDecoder
*iface
)
157 return CONTAINING_RECORD(iface
, JpegDecoder
, IWICBitmapDecoder_iface
);
160 static inline JpegDecoder
*impl_from_IWICBitmapFrameDecode(IWICBitmapFrameDecode
*iface
)
162 return CONTAINING_RECORD(iface
, JpegDecoder
, IWICBitmapFrameDecode_iface
);
165 static inline JpegDecoder
*decoder_from_decompress(j_decompress_ptr decompress
)
167 return CONTAINING_RECORD(decompress
, JpegDecoder
, cinfo
);
170 static HRESULT WINAPI
JpegDecoder_QueryInterface(IWICBitmapDecoder
*iface
, REFIID iid
,
173 JpegDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
174 TRACE("(%p,%s,%p)\n", iface
, debugstr_guid(iid
), ppv
);
176 if (!ppv
) return E_INVALIDARG
;
178 if (IsEqualIID(&IID_IUnknown
, iid
) || IsEqualIID(&IID_IWICBitmapDecoder
, iid
))
185 return E_NOINTERFACE
;
188 IUnknown_AddRef((IUnknown
*)*ppv
);
192 static ULONG WINAPI
JpegDecoder_AddRef(IWICBitmapDecoder
*iface
)
194 JpegDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
195 ULONG ref
= InterlockedIncrement(&This
->ref
);
197 TRACE("(%p) refcount=%u\n", iface
, ref
);
202 static ULONG WINAPI
JpegDecoder_Release(IWICBitmapDecoder
*iface
)
204 JpegDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
205 ULONG ref
= InterlockedDecrement(&This
->ref
);
207 TRACE("(%p) refcount=%u\n", iface
, ref
);
211 This
->lock
.DebugInfo
->Spare
[0] = 0;
212 DeleteCriticalSection(&This
->lock
);
213 if (This
->cinfo_initialized
) pjpeg_destroy_decompress(&This
->cinfo
);
214 if (This
->stream
) IStream_Release(This
->stream
);
215 HeapFree(GetProcessHeap(), 0, This
->image_data
);
216 HeapFree(GetProcessHeap(), 0, This
);
222 static HRESULT WINAPI
JpegDecoder_QueryCapability(IWICBitmapDecoder
*iface
, IStream
*pIStream
,
223 DWORD
*pdwCapability
)
225 FIXME("(%p,%p,%p): stub\n", iface
, pIStream
, pdwCapability
);
229 static void source_mgr_init_source(j_decompress_ptr cinfo
)
233 static jpeg_boolean
source_mgr_fill_input_buffer(j_decompress_ptr cinfo
)
235 JpegDecoder
*This
= decoder_from_decompress(cinfo
);
239 hr
= IStream_Read(This
->stream
, This
->source_buffer
, 1024, &bytesread
);
241 if (hr
!= S_OK
|| bytesread
== 0)
247 This
->source_mgr
.next_input_byte
= This
->source_buffer
;
248 This
->source_mgr
.bytes_in_buffer
= bytesread
;
253 static void source_mgr_skip_input_data(j_decompress_ptr cinfo
, long num_bytes
)
255 JpegDecoder
*This
= decoder_from_decompress(cinfo
);
258 if (num_bytes
> This
->source_mgr
.bytes_in_buffer
)
260 seek
.QuadPart
= num_bytes
- This
->source_mgr
.bytes_in_buffer
;
261 IStream_Seek(This
->stream
, seek
, STREAM_SEEK_CUR
, NULL
);
262 This
->source_mgr
.bytes_in_buffer
= 0;
264 else if (num_bytes
> 0)
266 This
->source_mgr
.next_input_byte
+= num_bytes
;
267 This
->source_mgr
.bytes_in_buffer
-= num_bytes
;
271 static void source_mgr_term_source(j_decompress_ptr cinfo
)
275 static HRESULT WINAPI
JpegDecoder_Initialize(IWICBitmapDecoder
*iface
, IStream
*pIStream
,
276 WICDecodeOptions cacheOptions
)
278 JpegDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
282 TRACE("(%p,%p,%u)\n", iface
, pIStream
, cacheOptions
);
284 EnterCriticalSection(&This
->lock
);
286 if (This
->cinfo_initialized
)
288 LeaveCriticalSection(&This
->lock
);
289 return WINCODEC_ERR_WRONGSTATE
;
292 pjpeg_std_error(&This
->jerr
);
294 This
->jerr
.error_exit
= error_exit_fn
;
295 This
->jerr
.emit_message
= emit_message_fn
;
297 This
->cinfo
.err
= &This
->jerr
;
299 This
->cinfo
.client_data
= jmpbuf
;
303 LeaveCriticalSection(&This
->lock
);
307 pjpeg_CreateDecompress(&This
->cinfo
, JPEG_LIB_VERSION
, sizeof(struct jpeg_decompress_struct
));
309 This
->cinfo_initialized
= TRUE
;
311 This
->stream
= pIStream
;
312 IStream_AddRef(pIStream
);
315 IStream_Seek(This
->stream
, seek
, STREAM_SEEK_SET
, NULL
);
317 This
->source_mgr
.bytes_in_buffer
= 0;
318 This
->source_mgr
.init_source
= source_mgr_init_source
;
319 This
->source_mgr
.fill_input_buffer
= source_mgr_fill_input_buffer
;
320 This
->source_mgr
.skip_input_data
= source_mgr_skip_input_data
;
321 This
->source_mgr
.resync_to_restart
= pjpeg_resync_to_restart
;
322 This
->source_mgr
.term_source
= source_mgr_term_source
;
324 This
->cinfo
.src
= &This
->source_mgr
;
326 ret
= pjpeg_read_header(&This
->cinfo
, TRUE
);
328 if (ret
!= JPEG_HEADER_OK
) {
329 WARN("Jpeg image in stream has bad format, read header returned %d.\n",ret
);
330 LeaveCriticalSection(&This
->lock
);
334 switch (This
->cinfo
.jpeg_color_space
)
337 This
->cinfo
.out_color_space
= JCS_GRAYSCALE
;
341 This
->cinfo
.out_color_space
= JCS_RGB
;
345 This
->cinfo
.out_color_space
= JCS_CMYK
;
348 ERR("Unknown JPEG color space %i\n", This
->cinfo
.jpeg_color_space
);
349 LeaveCriticalSection(&This
->lock
);
353 if (!pjpeg_start_decompress(&This
->cinfo
))
355 ERR("jpeg_start_decompress failed\n");
356 LeaveCriticalSection(&This
->lock
);
360 This
->initialized
= TRUE
;
362 LeaveCriticalSection(&This
->lock
);
367 static HRESULT WINAPI
JpegDecoder_GetContainerFormat(IWICBitmapDecoder
*iface
,
368 GUID
*pguidContainerFormat
)
370 memcpy(pguidContainerFormat
, &GUID_ContainerFormatJpeg
, sizeof(GUID
));
374 static HRESULT WINAPI
JpegDecoder_GetDecoderInfo(IWICBitmapDecoder
*iface
,
375 IWICBitmapDecoderInfo
**ppIDecoderInfo
)
378 IWICComponentInfo
*compinfo
;
380 TRACE("(%p,%p)\n", iface
, ppIDecoderInfo
);
382 hr
= CreateComponentInfo(&CLSID_WICJpegDecoder
, &compinfo
);
383 if (FAILED(hr
)) return hr
;
385 hr
= IWICComponentInfo_QueryInterface(compinfo
, &IID_IWICBitmapDecoderInfo
,
386 (void**)ppIDecoderInfo
);
388 IWICComponentInfo_Release(compinfo
);
393 static HRESULT WINAPI
JpegDecoder_CopyPalette(IWICBitmapDecoder
*iface
,
394 IWICPalette
*pIPalette
)
396 TRACE("(%p,%p)\n", iface
, pIPalette
);
398 return WINCODEC_ERR_PALETTEUNAVAILABLE
;
401 static HRESULT WINAPI
JpegDecoder_GetMetadataQueryReader(IWICBitmapDecoder
*iface
,
402 IWICMetadataQueryReader
**ppIMetadataQueryReader
)
404 FIXME("(%p,%p): stub\n", iface
, ppIMetadataQueryReader
);
408 static HRESULT WINAPI
JpegDecoder_GetPreview(IWICBitmapDecoder
*iface
,
409 IWICBitmapSource
**ppIBitmapSource
)
411 FIXME("(%p,%p): stub\n", iface
, ppIBitmapSource
);
412 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
415 static HRESULT WINAPI
JpegDecoder_GetColorContexts(IWICBitmapDecoder
*iface
,
416 UINT cCount
, IWICColorContext
**ppIColorContexts
, UINT
*pcActualCount
)
418 FIXME("(%p,%u,%p,%p): stub\n", iface
, cCount
, ppIColorContexts
, pcActualCount
);
419 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
422 static HRESULT WINAPI
JpegDecoder_GetThumbnail(IWICBitmapDecoder
*iface
,
423 IWICBitmapSource
**ppIThumbnail
)
425 FIXME("(%p,%p): stub\n", iface
, ppIThumbnail
);
426 return WINCODEC_ERR_CODECNOTHUMBNAIL
;
429 static HRESULT WINAPI
JpegDecoder_GetFrameCount(IWICBitmapDecoder
*iface
,
436 static HRESULT WINAPI
JpegDecoder_GetFrame(IWICBitmapDecoder
*iface
,
437 UINT index
, IWICBitmapFrameDecode
**ppIBitmapFrame
)
439 JpegDecoder
*This
= impl_from_IWICBitmapDecoder(iface
);
440 TRACE("(%p,%u,%p)\n", iface
, index
, ppIBitmapFrame
);
442 if (!This
->initialized
) return WINCODEC_ERR_NOTINITIALIZED
;
444 if (index
!= 0) return E_INVALIDARG
;
446 IWICBitmapDecoder_AddRef(iface
);
447 *ppIBitmapFrame
= &This
->IWICBitmapFrameDecode_iface
;
452 static const IWICBitmapDecoderVtbl JpegDecoder_Vtbl
= {
453 JpegDecoder_QueryInterface
,
456 JpegDecoder_QueryCapability
,
457 JpegDecoder_Initialize
,
458 JpegDecoder_GetContainerFormat
,
459 JpegDecoder_GetDecoderInfo
,
460 JpegDecoder_CopyPalette
,
461 JpegDecoder_GetMetadataQueryReader
,
462 JpegDecoder_GetPreview
,
463 JpegDecoder_GetColorContexts
,
464 JpegDecoder_GetThumbnail
,
465 JpegDecoder_GetFrameCount
,
469 static HRESULT WINAPI
JpegDecoder_Frame_QueryInterface(IWICBitmapFrameDecode
*iface
, REFIID iid
,
472 TRACE("(%p,%s,%p)\n", iface
, debugstr_guid(iid
), ppv
);
474 if (!ppv
) return E_INVALIDARG
;
476 if (IsEqualIID(&IID_IUnknown
, iid
) ||
477 IsEqualIID(&IID_IWICBitmapSource
, iid
) ||
478 IsEqualIID(&IID_IWICBitmapFrameDecode
, iid
))
485 return E_NOINTERFACE
;
488 IUnknown_AddRef((IUnknown
*)*ppv
);
492 static ULONG WINAPI
JpegDecoder_Frame_AddRef(IWICBitmapFrameDecode
*iface
)
494 JpegDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
495 return IUnknown_AddRef((IUnknown
*)This
);
498 static ULONG WINAPI
JpegDecoder_Frame_Release(IWICBitmapFrameDecode
*iface
)
500 JpegDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
501 return IUnknown_Release((IUnknown
*)This
);
504 static HRESULT WINAPI
JpegDecoder_Frame_GetSize(IWICBitmapFrameDecode
*iface
,
505 UINT
*puiWidth
, UINT
*puiHeight
)
507 JpegDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
508 *puiWidth
= This
->cinfo
.output_width
;
509 *puiHeight
= This
->cinfo
.output_height
;
510 TRACE("(%p)->(%u,%u)\n", iface
, *puiWidth
, *puiHeight
);
514 static HRESULT WINAPI
JpegDecoder_Frame_GetPixelFormat(IWICBitmapFrameDecode
*iface
,
515 WICPixelFormatGUID
*pPixelFormat
)
517 JpegDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
518 TRACE("(%p,%p)\n", iface
, pPixelFormat
);
519 if (This
->cinfo
.out_color_space
== JCS_RGB
)
520 memcpy(pPixelFormat
, &GUID_WICPixelFormat24bppBGR
, sizeof(GUID
));
521 else if (This
->cinfo
.out_color_space
== JCS_CMYK
)
522 memcpy(pPixelFormat
, &GUID_WICPixelFormat32bppCMYK
, sizeof(GUID
));
523 else /* This->cinfo.out_color_space == JCS_GRAYSCALE */
524 memcpy(pPixelFormat
, &GUID_WICPixelFormat8bppGray
, sizeof(GUID
));
528 static HRESULT WINAPI
JpegDecoder_Frame_GetResolution(IWICBitmapFrameDecode
*iface
,
529 double *pDpiX
, double *pDpiY
)
531 FIXME("(%p,%p,%p): stub\n", iface
, pDpiX
, pDpiY
);
535 static HRESULT WINAPI
JpegDecoder_Frame_CopyPalette(IWICBitmapFrameDecode
*iface
,
536 IWICPalette
*pIPalette
)
538 FIXME("(%p,%p): stub\n", iface
, pIPalette
);
542 static HRESULT WINAPI
JpegDecoder_Frame_CopyPixels(IWICBitmapFrameDecode
*iface
,
543 const WICRect
*prc
, UINT cbStride
, UINT cbBufferSize
, BYTE
*pbBuffer
)
545 JpegDecoder
*This
= impl_from_IWICBitmapFrameDecode(iface
);
552 TRACE("(%p,%p,%u,%u,%p)\n", iface
, prc
, cbStride
, cbBufferSize
, pbBuffer
);
558 rect
.Width
= This
->cinfo
.output_width
;
559 rect
.Height
= This
->cinfo
.output_height
;
564 if (prc
->X
< 0 || prc
->Y
< 0 || prc
->X
+prc
->Width
> This
->cinfo
.output_width
||
565 prc
->Y
+prc
->Height
> This
->cinfo
.output_height
)
569 if (This
->cinfo
.out_color_space
== JCS_GRAYSCALE
) bpp
= 8;
570 else if (This
->cinfo
.out_color_space
== JCS_CMYK
) bpp
= 32;
573 stride
= bpp
* This
->cinfo
.output_width
;
574 data_size
= stride
* This
->cinfo
.output_height
;
576 max_row_needed
= prc
->Y
+ prc
->Height
;
577 if (max_row_needed
> This
->cinfo
.output_height
) return E_INVALIDARG
;
579 EnterCriticalSection(&This
->lock
);
581 if (!This
->image_data
)
583 This
->image_data
= HeapAlloc(GetProcessHeap(), 0, data_size
);
584 if (!This
->image_data
)
586 LeaveCriticalSection(&This
->lock
);
587 return E_OUTOFMEMORY
;
591 This
->cinfo
.client_data
= jmpbuf
;
595 LeaveCriticalSection(&This
->lock
);
599 while (max_row_needed
> This
->cinfo
.output_scanline
)
601 UINT first_scanline
= This
->cinfo
.output_scanline
;
603 JSAMPROW out_rows
[4];
607 max_rows
= min(This
->cinfo
.output_height
-first_scanline
, 4);
608 for (i
=0; i
<max_rows
; i
++)
609 out_rows
[i
] = This
->image_data
+ stride
* (first_scanline
+i
);
611 ret
= pjpeg_read_scanlines(&This
->cinfo
, out_rows
, max_rows
);
615 ERR("read_scanlines failed\n");
616 LeaveCriticalSection(&This
->lock
);
622 /* libjpeg gives us RGB data and we want BGR, so byteswap the data */
623 reverse_bgr8(3, This
->image_data
+ stride
* first_scanline
,
624 This
->cinfo
.output_width
, This
->cinfo
.output_scanline
- first_scanline
,
628 if (This
->cinfo
.out_color_space
== JCS_CMYK
&& This
->cinfo
.saw_Adobe_marker
)
629 /* Adobe JPEG's have inverted CMYK data. */
630 for (i
=0; i
<data_size
; i
++)
631 This
->image_data
[i
] ^= 0xff;
634 LeaveCriticalSection(&This
->lock
);
636 return copy_pixels(bpp
, This
->image_data
,
637 This
->cinfo
.output_width
, This
->cinfo
.output_height
, stride
,
638 prc
, cbStride
, cbBufferSize
, pbBuffer
);
641 static HRESULT WINAPI
JpegDecoder_Frame_GetMetadataQueryReader(IWICBitmapFrameDecode
*iface
,
642 IWICMetadataQueryReader
**ppIMetadataQueryReader
)
644 FIXME("(%p,%p): stub\n", iface
, ppIMetadataQueryReader
);
645 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
648 static HRESULT WINAPI
JpegDecoder_Frame_GetColorContexts(IWICBitmapFrameDecode
*iface
,
649 UINT cCount
, IWICColorContext
**ppIColorContexts
, UINT
*pcActualCount
)
651 FIXME("(%p,%u,%p,%p): stub\n", iface
, cCount
, ppIColorContexts
, pcActualCount
);
652 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
655 static HRESULT WINAPI
JpegDecoder_Frame_GetThumbnail(IWICBitmapFrameDecode
*iface
,
656 IWICBitmapSource
**ppIThumbnail
)
658 FIXME("(%p,%p): stub\n", iface
, ppIThumbnail
);
659 return WINCODEC_ERR_CODECNOTHUMBNAIL
;
662 static const IWICBitmapFrameDecodeVtbl JpegDecoder_Frame_Vtbl
= {
663 JpegDecoder_Frame_QueryInterface
,
664 JpegDecoder_Frame_AddRef
,
665 JpegDecoder_Frame_Release
,
666 JpegDecoder_Frame_GetSize
,
667 JpegDecoder_Frame_GetPixelFormat
,
668 JpegDecoder_Frame_GetResolution
,
669 JpegDecoder_Frame_CopyPalette
,
670 JpegDecoder_Frame_CopyPixels
,
671 JpegDecoder_Frame_GetMetadataQueryReader
,
672 JpegDecoder_Frame_GetColorContexts
,
673 JpegDecoder_Frame_GetThumbnail
676 HRESULT
JpegDecoder_CreateInstance(IUnknown
*pUnkOuter
, REFIID iid
, void** ppv
)
681 TRACE("(%p,%s,%p)\n", pUnkOuter
, debugstr_guid(iid
), ppv
);
683 if (!libjpeg_handle
&& !load_libjpeg())
685 ERR("Failed reading JPEG because unable to find %s\n", SONAME_LIBJPEG
);
691 if (pUnkOuter
) return CLASS_E_NOAGGREGATION
;
693 This
= HeapAlloc(GetProcessHeap(), 0, sizeof(JpegDecoder
));
694 if (!This
) return E_OUTOFMEMORY
;
696 This
->IWICBitmapDecoder_iface
.lpVtbl
= &JpegDecoder_Vtbl
;
697 This
->IWICBitmapFrameDecode_iface
.lpVtbl
= &JpegDecoder_Frame_Vtbl
;
699 This
->initialized
= FALSE
;
700 This
->cinfo_initialized
= FALSE
;
702 This
->image_data
= NULL
;
703 InitializeCriticalSection(&This
->lock
);
704 This
->lock
.DebugInfo
->Spare
[0] = (DWORD_PTR
)(__FILE__
": JpegDecoder.lock");
706 ret
= IUnknown_QueryInterface((IUnknown
*)This
, iid
, ppv
);
707 IUnknown_Release((IUnknown
*)This
);
712 typedef struct jpeg_compress_format
{
713 const WICPixelFormatGUID
*guid
;
716 J_COLOR_SPACE color_space
;
718 } jpeg_compress_format
;
720 static const jpeg_compress_format compress_formats
[] = {
721 { &GUID_WICPixelFormat24bppBGR
, 24, 3, JCS_RGB
, 1 },
722 { &GUID_WICPixelFormat32bppCMYK
, 32, 4, JCS_CMYK
},
723 { &GUID_WICPixelFormat8bppGray
, 8, 1, JCS_GRAYSCALE
},
727 typedef struct JpegEncoder
{
728 IWICBitmapEncoder IWICBitmapEncoder_iface
;
729 IWICBitmapFrameEncode IWICBitmapFrameEncode_iface
;
731 struct jpeg_compress_struct cinfo
;
732 struct jpeg_error_mgr jerr
;
733 struct jpeg_destination_mgr dest_mgr
;
736 int frame_initialized
;
737 int started_compress
;
742 const jpeg_compress_format
*format
;
744 CRITICAL_SECTION lock
;
745 BYTE dest_buffer
[1024];
748 static inline JpegEncoder
*impl_from_IWICBitmapEncoder(IWICBitmapEncoder
*iface
)
750 return CONTAINING_RECORD(iface
, JpegEncoder
, IWICBitmapEncoder_iface
);
753 static inline JpegEncoder
*impl_from_IWICBitmapFrameEncode(IWICBitmapFrameEncode
*iface
)
755 return CONTAINING_RECORD(iface
, JpegEncoder
, IWICBitmapFrameEncode_iface
);
758 static inline JpegEncoder
*encoder_from_compress(j_compress_ptr compress
)
760 return CONTAINING_RECORD(compress
, JpegEncoder
, cinfo
);
763 static void dest_mgr_init_destination(j_compress_ptr cinfo
)
765 JpegEncoder
*This
= encoder_from_compress(cinfo
);
767 This
->dest_mgr
.next_output_byte
= This
->dest_buffer
;
768 This
->dest_mgr
.free_in_buffer
= sizeof(This
->dest_buffer
);
771 static jpeg_boolean
dest_mgr_empty_output_buffer(j_compress_ptr cinfo
)
773 JpegEncoder
*This
= encoder_from_compress(cinfo
);
777 hr
= IStream_Write(This
->stream
, This
->dest_buffer
,
778 sizeof(This
->dest_buffer
), &byteswritten
);
780 if (hr
!= S_OK
|| byteswritten
== 0)
782 ERR("Failed writing data, hr=%x\n", hr
);
786 This
->dest_mgr
.next_output_byte
= This
->dest_buffer
;
787 This
->dest_mgr
.free_in_buffer
= sizeof(This
->dest_buffer
);
791 static void dest_mgr_term_destination(j_compress_ptr cinfo
)
793 JpegEncoder
*This
= encoder_from_compress(cinfo
);
797 if (This
->dest_mgr
.free_in_buffer
!= sizeof(This
->dest_buffer
))
799 hr
= IStream_Write(This
->stream
, This
->dest_buffer
,
800 sizeof(This
->dest_buffer
) - This
->dest_mgr
.free_in_buffer
, &byteswritten
);
802 if (hr
!= S_OK
|| byteswritten
== 0)
803 ERR("Failed writing data, hr=%x\n", hr
);
807 static HRESULT WINAPI
JpegEncoder_Frame_QueryInterface(IWICBitmapFrameEncode
*iface
, REFIID iid
,
810 JpegEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
811 TRACE("(%p,%s,%p)\n", iface
, debugstr_guid(iid
), ppv
);
813 if (!ppv
) return E_INVALIDARG
;
815 if (IsEqualIID(&IID_IUnknown
, iid
) ||
816 IsEqualIID(&IID_IWICBitmapFrameEncode
, iid
))
818 *ppv
= &This
->IWICBitmapFrameEncode_iface
;
823 return E_NOINTERFACE
;
826 IUnknown_AddRef((IUnknown
*)*ppv
);
830 static ULONG WINAPI
JpegEncoder_Frame_AddRef(IWICBitmapFrameEncode
*iface
)
832 JpegEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
833 return IWICBitmapEncoder_AddRef(&This
->IWICBitmapEncoder_iface
);
836 static ULONG WINAPI
JpegEncoder_Frame_Release(IWICBitmapFrameEncode
*iface
)
838 JpegEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
839 return IWICBitmapEncoder_Release(&This
->IWICBitmapEncoder_iface
);
842 static HRESULT WINAPI
JpegEncoder_Frame_Initialize(IWICBitmapFrameEncode
*iface
,
843 IPropertyBag2
*pIEncoderOptions
)
845 JpegEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
846 TRACE("(%p,%p)\n", iface
, pIEncoderOptions
);
848 EnterCriticalSection(&This
->lock
);
850 if (This
->frame_initialized
)
852 LeaveCriticalSection(&This
->lock
);
853 return WINCODEC_ERR_WRONGSTATE
;
856 This
->frame_initialized
= TRUE
;
858 LeaveCriticalSection(&This
->lock
);
863 static HRESULT WINAPI
JpegEncoder_Frame_SetSize(IWICBitmapFrameEncode
*iface
,
864 UINT uiWidth
, UINT uiHeight
)
866 JpegEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
867 TRACE("(%p,%u,%u)\n", iface
, uiWidth
, uiHeight
);
869 EnterCriticalSection(&This
->lock
);
871 if (!This
->frame_initialized
|| This
->started_compress
)
873 LeaveCriticalSection(&This
->lock
);
874 return WINCODEC_ERR_WRONGSTATE
;
877 This
->width
= uiWidth
;
878 This
->height
= uiHeight
;
880 LeaveCriticalSection(&This
->lock
);
885 static HRESULT WINAPI
JpegEncoder_Frame_SetResolution(IWICBitmapFrameEncode
*iface
,
886 double dpiX
, double dpiY
)
888 JpegEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
889 TRACE("(%p,%0.2f,%0.2f)\n", iface
, dpiX
, dpiY
);
891 EnterCriticalSection(&This
->lock
);
893 if (!This
->frame_initialized
|| This
->started_compress
)
895 LeaveCriticalSection(&This
->lock
);
896 return WINCODEC_ERR_WRONGSTATE
;
902 LeaveCriticalSection(&This
->lock
);
907 static HRESULT WINAPI
JpegEncoder_Frame_SetPixelFormat(IWICBitmapFrameEncode
*iface
,
908 WICPixelFormatGUID
*pPixelFormat
)
910 JpegEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
912 TRACE("(%p,%s)\n", iface
, debugstr_guid(pPixelFormat
));
914 EnterCriticalSection(&This
->lock
);
916 if (!This
->frame_initialized
|| This
->started_compress
)
918 LeaveCriticalSection(&This
->lock
);
919 return WINCODEC_ERR_WRONGSTATE
;
922 for (i
=0; compress_formats
[i
].guid
; i
++)
924 if (memcmp(compress_formats
[i
].guid
, pPixelFormat
, sizeof(GUID
)) == 0)
928 if (!compress_formats
[i
].guid
) i
= 0;
930 This
->format
= &compress_formats
[i
];
931 memcpy(pPixelFormat
, This
->format
->guid
, sizeof(GUID
));
933 LeaveCriticalSection(&This
->lock
);
938 static HRESULT WINAPI
JpegEncoder_Frame_SetColorContexts(IWICBitmapFrameEncode
*iface
,
939 UINT cCount
, IWICColorContext
**ppIColorContext
)
941 FIXME("(%p,%u,%p): stub\n", iface
, cCount
, ppIColorContext
);
945 static HRESULT WINAPI
JpegEncoder_Frame_SetPalette(IWICBitmapFrameEncode
*iface
,
946 IWICPalette
*pIPalette
)
948 FIXME("(%p,%p): stub\n", iface
, pIPalette
);
949 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
952 static HRESULT WINAPI
JpegEncoder_Frame_SetThumbnail(IWICBitmapFrameEncode
*iface
,
953 IWICBitmapSource
*pIThumbnail
)
955 FIXME("(%p,%p): stub\n", iface
, pIThumbnail
);
956 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
959 static HRESULT WINAPI
JpegEncoder_Frame_WritePixels(IWICBitmapFrameEncode
*iface
,
960 UINT lineCount
, UINT cbStride
, UINT cbBufferSize
, BYTE
*pbPixels
)
962 JpegEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
964 BYTE
*swapped_data
= NULL
, *current_row
;
966 TRACE("(%p,%u,%u,%u,%p)\n", iface
, lineCount
, cbStride
, cbBufferSize
, pbPixels
);
968 EnterCriticalSection(&This
->lock
);
970 if (!This
->frame_initialized
|| !This
->width
|| !This
->height
|| !This
->format
)
972 LeaveCriticalSection(&This
->lock
);
973 return WINCODEC_ERR_WRONGSTATE
;
976 if (lineCount
== 0 || lineCount
+ This
->lines_written
> This
->height
)
978 LeaveCriticalSection(&This
->lock
);
982 /* set up setjmp/longjmp error handling */
985 LeaveCriticalSection(&This
->lock
);
986 HeapFree(GetProcessHeap(), 0, swapped_data
);
989 This
->cinfo
.client_data
= &jmpbuf
;
991 if (!This
->started_compress
)
993 This
->cinfo
.image_width
= This
->width
;
994 This
->cinfo
.image_height
= This
->height
;
995 This
->cinfo
.input_components
= This
->format
->num_components
;
996 This
->cinfo
.in_color_space
= This
->format
->color_space
;
998 pjpeg_set_defaults(&This
->cinfo
);
1000 if (This
->xres
!= 0.0 && This
->yres
!= 0.0)
1002 This
->cinfo
.density_unit
= 1; /* dots per inch */
1003 This
->cinfo
.X_density
= This
->xres
;
1004 This
->cinfo
.Y_density
= This
->yres
;
1007 pjpeg_start_compress(&This
->cinfo
, TRUE
);
1009 This
->started_compress
= 1;
1012 row_size
= This
->format
->bpp
/ 8 * This
->width
;
1014 if (This
->format
->swap_rgb
)
1016 swapped_data
= HeapAlloc(GetProcessHeap(), 0, row_size
);
1019 LeaveCriticalSection(&This
->lock
);
1020 return E_OUTOFMEMORY
;
1024 for (line
=0; line
< lineCount
; line
++)
1026 if (This
->format
->swap_rgb
)
1030 memcpy(swapped_data
, pbPixels
+ (cbStride
* line
), row_size
);
1032 for (x
=0; x
< This
->width
; x
++)
1036 b
= swapped_data
[x
*3];
1037 swapped_data
[x
*3] = swapped_data
[x
*3+2];
1038 swapped_data
[x
*3+2] = b
;
1041 current_row
= swapped_data
;
1044 current_row
= pbPixels
+ (cbStride
* line
);
1046 if (!pjpeg_write_scanlines(&This
->cinfo
, ¤t_row
, 1))
1048 ERR("failed writing scanlines\n");
1049 LeaveCriticalSection(&This
->lock
);
1050 HeapFree(GetProcessHeap(), 0, swapped_data
);
1054 This
->lines_written
++;
1057 LeaveCriticalSection(&This
->lock
);
1058 HeapFree(GetProcessHeap(), 0, swapped_data
);
1063 static HRESULT WINAPI
JpegEncoder_Frame_WriteSource(IWICBitmapFrameEncode
*iface
,
1064 IWICBitmapSource
*pIBitmapSource
, WICRect
*prc
)
1066 FIXME("(%p,%p,%p): stub\n", iface
, pIBitmapSource
, prc
);
1070 static HRESULT WINAPI
JpegEncoder_Frame_Commit(IWICBitmapFrameEncode
*iface
)
1072 JpegEncoder
*This
= impl_from_IWICBitmapFrameEncode(iface
);
1074 TRACE("(%p)\n", iface
);
1076 EnterCriticalSection(&This
->lock
);
1078 if (!This
->started_compress
|| This
->lines_written
!= This
->height
|| This
->frame_committed
)
1080 LeaveCriticalSection(&This
->lock
);
1081 return WINCODEC_ERR_WRONGSTATE
;
1084 /* set up setjmp/longjmp error handling */
1087 LeaveCriticalSection(&This
->lock
);
1090 This
->cinfo
.client_data
= &jmpbuf
;
1092 pjpeg_finish_compress(&This
->cinfo
);
1094 This
->frame_committed
= TRUE
;
1096 LeaveCriticalSection(&This
->lock
);
1101 static HRESULT WINAPI
JpegEncoder_Frame_GetMetadataQueryWriter(IWICBitmapFrameEncode
*iface
,
1102 IWICMetadataQueryWriter
**ppIMetadataQueryWriter
)
1104 FIXME("(%p, %p): stub\n", iface
, ppIMetadataQueryWriter
);
1108 static const IWICBitmapFrameEncodeVtbl JpegEncoder_FrameVtbl
= {
1109 JpegEncoder_Frame_QueryInterface
,
1110 JpegEncoder_Frame_AddRef
,
1111 JpegEncoder_Frame_Release
,
1112 JpegEncoder_Frame_Initialize
,
1113 JpegEncoder_Frame_SetSize
,
1114 JpegEncoder_Frame_SetResolution
,
1115 JpegEncoder_Frame_SetPixelFormat
,
1116 JpegEncoder_Frame_SetColorContexts
,
1117 JpegEncoder_Frame_SetPalette
,
1118 JpegEncoder_Frame_SetThumbnail
,
1119 JpegEncoder_Frame_WritePixels
,
1120 JpegEncoder_Frame_WriteSource
,
1121 JpegEncoder_Frame_Commit
,
1122 JpegEncoder_Frame_GetMetadataQueryWriter
1125 static HRESULT WINAPI
JpegEncoder_QueryInterface(IWICBitmapEncoder
*iface
, REFIID iid
,
1128 JpegEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1129 TRACE("(%p,%s,%p)\n", iface
, debugstr_guid(iid
), ppv
);
1131 if (!ppv
) return E_INVALIDARG
;
1133 if (IsEqualIID(&IID_IUnknown
, iid
) ||
1134 IsEqualIID(&IID_IWICBitmapEncoder
, iid
))
1141 return E_NOINTERFACE
;
1144 IUnknown_AddRef((IUnknown
*)*ppv
);
1148 static ULONG WINAPI
JpegEncoder_AddRef(IWICBitmapEncoder
*iface
)
1150 JpegEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1151 ULONG ref
= InterlockedIncrement(&This
->ref
);
1153 TRACE("(%p) refcount=%u\n", iface
, ref
);
1158 static ULONG WINAPI
JpegEncoder_Release(IWICBitmapEncoder
*iface
)
1160 JpegEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1161 ULONG ref
= InterlockedDecrement(&This
->ref
);
1163 TRACE("(%p) refcount=%u\n", iface
, ref
);
1167 This
->lock
.DebugInfo
->Spare
[0] = 0;
1168 DeleteCriticalSection(&This
->lock
);
1169 if (This
->initialized
) pjpeg_destroy_compress(&This
->cinfo
);
1170 if (This
->stream
) IStream_Release(This
->stream
);
1171 HeapFree(GetProcessHeap(), 0, This
);
1177 static HRESULT WINAPI
JpegEncoder_Initialize(IWICBitmapEncoder
*iface
,
1178 IStream
*pIStream
, WICBitmapEncoderCacheOption cacheOption
)
1180 JpegEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1183 TRACE("(%p,%p,%u)\n", iface
, pIStream
, cacheOption
);
1185 EnterCriticalSection(&This
->lock
);
1187 if (This
->initialized
)
1189 LeaveCriticalSection(&This
->lock
);
1190 return WINCODEC_ERR_WRONGSTATE
;
1193 pjpeg_std_error(&This
->jerr
);
1195 This
->jerr
.error_exit
= error_exit_fn
;
1196 This
->jerr
.emit_message
= emit_message_fn
;
1198 This
->cinfo
.err
= &This
->jerr
;
1200 This
->cinfo
.client_data
= &jmpbuf
;
1204 LeaveCriticalSection(&This
->lock
);
1208 pjpeg_CreateCompress(&This
->cinfo
, JPEG_LIB_VERSION
, sizeof(struct jpeg_compress_struct
));
1210 This
->stream
= pIStream
;
1211 IStream_AddRef(pIStream
);
1213 This
->dest_mgr
.next_output_byte
= This
->dest_buffer
;
1214 This
->dest_mgr
.free_in_buffer
= sizeof(This
->dest_buffer
);
1216 This
->dest_mgr
.init_destination
= dest_mgr_init_destination
;
1217 This
->dest_mgr
.empty_output_buffer
= dest_mgr_empty_output_buffer
;
1218 This
->dest_mgr
.term_destination
= dest_mgr_term_destination
;
1220 This
->cinfo
.dest
= &This
->dest_mgr
;
1222 This
->initialized
= TRUE
;
1224 LeaveCriticalSection(&This
->lock
);
1229 static HRESULT WINAPI
JpegEncoder_GetContainerFormat(IWICBitmapEncoder
*iface
,
1230 GUID
*pguidContainerFormat
)
1232 FIXME("(%p,%s): stub\n", iface
, debugstr_guid(pguidContainerFormat
));
1236 static HRESULT WINAPI
JpegEncoder_GetEncoderInfo(IWICBitmapEncoder
*iface
,
1237 IWICBitmapEncoderInfo
**ppIEncoderInfo
)
1239 FIXME("(%p,%p): stub\n", iface
, ppIEncoderInfo
);
1243 static HRESULT WINAPI
JpegEncoder_SetColorContexts(IWICBitmapEncoder
*iface
,
1244 UINT cCount
, IWICColorContext
**ppIColorContext
)
1246 FIXME("(%p,%u,%p): stub\n", iface
, cCount
, ppIColorContext
);
1250 static HRESULT WINAPI
JpegEncoder_SetPalette(IWICBitmapEncoder
*iface
, IWICPalette
*pIPalette
)
1252 TRACE("(%p,%p)\n", iface
, pIPalette
);
1253 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
1256 static HRESULT WINAPI
JpegEncoder_SetThumbnail(IWICBitmapEncoder
*iface
, IWICBitmapSource
*pIThumbnail
)
1258 TRACE("(%p,%p)\n", iface
, pIThumbnail
);
1259 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
1262 static HRESULT WINAPI
JpegEncoder_SetPreview(IWICBitmapEncoder
*iface
, IWICBitmapSource
*pIPreview
)
1264 TRACE("(%p,%p)\n", iface
, pIPreview
);
1265 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
1268 static HRESULT WINAPI
JpegEncoder_CreateNewFrame(IWICBitmapEncoder
*iface
,
1269 IWICBitmapFrameEncode
**ppIFrameEncode
, IPropertyBag2
**ppIEncoderOptions
)
1271 JpegEncoder
*This
= impl_from_IWICBitmapEncoder(iface
);
1274 TRACE("(%p,%p,%p)\n", iface
, ppIFrameEncode
, ppIEncoderOptions
);
1276 EnterCriticalSection(&This
->lock
);
1278 if (This
->frame_count
!= 0)
1280 LeaveCriticalSection(&This
->lock
);
1281 return WINCODEC_ERR_UNSUPPORTEDOPERATION
;
1284 if (!This
->initialized
)
1286 LeaveCriticalSection(&This
->lock
);
1287 return WINCODEC_ERR_NOTINITIALIZED
;
1290 hr
= CreatePropertyBag2(ppIEncoderOptions
);
1293 LeaveCriticalSection(&This
->lock
);
1297 This
->frame_count
= 1;
1299 LeaveCriticalSection(&This
->lock
);
1301 IWICBitmapEncoder_AddRef(iface
);
1302 *ppIFrameEncode
= &This
->IWICBitmapFrameEncode_iface
;
1307 static HRESULT WINAPI
JpegEncoder_Commit(IWICBitmapEncoder
*iface
)
1309 FIXME("(%p): stub\n", iface
);
1313 static HRESULT WINAPI
JpegEncoder_GetMetadataQueryWriter(IWICBitmapEncoder
*iface
,
1314 IWICMetadataQueryWriter
**ppIMetadataQueryWriter
)
1316 FIXME("(%p,%p): stub\n", iface
, ppIMetadataQueryWriter
);
1320 static const IWICBitmapEncoderVtbl JpegEncoder_Vtbl
= {
1321 JpegEncoder_QueryInterface
,
1323 JpegEncoder_Release
,
1324 JpegEncoder_Initialize
,
1325 JpegEncoder_GetContainerFormat
,
1326 JpegEncoder_GetEncoderInfo
,
1327 JpegEncoder_SetColorContexts
,
1328 JpegEncoder_SetPalette
,
1329 JpegEncoder_SetThumbnail
,
1330 JpegEncoder_SetPreview
,
1331 JpegEncoder_CreateNewFrame
,
1333 JpegEncoder_GetMetadataQueryWriter
1336 HRESULT
JpegEncoder_CreateInstance(IUnknown
*pUnkOuter
, REFIID iid
, void** ppv
)
1341 TRACE("(%p,%s,%p)\n", pUnkOuter
, debugstr_guid(iid
), ppv
);
1345 if (pUnkOuter
) return CLASS_E_NOAGGREGATION
;
1347 if (!libjpeg_handle
&& !load_libjpeg())
1349 ERR("Failed writing JPEG because unable to find %s\n",SONAME_LIBJPEG
);
1353 This
= HeapAlloc(GetProcessHeap(), 0, sizeof(JpegEncoder
));
1354 if (!This
) return E_OUTOFMEMORY
;
1356 This
->IWICBitmapEncoder_iface
.lpVtbl
= &JpegEncoder_Vtbl
;
1357 This
->IWICBitmapFrameEncode_iface
.lpVtbl
= &JpegEncoder_FrameVtbl
;
1359 This
->initialized
= 0;
1360 This
->frame_count
= 0;
1361 This
->frame_initialized
= 0;
1362 This
->started_compress
= 0;
1363 This
->lines_written
= 0;
1364 This
->frame_committed
= 0;
1365 This
->width
= This
->height
= 0;
1366 This
->xres
= This
->yres
= 0.0;
1367 This
->format
= NULL
;
1368 This
->stream
= NULL
;
1369 InitializeCriticalSection(&This
->lock
);
1370 This
->lock
.DebugInfo
->Spare
[0] = (DWORD_PTR
)(__FILE__
": JpegEncoder.lock");
1372 ret
= IUnknown_QueryInterface((IUnknown
*)This
, iid
, ppv
);
1373 IUnknown_Release((IUnknown
*)This
);
1378 #else /* !defined(SONAME_LIBJPEG) */
1380 HRESULT
JpegDecoder_CreateInstance(IUnknown
*pUnkOuter
, REFIID iid
, void** ppv
)
1382 ERR("Trying to load JPEG picture, but JPEG support is not compiled in.\n");
1386 HRESULT
JpegEncoder_CreateInstance(IUnknown
*pUnkOuter
, REFIID iid
, void** ppv
)
1388 ERR("Trying to save JPEG picture, but JPEG support is not compiled in.\n");