| // Copyright 2011 Google Inc. All Rights Reserved. |
| // |
| // Use of this source code is governed by a BSD-style license |
| // that can be found in the COPYING file in the root of the source |
| // tree. An additional intellectual property rights grant can be found |
| // in the file PATENTS. All contributing project authors may |
| // be found in the AUTHORS file in the root of the source tree. |
| // ----------------------------------------------------------------------------- |
| // |
| // simple command line calling the WebPEncode function. |
| // Encodes a raw .YUV into WebP bitstream |
| // |
| // Author: Skal (pascal.massimino@gmail.com) |
| |
| #include <assert.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| |
| #ifdef HAVE_CONFIG_H |
| #include "webp/config.h" |
| #endif |
| |
| #include "../examples/example_util.h" |
| #include "../imageio/image_dec.h" |
| #include "../imageio/imageio_util.h" |
| #include "../imageio/webpdec.h" |
| #include "./stopwatch.h" |
| #include "./unicode.h" |
| #include "sharpyuv/sharpyuv.h" |
| #include "webp/encode.h" |
| |
| #ifndef WEBP_DLL |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| extern void* VP8GetCPUInfo; // opaque forward declaration. |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| #endif // WEBP_DLL |
| |
| //------------------------------------------------------------------------------ |
| |
| static int verbose = 0; |
| |
| static int ReadYUV(const uint8_t* const data, size_t data_size, |
| WebPPicture* const pic) { |
| const int use_argb = pic->use_argb; |
| const int uv_width = (pic->width + 1) / 2; |
| const int uv_height = (pic->height + 1) / 2; |
| const int y_plane_size = pic->width * pic->height; |
| const int uv_plane_size = uv_width * uv_height; |
| const size_t expected_data_size = y_plane_size + 2 * uv_plane_size; |
| |
| if (data_size != expected_data_size) { |
| fprintf(stderr, |
| "input data doesn't have the expected size (%d instead of %d)\n", |
| (int)data_size, (int)expected_data_size); |
| return 0; |
| } |
| |
| pic->use_argb = 0; |
| if (!WebPPictureAlloc(pic)) return 0; |
| ImgIoUtilCopyPlane(data, pic->width, pic->y, pic->y_stride, |
| pic->width, pic->height); |
| ImgIoUtilCopyPlane(data + y_plane_size, uv_width, |
| pic->u, pic->uv_stride, uv_width, uv_height); |
| ImgIoUtilCopyPlane(data + y_plane_size + uv_plane_size, uv_width, |
| pic->v, pic->uv_stride, uv_width, uv_height); |
| return use_argb ? WebPPictureYUVAToARGB(pic) : 1; |
| } |
| |
| #ifdef HAVE_WINCODEC_H |
| |
| static int ReadPicture(const char* const filename, WebPPicture* const pic, |
| int keep_alpha, Metadata* const metadata) { |
| int ok = 0; |
| const uint8_t* data = NULL; |
| size_t data_size = 0; |
| if (pic->width != 0 && pic->height != 0) { |
| ok = ImgIoUtilReadFile(filename, &data, &data_size); |
| ok = ok && ReadYUV(data, data_size, pic); |
| } else { |
| // If no size specified, try to decode it using WIC. |
| ok = ReadPictureWithWIC(filename, pic, keep_alpha, metadata); |
| if (!ok) { |
| ok = ImgIoUtilReadFile(filename, &data, &data_size); |
| ok = ok && ReadWebP(data, data_size, pic, keep_alpha, metadata); |
| } |
| } |
| if (!ok) { |
| WFPRINTF(stderr, "Error! Could not process file %s\n", |
| (const W_CHAR*)filename); |
| } |
| WebPFree((void*)data); |
| return ok; |
| } |
| |
| #else // !HAVE_WINCODEC_H |
| |
| static int ReadPicture(const char* const filename, WebPPicture* const pic, |
| int keep_alpha, Metadata* const metadata) { |
| const uint8_t* data = NULL; |
| size_t data_size = 0; |
| int ok = 0; |
| |
| ok = ImgIoUtilReadFile(filename, &data, &data_size); |
| if (!ok) goto End; |
| |
| if (pic->width == 0 || pic->height == 0) { |
| WebPImageReader reader = WebPGuessImageReader(data, data_size); |
| ok = reader(data, data_size, pic, keep_alpha, metadata); |
| } else { |
| // If image size is specified, infer it as YUV format. |
| ok = ReadYUV(data, data_size, pic); |
| } |
| End: |
| if (!ok) { |
| WFPRINTF(stderr, "Error! Could not process file %s\n", |
| (const W_CHAR*)filename); |
| } |
| WebPFree((void*)data); |
| return ok; |
| } |
| |
| #endif // !HAVE_WINCODEC_H |
| |
| static void AllocExtraInfo(WebPPicture* const pic) { |
| const int mb_w = (pic->width + 15) / 16; |
| const int mb_h = (pic->height + 15) / 16; |
| pic->extra_info = |
| (uint8_t*)WebPMalloc(mb_w * mb_h * sizeof(*pic->extra_info)); |
| } |
| |
| static void PrintByteCount(const int bytes[4], int total_size, |
| int* const totals) { |
| int s; |
| int total = 0; |
| for (s = 0; s < 4; ++s) { |
| fprintf(stderr, "| %7d ", bytes[s]); |
| total += bytes[s]; |
| if (totals) totals[s] += bytes[s]; |
| } |
| fprintf(stderr, "| %7d (%.1f%%)\n", total, 100.f * total / total_size); |
| } |
| |
| static void PrintPercents(const int counts[4]) { |
| int s; |
| const int total = counts[0] + counts[1] + counts[2] + counts[3]; |
| for (s = 0; s < 4; ++s) { |
| fprintf(stderr, "| %3d%%", (int)(100. * counts[s] / total + .5)); |
| } |
| fprintf(stderr, "| %7d\n", total); |
| } |
| |
| static void PrintValues(const int values[4]) { |
| int s; |
| for (s = 0; s < 4; ++s) { |
| fprintf(stderr, "| %7d ", values[s]); |
| } |
| fprintf(stderr, "|\n"); |
| } |
| |
| static void PrintFullLosslessInfo(const WebPAuxStats* const stats, |
| const char* const description) { |
| fprintf(stderr, "Lossless-%s compressed size: %d bytes\n", |
| description, stats->lossless_size); |
| fprintf(stderr, " * Header size: %d bytes, image data size: %d\n", |
| stats->lossless_hdr_size, stats->lossless_data_size); |
| if (stats->lossless_features) { |
| fprintf(stderr, " * Lossless features used:"); |
| if (stats->lossless_features & 1) fprintf(stderr, " PREDICTION"); |
| if (stats->lossless_features & 2) fprintf(stderr, " CROSS-COLOR-TRANSFORM"); |
| if (stats->lossless_features & 4) fprintf(stderr, " SUBTRACT-GREEN"); |
| if (stats->lossless_features & 8) fprintf(stderr, " PALETTE"); |
| fprintf(stderr, "\n"); |
| } |
| fprintf(stderr, " * Precision Bits: histogram=%d transform=%d cache=%d\n", |
| stats->histogram_bits, stats->transform_bits, stats->cache_bits); |
| if (stats->palette_size > 0) { |
| fprintf(stderr, " * Palette size: %d\n", stats->palette_size); |
| } |
| } |
| |
| static void PrintExtraInfoLossless(const WebPPicture* const pic, |
| int short_output, |
| const char* const file_name) { |
| const WebPAuxStats* const stats = pic->stats; |
| if (short_output) { |
| fprintf(stderr, "%7d %2.2f\n", stats->coded_size, stats->PSNR[3]); |
| } else { |
| WFPRINTF(stderr, "File: %s\n", (const W_CHAR*)file_name); |
| fprintf(stderr, "Dimension: %d x %d\n", pic->width, pic->height); |
| fprintf(stderr, "Output: %d bytes (%.2f bpp)\n", stats->coded_size, |
| 8.f * stats->coded_size / pic->width / pic->height); |
| PrintFullLosslessInfo(stats, "ARGB"); |
| } |
| } |
| |
| static void PrintExtraInfoLossy(const WebPPicture* const pic, int short_output, |
| int full_details, |
| const char* const file_name) { |
| const WebPAuxStats* const stats = pic->stats; |
| if (short_output) { |
| fprintf(stderr, "%7d %2.2f\n", stats->coded_size, stats->PSNR[3]); |
| } else { |
| const int num_i4 = stats->block_count[0]; |
| const int num_i16 = stats->block_count[1]; |
| const int num_skip = stats->block_count[2]; |
| const int total = num_i4 + num_i16; |
| WFPRINTF(stderr, "File: %s\n", (const W_CHAR*)file_name); |
| fprintf(stderr, "Dimension: %d x %d%s\n", |
| pic->width, pic->height, |
| stats->alpha_data_size ? " (with alpha)" : ""); |
| fprintf(stderr, "Output: " |
| "%d bytes Y-U-V-All-PSNR %2.2f %2.2f %2.2f %2.2f dB\n" |
| " (%.2f bpp)\n", |
| stats->coded_size, |
| stats->PSNR[0], stats->PSNR[1], stats->PSNR[2], stats->PSNR[3], |
| 8.f * stats->coded_size / pic->width / pic->height); |
| if (total > 0) { |
| int totals[4] = { 0, 0, 0, 0 }; |
| fprintf(stderr, "block count: intra4: %6d (%.2f%%)\n" |
| " intra16: %6d (%.2f%%)\n" |
| " skipped: %6d (%.2f%%)\n", |
| num_i4, 100.f * num_i4 / total, |
| num_i16, 100.f * num_i16 / total, |
| num_skip, 100.f * num_skip / total); |
| fprintf(stderr, "bytes used: header: %6d (%.1f%%)\n" |
| " mode-partition: %6d (%.1f%%)\n", |
| stats->header_bytes[0], |
| 100.f * stats->header_bytes[0] / stats->coded_size, |
| stats->header_bytes[1], |
| 100.f * stats->header_bytes[1] / stats->coded_size); |
| if (stats->alpha_data_size > 0) { |
| fprintf(stderr, " transparency: %6d (%.1f dB)\n", |
| stats->alpha_data_size, stats->PSNR[4]); |
| } |
| fprintf(stderr, " Residuals bytes " |
| "|segment 1|segment 2|segment 3" |
| "|segment 4| total\n"); |
| if (full_details) { |
| fprintf(stderr, " intra4-coeffs: "); |
| PrintByteCount(stats->residual_bytes[0], stats->coded_size, totals); |
| fprintf(stderr, " intra16-coeffs: "); |
| PrintByteCount(stats->residual_bytes[1], stats->coded_size, totals); |
| fprintf(stderr, " chroma coeffs: "); |
| PrintByteCount(stats->residual_bytes[2], stats->coded_size, totals); |
| } |
| fprintf(stderr, " macroblocks: "); |
| PrintPercents(stats->segment_size); |
| fprintf(stderr, " quantizer: "); |
| PrintValues(stats->segment_quant); |
| fprintf(stderr, " filter level: "); |
| PrintValues(stats->segment_level); |
| if (full_details) { |
| fprintf(stderr, "------------------+---------"); |
| fprintf(stderr, "+---------+---------+---------+-----------------\n"); |
| fprintf(stderr, " segments total: "); |
| PrintByteCount(totals, stats->coded_size, NULL); |
| } |
| } |
| if (stats->lossless_size > 0) { |
| PrintFullLosslessInfo(stats, "alpha"); |
| } |
| } |
| } |
| |
| static void PrintMapInfo(const WebPPicture* const pic) { |
| if (pic->extra_info != NULL) { |
| const int mb_w = (pic->width + 15) / 16; |
| const int mb_h = (pic->height + 15) / 16; |
| const int type = pic->extra_info_type; |
| int x, y; |
| for (y = 0; y < mb_h; ++y) { |
| for (x = 0; x < mb_w; ++x) { |
| const int c = pic->extra_info[x + y * mb_w]; |
| if (type == 1) { // intra4/intra16 |
| fprintf(stderr, "%c", "+."[c]); |
| } else if (type == 2) { // segments |
| fprintf(stderr, "%c", ".-*X"[c]); |
| } else if (type == 3) { // quantizers |
| fprintf(stderr, "%.2d ", c); |
| } else if (type == 6 || type == 7) { |
| fprintf(stderr, "%3d ", c); |
| } else { |
| fprintf(stderr, "0x%.2x ", c); |
| } |
| } |
| fprintf(stderr, "\n"); |
| } |
| } |
| } |
| |
| //------------------------------------------------------------------------------ |
| |
| static int MyWriter(const uint8_t* data, size_t data_size, |
| const WebPPicture* const pic) { |
| FILE* const out = (FILE*)pic->custom_ptr; |
| return data_size ? (fwrite(data, data_size, 1, out) == 1) : 1; |
| } |
| |
| // Dumps a picture as a PGM file using the IMC4 layout. |
| static int DumpPicture(const WebPPicture* const picture, const char* PGM_name) { |
| int y; |
| const int uv_width = (picture->width + 1) / 2; |
| const int uv_height = (picture->height + 1) / 2; |
| const int stride = (picture->width + 1) & ~1; |
| const uint8_t* src_y = picture->y; |
| const uint8_t* src_u = picture->u; |
| const uint8_t* src_v = picture->v; |
| const uint8_t* src_a = picture->a; |
| const int alpha_height = |
| WebPPictureHasTransparency(picture) ? picture->height : 0; |
| const int height = picture->height + uv_height + alpha_height; |
| FILE* const f = WFOPEN(PGM_name, "wb"); |
| if (f == NULL) return 0; |
| fprintf(f, "P5\n%d %d\n255\n", stride, height); |
| for (y = 0; y < picture->height; ++y) { |
| if (fwrite(src_y, picture->width, 1, f) != 1) return 0; |
| if (picture->width & 1) fputc(0, f); // pad |
| src_y += picture->y_stride; |
| } |
| for (y = 0; y < uv_height; ++y) { |
| if (fwrite(src_u, uv_width, 1, f) != 1) return 0; |
| if (fwrite(src_v, uv_width, 1, f) != 1) return 0; |
| src_u += picture->uv_stride; |
| src_v += picture->uv_stride; |
| } |
| for (y = 0; y < alpha_height; ++y) { |
| if (fwrite(src_a, picture->width, 1, f) != 1) return 0; |
| if (picture->width & 1) fputc(0, f); // pad |
| src_a += picture->a_stride; |
| } |
| fclose(f); |
| return 1; |
| } |
| |
| // ----------------------------------------------------------------------------- |
| // Metadata writing. |
| |
| enum { |
| METADATA_EXIF = (1 << 0), |
| METADATA_ICC = (1 << 1), |
| METADATA_XMP = (1 << 2), |
| METADATA_ALL = METADATA_EXIF | METADATA_ICC | METADATA_XMP |
| }; |
| |
| static const int kChunkHeaderSize = 8; |
| static const int kTagSize = 4; |
| |
| static void PrintMetadataInfo(const Metadata* const metadata, |
| int metadata_written) { |
| if (metadata == NULL || metadata_written == 0) return; |
| |
| fprintf(stderr, "Metadata:\n"); |
| if (metadata_written & METADATA_ICC) { |
| fprintf(stderr, " * ICC profile: %6d bytes\n", (int)metadata->iccp.size); |
| } |
| if (metadata_written & METADATA_EXIF) { |
| fprintf(stderr, " * EXIF data: %6d bytes\n", (int)metadata->exif.size); |
| } |
| if (metadata_written & METADATA_XMP) { |
| fprintf(stderr, " * XMP data: %6d bytes\n", (int)metadata->xmp.size); |
| } |
| } |
| |
| // Outputs, in little endian, 'num' bytes from 'val' to 'out'. |
| static int WriteLE(FILE* const out, uint32_t val, int num) { |
| uint8_t buf[4]; |
| int i; |
| for (i = 0; i < num; ++i) { |
| buf[i] = (uint8_t)(val & 0xff); |
| val >>= 8; |
| } |
| return (fwrite(buf, num, 1, out) == 1); |
| } |
| |
| static int WriteLE24(FILE* const out, uint32_t val) { |
| return WriteLE(out, val, 3); |
| } |
| |
| static int WriteLE32(FILE* const out, uint32_t val) { |
| return WriteLE(out, val, 4); |
| } |
| |
| static int WriteMetadataChunk(FILE* const out, const char fourcc[4], |
| const MetadataPayload* const payload) { |
| const uint8_t zero = 0; |
| const size_t need_padding = payload->size & 1; |
| int ok = (fwrite(fourcc, kTagSize, 1, out) == 1); |
| ok = ok && WriteLE32(out, (uint32_t)payload->size); |
| ok = ok && (fwrite(payload->bytes, payload->size, 1, out) == 1); |
| return ok && (fwrite(&zero, need_padding, need_padding, out) == need_padding); |
| } |
| |
| // Sets 'flag' in 'vp8x_flags' and updates 'metadata_size' with the size of the |
| // chunk if there is metadata and 'keep' is true. |
| static int UpdateFlagsAndSize(const MetadataPayload* const payload, |
| int keep, int flag, |
| uint32_t* vp8x_flags, uint64_t* metadata_size) { |
| if (keep && payload->bytes != NULL && payload->size > 0) { |
| *vp8x_flags |= flag; |
| *metadata_size += kChunkHeaderSize + payload->size + (payload->size & 1); |
| return 1; |
| } |
| return 0; |
| } |
| |
| // Writes a WebP file using the image contained in 'memory_writer' and the |
| // metadata from 'metadata'. Metadata is controlled by 'keep_metadata' and the |
| // availability in 'metadata'. Returns true on success. |
| // For details see doc/webp-container-spec.txt#extended-file-format. |
| static int WriteWebPWithMetadata(FILE* const out, |
| const WebPPicture* const picture, |
| const WebPMemoryWriter* const memory_writer, |
| const Metadata* const metadata, |
| int keep_metadata, |
| int* const metadata_written) { |
| const char kVP8XHeader[] = "VP8X\x0a\x00\x00\x00"; |
| const int kAlphaFlag = 0x10; |
| const int kEXIFFlag = 0x08; |
| const int kICCPFlag = 0x20; |
| const int kXMPFlag = 0x04; |
| const size_t kRiffHeaderSize = 12; |
| const size_t kMaxChunkPayload = ~0 - kChunkHeaderSize - 1; |
| const size_t kMinSize = kRiffHeaderSize + kChunkHeaderSize; |
| uint32_t flags = 0; |
| uint64_t metadata_size = 0; |
| const int write_exif = UpdateFlagsAndSize(&metadata->exif, |
| !!(keep_metadata & METADATA_EXIF), |
| kEXIFFlag, &flags, &metadata_size); |
| const int write_iccp = UpdateFlagsAndSize(&metadata->iccp, |
| !!(keep_metadata & METADATA_ICC), |
| kICCPFlag, &flags, &metadata_size); |
| const int write_xmp = UpdateFlagsAndSize(&metadata->xmp, |
| !!(keep_metadata & METADATA_XMP), |
| kXMPFlag, &flags, &metadata_size); |
| uint8_t* webp = memory_writer->mem; |
| size_t webp_size = memory_writer->size; |
| |
| *metadata_written = 0; |
| |
| if (webp_size < kMinSize) return 0; |
| if (webp_size - kChunkHeaderSize + metadata_size > kMaxChunkPayload) { |
| fprintf(stderr, "Error! Addition of metadata would exceed " |
| "container size limit.\n"); |
| return 0; |
| } |
| |
| if (metadata_size > 0) { |
| const int kVP8XChunkSize = 18; |
| const int has_vp8x = !memcmp(webp + kRiffHeaderSize, "VP8X", kTagSize); |
| const uint32_t riff_size = (uint32_t)(webp_size - kChunkHeaderSize + |
| (has_vp8x ? 0 : kVP8XChunkSize) + |
| metadata_size); |
| // RIFF |
| int ok = (fwrite(webp, kTagSize, 1, out) == 1); |
| // RIFF size (file header size is not recorded) |
| ok = ok && WriteLE32(out, riff_size); |
| webp += kChunkHeaderSize; |
| webp_size -= kChunkHeaderSize; |
| // WEBP |
| ok = ok && (fwrite(webp, kTagSize, 1, out) == 1); |
| webp += kTagSize; |
| webp_size -= kTagSize; |
| if (has_vp8x) { // update the existing VP8X flags |
| webp[kChunkHeaderSize] |= (uint8_t)(flags & 0xff); |
| ok = ok && (fwrite(webp, kVP8XChunkSize, 1, out) == 1); |
| webp += kVP8XChunkSize; |
| webp_size -= kVP8XChunkSize; |
| } else { |
| const int is_lossless = !memcmp(webp, "VP8L", kTagSize); |
| if (is_lossless) { |
| // Presence of alpha is stored in the 37th bit (29th after the |
| // signature) of VP8L data. |
| if (webp[kChunkHeaderSize + 4] & (1 << 4)) flags |= kAlphaFlag; |
| } |
| ok = ok && (fwrite(kVP8XHeader, kChunkHeaderSize, 1, out) == 1); |
| ok = ok && WriteLE32(out, flags); |
| ok = ok && WriteLE24(out, picture->width - 1); |
| ok = ok && WriteLE24(out, picture->height - 1); |
| } |
| if (write_iccp) { |
| ok = ok && WriteMetadataChunk(out, "ICCP", &metadata->iccp); |
| *metadata_written |= METADATA_ICC; |
| } |
| // Image |
| ok = ok && (fwrite(webp, webp_size, 1, out) == 1); |
| if (write_exif) { |
| ok = ok && WriteMetadataChunk(out, "EXIF", &metadata->exif); |
| *metadata_written |= METADATA_EXIF; |
| } |
| if (write_xmp) { |
| ok = ok && WriteMetadataChunk(out, "XMP ", &metadata->xmp); |
| *metadata_written |= METADATA_XMP; |
| } |
| return ok; |
| } |
| |
| // No metadata, just write the original image file. |
| return (fwrite(webp, webp_size, 1, out) == 1); |
| } |
| |
| //------------------------------------------------------------------------------ |
| |
| static int ProgressReport(int percent, const WebPPicture* const picture) { |
| fprintf(stderr, "[%s]: %3d %% \r", |
| (char*)picture->user_data, percent); |
| return 1; // all ok |
| } |
| |
| //------------------------------------------------------------------------------ |
| |
| static void HelpShort(void) { |
| printf("Usage:\n\n"); |
| printf(" cwebp [options] -q quality input.png -o output.webp\n\n"); |
| printf("where quality is between 0 (poor) to 100 (very good).\n"); |
| printf("Typical value is around 80.\n\n"); |
| printf("Try -longhelp for an exhaustive list of advanced options.\n"); |
| } |
| |
| static void HelpLong(void) { |
| printf("Usage:\n"); |
| printf(" cwebp [-preset <...>] [options] in_file [-o out_file]\n\n"); |
| printf("If input size (-s) for an image is not specified, it is\n" |
| "assumed to be a PNG, JPEG, TIFF or WebP file.\n"); |
| printf("Note: Animated PNG and WebP files are not supported.\n"); |
| #ifdef HAVE_WINCODEC_H |
| printf("Windows builds can take as input any of the files handled by WIC.\n"); |
| #endif |
| printf("\nOptions:\n"); |
| printf(" -h / -help ............. short help\n"); |
| printf(" -H / -longhelp ......... long help\n"); |
| printf(" -q <float> ............. quality factor (0:small..100:big), " |
| "default=75\n"); |
| printf(" -alpha_q <int> ......... transparency-compression quality (0..100)," |
| "\n default=100\n"); |
| printf(" -preset <string> ....... preset setting, one of:\n"); |
| printf(" default, photo, picture,\n"); |
| printf(" drawing, icon, text\n"); |
| printf(" -preset must come first, as it overwrites other parameters\n"); |
| printf(" -z <int> ............... activates lossless preset with given\n" |
| " level in [0:fast, ..., 9:slowest]\n"); |
| printf("\n"); |
| printf(" -m <int> ............... compression method (0=fast, 6=slowest), " |
| "default=4\n"); |
| printf(" -segments <int> ........ number of segments to use (1..4), " |
| "default=4\n"); |
| printf(" -size <int> ............ target size (in bytes)\n"); |
| printf(" -psnr <float> .......... target PSNR (in dB. typically: 42)\n"); |
| printf("\n"); |
| printf(" -s <int> <int> ......... input size (width x height) for YUV\n"); |
| printf(" -sns <int> ............. spatial noise shaping (0:off, 100:max), " |
| "default=50\n"); |
| printf(" -f <int> ............... filter strength (0=off..100), " |
| "default=60\n"); |
| printf(" -sharpness <int> ....... " |
| "filter sharpness (0:most .. 7:least sharp), default=0\n"); |
| printf(" -strong ................ use strong filter instead " |
| "of simple (default)\n"); |
| printf(" -nostrong .............. use simple filter instead of strong\n"); |
| printf(" -sharp_yuv ............. use sharper (and slower) RGB->YUV " |
| "conversion\n"); |
| printf(" -partition_limit <int> . limit quality to fit the 512k limit on\n"); |
| printf(" " |
| "the first partition (0=no degradation ... 100=full)\n"); |
| printf(" -pass <int> ............ analysis pass number (1..10)\n"); |
| printf(" -qrange <min> <max> .... specifies the permissible quality range\n" |
| " (default: 0 100)\n"); |
| printf(" -crop <x> <y> <w> <h> .. crop picture with the given rectangle\n"); |
| printf(" -resize <w> <h> ........ resize picture (*after* any cropping)\n"); |
| printf(" -mt .................... use multi-threading if available\n"); |
| printf(" -low_memory ............ reduce memory usage (slower encoding)\n"); |
| printf(" -map <int> ............. print map of extra info\n"); |
| printf(" -print_psnr ............ prints averaged PSNR distortion\n"); |
| printf(" -print_ssim ............ prints averaged SSIM distortion\n"); |
| printf(" -print_lsim ............ prints local-similarity distortion\n"); |
| printf(" -d <file.pgm> .......... dump the compressed output (PGM file)\n"); |
| printf(" -alpha_method <int> .... transparency-compression method (0..1), " |
| "default=1\n"); |
| printf(" -alpha_filter <string> . predictive filtering for alpha plane,\n"); |
| printf(" one of: none, fast (default) or best\n"); |
| printf(" -exact ................. preserve RGB values in transparent area, " |
| "default=off\n"); |
| printf(" -blend_alpha <hex> ..... blend colors against background color\n" |
| " expressed as RGB values written in\n" |
| " hexadecimal, e.g. 0xc0e0d0 for red=0xc0\n" |
| " green=0xe0 and blue=0xd0\n"); |
| printf(" -noalpha ............... discard any transparency information\n"); |
| printf(" -lossless .............. encode image losslessly, default=off\n"); |
| printf(" -near_lossless <int> ... use near-lossless image preprocessing\n" |
| " (0..100=off), default=100\n"); |
| printf(" -hint <string> ......... specify image characteristics hint,\n"); |
| printf(" one of: photo, picture or graph\n"); |
| |
| printf("\n"); |
| printf(" -metadata <string> ..... comma separated list of metadata to\n"); |
| printf(" "); |
| printf("copy from the input to the output if present.\n"); |
| printf(" " |
| "Valid values: all, none (default), exif, icc, xmp\n"); |
| |
| printf("\n"); |
| printf(" -short ................. condense printed message\n"); |
| printf(" -quiet ................. don't print anything\n"); |
| printf(" -version ............... print version number and exit\n"); |
| #ifndef WEBP_DLL |
| printf(" -noasm ................. disable all assembly optimizations\n"); |
| #endif |
| printf(" -v ..................... verbose, e.g. print encoding/decoding " |
| "times\n"); |
| printf(" -progress .............. report encoding progress\n"); |
| printf("\n"); |
| printf("Experimental Options:\n"); |
| printf(" -jpeg_like ............. roughly match expected JPEG size\n"); |
| printf(" -af .................... auto-adjust filter strength\n"); |
| printf(" -pre <int> ............. pre-processing filter\n"); |
| printf("\n"); |
| printf("Supported input formats:\n %s\n", WebPGetEnabledInputFileFormats()); |
| } |
| |
| //------------------------------------------------------------------------------ |
| // Error messages |
| |
| static const char* const kErrorMessages[VP8_ENC_ERROR_LAST] = { |
| "OK", |
| "OUT_OF_MEMORY: Out of memory allocating objects", |
| "BITSTREAM_OUT_OF_MEMORY: Out of memory re-allocating byte buffer", |
| "NULL_PARAMETER: NULL parameter passed to function", |
| "INVALID_CONFIGURATION: configuration is invalid", |
| "BAD_DIMENSION: Bad picture dimension. Maximum width and height " |
| "allowed is 16383 pixels.", |
| "PARTITION0_OVERFLOW: Partition #0 is too big to fit 512k.\n" |
| "To reduce the size of this partition, try using less segments " |
| "with the -segments option, and eventually reduce the number of " |
| "header bits using -partition_limit. More details are available " |
| "in the manual (`man cwebp`)", |
| "PARTITION_OVERFLOW: Partition is too big to fit 16M", |
| "BAD_WRITE: Picture writer returned an I/O error", |
| "FILE_TOO_BIG: File would be too big to fit in 4G", |
| "USER_ABORT: encoding abort requested by user" |
| }; |
| |
| //------------------------------------------------------------------------------ |
| |
| int main(int argc, const char* argv[]) { |
| int return_value = -1; |
| const char* in_file = NULL, *out_file = NULL, *dump_file = NULL; |
| FILE* out = NULL; |
| int c; |
| int short_output = 0; |
| int quiet = 0; |
| int keep_alpha = 1; |
| int blend_alpha = 0; |
| uint32_t background_color = 0xffffffu; |
| int crop = 0, crop_x = 0, crop_y = 0, crop_w = 0, crop_h = 0; |
| int resize_w = 0, resize_h = 0; |
| int lossless_preset = 6; |
| int use_lossless_preset = -1; // -1=unset, 0=don't use, 1=use it |
| int show_progress = 0; |
| int keep_metadata = 0; |
| int metadata_written = 0; |
| WebPPicture picture; |
| int print_distortion = -1; // -1=off, 0=PSNR, 1=SSIM, 2=LSIM |
| WebPPicture original_picture; // when PSNR or SSIM is requested |
| WebPConfig config; |
| WebPAuxStats stats; |
| WebPMemoryWriter memory_writer; |
| int use_memory_writer; |
| Metadata metadata; |
| Stopwatch stop_watch; |
| |
| INIT_WARGV(argc, argv); |
| |
| MetadataInit(&metadata); |
| WebPMemoryWriterInit(&memory_writer); |
| if (!WebPPictureInit(&picture) || |
| !WebPPictureInit(&original_picture) || |
| !WebPConfigInit(&config)) { |
| fprintf(stderr, "Error! Version mismatch!\n"); |
| FREE_WARGV_AND_RETURN(-1); |
| } |
| |
| if (argc == 1) { |
| HelpShort(); |
| FREE_WARGV_AND_RETURN(0); |
| } |
| |
| for (c = 1; c < argc; ++c) { |
| int parse_error = 0; |
| if (!strcmp(argv[c], "-h") || !strcmp(argv[c], "-help")) { |
| HelpShort(); |
| FREE_WARGV_AND_RETURN(0); |
| } else if (!strcmp(argv[c], "-H") || !strcmp(argv[c], "-longhelp")) { |
| HelpLong(); |
| FREE_WARGV_AND_RETURN(0); |
| } else if (!strcmp(argv[c], "-o") && c + 1 < argc) { |
| out_file = (const char*)GET_WARGV(argv, ++c); |
| } else if (!strcmp(argv[c], "-d") && c + 1 < argc) { |
| dump_file = (const char*)GET_WARGV(argv, ++c); |
| config.show_compressed = 1; |
| } else if (!strcmp(argv[c], "-print_psnr")) { |
| config.show_compressed = 1; |
| print_distortion = 0; |
| } else if (!strcmp(argv[c], "-print_ssim")) { |
| config.show_compressed = 1; |
| print_distortion = 1; |
| } else if (!strcmp(argv[c], "-print_lsim")) { |
| config.show_compressed = 1; |
| print_distortion = 2; |
| } else if (!strcmp(argv[c], "-short")) { |
| ++short_output; |
| } else if (!strcmp(argv[c], "-s") && c + 2 < argc) { |
| picture.width = ExUtilGetInt(argv[++c], 0, &parse_error); |
| picture.height = ExUtilGetInt(argv[++c], 0, &parse_error); |
| if (picture.width > WEBP_MAX_DIMENSION || picture.width < 0 || |
| picture.height > WEBP_MAX_DIMENSION || picture.height < 0) { |
| fprintf(stderr, |
| "Specified dimension (%d x %d) is out of range.\n", |
| picture.width, picture.height); |
| goto Error; |
| } |
| } else if (!strcmp(argv[c], "-m") && c + 1 < argc) { |
| config.method = ExUtilGetInt(argv[++c], 0, &parse_error); |
| use_lossless_preset = 0; // disable -z option |
| } else if (!strcmp(argv[c], "-q") && c + 1 < argc) { |
| config.quality = ExUtilGetFloat(argv[++c], &parse_error); |
| use_lossless_preset = 0; // disable -z option |
| } else if (!strcmp(argv[c], "-z") && c + 1 < argc) { |
| lossless_preset = ExUtilGetInt(argv[++c], 0, &parse_error); |
| if (use_lossless_preset != 0) use_lossless_preset = 1; |
| } else if (!strcmp(argv[c], "-alpha_q") && c + 1 < argc) { |
| config.alpha_quality = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-alpha_method") && c + 1 < argc) { |
| config.alpha_compression = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-alpha_cleanup")) { |
| // This flag is obsolete, does opposite of -exact. |
| config.exact = 0; |
| } else if (!strcmp(argv[c], "-exact")) { |
| config.exact = 1; |
| } else if (!strcmp(argv[c], "-blend_alpha") && c + 1 < argc) { |
| blend_alpha = 1; |
| // background color is given in hex with an optional '0x' prefix |
| background_color = ExUtilGetInt(argv[++c], 16, &parse_error); |
| background_color = background_color & 0x00ffffffu; |
| } else if (!strcmp(argv[c], "-alpha_filter") && c + 1 < argc) { |
| ++c; |
| if (!strcmp(argv[c], "none")) { |
| config.alpha_filtering = 0; |
| } else if (!strcmp(argv[c], "fast")) { |
| config.alpha_filtering = 1; |
| } else if (!strcmp(argv[c], "best")) { |
| config.alpha_filtering = 2; |
| } else { |
| fprintf(stderr, "Error! Unrecognized alpha filter: %s\n", argv[c]); |
| goto Error; |
| } |
| } else if (!strcmp(argv[c], "-noalpha")) { |
| keep_alpha = 0; |
| } else if (!strcmp(argv[c], "-lossless")) { |
| config.lossless = 1; |
| } else if (!strcmp(argv[c], "-near_lossless") && c + 1 < argc) { |
| config.near_lossless = ExUtilGetInt(argv[++c], 0, &parse_error); |
| config.lossless = 1; // use near-lossless only with lossless |
| } else if (!strcmp(argv[c], "-hint") && c + 1 < argc) { |
| ++c; |
| if (!strcmp(argv[c], "photo")) { |
| config.image_hint = WEBP_HINT_PHOTO; |
| } else if (!strcmp(argv[c], "picture")) { |
| config.image_hint = WEBP_HINT_PICTURE; |
| } else if (!strcmp(argv[c], "graph")) { |
| config.image_hint = WEBP_HINT_GRAPH; |
| } else { |
| fprintf(stderr, "Error! Unrecognized image hint: %s\n", argv[c]); |
| goto Error; |
| } |
| } else if (!strcmp(argv[c], "-size") && c + 1 < argc) { |
| config.target_size = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-psnr") && c + 1 < argc) { |
| config.target_PSNR = ExUtilGetFloat(argv[++c], &parse_error); |
| } else if (!strcmp(argv[c], "-sns") && c + 1 < argc) { |
| config.sns_strength = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-f") && c + 1 < argc) { |
| config.filter_strength = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-af")) { |
| config.autofilter = 1; |
| } else if (!strcmp(argv[c], "-jpeg_like")) { |
| config.emulate_jpeg_size = 1; |
| } else if (!strcmp(argv[c], "-mt")) { |
| ++config.thread_level; // increase thread level |
| } else if (!strcmp(argv[c], "-low_memory")) { |
| config.low_memory = 1; |
| } else if (!strcmp(argv[c], "-strong")) { |
| config.filter_type = 1; |
| } else if (!strcmp(argv[c], "-nostrong")) { |
| config.filter_type = 0; |
| } else if (!strcmp(argv[c], "-sharpness") && c + 1 < argc) { |
| config.filter_sharpness = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-sharp_yuv")) { |
| config.use_sharp_yuv = 1; |
| } else if (!strcmp(argv[c], "-pass") && c + 1 < argc) { |
| config.pass = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-qrange") && c + 2 < argc) { |
| config.qmin = ExUtilGetInt(argv[++c], 0, &parse_error); |
| config.qmax = ExUtilGetInt(argv[++c], 0, &parse_error); |
| if (config.qmin < 0) config.qmin = 0; |
| if (config.qmax > 100) config.qmax = 100; |
| } else if (!strcmp(argv[c], "-pre") && c + 1 < argc) { |
| config.preprocessing = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-segments") && c + 1 < argc) { |
| config.segments = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-partition_limit") && c + 1 < argc) { |
| config.partition_limit = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-map") && c + 1 < argc) { |
| picture.extra_info_type = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-crop") && c + 4 < argc) { |
| crop = 1; |
| crop_x = ExUtilGetInt(argv[++c], 0, &parse_error); |
| crop_y = ExUtilGetInt(argv[++c], 0, &parse_error); |
| crop_w = ExUtilGetInt(argv[++c], 0, &parse_error); |
| crop_h = ExUtilGetInt(argv[++c], 0, &parse_error); |
| } else if (!strcmp(argv[c], "-resize") && c + 2 < argc) { |
| resize_w = ExUtilGetInt(argv[++c], 0, &parse_error); |
| resize_h = ExUtilGetInt(argv[++c], 0, &parse_error); |
| #ifndef WEBP_DLL |
| } else if (!strcmp(argv[c], "-noasm")) { |
| VP8GetCPUInfo = NULL; |
| #endif |
| } else if (!strcmp(argv[c], "-version")) { |
| const int version = WebPGetEncoderVersion(); |
| const int sharpyuv_version = SharpYuvGetVersion(); |
| printf("%d.%d.%d\n", |
| (version >> 16) & 0xff, (version >> 8) & 0xff, version & 0xff); |
| printf("libsharpyuv: %d.%d.%d\n", |
| (sharpyuv_version >> 24) & 0xff, (sharpyuv_version >> 16) & 0xffff, |
| sharpyuv_version & 0xff); |
| FREE_WARGV_AND_RETURN(0); |
| } else if (!strcmp(argv[c], "-progress")) { |
| show_progress = 1; |
| } else if (!strcmp(argv[c], "-quiet")) { |
| quiet = 1; |
| } else if (!strcmp(argv[c], "-preset") && c + 1 < argc) { |
| WebPPreset preset; |
| ++c; |
| if (!strcmp(argv[c], "default")) { |
| preset = WEBP_PRESET_DEFAULT; |
| } else if (!strcmp(argv[c], "photo")) { |
| preset = WEBP_PRESET_PHOTO; |
| } else if (!strcmp(argv[c], "picture")) { |
| preset = WEBP_PRESET_PICTURE; |
| } else if (!strcmp(argv[c], "drawing")) { |
| preset = WEBP_PRESET_DRAWING; |
| } else if (!strcmp(argv[c], "icon")) { |
| preset = WEBP_PRESET_ICON; |
| } else if (!strcmp(argv[c], "text")) { |
| preset = WEBP_PRESET_TEXT; |
| } else { |
| fprintf(stderr, "Error! Unrecognized preset: %s\n", argv[c]); |
| goto Error; |
| } |
| if (!WebPConfigPreset(&config, preset, config.quality)) { |
| fprintf(stderr, "Error! Could initialize configuration with preset.\n"); |
| goto Error; |
| } |
| } else if (!strcmp(argv[c], "-metadata") && c + 1 < argc) { |
| static const struct { |
| const char* option; |
| int flag; |
| } kTokens[] = { |
| { "all", METADATA_ALL }, |
| { "none", 0 }, |
| { "exif", METADATA_EXIF }, |
| { "icc", METADATA_ICC }, |
| { "xmp", METADATA_XMP }, |
| }; |
| const size_t kNumTokens = sizeof(kTokens) / sizeof(kTokens[0]); |
| const char* start = argv[++c]; |
| const char* const end = start + strlen(start); |
| |
| while (start < end) { |
| size_t i; |
| const char* token = strchr(start, ','); |
| if (token == NULL) token = end; |
| |
| for (i = 0; i < kNumTokens; ++i) { |
| if ((size_t)(token - start) == strlen(kTokens[i].option) && |
| !strncmp(start, kTokens[i].option, strlen(kTokens[i].option))) { |
| if (kTokens[i].flag != 0) { |
| keep_metadata |= kTokens[i].flag; |
| } else { |
| keep_metadata = 0; |
| } |
| break; |
| } |
| } |
| if (i == kNumTokens) { |
| fprintf(stderr, "Error! Unknown metadata type '%.*s'\n", |
| (int)(token - start), start); |
| FREE_WARGV_AND_RETURN(-1); |
| } |
| start = token + 1; |
| } |
| #ifdef HAVE_WINCODEC_H |
| if (keep_metadata != 0 && keep_metadata != METADATA_ICC) { |
| // TODO(jzern): remove when -metadata is supported on all platforms. |
| fprintf(stderr, "Warning: only ICC profile extraction is currently" |
| " supported on this platform!\n"); |
| } |
| #endif |
| } else if (!strcmp(argv[c], "-v")) { |
| verbose = 1; |
| } else if (!strcmp(argv[c], "--")) { |
| if (c + 1 < argc) in_file = (const char*)GET_WARGV(argv, ++c); |
| break; |
| } else if (argv[c][0] == '-') { |
| fprintf(stderr, "Error! Unknown option '%s'\n", argv[c]); |
| HelpLong(); |
| FREE_WARGV_AND_RETURN(-1); |
| } else { |
| in_file = (const char*)GET_WARGV(argv, c); |
| } |
| |
| if (parse_error) { |
| HelpLong(); |
| FREE_WARGV_AND_RETURN(-1); |
| } |
| } |
| if (in_file == NULL) { |
| fprintf(stderr, "No input file specified!\n"); |
| HelpShort(); |
| goto Error; |
| } |
| |
| if (use_lossless_preset == 1) { |
| if (!WebPConfigLosslessPreset(&config, lossless_preset)) { |
| fprintf(stderr, "Invalid lossless preset (-z %d)\n", lossless_preset); |
| goto Error; |
| } |
| } |
| |
| // Check for unsupported command line options for lossless mode and log |
| // warning for such options. |
| if (!quiet && config.lossless == 1) { |
| if (config.target_size > 0 || config.target_PSNR > 0) { |
| fprintf(stderr, "Encoding for specified size or PSNR is not supported" |
| " for lossless encoding. Ignoring such option(s)!\n"); |
| } |
| if (config.partition_limit > 0) { |
| fprintf(stderr, "Partition limit option is not required for lossless" |
| " encoding. Ignoring this option!\n"); |
| } |
| } |
| // If a target size or PSNR was given, but somehow the -pass option was |
| // omitted, force a reasonable value. |
| if (config.target_size > 0 || config.target_PSNR > 0) { |
| if (config.pass == 1) config.pass = 6; |
| } |
| |
| if (!WebPValidateConfig(&config)) { |
| fprintf(stderr, "Error! Invalid configuration.\n"); |
| goto Error; |
| } |
| |
| // Read the input. We need to decide if we prefer ARGB or YUVA |
| // samples, depending on the expected compression mode (this saves |
| // some conversion steps). |
| picture.use_argb = (config.lossless || config.use_sharp_yuv || |
| config.preprocessing > 0 || |
| crop || (resize_w | resize_h) > 0); |
| if (verbose) { |
| StopwatchReset(&stop_watch); |
| } |
| if (!ReadPicture(in_file, &picture, keep_alpha, |
| (keep_metadata == 0) ? NULL : &metadata)) { |
| WFPRINTF(stderr, "Error! Cannot read input picture file '%s'\n", |
| (const W_CHAR*)in_file); |
| goto Error; |
| } |
| picture.progress_hook = (show_progress && !quiet) ? ProgressReport : NULL; |
| |
| if (blend_alpha) { |
| WebPBlendAlpha(&picture, background_color); |
| } |
| |
| if (verbose) { |
| const double read_time = StopwatchReadAndReset(&stop_watch); |
| fprintf(stderr, "Time to read input: %.3fs\n", read_time); |
| } |
| // The bitstream should be kept in memory when metadata must be appended |
| // before writing it to a file/stream, and/or when the near-losslessly encoded |
| // bitstream must be decoded for distortion computation (lossy will modify the |
| // 'picture' but not the lossless pipeline). |
| // Otherwise directly write the bitstream to a file. |
| use_memory_writer = (out_file != NULL && keep_metadata) || |
| (!quiet && print_distortion >= 0 && config.lossless && |
| config.near_lossless < 100); |
| |
| // Open the output |
| if (out_file != NULL) { |
| const int use_stdout = !WSTRCMP(out_file, "-"); |
| out = use_stdout ? ImgIoUtilSetBinaryMode(stdout) : WFOPEN(out_file, "wb"); |
| if (out == NULL) { |
| WFPRINTF(stderr, "Error! Cannot open output file '%s'\n", |
| (const W_CHAR*)out_file); |
| goto Error; |
| } else { |
| if (!short_output && !quiet) { |
| WFPRINTF(stderr, "Saving file '%s'\n", (const W_CHAR*)out_file); |
| } |
| } |
| if (use_memory_writer) { |
| picture.writer = WebPMemoryWrite; |
| picture.custom_ptr = (void*)&memory_writer; |
| } else { |
| picture.writer = MyWriter; |
| picture.custom_ptr = (void*)out; |
| } |
| } else { |
| out = NULL; |
| if (use_memory_writer) { |
| picture.writer = WebPMemoryWrite; |
| picture.custom_ptr = (void*)&memory_writer; |
| } |
| if (!quiet && !short_output) { |
| fprintf(stderr, "No output file specified (no -o flag). Encoding will\n"); |
| fprintf(stderr, "be performed, but its results discarded.\n\n"); |
| } |
| } |
| if (!quiet) { |
| picture.stats = &stats; |
| picture.user_data = (void*)in_file; |
| } |
| |
| // Crop & resize. |
| if (verbose) { |
| StopwatchReset(&stop_watch); |
| } |
| if (crop != 0) { |
| // We use self-cropping using a view. |
| if (!WebPPictureView(&picture, crop_x, crop_y, crop_w, crop_h, &picture)) { |
| fprintf(stderr, "Error! Cannot crop picture\n"); |
| goto Error; |
| } |
| } |
| if ((resize_w | resize_h) > 0) { |
| WebPPicture picture_no_alpha; |
| if (config.exact) { |
| // If -exact, we can't premultiply RGB by A otherwise RGB is lost if A=0. |
| // We rescale an opaque copy and assemble scaled A and non-premultiplied |
| // RGB channels. This is slower but it's a very uncommon use case. Color |
| // leak at sharp alpha edges is possible. |
| if (!WebPPictureCopy(&picture, &picture_no_alpha)) { |
| fprintf(stderr, "Error! Cannot copy temporary picture\n"); |
| goto Error; |
| } |
| |
| // We enforced picture.use_argb = 1 above. Now, remove the alpha values. |
| { |
| int x, y; |
| uint32_t* argb_no_alpha = picture_no_alpha.argb; |
| for (y = 0; y < picture_no_alpha.height; ++y) { |
| for (x = 0; x < picture_no_alpha.width; ++x) { |
| argb_no_alpha[x] |= 0xff000000; // Opaque copy. |
| } |
| argb_no_alpha += picture_no_alpha.argb_stride; |
| } |
| } |
| |
| if (!WebPPictureRescale(&picture_no_alpha, resize_w, resize_h)) { |
| fprintf(stderr, "Error! Cannot resize temporary picture\n"); |
| goto Error; |
| } |
| } |
| |
| if (!WebPPictureRescale(&picture, resize_w, resize_h)) { |
| fprintf(stderr, "Error! Cannot resize picture\n"); |
| goto Error; |
| } |
| |
| if (config.exact) { // Put back the alpha information. |
| int x, y; |
| uint32_t* argb_no_alpha = picture_no_alpha.argb; |
| uint32_t* argb = picture.argb; |
| for (y = 0; y < picture_no_alpha.height; ++y) { |
| for (x = 0; x < picture_no_alpha.width; ++x) { |
| argb[x] = (argb[x] & 0xff000000) | (argb_no_alpha[x] & 0x00ffffff); |
| } |
| argb_no_alpha += picture_no_alpha.argb_stride; |
| argb += picture.argb_stride; |
| } |
| WebPPictureFree(&picture_no_alpha); |
| } |
| } |
| if (verbose && (crop != 0 || (resize_w | resize_h) > 0)) { |
| const double preproc_time = StopwatchReadAndReset(&stop_watch); |
| fprintf(stderr, "Time to crop/resize picture: %.3fs\n", preproc_time); |
| } |
| |
| if (picture.extra_info_type > 0) { |
| AllocExtraInfo(&picture); |
| } |
| // Save original picture for later comparison. Only for lossy as lossless does |
| // not modify 'picture' (even near-lossless). |
| if (print_distortion >= 0 && !config.lossless && |
| !WebPPictureCopy(&picture, &original_picture)) { |
| fprintf(stderr, "Error! Cannot copy temporary picture\n"); |
| goto Error; |
| } |
| |
| // Compress. |
| if (verbose) { |
| StopwatchReset(&stop_watch); |
| } |
| if (!WebPEncode(&config, &picture)) { |
| fprintf(stderr, "Error! Cannot encode picture as WebP\n"); |
| fprintf(stderr, "Error code: %d (%s)\n", |
| picture.error_code, kErrorMessages[picture.error_code]); |
| goto Error; |
| } |
| if (verbose) { |
| const double encode_time = StopwatchReadAndReset(&stop_watch); |
| fprintf(stderr, "Time to encode picture: %.3fs\n", encode_time); |
| } |
| |
| // Get the decompressed image for the lossless pipeline. |
| if (!quiet && print_distortion >= 0 && config.lossless) { |
| if (config.near_lossless == 100) { |
| // Pure lossless: image was not modified, make 'original_picture' a view |
| // of 'picture' by copying all members except the freeable pointers. |
| original_picture = picture; |
| original_picture.memory_ = original_picture.memory_argb_ = NULL; |
| } else { |
| // Decode the bitstream stored in 'memory_writer' to get the altered image |
| // to 'picture'; save the 'original_picture' beforehand. |
| assert(use_memory_writer); |
| original_picture = picture; |
| if (!WebPPictureInit(&picture)) { // Do not free 'picture'. |
| fprintf(stderr, "Error! Version mismatch!\n"); |
| goto Error; |
| } |
| |
| picture.use_argb = 1; |
| if (!ReadWebP( |
| memory_writer.mem, memory_writer.size, &picture, |
| /*keep_alpha=*/WebPPictureHasTransparency(&original_picture), |
| /*metadata=*/NULL)) { |
| fprintf(stderr, "Error! Cannot decode encoded WebP bitstream\n"); |
| fprintf(stderr, "Error code: %d (%s)\n", picture.error_code, |
| kErrorMessages[picture.error_code]); |
| goto Error; |
| } |
| picture.stats = original_picture.stats; |
| } |
| original_picture.stats = NULL; |
| } |
| |
| // Write the YUV planes to a PGM file. Only available for lossy. |
| if (dump_file) { |
| if (picture.use_argb) { |
| fprintf(stderr, "Warning: can't dump file (-d option) " |
| "in lossless mode.\n"); |
| } else if (!DumpPicture(&picture, dump_file)) { |
| WFPRINTF(stderr, "Warning, couldn't dump picture %s\n", |
| (const W_CHAR*)dump_file); |
| } |
| } |
| |
| if (use_memory_writer && out != NULL && |
| !WriteWebPWithMetadata(out, &picture, &memory_writer, &metadata, |
| keep_metadata, &metadata_written)) { |
| fprintf(stderr, "Error writing WebP file!\n"); |
| goto Error; |
| } |
| |
| if (out == NULL && keep_metadata) { |
| // output is disabled, just display the metadata stats. |
| const struct { |
| const MetadataPayload* const payload; |
| int flag; |
| } *iter, info[] = {{&metadata.exif, METADATA_EXIF}, |
| {&metadata.iccp, METADATA_ICC}, |
| {&metadata.xmp, METADATA_XMP}, |
| {NULL, 0}}; |
| uint32_t unused1 = 0; |
| uint64_t unused2 = 0; |
| |
| for (iter = info; iter->payload != NULL; ++iter) { |
| if (UpdateFlagsAndSize(iter->payload, !!(keep_metadata & iter->flag), |
| /*flag=*/0, &unused1, &unused2)) { |
| metadata_written |= iter->flag; |
| } |
| } |
| } |
| |
| if (!quiet) { |
| if (!short_output || print_distortion < 0) { |
| if (config.lossless) { |
| PrintExtraInfoLossless(&picture, short_output, in_file); |
| } else { |
| PrintExtraInfoLossy(&picture, short_output, config.low_memory, in_file); |
| } |
| } |
| if (!short_output && picture.extra_info_type > 0) { |
| PrintMapInfo(&picture); |
| } |
| if (print_distortion >= 0) { // print distortion |
| static const char* distortion_names[] = { "PSNR", "SSIM", "LSIM" }; |
| float values[5]; |
| if (!WebPPictureDistortion(&picture, &original_picture, |
| print_distortion, values)) { |
| fprintf(stderr, "Error while computing the distortion.\n"); |
| goto Error; |
| } |
| if (!short_output) { |
| fprintf(stderr, "%s: ", distortion_names[print_distortion]); |
| fprintf(stderr, "B:%.2f G:%.2f R:%.2f A:%.2f Total:%.2f\n", |
| values[0], values[1], values[2], values[3], values[4]); |
| } else { |
| fprintf(stderr, "%7d %.4f\n", picture.stats->coded_size, values[4]); |
| } |
| } |
| if (!short_output) { |
| PrintMetadataInfo(&metadata, metadata_written); |
| } |
| } |
| return_value = 0; |
| |
| Error: |
| WebPMemoryWriterClear(&memory_writer); |
| WebPFree(picture.extra_info); |
| MetadataFree(&metadata); |
| WebPPictureFree(&picture); |
| WebPPictureFree(&original_picture); |
| if (out != NULL && out != stdout) { |
| fclose(out); |
| } |
| |
| FREE_WARGV_AND_RETURN(return_value); |
| } |
| |
| //------------------------------------------------------------------------------ |