/*

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%                                                                             %

%                                                                             %

%                                                                             %

%                         W   W  EEEEE  BBBB   PPPP                           %

%                         W   W  E      B   B  P   P                          %

%                         W W W  EEE    BBBB   PPPP                           %

%                         WW WW  E      B   B  P                              %

%                         W   W  EEEEE  BBBB   P                              %

%                                                                             %

%                                                                             %

%                         Read/Write WebP Image Format                        %

%                                                                             %

%                              Software Design                                %

%                                   Cristy                                    %

%                                 March 2011                                  %

%                                                                             %

%                                                                             %

%  Copyright @ 1999 ImageMagick Studio LLC, a non-profit organization         %

%  dedicated to making software imaging solutions freely available.           %

%                                                                             %

%  You may not use this file except in compliance with the License.  You may  %

%  obtain a copy of the License at                                            %

%                                                                             %

%    https://imagemagick.org/license/                                         %

%                                                                             %

%  Unless required by applicable law or agreed to in writing, software        %

%  distributed under the License is distributed on an "AS IS" BASIS,          %

%  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   %

%  See the License for the specific language governing permissions and        %

%  limitations under the License.                                             %

%                                                                             %

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%

%

*/

/*

  Include declarations.

*/

#include "MagickCore/studio.h"

#include "MagickCore/artifact.h"

#include "MagickCore/blob.h"

#include "MagickCore/blob-private.h"

#include "MagickCore/client.h"

#include "MagickCore/colorspace-private.h"

#include "MagickCore/display.h"

#include "MagickCore/exception.h"

#include "MagickCore/exception-private.h"

#include "MagickCore/image.h"

#include "MagickCore/image-private.h"

#include "MagickCore/layer.h"

#include "MagickCore/list.h"

#include "MagickCore/magick.h"

#include "MagickCore/monitor.h"

#include "MagickCore/monitor-private.h"

#include "MagickCore/memory_.h"

#include "MagickCore/option.h"

#include "MagickCore/pixel-accessor.h"

#include "MagickCore/profile-private.h"

#include "MagickCore/property.h"

#include "MagickCore/quantum-private.h"

#include "MagickCore/static.h"

#include "MagickCore/string_.h"

#include "MagickCore/string-private.h"

#include "MagickCore/module.h"

#include "MagickCore/utility.h"

#include "MagickCore/xwindow.h"

#include "MagickCore/xwindow-private.h"

#if defined(MAGICKCORE_WEBP_DELEGATE)

#include <webp/decode.h>

#include <webp/encode.h>

#if defined(MAGICKCORE_WEBPMUX_DELEGATE)

#include <webp/mux.h>

#include <webp/demux.h>

#endif

#endif

/*

  Forward declarations.

*/

#if defined(MAGICKCORE_WEBP_DELEGATE)

static MagickBooleanType

  WriteWEBPImage(const ImageInfo *,Image *,ExceptionInfo *);

#endif

/*

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%                                                                             %

%                                                                             %

%                                                                             %

%   I s W E B P                                                               %

%                                                                             %

%                                                                             %

%                                                                             %

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%

%  IsWEBP() returns MagickTrue if the image format type, identified by the

%  magick string, is WebP.

%

%  The format of the IsWEBP method is:

%

%      MagickBooleanType IsWEBP(const unsigned char *magick,const size_t length)

%

%  A description of each parameter follows:

%

%    o magick: compare image format pattern against these bytes.

%

%    o length: Specifies the length of the magick string.

%

*/

static MagickBooleanType IsWEBP(const unsigned char *magick,const size_t length)

{

  if (length < 12)

    return(MagickFalse);

  if (LocaleNCompare((const char *) magick+8,"WEBP",4) == 0)

    return(MagickTrue);

  return(MagickFalse);

}

#if defined(MAGICKCORE_WEBP_DELEGATE)

