Blame imageio/image_enc.c

Packit 9c6abc
// Copyright 2016 Google Inc. All Rights Reserved.
Packit 9c6abc
//
Packit 9c6abc
// Use of this source code is governed by a BSD-style license
Packit 9c6abc
// that can be found in the COPYING file in the root of the source
Packit 9c6abc
// tree. An additional intellectual property rights grant can be found
Packit 9c6abc
// in the file PATENTS. All contributing project authors may
Packit 9c6abc
// be found in the AUTHORS file in the root of the source tree.
Packit 9c6abc
// -----------------------------------------------------------------------------
Packit 9c6abc
//
Packit 9c6abc
// Save image
Packit 9c6abc
Packit 9c6abc
#include "./image_enc.h"
Packit 9c6abc
Packit 9c6abc
#include <assert.h>
Packit 9c6abc
#include <string.h>
Packit 9c6abc
Packit 9c6abc
#ifdef WEBP_HAVE_PNG
Packit 9c6abc
#include <png.h>
Packit 9c6abc
#include <setjmp.h>   // note: this must be included *after* png.h
Packit 9c6abc
#endif
Packit 9c6abc
Packit 9c6abc
#ifdef HAVE_WINCODEC_H
Packit 9c6abc
#ifdef __MINGW32__
Packit 9c6abc
#define INITGUID  // Without this GUIDs are declared extern and fail to link
Packit 9c6abc
#endif
Packit 9c6abc
#define CINTERFACE
Packit 9c6abc
#define COBJMACROS
Packit 9c6abc
#define _WIN32_IE 0x500  // Workaround bug in shlwapi.h when compiling C++
Packit 9c6abc
                         // code with COBJMACROS.
