Release 1.6-rc2.
[wine/testsucceed.git] / dlls / windowscodecs / jpegformat.c
blobc742f786f9a18e927989948feabf342a90069726
1 /*
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
19 #include "config.h"
20 #include "wine/port.h"
22 #ifdef HAVE_UNISTD_H
23 # include <unistd.h>
24 #endif
25 #include <stdarg.h>
26 #include <stdio.h>
27 #include <string.h>
28 #include <setjmp.h>
30 #ifdef SONAME_LIBJPEG
31 /* This is a hack, so jpeglib.h does not redefine INT32 and the like*/
32 #define XMD_H
33 #define UINT8 JPEG_UINT8
34 #define UINT16 JPEG_UINT16
35 #define boolean jpeg_boolean
36 #undef HAVE_STDLIB_H
37 # include <jpeglib.h>
38 #undef HAVE_STDLIB_H
39 #define HAVE_STDLIB_H 1
40 #undef UINT8
41 #undef UINT16
42 #undef boolean
43 #endif
45 #define COBJMACROS
47 #include "windef.h"
48 #include "winbase.h"
49 #include "objbase.h"
50 #include "wincodec.h"
52 #include "wincodecs_private.h"
54 #include "wine/debug.h"
55 #include "wine/library.h"
57 WINE_DEFAULT_DEBUG_CHANNEL(wincodecs);
59 #ifdef SONAME_LIBJPEG
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);
78 #undef MAKE_FUNCPTR
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; \
87 return 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);
103 #undef LOAD_FUNCPTR
105 return libjpeg_handle;
108 static void error_exit_fn(j_common_ptr cinfo)
110 char message[JMSG_LENGTH_MAX];
111 if (ERR_ON(jpeg))
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);
140 typedef struct {
141 IWICBitmapDecoder IWICBitmapDecoder_iface;
142 IWICBitmapFrameDecode IWICBitmapFrameDecode_iface;
143 LONG ref;
144 BOOL initialized;
145 BOOL cinfo_initialized;
146 IStream *stream;
147 struct jpeg_decompress_struct cinfo;
148 struct jpeg_error_mgr jerr;
149 struct jpeg_source_mgr source_mgr;
150 BYTE source_buffer[1024];
151 BYTE *image_data;
152 CRITICAL_SECTION lock;
153 } JpegDecoder;
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,
171 void **ppv)
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) ||
179 IsEqualIID(&IID_IWICBitmapDecoder, iid))
181 *ppv = &This->IWICBitmapDecoder_iface;
183 else
185 *ppv = NULL;
186 return E_NOINTERFACE;
189 IUnknown_AddRef((IUnknown*)*ppv);
190 return S_OK;
193 static ULONG WINAPI JpegDecoder_AddRef(IWICBitmapDecoder *iface)
195 JpegDecoder *This = impl_from_IWICBitmapDecoder(iface);
196 ULONG ref = InterlockedIncrement(&This->ref);
198 TRACE("(%p) refcount=%u\n", iface, ref);
200 return ref;
203 static ULONG WINAPI JpegDecoder_Release(IWICBitmapDecoder *iface)
205 JpegDecoder *This = impl_from_IWICBitmapDecoder(iface);
206 ULONG ref = InterlockedDecrement(&This->ref);
208 TRACE("(%p) refcount=%u\n", iface, ref);
210 if (ref == 0)
212 This->lock.DebugInfo->Spare[0] = 0;
213 DeleteCriticalSection(&This->lock);
214 if (This->cinfo_initialized) pjpeg_destroy_decompress(&This->cinfo);
215 if (This->stream) IStream_Release(This->stream);
216 HeapFree(GetProcessHeap(), 0, This->image_data);
217 HeapFree(GetProcessHeap(), 0, This);
220 return ref;
223 static HRESULT WINAPI JpegDecoder_QueryCapability(IWICBitmapDecoder *iface, IStream *stream,
224 DWORD *capability)
226 HRESULT hr;
228 TRACE("(%p,%p,%p)\n", iface, stream, capability);
230 if (!stream || !capability) return E_INVALIDARG;
232 hr = IWICBitmapDecoder_Initialize(iface, stream, WICDecodeMetadataCacheOnDemand);
233 if (hr != S_OK) return hr;
235 *capability = WICBitmapDecoderCapabilityCanDecodeAllImages |
236 WICBitmapDecoderCapabilityCanDecodeSomeImages;
237 /* FIXME: WICBitmapDecoderCapabilityCanEnumerateMetadata */
238 return S_OK;
241 static void source_mgr_init_source(j_decompress_ptr cinfo)
245 static jpeg_boolean source_mgr_fill_input_buffer(j_decompress_ptr cinfo)
247 JpegDecoder *This = decoder_from_decompress(cinfo);
248 HRESULT hr;
249 ULONG bytesread;
251 hr = IStream_Read(This->stream, This->source_buffer, 1024, &bytesread);
253 if (hr != S_OK || bytesread == 0)
255 return FALSE;
257 else
259 This->source_mgr.next_input_byte = This->source_buffer;
260 This->source_mgr.bytes_in_buffer = bytesread;
261 return TRUE;
265 static void source_mgr_skip_input_data(j_decompress_ptr cinfo, long num_bytes)
267 JpegDecoder *This = decoder_from_decompress(cinfo);
268 LARGE_INTEGER seek;
270 if (num_bytes > This->source_mgr.bytes_in_buffer)
272 seek.QuadPart = num_bytes - This->source_mgr.bytes_in_buffer;
273 IStream_Seek(This->stream, seek, STREAM_SEEK_CUR, NULL);
274 This->source_mgr.bytes_in_buffer = 0;
276 else if (num_bytes > 0)
278 This->source_mgr.next_input_byte += num_bytes;
279 This->source_mgr.bytes_in_buffer -= num_bytes;
283 static void source_mgr_term_source(j_decompress_ptr cinfo)
287 static HRESULT WINAPI JpegDecoder_Initialize(IWICBitmapDecoder *iface, IStream *pIStream,
288 WICDecodeOptions cacheOptions)
290 JpegDecoder *This = impl_from_IWICBitmapDecoder(iface);
291 int ret;
292 LARGE_INTEGER seek;
293 jmp_buf jmpbuf;
294 TRACE("(%p,%p,%u)\n", iface, pIStream, cacheOptions);
296 EnterCriticalSection(&This->lock);
298 if (This->cinfo_initialized)
300 LeaveCriticalSection(&This->lock);
301 return WINCODEC_ERR_WRONGSTATE;
304 pjpeg_std_error(&This->jerr);
306 This->jerr.error_exit = error_exit_fn;
307 This->jerr.emit_message = emit_message_fn;
309 This->cinfo.err = &This->jerr;
311 This->cinfo.client_data = jmpbuf;
313 if (setjmp(jmpbuf))
315 LeaveCriticalSection(&This->lock);
316 return E_FAIL;
319 pjpeg_CreateDecompress(&This->cinfo, JPEG_LIB_VERSION, sizeof(struct jpeg_decompress_struct));
321 This->cinfo_initialized = TRUE;
323 This->stream = pIStream;
324 IStream_AddRef(pIStream);
326 seek.QuadPart = 0;
327 IStream_Seek(This->stream, seek, STREAM_SEEK_SET, NULL);
329 This->source_mgr.bytes_in_buffer = 0;
330 This->source_mgr.init_source = source_mgr_init_source;
331 This->source_mgr.fill_input_buffer = source_mgr_fill_input_buffer;
332 This->source_mgr.skip_input_data = source_mgr_skip_input_data;
333 This->source_mgr.resync_to_restart = pjpeg_resync_to_restart;
334 This->source_mgr.term_source = source_mgr_term_source;
336 This->cinfo.src = &This->source_mgr;
338 ret = pjpeg_read_header(&This->cinfo, TRUE);
340 if (ret != JPEG_HEADER_OK) {
341 WARN("Jpeg image in stream has bad format, read header returned %d.\n",ret);
342 LeaveCriticalSection(&This->lock);
343 return E_FAIL;
346 switch (This->cinfo.jpeg_color_space)
348 case JCS_GRAYSCALE:
349 This->cinfo.out_color_space = JCS_GRAYSCALE;
350 break;
351 case JCS_RGB:
352 case JCS_YCbCr:
353 This->cinfo.out_color_space = JCS_RGB;
354 break;
355 case JCS_CMYK:
356 case JCS_YCCK:
357 This->cinfo.out_color_space = JCS_CMYK;
358 break;
359 default:
360 ERR("Unknown JPEG color space %i\n", This->cinfo.jpeg_color_space);
361 LeaveCriticalSection(&This->lock);
362 return E_FAIL;
365 if (!pjpeg_start_decompress(&This->cinfo))
367 ERR("jpeg_start_decompress failed\n");
368 LeaveCriticalSection(&This->lock);
369 return E_FAIL;
372 This->initialized = TRUE;
374 LeaveCriticalSection(&This->lock);
376 return S_OK;
379 static HRESULT WINAPI JpegDecoder_GetContainerFormat(IWICBitmapDecoder *iface,
380 GUID *pguidContainerFormat)
382 memcpy(pguidContainerFormat, &GUID_ContainerFormatJpeg, sizeof(GUID));
383 return S_OK;
386 static HRESULT WINAPI JpegDecoder_GetDecoderInfo(IWICBitmapDecoder *iface,
387 IWICBitmapDecoderInfo **ppIDecoderInfo)
389 HRESULT hr;
390 IWICComponentInfo *compinfo;
392 TRACE("(%p,%p)\n", iface, ppIDecoderInfo);
394 hr = CreateComponentInfo(&CLSID_WICJpegDecoder, &compinfo);
395 if (FAILED(hr)) return hr;
397 hr = IWICComponentInfo_QueryInterface(compinfo, &IID_IWICBitmapDecoderInfo,
398 (void**)ppIDecoderInfo);
400 IWICComponentInfo_Release(compinfo);
402 return hr;
405 static HRESULT WINAPI JpegDecoder_CopyPalette(IWICBitmapDecoder *iface,
406 IWICPalette *pIPalette)
408 TRACE("(%p,%p)\n", iface, pIPalette);
410 return WINCODEC_ERR_PALETTEUNAVAILABLE;
413 static HRESULT WINAPI JpegDecoder_GetMetadataQueryReader(IWICBitmapDecoder *iface,
414 IWICMetadataQueryReader **ppIMetadataQueryReader)
416 FIXME("(%p,%p): stub\n", iface, ppIMetadataQueryReader);
417 return E_NOTIMPL;
420 static HRESULT WINAPI JpegDecoder_GetPreview(IWICBitmapDecoder *iface,
421 IWICBitmapSource **ppIBitmapSource)
423 FIXME("(%p,%p): stub\n", iface, ppIBitmapSource);
424 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
427 static HRESULT WINAPI JpegDecoder_GetColorContexts(IWICBitmapDecoder *iface,
428 UINT cCount, IWICColorContext **ppIColorContexts, UINT *pcActualCount)
430 FIXME("(%p,%u,%p,%p): stub\n", iface, cCount, ppIColorContexts, pcActualCount);
431 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
434 static HRESULT WINAPI JpegDecoder_GetThumbnail(IWICBitmapDecoder *iface,
435 IWICBitmapSource **ppIThumbnail)
437 FIXME("(%p,%p): stub\n", iface, ppIThumbnail);
438 return WINCODEC_ERR_CODECNOTHUMBNAIL;
441 static HRESULT WINAPI JpegDecoder_GetFrameCount(IWICBitmapDecoder *iface,
442 UINT *pCount)
444 if (!pCount) return E_INVALIDARG;
446 *pCount = 1;
447 return S_OK;
450 static HRESULT WINAPI JpegDecoder_GetFrame(IWICBitmapDecoder *iface,
451 UINT index, IWICBitmapFrameDecode **ppIBitmapFrame)
453 JpegDecoder *This = impl_from_IWICBitmapDecoder(iface);
454 TRACE("(%p,%u,%p)\n", iface, index, ppIBitmapFrame);
456 if (!This->initialized) return WINCODEC_ERR_FRAMEMISSING;
458 if (index != 0) return E_INVALIDARG;
460 IWICBitmapDecoder_AddRef(iface);
461 *ppIBitmapFrame = &This->IWICBitmapFrameDecode_iface;
463 return S_OK;
466 static const IWICBitmapDecoderVtbl JpegDecoder_Vtbl = {
467 JpegDecoder_QueryInterface,
468 JpegDecoder_AddRef,
469 JpegDecoder_Release,
470 JpegDecoder_QueryCapability,
471 JpegDecoder_Initialize,
472 JpegDecoder_GetContainerFormat,
473 JpegDecoder_GetDecoderInfo,
474 JpegDecoder_CopyPalette,
475 JpegDecoder_GetMetadataQueryReader,
476 JpegDecoder_GetPreview,
477 JpegDecoder_GetColorContexts,
478 JpegDecoder_GetThumbnail,
479 JpegDecoder_GetFrameCount,
480 JpegDecoder_GetFrame
483 static HRESULT WINAPI JpegDecoder_Frame_QueryInterface(IWICBitmapFrameDecode *iface, REFIID iid,
484 void **ppv)
486 JpegDecoder *This = impl_from_IWICBitmapFrameDecode(iface);
488 TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
490 if (!ppv) return E_INVALIDARG;
492 if (IsEqualIID(&IID_IUnknown, iid) ||
493 IsEqualIID(&IID_IWICBitmapSource, iid) ||
494 IsEqualIID(&IID_IWICBitmapFrameDecode, iid))
496 *ppv = &This->IWICBitmapFrameDecode_iface;
498 else
500 *ppv = NULL;
501 return E_NOINTERFACE;
504 IUnknown_AddRef((IUnknown*)*ppv);
505 return S_OK;
508 static ULONG WINAPI JpegDecoder_Frame_AddRef(IWICBitmapFrameDecode *iface)
510 JpegDecoder *This = impl_from_IWICBitmapFrameDecode(iface);
511 return IWICBitmapDecoder_AddRef(&This->IWICBitmapDecoder_iface);
514 static ULONG WINAPI JpegDecoder_Frame_Release(IWICBitmapFrameDecode *iface)
516 JpegDecoder *This = impl_from_IWICBitmapFrameDecode(iface);
517 return IWICBitmapDecoder_Release(&This->IWICBitmapDecoder_iface);
520 static HRESULT WINAPI JpegDecoder_Frame_GetSize(IWICBitmapFrameDecode *iface,
521 UINT *puiWidth, UINT *puiHeight)
523 JpegDecoder *This = impl_from_IWICBitmapFrameDecode(iface);
524 *puiWidth = This->cinfo.output_width;
525 *puiHeight = This->cinfo.output_height;
526 TRACE("(%p)->(%u,%u)\n", iface, *puiWidth, *puiHeight);
527 return S_OK;
530 static HRESULT WINAPI JpegDecoder_Frame_GetPixelFormat(IWICBitmapFrameDecode *iface,
531 WICPixelFormatGUID *pPixelFormat)
533 JpegDecoder *This = impl_from_IWICBitmapFrameDecode(iface);
534 TRACE("(%p,%p)\n", iface, pPixelFormat);
535 if (This->cinfo.out_color_space == JCS_RGB)
536 memcpy(pPixelFormat, &GUID_WICPixelFormat24bppBGR, sizeof(GUID));
537 else if (This->cinfo.out_color_space == JCS_CMYK)
538 memcpy(pPixelFormat, &GUID_WICPixelFormat32bppCMYK, sizeof(GUID));
539 else /* This->cinfo.out_color_space == JCS_GRAYSCALE */
540 memcpy(pPixelFormat, &GUID_WICPixelFormat8bppGray, sizeof(GUID));
541 return S_OK;
544 static HRESULT WINAPI JpegDecoder_Frame_GetResolution(IWICBitmapFrameDecode *iface,
545 double *pDpiX, double *pDpiY)
547 JpegDecoder *This = impl_from_IWICBitmapFrameDecode(iface);
549 EnterCriticalSection(&This->lock);
551 switch (This->cinfo.density_unit)
553 case 2: /* pixels per centimeter */
554 *pDpiX = This->cinfo.X_density * 2.54;
555 *pDpiY = This->cinfo.Y_density * 2.54;
556 break;
558 case 1: /* pixels per inch */
559 *pDpiX = This->cinfo.X_density;
560 *pDpiY = This->cinfo.Y_density;
561 break;
563 case 0: /* unknown */
564 default:
565 *pDpiX = 96.0;
566 *pDpiY = 96.0;
567 break;
570 LeaveCriticalSection(&This->lock);
572 TRACE("(%p)->(%0.2f,%0.2f)\n", iface, *pDpiX, *pDpiY);
574 return S_OK;
577 static HRESULT WINAPI JpegDecoder_Frame_CopyPalette(IWICBitmapFrameDecode *iface,
578 IWICPalette *pIPalette)
580 FIXME("(%p,%p): stub\n", iface, pIPalette);
581 return E_NOTIMPL;
584 static HRESULT WINAPI JpegDecoder_Frame_CopyPixels(IWICBitmapFrameDecode *iface,
585 const WICRect *prc, UINT cbStride, UINT cbBufferSize, BYTE *pbBuffer)
587 JpegDecoder *This = impl_from_IWICBitmapFrameDecode(iface);
588 UINT bpp;
589 UINT stride;
590 UINT data_size;
591 UINT max_row_needed;
592 jmp_buf jmpbuf;
593 WICRect rect;
594 TRACE("(%p,%p,%u,%u,%p)\n", iface, prc, cbStride, cbBufferSize, pbBuffer);
596 if (!prc)
598 rect.X = 0;
599 rect.Y = 0;
600 rect.Width = This->cinfo.output_width;
601 rect.Height = This->cinfo.output_height;
602 prc = &rect;
604 else
606 if (prc->X < 0 || prc->Y < 0 || prc->X+prc->Width > This->cinfo.output_width ||
607 prc->Y+prc->Height > This->cinfo.output_height)
608 return E_INVALIDARG;
611 if (This->cinfo.out_color_space == JCS_GRAYSCALE) bpp = 8;
612 else if (This->cinfo.out_color_space == JCS_CMYK) bpp = 32;
613 else bpp = 24;
615 stride = bpp * This->cinfo.output_width;
616 data_size = stride * This->cinfo.output_height;
618 max_row_needed = prc->Y + prc->Height;
619 if (max_row_needed > This->cinfo.output_height) return E_INVALIDARG;
621 EnterCriticalSection(&This->lock);
623 if (!This->image_data)
625 This->image_data = HeapAlloc(GetProcessHeap(), 0, data_size);
626 if (!This->image_data)
628 LeaveCriticalSection(&This->lock);
629 return E_OUTOFMEMORY;
633 This->cinfo.client_data = jmpbuf;
635 if (setjmp(jmpbuf))
637 LeaveCriticalSection(&This->lock);
638 return E_FAIL;
641 while (max_row_needed > This->cinfo.output_scanline)
643 UINT first_scanline = This->cinfo.output_scanline;
644 UINT max_rows;
645 JSAMPROW out_rows[4];
646 UINT i;
647 JDIMENSION ret;
649 max_rows = min(This->cinfo.output_height-first_scanline, 4);
650 for (i=0; i<max_rows; i++)
651 out_rows[i] = This->image_data + stride * (first_scanline+i);
653 ret = pjpeg_read_scanlines(&This->cinfo, out_rows, max_rows);
655 if (ret == 0)
657 ERR("read_scanlines failed\n");
658 LeaveCriticalSection(&This->lock);
659 return E_FAIL;
662 if (bpp == 24)
664 /* libjpeg gives us RGB data and we want BGR, so byteswap the data */
665 reverse_bgr8(3, This->image_data + stride * first_scanline,
666 This->cinfo.output_width, This->cinfo.output_scanline - first_scanline,
667 stride);
670 if (This->cinfo.out_color_space == JCS_CMYK && This->cinfo.saw_Adobe_marker)
671 /* Adobe JPEG's have inverted CMYK data. */
672 for (i=0; i<data_size; i++)
673 This->image_data[i] ^= 0xff;
676 LeaveCriticalSection(&This->lock);
678 return copy_pixels(bpp, This->image_data,
679 This->cinfo.output_width, This->cinfo.output_height, stride,
680 prc, cbStride, cbBufferSize, pbBuffer);
683 static HRESULT WINAPI JpegDecoder_Frame_GetMetadataQueryReader(IWICBitmapFrameDecode *iface,
684 IWICMetadataQueryReader **ppIMetadataQueryReader)
686 FIXME("(%p,%p): stub\n", iface, ppIMetadataQueryReader);
687 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
690 static HRESULT WINAPI JpegDecoder_Frame_GetColorContexts(IWICBitmapFrameDecode *iface,
691 UINT cCount, IWICColorContext **ppIColorContexts, UINT *pcActualCount)
693 FIXME("(%p,%u,%p,%p): stub\n", iface, cCount, ppIColorContexts, pcActualCount);
694 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
697 static HRESULT WINAPI JpegDecoder_Frame_GetThumbnail(IWICBitmapFrameDecode *iface,
698 IWICBitmapSource **ppIThumbnail)
700 FIXME("(%p,%p): stub\n", iface, ppIThumbnail);
701 return WINCODEC_ERR_CODECNOTHUMBNAIL;
704 static const IWICBitmapFrameDecodeVtbl JpegDecoder_Frame_Vtbl = {
705 JpegDecoder_Frame_QueryInterface,
706 JpegDecoder_Frame_AddRef,
707 JpegDecoder_Frame_Release,
708 JpegDecoder_Frame_GetSize,
709 JpegDecoder_Frame_GetPixelFormat,
710 JpegDecoder_Frame_GetResolution,
711 JpegDecoder_Frame_CopyPalette,
712 JpegDecoder_Frame_CopyPixels,
713 JpegDecoder_Frame_GetMetadataQueryReader,
714 JpegDecoder_Frame_GetColorContexts,
715 JpegDecoder_Frame_GetThumbnail
718 HRESULT JpegDecoder_CreateInstance(IUnknown *pUnkOuter, REFIID iid, void** ppv)
720 JpegDecoder *This;
721 HRESULT ret;
723 TRACE("(%p,%s,%p)\n", pUnkOuter, debugstr_guid(iid), ppv);
725 if (!libjpeg_handle && !load_libjpeg())
727 ERR("Failed reading JPEG because unable to find %s\n", SONAME_LIBJPEG);
728 return E_FAIL;
731 *ppv = NULL;
733 if (pUnkOuter) return CLASS_E_NOAGGREGATION;
735 This = HeapAlloc(GetProcessHeap(), 0, sizeof(JpegDecoder));
736 if (!This) return E_OUTOFMEMORY;
738 This->IWICBitmapDecoder_iface.lpVtbl = &JpegDecoder_Vtbl;
739 This->IWICBitmapFrameDecode_iface.lpVtbl = &JpegDecoder_Frame_Vtbl;
740 This->ref = 1;
741 This->initialized = FALSE;
742 This->cinfo_initialized = FALSE;
743 This->stream = NULL;
744 This->image_data = NULL;
745 InitializeCriticalSection(&This->lock);
746 This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": JpegDecoder.lock");
748 ret = IWICBitmapDecoder_QueryInterface(&This->IWICBitmapDecoder_iface, iid, ppv);
749 IWICBitmapDecoder_Release(&This->IWICBitmapDecoder_iface);
751 return ret;
754 typedef struct jpeg_compress_format {
755 const WICPixelFormatGUID *guid;
756 int bpp;
757 int num_components;
758 J_COLOR_SPACE color_space;
759 int swap_rgb;
760 } jpeg_compress_format;
762 static const jpeg_compress_format compress_formats[] = {
763 { &GUID_WICPixelFormat24bppBGR, 24, 3, JCS_RGB, 1 },
764 { &GUID_WICPixelFormat32bppCMYK, 32, 4, JCS_CMYK },
765 { &GUID_WICPixelFormat8bppGray, 8, 1, JCS_GRAYSCALE },
766 { 0 }
769 typedef struct JpegEncoder {
770 IWICBitmapEncoder IWICBitmapEncoder_iface;
771 IWICBitmapFrameEncode IWICBitmapFrameEncode_iface;
772 LONG ref;
773 struct jpeg_compress_struct cinfo;
774 struct jpeg_error_mgr jerr;
775 struct jpeg_destination_mgr dest_mgr;
776 int initialized;
777 int frame_count;
778 int frame_initialized;
779 int started_compress;
780 int lines_written;
781 int frame_committed;
782 int committed;
783 UINT width, height;
784 double xres, yres;
785 const jpeg_compress_format *format;
786 IStream *stream;
787 CRITICAL_SECTION lock;
788 BYTE dest_buffer[1024];
789 } JpegEncoder;
791 static inline JpegEncoder *impl_from_IWICBitmapEncoder(IWICBitmapEncoder *iface)
793 return CONTAINING_RECORD(iface, JpegEncoder, IWICBitmapEncoder_iface);
796 static inline JpegEncoder *impl_from_IWICBitmapFrameEncode(IWICBitmapFrameEncode *iface)
798 return CONTAINING_RECORD(iface, JpegEncoder, IWICBitmapFrameEncode_iface);
801 static inline JpegEncoder *encoder_from_compress(j_compress_ptr compress)
803 return CONTAINING_RECORD(compress, JpegEncoder, cinfo);
806 static void dest_mgr_init_destination(j_compress_ptr cinfo)
808 JpegEncoder *This = encoder_from_compress(cinfo);
810 This->dest_mgr.next_output_byte = This->dest_buffer;
811 This->dest_mgr.free_in_buffer = sizeof(This->dest_buffer);
814 static jpeg_boolean dest_mgr_empty_output_buffer(j_compress_ptr cinfo)
816 JpegEncoder *This = encoder_from_compress(cinfo);
817 HRESULT hr;
818 ULONG byteswritten;
820 hr = IStream_Write(This->stream, This->dest_buffer,
821 sizeof(This->dest_buffer), &byteswritten);
823 if (hr != S_OK || byteswritten == 0)
825 ERR("Failed writing data, hr=%x\n", hr);
826 return FALSE;
829 This->dest_mgr.next_output_byte = This->dest_buffer;
830 This->dest_mgr.free_in_buffer = sizeof(This->dest_buffer);
831 return TRUE;
834 static void dest_mgr_term_destination(j_compress_ptr cinfo)
836 JpegEncoder *This = encoder_from_compress(cinfo);
837 ULONG byteswritten;
838 HRESULT hr;
840 if (This->dest_mgr.free_in_buffer != sizeof(This->dest_buffer))
842 hr = IStream_Write(This->stream, This->dest_buffer,
843 sizeof(This->dest_buffer) - This->dest_mgr.free_in_buffer, &byteswritten);
845 if (hr != S_OK || byteswritten == 0)
846 ERR("Failed writing data, hr=%x\n", hr);
850 static HRESULT WINAPI JpegEncoder_Frame_QueryInterface(IWICBitmapFrameEncode *iface, REFIID iid,
851 void **ppv)
853 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
854 TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
856 if (!ppv) return E_INVALIDARG;
858 if (IsEqualIID(&IID_IUnknown, iid) ||
859 IsEqualIID(&IID_IWICBitmapFrameEncode, iid))
861 *ppv = &This->IWICBitmapFrameEncode_iface;
863 else
865 *ppv = NULL;
866 return E_NOINTERFACE;
869 IUnknown_AddRef((IUnknown*)*ppv);
870 return S_OK;
873 static ULONG WINAPI JpegEncoder_Frame_AddRef(IWICBitmapFrameEncode *iface)
875 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
876 return IWICBitmapEncoder_AddRef(&This->IWICBitmapEncoder_iface);
879 static ULONG WINAPI JpegEncoder_Frame_Release(IWICBitmapFrameEncode *iface)
881 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
882 return IWICBitmapEncoder_Release(&This->IWICBitmapEncoder_iface);
885 static HRESULT WINAPI JpegEncoder_Frame_Initialize(IWICBitmapFrameEncode *iface,
886 IPropertyBag2 *pIEncoderOptions)
888 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
889 TRACE("(%p,%p)\n", iface, pIEncoderOptions);
891 EnterCriticalSection(&This->lock);
893 if (This->frame_initialized)
895 LeaveCriticalSection(&This->lock);
896 return WINCODEC_ERR_WRONGSTATE;
899 This->frame_initialized = TRUE;
901 LeaveCriticalSection(&This->lock);
903 return S_OK;
906 static HRESULT WINAPI JpegEncoder_Frame_SetSize(IWICBitmapFrameEncode *iface,
907 UINT uiWidth, UINT uiHeight)
909 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
910 TRACE("(%p,%u,%u)\n", iface, uiWidth, uiHeight);
912 EnterCriticalSection(&This->lock);
914 if (!This->frame_initialized || This->started_compress)
916 LeaveCriticalSection(&This->lock);
917 return WINCODEC_ERR_WRONGSTATE;
920 This->width = uiWidth;
921 This->height = uiHeight;
923 LeaveCriticalSection(&This->lock);
925 return S_OK;
928 static HRESULT WINAPI JpegEncoder_Frame_SetResolution(IWICBitmapFrameEncode *iface,
929 double dpiX, double dpiY)
931 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
932 TRACE("(%p,%0.2f,%0.2f)\n", iface, dpiX, dpiY);
934 EnterCriticalSection(&This->lock);
936 if (!This->frame_initialized || This->started_compress)
938 LeaveCriticalSection(&This->lock);
939 return WINCODEC_ERR_WRONGSTATE;
942 This->xres = dpiX;
943 This->yres = dpiY;
945 LeaveCriticalSection(&This->lock);
947 return S_OK;
950 static HRESULT WINAPI JpegEncoder_Frame_SetPixelFormat(IWICBitmapFrameEncode *iface,
951 WICPixelFormatGUID *pPixelFormat)
953 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
954 int i;
955 TRACE("(%p,%s)\n", iface, debugstr_guid(pPixelFormat));
957 EnterCriticalSection(&This->lock);
959 if (!This->frame_initialized || This->started_compress)
961 LeaveCriticalSection(&This->lock);
962 return WINCODEC_ERR_WRONGSTATE;
965 for (i=0; compress_formats[i].guid; i++)
967 if (memcmp(compress_formats[i].guid, pPixelFormat, sizeof(GUID)) == 0)
968 break;
971 if (!compress_formats[i].guid) i = 0;
973 This->format = &compress_formats[i];
974 memcpy(pPixelFormat, This->format->guid, sizeof(GUID));
976 LeaveCriticalSection(&This->lock);
978 return S_OK;
981 static HRESULT WINAPI JpegEncoder_Frame_SetColorContexts(IWICBitmapFrameEncode *iface,
982 UINT cCount, IWICColorContext **ppIColorContext)
984 FIXME("(%p,%u,%p): stub\n", iface, cCount, ppIColorContext);
985 return E_NOTIMPL;
988 static HRESULT WINAPI JpegEncoder_Frame_SetPalette(IWICBitmapFrameEncode *iface,
989 IWICPalette *pIPalette)
991 FIXME("(%p,%p): stub\n", iface, pIPalette);
992 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
995 static HRESULT WINAPI JpegEncoder_Frame_SetThumbnail(IWICBitmapFrameEncode *iface,
996 IWICBitmapSource *pIThumbnail)
998 FIXME("(%p,%p): stub\n", iface, pIThumbnail);
999 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
1002 static HRESULT WINAPI JpegEncoder_Frame_WritePixels(IWICBitmapFrameEncode *iface,
1003 UINT lineCount, UINT cbStride, UINT cbBufferSize, BYTE *pbPixels)
1005 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
1006 jmp_buf jmpbuf;
1007 BYTE *swapped_data = NULL, *current_row;
1008 UINT line;
1009 int row_size;
1010 TRACE("(%p,%u,%u,%u,%p)\n", iface, lineCount, cbStride, cbBufferSize, pbPixels);
1012 EnterCriticalSection(&This->lock);
1014 if (!This->frame_initialized || !This->width || !This->height || !This->format)
1016 LeaveCriticalSection(&This->lock);
1017 return WINCODEC_ERR_WRONGSTATE;
1020 if (lineCount == 0 || lineCount + This->lines_written > This->height)
1022 LeaveCriticalSection(&This->lock);
1023 return E_INVALIDARG;
1026 /* set up setjmp/longjmp error handling */
1027 if (setjmp(jmpbuf))
1029 LeaveCriticalSection(&This->lock);
1030 HeapFree(GetProcessHeap(), 0, swapped_data);
1031 return E_FAIL;
1033 This->cinfo.client_data = jmpbuf;
1035 if (!This->started_compress)
1037 This->cinfo.image_width = This->width;
1038 This->cinfo.image_height = This->height;
1039 This->cinfo.input_components = This->format->num_components;
1040 This->cinfo.in_color_space = This->format->color_space;
1042 pjpeg_set_defaults(&This->cinfo);
1044 if (This->xres != 0.0 && This->yres != 0.0)
1046 This->cinfo.density_unit = 1; /* dots per inch */
1047 This->cinfo.X_density = This->xres;
1048 This->cinfo.Y_density = This->yres;
1051 pjpeg_start_compress(&This->cinfo, TRUE);
1053 This->started_compress = 1;
1056 row_size = This->format->bpp / 8 * This->width;
1058 if (This->format->swap_rgb)
1060 swapped_data = HeapAlloc(GetProcessHeap(), 0, row_size);
1061 if (!swapped_data)
1063 LeaveCriticalSection(&This->lock);
1064 return E_OUTOFMEMORY;
1068 for (line=0; line < lineCount; line++)
1070 if (This->format->swap_rgb)
1072 UINT x;
1074 memcpy(swapped_data, pbPixels + (cbStride * line), row_size);
1076 for (x=0; x < This->width; x++)
1078 BYTE b;
1080 b = swapped_data[x*3];
1081 swapped_data[x*3] = swapped_data[x*3+2];
1082 swapped_data[x*3+2] = b;
1085 current_row = swapped_data;
1087 else
1088 current_row = pbPixels + (cbStride * line);
1090 if (!pjpeg_write_scanlines(&This->cinfo, &current_row, 1))
1092 ERR("failed writing scanlines\n");
1093 LeaveCriticalSection(&This->lock);
1094 HeapFree(GetProcessHeap(), 0, swapped_data);
1095 return E_FAIL;
1098 This->lines_written++;
1101 LeaveCriticalSection(&This->lock);
1102 HeapFree(GetProcessHeap(), 0, swapped_data);
1104 return S_OK;
1107 static HRESULT WINAPI JpegEncoder_Frame_WriteSource(IWICBitmapFrameEncode *iface,
1108 IWICBitmapSource *pIBitmapSource, WICRect *prc)
1110 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
1111 HRESULT hr;
1112 WICRect rc;
1113 WICPixelFormatGUID guid;
1114 UINT stride;
1115 BYTE *pixeldata;
1116 TRACE("(%p,%p,%p)\n", iface, pIBitmapSource, prc);
1118 if (!This->frame_initialized || !This->width || !This->height)
1119 return WINCODEC_ERR_WRONGSTATE;
1121 if (!This->format)
1123 hr = IWICBitmapSource_GetPixelFormat(pIBitmapSource, &guid);
1124 if (FAILED(hr)) return hr;
1125 hr = IWICBitmapFrameEncode_SetPixelFormat(iface, &guid);
1126 if (FAILED(hr)) return hr;
1129 hr = IWICBitmapSource_GetPixelFormat(pIBitmapSource, &guid);
1130 if (FAILED(hr)) return hr;
1131 if (memcmp(&guid, This->format->guid, sizeof(GUID)) != 0)
1133 /* FIXME: should use WICConvertBitmapSource to convert */
1134 ERR("format %s unsupported\n", debugstr_guid(&guid));
1135 return E_FAIL;
1138 if (This->xres == 0.0 || This->yres == 0.0)
1140 double xres, yres;
1141 hr = IWICBitmapSource_GetResolution(pIBitmapSource, &xres, &yres);
1142 if (FAILED(hr)) return hr;
1143 hr = IWICBitmapFrameEncode_SetResolution(iface, xres, yres);
1144 if (FAILED(hr)) return hr;
1147 if (!prc)
1149 UINT width, height;
1150 hr = IWICBitmapSource_GetSize(pIBitmapSource, &width, &height);
1151 if (FAILED(hr)) return hr;
1152 rc.X = 0;
1153 rc.Y = 0;
1154 rc.Width = width;
1155 rc.Height = height;
1156 prc = &rc;
1159 if (prc->Width != This->width) return E_INVALIDARG;
1161 stride = (This->format->bpp * This->width + 7)/8;
1163 pixeldata = HeapAlloc(GetProcessHeap(), 0, stride * prc->Height);
1164 if (!pixeldata) return E_OUTOFMEMORY;
1166 hr = IWICBitmapSource_CopyPixels(pIBitmapSource, prc, stride,
1167 stride*prc->Height, pixeldata);
1169 if (SUCCEEDED(hr))
1171 hr = IWICBitmapFrameEncode_WritePixels(iface, prc->Height, stride,
1172 stride*prc->Height, pixeldata);
1175 HeapFree(GetProcessHeap(), 0, pixeldata);
1177 return hr;
1180 static HRESULT WINAPI JpegEncoder_Frame_Commit(IWICBitmapFrameEncode *iface)
1182 JpegEncoder *This = impl_from_IWICBitmapFrameEncode(iface);
1183 jmp_buf jmpbuf;
1184 TRACE("(%p)\n", iface);
1186 EnterCriticalSection(&This->lock);
1188 if (!This->started_compress || This->lines_written != This->height || This->frame_committed)
1190 LeaveCriticalSection(&This->lock);
1191 return WINCODEC_ERR_WRONGSTATE;
1194 /* set up setjmp/longjmp error handling */
1195 if (setjmp(jmpbuf))
1197 LeaveCriticalSection(&This->lock);
1198 return E_FAIL;
1200 This->cinfo.client_data = jmpbuf;
1202 pjpeg_finish_compress(&This->cinfo);
1204 This->frame_committed = TRUE;
1206 LeaveCriticalSection(&This->lock);
1208 return S_OK;
1211 static HRESULT WINAPI JpegEncoder_Frame_GetMetadataQueryWriter(IWICBitmapFrameEncode *iface,
1212 IWICMetadataQueryWriter **ppIMetadataQueryWriter)
1214 FIXME("(%p, %p): stub\n", iface, ppIMetadataQueryWriter);
1215 return E_NOTIMPL;
1218 static const IWICBitmapFrameEncodeVtbl JpegEncoder_FrameVtbl = {
1219 JpegEncoder_Frame_QueryInterface,
1220 JpegEncoder_Frame_AddRef,
1221 JpegEncoder_Frame_Release,
1222 JpegEncoder_Frame_Initialize,
1223 JpegEncoder_Frame_SetSize,
1224 JpegEncoder_Frame_SetResolution,
1225 JpegEncoder_Frame_SetPixelFormat,
1226 JpegEncoder_Frame_SetColorContexts,
1227 JpegEncoder_Frame_SetPalette,
1228 JpegEncoder_Frame_SetThumbnail,
1229 JpegEncoder_Frame_WritePixels,
1230 JpegEncoder_Frame_WriteSource,
1231 JpegEncoder_Frame_Commit,
1232 JpegEncoder_Frame_GetMetadataQueryWriter
1235 static HRESULT WINAPI JpegEncoder_QueryInterface(IWICBitmapEncoder *iface, REFIID iid,
1236 void **ppv)
1238 JpegEncoder *This = impl_from_IWICBitmapEncoder(iface);
1239 TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
1241 if (!ppv) return E_INVALIDARG;
1243 if (IsEqualIID(&IID_IUnknown, iid) ||
1244 IsEqualIID(&IID_IWICBitmapEncoder, iid))
1246 *ppv = &This->IWICBitmapEncoder_iface;
1248 else
1250 *ppv = NULL;
1251 return E_NOINTERFACE;
1254 IUnknown_AddRef((IUnknown*)*ppv);
1255 return S_OK;
1258 static ULONG WINAPI JpegEncoder_AddRef(IWICBitmapEncoder *iface)
1260 JpegEncoder *This = impl_from_IWICBitmapEncoder(iface);
1261 ULONG ref = InterlockedIncrement(&This->ref);
1263 TRACE("(%p) refcount=%u\n", iface, ref);
1265 return ref;
1268 static ULONG WINAPI JpegEncoder_Release(IWICBitmapEncoder *iface)
1270 JpegEncoder *This = impl_from_IWICBitmapEncoder(iface);
1271 ULONG ref = InterlockedDecrement(&This->ref);
1273 TRACE("(%p) refcount=%u\n", iface, ref);
1275 if (ref == 0)
1277 This->lock.DebugInfo->Spare[0] = 0;
1278 DeleteCriticalSection(&This->lock);
1279 if (This->initialized) pjpeg_destroy_compress(&This->cinfo);
1280 if (This->stream) IStream_Release(This->stream);
1281 HeapFree(GetProcessHeap(), 0, This);
1284 return ref;
1287 static HRESULT WINAPI JpegEncoder_Initialize(IWICBitmapEncoder *iface,
1288 IStream *pIStream, WICBitmapEncoderCacheOption cacheOption)
1290 JpegEncoder *This = impl_from_IWICBitmapEncoder(iface);
1291 jmp_buf jmpbuf;
1293 TRACE("(%p,%p,%u)\n", iface, pIStream, cacheOption);
1295 EnterCriticalSection(&This->lock);
1297 if (This->initialized)
1299 LeaveCriticalSection(&This->lock);
1300 return WINCODEC_ERR_WRONGSTATE;
1303 pjpeg_std_error(&This->jerr);
1305 This->jerr.error_exit = error_exit_fn;
1306 This->jerr.emit_message = emit_message_fn;
1308 This->cinfo.err = &This->jerr;
1310 This->cinfo.client_data = jmpbuf;
1312 if (setjmp(jmpbuf))
1314 LeaveCriticalSection(&This->lock);
1315 return E_FAIL;
1318 pjpeg_CreateCompress(&This->cinfo, JPEG_LIB_VERSION, sizeof(struct jpeg_compress_struct));
1320 This->stream = pIStream;
1321 IStream_AddRef(pIStream);
1323 This->dest_mgr.next_output_byte = This->dest_buffer;
1324 This->dest_mgr.free_in_buffer = sizeof(This->dest_buffer);
1326 This->dest_mgr.init_destination = dest_mgr_init_destination;
1327 This->dest_mgr.empty_output_buffer = dest_mgr_empty_output_buffer;
1328 This->dest_mgr.term_destination = dest_mgr_term_destination;
1330 This->cinfo.dest = &This->dest_mgr;
1332 This->initialized = TRUE;
1334 LeaveCriticalSection(&This->lock);
1336 return S_OK;
1339 static HRESULT WINAPI JpegEncoder_GetContainerFormat(IWICBitmapEncoder *iface,
1340 GUID *pguidContainerFormat)
1342 FIXME("(%p,%s): stub\n", iface, debugstr_guid(pguidContainerFormat));
1343 return E_NOTIMPL;
1346 static HRESULT WINAPI JpegEncoder_GetEncoderInfo(IWICBitmapEncoder *iface,
1347 IWICBitmapEncoderInfo **ppIEncoderInfo)
1349 FIXME("(%p,%p): stub\n", iface, ppIEncoderInfo);
1350 return E_NOTIMPL;
1353 static HRESULT WINAPI JpegEncoder_SetColorContexts(IWICBitmapEncoder *iface,
1354 UINT cCount, IWICColorContext **ppIColorContext)
1356 FIXME("(%p,%u,%p): stub\n", iface, cCount, ppIColorContext);
1357 return E_NOTIMPL;
1360 static HRESULT WINAPI JpegEncoder_SetPalette(IWICBitmapEncoder *iface, IWICPalette *pIPalette)
1362 TRACE("(%p,%p)\n", iface, pIPalette);
1363 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
1366 static HRESULT WINAPI JpegEncoder_SetThumbnail(IWICBitmapEncoder *iface, IWICBitmapSource *pIThumbnail)
1368 TRACE("(%p,%p)\n", iface, pIThumbnail);
1369 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
1372 static HRESULT WINAPI JpegEncoder_SetPreview(IWICBitmapEncoder *iface, IWICBitmapSource *pIPreview)
1374 TRACE("(%p,%p)\n", iface, pIPreview);
1375 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
1378 static HRESULT WINAPI JpegEncoder_CreateNewFrame(IWICBitmapEncoder *iface,
1379 IWICBitmapFrameEncode **ppIFrameEncode, IPropertyBag2 **ppIEncoderOptions)
1381 JpegEncoder *This = impl_from_IWICBitmapEncoder(iface);
1382 HRESULT hr;
1384 TRACE("(%p,%p,%p)\n", iface, ppIFrameEncode, ppIEncoderOptions);
1386 EnterCriticalSection(&This->lock);
1388 if (This->frame_count != 0)
1390 LeaveCriticalSection(&This->lock);
1391 return WINCODEC_ERR_UNSUPPORTEDOPERATION;
1394 if (!This->initialized)
1396 LeaveCriticalSection(&This->lock);
1397 return WINCODEC_ERR_NOTINITIALIZED;
1400 hr = CreatePropertyBag2(NULL, 0, ppIEncoderOptions);
1401 if (FAILED(hr))
1403 LeaveCriticalSection(&This->lock);
1404 return hr;
1407 This->frame_count = 1;
1409 LeaveCriticalSection(&This->lock);
1411 IWICBitmapEncoder_AddRef(iface);
1412 *ppIFrameEncode = &This->IWICBitmapFrameEncode_iface;
1414 return S_OK;
1417 static HRESULT WINAPI JpegEncoder_Commit(IWICBitmapEncoder *iface)
1419 JpegEncoder *This = impl_from_IWICBitmapEncoder(iface);
1420 TRACE("(%p)\n", iface);
1422 EnterCriticalSection(&This->lock);
1424 if (!This->frame_committed || This->committed)
1426 LeaveCriticalSection(&This->lock);
1427 return WINCODEC_ERR_WRONGSTATE;
1430 This->committed = TRUE;
1432 LeaveCriticalSection(&This->lock);
1434 return S_OK;
1437 static HRESULT WINAPI JpegEncoder_GetMetadataQueryWriter(IWICBitmapEncoder *iface,
1438 IWICMetadataQueryWriter **ppIMetadataQueryWriter)
1440 FIXME("(%p,%p): stub\n", iface, ppIMetadataQueryWriter);
1441 return E_NOTIMPL;
1444 static const IWICBitmapEncoderVtbl JpegEncoder_Vtbl = {
1445 JpegEncoder_QueryInterface,
1446 JpegEncoder_AddRef,
1447 JpegEncoder_Release,
1448 JpegEncoder_Initialize,
1449 JpegEncoder_GetContainerFormat,
1450 JpegEncoder_GetEncoderInfo,
1451 JpegEncoder_SetColorContexts,
1452 JpegEncoder_SetPalette,
1453 JpegEncoder_SetThumbnail,
1454 JpegEncoder_SetPreview,
1455 JpegEncoder_CreateNewFrame,
1456 JpegEncoder_Commit,
1457 JpegEncoder_GetMetadataQueryWriter
1460 HRESULT JpegEncoder_CreateInstance(IUnknown *pUnkOuter, REFIID iid, void** ppv)
1462 JpegEncoder *This;
1463 HRESULT ret;
1465 TRACE("(%p,%s,%p)\n", pUnkOuter, debugstr_guid(iid), ppv);
1467 *ppv = NULL;
1469 if (pUnkOuter) return CLASS_E_NOAGGREGATION;
1471 if (!libjpeg_handle && !load_libjpeg())
1473 ERR("Failed writing JPEG because unable to find %s\n",SONAME_LIBJPEG);
1474 return E_FAIL;
1477 This = HeapAlloc(GetProcessHeap(), 0, sizeof(JpegEncoder));
1478 if (!This) return E_OUTOFMEMORY;
1480 This->IWICBitmapEncoder_iface.lpVtbl = &JpegEncoder_Vtbl;
1481 This->IWICBitmapFrameEncode_iface.lpVtbl = &JpegEncoder_FrameVtbl;
1482 This->ref = 1;
1483 This->initialized = 0;
1484 This->frame_count = 0;
1485 This->frame_initialized = 0;
1486 This->started_compress = 0;
1487 This->lines_written = 0;
1488 This->frame_committed = 0;
1489 This->committed = 0;
1490 This->width = This->height = 0;
1491 This->xres = This->yres = 0.0;
1492 This->format = NULL;
1493 This->stream = NULL;
1494 InitializeCriticalSection(&This->lock);
1495 This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": JpegEncoder.lock");
1497 ret = IWICBitmapEncoder_QueryInterface(&This->IWICBitmapEncoder_iface, iid, ppv);
1498 IWICBitmapEncoder_Release(&This->IWICBitmapEncoder_iface);
1500 return ret;
1503 #else /* !defined(SONAME_LIBJPEG) */
1505 HRESULT JpegDecoder_CreateInstance(IUnknown *pUnkOuter, REFIID iid, void** ppv)
1507 ERR("Trying to load JPEG picture, but JPEG support is not compiled in.\n");
1508 return E_FAIL;
1511 HRESULT JpegEncoder_CreateInstance(IUnknown *pUnkOuter, REFIID iid, void** ppv)
1513 ERR("Trying to save JPEG picture, but JPEG support is not compiled in.\n");
1514 return E_FAIL;
1517 #endif