/*

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%                                                                             %

%                                                                             %

%                                                                             %

%   R e a d W E B P I m a g e                                                 %

%                                                                             %

%                                                                             %

%                                                                             %

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%

%  ReadWEBPImage() reads an image in the WebP image format.

%

%  The format of the ReadWEBPImage method is:

%

%      Image *ReadWEBPImage(const ImageInfo *image_info,

%        ExceptionInfo *exception)

%

%  A description of each parameter follows:

%

%    o image_info: the image info.

%

%    o exception: return any errors or warnings in this structure.

%

*/

static inline uint32_t ReadWebPLSBWord(

  const unsigned char *magick_restrict data)

{

  const unsigned char

    *p;

  uint32_t

    value;

  p=data;

  value=(uint32_t) (*p++);

  value|=((uint32_t) (*p++)) << 8;

  value|=((uint32_t) (*p++)) << 16;

  value|=((uint32_t) (*p++)) << 24;

  return(value);

}

static inline void WriteWebPLSBWord(unsigned char *magick_restrict data,

  const size_t value)

{

  unsigned char

    *p;

  p=data;

  *(p++)=(unsigned char) value;

  *(p++)=(unsigned char) (value >> 8);

  *(p++)=(unsigned char) (value >> 16);

  *(p++)=(unsigned char) (value >> 24);

}

static MagickBooleanType IsWEBPImageLossless(const unsigned char *stream,

  const size_t length)

{

#define VP8_CHUNK_INDEX  15

#define LOSSLESS_FLAG  'L'

#define EXTENDED_HEADER  'X'

#define VP8_CHUNK_HEADER  "VP8"

#define VP8_CHUNK_HEADER_SIZE  3

#define RIFF_HEADER_SIZE  12

#define VP8X_CHUNK_SIZE  10

#define TAG_SIZE  4

#define CHUNK_SIZE_BYTES  4

#define CHUNK_HEADER_SIZE  8

#define MAX_CHUNK_PAYLOAD  (~0U-CHUNK_HEADER_SIZE-1)

  size_t

    offset;

  /*

    Read simple header.

  */

  if (length <= VP8_CHUNK_INDEX)

    return(MagickFalse);

  if (stream[VP8_CHUNK_INDEX] != EXTENDED_HEADER)

    return(stream[VP8_CHUNK_INDEX] == LOSSLESS_FLAG ? MagickTrue : MagickFalse);

  /*

    Read extended header.

  */

  offset=RIFF_HEADER_SIZE+TAG_SIZE+CHUNK_SIZE_BYTES+VP8X_CHUNK_SIZE;

  while (offset <= (length-TAG_SIZE-TAG_SIZE-4))

  {

    uint32_t

      chunk_size,

      chunk_size_pad;

    chunk_size=ReadWebPLSBWord(stream+offset+TAG_SIZE);

    if (chunk_size > MAX_CHUNK_PAYLOAD)

      break;

    chunk_size_pad=(CHUNK_HEADER_SIZE+chunk_size+1) & (unsigned int) ~1;

    if (memcmp(stream+offset,VP8_CHUNK_HEADER,VP8_CHUNK_HEADER_SIZE) == 0)

      return(*(stream+offset+VP8_CHUNK_HEADER_SIZE) == LOSSLESS_FLAG ?

        MagickTrue : MagickFalse);

    offset+=chunk_size_pad;

  }

  return(MagickFalse);

}

static int FillBasicWEBPInfo(Image *image,const uint8_t *stream,size_t length,

  WebPDecoderConfig *configure)

{

  int

    webp_status;

  WebPBitstreamFeatures

    *magick_restrict features = &configure->input;

  webp_status=(int) WebPGetFeatures(stream,length,features);

  if (webp_status != VP8_STATUS_OK)

    return(webp_status);

  image->columns=(size_t) features->width;

  image->rows=(size_t) features->height;

  image->depth=8;

  image->alpha_trait=features->has_alpha != 0 ? BlendPixelTrait :

    UndefinedPixelTrait;

  return(webp_status);

}

static int ReadSingleWEBPImage(const ImageInfo *image_info,Image *image,

  const uint8_t *stream,size_t length,WebPDecoderConfig *configure,

  ExceptionInfo *exception,MagickBooleanType is_first)

