WebPCompressor.h

/*  WebP Compressor Class:
    Handles alpha channels, ICC profiles and XMP metadata
 
    Copyright (C) 2022-2023 Ruven Pillay
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
 
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
 
    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software Foundation,
    Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
 
*/
 
 
#ifndef _WEBPCOMPRESSOR_H
#define _WEBPCOMPRESSOR_H
 
 
#include "Compressor.h"
#include <webp/encode.h>
#include <webp/mux.h>
 
 
 
class WebPCompressor : public Compressor {
 
 private:
 
  WebPConfig config;
  WebPPicture pic;
  WebPMemoryWriter writer;
  WebPMux* mux;
 
  RawTile tile;                   
  unsigned int chunk_size;        
  size_t current_chunk;           
 
 
  void writeICCProfile();
 
  void writeXMPMetadata();
 
 
 public:
 
 
  WebPCompressor( int compressionLevel ) : Compressor(compressionLevel) {
 
    // Initialize our configuration structure with default values
    WebPConfigInit( &config );
    config.method = 0;         // Fastest encoding
    config.thread_level = 1;   // Enable threading
    config.quality = this->Q;  // WebP's quality range is 0-100
 
    // Create our muxer object
    mux = WebPMuxNew();
 
  };
 
 
  ~WebPCompressor(){
    // Delete our muxer object
    WebPMuxDelete( mux );
  };
 
 
 
  void InitCompression( const RawTile& rawtile, unsigned int strip_height );
 
 
 
  unsigned int CompressStrip( unsigned char* source, unsigned char* output, unsigned int tile_height );
 
 
 
  unsigned int Finish( unsigned char* output );
 
 
 
  unsigned int Compress( RawTile& t );
 
 
  inline unsigned int getHeaderSize() const { return header_size; }
 
  inline unsigned char* getHeader() { return header; }
 
  inline const char* getMimeType() const { return "image/webp"; }
 
  inline const char* getSuffix() const { return "webp"; }
 
  inline CompressionType getCompressionType() const { return WEBP; };
 
 
 
  inline int getQuality() const { return Q; }
 
 
 
  inline void setQuality( int quality ){
 
    // WebP quality ranges from 0 (best compression - smallest size) to 100 (worst compression - largest size)
    this->Q = quality;
 
    // WebP compression level
    if( Q < 0 ) Q = 0;
    else if( Q > 100 ) Q = 100;
 
    // Update our WebP config structure
    config.quality = Q;
  }
 
 
};
 
#endif