blob: ef903284f48589ee16f747857aac94713e4eb7bb [file] [log] [blame]
Pascal Massiminof61d14a2011-02-18 23:33:46 -08001// Copyright 2011 Google Inc.
2//
3// This code is licensed under the same terms as WebM:
4// Software License Agreement: http://www.webmproject.org/license/software/
5// Additional IP Rights Grant: http://www.webmproject.org/license/additional/
6// -----------------------------------------------------------------------------
7//
8// simple command line calling the WebPEncode function.
9// Encodes a raw .YUV into WebP bitstream
10//
11// Author: Skal (pascal.massimino@gmail.com)
12
13#include <stdio.h>
Pascal Massimino4b0b0d62011-03-26 09:27:45 -070014#include <stdlib.h>
Pascal Massiminof61d14a2011-02-18 23:33:46 -080015#include <string.h>
16
James Zern31f9dc62011-06-06 17:56:50 -070017#ifdef HAVE_CONFIG_H
18#include "config.h"
19#endif
20
Pascal Massiminof61d14a2011-02-18 23:33:46 -080021#ifdef WEBP_HAVE_PNG
22#include <png.h>
23#endif
24
25#ifdef WEBP_HAVE_JPEG
26#include <setjmp.h> // note: this must be included *after* png.h
27#include <jpeglib.h>
28#endif
29
James Zern31f9dc62011-06-06 17:56:50 -070030#ifdef HAVE_WINCODEC_H
31#ifdef __MINGW32__
32#define INITGUID // Without this GUIDs are declared extern and fail to link
33#endif
Pascal Massiminof61d14a2011-02-18 23:33:46 -080034#define CINTERFACE
35#define COBJMACROS
36#define _WIN32_IE 0x500 // Workaround bug in shlwapi.h when compiling C++
37 // code with COBJMACROS.
38#include <shlwapi.h>
39#include <windows.h>
40#include <wincodec.h>
James Zern31f9dc62011-06-06 17:56:50 -070041
42#ifndef GUID_WICPixelFormat24bppRGB
43// From Microsoft SDK 7.0a
44DEFINE_GUID(GUID_WICPixelFormat24bppRGB,
45 0x6fddc324, 0x4e03, 0x4bfe, 0xb1, 0x85, 0x3d, 0x77, 0x76, 0x8d, 0xc9, 0x0d);
Pascal Massiminof61d14a2011-02-18 23:33:46 -080046#endif
James Zern3cfe0882011-06-21 18:29:30 -070047#ifndef GUID_WICPixelFormat32bppRGBA
48DEFINE_GUID(GUID_WICPixelFormat32bppRGBA,
49 0xf5c7ad2d, 0x6a8d, 0x43dd, 0xa7, 0xa8, 0xa2, 0x99, 0x35, 0x26, 0x1a, 0xe9);
50#endif
James Zern31f9dc62011-06-06 17:56:50 -070051#endif /* HAVE_WINCODEC_H */
Pascal Massiminof61d14a2011-02-18 23:33:46 -080052
53
54#include "webp/encode.h"
55#include "stopwatch.h"
James Zernb4d0ef82011-07-15 14:53:03 -070056#ifndef WEBP_DLL
James Zern04e84cf2011-11-04 15:20:08 -070057#if defined(__cplusplus) || defined(c_plusplus)
58extern "C" {
James Zernb4d0ef82011-07-15 14:53:03 -070059#endif
Pascal Massiminocfbf88a2011-04-22 12:14:45 -070060
James Zern04e84cf2011-11-04 15:20:08 -070061extern void* VP8GetCPUInfo; // opaque forward declaration.
62
63#if defined(__cplusplus) || defined(c_plusplus)
64} // extern "C"
65#endif
66#endif // WEBP_DLL
67
James Zernc7e86ab2011-08-25 14:22:32 -070068//------------------------------------------------------------------------------
Pascal Massiminof61d14a2011-02-18 23:33:46 -080069
70static int verbose = 0;
71
72static int ReadYUV(FILE* in_file, WebPPicture* const pic) {
73 const int uv_width = (pic->width + 1) / 2;
74 const int uv_height = (pic->height + 1) / 2;
75 int y;
76 int ok = 0;
77
78 if (!WebPPictureAlloc(pic)) return ok;
79
80 for (y = 0; y < pic->height; ++y) {
81 if (fread(pic->y + y * pic->y_stride, pic->width, 1, in_file) != 1) {
82 goto End;
83 }
84 }
85 for (y = 0; y < uv_height; ++y) {
86 if (fread(pic->u + y * pic->uv_stride, uv_width, 1, in_file) != 1)
87 goto End;
88 }
89 for (y = 0; y < uv_height; ++y) {
90 if (fread(pic->v + y * pic->uv_stride, uv_width, 1, in_file) != 1)
91 goto End;
92 }
93 ok = 1;
94
95 End:
96 return ok;
97}
98
James Zern31f9dc62011-06-06 17:56:50 -070099#ifdef HAVE_WINCODEC_H
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800100
101#define IFS(fn) \
102 do { \
103 if (SUCCEEDED(hr)) \
104 { \
105 hr = (fn); \
106 if (FAILED(hr) && verbose) \
107 printf(#fn " failed %08x\n", hr); \
108 } \
109 } while (0)
110
111#ifdef __cplusplus
112#define MAKE_REFGUID(x) (x)
113#else
114#define MAKE_REFGUID(x) &(x)
115#endif
116
117static HRESULT OpenInputStream(const char* filename, IStream** ppStream) {
118 HRESULT hr = S_OK;
119 IFS(SHCreateStreamOnFileA(filename, STGM_READ, ppStream));
120 if (FAILED(hr))
121 printf("Error opening input file %s (%08x)\n", filename, hr);
122 return hr;
123}
124
125static HRESULT ReadPictureWithWIC(const char* filename,
James Zern3cfe0882011-06-21 18:29:30 -0700126 WebPPicture* const pic, int keep_alpha) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800127 HRESULT hr = S_OK;
128 IWICBitmapFrameDecode* pFrame = NULL;
129 IWICFormatConverter* pConverter = NULL;
130 IWICImagingFactory* pFactory = NULL;
131 IWICBitmapDecoder* pDecoder = NULL;
132 IStream* pStream = NULL;
133 UINT frameCount = 0;
134 UINT width, height = 0;
135 BYTE* rgb = NULL;
James Zern3cfe0882011-06-21 18:29:30 -0700136 WICPixelFormatGUID srcPixelFormat = { 0 };
137 GUID srcContainerFormat = { 0 };
138 const GUID* alphaContainers[] = {
139 &GUID_ContainerFormatBmp,
140 &GUID_ContainerFormatPng,
141 &GUID_ContainerFormatTiff
142 };
143 int has_alpha = 0;
144 int i, stride;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800145
146 IFS(CoInitialize(NULL));
147 IFS(CoCreateInstance(MAKE_REFGUID(CLSID_WICImagingFactory), NULL,
148 CLSCTX_INPROC_SERVER, MAKE_REFGUID(IID_IWICImagingFactory),
149 (LPVOID*)&pFactory));
150 if (hr == REGDB_E_CLASSNOTREG) {
151 printf("Couldn't access Windows Imaging Component (are you running \n");
152 printf("Windows XP SP3 or newer?). Most formats not available.\n");
153 printf("Use -s for the available YUV input.\n");
154 }
155 // Prepare for image decoding.
156 IFS(OpenInputStream(filename, &pStream));
157 IFS(IWICImagingFactory_CreateDecoderFromStream(pFactory, pStream, NULL,
158 WICDecodeMetadataCacheOnDemand, &pDecoder));
159 IFS(IWICBitmapDecoder_GetFrameCount(pDecoder, &frameCount));
160 if (SUCCEEDED(hr) && frameCount == 0) {
161 printf("No frame found in input file.\n");
162 hr = E_FAIL;
163 }
164 IFS(IWICBitmapDecoder_GetFrame(pDecoder, 0, &pFrame));
James Zern3cfe0882011-06-21 18:29:30 -0700165 IFS(IWICBitmapFrameDecode_GetPixelFormat(pFrame, &srcPixelFormat));
166 IFS(IWICBitmapDecoder_GetContainerFormat(pDecoder, &srcContainerFormat));
167
168 has_alpha = keep_alpha;
169 for (i = 0;
170 has_alpha && i < sizeof(alphaContainers)/sizeof(alphaContainers[0]);
171 ++i) {
172 if (IsEqualGUID(&srcContainerFormat, alphaContainers[i])) {
173 has_alpha =
174 IsEqualGUID(&srcPixelFormat, &GUID_WICPixelFormat32bppRGBA) ||
175 IsEqualGUID(&srcPixelFormat, &GUID_WICPixelFormat32bppBGRA);
176 break;
177 }
178 }
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800179
180 // Prepare for pixel format conversion (if necessary).
181 IFS(IWICImagingFactory_CreateFormatConverter(pFactory, &pConverter));
182 IFS(IWICFormatConverter_Initialize(pConverter, (IWICBitmapSource*)pFrame,
James Zern3cfe0882011-06-21 18:29:30 -0700183 has_alpha ? MAKE_REFGUID(GUID_WICPixelFormat32bppRGBA)
184 : MAKE_REFGUID(GUID_WICPixelFormat24bppRGB),
185 WICBitmapDitherTypeNone,
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800186 NULL, 0.0, WICBitmapPaletteTypeCustom));
187
188 // Decode.
189 IFS(IWICFormatConverter_GetSize(pConverter, &width, &height));
James Zern3cfe0882011-06-21 18:29:30 -0700190 stride = (has_alpha ? 4 : 3) * width * sizeof(*rgb);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800191 if (SUCCEEDED(hr)) {
James Zern3cfe0882011-06-21 18:29:30 -0700192 rgb = (BYTE*)malloc(stride * height);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800193 if (rgb == NULL)
194 hr = E_OUTOFMEMORY;
195 }
James Zern3cfe0882011-06-21 18:29:30 -0700196 IFS(IWICFormatConverter_CopyPixels(pConverter, NULL, stride,
197 stride * height, rgb));
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800198
199 // WebP conversion.
200 if (SUCCEEDED(hr)) {
James Zern3cfe0882011-06-21 18:29:30 -0700201 int ok;
202#ifdef WEBP_EXPERIMENTAL_FEATURES
203 if (has_alpha) {
204 pic->colorspace |= WEBP_CSP_ALPHA_BIT;
205 }
206#endif
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800207 pic->width = width;
208 pic->height = height;
James Zern3cfe0882011-06-21 18:29:30 -0700209 ok = has_alpha ? WebPPictureImportRGBA(pic, rgb, stride)
210 : WebPPictureImportRGB(pic, rgb, stride);
211 if (!ok)
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800212 hr = E_FAIL;
213 }
214
215 // Cleanup.
216 if (pConverter != NULL) IUnknown_Release(pConverter);
217 if (pFrame != NULL) IUnknown_Release(pFrame);
218 if (pDecoder != NULL) IUnknown_Release(pDecoder);
219 if (pFactory != NULL) IUnknown_Release(pFactory);
220 if (pStream != NULL) IUnknown_Release(pStream);
221 free(rgb);
222 return hr;
223}
224
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700225static int ReadPicture(const char* const filename, WebPPicture* const pic,
226 int keep_alpha) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800227 int ok;
228 if (pic->width != 0 && pic->height != 0) {
229 // If image size is specified, infer it as YUV format.
230 FILE* in_file = fopen(filename, "rb");
231 if (in_file == NULL) {
232 fprintf(stderr, "Error! Cannot open input file '%s'\n", filename);
233 return 0;
234 }
235 ok = ReadYUV(in_file, pic);
236 fclose(in_file);
237 } else {
238 // If no size specified, try to decode it using WIC.
James Zern3cfe0882011-06-21 18:29:30 -0700239 ok = SUCCEEDED(ReadPictureWithWIC(filename, pic, keep_alpha));
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800240 }
241 if (!ok) {
242 fprintf(stderr, "Error! Could not process file %s\n", filename);
243 }
244 return ok;
245}
246
James Zern31f9dc62011-06-06 17:56:50 -0700247#else // !HAVE_WINCODEC_H
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800248
249#ifdef WEBP_HAVE_JPEG
250struct my_error_mgr {
251 struct jpeg_error_mgr pub;
252 jmp_buf setjmp_buffer;
253};
254
255static void my_error_exit(j_common_ptr dinfo) {
256 struct my_error_mgr* myerr = (struct my_error_mgr*) dinfo->err;
257 (*dinfo->err->output_message) (dinfo);
258 longjmp(myerr->setjmp_buffer, 1);
259}
260
261static int ReadJPEG(FILE* in_file, WebPPicture* const pic) {
262 int ok = 0;
263 int stride, width, height;
264 uint8_t* rgb = NULL;
265 uint8_t* row_ptr = NULL;
266 struct jpeg_decompress_struct dinfo;
267 struct my_error_mgr jerr;
268 JSAMPARRAY buffer;
269
270 dinfo.err = jpeg_std_error(&jerr.pub);
271 jerr.pub.error_exit = my_error_exit;
272
James Zern04e84cf2011-11-04 15:20:08 -0700273 if (setjmp(jerr.setjmp_buffer)) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800274 Error:
275 jpeg_destroy_decompress(&dinfo);
276 goto End;
277 }
278
279 jpeg_create_decompress(&dinfo);
280 jpeg_stdio_src(&dinfo, in_file);
281 jpeg_read_header(&dinfo, TRUE);
282
283 dinfo.out_color_space = JCS_RGB;
284 dinfo.dct_method = JDCT_IFAST;
285 dinfo.do_fancy_upsampling = TRUE;
286
287 jpeg_start_decompress(&dinfo);
288
289 if (dinfo.output_components != 3) {
290 goto Error;
291 }
292
293 width = dinfo.output_width;
294 height = dinfo.output_height;
295 stride = dinfo.output_width * dinfo.output_components * sizeof(*rgb);
296
297 rgb = (uint8_t*)malloc(stride * height);
298 if (rgb == NULL) {
299 goto End;
300 }
301 row_ptr = rgb;
302
303 buffer = (*dinfo.mem->alloc_sarray) ((j_common_ptr) &dinfo,
304 JPOOL_IMAGE, stride, 1);
305 if (buffer == NULL) {
306 goto End;
307 }
308
309 while (dinfo.output_scanline < dinfo.output_height) {
310 if (jpeg_read_scanlines(&dinfo, buffer, 1) != 1) {
311 goto End;
312 }
313 memcpy(row_ptr, buffer[0], stride);
314 row_ptr += stride;
315 }
316
James Zern04e84cf2011-11-04 15:20:08 -0700317 jpeg_finish_decompress(&dinfo);
318 jpeg_destroy_decompress(&dinfo);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800319
320 // WebP conversion.
321 pic->width = width;
322 pic->height = height;
323 ok = WebPPictureImportRGB(pic, rgb, stride);
324
325 End:
326 if (rgb) {
327 free(rgb);
328 }
329 return ok;
330}
331
332#else
333static int ReadJPEG(FILE* in_file, WebPPicture* const pic) {
James Zern08057062011-06-24 16:50:02 -0400334 (void)in_file;
335 (void)pic;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800336 printf("JPEG support not compiled. Please install the libjpeg development "
337 "package before building.\n");
338 return 0;
339}
340#endif
341
342#ifdef WEBP_HAVE_PNG
343static void PNGAPI error_function(png_structp png, png_const_charp dummy) {
Pascal Massiminof8db5d52011-03-25 15:04:11 -0700344 (void)dummy; // remove variable-unused warning
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800345 longjmp(png_jmpbuf(png), 1);
346}
347
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700348static int ReadPNG(FILE* in_file, WebPPicture* const pic, int keep_alpha) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800349 png_structp png;
350 png_infop info;
351 int color_type, bit_depth, interlaced;
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700352 int has_alpha;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800353 int num_passes;
Pascal Massiminof8db5d52011-03-25 15:04:11 -0700354 int p;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800355 int ok = 0;
Pascal Massiminof8db5d52011-03-25 15:04:11 -0700356 png_uint_32 width, height, y;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800357 int stride;
358 uint8_t* rgb = NULL;
359
360 png = png_create_read_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0);
361 if (png == NULL) {
362 goto End;
363 }
364
365 png_set_error_fn(png, 0, error_function, NULL);
366 if (setjmp(png_jmpbuf(png))) {
367 Error:
368 png_destroy_read_struct(&png, NULL, NULL);
Pascal Massiminof6a7d752011-11-01 05:18:46 -0700369 free(rgb);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800370 goto End;
371 }
372
373 info = png_create_info_struct(png);
374 if (info == NULL) goto Error;
375
376 png_init_io(png, in_file);
377 png_read_info(png, info);
378 if (!png_get_IHDR(png, info,
379 &width, &height, &bit_depth, &color_type, &interlaced,
380 NULL, NULL)) goto Error;
381
382 png_set_strip_16(png);
383 png_set_packing(png);
384 if (color_type == PNG_COLOR_TYPE_PALETTE) png_set_palette_to_rgb(png);
385 if (color_type == PNG_COLOR_TYPE_GRAY) {
386 if (bit_depth < 8) {
387 png_set_expand_gray_1_2_4_to_8(png);
388 }
389 png_set_gray_to_rgb(png);
390 }
391 if (png_get_valid(png, info, PNG_INFO_tRNS)) {
392 png_set_tRNS_to_alpha(png);
Pascal Massimino28ad70c2011-09-20 07:51:02 -0700393 has_alpha = 1;
394 } else {
395 has_alpha = !!(color_type & PNG_COLOR_MASK_ALPHA);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800396 }
397
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700398 if (!keep_alpha) {
399 png_set_strip_alpha(png);
400 has_alpha = 0;
401 }
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700402#ifdef WEBP_EXPERIMENTAL_FEATURES
403 if (has_alpha) {
404 pic->colorspace |= WEBP_CSP_ALPHA_BIT;
405 }
406#endif
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700407
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800408 num_passes = png_set_interlace_handling(png);
409 png_read_update_info(png, info);
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700410 stride = (has_alpha ? 4 : 3) * width * sizeof(*rgb);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800411 rgb = (uint8_t*)malloc(stride * height);
412 if (rgb == NULL) goto Error;
413 for (p = 0; p < num_passes; ++p) {
414 for (y = 0; y < height; ++y) {
415 png_bytep row = rgb + y * stride;
416 png_read_rows(png, &row, NULL, 1);
417 }
418 }
419 png_read_end(png, info);
420 png_destroy_read_struct(&png, &info, NULL);
421
422 pic->width = width;
423 pic->height = height;
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700424 ok = has_alpha ? WebPPictureImportRGBA(pic, rgb, stride)
425 : WebPPictureImportRGB(pic, rgb, stride);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800426 free(rgb);
427
428 End:
429 return ok;
430}
431#else
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700432static int ReadPNG(FILE* in_file, WebPPicture* const pic, int keep_alpha) {
James Zern08057062011-06-24 16:50:02 -0400433 (void)in_file;
434 (void)pic;
435 (void)keep_alpha;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800436 printf("PNG support not compiled. Please install the libpng development "
437 "package before building.\n");
438 return 0;
439}
440#endif
441
442typedef enum {
443 PNG = 0,
444 JPEG,
445 UNSUPPORTED,
446} InputFileFormat;
447
448static InputFileFormat GetImageType(FILE* in_file) {
449 InputFileFormat format = UNSUPPORTED;
450 unsigned int magic;
451 unsigned char buf[4];
452
453 if ((fread(&buf[0], 4, 1, in_file) != 1) ||
454 (fseek(in_file, 0, SEEK_SET) != 0)) {
455 return format;
456 }
457
458 magic = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
459 if (magic == 0x89504E47U) {
460 format = PNG;
461 } else if (magic >= 0xFFD8FF00U && magic <= 0xFFD8FFFFU) {
462 format = JPEG;
463 }
464 return format;
465}
466
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700467static int ReadPicture(const char* const filename, WebPPicture* const pic,
468 int keep_alpha) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800469 int ok = 0;
470 FILE* in_file = fopen(filename, "rb");
471 if (in_file == NULL) {
472 fprintf(stderr, "Error! Cannot open input file '%s'\n", filename);
473 return ok;
474 }
475
476 if (pic->width == 0 || pic->height == 0) {
477 // If no size specified, try to decode it as PNG/JPEG (as appropriate).
478 const InputFileFormat format = GetImageType(in_file);
479 if (format == PNG) {
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700480 ok = ReadPNG(in_file, pic, keep_alpha);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800481 } else if (format == JPEG) {
482 ok = ReadJPEG(in_file, pic);
483 }
484 } else {
485 // If image size is specified, infer it as YUV format.
486 ok = ReadYUV(in_file, pic);
487 }
488 if (!ok) {
489 fprintf(stderr, "Error! Could not process file %s\n", filename);
490 }
491
492 fclose(in_file);
493 return ok;
494}
495
James Zern31f9dc62011-06-06 17:56:50 -0700496#endif // !HAVE_WINCODEC_H
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800497
498static void AllocExtraInfo(WebPPicture* const pic) {
499 const int mb_w = (pic->width + 15) / 16;
500 const int mb_h = (pic->height + 15) / 16;
501 pic->extra_info = (uint8_t*)malloc(mb_w * mb_h * sizeof(*pic->extra_info));
502}
503
504static void PrintByteCount(const int bytes[4], int total_size,
505 int* const totals) {
506 int s;
507 int total = 0;
508 for (s = 0; s < 4; ++s) {
509 fprintf(stderr, "| %7d ", bytes[s]);
510 total += bytes[s];
511 if (totals) totals[s] += bytes[s];
512 }
James Zern04e84cf2011-11-04 15:20:08 -0700513 fprintf(stderr, "| %7d (%.1f%%)\n", total, 100.f * total / total_size);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800514}
515
516static void PrintPercents(const int counts[4], int total) {
517 int s;
518 for (s = 0; s < 4; ++s) {
519 fprintf(stderr, "| %2d%%", 100 * counts[s] / total);
520 }
James Zern04e84cf2011-11-04 15:20:08 -0700521 fprintf(stderr, "| %7d\n", total);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800522}
523
524static void PrintValues(const int values[4]) {
525 int s;
526 for (s = 0; s < 4; ++s) {
527 fprintf(stderr, "| %7d ", values[s]);
528 }
James Zern04e84cf2011-11-04 15:20:08 -0700529 fprintf(stderr, "|\n");
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800530}
531
Pascal Massiminof8db5d52011-03-25 15:04:11 -0700532static void PrintExtraInfo(const WebPPicture* const pic, int short_output) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800533 const WebPAuxStats* const stats = pic->stats;
534 if (short_output) {
535 fprintf(stderr, "%7d %2.2f\n", stats->coded_size, stats->PSNR[3]);
James Zern04e84cf2011-11-04 15:20:08 -0700536 } else {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800537 const int num_i4 = stats->block_count[0];
538 const int num_i16 = stats->block_count[1];
539 const int num_skip = stats->block_count[2];
540 const int total = num_i4 + num_i16;
541 fprintf(stderr,
542 "%7d bytes Y-U-V-All-PSNR %2.2f %2.2f %2.2f %2.2f dB\n",
543 stats->coded_size,
544 stats->PSNR[0], stats->PSNR[1], stats->PSNR[2], stats->PSNR[3]);
545 if (total > 0) {
546 int totals[4] = { 0, 0, 0, 0 };
547 fprintf(stderr, "block count: intra4: %d\n"
548 " intra16: %d (-> %.2f%%)\n",
549 num_i4, num_i16, 100.f * num_i16 / total);
550 fprintf(stderr, " skipped block: %d (%.2f%%)\n",
551 num_skip, 100.f * num_skip / total);
552 fprintf(stderr, "bytes used: header: %6d (%.1f%%)\n"
553 " mode-partition: %6d (%.1f%%)\n",
554 stats->header_bytes[0],
555 100.f * stats->header_bytes[0] / stats->coded_size,
556 stats->header_bytes[1],
557 100.f * stats->header_bytes[1] / stats->coded_size);
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700558 if (stats->alpha_data_size) {
559 fprintf(stderr, " transparency: %6d\n",
560 stats->alpha_data_size);
561 }
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700562 if (stats->layer_data_size) {
563 fprintf(stderr, " enhancement: %6d\n",
564 stats->layer_data_size);
565 }
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800566 fprintf(stderr, " Residuals bytes "
567 "|segment 1|segment 2|segment 3"
568 "|segment 4| total\n");
569 fprintf(stderr, " intra4-coeffs: ");
570 PrintByteCount(stats->residual_bytes[0], stats->coded_size, totals);
571 fprintf(stderr, " intra16-coeffs: ");
572 PrintByteCount(stats->residual_bytes[1], stats->coded_size, totals);
573 fprintf(stderr, " chroma coeffs: ");
574 PrintByteCount(stats->residual_bytes[2], stats->coded_size, totals);
575 fprintf(stderr, " macroblocks: ");
576 PrintPercents(stats->segment_size, total);
577 fprintf(stderr, " quantizer: ");
578 PrintValues(stats->segment_quant);
579 fprintf(stderr, " filter level: ");
580 PrintValues(stats->segment_level);
581 fprintf(stderr, "------------------+---------");
582 fprintf(stderr, "+---------+---------+---------+-----------------\n");
583 fprintf(stderr, " segments total: ");
584 PrintByteCount(totals, stats->coded_size, NULL);
585 }
586 }
587 if (pic->extra_info) {
588 const int mb_w = (pic->width + 15) / 16;
589 const int mb_h = (pic->height + 15) / 16;
590 const int type = pic->extra_info_type;
591 int x, y;
592 for (y = 0; y < mb_h; ++y) {
593 for (x = 0; x < mb_w; ++x) {
594 const int c = pic->extra_info[x + y * mb_w];
595 if (type == 1) { // intra4/intra16
596 printf("%c", "+."[c]);
597 } else if (type == 2) { // segments
598 printf("%c", ".-*X"[c]);
599 } else if (type == 3) { // quantizers
600 printf("%.2d ", c);
601 } else if (type == 6 || type == 7) {
602 printf("%3d ", c);
603 } else {
604 printf("0x%.2x ", c);
605 }
606 }
607 printf("\n");
608 }
609 }
610}
611
James Zernc7e86ab2011-08-25 14:22:32 -0700612//------------------------------------------------------------------------------
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800613
614static int MyWriter(const uint8_t* data, size_t data_size,
615 const WebPPicture* const pic) {
616 FILE* const out = (FILE*)pic->custom_ptr;
617 return data_size ? (fwrite(data, data_size, 1, out) == 1) : 1;
618}
619
620// Dumps a picture as a PGM file using the IMC4 layout.
621static int DumpPicture(const WebPPicture* const picture, const char* PGM_name) {
622 int y;
623 const int uv_width = (picture->width + 1) / 2;
624 const int uv_height = (picture->height + 1) / 2;
625 const int stride = (picture->width + 1) & ~1;
Pascal Massimino5142a0b2011-07-07 16:08:15 -0700626 const int alpha_height = picture->a ? picture->height : 0;
627 const int height = picture->height + uv_height + alpha_height;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800628 FILE* const f = fopen(PGM_name, "wb");
629 if (!f) return 0;
630 fprintf(f, "P5\n%d %d\n255\n", stride, height);
631 for (y = 0; y < picture->height; ++y) {
632 if (fwrite(picture->y + y * picture->y_stride, picture->width, 1, f) != 1)
633 return 0;
634 if (picture->width & 1) fputc(0, f); // pad
635 }
636 for (y = 0; y < uv_height; ++y) {
637 if (fwrite(picture->u + y * picture->uv_stride, uv_width, 1, f) != 1)
638 return 0;
639 if (fwrite(picture->v + y * picture->uv_stride, uv_width, 1, f) != 1)
640 return 0;
641 }
Pascal Massimino5142a0b2011-07-07 16:08:15 -0700642 for (y = 0; y < alpha_height; ++y) {
643 if (fwrite(picture->a + y * picture->a_stride, picture->width, 1, f) != 1)
644 return 0;
645 if (picture->width & 1) fputc(0, f); // pad
646 }
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800647 fclose(f);
648 return 1;
649}
650
James Zernc7e86ab2011-08-25 14:22:32 -0700651//------------------------------------------------------------------------------
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800652
Pascal Massiminof8db5d52011-03-25 15:04:11 -0700653static void HelpShort(void) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800654 printf("Usage:\n\n");
655 printf(" cwebp [options] -q quality input.png -o output.webp\n\n");
656 printf("where quality is between 0 (poor) to 100 (very good).\n");
657 printf("Typical value is around 80.\n\n");
658 printf("Try -longhelp for an exhaustive list of advanced options.\n");
659}
660
Pascal Massiminof8db5d52011-03-25 15:04:11 -0700661static void HelpLong(void) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800662 printf("Usage:\n");
663 printf(" cwebp [-preset <...>] [options] in_file [-o out_file]\n\n");
664 printf("If input size (-s) for an image is not specified, "
665 "it is assumed to be a PNG or JPEG file.\n");
James Zern31f9dc62011-06-06 17:56:50 -0700666#ifdef HAVE_WINCODEC_H
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800667 printf("Windows builds can take as input any of the files handled by WIC\n");
668#endif
669 printf("options:\n");
670 printf(" -h / -help ............ short help\n");
671 printf(" -H / -longhelp ........ long help\n");
672 printf(" -q <float> ............. quality factor (0:small..100:big)\n");
673 printf(" -preset <string> ....... Preset setting, one of:\n");
674 printf(" default, photo, picture,\n");
675 printf(" drawing, icon, text\n");
676 printf(" -preset must come first, as it overwrites other parameters.");
677 printf("\n");
678 printf(" -m <int> ............... compression method (0=fast, 6=slowest)\n");
679 printf(" -segments <int> ........ number of segments to use (1..4)\n");
Pascal Massimino38369c02011-05-04 22:59:51 -0700680 printf(" -size <int> ............ Target size (in bytes)\n");
681 printf(" -psnr <float> .......... Target PSNR (in dB. typically: 42)\n");
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800682 printf("\n");
683 printf(" -s <int> <int> ......... Input size (width x height) for YUV\n");
684 printf(" -sns <int> ............. Spatial Noise Shaping (0:off, 100:max)\n");
685 printf(" -f <int> ............... filter strength (0=off..100)\n");
686 printf(" -sharpness <int> ....... "
687 "filter sharpness (0:most .. 7:least sharp)\n");
688 printf(" -strong ................ use strong filter instead of simple.\n");
Pascal Massimino900286e2011-08-23 15:58:22 -0700689 printf(" -partition_limit <int> . limit quality to fit the 512k limit on\n");
690 printf(" "
691 "the first partition (0=no degradation ... 100=full)\n");
Pascal Massiminoa5b78c82011-08-29 21:49:18 -0700692#ifdef WEBP_EXPERIMENTAL_FEATURES
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700693 printf(" -alpha_comp <int> ...... set the transparency-compression\n");
694 printf(" -noalpha ............... discard any transparency information.\n");
Pascal Massiminoa5b78c82011-08-29 21:49:18 -0700695#endif
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800696 printf(" -pass <int> ............ analysis pass number (1..10)\n");
697 printf(" -crop <x> <y> <w> <h> .. crop picture with the given rectangle\n");
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700698 printf(" -resize <w> <h> ........ resize picture (after any cropping)\n");
699#ifdef WEBP_EXPERIMENTAL_FEATURES
700 printf(" -444 / -422 / -gray ..... Change colorspace\n");
701#endif
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800702 printf(" -map <int> ............. print map of extra info.\n");
703 printf(" -d <file.pgm> .......... dump the compressed output (PGM file).\n");
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700704
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800705 printf("\n");
706 printf(" -short ................. condense printed message\n");
707 printf(" -quiet ................. don't print anything.\n");
Pascal Massimino650ffa32011-03-24 16:17:10 -0700708 printf(" -version ............... print version number and exit.\n");
James Zernb4d0ef82011-07-15 14:53:03 -0700709#ifndef WEBP_DLL
Pascal Massiminocfbf88a2011-04-22 12:14:45 -0700710 printf(" -noasm ................. disable all assembly optimizations.\n");
James Zernb4d0ef82011-07-15 14:53:03 -0700711#endif
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800712 printf(" -v ..................... verbose, e.g. print encoding/decoding "
713 "times\n");
714 printf("\n");
715 printf("Experimental Options:\n");
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800716 printf(" -af .................... auto-adjust filter strength.\n");
717 printf(" -pre <int> ............. pre-processing filter\n");
718 printf("\n");
719}
720
James Zernc7e86ab2011-08-25 14:22:32 -0700721//------------------------------------------------------------------------------
Pascal Massiminoca7a2fd2011-06-02 06:55:03 -0700722// Error messages
723
724static const char* const kErrorMessages[] = {
725 "OK",
726 "OUT_OF_MEMORY: Out of memory allocating objects",
727 "BITSTREAM_OUT_OF_MEMORY: Out of memory re-allocating byte buffer",
728 "NULL_PARAMETER: NULL parameter passed to function",
729 "INVALID_CONFIGURATION: configuration is invalid",
Pascal Massimino900286e2011-08-23 15:58:22 -0700730 "BAD_DIMENSION: Bad picture dimension. Maximum width and height "
731 "allowed is 16383 pixels.",
732 "PARTITION0_OVERFLOW: Partition #0 is too big to fit 512k.\n"
733 "To reduce the size of this partition, try using less segments "
734 "with the -segments option, and eventually reduce the number of "
735 "header bits using -partition_limit. More details are available "
736 "in the manual (`man cwebp`)",
737 "PARTITION_OVERFLOW: Partition is too big to fit 16M",
738 "BAD_WRITE: Picture writer returned an I/O error"
Mikolaj Zalewski2aa6b802011-06-28 16:02:56 +0200739 "FILE_TOO_BIG: File would be too big to fit in 4G"
Pascal Massiminoca7a2fd2011-06-02 06:55:03 -0700740};
741
James Zernc7e86ab2011-08-25 14:22:32 -0700742//------------------------------------------------------------------------------
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800743
744int main(int argc, const char *argv[]) {
745 const char *in_file = NULL, *out_file = NULL, *dump_file = NULL;
746 FILE *out = NULL;
747 int c;
748 int short_output = 0;
749 int quiet = 0;
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700750 int keep_alpha = 0;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800751 int crop = 0, crop_x = 0, crop_y = 0, crop_w = 0, crop_h = 0;
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700752 int resize_w = 0, resize_h = 0;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800753 WebPPicture picture;
754 WebPConfig config;
755 WebPAuxStats stats;
756 Stopwatch stop_watch;
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700757
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700758#ifdef WEBP_EXPERIMENTAL_FEATURES
759 keep_alpha = 1;
760#endif
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700761
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800762 if (!WebPPictureInit(&picture) || !WebPConfigInit(&config)) {
763 fprintf(stderr, "Error! Version mismatch!\n");
764 goto Error;
765 }
766
767 if (argc == 1) {
768 HelpShort();
769 return 0;
770 }
771
772 for (c = 1; c < argc; ++c) {
773 if (!strcmp(argv[c], "-h") || !strcmp(argv[c], "-help")) {
774 HelpShort();
775 return 0;
776 } else if (!strcmp(argv[c], "-H") || !strcmp(argv[c], "-longhelp")) {
777 HelpLong();
778 return 0;
779 } else if (!strcmp(argv[c], "-o") && c < argc - 1) {
780 out_file = argv[++c];
781 } else if (!strcmp(argv[c], "-d") && c < argc - 1) {
782 dump_file = argv[++c];
783 config.show_compressed = 1;
784 } else if (!strcmp(argv[c], "-short")) {
785 short_output++;
786 } else if (!strcmp(argv[c], "-s") && c < argc - 2) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700787 picture.width = strtol(argv[++c], NULL, 0);
788 picture.height = strtol(argv[++c], NULL, 0);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800789 } else if (!strcmp(argv[c], "-m") && c < argc - 1) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700790 config.method = strtol(argv[++c], NULL, 0);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800791 } else if (!strcmp(argv[c], "-q") && c < argc - 1) {
Mikolaj Zalewski2aa6b802011-06-28 16:02:56 +0200792 config.quality = (float)strtod(argv[++c], NULL);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800793 } else if (!strcmp(argv[c], "-size") && c < argc - 1) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700794 config.target_size = strtol(argv[++c], NULL, 0);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800795 } else if (!strcmp(argv[c], "-psnr") && c < argc - 1) {
Mikolaj Zalewski2aa6b802011-06-28 16:02:56 +0200796 config.target_PSNR = (float)strtod(argv[++c], NULL);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800797 } else if (!strcmp(argv[c], "-sns") && c < argc - 1) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700798 config.sns_strength = strtol(argv[++c], NULL, 0);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800799 } else if (!strcmp(argv[c], "-f") && c < argc - 1) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700800 config.filter_strength = strtol(argv[++c], NULL, 0);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800801 } else if (!strcmp(argv[c], "-af")) {
802 config.autofilter = 1;
Pascal Massimino842c0092011-05-05 18:10:08 -0700803 } else if (!strcmp(argv[c], "-strong")) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800804 config.filter_type = 1;
805 } else if (!strcmp(argv[c], "-sharpness") && c < argc - 1) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700806 config.filter_sharpness = strtol(argv[++c], NULL, 0);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800807 } else if (!strcmp(argv[c], "-pass") && c < argc - 1) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700808 config.pass = strtol(argv[++c], NULL, 0);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800809 } else if (!strcmp(argv[c], "-pre") && c < argc - 1) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700810 config.preprocessing = strtol(argv[++c], NULL, 0);
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800811 } else if (!strcmp(argv[c], "-segments") && c < argc - 1) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700812 config.segments = strtol(argv[++c], NULL, 0);
Pascal Massimino900286e2011-08-23 15:58:22 -0700813 } else if (!strcmp(argv[c], "-partition_limit") && c < argc - 1) {
814 config.partition_limit = strtol(argv[++c], NULL, 0);
Pascal Massiminoa5b78c82011-08-29 21:49:18 -0700815#ifdef WEBP_EXPERIMENTAL_FEATURES
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700816 } else if (!strcmp(argv[c], "-alpha_comp") && c < argc - 1) {
817 config.alpha_compression = strtol(argv[++c], NULL, 0);
818 } else if (!strcmp(argv[c], "-noalpha")) {
819 keep_alpha = 0;
Pascal Massiminoa5b78c82011-08-29 21:49:18 -0700820#endif
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800821 } else if (!strcmp(argv[c], "-map") && c < argc - 1) {
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700822 picture.extra_info_type = strtol(argv[++c], NULL, 0);
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700823#ifdef WEBP_EXPERIMENTAL_FEATURES
824 } else if (!strcmp(argv[c], "-444")) {
825 picture.colorspace = WEBP_YUV444;
826 } else if (!strcmp(argv[c], "-422")) {
827 picture.colorspace = WEBP_YUV422;
828 } else if (!strcmp(argv[c], "-gray")) {
829 picture.colorspace = WEBP_YUV400;
830#endif
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800831 } else if (!strcmp(argv[c], "-crop") && c < argc - 4) {
832 crop = 1;
Pascal Massimino4b0b0d62011-03-26 09:27:45 -0700833 crop_x = strtol(argv[++c], NULL, 0);
834 crop_y = strtol(argv[++c], NULL, 0);
835 crop_w = strtol(argv[++c], NULL, 0);
836 crop_h = strtol(argv[++c], NULL, 0);
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700837 } else if (!strcmp(argv[c], "-resize") && c < argc - 2) {
838 resize_w = strtol(argv[++c], NULL, 0);
839 resize_h = strtol(argv[++c], NULL, 0);
James Zernb4d0ef82011-07-15 14:53:03 -0700840#ifndef WEBP_DLL
Pascal Massiminocfbf88a2011-04-22 12:14:45 -0700841 } else if (!strcmp(argv[c], "-noasm")) {
Pascal Massiminoe06ac082011-09-02 21:30:08 +0000842 VP8GetCPUInfo = NULL;
James Zernb4d0ef82011-07-15 14:53:03 -0700843#endif
Pascal Massimino650ffa32011-03-24 16:17:10 -0700844 } else if (!strcmp(argv[c], "-version")) {
845 const int version = WebPGetEncoderVersion();
846 printf("%d.%d.%d\n",
847 (version >> 16) & 0xff, (version >> 8) & 0xff, version & 0xff);
848 return 0;
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800849 } else if (!strcmp(argv[c], "-quiet")) {
850 quiet = 1;
851 } else if (!strcmp(argv[c], "-preset") && c < argc - 1) {
852 WebPPreset preset;
853 ++c;
854 if (!strcmp(argv[c], "default")) {
855 preset = WEBP_PRESET_DEFAULT;
856 } else if (!strcmp(argv[c], "photo")) {
857 preset = WEBP_PRESET_PHOTO;
858 } else if (!strcmp(argv[c], "picture")) {
859 preset = WEBP_PRESET_PICTURE;
860 } else if (!strcmp(argv[c], "drawing")) {
861 preset = WEBP_PRESET_DRAWING;
862 } else if (!strcmp(argv[c], "icon")) {
863 preset = WEBP_PRESET_ICON;
864 } else if (!strcmp(argv[c], "text")) {
865 preset = WEBP_PRESET_TEXT;
866 } else {
867 fprintf(stderr, "Error! Unrecognized preset: %s\n", argv[c]);
868 goto Error;
869 }
870 if (!WebPConfigPreset(&config, preset, config.quality)) {
Pascal Massimino6d978a62011-03-17 14:49:19 -0700871 fprintf(stderr, "Error! Could initialize configuration with preset.\n");
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800872 goto Error;
873 }
874 } else if (!strcmp(argv[c], "-v")) {
875 verbose = 1;
876 } else if (argv[c][0] == '-') {
877 fprintf(stderr, "Error! Unknown option '%s'\n", argv[c]);
878 HelpLong();
879 return -1;
880 } else {
881 in_file = argv[c];
882 }
883 }
884
885 if (!WebPValidateConfig(&config)) {
886 fprintf(stderr, "Error! Invalid configuration.\n");
887 goto Error;
888 }
889
Pascal Massimino0744e842011-02-25 12:03:27 -0800890 // Read the input
Pascal Massimino38369c02011-05-04 22:59:51 -0700891 if (verbose) {
Pascal Massimino0744e842011-02-25 12:03:27 -0800892 StopwatchReadAndReset(&stop_watch);
Pascal Massimino38369c02011-05-04 22:59:51 -0700893 }
Pascal Massimino2ab4b722011-04-25 16:58:04 -0700894 if (!ReadPicture(in_file, &picture, keep_alpha)) {
Pascal Massimino6d978a62011-03-17 14:49:19 -0700895 fprintf(stderr, "Error! Cannot read input picture\n");
896 goto Error;
897 }
Pascal Massimino0744e842011-02-25 12:03:27 -0800898 if (verbose) {
899 const double time = StopwatchReadAndReset(&stop_watch);
900 fprintf(stderr, "Time to read input: %.3fs\n", time);
901 }
902
903 // Open the output
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800904 if (out_file) {
905 out = fopen(out_file, "wb");
906 if (!out) {
907 fprintf(stderr, "Error! Cannot open output file '%s'\n", out_file);
908 goto Error;
909 } else {
910 if (!short_output && !quiet) {
911 fprintf(stderr, "Saving file '%s'\n", out_file);
912 }
913 }
914 picture.writer = MyWriter;
915 picture.custom_ptr = (void*)out;
916 } else {
917 out = NULL;
918 if (!quiet && !short_output) {
919 fprintf(stderr, "No output file specified (no -o flag). Encoding will\n");
920 fprintf(stderr, "be performed, but its results discarded.\n\n");
921 }
922 }
923 picture.stats = &stats;
924
Pascal Massimino0744e842011-02-25 12:03:27 -0800925 // Compress
Pascal Massimino38369c02011-05-04 22:59:51 -0700926 if (verbose) {
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800927 StopwatchReadAndReset(&stop_watch);
Pascal Massimino38369c02011-05-04 22:59:51 -0700928 }
Pascal Massimino6d978a62011-03-17 14:49:19 -0700929 if (crop != 0 && !WebPPictureCrop(&picture, crop_x, crop_y, crop_w, crop_h)) {
930 fprintf(stderr, "Error! Cannot crop picture\n");
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800931 goto Error;
Pascal Massimino6d978a62011-03-17 14:49:19 -0700932 }
Pascal Massimino6d0e66c2011-05-02 17:19:00 -0700933 if ((resize_w | resize_h) > 0) {
934 if (!WebPPictureRescale(&picture, resize_w, resize_h)) {
935 fprintf(stderr, "Error! Cannot resize picture\n");
936 goto Error;
937 }
938 }
939 if (picture.extra_info_type > 0) {
940 AllocExtraInfo(&picture);
941 }
Pascal Massimino6d978a62011-03-17 14:49:19 -0700942 if (!WebPEncode(&config, &picture)) {
943 fprintf(stderr, "Error! Cannot encode picture as WebP\n");
Pascal Massiminoca7a2fd2011-06-02 06:55:03 -0700944 fprintf(stderr, "Error code: %d (%s)\n",
945 picture.error_code, kErrorMessages[picture.error_code]);
Pascal Massimino6d978a62011-03-17 14:49:19 -0700946 goto Error;
947 }
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800948 if (verbose) {
949 const double time = StopwatchReadAndReset(&stop_watch);
950 fprintf(stderr, "Time to encode picture: %.3fs\n", time);
951 }
Pascal Massimino0744e842011-02-25 12:03:27 -0800952
953 // Write info
Pascal Massimino38369c02011-05-04 22:59:51 -0700954 if (dump_file) {
955 DumpPicture(&picture, dump_file);
956 }
957 if (!quiet) {
958 PrintExtraInfo(&picture, short_output);
959 }
Pascal Massiminof61d14a2011-02-18 23:33:46 -0800960
961 Error:
962 free(picture.extra_info);
963 WebPPictureFree(&picture);
964 if (out != NULL) {
965 fclose(out);
966 }
967
968 return 0;
969}
970
James Zernc7e86ab2011-08-25 14:22:32 -0700971//------------------------------------------------------------------------------