{

  int

    webp_status;

  MagickBooleanType

    status;

  size_t

    canvas_width,

    canvas_height,

    image_width,

    image_height;

  ssize_t

    x_offset,

    y_offset,

    y;

  unsigned char

    *p;

  WebPDecBuffer

    *magick_restrict webp_image;

  webp_image=&configure->output;

  if (is_first != MagickFalse)

    {

      canvas_width=image->columns;

      canvas_height=image->rows;

      x_offset=image->page.x;

      y_offset=image->page.y;

      image->page.x=0;

      image->page.y=0;

    }

  else

    {

      canvas_width=0;

      canvas_height=0;

      x_offset=0;

      y_offset=0;

    }

  webp_status=FillBasicWEBPInfo(image,stream,length,configure);

  image_width=image->columns;

  image_height=image->rows;

  if (is_first)

    {

      image->columns=canvas_width;

      image->rows=canvas_height;

    }

  if (webp_status != VP8_STATUS_OK)

    return(webp_status);

  if (IsWEBPImageLossless((unsigned char *) stream,length) != MagickFalse)

    image->quality=100;

  if (image_info->ping != MagickFalse)

    return(webp_status);

  webp_status=(int) WebPDecode(stream,length,configure);

  if (webp_status != VP8_STATUS_OK)

    return(webp_status);

  p=(unsigned char *) webp_image->u.RGBA.rgba;

  for (y=0; y < (ssize_t) image->rows; y++)

  {

    Quantum

      *q;

    ssize_t

      x;

    q=QueueAuthenticPixels(image,0,y,image->columns,1,exception);

    if (q == (Quantum *) NULL)

      break;

    for (x=0; x < (ssize_t) image->columns; x++)

    {

      if (((x >= x_offset) && (x < (x_offset+(ssize_t) image_width))) &&

          ((y >= y_offset) && (y < (y_offset+(ssize_t) image_height))))

        {

          SetPixelRed(image,ScaleCharToQuantum(*p++),q);

          SetPixelGreen(image,ScaleCharToQuantum(*p++),q);

          SetPixelBlue(image,ScaleCharToQuantum(*p++),q);

          SetPixelAlpha(image,ScaleCharToQuantum(*p++),q);

        }

      else

        {

          SetPixelRed(image,(Quantum) 0,q);

          SetPixelGreen(image,(Quantum) 0,q);

          SetPixelBlue(image,(Quantum) 0,q);

          SetPixelAlpha(image,(Quantum) 0,q);

        }

      q+=(ptrdiff_t) GetPixelChannels(image);

    }

    if (SyncAuthenticPixels(image,exception) == MagickFalse)

      break;

    status=SetImageProgress(image,LoadImageTag,(MagickOffsetType) y,

      image->rows);

    if (status == MagickFalse)

      break;

  }

  WebPFreeDecBuffer(webp_image);

#if defined(MAGICKCORE_WEBPMUX_DELEGATE)

  {

    StringInfo

      *profile;

    uint32_t

      webp_flags = 0;

    WebPData

     chunk,

     content;

    WebPMux

      *mux;

    /*

      Extract any profiles:

      https://developers.google.com/speed/webp/docs/container-api.

    */

    content.bytes=stream;

    content.size=length;

    mux=WebPMuxCreate(&content,0);

    (void) memset(&chunk,0,sizeof(chunk));

    (void) WebPMuxGetFeatures(mux,&webp_flags);

    if ((webp_flags & ICCP_FLAG) &&

        (WebPMuxGetChunk(mux,"ICCP",&chunk) == WEBP_MUX_OK))

      if (chunk.size != 0)

        {

          profile=BlobToProfileStringInfo("icc",chunk.bytes,chunk.size,

            exception);

          (void) SetImageProfilePrivate(image,profile,exception);

        }

    if ((webp_flags & EXIF_FLAG) &&

        (WebPMuxGetChunk(mux,"EXIF",&chunk) == WEBP_MUX_OK))

      if (chunk.size != 0)

        {

          profile=BlobToProfileStringInfo("exif",chunk.bytes,chunk.size,

            exception);

          (void) SetImageProfilePrivate(image,profile,exception);

        }

    if (((webp_flags & XMP_FLAG) &&

         (WebPMuxGetChunk(mux,"XMP ",&chunk) == WEBP_MUX_OK)) ||

         (WebPMuxGetChunk(mux,"XMP\0",&chunk) == WEBP_MUX_OK))

      if (chunk.size != 0)

        {

          profile=BlobToProfileStringInfo("xmp",chunk.bytes,chunk.size,

            exception);

          (void) SetImageProfilePrivate(image,profile,exception);

        }

    WebPMuxDelete(mux);

  }

#endif

  return(webp_status);

}