Packit 9c6abc
#include <ole2.h>  // CreateStreamOnHGlobal()
Packit 9c6abc
#include <shlwapi.h>
Packit 9c6abc
#include <windows.h>
Packit 9c6abc
#include <wincodec.h>
Packit 9c6abc
#endif
Packit 9c6abc
Packit 9c6abc
#include "./imageio_util.h"
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// PNG
Packit 9c6abc
Packit 9c6abc
#ifdef HAVE_WINCODEC_H
Packit 9c6abc
Packit 9c6abc
#define IFS(fn)                                                     \
Packit 9c6abc
  do {                                                              \
Packit 9c6abc
    if (SUCCEEDED(hr)) {                                            \
Packit 9c6abc
      hr = (fn);                                                    \
Packit 9c6abc
      if (FAILED(hr)) fprintf(stderr, #fn " failed %08lx\n", hr);   \
Packit 9c6abc
    }                                                               \
Packit 9c6abc
  } while (0)
Packit 9c6abc
Packit 9c6abc
#ifdef __cplusplus
Packit 9c6abc
#define MAKE_REFGUID(x) (x)
Packit 9c6abc
#else
Packit 9c6abc
#define MAKE_REFGUID(x) &(x)
Packit 9c6abc
#endif
Packit 9c6abc
Packit 9c6abc
static HRESULT CreateOutputStream(const char* out_file_name,
Packit 9c6abc
                                  int write_to_mem, IStream** stream) {
Packit 9c6abc
  HRESULT hr = S_OK;
Packit 9c6abc
  if (write_to_mem) {
Packit 9c6abc
    // Output to a memory buffer. This is freed when 'stream' is released.
Packit 9c6abc
    IFS(CreateStreamOnHGlobal(NULL, TRUE, stream));
Packit 9c6abc
  } else {
Packit 9c6abc
    IFS(SHCreateStreamOnFileA(out_file_name, STGM_WRITE | STGM_CREATE, stream));
Packit 9c6abc
  }
Packit 9c6abc
  if (FAILED(hr)) {
Packit 9c6abc
    fprintf(stderr, "Error opening output file %s (%08lx)\n",
Packit 9c6abc
            out_file_name, hr);
Packit 9c6abc
  }
Packit 9c6abc
  return hr;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
static HRESULT WriteUsingWIC(const char* out_file_name, int use_stdout,
Packit 9c6abc
                             REFGUID container_guid,
Packit 9c6abc
                             uint8_t* rgb, int stride,
Packit 9c6abc
                             uint32_t width, uint32_t height, int has_alpha) {
Packit 9c6abc
  HRESULT hr = S_OK;
Packit 9c6abc
  IWICImagingFactory* factory = NULL;
Packit 9c6abc
  IWICBitmapFrameEncode* frame = NULL;
Packit 9c6abc
  IWICBitmapEncoder* encoder = NULL;
Packit 9c6abc
  IStream* stream = NULL;
Packit 9c6abc
  WICPixelFormatGUID pixel_format = has_alpha ? GUID_WICPixelFormat32bppBGRA
Packit 9c6abc
                                              : GUID_WICPixelFormat24bppBGR;
Packit 9c6abc
Packit 9c6abc
  if (out_file_name == NULL || rgb == NULL) return E_INVALIDARG;
Packit 9c6abc
Packit 9c6abc
  IFS(CoInitialize(NULL));
Packit 9c6abc
  IFS(CoCreateInstance(MAKE_REFGUID(CLSID_WICImagingFactory), NULL,
Packit 9c6abc
                       CLSCTX_INPROC_SERVER,
Packit 9c6abc
                       MAKE_REFGUID(IID_IWICImagingFactory),
Packit 9c6abc
                       (LPVOID*)&factory));
Packit 9c6abc
  if (hr == REGDB_E_CLASSNOTREG) {
Packit 9c6abc
    fprintf(stderr,
Packit 9c6abc
            "Couldn't access Windows Imaging Component (are you running "
Packit 9c6abc
            "Windows XP SP3 or newer?). PNG support not available. "
Packit 9c6abc
            "Use -ppm or -pgm for available PPM and PGM formats.\n");
Packit 9c6abc
  }
Packit 9c6abc
  IFS(CreateOutputStream(out_file_name, use_stdout, &stream));
Packit 9c6abc
  IFS(IWICImagingFactory_CreateEncoder(factory, container_guid, NULL,
Packit 9c6abc
                                       &encoder));
Packit 9c6abc
  IFS(IWICBitmapEncoder_Initialize(encoder, stream,
Packit 9c6abc
                                   WICBitmapEncoderNoCache));
Packit 9c6abc
  IFS(IWICBitmapEncoder_CreateNewFrame(encoder, &frame, NULL));
Packit 9c6abc
  IFS(IWICBitmapFrameEncode_Initialize(frame, NULL));
Packit 9c6abc
  IFS(IWICBitmapFrameEncode_SetSize(frame, width, height));
Packit 9c6abc
  IFS(IWICBitmapFrameEncode_SetPixelFormat(frame, &pixel_format));
Packit 9c6abc
  IFS(IWICBitmapFrameEncode_WritePixels(frame, height, stride,
Packit 9c6abc
                                        height * stride, rgb));
Packit 9c6abc
  IFS(IWICBitmapFrameEncode_Commit(frame));
Packit 9c6abc
  IFS(IWICBitmapEncoder_Commit(encoder));
Packit 9c6abc
Packit 9c6abc
  if (SUCCEEDED(hr) && use_stdout) {
Packit 9c6abc
    HGLOBAL image;
Packit 9c6abc
    IFS(GetHGlobalFromStream(stream, &image);;
Packit 9c6abc
    if (SUCCEEDED(hr)) {
Packit 9c6abc
      HANDLE std_output = GetStdHandle(STD_OUTPUT_HANDLE);
Packit 9c6abc
      DWORD mode;
Packit 9c6abc
      const BOOL update_mode = GetConsoleMode(std_output, &mode);
Packit 9c6abc
      const void* const image_mem = GlobalLock(image);
Packit 9c6abc
      DWORD bytes_written = 0;
Packit 9c6abc
Packit 9c6abc
      // Clear output processing if necessary, then output the image.
Packit 9c6abc
      if (update_mode) SetConsoleMode(std_output, 0);
Packit 9c6abc
      if (!WriteFile(std_output, image_mem, (DWORD)GlobalSize(image),
Packit 9c6abc
                     &bytes_written, NULL) ||
Packit 9c6abc
          bytes_written != GlobalSize(image)) {
Packit 9c6abc
        hr = E_FAIL;
Packit 9c6abc
      }
Packit 9c6abc
      if (update_mode) SetConsoleMode(std_output, mode);
Packit 9c6abc
      GlobalUnlock(image);
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
Packit 9c6abc
  if (frame != NULL) IUnknown_Release(frame);
Packit 9c6abc
  if (encoder != NULL) IUnknown_Release(encoder);
Packit 9c6abc
  if (factory != NULL) IUnknown_Release(factory);
Packit 9c6abc
  if (stream != NULL) IUnknown_Release(stream);
Packit 9c6abc
  return hr;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
int WebPWritePNG(const char* out_file_name, int use_stdout,
Packit 9c6abc
                 const WebPDecBuffer* const buffer) {
Packit 9c6abc
  const uint32_t width = buffer->width;
Packit 9c6abc
  const uint32_t height = buffer->height;
Packit 9c6abc
  uint8_t* const rgb = buffer->u.RGBA.rgba;
Packit 9c6abc
  const int stride = buffer->u.RGBA.stride;
Packit 9c6abc
  const int has_alpha = WebPIsAlphaMode(buffer->colorspace);
Packit 9c6abc
Packit 9c6abc
  return SUCCEEDED(WriteUsingWIC(out_file_name, use_stdout,
Packit 9c6abc
                                 MAKE_REFGUID(GUID_ContainerFormatPng),
Packit 9c6abc
                                 rgb, stride, width, height, has_alpha));
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
#elif defined(WEBP_HAVE_PNG)    // !HAVE_WINCODEC_H
Packit 9c6abc
static void PNGAPI PNGErrorFunction(png_structp png, png_const_charp dummy) {
Packit 9c6abc
  (void)dummy;  // remove variable-unused warning
Packit 9c6abc
  longjmp(png_jmpbuf(png), 1);
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
int WebPWritePNG(FILE* out_file, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  volatile png_structp png;
Packit 9c6abc
  volatile png_infop info;
Packit 9c6abc
Packit 9c6abc
  if (out_file == NULL || buffer == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
  png = png_create_write_struct(PNG_LIBPNG_VER_STRING,
Packit 9c6abc
                                NULL, PNGErrorFunction, NULL);
Packit 9c6abc
  if (png == NULL) {
Packit 9c6abc
    return 0;
Packit 9c6abc
  }
Packit 9c6abc
  info = png_create_info_struct(png);
Packit 9c6abc
  if (info == NULL) {
Packit 9c6abc
    png_destroy_write_struct((png_structpp)&png, NULL);
Packit 9c6abc
    return 0;
Packit 9c6abc
  }
Packit 9c6abc
  if (setjmp(png_jmpbuf(png))) {
Packit 9c6abc
    png_destroy_write_struct((png_structpp)&png, (png_infopp)&info;;
Packit 9c6abc
    return 0;
Packit 9c6abc
  }
Packit 9c6abc
  png_init_io(png, out_file);
Packit 9c6abc
  {
Packit 9c6abc
    const uint32_t width = buffer->width;
Packit 9c6abc
    const uint32_t height = buffer->height;
Packit 9c6abc
    png_bytep row = buffer->u.RGBA.rgba;
Packit 9c6abc
    const int stride = buffer->u.RGBA.stride;
Packit 9c6abc
    const int has_alpha = WebPIsAlphaMode(buffer->colorspace);
Packit 9c6abc
    uint32_t y;
Packit 9c6abc
Packit 9c6abc
    png_set_IHDR(png, info, width, height, 8,
Packit 9c6abc
                 has_alpha ? PNG_COLOR_TYPE_RGBA : PNG_COLOR_TYPE_RGB,
Packit 9c6abc
                 PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT,
Packit 9c6abc
                 PNG_FILTER_TYPE_DEFAULT);
Packit 9c6abc
    png_write_info(png, info);
Packit 9c6abc
    for (y = 0; y < height; ++y) {
Packit 9c6abc
      png_write_rows(png, &row, 1);
Packit 9c6abc
      row += stride;
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
  png_write_end(png, info);
Packit 9c6abc
  png_destroy_write_struct((png_structpp)&png, (png_infopp)&info;;
Packit 9c6abc
  return 1;
Packit 9c6abc
}
Packit 9c6abc
#else    // !HAVE_WINCODEC_H && !WEBP_HAVE_PNG
Packit 9c6abc
int WebPWritePNG(FILE* fout, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  if (fout == NULL || buffer == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
  fprintf(stderr, "PNG support not compiled. Please install the libpng "
Packit 9c6abc
          "development package before building.\n");
Packit 9c6abc
  fprintf(stderr, "You can run with -ppm flag to decode in PPM format.\n");
Packit 9c6abc
  return 0;
Packit 9c6abc
}
Packit 9c6abc
#endif
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// PPM / PAM
Packit 9c6abc
Packit 9c6abc
static int WritePPMPAM(FILE* fout, const WebPDecBuffer* const buffer,
Packit 9c6abc
                       int alpha) {
Packit 9c6abc
  if (fout == NULL || buffer == NULL) {
Packit 9c6abc
    return 0;
Packit 9c6abc
  } else {
Packit 9c6abc
    const uint32_t width = buffer->width;
Packit 9c6abc
    const uint32_t height = buffer->height;
Packit 9c6abc
    const uint8_t* row = buffer->u.RGBA.rgba;
Packit 9c6abc
    const int stride = buffer->u.RGBA.stride;
Packit 9c6abc
    const size_t bytes_per_px = alpha ? 4 : 3;
Packit 9c6abc
    uint32_t y;
Packit 9c6abc
Packit 9c6abc
    if (row == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
    if (alpha) {
Packit 9c6abc
      fprintf(fout, "P7\nWIDTH %u\nHEIGHT %u\nDEPTH 4\nMAXVAL 255\n"
Packit 9c6abc
                    "TUPLTYPE RGB_ALPHA\nENDHDR\n", width, height);
Packit 9c6abc
    } else {
Packit 9c6abc
      fprintf(fout, "P6\n%u %u\n255\n", width, height);
Packit 9c6abc
    }
Packit 9c6abc
    for (y = 0; y < height; ++y) {
Packit 9c6abc
      if (fwrite(row, width, bytes_per_px, fout) != bytes_per_px) {
Packit 9c6abc
        return 0;
Packit 9c6abc
      }
Packit 9c6abc
      row += stride;
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
  return 1;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
int WebPWritePPM(FILE* fout, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  return WritePPMPAM(fout, buffer, 0);
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
int WebPWritePAM(FILE* fout, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  return WritePPMPAM(fout, buffer, 1);
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// Raw PGM
Packit 9c6abc
Packit 9c6abc
// Save 16b mode (RGBA4444, RGB565, ...) for debugging purpose.
Packit 9c6abc
int WebPWrite16bAsPGM(FILE* fout, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  const uint32_t width = buffer->width;
Packit 9c6abc
  const uint32_t height = buffer->height;
Packit 9c6abc
  const uint8_t* rgba = buffer->u.RGBA.rgba;
Packit 9c6abc
  const int stride = buffer->u.RGBA.stride;
Packit 9c6abc
  const uint32_t bytes_per_px = 2;
Packit 9c6abc
  uint32_t y;
Packit 9c6abc
Packit 9c6abc
  if (fout == NULL || buffer == NULL || rgba == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
  fprintf(fout, "P5\n%u %u\n255\n", width * bytes_per_px, height);
Packit 9c6abc
  for (y = 0; y < height; ++y) {
Packit 9c6abc
    if (fwrite(rgba, width, bytes_per_px, fout) != bytes_per_px) {
Packit 9c6abc
      return 0;
Packit 9c6abc
    }
Packit 9c6abc
    rgba += stride;
Packit 9c6abc
  }
Packit 9c6abc
  return 1;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// BMP
Packit 9c6abc
Packit 9c6abc
static void PutLE16(uint8_t* const dst, uint32_t value) {
Packit 9c6abc
  dst[0] = (value >> 0) & 0xff;
Packit 9c6abc
  dst[1] = (value >> 8) & 0xff;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
static void PutLE32(uint8_t* const dst, uint32_t value) {
Packit 9c6abc
  PutLE16(dst + 0, (value >>  0) & 0xffff);
Packit 9c6abc
  PutLE16(dst + 2, (value >> 16) & 0xffff);
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
#define BMP_HEADER_SIZE 54
Packit 9c6abc
int WebPWriteBMP(FILE* fout, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  const int has_alpha = WebPIsAlphaMode(buffer->colorspace);
Packit 9c6abc
  const uint32_t width = buffer->width;
Packit 9c6abc
  const uint32_t height = buffer->height;
Packit 9c6abc
  const uint8_t* rgba = buffer->u.RGBA.rgba;
Packit 9c6abc
  const int stride = buffer->u.RGBA.stride;
Packit 9c6abc
  const uint32_t bytes_per_px = has_alpha ? 4 : 3;
Packit 9c6abc
  uint32_t y;
Packit 9c6abc
  const uint32_t line_size = bytes_per_px * width;
Packit 9c6abc
  const uint32_t bmp_stride = (line_size + 3) & ~3;   // pad to 4
Packit 9c6abc
  const uint32_t total_size = bmp_stride * height + BMP_HEADER_SIZE;
Packit 9c6abc
  uint8_t bmp_header[BMP_HEADER_SIZE] = { 0 };
Packit 9c6abc
Packit 9c6abc
  if (fout == NULL || buffer == NULL || rgba == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
  // bitmap file header
Packit 9c6abc
  PutLE16(bmp_header + 0, 0x4d42);                // signature 'BM'
Packit 9c6abc
  PutLE32(bmp_header + 2, total_size);            // size including header
Packit 9c6abc
  PutLE32(bmp_header + 6, 0);                     // reserved
Packit 9c6abc
  PutLE32(bmp_header + 10, BMP_HEADER_SIZE);      // offset to pixel array
Packit 9c6abc
  // bitmap info header
Packit 9c6abc
  PutLE32(bmp_header + 14, 40);                   // DIB header size
Packit 9c6abc
  PutLE32(bmp_header + 18, width);                // dimensions
Packit 9c6abc
  PutLE32(bmp_header + 22, -(int)height);         // vertical flip!
Packit 9c6abc
  PutLE16(bmp_header + 26, 1);                    // number of planes
Packit 9c6abc
  PutLE16(bmp_header + 28, bytes_per_px * 8);     // bits per pixel
Packit 9c6abc
  PutLE32(bmp_header + 30, 0);                    // no compression (BI_RGB)
Packit 9c6abc
  PutLE32(bmp_header + 34, 0);                    // image size (dummy)
Packit 9c6abc
  PutLE32(bmp_header + 38, 2400);                 // x pixels/meter
Packit 9c6abc
  PutLE32(bmp_header + 42, 2400);                 // y pixels/meter
Packit 9c6abc
  PutLE32(bmp_header + 46, 0);                    // number of palette colors
Packit 9c6abc
  PutLE32(bmp_header + 50, 0);                    // important color count
Packit 9c6abc
Packit 9c6abc
  // TODO(skal): color profile
Packit 9c6abc
Packit 9c6abc
  // write header
Packit 9c6abc
  if (fwrite(bmp_header, sizeof(bmp_header), 1, fout) != 1) {
Packit 9c6abc
    return 0;
Packit 9c6abc
  }
Packit 9c6abc
Packit 9c6abc
  // write pixel array
Packit 9c6abc
  for (y = 0; y < height; ++y) {
Packit 9c6abc
    if (fwrite(rgba, line_size, 1, fout) != 1) {
Packit 9c6abc
      return 0;
Packit 9c6abc
    }
Packit 9c6abc
    // write padding zeroes
Packit 9c6abc
    if (bmp_stride != line_size) {
Packit 9c6abc
      const uint8_t zeroes[3] = { 0 };
Packit 9c6abc
      if (fwrite(zeroes, bmp_stride - line_size, 1, fout) != 1) {
Packit 9c6abc
        return 0;
Packit 9c6abc
      }
Packit 9c6abc
    }
Packit 9c6abc
    rgba += stride;
Packit 9c6abc
  }
Packit 9c6abc
  return 1;
Packit 9c6abc
}
Packit 9c6abc
#undef BMP_HEADER_SIZE
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// TIFF
Packit 9c6abc
Packit 9c6abc
#define NUM_IFD_ENTRIES 15
Packit 9c6abc
#define EXTRA_DATA_SIZE 16
Packit 9c6abc
// 10b for signature/header + n * 12b entries + 4b for IFD terminator:
Packit 9c6abc
#define EXTRA_DATA_OFFSET (10 + 12 * NUM_IFD_ENTRIES + 4)
Packit 9c6abc
#define TIFF_HEADER_SIZE (EXTRA_DATA_OFFSET + EXTRA_DATA_SIZE)
Packit 9c6abc
Packit 9c6abc
int WebPWriteTIFF(FILE* fout, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  const int has_alpha = WebPIsAlphaMode(buffer->colorspace);
Packit 9c6abc
  const uint32_t width = buffer->width;
Packit 9c6abc
  const uint32_t height = buffer->height;
Packit 9c6abc
  const uint8_t* rgba = buffer->u.RGBA.rgba;
Packit 9c6abc
  const int stride = buffer->u.RGBA.stride;
Packit 9c6abc
  const uint8_t bytes_per_px = has_alpha ? 4 : 3;
Packit 9c6abc
  const uint8_t assoc_alpha =
Packit 9c6abc
      WebPIsPremultipliedMode(buffer->colorspace) ? 1 : 2;
Packit 9c6abc
  // For non-alpha case, we omit tag 0x152 (ExtraSamples).
Packit 9c6abc
  const uint8_t num_ifd_entries = has_alpha ? NUM_IFD_ENTRIES
Packit 9c6abc
                                            : NUM_IFD_ENTRIES - 1;
Packit 9c6abc
  uint8_t tiff_header[TIFF_HEADER_SIZE] = {
Packit 9c6abc
    0x49, 0x49, 0x2a, 0x00,   // little endian signature
Packit 9c6abc
    8, 0, 0, 0,               // offset to the unique IFD that follows
Packit 9c6abc
    // IFD (offset = 8). Entries must be written in increasing tag order.
Packit 9c6abc
    num_ifd_entries, 0,       // Number of entries in the IFD (12 bytes each).
Packit 9c6abc
    0x00, 0x01, 3, 0, 1, 0, 0, 0, 0, 0, 0, 0,    //  10: Width  (TBD)
Packit 9c6abc
    0x01, 0x01, 3, 0, 1, 0, 0, 0, 0, 0, 0, 0,    //  22: Height (TBD)
Packit 9c6abc
    0x02, 0x01, 3, 0, bytes_per_px, 0, 0, 0,     //  34: BitsPerSample: 8888
Packit 9c6abc
        EXTRA_DATA_OFFSET + 0, 0, 0, 0,
Packit 9c6abc
    0x03, 0x01, 3, 0, 1, 0, 0, 0, 1, 0, 0, 0,    //  46: Compression: none
Packit 9c6abc
    0x06, 0x01, 3, 0, 1, 0, 0, 0, 2, 0, 0, 0,    //  58: Photometric: RGB
Packit 9c6abc
    0x11, 0x01, 4, 0, 1, 0, 0, 0,                //  70: Strips offset:
Packit 9c6abc
        TIFF_HEADER_SIZE, 0, 0, 0,               //      data follows header
Packit 9c6abc
    0x12, 0x01, 3, 0, 1, 0, 0, 0, 1, 0, 0, 0,    //  82: Orientation: topleft
Packit 9c6abc
    0x15, 0x01, 3, 0, 1, 0, 0, 0,                //  94: SamplesPerPixels
Packit 9c6abc
        bytes_per_px, 0, 0, 0,
Packit 9c6abc
    0x16, 0x01, 3, 0, 1, 0, 0, 0, 0, 0, 0, 0,    // 106: Rows per strip (TBD)
Packit 9c6abc
    0x17, 0x01, 4, 0, 1, 0, 0, 0, 0, 0, 0, 0,    // 118: StripByteCount (TBD)
Packit 9c6abc
    0x1a, 0x01, 5, 0, 1, 0, 0, 0,                // 130: X-resolution
Packit 9c6abc
        EXTRA_DATA_OFFSET + 8, 0, 0, 0,
Packit 9c6abc
    0x1b, 0x01, 5, 0, 1, 0, 0, 0,                // 142: Y-resolution
Packit 9c6abc
        EXTRA_DATA_OFFSET + 8, 0, 0, 0,
Packit 9c6abc
    0x1c, 0x01, 3, 0, 1, 0, 0, 0, 1, 0, 0, 0,    // 154: PlanarConfiguration
Packit 9c6abc
    0x28, 0x01, 3, 0, 1, 0, 0, 0, 2, 0, 0, 0,    // 166: ResolutionUnit (inch)
Packit 9c6abc
    0x52, 0x01, 3, 0, 1, 0, 0, 0,
Packit 9c6abc
        assoc_alpha, 0, 0, 0,                    // 178: ExtraSamples: rgbA/RGBA
Packit 9c6abc
    0, 0, 0, 0,                                  // 190: IFD terminator
Packit 9c6abc
    // EXTRA_DATA_OFFSET:
Packit 9c6abc
    8, 0, 8, 0, 8, 0, 8, 0,      // BitsPerSample
Packit 9c6abc
    72, 0, 0, 0, 1, 0, 0, 0      // 72 pixels/inch, for X/Y-resolution
Packit 9c6abc
  };
Packit 9c6abc
  uint32_t y;
Packit 9c6abc
Packit 9c6abc
  if (fout == NULL || buffer == NULL || rgba == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
  // Fill placeholders in IFD:
Packit 9c6abc
  PutLE32(tiff_header + 10 + 8, width);
Packit 9c6abc
  PutLE32(tiff_header + 22 + 8, height);
Packit 9c6abc
  PutLE32(tiff_header + 106 + 8, height);
Packit 9c6abc
  PutLE32(tiff_header + 118 + 8, width * bytes_per_px * height);
Packit 9c6abc
  if (!has_alpha) PutLE32(tiff_header + 178, 0);  // IFD terminator
Packit 9c6abc
Packit 9c6abc
  // write header
Packit 9c6abc
  if (fwrite(tiff_header, sizeof(tiff_header), 1, fout) != 1) {
Packit 9c6abc
    return 0;
Packit 9c6abc
  }
Packit 9c6abc
  // write pixel values
Packit 9c6abc
  for (y = 0; y < height; ++y) {
Packit 9c6abc
    if (fwrite(rgba, bytes_per_px, width, fout) != width) {
Packit 9c6abc
      return 0;
Packit 9c6abc
    }
Packit 9c6abc
    rgba += stride;
Packit 9c6abc
  }
Packit 9c6abc
Packit 9c6abc
  return 1;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
#undef TIFF_HEADER_SIZE
Packit 9c6abc
#undef EXTRA_DATA_OFFSET
Packit 9c6abc
#undef EXTRA_DATA_SIZE
Packit 9c6abc
#undef NUM_IFD_ENTRIES
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// Raw Alpha
Packit 9c6abc
Packit 9c6abc
int WebPWriteAlphaPlane(FILE* fout, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  if (fout == NULL || buffer == NULL) {
Packit 9c6abc
    return 0;
Packit 9c6abc
  } else {
Packit 9c6abc
    const uint32_t width = buffer->width;
Packit 9c6abc
    const uint32_t height = buffer->height;
Packit 9c6abc
    const uint8_t* a = buffer->u.YUVA.a;
Packit 9c6abc
    const int a_stride = buffer->u.YUVA.a_stride;
Packit 9c6abc
    uint32_t y;
Packit 9c6abc
Packit 9c6abc
    if (a == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
    fprintf(fout, "P5\n%u %u\n255\n", width, height);
Packit 9c6abc
    for (y = 0; y < height; ++y) {
Packit 9c6abc
      if (fwrite(a, width, 1, fout) != 1) return 0;
Packit 9c6abc
      a += a_stride;
Packit 9c6abc
    }
Packit 9c6abc
    return 1;
Packit 9c6abc
  }
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// PGM with IMC4 layout
Packit 9c6abc
Packit 9c6abc
int WebPWritePGM(FILE* fout, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  if (fout == NULL || buffer == NULL) {
Packit 9c6abc
    return 0;
Packit 9c6abc
  } else {
Packit 9c6abc
    const int width = buffer->width;
Packit 9c6abc
    const int height = buffer->height;
Packit 9c6abc
    const WebPYUVABuffer* const yuv = &buffer->u.YUVA;
Packit 9c6abc
    const uint8_t* src_y = yuv->y;
Packit 9c6abc
    const uint8_t* src_u = yuv->u;
Packit 9c6abc
    const uint8_t* src_v = yuv->v;
Packit 9c6abc
    const uint8_t* src_a = yuv->a;
Packit 9c6abc
    const int uv_width = (width + 1) / 2;
Packit 9c6abc
    const int uv_height = (height + 1) / 2;
Packit 9c6abc
    const int a_height = (src_a != NULL) ? height : 0;
Packit 9c6abc
    int ok = 1;
Packit 9c6abc
    int y;
Packit 9c6abc
Packit 9c6abc
    if (src_y == NULL || src_u == NULL || src_v == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
    fprintf(fout, "P5\n%d %d\n255\n",
Packit 9c6abc
            (width + 1) & ~1, height + uv_height + a_height);
Packit 9c6abc
    for (y = 0; ok && y < height; ++y) {
Packit 9c6abc
      ok &= (fwrite(src_y, width, 1, fout) == 1);
Packit 9c6abc
      if (width & 1) fputc(0, fout);    // padding byte
Packit 9c6abc
      src_y += yuv->y_stride;
Packit 9c6abc
    }
Packit 9c6abc
    for (y = 0; ok && y < uv_height; ++y) {
Packit 9c6abc
      ok &= (fwrite(src_u, uv_width, 1, fout) == 1);
Packit 9c6abc
      ok &= (fwrite(src_v, uv_width, 1, fout) == 1);
Packit 9c6abc
      src_u += yuv->u_stride;
Packit 9c6abc
      src_v += yuv->v_stride;
Packit 9c6abc
    }
Packit 9c6abc
    for (y = 0; ok && y < a_height; ++y) {
Packit 9c6abc
      ok &= (fwrite(src_a, width, 1, fout) == 1);
Packit 9c6abc
      if (width & 1) fputc(0, fout);    // padding byte
Packit 9c6abc
      src_a += yuv->a_stride;
Packit 9c6abc
    }
Packit 9c6abc
    return ok;
Packit 9c6abc
  }
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// Raw YUV(A) planes
Packit 9c6abc
Packit 9c6abc
int WebPWriteYUV(FILE* fout, const WebPDecBuffer* const buffer) {
Packit 9c6abc
  if (fout == NULL || buffer == NULL) {
Packit 9c6abc
    return 0;
Packit 9c6abc
  } else {
Packit 9c6abc
    const int width = buffer->width;
Packit 9c6abc
    const int height = buffer->height;
Packit 9c6abc
    const WebPYUVABuffer* const yuv = &buffer->u.YUVA;
Packit 9c6abc
    const uint8_t* src_y = yuv->y;
Packit 9c6abc
    const uint8_t* src_u = yuv->u;
Packit 9c6abc
    const uint8_t* src_v = yuv->v;
Packit 9c6abc
    const uint8_t* src_a = yuv->a;
Packit 9c6abc
    const int uv_width = (width + 1) / 2;
Packit 9c6abc
    const int uv_height = (height + 1) / 2;
Packit 9c6abc
    const int a_height = (src_a != NULL) ? height : 0;
Packit 9c6abc
    int ok = 1;
Packit 9c6abc
    int y;
Packit 9c6abc
Packit 9c6abc
    if (src_y == NULL || src_u == NULL || src_v == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
    for (y = 0; ok && y < height; ++y) {
Packit 9c6abc
      ok &= (fwrite(src_y, width, 1, fout) == 1);
Packit 9c6abc
      src_y += yuv->y_stride;
Packit 9c6abc
    }
Packit 9c6abc
    for (y = 0; ok && y < uv_height; ++y) {
Packit 9c6abc
      ok &= (fwrite(src_u, uv_width, 1, fout) == 1);
Packit 9c6abc
      src_u += yuv->u_stride;
Packit 9c6abc
    }
Packit 9c6abc
    for (y = 0; ok && y < uv_height; ++y) {
Packit 9c6abc
      ok &= (fwrite(src_v, uv_width, 1, fout) == 1);
Packit 9c6abc
      src_v += yuv->v_stride;
Packit 9c6abc
    }
Packit 9c6abc
    for (y = 0; ok && y < a_height; ++y) {
Packit 9c6abc
      ok &= (fwrite(src_a, width, 1, fout) == 1);
Packit 9c6abc
      src_a += yuv->a_stride;
Packit 9c6abc
    }
Packit 9c6abc
    return ok;
Packit 9c6abc
  }
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// Generic top-level call
Packit 9c6abc
Packit 9c6abc
int WebPSaveImage(const WebPDecBuffer* const buffer,
Packit 9c6abc
                  WebPOutputFileFormat format,
Packit 9c6abc
                  const char* const out_file_name) {
Packit 9c6abc
  FILE* fout = NULL;
Packit 9c6abc
  int needs_open_file = 1;
Packit 9c6abc
  const int use_stdout = (out_file_name != NULL) && !strcmp(out_file_name, "-");
Packit 9c6abc
  int ok = 1;
Packit 9c6abc
Packit 9c6abc
  if (buffer == NULL || out_file_name == NULL) return 0;
Packit 9c6abc
Packit 9c6abc
#ifdef HAVE_WINCODEC_H
Packit 9c6abc
  needs_open_file = (format != PNG);
Packit 9c6abc
#endif
Packit 9c6abc
Packit 9c6abc
  if (needs_open_file) {
Packit 9c6abc
    fout = use_stdout ? ImgIoUtilSetBinaryMode(stdout)
Packit 9c6abc
                      : fopen(out_file_name, "wb");
Packit 9c6abc
    if (fout == NULL) {
Packit 9c6abc
      fprintf(stderr, "Error opening output file %s\n", out_file_name);
Packit 9c6abc
      return 0;
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
Packit 9c6abc
  if (format == PNG ||
Packit 9c6abc
      format == RGBA || format == BGRA || format == ARGB ||
Packit 9c6abc
      format == rgbA || format == bgrA || format == Argb) {
Packit 9c6abc
#ifdef HAVE_WINCODEC_H
Packit 9c6abc
    ok &= WebPWritePNG(out_file_name, use_stdout, buffer);
Packit 9c6abc
#else
Packit 9c6abc
    ok &= WebPWritePNG(fout, buffer);
Packit 9c6abc
#endif
Packit 9c6abc
  } else if (format == PAM) {
Packit 9c6abc
    ok &= WebPWritePAM(fout, buffer);
Packit 9c6abc
  } else if (format == PPM || format == RGB || format == BGR) {
Packit 9c6abc
    ok &= WebPWritePPM(fout, buffer);
Packit 9c6abc
  } else if (format == RGBA_4444 || format == RGB_565 || format == rgbA_4444) {
Packit 9c6abc
    ok &= WebPWrite16bAsPGM(fout, buffer);
Packit 9c6abc
  } else if (format == BMP) {
Packit 9c6abc
    ok &= WebPWriteBMP(fout, buffer);
Packit 9c6abc
  } else if (format == TIFF) {
Packit 9c6abc
    ok &= WebPWriteTIFF(fout, buffer);
Packit 9c6abc
  } else if (format == RAW_YUV) {
Packit 9c6abc
    ok &= WebPWriteYUV(fout, buffer);
Packit 9c6abc
  } else if (format == PGM || format == YUV || format == YUVA) {
Packit 9c6abc
    ok &= WebPWritePGM(fout, buffer);
Packit 9c6abc
  } else if (format == ALPHA_PLANE_ONLY) {
Packit 9c6abc
    ok &= WebPWriteAlphaPlane(fout, buffer);
Packit 9c6abc
  }
Packit 9c6abc
  if (fout != NULL && fout != stdout) {
Packit 9c6abc
    fclose(fout);
Packit 9c6abc
  }
Packit 9c6abc
  return ok;
Packit 9c6abc
}