#if defined(MAGICKCORE_WEBPMUX_DELEGATE)

static int ReadAnimatedWEBPImage(const ImageInfo *image_info,Image *image,

  uint8_t *stream,size_t length,WebPDecoderConfig *configure,

  ExceptionInfo *exception)

{

  Image

    *original_image;

  int

    image_count,

    webp_status;

  size_t

    canvas_width,

    canvas_height;

  WebPData

    data;

  WebPDemuxer

    *demux;

  WebPIterator

    iter;

  image_count=0;

  webp_status=0;

  original_image=image;

  webp_status=FillBasicWEBPInfo(image,stream,length,configure);

  canvas_width=image->columns;

  canvas_height=image->rows;

  data.bytes=stream;

  data.size=length;

  {

    WebPMux

      *mux;

    WebPMuxAnimParams

      params;

    WebPMuxError

      status;

    mux=WebPMuxCreate(&data,0);

    status=WebPMuxGetAnimationParams(mux,&params);

    if (status >= 0)

      image->iterations=(size_t) params.loop_count;

    WebPMuxDelete(mux);

  }

  demux=WebPDemux(&data);

  if (WebPDemuxGetFrame(demux,1,&iter))

    {

      do

      {

        if (image_count != 0)

          {

            AcquireNextImage(image_info,image,exception);

            if (GetNextImageInList(image) == (Image *) NULL)

              break;

            image=SyncNextImageInList(image);

            CloneImageProperties(image,original_image);

            image->page.x=(ssize_t) iter.x_offset;

            image->page.y=(ssize_t) iter.y_offset;

            webp_status=ReadSingleWEBPImage(image_info,image,

              iter.fragment.bytes,iter.fragment.size,configure,exception,

              MagickFalse);

          }

        else

          {

            image->page.x=(ssize_t) iter.x_offset;

            image->page.y=(ssize_t) iter.y_offset;

            webp_status=ReadSingleWEBPImage(image_info,image,

              iter.fragment.bytes,iter.fragment.size,configure,exception,

              MagickTrue);

          }

        if (webp_status != VP8_STATUS_OK)

          break;

        image->page.width=canvas_width;

        image->page.height=canvas_height;

        image->ticks_per_second=100;

        image->delay=(size_t) round(iter.duration/10.0);

        image->dispose=NoneDispose;

        if (iter.dispose_method == WEBP_MUX_DISPOSE_BACKGROUND)

          image->dispose=BackgroundDispose;

        (void) SetImageProperty(image,"webp:mux-blend",

          "AtopPreviousAlphaBlend",exception);

        if (iter.blend_method == WEBP_MUX_BLEND)

          (void) SetImageProperty(image,"webp:mux-blend",

            "AtopBackgroundAlphaBlend",exception);

        image_count++;

      } while (WebPDemuxNextFrame(&iter));

      WebPDemuxReleaseIterator(&iter);

    }

  WebPDemuxDelete(demux);

  return(webp_status);

}

#endif

static Image *ReadWEBPImage(const ImageInfo *image_info,

  ExceptionInfo *exception)

{

#define ThrowWEBPException(severity,tag) \

{ \

  if (stream != (unsigned char *) NULL) \

    stream=(unsigned char*) RelinquishMagickMemory(stream); \

  if (webp_image != (WebPDecBuffer *) NULL) \

    WebPFreeDecBuffer(webp_image); \

  ThrowReaderException(severity,tag); \

}

  Image

    *image;

  int

    webp_status;

  MagickBooleanType

    status;

  size_t

    blob_size,

    length;

  ssize_t

    count;

  unsigned char

    header[12],

    *stream;

  WebPDecoderConfig

    configure;

  WebPDecBuffer

    *magick_restrict webp_image = &configure.output;

  /*

    Open image file.

  */

  assert(image_info != (const ImageInfo *) NULL);

  assert(image_info->signature == MagickCoreSignature);

  assert(exception != (ExceptionInfo *) NULL);

  assert(exception->signature == MagickCoreSignature);

  if (IsEventLogging() != MagickFalse)

    (void) LogMagickEvent(TraceEvent,GetMagickModule(),"%s",

      image_info->filename);

  image=AcquireImage(image_info,exception);

  status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);

  if (status == MagickFalse)

    {

      image=DestroyImageList(image);

      return((Image *) NULL);

    }

  stream=(unsigned char *) NULL;

  if (WebPInitDecoderConfig(&configure) == 0)

    ThrowReaderException(ResourceLimitError,"UnableToDecodeImageFile");

  webp_image->colorspace=MODE_RGBA;

  count=ReadBlob(image,12,header);

  if (count != 12)

    ThrowWEBPException(CorruptImageError,"InsufficientImageDataInFile");

  status=IsWEBP(header,(size_t) count);

  if (status == MagickFalse)

    ThrowWEBPException(CorruptImageError,"CorruptImage");

  length=(size_t) (ReadWebPLSBWord(header+4)+8);

  if (length < 12)

    ThrowWEBPException(CorruptImageError,"CorruptImage");

  blob_size=(size_t) GetBlobSize(image);

  if (length > blob_size)

    {

      size_t

        delta=length-blob_size;

      if (delta != 12 && delta != (12 + 8))

        ThrowWEBPException(CorruptImageError,"InsufficientImageDataInFile");

      length-=delta;

      WriteWebPLSBWord(header+4,length-8);

    }

  stream=(unsigned char *) AcquireQuantumMemory(length,sizeof(*stream));

  if (stream == (unsigned char *) NULL)

    ThrowWEBPException(ResourceLimitError,"MemoryAllocationFailed");

  (void) memcpy(stream,header,12);

  count=ReadBlob(image,length-12,stream+12);

  if (count != (ssize_t) (length-12))

    ThrowWEBPException(CorruptImageError,"InsufficientImageDataInFile");

  webp_status=FillBasicWEBPInfo(image,stream,length,&configure);

  if (webp_status == VP8_STATUS_OK) {

    if (configure.input.has_animation) {

#if defined(MAGICKCORE_WEBPMUX_DELEGATE)

      webp_status=ReadAnimatedWEBPImage(image_info,image,stream,length,

        &configure,exception);

#else

      webp_status=VP8_STATUS_UNSUPPORTED_FEATURE;

#endif

    } else {

      webp_status=ReadSingleWEBPImage(image_info,image,stream,length,

        &configure,exception,MagickFalse);

    }

  }

  if (webp_status != VP8_STATUS_OK)

    switch (webp_status)

    {

      case VP8_STATUS_OUT_OF_MEMORY:

      {

        ThrowWEBPException(ResourceLimitError,"MemoryAllocationFailed");

        break;

      }

      case VP8_STATUS_INVALID_PARAM:

      {

        ThrowWEBPException(CorruptImageError,"invalid parameter");

        break;

      }

      case VP8_STATUS_BITSTREAM_ERROR:

      {

        ThrowWEBPException(CorruptImageError,"CorruptImage");

        break;

      }

      case VP8_STATUS_UNSUPPORTED_FEATURE:

      {

        ThrowWEBPException(CoderError,"DataEncodingSchemeIsNotSupported");

        break;

      }

      case VP8_STATUS_SUSPENDED:

      {

        ThrowWEBPException(CorruptImageError,"decoder suspended");

        break;

      }

      case VP8_STATUS_USER_ABORT:

      {

        ThrowWEBPException(CorruptImageError,"user abort");

        break;

      }

      case VP8_STATUS_NOT_ENOUGH_DATA:

      {

        ThrowWEBPException(CorruptImageError,"InsufficientImageDataInFile");

        break;

      }

      default:

        ThrowWEBPException(CorruptImageError,"CorruptImage");

    }

  stream=(unsigned char*) RelinquishMagickMemory(stream);

  (void) CloseBlob(image);

  return(image);

}

#endif

/*

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%                                                                             %

%                                                                             %

%                                                                             %

%   R e g i s t e r W E B P I m a g e                                         %

%                                                                             %

%                                                                             %

%                                                                             %

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%

%  RegisterWEBPImage() adds attributes for the WebP image format to

%  the list of supported formats.  The attributes include the image format

%  tag, a method to read and/or write the format, whether the format

%  supports the saving of more than one frame to the same file or blob,

%  whether the format supports native in-memory I/O, and a brief

%  description of the format.

%

%  The format of the RegisterWEBPImage method is:

%

%      size_t RegisterWEBPImage(void)

%

*/

ModuleExport size_t RegisterWEBPImage(void)

{

  char

    version[MagickPathExtent];

  MagickInfo

    *entry;

  *version='\0';

  entry=AcquireMagickInfo("WEBP","WEBP","WebP Image Format");

#if defined(MAGICKCORE_WEBP_DELEGATE)

  entry->decoder=(DecodeImageHandler *) ReadWEBPImage;

  entry->encoder=(EncodeImageHandler *) WriteWEBPImage;

  (void) FormatLocaleString(version,MagickPathExtent,"libwebp %d.%d.%d [%04X]",

    (WebPGetEncoderVersion() >> 16) & 0xff,

    (WebPGetEncoderVersion() >> 8) & 0xff,

    (WebPGetEncoderVersion() >> 0) & 0xff,WEBP_ENCODER_ABI_VERSION);

#endif

  entry->mime_type=ConstantString("image/webp");

  entry->flags|=CoderDecoderSeekableStreamFlag;

#if !defined(MAGICKCORE_WEBPMUX_DELEGATE)

  entry->flags^=CoderAdjoinFlag;

#endif

  entry->magick=(IsImageFormatHandler *) IsWEBP;

  if (*version != '\0')

    entry->version=ConstantString(version);

  (void) RegisterMagickInfo(entry);

  return(MagickImageCoderSignature);

}

/*

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%                                                                             %

%                                                                             %

%                                                                             %

%   U n r e g i s t e r W E B P I m a g e                                     %

%                                                                             %

%                                                                             %

%                                                                             %

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%

%  UnregisterWEBPImage() removes format registrations made by the WebP module

%  from the list of supported formats.

%

%  The format of the UnregisterWEBPImage method is:

%

%      UnregisterWEBPImage(void)

%

*/

ModuleExport void UnregisterWEBPImage(void)

{

  (void) UnregisterMagickInfo("WEBP");

}

#if defined(MAGICKCORE_WEBP_DELEGATE)

/*

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%                                                                             %

%                                                                             %

%                                                                             %

%   W r i t e W E B P I m a g e                                               %

%                                                                             %

%                                                                             %

%                                                                             %

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%

%  WriteWEBPImage() writes an image in the WebP image format.

%

%  The format of the WriteWEBPImage method is:

%

%      MagickBooleanType WriteWEBPImage(const ImageInfo *image_info,

%        Image *image)

%

%  A description of each parameter follows.

%

%    o image_info: the image info.

%

%    o image:  The image.

%

*/

#if WEBP_ENCODER_ABI_VERSION >= 0x0100

static int WebPEncodeProgress(int percent,const WebPPicture* picture)

{

#define EncodeImageTag  "Encode/Image"

  Image

    *image;

  MagickBooleanType

    status;

  image=(Image *) picture->user_data;

  status=SetImageProgress(image,EncodeImageTag,percent-1,100);

  return(status == MagickFalse ? 0 : 1);

}

#endif

static const char * WebPErrorCodeMessage(WebPEncodingError error_code)

{

  switch (error_code)

  {

    case VP8_ENC_OK:

      return "";

    case VP8_ENC_ERROR_OUT_OF_MEMORY:

      return "out of memory";

    case VP8_ENC_ERROR_BITSTREAM_OUT_OF_MEMORY:

      return "bitstream out of memory";

    case VP8_ENC_ERROR_NULL_PARAMETER:

      return "NULL parameter";

    case VP8_ENC_ERROR_INVALID_CONFIGURATION:

      return "invalid configuration";

    case VP8_ENC_ERROR_BAD_DIMENSION:

      return "bad dimension";

    case VP8_ENC_ERROR_PARTITION0_OVERFLOW:

      return "partition 0 overflow (> 512K)";

    case VP8_ENC_ERROR_PARTITION_OVERFLOW:

      return "partition overflow (> 16M)";

    case VP8_ENC_ERROR_BAD_WRITE:

      return "bad write";

    case VP8_ENC_ERROR_FILE_TOO_BIG:

      return "file too big (> 4GB)";

#if WEBP_ENCODER_ABI_VERSION >= 0x0100

    case VP8_ENC_ERROR_USER_ABORT:

      return "user abort";

    case VP8_ENC_ERROR_LAST:

      return "error last";

#endif

  }

  return "unknown exception";

}

static MagickBooleanType WriteSingleWEBPPicture(const ImageInfo *image_info,

  Image *image,WebPPicture *picture,MemoryInfo **memory_info,

  ExceptionInfo *exception)

{

  MagickBooleanType

    status;

  ssize_t

    y;

  uint32_t

    *magick_restrict q;

#if WEBP_ENCODER_ABI_VERSION >= 0x0100

  if (image->progress_monitor != (MagickProgressMonitor) NULL)

    {

      picture->progress_hook=WebPEncodeProgress;

      picture->user_data=(void *) image;

    }

#endif

  picture->width=(int) image->columns;

  picture->height=(int) image->rows;

  picture->argb_stride=(int) image->columns;

  picture->use_argb=1;

  /*

    Allocate memory for pixels.

  */

  if (IssRGBCompatibleColorspace(image->colorspace) == MagickFalse)

    (void) TransformImageColorspace(image,sRGBColorspace,exception);

  *memory_info=AcquireVirtualMemory(image->columns,image->rows*

    sizeof(*(picture->argb)));

  if (*memory_info == (MemoryInfo *) NULL)

    ThrowWriterException(ResourceLimitError,"MemoryAllocationFailed");

  picture->argb=(uint32_t *) GetVirtualMemoryBlob(*memory_info);

  /*

    Convert image to WebP raster pixels.

  */

  status=MagickFalse;

  q=picture->argb;

  for (y=0; y < (ssize_t) image->rows; y++)

  {

    const Quantum

      *magick_restrict p;

    ssize_t

      x;

    p=GetVirtualPixels(image,0,y,image->columns,1,exception);

    if (p == (const Quantum *) NULL)

      break;

    for (x=0; x < (ssize_t) image->columns; x++)

    {

      *q++=(uint32_t) (image->alpha_trait != UndefinedPixelTrait ? (uint32_t)

        ScaleQuantumToChar(GetPixelAlpha(image,p)) << 24 : 0xff000000) |

        ((uint32_t) ScaleQuantumToChar(GetPixelRed(image,p)) << 16) |

        ((uint32_t) ScaleQuantumToChar(GetPixelGreen(image,p)) << 8) |

        ((uint32_t) ScaleQuantumToChar(GetPixelBlue(image,p)));

      p+=(ptrdiff_t) GetPixelChannels(image);

    }

    status=SetImageProgress(image,SaveImageTag,(MagickOffsetType) y,

      image->rows);

    if (status == MagickFalse)

      break;

  }

  return(status);

}

static MagickBooleanType WriteSingleWEBPImage(const ImageInfo *image_info,

  Image *image,WebPConfig *configure,WebPMemoryWriter *writer,

  ExceptionInfo *exception)

{

  MagickBooleanType

    status;

  MemoryInfo

    *memory_info;

  WebPPicture

    picture;

  if (WebPPictureInit(&picture) == 0)

    ThrowWriterException(ResourceLimitError,"UnableToEncodeImageFile");

  picture.writer=WebPMemoryWrite;

  picture.custom_ptr=writer;

  status=WriteSingleWEBPPicture(image_info,image,&picture,&memory_info,

    exception);

  if (status != MagickFalse)

    {

      status=(MagickBooleanType) WebPEncode(configure,&picture);

      if (status == MagickFalse)

        (void) ThrowMagickException(exception,GetMagickModule(),

          CorruptImageError,WebPErrorCodeMessage(picture.error_code),"`%s'",

          image->filename);

  }

  if (memory_info != (MemoryInfo *) NULL)

    memory_info=RelinquishVirtualMemory(memory_info);

  WebPPictureFree(&picture);

  return(status);

}

#if defined(MAGICKCORE_WEBPMUX_DELEGATE)

static void *WebPDestroyMemoryInfo(void *memory_info)

{

  return((void *) RelinquishVirtualMemory((MemoryInfo *) memory_info));

}

static MagickBooleanType WriteAnimatedWEBPImage(const ImageInfo *image_info,

  Image *image,const WebPConfig *configure,WebPData *webp_data,

  ExceptionInfo *exception)

{

  Image

    *frame;

  LinkedListInfo

    *memory_info_list;

  MagickBooleanType

    status;

  MemoryInfo

    *memory_info;

  size_t

    effective_delta,

    frame_timestamp;

  WebPAnimEncoder

    *enc;

  WebPAnimEncoderOptions

    enc_options;

  WebPPicture

    picture;

  (void) WebPAnimEncoderOptionsInit(&enc_options);

  if (image_info->verbose != MagickFalse)

    enc_options.verbose=1;

  /*

    Appropriate default kmin, kmax values for lossy and lossless.

  */

  enc_options.kmin = configure->lossless ? 9 : 3;

  enc_options.kmax = configure->lossless ? 17 : 5;

  enc=WebPAnimEncoderNew((int) image->columns,(int) image->rows,&enc_options);

  if (enc == (WebPAnimEncoder*) NULL)

    return(MagickFalse);

  status=MagickTrue;

  effective_delta=0;

  frame_timestamp=0;

  memory_info_list=NewLinkedList(GetImageListLength(image));

  frame=image;

  while (frame != NULL)

  {

    status=(MagickBooleanType) WebPPictureInit(&picture);

    if (status == MagickFalse)

      {

        (void) ThrowMagickException(exception,GetMagickModule(),

          ResourceLimitError,"UnableToEncodeImageFile","`%s'",image->filename);

        break;

      }

    status=WriteSingleWEBPPicture(image_info,frame,&picture,

      &memory_info,exception);

    if (status != MagickFalse)

      {

        status=(MagickBooleanType) WebPAnimEncoderAdd(enc,&picture,

          (int) frame_timestamp,configure);

        if (status == MagickFalse)

          (void) ThrowMagickException(exception,GetMagickModule(),

            CorruptImageError,WebPErrorCodeMessage(picture.error_code),"`%s'",

            image->filename);

      }

    if (memory_info != (MemoryInfo *) NULL)

      (void) AppendValueToLinkedList(memory_info_list,memory_info);

    WebPPictureFree(&picture);

    effective_delta=(size_t) (frame->delay*1000*MagickSafeReciprocal(

      (double) frame->ticks_per_second));

    if (effective_delta < 10)

      effective_delta=100; /* Consistent with gif2webp */

    frame_timestamp+=effective_delta;

    frame=GetNextImageInList(frame);

  }

  if (status != MagickFalse)

    {

      /*

        Add last null frame and assemble picture.

      */

      status=(MagickBooleanType) WebPAnimEncoderAdd(enc,(WebPPicture *) NULL,

        (int) frame_timestamp,configure);

      if (status != MagickFalse)

        status=(MagickBooleanType) WebPAnimEncoderAssemble(enc,webp_data);

      if (status == MagickFalse)

        (void) ThrowMagickException(exception,GetMagickModule(),CoderError,

          WebPAnimEncoderGetError(enc),"`%s'",image->filename);

    }

  memory_info_list=DestroyLinkedList(memory_info_list,WebPDestroyMemoryInfo);

  WebPAnimEncoderDelete(enc);

  return(status);

}

static MagickBooleanType WriteWEBPImageProfile(Image *image,

  WebPData *webp_data,ExceptionInfo *exception)