shun-iwasawa 7bad49
#ifndef TINYEXR_H_
shun-iwasawa 7bad49
#define TINYEXR_H_
shun-iwasawa 7bad49
/*
shun-iwasawa 7bad49
Copyright (c) 2014 - 2021, Syoyo Fujita and many contributors.
shun-iwasawa 7bad49
All rights reserved.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
Redistribution and use in source and binary forms, with or without
shun-iwasawa 7bad49
modification, are permitted provided that the following conditions are met:
shun-iwasawa 7bad49
    * Redistributions of source code must retain the above copyright
shun-iwasawa 7bad49
      notice, this list of conditions and the following disclaimer.
shun-iwasawa 7bad49
    * Redistributions in binary form must reproduce the above copyright
shun-iwasawa 7bad49
      notice, this list of conditions and the following disclaimer in the
shun-iwasawa 7bad49
      documentation and/or other materials provided with the distribution.
shun-iwasawa 7bad49
    * Neither the name of the Syoyo Fujita nor the
shun-iwasawa 7bad49
      names of its contributors may be used to endorse or promote products
shun-iwasawa 7bad49
      derived from this software without specific prior written permission.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
shun-iwasawa 7bad49
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
shun-iwasawa 7bad49
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
shun-iwasawa 7bad49
DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
shun-iwasawa 7bad49
DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
shun-iwasawa 7bad49
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
shun-iwasawa 7bad49
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
shun-iwasawa 7bad49
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
shun-iwasawa 7bad49
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
shun-iwasawa 7bad49
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
shun-iwasawa 7bad49
*/
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// TinyEXR contains some OpenEXR code, which is licensed under ------------
shun-iwasawa 7bad49
shun-iwasawa 7bad49
///////////////////////////////////////////////////////////////////////////
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Copyright (c) 2002, Industrial Light & Magic, a division of Lucas
shun-iwasawa 7bad49
// Digital Ltd. LLC
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// All rights reserved.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Redistribution and use in source and binary forms, with or without
shun-iwasawa 7bad49
// modification, are permitted provided that the following conditions are
shun-iwasawa 7bad49
// met:
shun-iwasawa 7bad49
// *       Redistributions of source code must retain the above copyright
shun-iwasawa 7bad49
// notice, this list of conditions and the following disclaimer.
shun-iwasawa 7bad49
// *       Redistributions in binary form must reproduce the above
shun-iwasawa 7bad49
// copyright notice, this list of conditions and the following disclaimer
shun-iwasawa 7bad49
// in the documentation and/or other materials provided with the
shun-iwasawa 7bad49
// distribution.
shun-iwasawa 7bad49
// *       Neither the name of Industrial Light & Magic nor the names of
shun-iwasawa 7bad49
// its contributors may be used to endorse or promote products derived
shun-iwasawa 7bad49
// from this software without specific prior written permission.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
shun-iwasawa 7bad49
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
shun-iwasawa 7bad49
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
shun-iwasawa 7bad49
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
shun-iwasawa 7bad49
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
shun-iwasawa 7bad49
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
shun-iwasawa 7bad49
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
shun-iwasawa 7bad49
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
shun-iwasawa 7bad49
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
shun-iwasawa 7bad49
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
shun-iwasawa 7bad49
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
///////////////////////////////////////////////////////////////////////////
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// End of OpenEXR license -------------------------------------------------
shun-iwasawa 7bad49
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//   Do this:
shun-iwasawa 7bad49
//    #define TINYEXR_IMPLEMENTATION
shun-iwasawa 7bad49
//   before you include this file in *one* C or C++ file to create the
shun-iwasawa 7bad49
//   implementation.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//   // i.e. it should look like this:
shun-iwasawa 7bad49
//   #include ...
shun-iwasawa 7bad49
//   #include ...
shun-iwasawa 7bad49
//   #include ...
shun-iwasawa 7bad49
//   #define TINYEXR_IMPLEMENTATION
shun-iwasawa 7bad49
//   #include "tinyexr.h"
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#include <stddef.h>  // for size_t
shun-iwasawa 7bad49
#include <stdint.h>  // guess stdint.h is available(C99)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __cplusplus
shun-iwasawa 7bad49
extern "C" {
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if defined(_M_IX86) || defined(_M_X64) || defined(__i386__) || \
shun-iwasawa 7bad49
    defined(__i386) || defined(__i486__) || defined(__i486) ||  \
shun-iwasawa 7bad49
    defined(i386) || defined(__ia64__) || defined(__x86_64__)
shun-iwasawa 7bad49
#define TINYEXR_X86_OR_X64_CPU 1
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
#define TINYEXR_X86_OR_X64_CPU 0
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if (__BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__) || TINYEXR_X86_OR_X64_CPU
shun-iwasawa 7bad49
#define TINYEXR_LITTLE_ENDIAN 1
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
#define TINYEXR_LITTLE_ENDIAN 0
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Use miniz or not to decode ZIP format pixel. Linking with zlib
shun-iwasawa 7bad49
// required if this flas is 0.
shun-iwasawa 7bad49
#ifndef TINYEXR_USE_MINIZ
shun-iwasawa 7bad49
#define TINYEXR_USE_MINIZ (1)
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Disable PIZ comporession when applying cpplint.
shun-iwasawa 7bad49
#ifndef TINYEXR_USE_PIZ
shun-iwasawa 7bad49
#define TINYEXR_USE_PIZ (1)
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifndef TINYEXR_USE_ZFP
shun-iwasawa 7bad49
#define TINYEXR_USE_ZFP (0)  // TinyEXR extension.
shun-iwasawa 7bad49
// http://computation.llnl.gov/projects/floating-point-compression
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifndef TINYEXR_USE_THREAD
shun-iwasawa 7bad49
#define TINYEXR_USE_THREAD (0)  // No threaded loading.
shun-iwasawa 7bad49
// http://computation.llnl.gov/projects/floating-point-compression
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifndef TINYEXR_USE_OPENMP
shun-iwasawa 7bad49
#ifdef _OPENMP
shun-iwasawa 7bad49
#define TINYEXR_USE_OPENMP (1)
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
#define TINYEXR_USE_OPENMP (0)
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#define TINYEXR_SUCCESS (0)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_INVALID_MAGIC_NUMBER (-1)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_INVALID_EXR_VERSION (-2)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_INVALID_ARGUMENT (-3)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_INVALID_DATA (-4)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_INVALID_FILE (-5)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_INVALID_PARAMETER (-6)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_CANT_OPEN_FILE (-7)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_UNSUPPORTED_FORMAT (-8)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_INVALID_HEADER (-9)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_UNSUPPORTED_FEATURE (-10)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_CANT_WRITE_FILE (-11)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_SERIALZATION_FAILED (-12)
shun-iwasawa 7bad49
#define TINYEXR_ERROR_LAYER_NOT_FOUND (-13)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// @note { OpenEXR file format: http://www.openexr.com/openexrfilelayout.pdf }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// pixel type: possible values are: UINT = 0 HALF = 1 FLOAT = 2
shun-iwasawa 7bad49
#define TINYEXR_PIXELTYPE_UINT (0)
shun-iwasawa 7bad49
#define TINYEXR_PIXELTYPE_HALF (1)
shun-iwasawa 7bad49
#define TINYEXR_PIXELTYPE_FLOAT (2)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#define TINYEXR_MAX_HEADER_ATTRIBUTES (1024)
shun-iwasawa 7bad49
#define TINYEXR_MAX_CUSTOM_ATTRIBUTES (128)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#define TINYEXR_COMPRESSIONTYPE_NONE (0)
shun-iwasawa 7bad49
#define TINYEXR_COMPRESSIONTYPE_RLE (1)
shun-iwasawa 7bad49
#define TINYEXR_COMPRESSIONTYPE_ZIPS (2)
shun-iwasawa 7bad49
#define TINYEXR_COMPRESSIONTYPE_ZIP (3)
shun-iwasawa 7bad49
#define TINYEXR_COMPRESSIONTYPE_PIZ (4)
shun-iwasawa 7bad49
#define TINYEXR_COMPRESSIONTYPE_ZFP (128)  // TinyEXR extension
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#define TINYEXR_ZFP_COMPRESSIONTYPE_RATE (0)
shun-iwasawa 7bad49
#define TINYEXR_ZFP_COMPRESSIONTYPE_PRECISION (1)
shun-iwasawa 7bad49
#define TINYEXR_ZFP_COMPRESSIONTYPE_ACCURACY (2)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#define TINYEXR_TILE_ONE_LEVEL (0)
shun-iwasawa 7bad49
#define TINYEXR_TILE_MIPMAP_LEVELS (1)
shun-iwasawa 7bad49
#define TINYEXR_TILE_RIPMAP_LEVELS (2)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#define TINYEXR_TILE_ROUND_DOWN (0)
shun-iwasawa 7bad49
#define TINYEXR_TILE_ROUND_UP (1)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _EXRVersion {
shun-iwasawa 7bad49
  int version;    // this must be 2
shun-iwasawa 7bad49
  // tile format image;
shun-iwasawa 7bad49
  // not zero for only a single-part "normal" tiled file (according to spec.)
shun-iwasawa 7bad49
  int tiled;      
shun-iwasawa 7bad49
  int long_name;  // long name attribute
shun-iwasawa 7bad49
  // deep image(EXR 2.0);
shun-iwasawa 7bad49
  // for a multi-part file, indicates that at least one part is of type deep* (according to spec.)
shun-iwasawa 7bad49
  int non_image;  
shun-iwasawa 7bad49
  int multipart;  // multi-part(EXR 2.0)
shun-iwasawa 7bad49
} EXRVersion;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _EXRAttribute {
shun-iwasawa 7bad49
  char name[256];  // name and type are up to 255 chars long.
shun-iwasawa 7bad49
  char type[256];
shun-iwasawa 7bad49
  unsigned char *value;  // uint8_t*
shun-iwasawa 7bad49
  int size;
shun-iwasawa 7bad49
  int pad0;
shun-iwasawa 7bad49
} EXRAttribute;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _EXRChannelInfo {
shun-iwasawa 7bad49
  char name[256];  // less than 255 bytes long
shun-iwasawa 7bad49
  int pixel_type;
shun-iwasawa 7bad49
  int x_sampling;
shun-iwasawa 7bad49
  int y_sampling;
shun-iwasawa 7bad49
  unsigned char p_linear;
shun-iwasawa 7bad49
  unsigned char pad[3];
shun-iwasawa 7bad49
} EXRChannelInfo;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _EXRTile {
shun-iwasawa 7bad49
  int offset_x;
shun-iwasawa 7bad49
  int offset_y;
shun-iwasawa 7bad49
  int level_x;
shun-iwasawa 7bad49
  int level_y;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int width;   // actual width in a tile.
shun-iwasawa 7bad49
  int height;  // actual height int a tile.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  unsigned char **images;  // image[channels][pixels]
shun-iwasawa 7bad49
} EXRTile;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _EXRBox2i {
shun-iwasawa 7bad49
  int min_x;
shun-iwasawa 7bad49
  int min_y;
shun-iwasawa 7bad49
  int max_x;
shun-iwasawa 7bad49
  int max_y;
shun-iwasawa 7bad49
} EXRBox2i;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _EXRHeader {
shun-iwasawa 7bad49
  float pixel_aspect_ratio;
shun-iwasawa 7bad49
  int line_order;
shun-iwasawa 7bad49
  EXRBox2i data_window;
shun-iwasawa 7bad49
  EXRBox2i display_window;
shun-iwasawa 7bad49
  float screen_window_center[2];
shun-iwasawa 7bad49
  float screen_window_width;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int chunk_count;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Properties for tiled format(`tiledesc`).
shun-iwasawa 7bad49
  int tiled;
shun-iwasawa 7bad49
  int tile_size_x;
shun-iwasawa 7bad49
  int tile_size_y;
shun-iwasawa 7bad49
  int tile_level_mode;
shun-iwasawa 7bad49
  int tile_rounding_mode;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int long_name;
shun-iwasawa 7bad49
  // for a single-part file, agree with the version field bit 11
shun-iwasawa 7bad49
  // for a multi-part file, it is consistent with the type of part
shun-iwasawa 7bad49
  int non_image;
shun-iwasawa 7bad49
  int multipart;
shun-iwasawa 7bad49
  unsigned int header_len;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Custom attributes(exludes required attributes(e.g. `channels`,
shun-iwasawa 7bad49
  // `compression`, etc)
shun-iwasawa 7bad49
  int num_custom_attributes;
shun-iwasawa 7bad49
  EXRAttribute *custom_attributes;  // array of EXRAttribute. size =
shun-iwasawa 7bad49
                                    // `num_custom_attributes`.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  EXRChannelInfo *channels;  // [num_channels]
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int *pixel_types;  // Loaded pixel type(TINYEXR_PIXELTYPE_*) of `images` for
shun-iwasawa 7bad49
  // each channel. This is overwritten with `requested_pixel_types` when
shun-iwasawa 7bad49
  // loading.
shun-iwasawa 7bad49
  int num_channels;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int compression_type;        // compression type(TINYEXR_COMPRESSIONTYPE_*)
shun-iwasawa 7bad49
  int *requested_pixel_types;  // Filled initially by
shun-iwasawa 7bad49
                               // ParseEXRHeaderFrom(Meomory|File), then users
shun-iwasawa 7bad49
                               // can edit it(only valid for HALF pixel type
shun-iwasawa 7bad49
                               // channel)
shun-iwasawa 7bad49
  // name attribute required for multipart files;
shun-iwasawa 7bad49
  // must be unique and non empty (according to spec.);
shun-iwasawa 7bad49
  // use EXRSetNameAttr for setting value;
shun-iwasawa 7bad49
  // max 255 character allowed - excluding terminating zero
shun-iwasawa 7bad49
  char name[256];
shun-iwasawa 7bad49
} EXRHeader;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _EXRMultiPartHeader {
shun-iwasawa 7bad49
  int num_headers;
shun-iwasawa 7bad49
  EXRHeader *headers;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
} EXRMultiPartHeader;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _EXRImage {
shun-iwasawa 7bad49
  EXRTile *tiles;  // Tiled pixel data. The application must reconstruct image
shun-iwasawa 7bad49
                   // from tiles manually. NULL if scanline format.
shun-iwasawa 7bad49
  struct _EXRImage* next_level; // NULL if scanline format or image is the last level.
shun-iwasawa 7bad49
  int level_x; // x level index
shun-iwasawa 7bad49
  int level_y; // y level index
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  unsigned char **images;  // image[channels][pixels]. NULL if tiled format.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int width;
shun-iwasawa 7bad49
  int height;
shun-iwasawa 7bad49
  int num_channels;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Properties for tile format.
shun-iwasawa 7bad49
  int num_tiles;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
} EXRImage;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _EXRMultiPartImage {
shun-iwasawa 7bad49
  int num_images;
shun-iwasawa 7bad49
  EXRImage *images;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
} EXRMultiPartImage;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct _DeepImage {
shun-iwasawa 7bad49
  const char **channel_names;
shun-iwasawa 7bad49
  float ***image;      // image[channels][scanlines][samples]
shun-iwasawa 7bad49
  int **offset_table;  // offset_table[scanline][offsets]
shun-iwasawa 7bad49
  int num_channels;
shun-iwasawa 7bad49
  int width;
shun-iwasawa 7bad49
  int height;
shun-iwasawa 7bad49
  int pad0;
shun-iwasawa 7bad49
} DeepImage;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// @deprecated { For backward compatibility. Not recommended to use. }
shun-iwasawa 7bad49
// Loads single-frame OpenEXR image. Assume EXR image contains A(single channel
shun-iwasawa 7bad49
// alpha) or RGB(A) channels.
shun-iwasawa 7bad49
// Application must free image data as returned by `out_rgba`
shun-iwasawa 7bad49
// Result image format is: float x RGBA x width x hight
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
extern int LoadEXR(float **out_rgba, int *width, int *height,
shun-iwasawa 7bad49
                   const char *filename, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Loads single-frame OpenEXR image by specifying layer name. Assume EXR image
shun-iwasawa 7bad49
// contains A(single channel alpha) or RGB(A) channels. Application must free
shun-iwasawa 7bad49
// image data as returned by `out_rgba` Result image format is: float x RGBA x
shun-iwasawa 7bad49
// width x hight Returns negative value and may set error string in `err` when
shun-iwasawa 7bad49
// there's an error When the specified layer name is not found in the EXR file,
shun-iwasawa 7bad49
// the function will return `TINYEXR_ERROR_LAYER_NOT_FOUND`.
shun-iwasawa 7bad49
extern int LoadEXRWithLayer(float **out_rgba, int *width, int *height,
shun-iwasawa 7bad49
                            const char *filename, const char *layer_name,
shun-iwasawa 7bad49
                            const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Get layer infos from EXR file.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// @param[out] layer_names List of layer names. Application must free memory
shun-iwasawa 7bad49
// after using this.
shun-iwasawa 7bad49
// @param[out] num_layers The number of layers
shun-iwasawa 7bad49
// @param[out] err Error string(will be filled when the function returns error
shun-iwasawa 7bad49
// code). Free it using FreeEXRErrorMessage after using this value.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// @return TINYEXR_SUCCEES upon success.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
extern int EXRLayers(const char *filename, const char **layer_names[],
shun-iwasawa 7bad49
                     int *num_layers, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// @deprecated { to be removed. }
shun-iwasawa 7bad49
// Simple wrapper API for ParseEXRHeaderFromFile.
shun-iwasawa 7bad49
// checking given file is a EXR file(by just look up header)
shun-iwasawa 7bad49
// @return TINYEXR_SUCCEES for EXR image, TINYEXR_ERROR_INVALID_HEADER for
shun-iwasawa 7bad49
// others
shun-iwasawa 7bad49
extern int IsEXR(const char *filename);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// @deprecated { to be removed. }
shun-iwasawa 7bad49
// Saves single-frame OpenEXR image. Assume EXR image contains RGB(A) channels.
shun-iwasawa 7bad49
// components must be 1(Grayscale), 3(RGB) or 4(RGBA).
shun-iwasawa 7bad49
// Input image format is: `float x width x height`, or `float x RGB(A) x width x
shun-iwasawa 7bad49
// hight`
shun-iwasawa 7bad49
// Save image as fp16(HALF) format when `save_as_fp16` is positive non-zero
shun-iwasawa 7bad49
// value.
shun-iwasawa 7bad49
// Save image as fp32(FLOAT) format when `save_as_fp16` is 0.
shun-iwasawa 7bad49
// Use ZIP compression by default.
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
extern int SaveEXR(const float *data, const int width, const int height,
shun-iwasawa 7bad49
                   const int components, const int save_as_fp16,
shun-iwasawa 7bad49
                   const char *filename, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Returns the number of resolution levels of the image (including the base)
shun-iwasawa 7bad49
extern int EXRNumLevels(const EXRImage* exr_image);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Initialize EXRHeader struct
shun-iwasawa 7bad49
extern void InitEXRHeader(EXRHeader *exr_header);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Set name attribute of EXRHeader struct (it makes a copy)
shun-iwasawa 7bad49
extern void EXRSetNameAttr(EXRHeader *exr_header, const char* name);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Initialize EXRImage struct
shun-iwasawa 7bad49
extern void InitEXRImage(EXRImage *exr_image);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Frees internal data of EXRHeader struct
shun-iwasawa 7bad49
extern int FreeEXRHeader(EXRHeader *exr_header);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Frees internal data of EXRImage struct
shun-iwasawa 7bad49
extern int FreeEXRImage(EXRImage *exr_image);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Frees error message
shun-iwasawa 7bad49
extern void FreeEXRErrorMessage(const char *msg);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Parse EXR version header of a file.
shun-iwasawa 7bad49
extern int ParseEXRVersionFromFile(EXRVersion *version, const char *filename);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Parse EXR version header from memory-mapped EXR data.
shun-iwasawa 7bad49
extern int ParseEXRVersionFromMemory(EXRVersion *version,
shun-iwasawa 7bad49
                                     const unsigned char *memory, size_t size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Parse single-part OpenEXR header from a file and initialize `EXRHeader`.
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int ParseEXRHeaderFromFile(EXRHeader *header, const EXRVersion *version,
shun-iwasawa 7bad49
                                  const char *filename, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Parse single-part OpenEXR header from a memory and initialize `EXRHeader`.
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int ParseEXRHeaderFromMemory(EXRHeader *header,
shun-iwasawa 7bad49
                                    const EXRVersion *version,
shun-iwasawa 7bad49
                                    const unsigned char *memory, size_t size,
shun-iwasawa 7bad49
                                    const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Parse multi-part OpenEXR headers from a file and initialize `EXRHeader*`
shun-iwasawa 7bad49
// array.
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int ParseEXRMultipartHeaderFromFile(EXRHeader ***headers,
shun-iwasawa 7bad49
                                           int *num_headers,
shun-iwasawa 7bad49
                                           const EXRVersion *version,
shun-iwasawa 7bad49
                                           const char *filename,
shun-iwasawa 7bad49
                                           const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Parse multi-part OpenEXR headers from a memory and initialize `EXRHeader*`
shun-iwasawa 7bad49
// array
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int ParseEXRMultipartHeaderFromMemory(EXRHeader ***headers,
shun-iwasawa 7bad49
                                             int *num_headers,
shun-iwasawa 7bad49
                                             const EXRVersion *version,
shun-iwasawa 7bad49
                                             const unsigned char *memory,
shun-iwasawa 7bad49
                                             size_t size, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Loads single-part OpenEXR image from a file.
shun-iwasawa 7bad49
// Application must setup `ParseEXRHeaderFromFile` before calling this function.
shun-iwasawa 7bad49
// Application can free EXRImage using `FreeEXRImage`
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int LoadEXRImageFromFile(EXRImage *image, const EXRHeader *header,
shun-iwasawa 7bad49
                                const char *filename, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Loads single-part OpenEXR image from a memory.
shun-iwasawa 7bad49
// Application must setup `EXRHeader` with
shun-iwasawa 7bad49
// `ParseEXRHeaderFromMemory` before calling this function.
shun-iwasawa 7bad49
// Application can free EXRImage using `FreeEXRImage`
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int LoadEXRImageFromMemory(EXRImage *image, const EXRHeader *header,
shun-iwasawa 7bad49
                                  const unsigned char *memory,
shun-iwasawa 7bad49
                                  const size_t size, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Loads multi-part OpenEXR image from a file.
shun-iwasawa 7bad49
// Application must setup `ParseEXRMultipartHeaderFromFile` before calling this
shun-iwasawa 7bad49
// function.
shun-iwasawa 7bad49
// Application can free EXRImage using `FreeEXRImage`
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int LoadEXRMultipartImageFromFile(EXRImage *images,
shun-iwasawa 7bad49
                                         const EXRHeader **headers,
shun-iwasawa 7bad49
                                         unsigned int num_parts,
shun-iwasawa 7bad49
                                         const char *filename,
shun-iwasawa 7bad49
                                         const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Loads multi-part OpenEXR image from a memory.
shun-iwasawa 7bad49
// Application must setup `EXRHeader*` array with
shun-iwasawa 7bad49
// `ParseEXRMultipartHeaderFromMemory` before calling this function.
shun-iwasawa 7bad49
// Application can free EXRImage using `FreeEXRImage`
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int LoadEXRMultipartImageFromMemory(EXRImage *images,
shun-iwasawa 7bad49
                                           const EXRHeader **headers,
shun-iwasawa 7bad49
                                           unsigned int num_parts,
shun-iwasawa 7bad49
                                           const unsigned char *memory,
shun-iwasawa 7bad49
                                           const size_t size, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Saves multi-channel, single-frame OpenEXR image to a file.
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int SaveEXRImageToFile(const EXRImage *image,
shun-iwasawa 7bad49
                              const EXRHeader *exr_header, const char *filename,
shun-iwasawa 7bad49
                              const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Saves multi-channel, single-frame OpenEXR image to a memory.
shun-iwasawa 7bad49
// Image is compressed using EXRImage.compression value.
shun-iwasawa 7bad49
// Return the number of bytes if success.
shun-iwasawa 7bad49
// Return zero and will set error string in `err` when there's an
shun-iwasawa 7bad49
// error.
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern size_t SaveEXRImageToMemory(const EXRImage *image,
shun-iwasawa 7bad49
                                   const EXRHeader *exr_header,
shun-iwasawa 7bad49
                                   unsigned char **memory, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Saves multi-channel, multi-frame OpenEXR image to a memory.
shun-iwasawa 7bad49
// Image is compressed using EXRImage.compression value.
shun-iwasawa 7bad49
// File global attributes (eg. display_window) must be set in the first header.
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int SaveEXRMultipartImageToFile(const EXRImage *images,
shun-iwasawa 7bad49
                                       const EXRHeader **exr_headers,
shun-iwasawa 7bad49
                                       unsigned int num_parts,
shun-iwasawa 7bad49
                                       const char *filename, const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Saves multi-channel, multi-frame OpenEXR image to a memory.
shun-iwasawa 7bad49
// Image is compressed using EXRImage.compression value.
shun-iwasawa 7bad49
// File global attributes (eg. display_window) must be set in the first header.
shun-iwasawa 7bad49
// Return the number of bytes if success.
shun-iwasawa 7bad49
// Return zero and will set error string in `err` when there's an
shun-iwasawa 7bad49
// error.
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern size_t SaveEXRMultipartImageToMemory(const EXRImage *images,
shun-iwasawa 7bad49
                                            const EXRHeader **exr_headers,
shun-iwasawa 7bad49
                                            unsigned int num_parts,
shun-iwasawa 7bad49
                                            unsigned char **memory, const char **err);
shun-iwasawa 7bad49
// Loads single-frame OpenEXR deep image.
shun-iwasawa 7bad49
// Application must free memory of variables in DeepImage(image, offset_table)
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int LoadDeepEXR(DeepImage *out_image, const char *filename,
shun-iwasawa 7bad49
                       const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// NOT YET IMPLEMENTED:
shun-iwasawa 7bad49
// Saves single-frame OpenEXR deep image.
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
// extern int SaveDeepEXR(const DeepImage *in_image, const char *filename,
shun-iwasawa 7bad49
//                       const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// NOT YET IMPLEMENTED:
shun-iwasawa 7bad49
// Loads multi-part OpenEXR deep image.
shun-iwasawa 7bad49
// Application must free memory of variables in DeepImage(image, offset_table)
shun-iwasawa 7bad49
// extern int LoadMultiPartDeepEXR(DeepImage **out_image, int num_parts, const
shun-iwasawa 7bad49
// char *filename,
shun-iwasawa 7bad49
//                       const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// For emscripten.
shun-iwasawa 7bad49
// Loads single-frame OpenEXR image from memory. Assume EXR image contains
shun-iwasawa 7bad49
// RGB(A) channels.
shun-iwasawa 7bad49
// Returns negative value and may set error string in `err` when there's an
shun-iwasawa 7bad49
// error
shun-iwasawa 7bad49
// When there was an error message, Application must free `err` with
shun-iwasawa 7bad49
// FreeEXRErrorMessage()
shun-iwasawa 7bad49
extern int LoadEXRFromMemory(float **out_rgba, int *width, int *height,
shun-iwasawa 7bad49
                             const unsigned char *memory, size_t size,
shun-iwasawa 7bad49
                             const char **err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __cplusplus
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif  // TINYEXR_H_
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef TINYEXR_IMPLEMENTATION
shun-iwasawa 7bad49
#ifndef TINYEXR_IMPLEMENTATION_DEFINED
shun-iwasawa 7bad49
#define TINYEXR_IMPLEMENTATION_DEFINED
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifndef WIN32_LEAN_AND_MEAN
shun-iwasawa 7bad49
#define WIN32_LEAN_AND_MEAN
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#ifndef NOMINMAX
shun-iwasawa 7bad49
#define NOMINMAX
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#include <windows.h>  // for UTF-8
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#include <algorithm>
shun-iwasawa 7bad49
#include <cassert>
shun-iwasawa 7bad49
#include <cstdio>
shun-iwasawa 7bad49
#include <cstdlib>
shun-iwasawa 7bad49
#include <cstring>
shun-iwasawa 7bad49
#include <sstream>
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// #include <iostream> // debug
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#include <limits>
shun-iwasawa 7bad49
#include <string>
shun-iwasawa 7bad49
#include <vector>
shun-iwasawa 7bad49
#include <set>
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// https://stackoverflow.com/questions/5047971/how-do-i-check-for-c11-support
shun-iwasawa 7bad49
#if __cplusplus > 199711L || (defined(_MSC_VER) && _MSC_VER >= 1900)
shun-iwasawa 7bad49
#define TINYEXR_HAS_CXX11 (1)
shun-iwasawa 7bad49
// C++11
shun-iwasawa 7bad49
#include <cstdint>
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_THREAD
shun-iwasawa 7bad49
#include <atomic>
shun-iwasawa 7bad49
#include <thread>
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif  // __cplusplus > 199711L
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_OPENMP
shun-iwasawa 7bad49
#include <omp.h>
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_MINIZ
shun-iwasawa 7bad49
#include <miniz.h>
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
//  Issue #46. Please include your own zlib-compatible API header before
shun-iwasawa 7bad49
//  including `tinyexr.h`
shun-iwasawa 7bad49
//#include "zlib.h"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_ZFP
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic push
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Weverything"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#include "zfp.h"
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic pop
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
namespace tinyexr {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if __cplusplus > 199711L
shun-iwasawa 7bad49
// C++11
shun-iwasawa 7bad49
typedef uint64_t tinyexr_uint64;
shun-iwasawa 7bad49
typedef int64_t tinyexr_int64;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
// Although `long long` is not a standard type pre C++11, assume it is defined
shun-iwasawa 7bad49
// as a compiler's extension.
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic push
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wc++11-long-long"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
typedef unsigned long long tinyexr_uint64;
shun-iwasawa 7bad49
typedef long long tinyexr_int64;
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic pop
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// static bool IsBigEndian(void) {
shun-iwasawa 7bad49
//  union {
shun-iwasawa 7bad49
//    unsigned int i;
shun-iwasawa 7bad49
//    char c[4];
shun-iwasawa 7bad49
//  } bint = {0x01020304};
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//  return bint.c[0] == 1;
shun-iwasawa 7bad49
//}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void SetErrorMessage(const std::string &msg, const char **err) {
shun-iwasawa 7bad49
  if (err) {
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
    (*err) = _strdup(msg.c_str());
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    (*err) = strdup(msg.c_str());
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static const int kEXRVersionSize = 8;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void cpy2(unsigned short *dst_val, const unsigned short *src_val) {
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(dst_val);
shun-iwasawa 7bad49
  const unsigned char *src = reinterpret_cast<const unsigned char *>(src_val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[0];
shun-iwasawa 7bad49
  dst[1] = src[1];
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void swap2(unsigned short *val) {
shun-iwasawa 7bad49
#ifdef TINYEXR_LITTLE_ENDIAN
shun-iwasawa 7bad49
  (void)val;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  unsigned short tmp = *val;
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(val);
shun-iwasawa 7bad49
  unsigned char *src = reinterpret_cast<unsigned char *>(&tmp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[1];
shun-iwasawa 7bad49
  dst[1] = src[0];
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic push
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wunused-function"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __GNUC__
shun-iwasawa 7bad49
#pragma GCC diagnostic push
shun-iwasawa 7bad49
#pragma GCC diagnostic ignored "-Wunused-function"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
static void cpy4(int *dst_val, const int *src_val) {
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(dst_val);
shun-iwasawa 7bad49
  const unsigned char *src = reinterpret_cast<const unsigned char *>(src_val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[0];
shun-iwasawa 7bad49
  dst[1] = src[1];
shun-iwasawa 7bad49
  dst[2] = src[2];
shun-iwasawa 7bad49
  dst[3] = src[3];
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void cpy4(unsigned int *dst_val, const unsigned int *src_val) {
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(dst_val);
shun-iwasawa 7bad49
  const unsigned char *src = reinterpret_cast<const unsigned char *>(src_val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[0];
shun-iwasawa 7bad49
  dst[1] = src[1];
shun-iwasawa 7bad49
  dst[2] = src[2];
shun-iwasawa 7bad49
  dst[3] = src[3];
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void cpy4(float *dst_val, const float *src_val) {
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(dst_val);
shun-iwasawa 7bad49
  const unsigned char *src = reinterpret_cast<const unsigned char *>(src_val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[0];
shun-iwasawa 7bad49
  dst[1] = src[1];
shun-iwasawa 7bad49
  dst[2] = src[2];
shun-iwasawa 7bad49
  dst[3] = src[3];
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic pop
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __GNUC__
shun-iwasawa 7bad49
#pragma GCC diagnostic pop
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void swap4(unsigned int *val) {
shun-iwasawa 7bad49
#ifdef TINYEXR_LITTLE_ENDIAN
shun-iwasawa 7bad49
  (void)val;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  unsigned int tmp = *val;
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(val);
shun-iwasawa 7bad49
  unsigned char *src = reinterpret_cast<unsigned char *>(&tmp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[3];
shun-iwasawa 7bad49
  dst[1] = src[2];
shun-iwasawa 7bad49
  dst[2] = src[1];
shun-iwasawa 7bad49
  dst[3] = src[0];
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void swap4(int *val) {
shun-iwasawa 7bad49
#ifdef TINYEXR_LITTLE_ENDIAN
shun-iwasawa 7bad49
  (void)val;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  int tmp = *val;
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(val);
shun-iwasawa 7bad49
  unsigned char *src = reinterpret_cast<unsigned char *>(&tmp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[3];
shun-iwasawa 7bad49
  dst[1] = src[2];
shun-iwasawa 7bad49
  dst[2] = src[1];
shun-iwasawa 7bad49
  dst[3] = src[0];
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void swap4(float *val) {
shun-iwasawa 7bad49
#ifdef TINYEXR_LITTLE_ENDIAN
shun-iwasawa 7bad49
  (void)val;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  float tmp = *val;
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(val);
shun-iwasawa 7bad49
  unsigned char *src = reinterpret_cast<unsigned char *>(&tmp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[3];
shun-iwasawa 7bad49
  dst[1] = src[2];
shun-iwasawa 7bad49
  dst[2] = src[1];
shun-iwasawa 7bad49
  dst[3] = src[0];
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if 0
shun-iwasawa 7bad49
static void cpy8(tinyexr::tinyexr_uint64 *dst_val, const tinyexr::tinyexr_uint64 *src_val) {
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(dst_val);
shun-iwasawa 7bad49
  const unsigned char *src = reinterpret_cast<const unsigned char *>(src_val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[0];
shun-iwasawa 7bad49
  dst[1] = src[1];
shun-iwasawa 7bad49
  dst[2] = src[2];
shun-iwasawa 7bad49
  dst[3] = src[3];
shun-iwasawa 7bad49
  dst[4] = src[4];
shun-iwasawa 7bad49
  dst[5] = src[5];
shun-iwasawa 7bad49
  dst[6] = src[6];
shun-iwasawa 7bad49
  dst[7] = src[7];
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void swap8(tinyexr::tinyexr_uint64 *val) {
shun-iwasawa 7bad49
#ifdef TINYEXR_LITTLE_ENDIAN
shun-iwasawa 7bad49
  (void)val;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  tinyexr::tinyexr_uint64 tmp = (*val);
shun-iwasawa 7bad49
  unsigned char *dst = reinterpret_cast<unsigned char *>(val);
shun-iwasawa 7bad49
  unsigned char *src = reinterpret_cast<unsigned char *>(&tmp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  dst[0] = src[7];
shun-iwasawa 7bad49
  dst[1] = src[6];
shun-iwasawa 7bad49
  dst[2] = src[5];
shun-iwasawa 7bad49
  dst[3] = src[4];
shun-iwasawa 7bad49
  dst[4] = src[3];
shun-iwasawa 7bad49
  dst[5] = src[2];
shun-iwasawa 7bad49
  dst[6] = src[1];
shun-iwasawa 7bad49
  dst[7] = src[0];
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// https://gist.github.com/rygorous/2156668
shun-iwasawa 7bad49
union FP32 {
shun-iwasawa 7bad49
  unsigned int u;
shun-iwasawa 7bad49
  float f;
shun-iwasawa 7bad49
  struct {
shun-iwasawa 7bad49
#if TINYEXR_LITTLE_ENDIAN
shun-iwasawa 7bad49
    unsigned int Mantissa : 23;
shun-iwasawa 7bad49
    unsigned int Exponent : 8;
shun-iwasawa 7bad49
    unsigned int Sign : 1;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    unsigned int Sign : 1;
shun-iwasawa 7bad49
    unsigned int Exponent : 8;
shun-iwasawa 7bad49
    unsigned int Mantissa : 23;
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  } s;
shun-iwasawa 7bad49
};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic push
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wpadded"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
union FP16 {
shun-iwasawa 7bad49
  unsigned short u;
shun-iwasawa 7bad49
  struct {
shun-iwasawa 7bad49
#if TINYEXR_LITTLE_ENDIAN
shun-iwasawa 7bad49
    unsigned int Mantissa : 10;
shun-iwasawa 7bad49
    unsigned int Exponent : 5;
shun-iwasawa 7bad49
    unsigned int Sign : 1;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    unsigned int Sign : 1;
shun-iwasawa 7bad49
    unsigned int Exponent : 5;
shun-iwasawa 7bad49
    unsigned int Mantissa : 10;
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  } s;
shun-iwasawa 7bad49
};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic pop
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static FP32 half_to_float(FP16 h) {
shun-iwasawa 7bad49
  static const FP32 magic = {113 << 23};
shun-iwasawa 7bad49
  static const unsigned int shifted_exp = 0x7c00
shun-iwasawa 7bad49
                                          << 13;  // exponent mask after shift
shun-iwasawa 7bad49
  FP32 o;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  o.u = (h.u & 0x7fffU) << 13U;           // exponent/mantissa bits
shun-iwasawa 7bad49
  unsigned int exp_ = shifted_exp & o.u;  // just the exponent
shun-iwasawa 7bad49
  o.u += (127 - 15) << 23;                // exponent adjust
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // handle exponent special cases
shun-iwasawa 7bad49
  if (exp_ == shifted_exp)    // Inf/NaN?
shun-iwasawa 7bad49
    o.u += (128 - 16) << 23;  // extra exp adjust
shun-iwasawa 7bad49
  else if (exp_ == 0)         // Zero/Denormal?
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    o.u += 1 << 23;  // extra exp adjust
shun-iwasawa 7bad49
    o.f -= magic.f;  // renormalize
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  o.u |= (h.u & 0x8000U) << 16U;  // sign bit
shun-iwasawa 7bad49
  return o;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static FP16 float_to_half_full(FP32 f) {
shun-iwasawa 7bad49
  FP16 o = {0};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Based on ISPC reference code (with minor modifications)
shun-iwasawa 7bad49
  if (f.s.Exponent == 0)  // Signed zero/denormal (which will underflow)
shun-iwasawa 7bad49
    o.s.Exponent = 0;
shun-iwasawa 7bad49
  else if (f.s.Exponent == 255)  // Inf or NaN (all exponent bits set)
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    o.s.Exponent = 31;
shun-iwasawa 7bad49
    o.s.Mantissa = f.s.Mantissa ? 0x200 : 0;  // NaN->qNaN and Inf->Inf
shun-iwasawa 7bad49
  } else                                      // Normalized number
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    // Exponent unbias the single, then bias the halfp
shun-iwasawa 7bad49
    int newexp = f.s.Exponent - 127 + 15;
shun-iwasawa 7bad49
    if (newexp >= 31)  // Overflow, return signed infinity
shun-iwasawa 7bad49
      o.s.Exponent = 31;
shun-iwasawa 7bad49
    else if (newexp <= 0)  // Underflow
shun-iwasawa 7bad49
    {
shun-iwasawa 7bad49
      if ((14 - newexp) <= 24)  // Mantissa might be non-zero
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        unsigned int mant = f.s.Mantissa | 0x800000;  // Hidden 1 bit
shun-iwasawa 7bad49
        o.s.Mantissa = mant >> (14 - newexp);
shun-iwasawa 7bad49
        if ((mant >> (13 - newexp)) & 1)  // Check for rounding
shun-iwasawa 7bad49
          o.u++;  // Round, might overflow into exp bit, but this is OK
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      o.s.Exponent = static_cast<unsigned int>(newexp);
shun-iwasawa 7bad49
      o.s.Mantissa = f.s.Mantissa >> 13;
shun-iwasawa 7bad49
      if (f.s.Mantissa & 0x1000)  // Check for rounding
shun-iwasawa 7bad49
        o.u++;                    // Round, might overflow to inf, this is OK
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  o.s.Sign = f.s.Sign;
shun-iwasawa 7bad49
  return o;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// NOTE: From OpenEXR code
shun-iwasawa 7bad49
// #define IMF_INCREASING_Y  0
shun-iwasawa 7bad49
// #define IMF_DECREASING_Y  1
shun-iwasawa 7bad49
// #define IMF_RAMDOM_Y    2
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// #define IMF_NO_COMPRESSION  0
shun-iwasawa 7bad49
// #define IMF_RLE_COMPRESSION 1
shun-iwasawa 7bad49
// #define IMF_ZIPS_COMPRESSION  2
shun-iwasawa 7bad49
// #define IMF_ZIP_COMPRESSION 3
shun-iwasawa 7bad49
// #define IMF_PIZ_COMPRESSION 4
shun-iwasawa 7bad49
// #define IMF_PXR24_COMPRESSION 5
shun-iwasawa 7bad49
// #define IMF_B44_COMPRESSION 6
shun-iwasawa 7bad49
// #define IMF_B44A_COMPRESSION  7
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic push
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if __has_warning("-Wzero-as-null-pointer-constant")
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wzero-as-null-pointer-constant"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static const char *ReadString(std::string *s, const char *ptr, size_t len) {
shun-iwasawa 7bad49
  // Read untile NULL(\0).
shun-iwasawa 7bad49
  const char *p = ptr;
shun-iwasawa 7bad49
  const char *q = ptr;
shun-iwasawa 7bad49
  while ((size_t(q - ptr) < len) && (*q) != 0) {
shun-iwasawa 7bad49
    q++;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (size_t(q - ptr) >= len) {
shun-iwasawa 7bad49
    (*s).clear();
shun-iwasawa 7bad49
    return NULL;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  (*s) = std::string(p, q);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return q + 1;  // skip '\0'
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool ReadAttribute(std::string *name, std::string *type,
shun-iwasawa 7bad49
                          std::vector<unsigned char> *data, size_t *marker_size,
shun-iwasawa 7bad49
                          const char *marker, size_t size) {
shun-iwasawa 7bad49
  size_t name_len = strnlen(marker, size);
shun-iwasawa 7bad49
  if (name_len == size) {
shun-iwasawa 7bad49
    // String does not have a terminating character.
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  *name = std::string(marker, name_len);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  marker += name_len + 1;
shun-iwasawa 7bad49
  size -= name_len + 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t type_len = strnlen(marker, size);
shun-iwasawa 7bad49
  if (type_len == size) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  *type = std::string(marker, type_len);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  marker += type_len + 1;
shun-iwasawa 7bad49
  size -= type_len + 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (size < sizeof(uint32_t)) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  uint32_t data_len;
shun-iwasawa 7bad49
  memcpy(&data_len, marker, sizeof(uint32_t));
shun-iwasawa 7bad49
  tinyexr::swap4(reinterpret_cast<unsigned int *>(&data_len));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (data_len == 0) {
shun-iwasawa 7bad49
    if ((*type).compare("string") == 0) {
shun-iwasawa 7bad49
      // Accept empty string attribute.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      marker += sizeof(uint32_t);
shun-iwasawa 7bad49
      size -= sizeof(uint32_t);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      *marker_size = name_len + 1 + type_len + 1 + sizeof(uint32_t);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      data->resize(1);
shun-iwasawa 7bad49
      (*data)[0] = '\0';
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      return true;
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  marker += sizeof(uint32_t);
shun-iwasawa 7bad49
  size -= sizeof(uint32_t);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (size < data_len) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  data->resize(static_cast<size_t>(data_len));
shun-iwasawa 7bad49
  memcpy(&data->at(0), marker, static_cast<size_t>(data_len));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  *marker_size = name_len + 1 + type_len + 1 + sizeof(uint32_t) + data_len;
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void WriteAttributeToMemory(std::vector<unsigned char> *out,
shun-iwasawa 7bad49
                                   const char *name, const char *type,
shun-iwasawa 7bad49
                                   const unsigned char *data, int len) {
shun-iwasawa 7bad49
  out->insert(out->end(), name, name + strlen(name) + 1);
shun-iwasawa 7bad49
  out->insert(out->end(), type, type + strlen(type) + 1);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int outLen = len;
shun-iwasawa 7bad49
  tinyexr::swap4(&outLen);
shun-iwasawa 7bad49
  out->insert(out->end(), reinterpret_cast<unsigned char *>(&outLen),
shun-iwasawa 7bad49
              reinterpret_cast<unsigned char *>(&outLen) + sizeof(int));
shun-iwasawa 7bad49
  out->insert(out->end(), data, data + len);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct {
shun-iwasawa 7bad49
  std::string name;  // less than 255 bytes long
shun-iwasawa 7bad49
  int pixel_type;
shun-iwasawa 7bad49
  int requested_pixel_type;
shun-iwasawa 7bad49
  int x_sampling;
shun-iwasawa 7bad49
  int y_sampling;
shun-iwasawa 7bad49
  unsigned char p_linear;
shun-iwasawa 7bad49
  unsigned char pad[3];
shun-iwasawa 7bad49
} ChannelInfo;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
typedef struct {
shun-iwasawa 7bad49
  int min_x;
shun-iwasawa 7bad49
  int min_y;
shun-iwasawa 7bad49
  int max_x;
shun-iwasawa 7bad49
  int max_y;
shun-iwasawa 7bad49
} Box2iInfo;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
struct HeaderInfo {
shun-iwasawa 7bad49
  std::vector<tinyexr::ChannelInfo> channels;
shun-iwasawa 7bad49
  std::vector<EXRAttribute> attributes;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  Box2iInfo data_window;
shun-iwasawa 7bad49
  int line_order;
shun-iwasawa 7bad49
  Box2iInfo display_window;
shun-iwasawa 7bad49
  float screen_window_center[2];
shun-iwasawa 7bad49
  float screen_window_width;
shun-iwasawa 7bad49
  float pixel_aspect_ratio;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int chunk_count;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Tiled format
shun-iwasawa 7bad49
  int tiled; // Non-zero if the part is tiled.
shun-iwasawa 7bad49
  int tile_size_x;
shun-iwasawa 7bad49
  int tile_size_y;
shun-iwasawa 7bad49
  int tile_level_mode;
shun-iwasawa 7bad49
  int tile_rounding_mode;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  unsigned int header_len;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int compression_type;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // required for multi-part or non-image files
shun-iwasawa 7bad49
  std::string name;
shun-iwasawa 7bad49
  // required for multi-part or non-image files
shun-iwasawa 7bad49
  std::string type;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  void clear() {
shun-iwasawa 7bad49
    channels.clear();
shun-iwasawa 7bad49
    attributes.clear();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    data_window.min_x = 0;
shun-iwasawa 7bad49
    data_window.min_y = 0;
shun-iwasawa 7bad49
    data_window.max_x = 0;
shun-iwasawa 7bad49
    data_window.max_y = 0;
shun-iwasawa 7bad49
    line_order = 0;
shun-iwasawa 7bad49
    display_window.min_x = 0;
shun-iwasawa 7bad49
    display_window.min_y = 0;
shun-iwasawa 7bad49
    display_window.max_x = 0;
shun-iwasawa 7bad49
    display_window.max_y = 0;
shun-iwasawa 7bad49
    screen_window_center[0] = 0.0f;
shun-iwasawa 7bad49
    screen_window_center[1] = 0.0f;
shun-iwasawa 7bad49
    screen_window_width = 0.0f;
shun-iwasawa 7bad49
    pixel_aspect_ratio = 0.0f;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    chunk_count = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // Tiled format
shun-iwasawa 7bad49
    tiled = 0;
shun-iwasawa 7bad49
    tile_size_x = 0;
shun-iwasawa 7bad49
    tile_size_y = 0;
shun-iwasawa 7bad49
    tile_level_mode = 0;
shun-iwasawa 7bad49
    tile_rounding_mode = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    header_len = 0;
shun-iwasawa 7bad49
    compression_type = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    name.clear();
shun-iwasawa 7bad49
    type.clear();
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool ReadChannelInfo(std::vector<ChannelInfo> &channels,
shun-iwasawa 7bad49
                            const std::vector<unsigned char> &data) {
shun-iwasawa 7bad49
  const char *p = reinterpret_cast<const char *>(&data.at(0));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (;;) {
shun-iwasawa 7bad49
    if ((*p) == 0) {
shun-iwasawa 7bad49
      break;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    ChannelInfo info;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tinyexr_int64 data_len = static_cast<tinyexr_int64>(data.size()) -
shun-iwasawa 7bad49
                             (p - reinterpret_cast<const char *>(data.data()));
shun-iwasawa 7bad49
    if (data_len < 0) {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    p = ReadString(&info.name, p, size_t(data_len));
shun-iwasawa 7bad49
    if ((p == NULL) && (info.name.empty())) {
shun-iwasawa 7bad49
      // Buffer overrun. Issue #51.
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    const unsigned char *data_end =
shun-iwasawa 7bad49
        reinterpret_cast<const unsigned char *>(p) + 16;
shun-iwasawa 7bad49
    if (data_end >= (data.data() + data.size())) {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    memcpy(&info.pixel_type, p, sizeof(int));
shun-iwasawa 7bad49
    p += 4;
shun-iwasawa 7bad49
    info.p_linear = static_cast<unsigned char>(p[0]);  // uchar
shun-iwasawa 7bad49
    p += 1 + 3;                                        // reserved: uchar[3]
shun-iwasawa 7bad49
    memcpy(&info.x_sampling, p, sizeof(int));          // int
shun-iwasawa 7bad49
    p += 4;
shun-iwasawa 7bad49
    memcpy(&info.y_sampling, p, sizeof(int));  // int
shun-iwasawa 7bad49
    p += 4;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tinyexr::swap4(&info.pixel_type);
shun-iwasawa 7bad49
    tinyexr::swap4(&info.x_sampling);
shun-iwasawa 7bad49
    tinyexr::swap4(&info.y_sampling);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    channels.push_back(info);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void WriteChannelInfo(std::vector<unsigned char> &data,
shun-iwasawa 7bad49
                             const std::vector<ChannelInfo> &channels) {
shun-iwasawa 7bad49
  size_t sz = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Calculate total size.
shun-iwasawa 7bad49
  for (size_t c = 0; c < channels.size(); c++) {
shun-iwasawa 7bad49
    sz += channels[c].name.length() + 1;  // +1 for \0
shun-iwasawa 7bad49
    sz += 16;                                    // 4 * int
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  data.resize(sz + 1);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  unsigned char *p = &data.at(0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t c = 0; c < channels.size(); c++) {
shun-iwasawa 7bad49
    memcpy(p, channels[c].name.c_str(), channels[c].name.length());
shun-iwasawa 7bad49
    p += channels[c].name.length();
shun-iwasawa 7bad49
    (*p) = '\0';
shun-iwasawa 7bad49
    p++;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int pixel_type = channels[c].requested_pixel_type;
shun-iwasawa 7bad49
    int x_sampling = channels[c].x_sampling;
shun-iwasawa 7bad49
    int y_sampling = channels[c].y_sampling;
shun-iwasawa 7bad49
    tinyexr::swap4(&pixel_type);
shun-iwasawa 7bad49
    tinyexr::swap4(&x_sampling);
shun-iwasawa 7bad49
    tinyexr::swap4(&y_sampling);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    memcpy(p, &pixel_type, sizeof(int));
shun-iwasawa 7bad49
    p += sizeof(int);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    (*p) = channels[c].p_linear;
shun-iwasawa 7bad49
    p += 4;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    memcpy(p, &x_sampling, sizeof(int));
shun-iwasawa 7bad49
    p += sizeof(int);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    memcpy(p, &y_sampling, sizeof(int));
shun-iwasawa 7bad49
    p += sizeof(int);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  (*p) = '\0';
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void CompressZip(unsigned char *dst,
shun-iwasawa 7bad49
                        tinyexr::tinyexr_uint64 &compressedSize,
shun-iwasawa 7bad49
                        const unsigned char *src, unsigned long src_size) {
shun-iwasawa 7bad49
  std::vector<unsigned char> tmpBuf(src_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Apply EXR-specific? postprocess. Grabbed from OpenEXR's
shun-iwasawa 7bad49
  // ImfZipCompressor.cpp
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Reorder the pixel data.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const char *srcPtr = reinterpret_cast<const char *>(src);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    char *t1 = reinterpret_cast<char *>(&tmpBuf.at(0));
shun-iwasawa 7bad49
    char *t2 = reinterpret_cast<char *>(&tmpBuf.at(0)) + (src_size + 1) / 2;
shun-iwasawa 7bad49
    const char *stop = srcPtr + src_size;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (;;) {
shun-iwasawa 7bad49
      if (srcPtr < stop)
shun-iwasawa 7bad49
        *(t1++) = *(srcPtr++);
shun-iwasawa 7bad49
      else
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (srcPtr < stop)
shun-iwasawa 7bad49
        *(t2++) = *(srcPtr++);
shun-iwasawa 7bad49
      else
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Predictor.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    unsigned char *t = &tmpBuf.at(0) + 1;
shun-iwasawa 7bad49
    unsigned char *stop = &tmpBuf.at(0) + src_size;
shun-iwasawa 7bad49
    int p = t[-1];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    while (t < stop) {
shun-iwasawa 7bad49
      int d = int(t[0]) - p + (128 + 256);
shun-iwasawa 7bad49
      p = t[0];
shun-iwasawa 7bad49
      t[0] = static_cast<unsigned char>(d);
shun-iwasawa 7bad49
      ++t;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_MINIZ
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Compress the data using miniz
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  mz_ulong outSize = mz_compressBound(src_size);
shun-iwasawa 7bad49
  int ret = mz_compress(
shun-iwasawa 7bad49
      dst, &outSize, static_cast<const unsigned char *>(&tmpBuf.at(0)),
shun-iwasawa 7bad49
      src_size);
shun-iwasawa 7bad49
  assert(ret == MZ_OK);
shun-iwasawa 7bad49
  (void)ret;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  compressedSize = outSize;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  uLong outSize = compressBound(static_cast<uLong>(src_size));
shun-iwasawa 7bad49
  int ret = compress(dst, &outSize, static_cast<const Bytef *>(&tmpBuf.at(0)),
shun-iwasawa 7bad49
                     src_size);
shun-iwasawa 7bad49
  assert(ret == Z_OK);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  compressedSize = outSize;
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Use uncompressed data when compressed data is larger than uncompressed.
shun-iwasawa 7bad49
  // (Issue 40)
shun-iwasawa 7bad49
  if (compressedSize >= src_size) {
shun-iwasawa 7bad49
    compressedSize = src_size;
shun-iwasawa 7bad49
    memcpy(dst, src, src_size);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool DecompressZip(unsigned char *dst,
shun-iwasawa 7bad49
                          unsigned long *uncompressed_size /* inout */,
shun-iwasawa 7bad49
                          const unsigned char *src, unsigned long src_size) {
shun-iwasawa 7bad49
  if ((*uncompressed_size) == src_size) {
shun-iwasawa 7bad49
    // Data is not compressed(Issue 40).
shun-iwasawa 7bad49
    memcpy(dst, src, src_size);
shun-iwasawa 7bad49
    return true;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  std::vector<unsigned char> tmpBuf(*uncompressed_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_MINIZ
shun-iwasawa 7bad49
  int ret =
shun-iwasawa 7bad49
      mz_uncompress(&tmpBuf.at(0), uncompressed_size, src, src_size);
shun-iwasawa 7bad49
  if (MZ_OK != ret) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  int ret = uncompress(&tmpBuf.at(0), uncompressed_size, src, src_size);
shun-iwasawa 7bad49
  if (Z_OK != ret) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Apply EXR-specific? postprocess. Grabbed from OpenEXR's
shun-iwasawa 7bad49
  // ImfZipCompressor.cpp
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Predictor.
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    unsigned char *t = &tmpBuf.at(0) + 1;
shun-iwasawa 7bad49
    unsigned char *stop = &tmpBuf.at(0) + (*uncompressed_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    while (t < stop) {
shun-iwasawa 7bad49
      int d = int(t[-1]) + int(t[0]) - 128;
shun-iwasawa 7bad49
      t[0] = static_cast<unsigned char>(d);
shun-iwasawa 7bad49
      ++t;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Reorder the pixel data.
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    const char *t1 = reinterpret_cast<const char *>(&tmpBuf.at(0));
shun-iwasawa 7bad49
    const char *t2 = reinterpret_cast<const char *>(&tmpBuf.at(0)) +
shun-iwasawa 7bad49
                     (*uncompressed_size + 1) / 2;
shun-iwasawa 7bad49
    char *s = reinterpret_cast<char *>(dst);
shun-iwasawa 7bad49
    char *stop = s + (*uncompressed_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (;;) {
shun-iwasawa 7bad49
      if (s < stop)
shun-iwasawa 7bad49
        *(s++) = *(t1++);
shun-iwasawa 7bad49
      else
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (s < stop)
shun-iwasawa 7bad49
        *(s++) = *(t2++);
shun-iwasawa 7bad49
      else
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// RLE code from OpenEXR --------------------------------------
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic push
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wsign-conversion"
shun-iwasawa 7bad49
#if __has_warning("-Wextra-semi-stmt")
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wextra-semi-stmt"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef _MSC_VER
shun-iwasawa 7bad49
#pragma warning(push)
shun-iwasawa 7bad49
#pragma warning(disable : 4204)  // nonstandard extension used : non-constant
shun-iwasawa 7bad49
                                 // aggregate initializer (also supported by GNU
shun-iwasawa 7bad49
                                 // C and C99, so no big deal)
shun-iwasawa 7bad49
#pragma warning(disable : 4244)  // 'initializing': conversion from '__int64' to
shun-iwasawa 7bad49
                                 // 'int', possible loss of data
shun-iwasawa 7bad49
#pragma warning(disable : 4267)  // 'argument': conversion from '__int64' to
shun-iwasawa 7bad49
                                 // 'int', possible loss of data
shun-iwasawa 7bad49
#pragma warning(disable : 4996)  // 'strdup': The POSIX name for this item is
shun-iwasawa 7bad49
                                 // deprecated. Instead, use the ISO C and C++
shun-iwasawa 7bad49
                                 // conformant name: _strdup.
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
const int MIN_RUN_LENGTH = 3;
shun-iwasawa 7bad49
const int MAX_RUN_LENGTH = 127;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Compress an array of bytes, using run-length encoding,
shun-iwasawa 7bad49
// and return the length of the compressed data.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int rleCompress(int inLength, const char in[], signed char out[]) {
shun-iwasawa 7bad49
  const char *inEnd = in + inLength;
shun-iwasawa 7bad49
  const char *runStart = in;
shun-iwasawa 7bad49
  const char *runEnd = in + 1;
shun-iwasawa 7bad49
  signed char *outWrite = out;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (runStart < inEnd) {
shun-iwasawa 7bad49
    while (runEnd < inEnd && *runStart == *runEnd &&
shun-iwasawa 7bad49
           runEnd - runStart - 1 < MAX_RUN_LENGTH) {
shun-iwasawa 7bad49
      ++runEnd;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (runEnd - runStart >= MIN_RUN_LENGTH) {
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
      // Compressible run
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      *outWrite++ = static_cast<char>(runEnd - runStart) - 1;
shun-iwasawa 7bad49
      *outWrite++ = *(reinterpret_cast<const signed char *>(runStart));
shun-iwasawa 7bad49
      runStart = runEnd;
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
      // Uncompressable run
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      while (runEnd < inEnd &&
shun-iwasawa 7bad49
             ((runEnd + 1 >= inEnd || *runEnd != *(runEnd + 1)) ||
shun-iwasawa 7bad49
              (runEnd + 2 >= inEnd || *(runEnd + 1) != *(runEnd + 2))) &&
shun-iwasawa 7bad49
             runEnd - runStart < MAX_RUN_LENGTH) {
shun-iwasawa 7bad49
        ++runEnd;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      *outWrite++ = static_cast<char>(runStart - runEnd);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      while (runStart < runEnd) {
shun-iwasawa 7bad49
        *outWrite++ = *(reinterpret_cast<const signed char *>(runStart++));
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    ++runEnd;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return static_cast<int>(outWrite - out);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Uncompress an array of bytes compressed with rleCompress().
shun-iwasawa 7bad49
// Returns the length of the oncompressed data, or 0 if the
shun-iwasawa 7bad49
// length of the uncompressed data would be more than maxLength.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int rleUncompress(int inLength, int maxLength, const signed char in[],
shun-iwasawa 7bad49
                         char out[]) {
shun-iwasawa 7bad49
  char *outStart = out;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (inLength > 0) {
shun-iwasawa 7bad49
    if (*in < 0) {
shun-iwasawa 7bad49
      int count = -(static_cast<int>(*in++));
shun-iwasawa 7bad49
      inLength -= count + 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      // Fixes #116: Add bounds check to in buffer.
shun-iwasawa 7bad49
      if ((0 > (maxLength -= count)) || (inLength < 0)) return 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      memcpy(out, in, count);
shun-iwasawa 7bad49
      out += count;
shun-iwasawa 7bad49
      in += count;
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      int count = *in++;
shun-iwasawa 7bad49
      inLength -= 2;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (0 > (maxLength -= count + 1)) return 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      memset(out, *reinterpret_cast<const char *>(in), count + 1);
shun-iwasawa 7bad49
      out += count + 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      in++;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return static_cast<int>(out - outStart);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic pop
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// End of RLE code from OpenEXR -----------------------------------
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void CompressRle(unsigned char *dst,
shun-iwasawa 7bad49
                        tinyexr::tinyexr_uint64 &compressedSize,
shun-iwasawa 7bad49
                        const unsigned char *src, unsigned long src_size) {
shun-iwasawa 7bad49
  std::vector<unsigned char> tmpBuf(src_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Apply EXR-specific? postprocess. Grabbed from OpenEXR's
shun-iwasawa 7bad49
  // ImfRleCompressor.cpp
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Reorder the pixel data.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const char *srcPtr = reinterpret_cast<const char *>(src);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    char *t1 = reinterpret_cast<char *>(&tmpBuf.at(0));
shun-iwasawa 7bad49
    char *t2 = reinterpret_cast<char *>(&tmpBuf.at(0)) + (src_size + 1) / 2;
shun-iwasawa 7bad49
    const char *stop = srcPtr + src_size;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (;;) {
shun-iwasawa 7bad49
      if (srcPtr < stop)
shun-iwasawa 7bad49
        *(t1++) = *(srcPtr++);
shun-iwasawa 7bad49
      else
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (srcPtr < stop)
shun-iwasawa 7bad49
        *(t2++) = *(srcPtr++);
shun-iwasawa 7bad49
      else
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Predictor.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    unsigned char *t = &tmpBuf.at(0) + 1;
shun-iwasawa 7bad49
    unsigned char *stop = &tmpBuf.at(0) + src_size;
shun-iwasawa 7bad49
    int p = t[-1];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    while (t < stop) {
shun-iwasawa 7bad49
      int d = int(t[0]) - p + (128 + 256);
shun-iwasawa 7bad49
      p = t[0];
shun-iwasawa 7bad49
      t[0] = static_cast<unsigned char>(d);
shun-iwasawa 7bad49
      ++t;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // outSize will be (srcSiz * 3) / 2 at max.
shun-iwasawa 7bad49
  int outSize = rleCompress(static_cast<int>(src_size),
shun-iwasawa 7bad49
                            reinterpret_cast<const char *>(&tmpBuf.at(0)),
shun-iwasawa 7bad49
                            reinterpret_cast<signed char *>(dst));
shun-iwasawa 7bad49
  assert(outSize > 0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  compressedSize = static_cast<tinyexr::tinyexr_uint64>(outSize);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Use uncompressed data when compressed data is larger than uncompressed.
shun-iwasawa 7bad49
  // (Issue 40)
shun-iwasawa 7bad49
  if (compressedSize >= src_size) {
shun-iwasawa 7bad49
    compressedSize = src_size;
shun-iwasawa 7bad49
    memcpy(dst, src, src_size);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool DecompressRle(unsigned char *dst,
shun-iwasawa 7bad49
                          const unsigned long uncompressed_size,
shun-iwasawa 7bad49
                          const unsigned char *src, unsigned long src_size) {
shun-iwasawa 7bad49
  if (uncompressed_size == src_size) {
shun-iwasawa 7bad49
    // Data is not compressed(Issue 40).
shun-iwasawa 7bad49
    memcpy(dst, src, src_size);
shun-iwasawa 7bad49
    return true;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Workaround for issue #112.
shun-iwasawa 7bad49
  // TODO(syoyo): Add more robust out-of-bounds check in `rleUncompress`.
shun-iwasawa 7bad49
  if (src_size <= 2) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned char> tmpBuf(uncompressed_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int ret = rleUncompress(static_cast<int>(src_size),
shun-iwasawa 7bad49
                          static_cast<int>(uncompressed_size),
shun-iwasawa 7bad49
                          reinterpret_cast<const signed char *>(src),
shun-iwasawa 7bad49
                          reinterpret_cast<char *>(&tmpBuf.at(0)));
shun-iwasawa 7bad49
  if (ret != static_cast<int>(uncompressed_size)) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Apply EXR-specific? postprocess. Grabbed from OpenEXR's
shun-iwasawa 7bad49
  // ImfRleCompressor.cpp
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Predictor.
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    unsigned char *t = &tmpBuf.at(0) + 1;
shun-iwasawa 7bad49
    unsigned char *stop = &tmpBuf.at(0) + uncompressed_size;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    while (t < stop) {
shun-iwasawa 7bad49
      int d = int(t[-1]) + int(t[0]) - 128;
shun-iwasawa 7bad49
      t[0] = static_cast<unsigned char>(d);
shun-iwasawa 7bad49
      ++t;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Reorder the pixel data.
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    const char *t1 = reinterpret_cast<const char *>(&tmpBuf.at(0));
shun-iwasawa 7bad49
    const char *t2 = reinterpret_cast<const char *>(&tmpBuf.at(0)) +
shun-iwasawa 7bad49
                     (uncompressed_size + 1) / 2;
shun-iwasawa 7bad49
    char *s = reinterpret_cast<char *>(dst);
shun-iwasawa 7bad49
    char *stop = s + uncompressed_size;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (;;) {
shun-iwasawa 7bad49
      if (s < stop)
shun-iwasawa 7bad49
        *(s++) = *(t1++);
shun-iwasawa 7bad49
      else
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (s < stop)
shun-iwasawa 7bad49
        *(s++) = *(t2++);
shun-iwasawa 7bad49
      else
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_PIZ
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic push
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wc++11-long-long"
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wold-style-cast"
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wpadded"
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wsign-conversion"
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wc++11-extensions"
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wconversion"
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wc++98-compat-pedantic"
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if __has_warning("-Wcast-qual")
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wcast-qual"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if __has_warning("-Wextra-semi-stmt")
shun-iwasawa 7bad49
#pragma clang diagnostic ignored "-Wextra-semi-stmt"
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// PIZ compress/uncompress, based on OpenEXR's ImfPizCompressor.cpp
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// -----------------------------------------------------------------
shun-iwasawa 7bad49
// Copyright (c) 2004, Industrial Light & Magic, a division of Lucas
shun-iwasawa 7bad49
// Digital Ltd. LLC)
shun-iwasawa 7bad49
// (3 clause BSD license)
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
struct PIZChannelData {
shun-iwasawa 7bad49
  unsigned short *start;
shun-iwasawa 7bad49
  unsigned short *end;
shun-iwasawa 7bad49
  int nx;
shun-iwasawa 7bad49
  int ny;
shun-iwasawa 7bad49
  int ys;
shun-iwasawa 7bad49
  int size;
shun-iwasawa 7bad49
};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//-----------------------------------------------------------------------------
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//  16-bit Haar Wavelet encoding and decoding
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//  The source code in this file is derived from the encoding
shun-iwasawa 7bad49
//  and decoding routines written by Christian Rouet for his
shun-iwasawa 7bad49
//  PIZ image file format.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//-----------------------------------------------------------------------------
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Wavelet basis functions without modulo arithmetic; they produce
shun-iwasawa 7bad49
// the best compression ratios when the wavelet-transformed data are
shun-iwasawa 7bad49
// Huffman-encoded, but the wavelet transform works only for 14-bit
shun-iwasawa 7bad49
// data (untransformed data values must be less than (1 << 14)).
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline void wenc14(unsigned short a, unsigned short b, unsigned short &l,
shun-iwasawa 7bad49
                   unsigned short &h) {
shun-iwasawa 7bad49
  short as = static_cast<short>(a);
shun-iwasawa 7bad49
  short bs = static_cast<short>(b);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  short ms = (as + bs) >> 1;
shun-iwasawa 7bad49
  short ds = as - bs;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  l = static_cast<unsigned short>(ms);
shun-iwasawa 7bad49
  h = static_cast<unsigned short>(ds);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline void wdec14(unsigned short l, unsigned short h, unsigned short &a,
shun-iwasawa 7bad49
                   unsigned short &b) {
shun-iwasawa 7bad49
  short ls = static_cast<short>(l);
shun-iwasawa 7bad49
  short hs = static_cast<short>(h);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int hi = hs;
shun-iwasawa 7bad49
  int ai = ls + (hi & 1) + (hi >> 1);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  short as = static_cast<short>(ai);
shun-iwasawa 7bad49
  short bs = static_cast<short>(ai - hi);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  a = static_cast<unsigned short>(as);
shun-iwasawa 7bad49
  b = static_cast<unsigned short>(bs);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Wavelet basis functions with modulo arithmetic; they work with full
shun-iwasawa 7bad49
// 16-bit data, but Huffman-encoding the wavelet-transformed data doesn't
shun-iwasawa 7bad49
// compress the data quite as well.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
const int NBITS = 16;
shun-iwasawa 7bad49
const int A_OFFSET = 1 << (NBITS - 1);
shun-iwasawa 7bad49
const int M_OFFSET = 1 << (NBITS - 1);
shun-iwasawa 7bad49
const int MOD_MASK = (1 << NBITS) - 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline void wenc16(unsigned short a, unsigned short b, unsigned short &l,
shun-iwasawa 7bad49
                   unsigned short &h) {
shun-iwasawa 7bad49
  int ao = (a + A_OFFSET) & MOD_MASK;
shun-iwasawa 7bad49
  int m = ((ao + b) >> 1);
shun-iwasawa 7bad49
  int d = ao - b;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (d < 0) m = (m + M_OFFSET) & MOD_MASK;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  d &= MOD_MASK;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  l = static_cast<unsigned short>(m);
shun-iwasawa 7bad49
  h = static_cast<unsigned short>(d);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline void wdec16(unsigned short l, unsigned short h, unsigned short &a,
shun-iwasawa 7bad49
                   unsigned short &b) {
shun-iwasawa 7bad49
  int m = l;
shun-iwasawa 7bad49
  int d = h;
shun-iwasawa 7bad49
  int bb = (m - (d >> 1)) & MOD_MASK;
shun-iwasawa 7bad49
  int aa = (d + bb - A_OFFSET) & MOD_MASK;
shun-iwasawa 7bad49
  b = static_cast<unsigned short>(bb);
shun-iwasawa 7bad49
  a = static_cast<unsigned short>(aa);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// 2D Wavelet encoding:
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void wav2Encode(
shun-iwasawa 7bad49
    unsigned short *in,  // io: values are transformed in place
shun-iwasawa 7bad49
    int nx,              // i : x size
shun-iwasawa 7bad49
    int ox,              // i : x offset
shun-iwasawa 7bad49
    int ny,              // i : y size
shun-iwasawa 7bad49
    int oy,              // i : y offset
shun-iwasawa 7bad49
    unsigned short mx)   // i : maximum in[x][y] value
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  bool w14 = (mx < (1 << 14));
shun-iwasawa 7bad49
  int n = (nx > ny) ? ny : nx;
shun-iwasawa 7bad49
  int p = 1;   // == 1 <<  level
shun-iwasawa 7bad49
  int p2 = 2;  // == 1 << (level+1)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Hierarchical loop on smaller dimension n
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (p2 <= n) {
shun-iwasawa 7bad49
    unsigned short *py = in;
shun-iwasawa 7bad49
    unsigned short *ey = in + oy * (ny - p2);
shun-iwasawa 7bad49
    int oy1 = oy * p;
shun-iwasawa 7bad49
    int oy2 = oy * p2;
shun-iwasawa 7bad49
    int ox1 = ox * p;
shun-iwasawa 7bad49
    int ox2 = ox * p2;
shun-iwasawa 7bad49
    unsigned short i00, i01, i10, i11;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Y loop
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (; py <= ey; py += oy2) {
shun-iwasawa 7bad49
      unsigned short *px = py;
shun-iwasawa 7bad49
      unsigned short *ex = py + ox * (nx - p2);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
      // X loop
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      for (; px <= ex; px += ox2) {
shun-iwasawa 7bad49
        unsigned short *p01 = px + ox1;
shun-iwasawa 7bad49
        unsigned short *p10 = px + oy1;
shun-iwasawa 7bad49
        unsigned short *p11 = p10 + ox1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
        // 2D wavelet encoding
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (w14) {
shun-iwasawa 7bad49
          wenc14(*px, *p01, i00, i01);
shun-iwasawa 7bad49
          wenc14(*p10, *p11, i10, i11);
shun-iwasawa 7bad49
          wenc14(i00, i10, *px, *p10);
shun-iwasawa 7bad49
          wenc14(i01, i11, *p01, *p11);
shun-iwasawa 7bad49
        } else {
shun-iwasawa 7bad49
          wenc16(*px, *p01, i00, i01);
shun-iwasawa 7bad49
          wenc16(*p10, *p11, i10, i11);
shun-iwasawa 7bad49
          wenc16(i00, i10, *px, *p10);
shun-iwasawa 7bad49
          wenc16(i01, i11, *p01, *p11);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
      // Encode (1D) odd column (still in Y loop)
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (nx & p) {
shun-iwasawa 7bad49
        unsigned short *p10 = px + oy1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (w14)
shun-iwasawa 7bad49
          wenc14(*px, *p10, i00, *p10);
shun-iwasawa 7bad49
        else
shun-iwasawa 7bad49
          wenc16(*px, *p10, i00, *p10);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        *px = i00;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Encode (1D) odd line (must loop in X)
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (ny & p) {
shun-iwasawa 7bad49
      unsigned short *px = py;
shun-iwasawa 7bad49
      unsigned short *ex = py + ox * (nx - p2);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      for (; px <= ex; px += ox2) {
shun-iwasawa 7bad49
        unsigned short *p01 = px + ox1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (w14)
shun-iwasawa 7bad49
          wenc14(*px, *p01, i00, *p01);
shun-iwasawa 7bad49
        else
shun-iwasawa 7bad49
          wenc16(*px, *p01, i00, *p01);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        *px = i00;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Next level
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    p = p2;
shun-iwasawa 7bad49
    p2 <<= 1;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// 2D Wavelet decoding:
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void wav2Decode(
shun-iwasawa 7bad49
    unsigned short *in,  // io: values are transformed in place
shun-iwasawa 7bad49
    int nx,              // i : x size
shun-iwasawa 7bad49
    int ox,              // i : x offset
shun-iwasawa 7bad49
    int ny,              // i : y size
shun-iwasawa 7bad49
    int oy,              // i : y offset
shun-iwasawa 7bad49
    unsigned short mx)   // i : maximum in[x][y] value
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  bool w14 = (mx < (1 << 14));
shun-iwasawa 7bad49
  int n = (nx > ny) ? ny : nx;
shun-iwasawa 7bad49
  int p = 1;
shun-iwasawa 7bad49
  int p2;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Search max level
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (p <= n) p <<= 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  p >>= 1;
shun-iwasawa 7bad49
  p2 = p;
shun-iwasawa 7bad49
  p >>= 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Hierarchical loop on smaller dimension n
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (p >= 1) {
shun-iwasawa 7bad49
    unsigned short *py = in;
shun-iwasawa 7bad49
    unsigned short *ey = in + oy * (ny - p2);
shun-iwasawa 7bad49
    int oy1 = oy * p;
shun-iwasawa 7bad49
    int oy2 = oy * p2;
shun-iwasawa 7bad49
    int ox1 = ox * p;
shun-iwasawa 7bad49
    int ox2 = ox * p2;
shun-iwasawa 7bad49
    unsigned short i00, i01, i10, i11;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Y loop
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (; py <= ey; py += oy2) {
shun-iwasawa 7bad49
      unsigned short *px = py;
shun-iwasawa 7bad49
      unsigned short *ex = py + ox * (nx - p2);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
      // X loop
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      for (; px <= ex; px += ox2) {
shun-iwasawa 7bad49
        unsigned short *p01 = px + ox1;
shun-iwasawa 7bad49
        unsigned short *p10 = px + oy1;
shun-iwasawa 7bad49
        unsigned short *p11 = p10 + ox1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
        // 2D wavelet decoding
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (w14) {
shun-iwasawa 7bad49
          wdec14(*px, *p10, i00, i10);
shun-iwasawa 7bad49
          wdec14(*p01, *p11, i01, i11);
shun-iwasawa 7bad49
          wdec14(i00, i01, *px, *p01);
shun-iwasawa 7bad49
          wdec14(i10, i11, *p10, *p11);
shun-iwasawa 7bad49
        } else {
shun-iwasawa 7bad49
          wdec16(*px, *p10, i00, i10);
shun-iwasawa 7bad49
          wdec16(*p01, *p11, i01, i11);
shun-iwasawa 7bad49
          wdec16(i00, i01, *px, *p01);
shun-iwasawa 7bad49
          wdec16(i10, i11, *p10, *p11);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
      // Decode (1D) odd column (still in Y loop)
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (nx & p) {
shun-iwasawa 7bad49
        unsigned short *p10 = px + oy1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (w14)
shun-iwasawa 7bad49
          wdec14(*px, *p10, i00, *p10);
shun-iwasawa 7bad49
        else
shun-iwasawa 7bad49
          wdec16(*px, *p10, i00, *p10);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        *px = i00;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Decode (1D) odd line (must loop in X)
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (ny & p) {
shun-iwasawa 7bad49
      unsigned short *px = py;
shun-iwasawa 7bad49
      unsigned short *ex = py + ox * (nx - p2);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      for (; px <= ex; px += ox2) {
shun-iwasawa 7bad49
        unsigned short *p01 = px + ox1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (w14)
shun-iwasawa 7bad49
          wdec14(*px, *p01, i00, *p01);
shun-iwasawa 7bad49
        else
shun-iwasawa 7bad49
          wdec16(*px, *p01, i00, *p01);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        *px = i00;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Next level
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    p2 = p;
shun-iwasawa 7bad49
    p >>= 1;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//-----------------------------------------------------------------------------
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//  16-bit Huffman compression and decompression.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//  The source code in this file is derived from the 8-bit
shun-iwasawa 7bad49
//  Huffman compression and decompression routines written
shun-iwasawa 7bad49
//  by Christian Rouet for his PIZ image file format.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//-----------------------------------------------------------------------------
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Adds some modification for tinyexr.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
const int HUF_ENCBITS = 16;  // literal (value) bit length
shun-iwasawa 7bad49
const int HUF_DECBITS = 14;  // decoding bit size (>= 8)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
const int HUF_ENCSIZE = (1 << HUF_ENCBITS) + 1;  // encoding table size
shun-iwasawa 7bad49
const int HUF_DECSIZE = 1 << HUF_DECBITS;        // decoding table size
shun-iwasawa 7bad49
const int HUF_DECMASK = HUF_DECSIZE - 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
struct HufDec {  // short code    long code
shun-iwasawa 7bad49
  //-------------------------------
shun-iwasawa 7bad49
  unsigned int len : 8;   // code length    0
shun-iwasawa 7bad49
  unsigned int lit : 24;  // lit      p size
shun-iwasawa 7bad49
  unsigned int *p;        // 0      lits
shun-iwasawa 7bad49
};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline long long hufLength(long long code) { return code & 63; }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline long long hufCode(long long code) { return code >> 6; }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline void outputBits(int nBits, long long bits, long long &c, int &lc,
shun-iwasawa 7bad49
                       char *&out) {
shun-iwasawa 7bad49
  c <<= nBits;
shun-iwasawa 7bad49
  lc += nBits;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  c |= bits;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (lc >= 8) *out++ = static_cast<char>((c >> (lc -= 8)));
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline long long getBits(int nBits, long long &c, int &lc, const char *&in) {
shun-iwasawa 7bad49
  while (lc < nBits) {
shun-iwasawa 7bad49
    c = (c << 8) | *(reinterpret_cast<const unsigned char *>(in++));
shun-iwasawa 7bad49
    lc += 8;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  lc -= nBits;
shun-iwasawa 7bad49
  return (c >> lc) & ((1 << nBits) - 1);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// ENCODING TABLE BUILDING & (UN)PACKING
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Build a "canonical" Huffman code table:
shun-iwasawa 7bad49
//  - for each (uncompressed) symbol, hcode contains the length
shun-iwasawa 7bad49
//    of the corresponding code (in the compressed data)
shun-iwasawa 7bad49
//  - canonical codes are computed and stored in hcode
shun-iwasawa 7bad49
//  - the rules for constructing canonical codes are as follows:
shun-iwasawa 7bad49
//    * shorter codes (if filled with zeroes to the right)
shun-iwasawa 7bad49
//      have a numerically higher value than longer codes
shun-iwasawa 7bad49
//    * for codes with the same length, numerical values
shun-iwasawa 7bad49
//      increase with numerical symbol values
shun-iwasawa 7bad49
//  - because the canonical code table can be constructed from
shun-iwasawa 7bad49
//    symbol lengths alone, the code table can be transmitted
shun-iwasawa 7bad49
//    without sending the actual code values
shun-iwasawa 7bad49
//  - see http://www.compressconsult.com/huffman/
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void hufCanonicalCodeTable(long long hcode[HUF_ENCSIZE]) {
shun-iwasawa 7bad49
  long long n[59];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // For each i from 0 through 58, count the
shun-iwasawa 7bad49
  // number of different codes of length i, and
shun-iwasawa 7bad49
  // store the count in n[i].
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i <= 58; ++i) n[i] = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < HUF_ENCSIZE; ++i) n[hcode[i]] += 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // For each i from 58 through 1, compute the
shun-iwasawa 7bad49
  // numerically lowest code with length i, and
shun-iwasawa 7bad49
  // store that code in n[i].
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  long long c = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 58; i > 0; --i) {
shun-iwasawa 7bad49
    long long nc = ((c + n[i]) >> 1);
shun-iwasawa 7bad49
    n[i] = c;
shun-iwasawa 7bad49
    c = nc;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // hcode[i] contains the length, l, of the
shun-iwasawa 7bad49
  // code for symbol i.  Assign the next available
shun-iwasawa 7bad49
  // code of length l to the symbol and store both
shun-iwasawa 7bad49
  // l and the code in hcode[i].
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < HUF_ENCSIZE; ++i) {
shun-iwasawa 7bad49
    int l = static_cast<int>(hcode[i]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (l > 0) hcode[i] = l | (n[l]++ << 6);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Compute Huffman codes (based on frq input) and store them in frq:
shun-iwasawa 7bad49
//  - code structure is : [63:lsb - 6:msb] | [5-0: bit length];
shun-iwasawa 7bad49
//  - max code length is 58 bits;
shun-iwasawa 7bad49
//  - codes outside the range [im-iM] have a null length (unused values);
shun-iwasawa 7bad49
//  - original frequencies are destroyed;
shun-iwasawa 7bad49
//  - encoding tables are used by hufEncode() and hufBuildDecTable();
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
struct FHeapCompare {
shun-iwasawa 7bad49
  bool operator()(long long *a, long long *b) { return *a > *b; }
shun-iwasawa 7bad49
};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void hufBuildEncTable(
shun-iwasawa 7bad49
    long long *frq,  // io: input frequencies [HUF_ENCSIZE], output table
shun-iwasawa 7bad49
    int *im,         //  o: min frq index
shun-iwasawa 7bad49
    int *iM)         //  o: max frq index
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // This function assumes that when it is called, array frq
shun-iwasawa 7bad49
  // indicates the frequency of all possible symbols in the data
shun-iwasawa 7bad49
  // that are to be Huffman-encoded.  (frq[i] contains the number
shun-iwasawa 7bad49
  // of occurrences of symbol i in the data.)
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // The loop below does three things:
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // 1) Finds the minimum and maximum indices that point
shun-iwasawa 7bad49
  //    to non-zero entries in frq:
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  //     frq[im] != 0, and frq[i] == 0 for all i < im
shun-iwasawa 7bad49
  //     frq[iM] != 0, and frq[i] == 0 for all i > iM
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // 2) Fills array fHeap with pointers to all non-zero
shun-iwasawa 7bad49
  //    entries in frq.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // 3) Initializes array hlink such that hlink[i] == i
shun-iwasawa 7bad49
  //    for all array entries.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<int> hlink(HUF_ENCSIZE);
shun-iwasawa 7bad49
  std::vector<long long *> fHeap(HUF_ENCSIZE);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  *im = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (!frq[*im]) (*im)++;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int nf = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = *im; i < HUF_ENCSIZE; i++) {
shun-iwasawa 7bad49
    hlink[i] = i;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (frq[i]) {
shun-iwasawa 7bad49
      fHeap[nf] = &frq[i];
shun-iwasawa 7bad49
      nf++;
shun-iwasawa 7bad49
      *iM = i;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Add a pseudo-symbol, with a frequency count of 1, to frq;
shun-iwasawa 7bad49
  // adjust the fHeap and hlink array accordingly.  Function
shun-iwasawa 7bad49
  // hufEncode() uses the pseudo-symbol for run-length encoding.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  (*iM)++;
shun-iwasawa 7bad49
  frq[*iM] = 1;
shun-iwasawa 7bad49
  fHeap[nf] = &frq[*iM];
shun-iwasawa 7bad49
  nf++;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Build an array, scode, such that scode[i] contains the number
shun-iwasawa 7bad49
  // of bits assigned to symbol i.  Conceptually this is done by
shun-iwasawa 7bad49
  // constructing a tree whose leaves are the symbols with non-zero
shun-iwasawa 7bad49
  // frequency:
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  //     Make a heap that contains all symbols with a non-zero frequency,
shun-iwasawa 7bad49
  //     with the least frequent symbol on top.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  //     Repeat until only one symbol is left on the heap:
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  //         Take the two least frequent symbols off the top of the heap.
shun-iwasawa 7bad49
  //         Create a new node that has first two nodes as children, and
shun-iwasawa 7bad49
  //         whose frequency is the sum of the frequencies of the first
shun-iwasawa 7bad49
  //         two nodes.  Put the new node back into the heap.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // The last node left on the heap is the root of the tree.  For each
shun-iwasawa 7bad49
  // leaf node, the distance between the root and the leaf is the length
shun-iwasawa 7bad49
  // of the code for the corresponding symbol.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // The loop below doesn't actually build the tree; instead we compute
shun-iwasawa 7bad49
  // the distances of the leaves from the root on the fly.  When a new
shun-iwasawa 7bad49
  // node is added to the heap, then that node's descendants are linked
shun-iwasawa 7bad49
  // into a single linear list that starts at the new node, and the code
shun-iwasawa 7bad49
  // lengths of the descendants (that is, their distance from the root
shun-iwasawa 7bad49
  // of the tree) are incremented by one.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::make_heap(&fHeap[0], &fHeap[nf], FHeapCompare());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<long long> scode(HUF_ENCSIZE);
shun-iwasawa 7bad49
  memset(scode.data(), 0, sizeof(long long) * HUF_ENCSIZE);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (nf > 1) {
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Find the indices, mm and m, of the two smallest non-zero frq
shun-iwasawa 7bad49
    // values in fHeap, add the smallest frq to the second-smallest
shun-iwasawa 7bad49
    // frq, and remove the smallest frq value from fHeap.
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int mm = fHeap[0] - frq;
shun-iwasawa 7bad49
    std::pop_heap(&fHeap[0], &fHeap[nf], FHeapCompare());
shun-iwasawa 7bad49
    --nf;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int m = fHeap[0] - frq;
shun-iwasawa 7bad49
    std::pop_heap(&fHeap[0], &fHeap[nf], FHeapCompare());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    frq[m] += frq[mm];
shun-iwasawa 7bad49
    std::push_heap(&fHeap[0], &fHeap[nf], FHeapCompare());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // The entries in scode are linked into lists with the
shun-iwasawa 7bad49
    // entries in hlink serving as "next" pointers and with
shun-iwasawa 7bad49
    // the end of a list marked by hlink[j] == j.
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Traverse the lists that start at scode[m] and scode[mm].
shun-iwasawa 7bad49
    // For each element visited, increment the length of the
shun-iwasawa 7bad49
    // corresponding code by one bit. (If we visit scode[j]
shun-iwasawa 7bad49
    // during the traversal, then the code for symbol j becomes
shun-iwasawa 7bad49
    // one bit longer.)
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Merge the lists that start at scode[m] and scode[mm]
shun-iwasawa 7bad49
    // into a single list that starts at scode[m].
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Add a bit to all codes in the first list.
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (int j = m;; j = hlink[j]) {
shun-iwasawa 7bad49
      scode[j]++;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      assert(scode[j] <= 58);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (hlink[j] == j) {
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
        // Merge the two lists.
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        hlink[j] = mm;
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Add a bit to all codes in the second list
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (int j = mm;; j = hlink[j]) {
shun-iwasawa 7bad49
      scode[j]++;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      assert(scode[j] <= 58);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (hlink[j] == j) break;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Build a canonical Huffman code table, replacing the code
shun-iwasawa 7bad49
  // lengths in scode with (code, code length) pairs.  Copy the
shun-iwasawa 7bad49
  // code table from scode into frq.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  hufCanonicalCodeTable(scode.data());
shun-iwasawa 7bad49
  memcpy(frq, scode.data(), sizeof(long long) * HUF_ENCSIZE);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Pack an encoding table:
shun-iwasawa 7bad49
//  - only code lengths, not actual codes, are stored
shun-iwasawa 7bad49
//  - runs of zeroes are compressed as follows:
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
//    unpacked    packed
shun-iwasawa 7bad49
//    --------------------------------
shun-iwasawa 7bad49
//    1 zero    0  (6 bits)
shun-iwasawa 7bad49
//    2 zeroes    59
shun-iwasawa 7bad49
//    3 zeroes    60
shun-iwasawa 7bad49
//    4 zeroes    61
shun-iwasawa 7bad49
//    5 zeroes    62
shun-iwasawa 7bad49
//    n zeroes (6 or more)  63 n-6  (6 + 8 bits)
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
const int SHORT_ZEROCODE_RUN = 59;
shun-iwasawa 7bad49
const int LONG_ZEROCODE_RUN = 63;
shun-iwasawa 7bad49
const int SHORTEST_LONG_RUN = 2 + LONG_ZEROCODE_RUN - SHORT_ZEROCODE_RUN;
shun-iwasawa 7bad49
const int LONGEST_LONG_RUN = 255 + SHORTEST_LONG_RUN;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void hufPackEncTable(
shun-iwasawa 7bad49
    const long long *hcode,  // i : encoding table [HUF_ENCSIZE]
shun-iwasawa 7bad49
    int im,                  // i : min hcode index
shun-iwasawa 7bad49
    int iM,                  // i : max hcode index
shun-iwasawa 7bad49
    char **pcode)            //  o: ptr to packed table (updated)
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  char *p = *pcode;
shun-iwasawa 7bad49
  long long c = 0;
shun-iwasawa 7bad49
  int lc = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (; im <= iM; im++) {
shun-iwasawa 7bad49
    int l = hufLength(hcode[im]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (l == 0) {
shun-iwasawa 7bad49
      int zerun = 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      while ((im < iM) && (zerun < LONGEST_LONG_RUN)) {
shun-iwasawa 7bad49
        if (hufLength(hcode[im + 1]) > 0) break;
shun-iwasawa 7bad49
        im++;
shun-iwasawa 7bad49
        zerun++;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (zerun >= 2) {
shun-iwasawa 7bad49
        if (zerun >= SHORTEST_LONG_RUN) {
shun-iwasawa 7bad49
          outputBits(6, LONG_ZEROCODE_RUN, c, lc, p);
shun-iwasawa 7bad49
          outputBits(8, zerun - SHORTEST_LONG_RUN, c, lc, p);
shun-iwasawa 7bad49
        } else {
shun-iwasawa 7bad49
          outputBits(6, SHORT_ZEROCODE_RUN + zerun - 2, c, lc, p);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        continue;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    outputBits(6, l, c, lc, p);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (lc > 0) *p++ = (unsigned char)(c << (8 - lc));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  *pcode = p;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Unpack an encoding table packed by hufPackEncTable():
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool hufUnpackEncTable(
shun-iwasawa 7bad49
    const char **pcode,  // io: ptr to packed table (updated)
shun-iwasawa 7bad49
    int ni,              // i : input size (in bytes)
shun-iwasawa 7bad49
    int im,              // i : min hcode index
shun-iwasawa 7bad49
    int iM,              // i : max hcode index
shun-iwasawa 7bad49
    long long *hcode)    //  o: encoding table [HUF_ENCSIZE]
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  memset(hcode, 0, sizeof(long long) * HUF_ENCSIZE);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const char *p = *pcode;
shun-iwasawa 7bad49
  long long c = 0;
shun-iwasawa 7bad49
  int lc = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (; im <= iM; im++) {
shun-iwasawa 7bad49
    if (p - *pcode >= ni) {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    long long l = hcode[im] = getBits(6, c, lc, p);  // code length
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (l == (long long)LONG_ZEROCODE_RUN) {
shun-iwasawa 7bad49
      if (p - *pcode > ni) {
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      int zerun = getBits(8, c, lc, p) + SHORTEST_LONG_RUN;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (im + zerun > iM + 1) {
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      while (zerun--) hcode[im++] = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      im--;
shun-iwasawa 7bad49
    } else if (l >= (long long)SHORT_ZEROCODE_RUN) {
shun-iwasawa 7bad49
      int zerun = l - SHORT_ZEROCODE_RUN + 2;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (im + zerun > iM + 1) {
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      while (zerun--) hcode[im++] = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      im--;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  *pcode = const_cast<char *>(p);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  hufCanonicalCodeTable(hcode);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// DECODING TABLE BUILDING
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Clear a newly allocated decoding table so that it contains only zeroes.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void hufClearDecTable(HufDec *hdecod)  // io: (allocated by caller)
shun-iwasawa 7bad49
//     decoding table [HUF_DECSIZE]
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  for (int i = 0; i < HUF_DECSIZE; i++) {
shun-iwasawa 7bad49
    hdecod[i].len = 0;
shun-iwasawa 7bad49
    hdecod[i].lit = 0;
shun-iwasawa 7bad49
    hdecod[i].p = NULL;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  // memset(hdecod, 0, sizeof(HufDec) * HUF_DECSIZE);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Build a decoding hash table based on the encoding table hcode:
shun-iwasawa 7bad49
//  - short codes (<= HUF_DECBITS) are resolved with a single table access;
shun-iwasawa 7bad49
//  - long code entry allocations are not optimized, because long codes are
shun-iwasawa 7bad49
//    unfrequent;
shun-iwasawa 7bad49
//  - decoding tables are used by hufDecode();
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool hufBuildDecTable(const long long *hcode,  // i : encoding table
shun-iwasawa 7bad49
                             int im,                  // i : min index in hcode
shun-iwasawa 7bad49
                             int iM,                  // i : max index in hcode
shun-iwasawa 7bad49
                             HufDec *hdecod)  //  o: (allocated by caller)
shun-iwasawa 7bad49
//     decoding table [HUF_DECSIZE]
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Init hashtable & loop on all codes.
shun-iwasawa 7bad49
  // Assumes that hufClearDecTable(hdecod) has already been called.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (; im <= iM; im++) {
shun-iwasawa 7bad49
    long long c = hufCode(hcode[im]);
shun-iwasawa 7bad49
    int l = hufLength(hcode[im]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (c >> l) {
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
      // Error: c is supposed to be an l-bit code,
shun-iwasawa 7bad49
      // but c contains a value that is greater
shun-iwasawa 7bad49
      // than the largest l-bit number.
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      // invalidTableEntry();
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (l > HUF_DECBITS) {
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
      // Long code: add a secondary entry
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      HufDec *pl = hdecod + (c >> (l - HUF_DECBITS));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (pl->len) {
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
        // Error: a short code has already
shun-iwasawa 7bad49
        // been stored in table entry *pl.
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        // invalidTableEntry();
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      pl->lit++;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (pl->p) {
shun-iwasawa 7bad49
        unsigned int *p = pl->p;
shun-iwasawa 7bad49
        pl->p = new unsigned int[pl->lit];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        for (int i = 0; i < pl->lit - 1; ++i) pl->p[i] = p[i];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        delete[] p;
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        pl->p = new unsigned int[1];
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      pl->p[pl->lit - 1] = im;
shun-iwasawa 7bad49
    } else if (l) {
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
      // Short code: init all primary entries
shun-iwasawa 7bad49
      //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      HufDec *pl = hdecod + (c << (HUF_DECBITS - l));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      for (long long i = 1ULL << (HUF_DECBITS - l); i > 0; i--, pl++) {
shun-iwasawa 7bad49
        if (pl->len || pl->p) {
shun-iwasawa 7bad49
          //
shun-iwasawa 7bad49
          // Error: a short code or a long code has
shun-iwasawa 7bad49
          // already been stored in table entry *pl.
shun-iwasawa 7bad49
          //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          // invalidTableEntry();
shun-iwasawa 7bad49
          return false;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        pl->len = l;
shun-iwasawa 7bad49
        pl->lit = im;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Free the long code entries of a decoding table built by hufBuildDecTable()
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void hufFreeDecTable(HufDec *hdecod)  // io: Decoding table
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  for (int i = 0; i < HUF_DECSIZE; i++) {
shun-iwasawa 7bad49
    if (hdecod[i].p) {
shun-iwasawa 7bad49
      delete[] hdecod[i].p;
shun-iwasawa 7bad49
      hdecod[i].p = 0;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// ENCODING
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline void outputCode(long long code, long long &c, int &lc, char *&out) {
shun-iwasawa 7bad49
  outputBits(hufLength(code), hufCode(code), c, lc, out);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
inline void sendCode(long long sCode, int runCount, long long runCode,
shun-iwasawa 7bad49
                     long long &c, int &lc, char *&out) {
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Output a run of runCount instances of the symbol sCount.
shun-iwasawa 7bad49
  // Output the symbols explicitly, or if that is shorter, output
shun-iwasawa 7bad49
  // the sCode symbol once followed by a runCode symbol and runCount
shun-iwasawa 7bad49
  // expressed as an 8-bit number.
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (hufLength(sCode) + hufLength(runCode) + 8 < hufLength(sCode) * runCount) {
shun-iwasawa 7bad49
    outputCode(sCode, c, lc, out);
shun-iwasawa 7bad49
    outputCode(runCode, c, lc, out);
shun-iwasawa 7bad49
    outputBits(8, runCount, c, lc, out);
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    while (runCount-- >= 0) outputCode(sCode, c, lc, out);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Encode (compress) ni values based on the Huffman encoding table hcode:
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int hufEncode            // return: output size (in bits)
shun-iwasawa 7bad49
    (const long long *hcode,    // i : encoding table
shun-iwasawa 7bad49
     const unsigned short *in,  // i : uncompressed input buffer
shun-iwasawa 7bad49
     const int ni,              // i : input buffer size (in bytes)
shun-iwasawa 7bad49
     int rlc,                   // i : rl code
shun-iwasawa 7bad49
     char *out)                 //  o: compressed output buffer
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  char *outStart = out;
shun-iwasawa 7bad49
  long long c = 0;  // bits not yet written to out
shun-iwasawa 7bad49
  int lc = 0;       // number of valid bits in c (LSB)
shun-iwasawa 7bad49
  int s = in[0];
shun-iwasawa 7bad49
  int cs = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Loop on input values
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 1; i < ni; i++) {
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Count same values or send code
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (s == in[i] && cs < 255) {
shun-iwasawa 7bad49
      cs++;
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      sendCode(hcode[s], cs, hcode[rlc], c, lc, out);
shun-iwasawa 7bad49
      cs = 0;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    s = in[i];
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Send remaining code
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  sendCode(hcode[s], cs, hcode[rlc], c, lc, out);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (lc) *out = (c << (8 - lc)) & 0xff;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return (out - outStart) * 8 + lc;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// DECODING
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// In order to force the compiler to inline them,
shun-iwasawa 7bad49
// getChar() and getCode() are implemented as macros
shun-iwasawa 7bad49
// instead of "inline" functions.
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#define getChar(c, lc, in)                   \
shun-iwasawa 7bad49
  {                                          \
shun-iwasawa 7bad49
    c = (c << 8) | *(unsigned char *)(in++); \
shun-iwasawa 7bad49
    lc += 8;                                 \
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if 0
shun-iwasawa 7bad49
#define getCode(po, rlc, c, lc, in, out, ob, oe) \
shun-iwasawa 7bad49
  {                                              \
shun-iwasawa 7bad49
    if (po == rlc) {                             \
shun-iwasawa 7bad49
      if (lc < 8) getChar(c, lc, in);            \
shun-iwasawa 7bad49
                                                 \
shun-iwasawa 7bad49
      lc -= 8;                                   \
shun-iwasawa 7bad49
                                                 \
shun-iwasawa 7bad49
      unsigned char cs = (c >> lc);              \
shun-iwasawa 7bad49
                                                 \
shun-iwasawa 7bad49
      if (out + cs > oe) return false;           \
shun-iwasawa 7bad49
                                                 \
shun-iwasawa 7bad49
      /* TinyEXR issue 78 */                     \
shun-iwasawa 7bad49
      unsigned short s = out[-1];                \
shun-iwasawa 7bad49
                                                 \
shun-iwasawa 7bad49
      while (cs-- > 0) *out++ = s;               \
shun-iwasawa 7bad49
    } else if (out < oe) {                       \
shun-iwasawa 7bad49
      *out++ = po;                               \
shun-iwasawa 7bad49
    } else {                                     \
shun-iwasawa 7bad49
      return false;                              \
shun-iwasawa 7bad49
    }                                            \
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
static bool getCode(int po, int rlc, long long &c, int &lc, const char *&in,
shun-iwasawa 7bad49
                    const char *in_end, unsigned short *&out,
shun-iwasawa 7bad49
                    const unsigned short *ob, const unsigned short *oe) {
shun-iwasawa 7bad49
  (void)ob;
shun-iwasawa 7bad49
  if (po == rlc) {
shun-iwasawa 7bad49
    if (lc < 8) {
shun-iwasawa 7bad49
      /* TinyEXR issue 78 */
shun-iwasawa 7bad49
      /* TinyEXR issue 160. in + 1 -> in */
shun-iwasawa 7bad49
      if (in >= in_end) {
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      getChar(c, lc, in);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    lc -= 8;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    unsigned char cs = (c >> lc);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (out + cs > oe) return false;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // Bounds check for safety
shun-iwasawa 7bad49
    // Issue 100.
shun-iwasawa 7bad49
    if ((out - 1) < ob) return false;
shun-iwasawa 7bad49
    unsigned short s = out[-1];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    while (cs-- > 0) *out++ = s;
shun-iwasawa 7bad49
  } else if (out < oe) {
shun-iwasawa 7bad49
    *out++ = po;
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Decode (uncompress) ni bits based on encoding & decoding tables:
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool hufDecode(const long long *hcode,  // i : encoding table
shun-iwasawa 7bad49
                      const HufDec *hdecod,    // i : decoding table
shun-iwasawa 7bad49
                      const char *in,          // i : compressed input buffer
shun-iwasawa 7bad49
                      int ni,                  // i : input size (in bits)
shun-iwasawa 7bad49
                      int rlc,                 // i : run-length code
shun-iwasawa 7bad49
                      int no,  // i : expected output size (in bytes)
shun-iwasawa 7bad49
                      unsigned short *out)  //  o: uncompressed output buffer
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  long long c = 0;
shun-iwasawa 7bad49
  int lc = 0;
shun-iwasawa 7bad49
  unsigned short *outb = out;          // begin
shun-iwasawa 7bad49
  unsigned short *oe = out + no;       // end
shun-iwasawa 7bad49
  const char *ie = in + (ni + 7) / 8;  // input byte size
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Loop on input bytes
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (in < ie) {
shun-iwasawa 7bad49
    getChar(c, lc, in);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Access decoding table
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    while (lc >= HUF_DECBITS) {
shun-iwasawa 7bad49
      const HufDec pl = hdecod[(c >> (lc - HUF_DECBITS)) & HUF_DECMASK];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (pl.len) {
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
        // Get short code
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        lc -= pl.len;
shun-iwasawa 7bad49
        // std::cout << "lit = " << pl.lit << std::endl;
shun-iwasawa 7bad49
        // std::cout << "rlc = " << rlc << std::endl;
shun-iwasawa 7bad49
        // std::cout << "c = " << c << std::endl;
shun-iwasawa 7bad49
        // std::cout << "lc = " << lc << std::endl;
shun-iwasawa 7bad49
        // std::cout << "in = " << in << std::endl;
shun-iwasawa 7bad49
        // std::cout << "out = " << out << std::endl;
shun-iwasawa 7bad49
        // std::cout << "oe = " << oe << std::endl;
shun-iwasawa 7bad49
        if (!getCode(pl.lit, rlc, c, lc, in, ie, out, outb, oe)) {
shun-iwasawa 7bad49
          return false;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        if (!pl.p) {
shun-iwasawa 7bad49
          return false;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        // invalidCode(); // wrong code
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
        // Search long code
shun-iwasawa 7bad49
        //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        int j;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        for (j = 0; j < pl.lit; j++) {
shun-iwasawa 7bad49
          int l = hufLength(hcode[pl.p[j]]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          while (lc < l && in < ie)  // get more bits
shun-iwasawa 7bad49
            getChar(c, lc, in);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          if (lc >= l) {
shun-iwasawa 7bad49
            if (hufCode(hcode[pl.p[j]]) ==
shun-iwasawa 7bad49
                ((c >> (lc - l)) & (((long long)(1) << l) - 1))) {
shun-iwasawa 7bad49
              //
shun-iwasawa 7bad49
              // Found : get long code
shun-iwasawa 7bad49
              //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              lc -= l;
shun-iwasawa 7bad49
              if (!getCode(pl.p[j], rlc, c, lc, in, ie, out, outb, oe)) {
shun-iwasawa 7bad49
                return false;
shun-iwasawa 7bad49
              }
shun-iwasawa 7bad49
              break;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (j == pl.lit) {
shun-iwasawa 7bad49
          return false;
shun-iwasawa 7bad49
          // invalidCode(); // Not found
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Get remaining (short) codes
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int i = (8 - ni) & 7;
shun-iwasawa 7bad49
  c >>= i;
shun-iwasawa 7bad49
  lc -= i;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (lc > 0) {
shun-iwasawa 7bad49
    const HufDec pl = hdecod[(c << (HUF_DECBITS - lc)) & HUF_DECMASK];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (pl.len) {
shun-iwasawa 7bad49
      lc -= pl.len;
shun-iwasawa 7bad49
      if (!getCode(pl.lit, rlc, c, lc, in, ie, out, outb, oe)) {
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
      // invalidCode(); // wrong (long) code
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (out - outb != no) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  // notEnoughData ();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void countFrequencies(std::vector<long long> &freq,
shun-iwasawa 7bad49
                             const unsigned short data[/*n*/], int n) {
shun-iwasawa 7bad49
  for (int i = 0; i < HUF_ENCSIZE; ++i) freq[i] = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < n; ++i) ++freq[data[i]];
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void writeUInt(char buf[4], unsigned int i) {
shun-iwasawa 7bad49
  unsigned char *b = (unsigned char *)buf;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  b[0] = i;
shun-iwasawa 7bad49
  b[1] = i >> 8;
shun-iwasawa 7bad49
  b[2] = i >> 16;
shun-iwasawa 7bad49
  b[3] = i >> 24;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static unsigned int readUInt(const char buf[4]) {
shun-iwasawa 7bad49
  const unsigned char *b = (const unsigned char *)buf;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return (b[0] & 0x000000ff) | ((b[1] << 8) & 0x0000ff00) |
shun-iwasawa 7bad49
         ((b[2] << 16) & 0x00ff0000) | ((b[3] << 24) & 0xff000000);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// EXTERNAL INTERFACE
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int hufCompress(const unsigned short raw[], int nRaw,
shun-iwasawa 7bad49
                       char compressed[]) {
shun-iwasawa 7bad49
  if (nRaw == 0) return 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<long long> freq(HUF_ENCSIZE);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  countFrequencies(freq, raw, nRaw);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int im = 0;
shun-iwasawa 7bad49
  int iM = 0;
shun-iwasawa 7bad49
  hufBuildEncTable(freq.data(), &im, &iM);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  char *tableStart = compressed + 20;
shun-iwasawa 7bad49
  char *tableEnd = tableStart;
shun-iwasawa 7bad49
  hufPackEncTable(freq.data(), im, iM, &tableEnd);
shun-iwasawa 7bad49
  int tableLength = tableEnd - tableStart;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  char *dataStart = tableEnd;
shun-iwasawa 7bad49
  int nBits = hufEncode(freq.data(), raw, nRaw, iM, dataStart);
shun-iwasawa 7bad49
  int data_length = (nBits + 7) / 8;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  writeUInt(compressed, im);
shun-iwasawa 7bad49
  writeUInt(compressed + 4, iM);
shun-iwasawa 7bad49
  writeUInt(compressed + 8, tableLength);
shun-iwasawa 7bad49
  writeUInt(compressed + 12, nBits);
shun-iwasawa 7bad49
  writeUInt(compressed + 16, 0);  // room for future extensions
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return dataStart + data_length - compressed;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool hufUncompress(const char compressed[], int nCompressed,
shun-iwasawa 7bad49
                          std::vector<unsigned short> *raw) {
shun-iwasawa 7bad49
  if (nCompressed == 0) {
shun-iwasawa 7bad49
    if (raw->size() != 0) return false;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int im = readUInt(compressed);
shun-iwasawa 7bad49
  int iM = readUInt(compressed + 4);
shun-iwasawa 7bad49
  // int tableLength = readUInt (compressed + 8);
shun-iwasawa 7bad49
  int nBits = readUInt(compressed + 12);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (im < 0 || im >= HUF_ENCSIZE || iM < 0 || iM >= HUF_ENCSIZE) return false;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const char *ptr = compressed + 20;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Fast decoder needs at least 2x64-bits of compressed data, and
shun-iwasawa 7bad49
  // needs to be run-able on this platform. Otherwise, fall back
shun-iwasawa 7bad49
  // to the original decoder
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // if (FastHufDecoder::enabled() && nBits > 128)
shun-iwasawa 7bad49
  //{
shun-iwasawa 7bad49
  //    FastHufDecoder fhd (ptr, nCompressed - (ptr - compressed), im, iM, iM);
shun-iwasawa 7bad49
  //    fhd.decode ((unsigned char*)ptr, nBits, raw, nRaw);
shun-iwasawa 7bad49
  //}
shun-iwasawa 7bad49
  // else
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    std::vector<long long> freq(HUF_ENCSIZE);
shun-iwasawa 7bad49
    std::vector<HufDec> hdec(HUF_DECSIZE);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    hufClearDecTable(&hdec.at(0));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    hufUnpackEncTable(&ptr, nCompressed - (ptr - compressed), im, iM,
shun-iwasawa 7bad49
                      &freq.at(0));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    {
shun-iwasawa 7bad49
      if (nBits > 8 * (nCompressed - (ptr - compressed))) {
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      hufBuildDecTable(&freq.at(0), im, iM, &hdec.at(0));
shun-iwasawa 7bad49
      hufDecode(&freq.at(0), &hdec.at(0), ptr, nBits, iM, raw->size(),
shun-iwasawa 7bad49
                raw->data());
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    // catch (...)
shun-iwasawa 7bad49
    //{
shun-iwasawa 7bad49
    //    hufFreeDecTable (hdec);
shun-iwasawa 7bad49
    //    throw;
shun-iwasawa 7bad49
    //}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    hufFreeDecTable(&hdec.at(0));
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// Functions to compress the range of values in the pixel data
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
const int USHORT_RANGE = (1 << 16);
shun-iwasawa 7bad49
const int BITMAP_SIZE = (USHORT_RANGE >> 3);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void bitmapFromData(const unsigned short data[/*nData*/], int nData,
shun-iwasawa 7bad49
                           unsigned char bitmap[BITMAP_SIZE],
shun-iwasawa 7bad49
                           unsigned short &minNonZero,
shun-iwasawa 7bad49
                           unsigned short &maxNonZero) {
shun-iwasawa 7bad49
  for (int i = 0; i < BITMAP_SIZE; ++i) bitmap[i] = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < nData; ++i) bitmap[data[i] >> 3] |= (1 << (data[i] & 7));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  bitmap[0] &= ~1;  // zero is not explicitly stored in
shun-iwasawa 7bad49
                    // the bitmap; we assume that the
shun-iwasawa 7bad49
                    // data always contain zeroes
shun-iwasawa 7bad49
  minNonZero = BITMAP_SIZE - 1;
shun-iwasawa 7bad49
  maxNonZero = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < BITMAP_SIZE; ++i) {
shun-iwasawa 7bad49
    if (bitmap[i]) {
shun-iwasawa 7bad49
      if (minNonZero > i) minNonZero = i;
shun-iwasawa 7bad49
      if (maxNonZero < i) maxNonZero = i;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static unsigned short forwardLutFromBitmap(
shun-iwasawa 7bad49
    const unsigned char bitmap[BITMAP_SIZE], unsigned short lut[USHORT_RANGE]) {
shun-iwasawa 7bad49
  int k = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < USHORT_RANGE; ++i) {
shun-iwasawa 7bad49
    if ((i == 0) || (bitmap[i >> 3] & (1 << (i & 7))))
shun-iwasawa 7bad49
      lut[i] = k++;
shun-iwasawa 7bad49
    else
shun-iwasawa 7bad49
      lut[i] = 0;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return k - 1;  // maximum value stored in lut[],
shun-iwasawa 7bad49
}  // i.e. number of ones in bitmap minus 1
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static unsigned short reverseLutFromBitmap(
shun-iwasawa 7bad49
    const unsigned char bitmap[BITMAP_SIZE], unsigned short lut[USHORT_RANGE]) {
shun-iwasawa 7bad49
  int k = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < USHORT_RANGE; ++i) {
shun-iwasawa 7bad49
    if ((i == 0) || (bitmap[i >> 3] & (1 << (i & 7)))) lut[k++] = i;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int n = k - 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  while (k < USHORT_RANGE) lut[k++] = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return n;  // maximum k where lut[k] is non-zero,
shun-iwasawa 7bad49
}  // i.e. number of ones in bitmap minus 1
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void applyLut(const unsigned short lut[USHORT_RANGE],
shun-iwasawa 7bad49
                     unsigned short data[/*nData*/], int nData) {
shun-iwasawa 7bad49
  for (int i = 0; i < nData; ++i) data[i] = lut[data[i]];
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
#pragma clang diagnostic pop
shun-iwasawa 7bad49
#endif  // __clang__
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef _MSC_VER
shun-iwasawa 7bad49
#pragma warning(pop)
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool CompressPiz(unsigned char *outPtr, unsigned int *outSize,
shun-iwasawa 7bad49
                        const unsigned char *inPtr, size_t inSize,
shun-iwasawa 7bad49
                        const std::vector<ChannelInfo> &channelInfo,
shun-iwasawa 7bad49
                        int data_width, int num_lines) {
shun-iwasawa 7bad49
  std::vector<unsigned char> bitmap(BITMAP_SIZE);
shun-iwasawa 7bad49
  unsigned short minNonZero;
shun-iwasawa 7bad49
  unsigned short maxNonZero;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if !TINYEXR_LITTLE_ENDIAN
shun-iwasawa 7bad49
  // @todo { PIZ compression on BigEndian architecture. }
shun-iwasawa 7bad49
  assert(0);
shun-iwasawa 7bad49
  return false;
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Assume `inSize` is multiple of 2 or 4.
shun-iwasawa 7bad49
  std::vector<unsigned short> tmpBuffer(inSize / sizeof(unsigned short));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<PIZChannelData> channelData(channelInfo.size());
shun-iwasawa 7bad49
  unsigned short *tmpBufferEnd = &tmpBuffer.at(0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t c = 0; c < channelData.size(); c++) {
shun-iwasawa 7bad49
    PIZChannelData &cd = channelData[c];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    cd.start = tmpBufferEnd;
shun-iwasawa 7bad49
    cd.end = cd.start;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    cd.nx = data_width;
shun-iwasawa 7bad49
    cd.ny = num_lines;
shun-iwasawa 7bad49
    // cd.ys = c.channel().ySampling;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    size_t pixelSize = sizeof(int);  // UINT and FLOAT
shun-iwasawa 7bad49
    if (channelInfo[c].requested_pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
      pixelSize = sizeof(short);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    cd.size = static_cast<int>(pixelSize / sizeof(short));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tmpBufferEnd += cd.nx * cd.ny * cd.size;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const unsigned char *ptr = inPtr;
shun-iwasawa 7bad49
  for (int y = 0; y < num_lines; ++y) {
shun-iwasawa 7bad49
    for (size_t i = 0; i < channelData.size(); ++i) {
shun-iwasawa 7bad49
      PIZChannelData &cd = channelData[i];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      // if (modp (y, cd.ys) != 0)
shun-iwasawa 7bad49
      //    continue;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      size_t n = static_cast<size_t>(cd.nx * cd.size);
shun-iwasawa 7bad49
      memcpy(cd.end, ptr, n * sizeof(unsigned short));
shun-iwasawa 7bad49
      ptr += n * sizeof(unsigned short);
shun-iwasawa 7bad49
      cd.end += n;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  bitmapFromData(&tmpBuffer.at(0), static_cast<int>(tmpBuffer.size()),
shun-iwasawa 7bad49
                 bitmap.data(), minNonZero, maxNonZero);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned short> lut(USHORT_RANGE);
shun-iwasawa 7bad49
  unsigned short maxValue = forwardLutFromBitmap(bitmap.data(), lut.data());
shun-iwasawa 7bad49
  applyLut(lut.data(), &tmpBuffer.at(0), static_cast<int>(tmpBuffer.size()));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Store range compression info in _outBuffer
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  char *buf = reinterpret_cast<char *>(outPtr);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  memcpy(buf, &minNonZero, sizeof(unsigned short));
shun-iwasawa 7bad49
  buf += sizeof(unsigned short);
shun-iwasawa 7bad49
  memcpy(buf, &maxNonZero, sizeof(unsigned short));
shun-iwasawa 7bad49
  buf += sizeof(unsigned short);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (minNonZero <= maxNonZero) {
shun-iwasawa 7bad49
    memcpy(buf, reinterpret_cast<char *>(&bitmap[0] + minNonZero),
shun-iwasawa 7bad49
           maxNonZero - minNonZero + 1);
shun-iwasawa 7bad49
    buf += maxNonZero - minNonZero + 1;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Apply wavelet encoding
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t i = 0; i < channelData.size(); ++i) {
shun-iwasawa 7bad49
    PIZChannelData &cd = channelData[i];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (int j = 0; j < cd.size; ++j) {
shun-iwasawa 7bad49
      wav2Encode(cd.start + j, cd.nx, cd.size, cd.ny, cd.nx * cd.size,
shun-iwasawa 7bad49
                 maxValue);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Apply Huffman encoding; append the result to _outBuffer
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // length header(4byte), then huff data. Initialize length header with zero,
shun-iwasawa 7bad49
  // then later fill it by `length`.
shun-iwasawa 7bad49
  char *lengthPtr = buf;
shun-iwasawa 7bad49
  int zero = 0;
shun-iwasawa 7bad49
  memcpy(buf, &zero, sizeof(int));
shun-iwasawa 7bad49
  buf += sizeof(int);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int length =
shun-iwasawa 7bad49
      hufCompress(&tmpBuffer.at(0), static_cast<int>(tmpBuffer.size()), buf);
shun-iwasawa 7bad49
  memcpy(lengthPtr, &length, sizeof(int));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  (*outSize) = static_cast<unsigned int>(
shun-iwasawa 7bad49
      (reinterpret_cast<unsigned char *>(buf) - outPtr) +
shun-iwasawa 7bad49
      static_cast<unsigned int>(length));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Use uncompressed data when compressed data is larger than uncompressed.
shun-iwasawa 7bad49
  // (Issue 40)
shun-iwasawa 7bad49
  if ((*outSize) >= inSize) {
shun-iwasawa 7bad49
    (*outSize) = static_cast<unsigned int>(inSize);
shun-iwasawa 7bad49
    memcpy(outPtr, inPtr, inSize);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool DecompressPiz(unsigned char *outPtr, const unsigned char *inPtr,
shun-iwasawa 7bad49
                          size_t tmpBufSizeInBytes, size_t inLen, int num_channels,
shun-iwasawa 7bad49
                          const EXRChannelInfo *channels, int data_width,
shun-iwasawa 7bad49
                          int num_lines) {
shun-iwasawa 7bad49
  if (inLen == tmpBufSizeInBytes) {
shun-iwasawa 7bad49
    // Data is not compressed(Issue 40).
shun-iwasawa 7bad49
    memcpy(outPtr, inPtr, inLen);
shun-iwasawa 7bad49
    return true;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned char> bitmap(BITMAP_SIZE);
shun-iwasawa 7bad49
  unsigned short minNonZero;
shun-iwasawa 7bad49
  unsigned short maxNonZero;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if !TINYEXR_LITTLE_ENDIAN
shun-iwasawa 7bad49
  // @todo { PIZ compression on BigEndian architecture. }
shun-iwasawa 7bad49
  assert(0);
shun-iwasawa 7bad49
  return false;
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  memset(bitmap.data(), 0, BITMAP_SIZE);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const unsigned char *ptr = inPtr;
shun-iwasawa 7bad49
  // minNonZero = *(reinterpret_cast<const unsigned short *>(ptr));
shun-iwasawa 7bad49
  tinyexr::cpy2(&minNonZero, reinterpret_cast<const unsigned short *>(ptr));
shun-iwasawa 7bad49
  // maxNonZero = *(reinterpret_cast<const unsigned short *>(ptr + 2));
shun-iwasawa 7bad49
  tinyexr::cpy2(&maxNonZero, reinterpret_cast<const unsigned short *>(ptr + 2));
shun-iwasawa 7bad49
  ptr += 4;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (maxNonZero >= BITMAP_SIZE) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (minNonZero <= maxNonZero) {
shun-iwasawa 7bad49
    memcpy(reinterpret_cast<char *>(&bitmap[0] + minNonZero), ptr,
shun-iwasawa 7bad49
           maxNonZero - minNonZero + 1);
shun-iwasawa 7bad49
    ptr += maxNonZero - minNonZero + 1;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned short> lut(USHORT_RANGE);
shun-iwasawa 7bad49
  memset(lut.data(), 0, sizeof(unsigned short) * USHORT_RANGE);
shun-iwasawa 7bad49
  unsigned short maxValue = reverseLutFromBitmap(bitmap.data(), lut.data());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Huffman decoding
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int length;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // length = *(reinterpret_cast<const int *>(ptr));
shun-iwasawa 7bad49
  tinyexr::cpy4(&length, reinterpret_cast<const int *>(ptr));
shun-iwasawa 7bad49
  ptr += sizeof(int);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (size_t((ptr - inPtr) + length) > inLen) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned short> tmpBuffer(tmpBufSizeInBytes / sizeof(unsigned short));
shun-iwasawa 7bad49
  hufUncompress(reinterpret_cast<const char *>(ptr), length, &tmpBuffer);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Wavelet decoding
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<PIZChannelData> channelData(static_cast<size_t>(num_channels));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  unsigned short *tmpBufferEnd = &tmpBuffer.at(0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t i = 0; i < static_cast<size_t>(num_channels); ++i) {
shun-iwasawa 7bad49
    const EXRChannelInfo &chan = channels[i];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    size_t pixelSize = sizeof(int);  // UINT and FLOAT
shun-iwasawa 7bad49
    if (chan.pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
      pixelSize = sizeof(short);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    channelData[i].start = tmpBufferEnd;
shun-iwasawa 7bad49
    channelData[i].end = channelData[i].start;
shun-iwasawa 7bad49
    channelData[i].nx = data_width;
shun-iwasawa 7bad49
    channelData[i].ny = num_lines;
shun-iwasawa 7bad49
    // channelData[i].ys = 1;
shun-iwasawa 7bad49
    channelData[i].size = static_cast<int>(pixelSize / sizeof(short));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tmpBufferEnd += channelData[i].nx * channelData[i].ny * channelData[i].size;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t i = 0; i < channelData.size(); ++i) {
shun-iwasawa 7bad49
    PIZChannelData &cd = channelData[i];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (int j = 0; j < cd.size; ++j) {
shun-iwasawa 7bad49
      wav2Decode(cd.start + j, cd.nx, cd.size, cd.ny, cd.nx * cd.size,
shun-iwasawa 7bad49
                 maxValue);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // Expand the pixel data to their original range
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  applyLut(lut.data(), &tmpBuffer.at(0), static_cast<int>(tmpBufSizeInBytes / sizeof(unsigned short)));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int y = 0; y < num_lines; y++) {
shun-iwasawa 7bad49
    for (size_t i = 0; i < channelData.size(); ++i) {
shun-iwasawa 7bad49
      PIZChannelData &cd = channelData[i];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      // if (modp (y, cd.ys) != 0)
shun-iwasawa 7bad49
      //    continue;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      size_t n = static_cast<size_t>(cd.nx * cd.size);
shun-iwasawa 7bad49
      memcpy(outPtr, cd.end, static_cast<size_t>(n * sizeof(unsigned short)));
shun-iwasawa 7bad49
      outPtr += n * sizeof(unsigned short);
shun-iwasawa 7bad49
      cd.end += n;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
#endif  // TINYEXR_USE_PIZ
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_ZFP
shun-iwasawa 7bad49
shun-iwasawa 7bad49
struct ZFPCompressionParam {
shun-iwasawa 7bad49
  double rate;
shun-iwasawa 7bad49
  unsigned int precision;
shun-iwasawa 7bad49
  unsigned int __pad0;
shun-iwasawa 7bad49
  double tolerance;
shun-iwasawa 7bad49
  int type;  // TINYEXR_ZFP_COMPRESSIONTYPE_*
shun-iwasawa 7bad49
  unsigned int __pad1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  ZFPCompressionParam() {
shun-iwasawa 7bad49
    type = TINYEXR_ZFP_COMPRESSIONTYPE_RATE;
shun-iwasawa 7bad49
    rate = 2.0;
shun-iwasawa 7bad49
    precision = 0;
shun-iwasawa 7bad49
    tolerance = 0.0;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool FindZFPCompressionParam(ZFPCompressionParam *param,
shun-iwasawa 7bad49
                                    const EXRAttribute *attributes,
shun-iwasawa 7bad49
                                    int num_attributes, std::string *err) {
shun-iwasawa 7bad49
  bool foundType = false;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < num_attributes; i++) {
shun-iwasawa 7bad49
    if ((strcmp(attributes[i].name, "zfpCompressionType") == 0)) {
shun-iwasawa 7bad49
      if (attributes[i].size == 1) {
shun-iwasawa 7bad49
        param->type = static_cast<int>(attributes[i].value[0]);
shun-iwasawa 7bad49
        foundType = true;
shun-iwasawa 7bad49
        break;
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          (*err) +=
shun-iwasawa 7bad49
              "zfpCompressionType attribute must be uchar(1 byte) type.\n";
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (!foundType) {
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "`zfpCompressionType` attribute not found.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (param->type == TINYEXR_ZFP_COMPRESSIONTYPE_RATE) {
shun-iwasawa 7bad49
    for (int i = 0; i < num_attributes; i++) {
shun-iwasawa 7bad49
      if ((strcmp(attributes[i].name, "zfpCompressionRate") == 0) &&
shun-iwasawa 7bad49
          (attributes[i].size == 8)) {
shun-iwasawa 7bad49
        param->rate = *(reinterpret_cast<double *>(attributes[i].value));
shun-iwasawa 7bad49
        return true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "`zfpCompressionRate` attribute not found.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  } else if (param->type == TINYEXR_ZFP_COMPRESSIONTYPE_PRECISION) {
shun-iwasawa 7bad49
    for (int i = 0; i < num_attributes; i++) {
shun-iwasawa 7bad49
      if ((strcmp(attributes[i].name, "zfpCompressionPrecision") == 0) &&
shun-iwasawa 7bad49
          (attributes[i].size == 4)) {
shun-iwasawa 7bad49
        param->rate = *(reinterpret_cast<int *>(attributes[i].value));
shun-iwasawa 7bad49
        return true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "`zfpCompressionPrecision` attribute not found.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  } else if (param->type == TINYEXR_ZFP_COMPRESSIONTYPE_ACCURACY) {
shun-iwasawa 7bad49
    for (int i = 0; i < num_attributes; i++) {
shun-iwasawa 7bad49
      if ((strcmp(attributes[i].name, "zfpCompressionTolerance") == 0) &&
shun-iwasawa 7bad49
          (attributes[i].size == 8)) {
shun-iwasawa 7bad49
        param->tolerance = *(reinterpret_cast<double *>(attributes[i].value));
shun-iwasawa 7bad49
        return true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "`zfpCompressionTolerance` attribute not found.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "Unknown value specified for `zfpCompressionType`.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return false;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Assume pixel format is FLOAT for all channels.
shun-iwasawa 7bad49
static bool DecompressZfp(float *dst, int dst_width, int dst_num_lines,
shun-iwasawa 7bad49
                          size_t num_channels, const unsigned char *src,
shun-iwasawa 7bad49
                          unsigned long src_size,
shun-iwasawa 7bad49
                          const ZFPCompressionParam &param) {
shun-iwasawa 7bad49
  size_t uncompressed_size =
shun-iwasawa 7bad49
      size_t(dst_width) * size_t(dst_num_lines) * num_channels;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (uncompressed_size == src_size) {
shun-iwasawa 7bad49
    // Data is not compressed(Issue 40).
shun-iwasawa 7bad49
    memcpy(dst, src, src_size);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  zfp_stream *zfp = NULL;
shun-iwasawa 7bad49
  zfp_field *field = NULL;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  assert((dst_width % 4) == 0);
shun-iwasawa 7bad49
  assert((dst_num_lines % 4) == 0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if ((size_t(dst_width) & 3U) || (size_t(dst_num_lines) & 3U)) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  field =
shun-iwasawa 7bad49
      zfp_field_2d(reinterpret_cast<void *>(const_cast<unsigned char *>(src)),
shun-iwasawa 7bad49
                   zfp_type_float, static_cast<unsigned int>(dst_width),
shun-iwasawa 7bad49
                   static_cast<unsigned int>(dst_num_lines) *
shun-iwasawa 7bad49
                       static_cast<unsigned int>(num_channels));
shun-iwasawa 7bad49
  zfp = zfp_stream_open(NULL);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (param.type == TINYEXR_ZFP_COMPRESSIONTYPE_RATE) {
shun-iwasawa 7bad49
    zfp_stream_set_rate(zfp, param.rate, zfp_type_float, /* dimension */ 2,
shun-iwasawa 7bad49
                        /* write random access */ 0);
shun-iwasawa 7bad49
  } else if (param.type == TINYEXR_ZFP_COMPRESSIONTYPE_PRECISION) {
shun-iwasawa 7bad49
    zfp_stream_set_precision(zfp, param.precision);
shun-iwasawa 7bad49
  } else if (param.type == TINYEXR_ZFP_COMPRESSIONTYPE_ACCURACY) {
shun-iwasawa 7bad49
    zfp_stream_set_accuracy(zfp, param.tolerance);
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    assert(0);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t buf_size = zfp_stream_maximum_size(zfp, field);
shun-iwasawa 7bad49
  std::vector<unsigned char> buf(buf_size);
shun-iwasawa 7bad49
  memcpy(&buf.at(0), src, src_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  bitstream *stream = stream_open(&buf.at(0), buf_size);
shun-iwasawa 7bad49
  zfp_stream_set_bit_stream(zfp, stream);
shun-iwasawa 7bad49
  zfp_stream_rewind(zfp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t image_size = size_t(dst_width) * size_t(dst_num_lines);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t c = 0; c < size_t(num_channels); c++) {
shun-iwasawa 7bad49
    // decompress 4x4 pixel block.
shun-iwasawa 7bad49
    for (size_t y = 0; y < size_t(dst_num_lines); y += 4) {
shun-iwasawa 7bad49
      for (size_t x = 0; x < size_t(dst_width); x += 4) {
shun-iwasawa 7bad49
        float fblock[16];
shun-iwasawa 7bad49
        zfp_decode_block_float_2(zfp, fblock);
shun-iwasawa 7bad49
        for (size_t j = 0; j < 4; j++) {
shun-iwasawa 7bad49
          for (size_t i = 0; i < 4; i++) {
shun-iwasawa 7bad49
            dst[c * image_size + ((y + j) * size_t(dst_width) + (x + i))] =
shun-iwasawa 7bad49
                fblock[j * 4 + i];
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  zfp_field_free(field);
shun-iwasawa 7bad49
  zfp_stream_close(zfp);
shun-iwasawa 7bad49
  stream_close(stream);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Assume pixel format is FLOAT for all channels.
shun-iwasawa 7bad49
static bool CompressZfp(std::vector<unsigned char> *outBuf,
shun-iwasawa 7bad49
                        unsigned int *outSize, const float *inPtr, int width,
shun-iwasawa 7bad49
                        int num_lines, int num_channels,
shun-iwasawa 7bad49
                        const ZFPCompressionParam &param) {
shun-iwasawa 7bad49
  zfp_stream *zfp = NULL;
shun-iwasawa 7bad49
  zfp_field *field = NULL;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  assert((width % 4) == 0);
shun-iwasawa 7bad49
  assert((num_lines % 4) == 0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if ((size_t(width) & 3U) || (size_t(num_lines) & 3U)) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // create input array.
shun-iwasawa 7bad49
  field = zfp_field_2d(reinterpret_cast<void *>(const_cast<float *>(inPtr)),
shun-iwasawa 7bad49
                       zfp_type_float, static_cast<unsigned int>(width),
shun-iwasawa 7bad49
                       static_cast<unsigned int>(num_lines * num_channels));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  zfp = zfp_stream_open(NULL);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (param.type == TINYEXR_ZFP_COMPRESSIONTYPE_RATE) {
shun-iwasawa 7bad49
    zfp_stream_set_rate(zfp, param.rate, zfp_type_float, 2, 0);
shun-iwasawa 7bad49
  } else if (param.type == TINYEXR_ZFP_COMPRESSIONTYPE_PRECISION) {
shun-iwasawa 7bad49
    zfp_stream_set_precision(zfp, param.precision);
shun-iwasawa 7bad49
  } else if (param.type == TINYEXR_ZFP_COMPRESSIONTYPE_ACCURACY) {
shun-iwasawa 7bad49
    zfp_stream_set_accuracy(zfp, param.tolerance);
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    assert(0);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t buf_size = zfp_stream_maximum_size(zfp, field);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  outBuf->resize(buf_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  bitstream *stream = stream_open(&outBuf->at(0), buf_size);
shun-iwasawa 7bad49
  zfp_stream_set_bit_stream(zfp, stream);
shun-iwasawa 7bad49
  zfp_field_free(field);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t image_size = size_t(width) * size_t(num_lines);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t c = 0; c < size_t(num_channels); c++) {
shun-iwasawa 7bad49
    // compress 4x4 pixel block.
shun-iwasawa 7bad49
    for (size_t y = 0; y < size_t(num_lines); y += 4) {
shun-iwasawa 7bad49
      for (size_t x = 0; x < size_t(width); x += 4) {
shun-iwasawa 7bad49
        float fblock[16];
shun-iwasawa 7bad49
        for (size_t j = 0; j < 4; j++) {
shun-iwasawa 7bad49
          for (size_t i = 0; i < 4; i++) {
shun-iwasawa 7bad49
            fblock[j * 4 + i] =
shun-iwasawa 7bad49
                inPtr[c * image_size + ((y + j) * size_t(width) + (x + i))];
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        zfp_encode_block_float_2(zfp, fblock);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  zfp_stream_flush(zfp);
shun-iwasawa 7bad49
  (*outSize) = static_cast<unsigned int>(zfp_stream_compressed_size(zfp));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  zfp_stream_close(zfp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
// -----------------------------------------------------------------
shun-iwasawa 7bad49
//
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// heuristics
shun-iwasawa 7bad49
#define TINYEXR_DIMENSION_THRESHOLD (1024 * 8192)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// TODO(syoyo): Refactor function arguments.
shun-iwasawa 7bad49
static bool DecodePixelData(/* out */ unsigned char **out_images,
shun-iwasawa 7bad49
                            const int *requested_pixel_types,
shun-iwasawa 7bad49
                            const unsigned char *data_ptr, size_t data_len,
shun-iwasawa 7bad49
                            int compression_type, int line_order, int width,
shun-iwasawa 7bad49
                            int height, int x_stride, int y, int line_no,
shun-iwasawa 7bad49
                            int num_lines, size_t pixel_data_size,
shun-iwasawa 7bad49
                            size_t num_attributes,
shun-iwasawa 7bad49
                            const EXRAttribute *attributes, size_t num_channels,
shun-iwasawa 7bad49
                            const EXRChannelInfo *channels,
shun-iwasawa 7bad49
                            const std::vector<size_t> &channel_offset_list) {
shun-iwasawa 7bad49
  if (compression_type == TINYEXR_COMPRESSIONTYPE_PIZ) {  // PIZ
shun-iwasawa 7bad49
#if TINYEXR_USE_PIZ
shun-iwasawa 7bad49
    if ((width == 0) || (num_lines == 0) || (pixel_data_size == 0)) {
shun-iwasawa 7bad49
      // Invalid input #90
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // Allocate original data size.
shun-iwasawa 7bad49
    std::vector<unsigned char> outBuf(static_cast<size_t>(
shun-iwasawa 7bad49
        static_cast<size_t>(width * num_lines) * pixel_data_size));
shun-iwasawa 7bad49
    size_t tmpBufLen = outBuf.size();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    bool ret = tinyexr::DecompressPiz(
shun-iwasawa 7bad49
        reinterpret_cast<unsigned char *>(&outBuf.at(0)), data_ptr, tmpBufLen,
shun-iwasawa 7bad49
        data_len, static_cast<int>(num_channels), channels, width, num_lines);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!ret) {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // For PIZ_COMPRESSION:
shun-iwasawa 7bad49
    //   pixel sample data for channel 0 for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel 1 for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel ... for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel n for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel 0 for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel 1 for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel ... for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel n for scanline 1
shun-iwasawa 7bad49
    //   ...
shun-iwasawa 7bad49
    for (size_t c = 0; c < static_cast<size_t>(num_channels); c++) {
shun-iwasawa 7bad49
      if (channels[c].pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const unsigned short *line_ptr = reinterpret_cast<unsigned short *>(
shun-iwasawa 7bad49
              &outBuf.at(v * pixel_data_size * static_cast<size_t>(width) +
shun-iwasawa 7bad49
                         channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            FP16 hf;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // hf.u = line_ptr[u];
shun-iwasawa 7bad49
            // use `cpy` to avoid unaligned memory access when compiler's
shun-iwasawa 7bad49
            // optimization is on.
shun-iwasawa 7bad49
            tinyexr::cpy2(&(hf.u), line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap2(reinterpret_cast<unsigned short *>(&hf.u));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            if (requested_pixel_types[c] == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
              unsigned short *image =
shun-iwasawa 7bad49
                  reinterpret_cast<unsigned short **>(out_images)[c];
shun-iwasawa 7bad49
              if (line_order == 0) {
shun-iwasawa 7bad49
                image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              } else {
shun-iwasawa 7bad49
                image += static_cast<size_t>(
shun-iwasawa 7bad49
                             (height - 1 - (line_no + static_cast<int>(v)))) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              }
shun-iwasawa 7bad49
              *image = hf.u;
shun-iwasawa 7bad49
            } else {  // HALF -> FLOAT
shun-iwasawa 7bad49
              FP32 f32 = half_to_float(hf);
shun-iwasawa 7bad49
              float *image = reinterpret_cast<float **>(out_images)[c];
shun-iwasawa 7bad49
              size_t offset = 0;
shun-iwasawa 7bad49
              if (line_order == 0) {
shun-iwasawa 7bad49
                offset = (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              } else {
shun-iwasawa 7bad49
                offset = static_cast<size_t>(
shun-iwasawa 7bad49
                             (height - 1 - (line_no + static_cast<int>(v)))) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              }
shun-iwasawa 7bad49
              image += offset;
shun-iwasawa 7bad49
              *image = f32.f;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_UINT) {
shun-iwasawa 7bad49
        assert(requested_pixel_types[c] == TINYEXR_PIXELTYPE_UINT);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const unsigned int *line_ptr = reinterpret_cast<unsigned int *>(
shun-iwasawa 7bad49
              &outBuf.at(v * pixel_data_size * static_cast<size_t>(width) +
shun-iwasawa 7bad49
                         channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            unsigned int val;
shun-iwasawa 7bad49
            // val = line_ptr[u];
shun-iwasawa 7bad49
            tinyexr::cpy4(&val, line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(&val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            unsigned int *image =
shun-iwasawa 7bad49
                reinterpret_cast<unsigned int **>(out_images)[c];
shun-iwasawa 7bad49
            if (line_order == 0) {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              image += static_cast<size_t>(
shun-iwasawa 7bad49
                           (height - 1 - (line_no + static_cast<int>(v)))) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            *image = val;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
        assert(requested_pixel_types[c] == TINYEXR_PIXELTYPE_FLOAT);
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const float *line_ptr = reinterpret_cast<float *>(&outBuf.at(
shun-iwasawa 7bad49
              v * pixel_data_size * static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
              channel_offset_list[c] * static_cast<size_t>(x_stride)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            float val;
shun-iwasawa 7bad49
            // val = line_ptr[u];
shun-iwasawa 7bad49
            tinyexr::cpy4(&val, line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(reinterpret_cast<unsigned int *>(&val));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            float *image = reinterpret_cast<float **>(out_images)[c];
shun-iwasawa 7bad49
            if (line_order == 0) {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              image += static_cast<size_t>(
shun-iwasawa 7bad49
                           (height - 1 - (line_no + static_cast<int>(v)))) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            *image = val;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        assert(0);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    assert(0 && "PIZ is enabled in this build");
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  } else if (compression_type == TINYEXR_COMPRESSIONTYPE_ZIPS ||
shun-iwasawa 7bad49
             compression_type == TINYEXR_COMPRESSIONTYPE_ZIP) {
shun-iwasawa 7bad49
    // Allocate original data size.
shun-iwasawa 7bad49
    std::vector<unsigned char> outBuf(static_cast<size_t>(width) *
shun-iwasawa 7bad49
                                      static_cast<size_t>(num_lines) *
shun-iwasawa 7bad49
                                      pixel_data_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    unsigned long dstLen = static_cast<unsigned long>(outBuf.size());
shun-iwasawa 7bad49
    assert(dstLen > 0);
shun-iwasawa 7bad49
    if (!tinyexr::DecompressZip(
shun-iwasawa 7bad49
            reinterpret_cast<unsigned char *>(&outBuf.at(0)), &dstLen, data_ptr,
shun-iwasawa 7bad49
            static_cast<unsigned long>(data_len))) {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // For ZIP_COMPRESSION:
shun-iwasawa 7bad49
    //   pixel sample data for channel 0 for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel 1 for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel ... for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel n for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel 0 for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel 1 for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel ... for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel n for scanline 1
shun-iwasawa 7bad49
    //   ...
shun-iwasawa 7bad49
    for (size_t c = 0; c < static_cast<size_t>(num_channels); c++) {
shun-iwasawa 7bad49
      if (channels[c].pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const unsigned short *line_ptr = reinterpret_cast<unsigned short *>(
shun-iwasawa 7bad49
              &outBuf.at(v * static_cast<size_t>(pixel_data_size) *
shun-iwasawa 7bad49
                             static_cast<size_t>(width) +
shun-iwasawa 7bad49
                         channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            tinyexr::FP16 hf;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // hf.u = line_ptr[u];
shun-iwasawa 7bad49
            tinyexr::cpy2(&(hf.u), line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap2(reinterpret_cast<unsigned short *>(&hf.u));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            if (requested_pixel_types[c] == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
              unsigned short *image =
shun-iwasawa 7bad49
                  reinterpret_cast<unsigned short **>(out_images)[c];
shun-iwasawa 7bad49
              if (line_order == 0) {
shun-iwasawa 7bad49
                image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              } else {
shun-iwasawa 7bad49
                image += (static_cast<size_t>(height) - 1U -
shun-iwasawa 7bad49
                          (static_cast<size_t>(line_no) + v)) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              }
shun-iwasawa 7bad49
              *image = hf.u;
shun-iwasawa 7bad49
            } else {  // HALF -> FLOAT
shun-iwasawa 7bad49
              tinyexr::FP32 f32 = half_to_float(hf);
shun-iwasawa 7bad49
              float *image = reinterpret_cast<float **>(out_images)[c];
shun-iwasawa 7bad49
              size_t offset = 0;
shun-iwasawa 7bad49
              if (line_order == 0) {
shun-iwasawa 7bad49
                offset = (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              } else {
shun-iwasawa 7bad49
                offset = (static_cast<size_t>(height) - 1U -
shun-iwasawa 7bad49
                          (static_cast<size_t>(line_no) + v)) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              }
shun-iwasawa 7bad49
              image += offset;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              *image = f32.f;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_UINT) {
shun-iwasawa 7bad49
        assert(requested_pixel_types[c] == TINYEXR_PIXELTYPE_UINT);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const unsigned int *line_ptr = reinterpret_cast<unsigned int *>(
shun-iwasawa 7bad49
              &outBuf.at(v * pixel_data_size * static_cast<size_t>(width) +
shun-iwasawa 7bad49
                         channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            unsigned int val;
shun-iwasawa 7bad49
            // val = line_ptr[u];
shun-iwasawa 7bad49
            tinyexr::cpy4(&val, line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(&val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            unsigned int *image =
shun-iwasawa 7bad49
                reinterpret_cast<unsigned int **>(out_images)[c];
shun-iwasawa 7bad49
            if (line_order == 0) {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(height) - 1U -
shun-iwasawa 7bad49
                        (static_cast<size_t>(line_no) + v)) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            *image = val;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
        assert(requested_pixel_types[c] == TINYEXR_PIXELTYPE_FLOAT);
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const float *line_ptr = reinterpret_cast<float *>(
shun-iwasawa 7bad49
              &outBuf.at(v * pixel_data_size * static_cast<size_t>(width) +
shun-iwasawa 7bad49
                         channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            float val;
shun-iwasawa 7bad49
            // val = line_ptr[u];
shun-iwasawa 7bad49
            tinyexr::cpy4(&val, line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(reinterpret_cast<unsigned int *>(&val));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            float *image = reinterpret_cast<float **>(out_images)[c];
shun-iwasawa 7bad49
            if (line_order == 0) {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(height) - 1U -
shun-iwasawa 7bad49
                        (static_cast<size_t>(line_no) + v)) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            *image = val;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        assert(0);
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  } else if (compression_type == TINYEXR_COMPRESSIONTYPE_RLE) {
shun-iwasawa 7bad49
    // Allocate original data size.
shun-iwasawa 7bad49
    std::vector<unsigned char> outBuf(static_cast<size_t>(width) *
shun-iwasawa 7bad49
                                      static_cast<size_t>(num_lines) *
shun-iwasawa 7bad49
                                      pixel_data_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    unsigned long dstLen = static_cast<unsigned long>(outBuf.size());
shun-iwasawa 7bad49
    if (dstLen == 0) {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!tinyexr::DecompressRle(
shun-iwasawa 7bad49
            reinterpret_cast<unsigned char *>(&outBuf.at(0)), dstLen, data_ptr,
shun-iwasawa 7bad49
            static_cast<unsigned long>(data_len))) {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // For RLE_COMPRESSION:
shun-iwasawa 7bad49
    //   pixel sample data for channel 0 for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel 1 for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel ... for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel n for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel 0 for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel 1 for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel ... for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel n for scanline 1
shun-iwasawa 7bad49
    //   ...
shun-iwasawa 7bad49
    for (size_t c = 0; c < static_cast<size_t>(num_channels); c++) {
shun-iwasawa 7bad49
      if (channels[c].pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const unsigned short *line_ptr = reinterpret_cast<unsigned short *>(
shun-iwasawa 7bad49
              &outBuf.at(v * static_cast<size_t>(pixel_data_size) *
shun-iwasawa 7bad49
                             static_cast<size_t>(width) +
shun-iwasawa 7bad49
                         channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            tinyexr::FP16 hf;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // hf.u = line_ptr[u];
shun-iwasawa 7bad49
            tinyexr::cpy2(&(hf.u), line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap2(reinterpret_cast<unsigned short *>(&hf.u));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            if (requested_pixel_types[c] == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
              unsigned short *image =
shun-iwasawa 7bad49
                  reinterpret_cast<unsigned short **>(out_images)[c];
shun-iwasawa 7bad49
              if (line_order == 0) {
shun-iwasawa 7bad49
                image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              } else {
shun-iwasawa 7bad49
                image += (static_cast<size_t>(height) - 1U -
shun-iwasawa 7bad49
                          (static_cast<size_t>(line_no) + v)) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              }
shun-iwasawa 7bad49
              *image = hf.u;
shun-iwasawa 7bad49
            } else {  // HALF -> FLOAT
shun-iwasawa 7bad49
              tinyexr::FP32 f32 = half_to_float(hf);
shun-iwasawa 7bad49
              float *image = reinterpret_cast<float **>(out_images)[c];
shun-iwasawa 7bad49
              if (line_order == 0) {
shun-iwasawa 7bad49
                image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              } else {
shun-iwasawa 7bad49
                image += (static_cast<size_t>(height) - 1U -
shun-iwasawa 7bad49
                          (static_cast<size_t>(line_no) + v)) *
shun-iwasawa 7bad49
                             static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                         u;
shun-iwasawa 7bad49
              }
shun-iwasawa 7bad49
              *image = f32.f;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_UINT) {
shun-iwasawa 7bad49
        assert(requested_pixel_types[c] == TINYEXR_PIXELTYPE_UINT);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const unsigned int *line_ptr = reinterpret_cast<unsigned int *>(
shun-iwasawa 7bad49
              &outBuf.at(v * pixel_data_size * static_cast<size_t>(width) +
shun-iwasawa 7bad49
                         channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            unsigned int val;
shun-iwasawa 7bad49
            // val = line_ptr[u];
shun-iwasawa 7bad49
            tinyexr::cpy4(&val, line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(&val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            unsigned int *image =
shun-iwasawa 7bad49
                reinterpret_cast<unsigned int **>(out_images)[c];
shun-iwasawa 7bad49
            if (line_order == 0) {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(height) - 1U -
shun-iwasawa 7bad49
                        (static_cast<size_t>(line_no) + v)) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            *image = val;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
        assert(requested_pixel_types[c] == TINYEXR_PIXELTYPE_FLOAT);
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const float *line_ptr = reinterpret_cast<float *>(
shun-iwasawa 7bad49
              &outBuf.at(v * pixel_data_size * static_cast<size_t>(width) +
shun-iwasawa 7bad49
                         channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            float val;
shun-iwasawa 7bad49
            // val = line_ptr[u];
shun-iwasawa 7bad49
            tinyexr::cpy4(&val, line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(reinterpret_cast<unsigned int *>(&val));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            float *image = reinterpret_cast<float **>(out_images)[c];
shun-iwasawa 7bad49
            if (line_order == 0) {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(height) - 1U -
shun-iwasawa 7bad49
                        (static_cast<size_t>(line_no) + v)) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            *image = val;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        assert(0);
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  } else if (compression_type == TINYEXR_COMPRESSIONTYPE_ZFP) {
shun-iwasawa 7bad49
#if TINYEXR_USE_ZFP
shun-iwasawa 7bad49
    tinyexr::ZFPCompressionParam zfp_compression_param;
shun-iwasawa 7bad49
    std::string e;
shun-iwasawa 7bad49
    if (!tinyexr::FindZFPCompressionParam(&zfp_compression_param, attributes,
shun-iwasawa 7bad49
                                          int(num_attributes), &e)) {
shun-iwasawa 7bad49
      // This code path should not be reachable.
shun-iwasawa 7bad49
      assert(0);
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // Allocate original data size.
shun-iwasawa 7bad49
    std::vector<unsigned char> outBuf(static_cast<size_t>(width) *
shun-iwasawa 7bad49
                                      static_cast<size_t>(num_lines) *
shun-iwasawa 7bad49
                                      pixel_data_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    unsigned long dstLen = outBuf.size();
shun-iwasawa 7bad49
    assert(dstLen > 0);
shun-iwasawa 7bad49
    tinyexr::DecompressZfp(reinterpret_cast<float *>(&outBuf.at(0)), width,
shun-iwasawa 7bad49
                           num_lines, num_channels, data_ptr,
shun-iwasawa 7bad49
                           static_cast<unsigned long>(data_len),
shun-iwasawa 7bad49
                           zfp_compression_param);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // For ZFP_COMPRESSION:
shun-iwasawa 7bad49
    //   pixel sample data for channel 0 for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel 1 for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel ... for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel n for scanline 0
shun-iwasawa 7bad49
    //   pixel sample data for channel 0 for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel 1 for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel ... for scanline 1
shun-iwasawa 7bad49
    //   pixel sample data for channel n for scanline 1
shun-iwasawa 7bad49
    //   ...
shun-iwasawa 7bad49
    for (size_t c = 0; c < static_cast<size_t>(num_channels); c++) {
shun-iwasawa 7bad49
      assert(channels[c].pixel_type == TINYEXR_PIXELTYPE_FLOAT);
shun-iwasawa 7bad49
      if (channels[c].pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
        assert(requested_pixel_types[c] == TINYEXR_PIXELTYPE_FLOAT);
shun-iwasawa 7bad49
        for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
          const float *line_ptr = reinterpret_cast<float *>(
shun-iwasawa 7bad49
              &outBuf.at(v * pixel_data_size * static_cast<size_t>(width) +
shun-iwasawa 7bad49
                         channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (size_t u = 0; u < static_cast<size_t>(width); u++) {
shun-iwasawa 7bad49
            float val;
shun-iwasawa 7bad49
            tinyexr::cpy4(&val, line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(reinterpret_cast<unsigned int *>(&val));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            float *image = reinterpret_cast<float **>(out_images)[c];
shun-iwasawa 7bad49
            if (line_order == 0) {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(line_no) + v) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              image += (static_cast<size_t>(height) - 1U -
shun-iwasawa 7bad49
                        (static_cast<size_t>(line_no) + v)) *
shun-iwasawa 7bad49
                           static_cast<size_t>(x_stride) +
shun-iwasawa 7bad49
                       u;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            *image = val;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        assert(0);
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    (void)attributes;
shun-iwasawa 7bad49
    (void)num_attributes;
shun-iwasawa 7bad49
    (void)num_channels;
shun-iwasawa 7bad49
    assert(0);
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  } else if (compression_type == TINYEXR_COMPRESSIONTYPE_NONE) {
shun-iwasawa 7bad49
    for (size_t c = 0; c < num_channels; c++) {
shun-iwasawa 7bad49
      for (size_t v = 0; v < static_cast<size_t>(num_lines); v++) {
shun-iwasawa 7bad49
        if (channels[c].pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
          const unsigned short *line_ptr =
shun-iwasawa 7bad49
              reinterpret_cast<const unsigned short *>(
shun-iwasawa 7bad49
                  data_ptr + v * pixel_data_size * size_t(width) +
shun-iwasawa 7bad49
                  channel_offset_list[c] * static_cast<size_t>(width));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          if (requested_pixel_types[c] == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
            unsigned short *outLine =
shun-iwasawa 7bad49
                reinterpret_cast<unsigned short *>(out_images[c]);
shun-iwasawa 7bad49
            if (line_order == 0) {
shun-iwasawa 7bad49
              outLine += (size_t(y) + v) * size_t(x_stride);
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              outLine +=
shun-iwasawa 7bad49
                  (size_t(height) - 1 - (size_t(y) + v)) * size_t(x_stride);
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            for (int u = 0; u < width; u++) {
shun-iwasawa 7bad49
              tinyexr::FP16 hf;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              // hf.u = line_ptr[u];
shun-iwasawa 7bad49
              tinyexr::cpy2(&(hf.u), line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              tinyexr::swap2(reinterpret_cast<unsigned short *>(&hf.u));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              outLine[u] = hf.u;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
          } else if (requested_pixel_types[c] == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
            float *outLine = reinterpret_cast<float *>(out_images[c]);
shun-iwasawa 7bad49
            if (line_order == 0) {
shun-iwasawa 7bad49
              outLine += (size_t(y) + v) * size_t(x_stride);
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              outLine +=
shun-iwasawa 7bad49
                  (size_t(height) - 1 - (size_t(y) + v)) * size_t(x_stride);
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            if (reinterpret_cast<const unsigned char *>(line_ptr + width) >
shun-iwasawa 7bad49
                (data_ptr + data_len)) {
shun-iwasawa 7bad49
              // Insufficient data size
shun-iwasawa 7bad49
              return false;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            for (int u = 0; u < width; u++) {
shun-iwasawa 7bad49
              tinyexr::FP16 hf;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              // address may not be aliged. use byte-wise copy for safety.#76
shun-iwasawa 7bad49
              // hf.u = line_ptr[u];
shun-iwasawa 7bad49
              tinyexr::cpy2(&(hf.u), line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              tinyexr::swap2(reinterpret_cast<unsigned short *>(&hf.u));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              tinyexr::FP32 f32 = half_to_float(hf);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              outLine[u] = f32.f;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
          } else {
shun-iwasawa 7bad49
            assert(0);
shun-iwasawa 7bad49
            return false;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
          const float *line_ptr = reinterpret_cast<const float *>(
shun-iwasawa 7bad49
              data_ptr + v * pixel_data_size * size_t(width) +
shun-iwasawa 7bad49
              channel_offset_list[c] * static_cast<size_t>(width));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          float *outLine = reinterpret_cast<float *>(out_images[c]);
shun-iwasawa 7bad49
          if (line_order == 0) {
shun-iwasawa 7bad49
            outLine += (size_t(y) + v) * size_t(x_stride);
shun-iwasawa 7bad49
          } else {
shun-iwasawa 7bad49
            outLine +=
shun-iwasawa 7bad49
                (size_t(height) - 1 - (size_t(y) + v)) * size_t(x_stride);
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          if (reinterpret_cast<const unsigned char *>(line_ptr + width) >
shun-iwasawa 7bad49
              (data_ptr + data_len)) {
shun-iwasawa 7bad49
            // Insufficient data size
shun-iwasawa 7bad49
            return false;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          for (int u = 0; u < width; u++) {
shun-iwasawa 7bad49
            float val;
shun-iwasawa 7bad49
            tinyexr::cpy4(&val, line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(reinterpret_cast<unsigned int *>(&val));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            outLine[u] = val;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_UINT) {
shun-iwasawa 7bad49
          const unsigned int *line_ptr = reinterpret_cast<const unsigned int *>(
shun-iwasawa 7bad49
              data_ptr + v * pixel_data_size * size_t(width) +
shun-iwasawa 7bad49
              channel_offset_list[c] * static_cast<size_t>(width));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          unsigned int *outLine =
shun-iwasawa 7bad49
              reinterpret_cast<unsigned int *>(out_images[c]);
shun-iwasawa 7bad49
          if (line_order == 0) {
shun-iwasawa 7bad49
            outLine += (size_t(y) + v) * size_t(x_stride);
shun-iwasawa 7bad49
          } else {
shun-iwasawa 7bad49
            outLine +=
shun-iwasawa 7bad49
                (size_t(height) - 1 - (size_t(y) + v)) * size_t(x_stride);
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          for (int u = 0; u < width; u++) {
shun-iwasawa 7bad49
            if (reinterpret_cast<const unsigned char *>(line_ptr + u) >=
shun-iwasawa 7bad49
                (data_ptr + data_len)) {
shun-iwasawa 7bad49
              // Corrupsed data?
shun-iwasawa 7bad49
              return false;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            unsigned int val;
shun-iwasawa 7bad49
            tinyexr::cpy4(&val, line_ptr + u);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(reinterpret_cast<unsigned int *>(&val));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            outLine[u] = val;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool DecodeTiledPixelData(
shun-iwasawa 7bad49
    unsigned char **out_images, int *width, int *height,
shun-iwasawa 7bad49
    const int *requested_pixel_types, const unsigned char *data_ptr,
shun-iwasawa 7bad49
    size_t data_len, int compression_type, int line_order, int data_width,
shun-iwasawa 7bad49
    int data_height, int tile_offset_x, int tile_offset_y, int tile_size_x,
shun-iwasawa 7bad49
    int tile_size_y, size_t pixel_data_size, size_t num_attributes,
shun-iwasawa 7bad49
    const EXRAttribute *attributes, size_t num_channels,
shun-iwasawa 7bad49
    const EXRChannelInfo *channels,
shun-iwasawa 7bad49
    const std::vector<size_t> &channel_offset_list) {
shun-iwasawa 7bad49
  // Here, data_width and data_height are the dimensions of the current (sub)level.
shun-iwasawa 7bad49
  if (tile_size_x * tile_offset_x > data_width ||
shun-iwasawa 7bad49
      tile_size_y * tile_offset_y > data_height) {
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Compute actual image size in a tile.
shun-iwasawa 7bad49
  if ((tile_offset_x + 1) * tile_size_x >= data_width) {
shun-iwasawa 7bad49
    (*width) = data_width - (tile_offset_x * tile_size_x);
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    (*width) = tile_size_x;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if ((tile_offset_y + 1) * tile_size_y >= data_height) {
shun-iwasawa 7bad49
    (*height) = data_height - (tile_offset_y * tile_size_y);
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    (*height) = tile_size_y;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Image size = tile size.
shun-iwasawa 7bad49
  return DecodePixelData(out_images, requested_pixel_types, data_ptr, data_len,
shun-iwasawa 7bad49
                         compression_type, line_order, (*width), tile_size_y,
shun-iwasawa 7bad49
                         /* stride */ tile_size_x, /* y */ 0, /* line_no */ 0,
shun-iwasawa 7bad49
                         (*height), pixel_data_size, num_attributes, attributes,
shun-iwasawa 7bad49
                         num_channels, channels, channel_offset_list);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool ComputeChannelLayout(std::vector<size_t> *channel_offset_list,
shun-iwasawa 7bad49
                                 int *pixel_data_size, size_t *channel_offset,
shun-iwasawa 7bad49
                                 int num_channels,
shun-iwasawa 7bad49
                                 const EXRChannelInfo *channels) {
shun-iwasawa 7bad49
  channel_offset_list->resize(static_cast<size_t>(num_channels));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  (*pixel_data_size) = 0;
shun-iwasawa 7bad49
  (*channel_offset) = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t c = 0; c < static_cast<size_t>(num_channels); c++) {
shun-iwasawa 7bad49
    (*channel_offset_list)[c] = (*channel_offset);
shun-iwasawa 7bad49
    if (channels[c].pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
      (*pixel_data_size) += sizeof(unsigned short);
shun-iwasawa 7bad49
      (*channel_offset) += sizeof(unsigned short);
shun-iwasawa 7bad49
    } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
      (*pixel_data_size) += sizeof(float);
shun-iwasawa 7bad49
      (*channel_offset) += sizeof(float);
shun-iwasawa 7bad49
    } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_UINT) {
shun-iwasawa 7bad49
      (*pixel_data_size) += sizeof(unsigned int);
shun-iwasawa 7bad49
      (*channel_offset) += sizeof(unsigned int);
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      // ???
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static unsigned char **AllocateImage(int num_channels,
shun-iwasawa 7bad49
                                     const EXRChannelInfo *channels,
shun-iwasawa 7bad49
                                     const int *requested_pixel_types,
shun-iwasawa 7bad49
                                     int data_width, int data_height) {
shun-iwasawa 7bad49
  unsigned char **images =
shun-iwasawa 7bad49
      reinterpret_cast<unsigned char **>(static_cast<float **>(
shun-iwasawa 7bad49
          malloc(sizeof(float *) * static_cast<size_t>(num_channels))));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t c = 0; c < static_cast<size_t>(num_channels); c++) {
shun-iwasawa 7bad49
    size_t data_len =
shun-iwasawa 7bad49
        static_cast<size_t>(data_width) * static_cast<size_t>(data_height);
shun-iwasawa 7bad49
    if (channels[c].pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
      // pixel_data_size += sizeof(unsigned short);
shun-iwasawa 7bad49
      // channel_offset += sizeof(unsigned short);
shun-iwasawa 7bad49
      // Alloc internal image for half type.
shun-iwasawa 7bad49
      if (requested_pixel_types[c] == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
        images[c] =
shun-iwasawa 7bad49
            reinterpret_cast<unsigned char *>(static_cast<unsigned short *>(
shun-iwasawa 7bad49
                malloc(sizeof(unsigned short) * data_len)));
shun-iwasawa 7bad49
      } else if (requested_pixel_types[c] == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
        images[c] = reinterpret_cast<unsigned char *>(
shun-iwasawa 7bad49
            static_cast<float *>(malloc(sizeof(float) * data_len)));
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        assert(0);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
      // pixel_data_size += sizeof(float);
shun-iwasawa 7bad49
      // channel_offset += sizeof(float);
shun-iwasawa 7bad49
      images[c] = reinterpret_cast<unsigned char *>(
shun-iwasawa 7bad49
          static_cast<float *>(malloc(sizeof(float) * data_len)));
shun-iwasawa 7bad49
    } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_UINT) {
shun-iwasawa 7bad49
      // pixel_data_size += sizeof(unsigned int);
shun-iwasawa 7bad49
      // channel_offset += sizeof(unsigned int);
shun-iwasawa 7bad49
      images[c] = reinterpret_cast<unsigned char *>(
shun-iwasawa 7bad49
          static_cast<unsigned int *>(malloc(sizeof(unsigned int) * data_len)));
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      assert(0);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return images;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
static inline std::wstring UTF8ToWchar(const std::string &str) {
shun-iwasawa 7bad49
  int wstr_size =
shun-iwasawa 7bad49
      MultiByteToWideChar(CP_UTF8, 0, str.data(), (int)str.size(), NULL, 0);
shun-iwasawa 7bad49
  std::wstring wstr(wstr_size, 0);
shun-iwasawa 7bad49
  MultiByteToWideChar(CP_UTF8, 0, str.data(), (int)str.size(), &wstr[0],
shun-iwasawa 7bad49
                      (int)wstr.size());
shun-iwasawa 7bad49
  return wstr;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int ParseEXRHeader(HeaderInfo *info, bool *empty_header,
shun-iwasawa 7bad49
                          const EXRVersion *version, std::string *err,
shun-iwasawa 7bad49
                          const unsigned char *buf, size_t size) {
shun-iwasawa 7bad49
  const char *marker = reinterpret_cast<const char *>(&buf[0]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (empty_header) {
shun-iwasawa 7bad49
    (*empty_header) = false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (version->multipart) {
shun-iwasawa 7bad49
    if (size > 0 && marker[0] == '\0') {
shun-iwasawa 7bad49
      // End of header list.
shun-iwasawa 7bad49
      if (empty_header) {
shun-iwasawa 7bad49
        (*empty_header) = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // According to the spec, the header of every OpenEXR file must contain at
shun-iwasawa 7bad49
  // least the following attributes:
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // channels chlist
shun-iwasawa 7bad49
  // compression compression
shun-iwasawa 7bad49
  // dataWindow box2i
shun-iwasawa 7bad49
  // displayWindow box2i
shun-iwasawa 7bad49
  // lineOrder lineOrder
shun-iwasawa 7bad49
  // pixelAspectRatio float
shun-iwasawa 7bad49
  // screenWindowCenter v2f
shun-iwasawa 7bad49
  // screenWindowWidth float
shun-iwasawa 7bad49
  bool has_channels = false;
shun-iwasawa 7bad49
  bool has_compression = false;
shun-iwasawa 7bad49
  bool has_data_window = false;
shun-iwasawa 7bad49
  bool has_display_window = false;
shun-iwasawa 7bad49
  bool has_line_order = false;
shun-iwasawa 7bad49
  bool has_pixel_aspect_ratio = false;
shun-iwasawa 7bad49
  bool has_screen_window_center = false;
shun-iwasawa 7bad49
  bool has_screen_window_width = false;
shun-iwasawa 7bad49
  bool has_name = false;
shun-iwasawa 7bad49
  bool has_type = false;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  info->name.clear();
shun-iwasawa 7bad49
  info->type.clear();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  info->data_window.min_x = 0;
shun-iwasawa 7bad49
  info->data_window.min_y = 0;
shun-iwasawa 7bad49
  info->data_window.max_x = 0;
shun-iwasawa 7bad49
  info->data_window.max_y = 0;
shun-iwasawa 7bad49
  info->line_order = 0;  // @fixme
shun-iwasawa 7bad49
  info->display_window.min_x = 0;
shun-iwasawa 7bad49
  info->display_window.min_y = 0;
shun-iwasawa 7bad49
  info->display_window.max_x = 0;
shun-iwasawa 7bad49
  info->display_window.max_y = 0;
shun-iwasawa 7bad49
  info->screen_window_center[0] = 0.0f;
shun-iwasawa 7bad49
  info->screen_window_center[1] = 0.0f;
shun-iwasawa 7bad49
  info->screen_window_width = -1.0f;
shun-iwasawa 7bad49
  info->pixel_aspect_ratio = -1.0f;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  info->tiled = 0;
shun-iwasawa 7bad49
  info->tile_size_x = -1;
shun-iwasawa 7bad49
  info->tile_size_y = -1;
shun-iwasawa 7bad49
  info->tile_level_mode = -1;
shun-iwasawa 7bad49
  info->tile_rounding_mode = -1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  info->attributes.clear();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Read attributes
shun-iwasawa 7bad49
  size_t orig_size = size;
shun-iwasawa 7bad49
  for (size_t nattr = 0; nattr < TINYEXR_MAX_HEADER_ATTRIBUTES; nattr++) {
shun-iwasawa 7bad49
    if (0 == size) {
shun-iwasawa 7bad49
      if (err) {
shun-iwasawa 7bad49
        (*err) += "Insufficient data size for attributes.\n";
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    } else if (marker[0] == '\0') {
shun-iwasawa 7bad49
      size--;
shun-iwasawa 7bad49
      break;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    std::string attr_name;
shun-iwasawa 7bad49
    std::string attr_type;
shun-iwasawa 7bad49
    std::vector<unsigned char> data;
shun-iwasawa 7bad49
    size_t marker_size;
shun-iwasawa 7bad49
    if (!tinyexr::ReadAttribute(&attr_name, &attr_type, &data, &marker_size,
shun-iwasawa 7bad49
                                marker, size)) {
shun-iwasawa 7bad49
      if (err) {
shun-iwasawa 7bad49
        (*err) += "Failed to read attribute.\n";
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    marker += marker_size;
shun-iwasawa 7bad49
    size -= marker_size;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // For a multipart file, the version field 9th bit is 0.
shun-iwasawa 7bad49
    if ((version->tiled || version->multipart || version->non_image) && attr_name.compare("tiles") == 0) {
shun-iwasawa 7bad49
      unsigned int x_size, y_size;
shun-iwasawa 7bad49
      unsigned char tile_mode;
shun-iwasawa 7bad49
      assert(data.size() == 9);
shun-iwasawa 7bad49
      memcpy(&x_size, &data.at(0), sizeof(int));
shun-iwasawa 7bad49
      memcpy(&y_size, &data.at(4), sizeof(int));
shun-iwasawa 7bad49
      tile_mode = data[8];
shun-iwasawa 7bad49
      tinyexr::swap4(&x_size);
shun-iwasawa 7bad49
      tinyexr::swap4(&y_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (x_size > static_cast<unsigned int>(std::numeric_limits<int>::max()) ||
shun-iwasawa 7bad49
          y_size > static_cast<unsigned int>(std::numeric_limits<int>::max())) {
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          (*err) = "Tile sizes were invalid.";
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return TINYEXR_ERROR_UNSUPPORTED_FORMAT;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      info->tile_size_x = static_cast<int>(x_size);
shun-iwasawa 7bad49
      info->tile_size_y = static_cast<int>(y_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      // mode = levelMode + roundingMode * 16
shun-iwasawa 7bad49
      info->tile_level_mode = tile_mode & 0x3;
shun-iwasawa 7bad49
      info->tile_rounding_mode = (tile_mode >> 4) & 0x1;
shun-iwasawa 7bad49
      info->tiled = 1;
shun-iwasawa 7bad49
    } else if (attr_name.compare("compression") == 0) {
shun-iwasawa 7bad49
      bool ok = false;
shun-iwasawa 7bad49
      if (data[0] < TINYEXR_COMPRESSIONTYPE_PIZ) {
shun-iwasawa 7bad49
        ok = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (data[0] == TINYEXR_COMPRESSIONTYPE_PIZ) {
shun-iwasawa 7bad49
#if TINYEXR_USE_PIZ
shun-iwasawa 7bad49
        ok = true;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          (*err) = "PIZ compression is not supported.";
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return TINYEXR_ERROR_UNSUPPORTED_FORMAT;
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (data[0] == TINYEXR_COMPRESSIONTYPE_ZFP) {
shun-iwasawa 7bad49
#if TINYEXR_USE_ZFP
shun-iwasawa 7bad49
        ok = true;
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          (*err) = "ZFP compression is not supported.";
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return TINYEXR_ERROR_UNSUPPORTED_FORMAT;
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (!ok) {
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          (*err) = "Unknown compression type.";
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return TINYEXR_ERROR_UNSUPPORTED_FORMAT;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      info->compression_type = static_cast<int>(data[0]);
shun-iwasawa 7bad49
      has_compression = true;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    } else if (attr_name.compare("channels") == 0) {
shun-iwasawa 7bad49
      // name: zero-terminated string, from 1 to 255 bytes long
shun-iwasawa 7bad49
      // pixel type: int, possible values are: UINT = 0 HALF = 1 FLOAT = 2
shun-iwasawa 7bad49
      // pLinear: unsigned char, possible values are 0 and 1
shun-iwasawa 7bad49
      // reserved: three chars, should be zero
shun-iwasawa 7bad49
      // xSampling: int
shun-iwasawa 7bad49
      // ySampling: int
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (!ReadChannelInfo(info->channels, data)) {
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          (*err) += "Failed to parse channel info.\n";
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (info->channels.size() < 1) {
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          (*err) += "# of channels is zero.\n";
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      has_channels = true;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    } else if (attr_name.compare("dataWindow") == 0) {
shun-iwasawa 7bad49
      if (data.size() >= 16) {
shun-iwasawa 7bad49
        memcpy(&info->data_window.min_x, &data.at(0), sizeof(int));
shun-iwasawa 7bad49
        memcpy(&info->data_window.min_y, &data.at(4), sizeof(int));
shun-iwasawa 7bad49
        memcpy(&info->data_window.max_x, &data.at(8), sizeof(int));
shun-iwasawa 7bad49
        memcpy(&info->data_window.max_y, &data.at(12), sizeof(int));
shun-iwasawa 7bad49
        tinyexr::swap4(&info->data_window.min_x);
shun-iwasawa 7bad49
        tinyexr::swap4(&info->data_window.min_y);
shun-iwasawa 7bad49
        tinyexr::swap4(&info->data_window.max_x);
shun-iwasawa 7bad49
        tinyexr::swap4(&info->data_window.max_y);
shun-iwasawa 7bad49
        has_data_window = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (attr_name.compare("displayWindow") == 0) {
shun-iwasawa 7bad49
      if (data.size() >= 16) {
shun-iwasawa 7bad49
        memcpy(&info->display_window.min_x, &data.at(0), sizeof(int));
shun-iwasawa 7bad49
        memcpy(&info->display_window.min_y, &data.at(4), sizeof(int));
shun-iwasawa 7bad49
        memcpy(&info->display_window.max_x, &data.at(8), sizeof(int));
shun-iwasawa 7bad49
        memcpy(&info->display_window.max_y, &data.at(12), sizeof(int));
shun-iwasawa 7bad49
        tinyexr::swap4(&info->display_window.min_x);
shun-iwasawa 7bad49
        tinyexr::swap4(&info->display_window.min_y);
shun-iwasawa 7bad49
        tinyexr::swap4(&info->display_window.max_x);
shun-iwasawa 7bad49
        tinyexr::swap4(&info->display_window.max_y);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        has_display_window = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (attr_name.compare("lineOrder") == 0) {
shun-iwasawa 7bad49
      if (data.size() >= 1) {
shun-iwasawa 7bad49
        info->line_order = static_cast<int>(data[0]);
shun-iwasawa 7bad49
        has_line_order = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (attr_name.compare("pixelAspectRatio") == 0) {
shun-iwasawa 7bad49
      if (data.size() >= sizeof(float)) {
shun-iwasawa 7bad49
        memcpy(&info->pixel_aspect_ratio, &data.at(0), sizeof(float));
shun-iwasawa 7bad49
        tinyexr::swap4(&info->pixel_aspect_ratio);
shun-iwasawa 7bad49
        has_pixel_aspect_ratio = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (attr_name.compare("screenWindowCenter") == 0) {
shun-iwasawa 7bad49
      if (data.size() >= 8) {
shun-iwasawa 7bad49
        memcpy(&info->screen_window_center[0], &data.at(0), sizeof(float));
shun-iwasawa 7bad49
        memcpy(&info->screen_window_center[1], &data.at(4), sizeof(float));
shun-iwasawa 7bad49
        tinyexr::swap4(&info->screen_window_center[0]);
shun-iwasawa 7bad49
        tinyexr::swap4(&info->screen_window_center[1]);
shun-iwasawa 7bad49
        has_screen_window_center = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (attr_name.compare("screenWindowWidth") == 0) {
shun-iwasawa 7bad49
      if (data.size() >= sizeof(float)) {
shun-iwasawa 7bad49
        memcpy(&info->screen_window_width, &data.at(0), sizeof(float));
shun-iwasawa 7bad49
        tinyexr::swap4(&info->screen_window_width);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        has_screen_window_width = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (attr_name.compare("chunkCount") == 0) {
shun-iwasawa 7bad49
      if (data.size() >= sizeof(int)) {
shun-iwasawa 7bad49
        memcpy(&info->chunk_count, &data.at(0), sizeof(int));
shun-iwasawa 7bad49
        tinyexr::swap4(&info->chunk_count);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (attr_name.compare("name") == 0) {
shun-iwasawa 7bad49
      if (!data.empty() && data[0]) {
shun-iwasawa 7bad49
        data.push_back(0);
shun-iwasawa 7bad49
        size_t len = strlen(reinterpret_cast<const char*>(&data[0]));
shun-iwasawa 7bad49
        info->name.resize(len);
shun-iwasawa 7bad49
        info->name.assign(reinterpret_cast<const char*>(&data[0]), len);
shun-iwasawa 7bad49
        has_name = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (attr_name.compare("type") == 0) {
shun-iwasawa 7bad49
      if (!data.empty() && data[0]) {
shun-iwasawa 7bad49
        data.push_back(0);
shun-iwasawa 7bad49
        size_t len = strlen(reinterpret_cast<const char*>(&data[0]));
shun-iwasawa 7bad49
        info->type.resize(len);
shun-iwasawa 7bad49
        info->type.assign(reinterpret_cast<const char*>(&data[0]), len);
shun-iwasawa 7bad49
        has_type = true;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      // Custom attribute(up to TINYEXR_MAX_CUSTOM_ATTRIBUTES)
shun-iwasawa 7bad49
      if (info->attributes.size() < TINYEXR_MAX_CUSTOM_ATTRIBUTES) {
shun-iwasawa 7bad49
        EXRAttribute attrib;
shun-iwasawa 7bad49
#ifdef _MSC_VER
shun-iwasawa 7bad49
        strncpy_s(attrib.name, attr_name.c_str(), 255);
shun-iwasawa 7bad49
        strncpy_s(attrib.type, attr_type.c_str(), 255);
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
        strncpy(attrib.name, attr_name.c_str(), 255);
shun-iwasawa 7bad49
        strncpy(attrib.type, attr_type.c_str(), 255);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
        attrib.name[255] = '\0';
shun-iwasawa 7bad49
        attrib.type[255] = '\0';
shun-iwasawa 7bad49
        attrib.size = static_cast<int>(data.size());
shun-iwasawa 7bad49
        attrib.value = static_cast<unsigned char *>(malloc(data.size()));
shun-iwasawa 7bad49
        memcpy(reinterpret_cast<char *>(attrib.value), &data.at(0),
shun-iwasawa 7bad49
               data.size());
shun-iwasawa 7bad49
        info->attributes.push_back(attrib);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Check if required attributes exist
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    std::stringstream ss_err;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!has_compression) {
shun-iwasawa 7bad49
      ss_err << "\"compression\" attribute not found in the header."
shun-iwasawa 7bad49
             << std::endl;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!has_channels) {
shun-iwasawa 7bad49
      ss_err << "\"channels\" attribute not found in the header." << std::endl;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!has_line_order) {
shun-iwasawa 7bad49
      ss_err << "\"lineOrder\" attribute not found in the header." << std::endl;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!has_display_window) {
shun-iwasawa 7bad49
      ss_err << "\"displayWindow\" attribute not found in the header."
shun-iwasawa 7bad49
             << std::endl;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!has_data_window) {
shun-iwasawa 7bad49
      ss_err << "\"dataWindow\" attribute not found in the header or invalid."
shun-iwasawa 7bad49
             << std::endl;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!has_pixel_aspect_ratio) {
shun-iwasawa 7bad49
      ss_err << "\"pixelAspectRatio\" attribute not found in the header."
shun-iwasawa 7bad49
             << std::endl;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!has_screen_window_width) {
shun-iwasawa 7bad49
      ss_err << "\"screenWindowWidth\" attribute not found in the header."
shun-iwasawa 7bad49
             << std::endl;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!has_screen_window_center) {
shun-iwasawa 7bad49
      ss_err << "\"screenWindowCenter\" attribute not found in the header."
shun-iwasawa 7bad49
             << std::endl;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (version->multipart || version->non_image) {
shun-iwasawa 7bad49
      if (!has_name) {
shun-iwasawa 7bad49
        ss_err << "\"name\" attribute not found in the header."
shun-iwasawa 7bad49
          << std::endl;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      if (!has_type) {
shun-iwasawa 7bad49
        ss_err << "\"type\" attribute not found in the header."
shun-iwasawa 7bad49
          << std::endl;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!(ss_err.str().empty())) {
shun-iwasawa 7bad49
      if (err) {
shun-iwasawa 7bad49
        (*err) += ss_err.str();
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_HEADER;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  info->header_len = static_cast<unsigned int>(orig_size - size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// C++ HeaderInfo to C EXRHeader conversion.
shun-iwasawa 7bad49
static void ConvertHeader(EXRHeader *exr_header, const HeaderInfo &info) {
shun-iwasawa 7bad49
  exr_header->pixel_aspect_ratio = info.pixel_aspect_ratio;
shun-iwasawa 7bad49
  exr_header->screen_window_center[0] = info.screen_window_center[0];
shun-iwasawa 7bad49
  exr_header->screen_window_center[1] = info.screen_window_center[1];
shun-iwasawa 7bad49
  exr_header->screen_window_width = info.screen_window_width;
shun-iwasawa 7bad49
  exr_header->chunk_count = info.chunk_count;
shun-iwasawa 7bad49
  exr_header->display_window.min_x = info.display_window.min_x;
shun-iwasawa 7bad49
  exr_header->display_window.min_y = info.display_window.min_y;
shun-iwasawa 7bad49
  exr_header->display_window.max_x = info.display_window.max_x;
shun-iwasawa 7bad49
  exr_header->display_window.max_y = info.display_window.max_y;
shun-iwasawa 7bad49
  exr_header->data_window.min_x = info.data_window.min_x;
shun-iwasawa 7bad49
  exr_header->data_window.min_y = info.data_window.min_y;
shun-iwasawa 7bad49
  exr_header->data_window.max_x = info.data_window.max_x;
shun-iwasawa 7bad49
  exr_header->data_window.max_y = info.data_window.max_y;
shun-iwasawa 7bad49
  exr_header->line_order = info.line_order;
shun-iwasawa 7bad49
  exr_header->compression_type = info.compression_type;
shun-iwasawa 7bad49
  exr_header->tiled = info.tiled;
shun-iwasawa 7bad49
  exr_header->tile_size_x = info.tile_size_x;
shun-iwasawa 7bad49
  exr_header->tile_size_y = info.tile_size_y;
shun-iwasawa 7bad49
  exr_header->tile_level_mode = info.tile_level_mode;
shun-iwasawa 7bad49
  exr_header->tile_rounding_mode = info.tile_rounding_mode;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  EXRSetNameAttr(exr_header, info.name.c_str());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (!info.type.empty()) {
shun-iwasawa 7bad49
    if (info.type == "scanlineimage") {
shun-iwasawa 7bad49
      assert(!exr_header->tiled);
shun-iwasawa 7bad49
    } else if (info.type == "tiledimage") {
shun-iwasawa 7bad49
      assert(exr_header->tiled);
shun-iwasawa 7bad49
    } else if (info.type == "deeptile") {
shun-iwasawa 7bad49
      exr_header->non_image = 1;
shun-iwasawa 7bad49
      assert(exr_header->tiled);
shun-iwasawa 7bad49
    } else if (info.type == "deepscanline") {
shun-iwasawa 7bad49
      exr_header->non_image = 1;
shun-iwasawa 7bad49
      assert(!exr_header->tiled);
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      assert(false);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  exr_header->num_channels = static_cast<int>(info.channels.size());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  exr_header->channels = static_cast<EXRChannelInfo *>(malloc(
shun-iwasawa 7bad49
      sizeof(EXRChannelInfo) * static_cast<size_t>(exr_header->num_channels)));
shun-iwasawa 7bad49
  for (size_t c = 0; c < static_cast<size_t>(exr_header->num_channels); c++) {
shun-iwasawa 7bad49
#ifdef _MSC_VER
shun-iwasawa 7bad49
    strncpy_s(exr_header->channels[c].name, info.channels[c].name.c_str(), 255);
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    strncpy(exr_header->channels[c].name, info.channels[c].name.c_str(), 255);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    // manually add '\0' for safety.
shun-iwasawa 7bad49
    exr_header->channels[c].name[255] = '\0';
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    exr_header->channels[c].pixel_type = info.channels[c].pixel_type;
shun-iwasawa 7bad49
    exr_header->channels[c].p_linear = info.channels[c].p_linear;
shun-iwasawa 7bad49
    exr_header->channels[c].x_sampling = info.channels[c].x_sampling;
shun-iwasawa 7bad49
    exr_header->channels[c].y_sampling = info.channels[c].y_sampling;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  exr_header->pixel_types = static_cast<int *>(
shun-iwasawa 7bad49
      malloc(sizeof(int) * static_cast<size_t>(exr_header->num_channels)));
shun-iwasawa 7bad49
  for (size_t c = 0; c < static_cast<size_t>(exr_header->num_channels); c++) {
shun-iwasawa 7bad49
    exr_header->pixel_types[c] = info.channels[c].pixel_type;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Initially fill with values of `pixel_types`
shun-iwasawa 7bad49
  exr_header->requested_pixel_types = static_cast<int *>(
shun-iwasawa 7bad49
      malloc(sizeof(int) * static_cast<size_t>(exr_header->num_channels)));
shun-iwasawa 7bad49
  for (size_t c = 0; c < static_cast<size_t>(exr_header->num_channels); c++) {
shun-iwasawa 7bad49
    exr_header->requested_pixel_types[c] = info.channels[c].pixel_type;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  exr_header->num_custom_attributes = static_cast<int>(info.attributes.size());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->num_custom_attributes > 0) {
shun-iwasawa 7bad49
    // TODO(syoyo): Report warning when # of attributes exceeds
shun-iwasawa 7bad49
    // `TINYEXR_MAX_CUSTOM_ATTRIBUTES`
shun-iwasawa 7bad49
    if (exr_header->num_custom_attributes > TINYEXR_MAX_CUSTOM_ATTRIBUTES) {
shun-iwasawa 7bad49
      exr_header->num_custom_attributes = TINYEXR_MAX_CUSTOM_ATTRIBUTES;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    exr_header->custom_attributes = static_cast<EXRAttribute *>(malloc(
shun-iwasawa 7bad49
        sizeof(EXRAttribute) * size_t(exr_header->num_custom_attributes)));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (size_t i = 0; i < info.attributes.size(); i++) {
shun-iwasawa 7bad49
      memcpy(exr_header->custom_attributes[i].name, info.attributes[i].name,
shun-iwasawa 7bad49
             256);
shun-iwasawa 7bad49
      memcpy(exr_header->custom_attributes[i].type, info.attributes[i].type,
shun-iwasawa 7bad49
             256);
shun-iwasawa 7bad49
      exr_header->custom_attributes[i].size = info.attributes[i].size;
shun-iwasawa 7bad49
      // Just copy pointer
shun-iwasawa 7bad49
      exr_header->custom_attributes[i].value = info.attributes[i].value;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    exr_header->custom_attributes = NULL;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  exr_header->header_len = info.header_len;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
struct OffsetData {
shun-iwasawa 7bad49
  OffsetData() : num_x_levels(0), num_y_levels(0) {}
shun-iwasawa 7bad49
  std::vector<std::vector<std::vector <tinyexr::tinyexr_uint64> > > offsets;
shun-iwasawa 7bad49
  int	num_x_levels;
shun-iwasawa 7bad49
  int	num_y_levels;
shun-iwasawa 7bad49
};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int LevelIndex(int lx, int ly, int tile_level_mode, int num_x_levels) {
shun-iwasawa 7bad49
  switch (tile_level_mode) {
shun-iwasawa 7bad49
  case TINYEXR_TILE_ONE_LEVEL:
shun-iwasawa 7bad49
    return 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  case TINYEXR_TILE_MIPMAP_LEVELS:
shun-iwasawa 7bad49
    return lx;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  case TINYEXR_TILE_RIPMAP_LEVELS:
shun-iwasawa 7bad49
    return lx + ly * num_x_levels;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  default:
shun-iwasawa 7bad49
    assert(false);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return 0;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int LevelSize(int toplevel_size, int level, int tile_rounding_mode) {
shun-iwasawa 7bad49
  assert(level >= 0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int b = (int)(1u << (unsigned)level);
shun-iwasawa 7bad49
  int level_size = toplevel_size / b;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (tile_rounding_mode == TINYEXR_TILE_ROUND_UP && level_size * b < toplevel_size)
shun-iwasawa 7bad49
    level_size += 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return std::max(level_size, 1);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int DecodeTiledLevel(EXRImage* exr_image, const EXRHeader* exr_header,
shun-iwasawa 7bad49
  const OffsetData& offset_data,
shun-iwasawa 7bad49
  const std::vector<size_t>& channel_offset_list,
shun-iwasawa 7bad49
  int pixel_data_size,
shun-iwasawa 7bad49
  const unsigned char* head, const size_t size,
shun-iwasawa 7bad49
  std::string* err) {
shun-iwasawa 7bad49
  int num_channels = exr_header->num_channels;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int level_index = LevelIndex(exr_image->level_x, exr_image->level_y, exr_header->tile_level_mode, offset_data.num_x_levels);
shun-iwasawa 7bad49
  int num_y_tiles = (int)offset_data.offsets[level_index].size();
shun-iwasawa 7bad49
  assert(num_y_tiles);
shun-iwasawa 7bad49
  int num_x_tiles = (int)offset_data.offsets[level_index][0].size();
shun-iwasawa 7bad49
  assert(num_x_tiles);
shun-iwasawa 7bad49
  int num_tiles = num_x_tiles * num_y_tiles;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int err_code = TINYEXR_SUCCESS;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  enum {
shun-iwasawa 7bad49
    EF_SUCCESS = 0,
shun-iwasawa 7bad49
    EF_INVALID_DATA = 1,
shun-iwasawa 7bad49
    EF_INSUFFICIENT_DATA = 2,
shun-iwasawa 7bad49
    EF_FAILED_TO_DECODE = 4
shun-iwasawa 7bad49
  };
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
  std::atomic<unsigned> error_flag(EF_SUCCESS);
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  unsigned error_flag(EF_SUCCESS);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  
shun-iwasawa 7bad49
  // Although the spec says : "...the data window is subdivided into an array of smaller rectangles...",
shun-iwasawa 7bad49
  // the IlmImf library allows the dimensions of the tile to be larger (or equal) than the dimensions of the data window.
shun-iwasawa 7bad49
#if 0
shun-iwasawa 7bad49
  if ((exr_header->tile_size_x > exr_image->width || exr_header->tile_size_y > exr_image->height) &&
shun-iwasawa 7bad49
    exr_image->level_x == 0 && exr_image->level_y == 0) {
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "Failed to decode tile data.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    err_code = TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  exr_image->tiles = static_cast<EXRTile*>(
shun-iwasawa 7bad49
    calloc(sizeof(EXRTile), static_cast<size_t>(num_tiles)));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
  std::vector<std::thread> workers;
shun-iwasawa 7bad49
  std::atomic<int> tile_count(0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int num_threads = std::max(1, int(std::thread::hardware_concurrency()));
shun-iwasawa 7bad49
  if (num_threads > int(num_tiles)) {
shun-iwasawa 7bad49
    num_threads = int(num_tiles);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int t = 0; t < num_threads; t++) {
shun-iwasawa 7bad49
    workers.emplace_back(std::thread([&]()
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        int tile_idx = 0;
shun-iwasawa 7bad49
        while ((tile_idx = tile_count++) < num_tiles) {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
#if TINYEXR_USE_OPENMP
shun-iwasawa 7bad49
#pragma omp parallel for
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  for (int tile_idx = 0; tile_idx < num_tiles; tile_idx++) {
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    // Allocate memory for each tile.
shun-iwasawa 7bad49
    exr_image->tiles[tile_idx].images = tinyexr::AllocateImage(
shun-iwasawa 7bad49
      num_channels, exr_header->channels,
shun-iwasawa 7bad49
      exr_header->requested_pixel_types, exr_header->tile_size_x,
shun-iwasawa 7bad49
      exr_header->tile_size_y);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int x_tile = tile_idx % num_x_tiles;
shun-iwasawa 7bad49
    int y_tile = tile_idx / num_x_tiles;
shun-iwasawa 7bad49
    // 16 byte: tile coordinates
shun-iwasawa 7bad49
    // 4 byte : data size
shun-iwasawa 7bad49
    // ~      : data(uncompressed or compressed)
shun-iwasawa 7bad49
    tinyexr::tinyexr_uint64 offset = offset_data.offsets[level_index][y_tile][x_tile];
shun-iwasawa 7bad49
    if (offset + sizeof(int) * 5 > size) {
shun-iwasawa 7bad49
      // Insufficient data size.
shun-iwasawa 7bad49
      error_flag |= EF_INSUFFICIENT_DATA; 
shun-iwasawa 7bad49
      continue;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    size_t data_size =
shun-iwasawa 7bad49
      size_t(size - (offset + sizeof(int) * 5));
shun-iwasawa 7bad49
    const unsigned char* data_ptr =
shun-iwasawa 7bad49
      reinterpret_cast<const unsigned char*>(head + offset);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int tile_coordinates[4];
shun-iwasawa 7bad49
    memcpy(tile_coordinates, data_ptr, sizeof(int) * 4);
shun-iwasawa 7bad49
    tinyexr::swap4(&tile_coordinates[0]);
shun-iwasawa 7bad49
    tinyexr::swap4(&tile_coordinates[1]);
shun-iwasawa 7bad49
    tinyexr::swap4(&tile_coordinates[2]);
shun-iwasawa 7bad49
    tinyexr::swap4(&tile_coordinates[3]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (tile_coordinates[2] != exr_image->level_x) {
shun-iwasawa 7bad49
      // Invalid data.
shun-iwasawa 7bad49
      error_flag |= EF_INVALID_DATA;
shun-iwasawa 7bad49
      continue;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    if (tile_coordinates[3] != exr_image->level_y) {
shun-iwasawa 7bad49
      // Invalid data.
shun-iwasawa 7bad49
      error_flag |= EF_INVALID_DATA;
shun-iwasawa 7bad49
      continue;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int data_len;
shun-iwasawa 7bad49
    memcpy(&data_len, data_ptr + 16,
shun-iwasawa 7bad49
      sizeof(int));  // 16 = sizeof(tile_coordinates)
shun-iwasawa 7bad49
    tinyexr::swap4(&data_len);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (data_len < 2 || size_t(data_len) > data_size) {
shun-iwasawa 7bad49
      // Insufficient data size.
shun-iwasawa 7bad49
      error_flag |= EF_INSUFFICIENT_DATA;
shun-iwasawa 7bad49
      continue;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // Move to data addr: 20 = 16 + 4;
shun-iwasawa 7bad49
    data_ptr += 20;
shun-iwasawa 7bad49
    bool ret = tinyexr::DecodeTiledPixelData(
shun-iwasawa 7bad49
      exr_image->tiles[tile_idx].images,
shun-iwasawa 7bad49
      &(exr_image->tiles[tile_idx].width),
shun-iwasawa 7bad49
      &(exr_image->tiles[tile_idx].height),
shun-iwasawa 7bad49
      exr_header->requested_pixel_types, data_ptr,
shun-iwasawa 7bad49
      static_cast<size_t>(data_len), exr_header->compression_type,
shun-iwasawa 7bad49
      exr_header->line_order,
shun-iwasawa 7bad49
      exr_image->width, exr_image->height,
shun-iwasawa 7bad49
      tile_coordinates[0], tile_coordinates[1], exr_header->tile_size_x,
shun-iwasawa 7bad49
      exr_header->tile_size_y, static_cast<size_t>(pixel_data_size),
shun-iwasawa 7bad49
      static_cast<size_t>(exr_header->num_custom_attributes),
shun-iwasawa 7bad49
      exr_header->custom_attributes,
shun-iwasawa 7bad49
      static_cast<size_t>(exr_header->num_channels),
shun-iwasawa 7bad49
      exr_header->channels, channel_offset_list);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!ret) {
shun-iwasawa 7bad49
      // Failed to decode tile data.
shun-iwasawa 7bad49
      error_flag |= EF_FAILED_TO_DECODE;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    exr_image->tiles[tile_idx].offset_x = tile_coordinates[0];
shun-iwasawa 7bad49
    exr_image->tiles[tile_idx].offset_y = tile_coordinates[1];
shun-iwasawa 7bad49
    exr_image->tiles[tile_idx].level_x = tile_coordinates[2];
shun-iwasawa 7bad49
    exr_image->tiles[tile_idx].level_y = tile_coordinates[3];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
  }  
shun-iwasawa 7bad49
        }));
shun-iwasawa 7bad49
    }  // num_thread loop
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (auto& t : workers) {
shun-iwasawa 7bad49
      t.join();
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  } // parallel for
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Even in the event of an error, the reserved memory may be freed.
shun-iwasawa 7bad49
  exr_image->num_channels = num_channels;
shun-iwasawa 7bad49
  exr_image->num_tiles = static_cast<int>(num_tiles);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (error_flag)  err_code = TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  if (err) {
shun-iwasawa 7bad49
    if (error_flag & EF_INSUFFICIENT_DATA) {
shun-iwasawa 7bad49
      (*err) += "Insufficient data length.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    if (error_flag & EF_FAILED_TO_DECODE) {
shun-iwasawa 7bad49
      (*err) += "Failed to decode tile data.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return err_code;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int DecodeChunk(EXRImage *exr_image, const EXRHeader *exr_header,
shun-iwasawa 7bad49
                       const OffsetData& offset_data,
shun-iwasawa 7bad49
                       const unsigned char *head, const size_t size,
shun-iwasawa 7bad49
                       std::string *err) {
shun-iwasawa 7bad49
  int num_channels = exr_header->num_channels;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int num_scanline_blocks = 1;
shun-iwasawa 7bad49
  if (exr_header->compression_type == TINYEXR_COMPRESSIONTYPE_ZIP) {
shun-iwasawa 7bad49
    num_scanline_blocks = 16;
shun-iwasawa 7bad49
  } else if (exr_header->compression_type == TINYEXR_COMPRESSIONTYPE_PIZ) {
shun-iwasawa 7bad49
    num_scanline_blocks = 32;
shun-iwasawa 7bad49
  } else if (exr_header->compression_type == TINYEXR_COMPRESSIONTYPE_ZFP) {
shun-iwasawa 7bad49
    num_scanline_blocks = 16;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_ZFP
shun-iwasawa 7bad49
    tinyexr::ZFPCompressionParam zfp_compression_param;
shun-iwasawa 7bad49
    if (!FindZFPCompressionParam(&zfp_compression_param,
shun-iwasawa 7bad49
                                 exr_header->custom_attributes,
shun-iwasawa 7bad49
                                 int(exr_header->num_custom_attributes), err)) {
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_HEADER;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->data_window.max_x < exr_header->data_window.min_x ||
shun-iwasawa 7bad49
      exr_header->data_window.max_y < exr_header->data_window.min_y) {
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "Invalid data window.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int data_width =
shun-iwasawa 7bad49
      exr_header->data_window.max_x - exr_header->data_window.min_x + 1;
shun-iwasawa 7bad49
  int data_height =
shun-iwasawa 7bad49
      exr_header->data_window.max_y - exr_header->data_window.min_y + 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Do not allow too large data_width and data_height. header invalid?
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    if ((data_width > TINYEXR_DIMENSION_THRESHOLD) || (data_height > TINYEXR_DIMENSION_THRESHOLD)) {
shun-iwasawa 7bad49
      if (err) {
shun-iwasawa 7bad49
        std::stringstream ss;
shun-iwasawa 7bad49
        ss << "data_with or data_height too large. data_width: " << data_width
shun-iwasawa 7bad49
           << ", "
shun-iwasawa 7bad49
           << "data_height = " << data_height << std::endl;
shun-iwasawa 7bad49
        (*err) += ss.str();
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    if (exr_header->tiled) {
shun-iwasawa 7bad49
      if ((exr_header->tile_size_x > TINYEXR_DIMENSION_THRESHOLD) || (exr_header->tile_size_y > TINYEXR_DIMENSION_THRESHOLD)) {
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          std::stringstream ss;
shun-iwasawa 7bad49
          ss << "tile with or tile height too large. tile width: " << exr_header->tile_size_x
shun-iwasawa 7bad49
            << ", "
shun-iwasawa 7bad49
            << "tile height = " << exr_header->tile_size_y << std::endl;
shun-iwasawa 7bad49
          (*err) += ss.str();
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const std::vector<tinyexr::tinyexr_uint64>& offsets = offset_data.offsets[0][0];
shun-iwasawa 7bad49
  size_t num_blocks = offsets.size();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<size_t> channel_offset_list;
shun-iwasawa 7bad49
  int pixel_data_size = 0;
shun-iwasawa 7bad49
  size_t channel_offset = 0;
shun-iwasawa 7bad49
  if (!tinyexr::ComputeChannelLayout(&channel_offset_list, &pixel_data_size,
shun-iwasawa 7bad49
                                     &channel_offset, num_channels,
shun-iwasawa 7bad49
                                     exr_header->channels)) {
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "Failed to compute channel layout.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
  std::atomic<bool> invalid_data(false);
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  bool invalid_data(false);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->tiled) {
shun-iwasawa 7bad49
    // value check
shun-iwasawa 7bad49
    if (exr_header->tile_size_x < 0) {
shun-iwasawa 7bad49
      if (err) {
shun-iwasawa 7bad49
        std::stringstream ss;
shun-iwasawa 7bad49
        ss << "Invalid tile size x : " << exr_header->tile_size_x << "\n";
shun-iwasawa 7bad49
        (*err) += ss.str();
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_HEADER;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (exr_header->tile_size_y < 0) {
shun-iwasawa 7bad49
      if (err) {
shun-iwasawa 7bad49
        std::stringstream ss;
shun-iwasawa 7bad49
        ss << "Invalid tile size y : " << exr_header->tile_size_y << "\n";
shun-iwasawa 7bad49
        (*err) += ss.str();
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_HEADER;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    if (exr_header->tile_level_mode != TINYEXR_TILE_RIPMAP_LEVELS) {
shun-iwasawa 7bad49
      EXRImage* level_image = NULL;
shun-iwasawa 7bad49
      for (int level = 0; level < offset_data.num_x_levels; ++level) {
shun-iwasawa 7bad49
        if (!level_image) {
shun-iwasawa 7bad49
          level_image = exr_image;
shun-iwasawa 7bad49
        } else {
shun-iwasawa 7bad49
          level_image->next_level = new EXRImage;
shun-iwasawa 7bad49
          InitEXRImage(level_image->next_level);
shun-iwasawa 7bad49
          level_image = level_image->next_level;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        level_image->width =
shun-iwasawa 7bad49
          LevelSize(exr_header->data_window.max_x - exr_header->data_window.min_x + 1, level, exr_header->tile_rounding_mode);
shun-iwasawa 7bad49
        level_image->height =
shun-iwasawa 7bad49
          LevelSize(exr_header->data_window.max_y - exr_header->data_window.min_y + 1, level, exr_header->tile_rounding_mode);
shun-iwasawa 7bad49
        level_image->level_x = level;
shun-iwasawa 7bad49
        level_image->level_y = level;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        int ret = DecodeTiledLevel(level_image, exr_header,
shun-iwasawa 7bad49
          offset_data,
shun-iwasawa 7bad49
          channel_offset_list,
shun-iwasawa 7bad49
          pixel_data_size,
shun-iwasawa 7bad49
          head, size,
shun-iwasawa 7bad49
          err);
shun-iwasawa 7bad49
        if (ret != TINYEXR_SUCCESS) return ret;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      EXRImage* level_image = NULL;
shun-iwasawa 7bad49
      for (int level_y = 0; level_y < offset_data.num_y_levels; ++level_y)
shun-iwasawa 7bad49
        for (int level_x = 0; level_x < offset_data.num_x_levels; ++level_x) {
shun-iwasawa 7bad49
          if (!level_image) {
shun-iwasawa 7bad49
            level_image = exr_image;
shun-iwasawa 7bad49
          } else {
shun-iwasawa 7bad49
            level_image->next_level = new EXRImage;
shun-iwasawa 7bad49
            InitEXRImage(level_image->next_level);
shun-iwasawa 7bad49
            level_image = level_image->next_level;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          level_image->width =
shun-iwasawa 7bad49
            LevelSize(exr_header->data_window.max_x - exr_header->data_window.min_x + 1, level_x, exr_header->tile_rounding_mode);
shun-iwasawa 7bad49
          level_image->height =
shun-iwasawa 7bad49
            LevelSize(exr_header->data_window.max_y - exr_header->data_window.min_y + 1, level_y, exr_header->tile_rounding_mode);
shun-iwasawa 7bad49
          level_image->level_x = level_x;
shun-iwasawa 7bad49
          level_image->level_y = level_y;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          int ret = DecodeTiledLevel(level_image, exr_header,
shun-iwasawa 7bad49
            offset_data,
shun-iwasawa 7bad49
            channel_offset_list,
shun-iwasawa 7bad49
            pixel_data_size,
shun-iwasawa 7bad49
            head, size,
shun-iwasawa 7bad49
            err);
shun-iwasawa 7bad49
          if (ret != TINYEXR_SUCCESS) return ret;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  } else {  // scanline format
shun-iwasawa 7bad49
    // Don't allow too large image(256GB * pixel_data_size or more). Workaround
shun-iwasawa 7bad49
    // for #104.
shun-iwasawa 7bad49
    size_t total_data_len =
shun-iwasawa 7bad49
        size_t(data_width) * size_t(data_height) * size_t(num_channels);
shun-iwasawa 7bad49
    const bool total_data_len_overflown =
shun-iwasawa 7bad49
        sizeof(void *) == 8 ? (total_data_len >= 0x4000000000) : false;
shun-iwasawa 7bad49
    if ((total_data_len == 0) || total_data_len_overflown) {
shun-iwasawa 7bad49
      if (err) {
shun-iwasawa 7bad49
        std::stringstream ss;
shun-iwasawa 7bad49
        ss << "Image data size is zero or too large: width = " << data_width
shun-iwasawa 7bad49
           << ", height = " << data_height << ", channels = " << num_channels
shun-iwasawa 7bad49
           << std::endl;
shun-iwasawa 7bad49
        (*err) += ss.str();
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    exr_image->images = tinyexr::AllocateImage(
shun-iwasawa 7bad49
        num_channels, exr_header->channels, exr_header->requested_pixel_types,
shun-iwasawa 7bad49
        data_width, data_height);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
    std::vector<std::thread> workers;
shun-iwasawa 7bad49
    std::atomic<int> y_count(0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int num_threads = std::max(1, int(std::thread::hardware_concurrency()));
shun-iwasawa 7bad49
    if (num_threads > int(num_blocks)) {
shun-iwasawa 7bad49
      num_threads = int(num_blocks);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (int t = 0; t < num_threads; t++) {
shun-iwasawa 7bad49
      workers.emplace_back(std::thread([&]() {
shun-iwasawa 7bad49
        int y = 0;
shun-iwasawa 7bad49
        while ((y = y_count++) < int(num_blocks)) {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_OPENMP
shun-iwasawa 7bad49
#pragma omp parallel for
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    for (int y = 0; y < static_cast<int>(num_blocks); y++) {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
          size_t y_idx = static_cast<size_t>(y);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          if (offsets[y_idx] + sizeof(int) * 2 > size) {
shun-iwasawa 7bad49
            invalid_data = true;
shun-iwasawa 7bad49
          } else {
shun-iwasawa 7bad49
            // 4 byte: scan line
shun-iwasawa 7bad49
            // 4 byte: data size
shun-iwasawa 7bad49
            // ~     : pixel data(uncompressed or compressed)
shun-iwasawa 7bad49
            size_t data_size =
shun-iwasawa 7bad49
                size_t(size - (offsets[y_idx] + sizeof(int) * 2));
shun-iwasawa 7bad49
            const unsigned char *data_ptr =
shun-iwasawa 7bad49
                reinterpret_cast<const unsigned char *>(head + offsets[y_idx]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            int line_no;
shun-iwasawa 7bad49
            memcpy(&line_no, data_ptr, sizeof(int));
shun-iwasawa 7bad49
            int data_len;
shun-iwasawa 7bad49
            memcpy(&data_len, data_ptr + 4, sizeof(int));
shun-iwasawa 7bad49
            tinyexr::swap4(&line_no);
shun-iwasawa 7bad49
            tinyexr::swap4(&data_len);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            if (size_t(data_len) > data_size) {
shun-iwasawa 7bad49
              invalid_data = true;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            } else if ((line_no > (2 << 20)) || (line_no < -(2 << 20))) {
shun-iwasawa 7bad49
              // Too large value. Assume this is invalid
shun-iwasawa 7bad49
              // 2**20 = 1048576 = heuristic value.
shun-iwasawa 7bad49
              invalid_data = true;
shun-iwasawa 7bad49
            } else if (data_len == 0) {
shun-iwasawa 7bad49
              // TODO(syoyo): May be ok to raise the threshold for example
shun-iwasawa 7bad49
              // `data_len < 4`
shun-iwasawa 7bad49
              invalid_data = true;
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              // line_no may be negative.
shun-iwasawa 7bad49
              int end_line_no = (std::min)(line_no + num_scanline_blocks,
shun-iwasawa 7bad49
                                           (exr_header->data_window.max_y + 1));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              int num_lines = end_line_no - line_no;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
              if (num_lines <= 0) {
shun-iwasawa 7bad49
                invalid_data = true;
shun-iwasawa 7bad49
              } else {
shun-iwasawa 7bad49
                // Move to data addr: 8 = 4 + 4;
shun-iwasawa 7bad49
                data_ptr += 8;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
                // Adjust line_no with data_window.bmin.y
shun-iwasawa 7bad49
shun-iwasawa 7bad49
                // overflow check
shun-iwasawa 7bad49
                tinyexr_int64 lno =
shun-iwasawa 7bad49
                    static_cast<tinyexr_int64>(line_no) -
shun-iwasawa 7bad49
                    static_cast<tinyexr_int64>(exr_header->data_window.min_y);
shun-iwasawa 7bad49
                if (lno > std::numeric_limits<int>::max()) {
shun-iwasawa 7bad49
                  line_no = -1;  // invalid
shun-iwasawa 7bad49
                } else if (lno < -std::numeric_limits<int>::max()) {
shun-iwasawa 7bad49
                  line_no = -1;  // invalid
shun-iwasawa 7bad49
                } else {
shun-iwasawa 7bad49
                  line_no -= exr_header->data_window.min_y;
shun-iwasawa 7bad49
                }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
                if (line_no < 0) {
shun-iwasawa 7bad49
                  invalid_data = true;
shun-iwasawa 7bad49
                } else {
shun-iwasawa 7bad49
                  if (!tinyexr::DecodePixelData(
shun-iwasawa 7bad49
                          exr_image->images, exr_header->requested_pixel_types,
shun-iwasawa 7bad49
                          data_ptr, static_cast<size_t>(data_len),
shun-iwasawa 7bad49
                          exr_header->compression_type, exr_header->line_order,
shun-iwasawa 7bad49
                          data_width, data_height, data_width, y, line_no,
shun-iwasawa 7bad49
                          num_lines, static_cast<size_t>(pixel_data_size),
shun-iwasawa 7bad49
                          static_cast<size_t>(
shun-iwasawa 7bad49
                              exr_header->num_custom_attributes),
shun-iwasawa 7bad49
                          exr_header->custom_attributes,
shun-iwasawa 7bad49
                          static_cast<size_t>(exr_header->num_channels),
shun-iwasawa 7bad49
                          exr_header->channels, channel_offset_list)) {
shun-iwasawa 7bad49
                    invalid_data = true;
shun-iwasawa 7bad49
                  }
shun-iwasawa 7bad49
                }
shun-iwasawa 7bad49
              }
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }));
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (auto &t : workers) {
shun-iwasawa 7bad49
      t.join();
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    }  // omp parallel
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (invalid_data) {
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "Invalid data found when decoding pixels.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Overwrite `pixel_type` with `requested_pixel_type`.
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    for (int c = 0; c < exr_header->num_channels; c++) {
shun-iwasawa 7bad49
      exr_header->pixel_types[c] = exr_header->requested_pixel_types[c];
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    exr_image->num_channels = num_channels;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    exr_image->width = data_width;
shun-iwasawa 7bad49
    exr_image->height = data_height;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool ReconstructLineOffsets(
shun-iwasawa 7bad49
    std::vector<tinyexr::tinyexr_uint64> *offsets, size_t n,
shun-iwasawa 7bad49
    const unsigned char *head, const unsigned char *marker, const size_t size) {
shun-iwasawa 7bad49
  assert(head < marker);
shun-iwasawa 7bad49
  assert(offsets->size() == n);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t i = 0; i < n; i++) {
shun-iwasawa 7bad49
    size_t offset = static_cast<size_t>(marker - head);
shun-iwasawa 7bad49
    // Offset should not exceed whole EXR file/data size.
shun-iwasawa 7bad49
    if ((offset + sizeof(tinyexr::tinyexr_uint64)) >= size) {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int y;
shun-iwasawa 7bad49
    unsigned int data_len;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    memcpy(&y, marker, sizeof(int));
shun-iwasawa 7bad49
    memcpy(&data_len, marker + 4, sizeof(unsigned int));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (data_len >= size) {
shun-iwasawa 7bad49
      return false;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tinyexr::swap4(&y);
shun-iwasawa 7bad49
    tinyexr::swap4(&data_len);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    (*offsets)[i] = offset;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    marker += data_len + 8;  // 8 = 4 bytes(y) + 4 bytes(data_len)
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int FloorLog2(unsigned x) {
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // For x > 0, floorLog2(y) returns floor(log(x)/log(2)).
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  int y = 0;
shun-iwasawa 7bad49
  while (x > 1) {
shun-iwasawa 7bad49
    y += 1;
shun-iwasawa 7bad49
    x >>= 1u;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return y;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int CeilLog2(unsigned x) {
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // For x > 0, ceilLog2(y) returns ceil(log(x)/log(2)).
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  int y = 0;
shun-iwasawa 7bad49
  int r = 0;
shun-iwasawa 7bad49
  while (x > 1) {
shun-iwasawa 7bad49
    if (x & 1)
shun-iwasawa 7bad49
      r = 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    y += 1;
shun-iwasawa 7bad49
    x >>= 1u;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return y + r;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int RoundLog2(int x, int tile_rounding_mode) {
shun-iwasawa 7bad49
  return (tile_rounding_mode == TINYEXR_TILE_ROUND_DOWN) ? FloorLog2(static_cast<unsigned>(x)) : CeilLog2(static_cast<unsigned>(x));
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int CalculateNumXLevels(const EXRHeader* exr_header) {
shun-iwasawa 7bad49
  int min_x = exr_header->data_window.min_x;
shun-iwasawa 7bad49
  int max_x = exr_header->data_window.max_x;
shun-iwasawa 7bad49
  int min_y = exr_header->data_window.min_y;
shun-iwasawa 7bad49
  int max_y = exr_header->data_window.max_y;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int num = 0;
shun-iwasawa 7bad49
  switch (exr_header->tile_level_mode) {
shun-iwasawa 7bad49
  case TINYEXR_TILE_ONE_LEVEL:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    num = 1;
shun-iwasawa 7bad49
    break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  case TINYEXR_TILE_MIPMAP_LEVELS:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    int w = max_x - min_x + 1;
shun-iwasawa 7bad49
    int h = max_y - min_y + 1;
shun-iwasawa 7bad49
    num = RoundLog2(std::max(w, h), exr_header->tile_rounding_mode) + 1;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  case TINYEXR_TILE_RIPMAP_LEVELS:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    int w = max_x - min_x + 1;
shun-iwasawa 7bad49
    num = RoundLog2(w, exr_header->tile_rounding_mode) + 1;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  default:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    assert(false);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return num;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int CalculateNumYLevels(const EXRHeader* exr_header) {
shun-iwasawa 7bad49
  int min_x = exr_header->data_window.min_x;
shun-iwasawa 7bad49
  int max_x = exr_header->data_window.max_x;
shun-iwasawa 7bad49
  int min_y = exr_header->data_window.min_y;
shun-iwasawa 7bad49
  int max_y = exr_header->data_window.max_y;
shun-iwasawa 7bad49
  int num = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  switch (exr_header->tile_level_mode) {
shun-iwasawa 7bad49
  case TINYEXR_TILE_ONE_LEVEL:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    num = 1;
shun-iwasawa 7bad49
    break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  case TINYEXR_TILE_MIPMAP_LEVELS:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    int w = max_x - min_x + 1;
shun-iwasawa 7bad49
    int h = max_y - min_y + 1;
shun-iwasawa 7bad49
    num = RoundLog2(std::max(w, h), exr_header->tile_rounding_mode) + 1;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  case TINYEXR_TILE_RIPMAP_LEVELS:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    int h = max_y - min_y + 1;
shun-iwasawa 7bad49
    num = RoundLog2(h, exr_header->tile_rounding_mode) + 1;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  default:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    assert(false);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return num;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void CalculateNumTiles(std::vector<int>& numTiles,
shun-iwasawa 7bad49
  int toplevel_size,
shun-iwasawa 7bad49
  int size,
shun-iwasawa 7bad49
  int tile_rounding_mode) {
shun-iwasawa 7bad49
  for (unsigned i = 0; i < numTiles.size(); i++) {
shun-iwasawa 7bad49
    int l = LevelSize(toplevel_size, i, tile_rounding_mode);
shun-iwasawa 7bad49
    assert(l <= std::numeric_limits<int>::max() - size + 1);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    numTiles[i] = (l + size - 1) / size;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void PrecalculateTileInfo(std::vector<int>& num_x_tiles,
shun-iwasawa 7bad49
  std::vector<int>& num_y_tiles,
shun-iwasawa 7bad49
  const EXRHeader* exr_header) {
shun-iwasawa 7bad49
  int min_x = exr_header->data_window.min_x;
shun-iwasawa 7bad49
  int max_x = exr_header->data_window.max_x;
shun-iwasawa 7bad49
  int min_y = exr_header->data_window.min_y;
shun-iwasawa 7bad49
  int max_y = exr_header->data_window.max_y;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int num_x_levels = CalculateNumXLevels(exr_header);
shun-iwasawa 7bad49
  int num_y_levels = CalculateNumYLevels(exr_header);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  num_x_tiles.resize(num_x_levels);
shun-iwasawa 7bad49
  num_y_tiles.resize(num_y_levels);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  CalculateNumTiles(num_x_tiles,
shun-iwasawa 7bad49
    max_x - min_x + 1,
shun-iwasawa 7bad49
    exr_header->tile_size_x,
shun-iwasawa 7bad49
    exr_header->tile_rounding_mode);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  CalculateNumTiles(num_y_tiles,
shun-iwasawa 7bad49
    max_y - min_y + 1,
shun-iwasawa 7bad49
    exr_header->tile_size_y,
shun-iwasawa 7bad49
    exr_header->tile_rounding_mode);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void InitSingleResolutionOffsets(OffsetData& offset_data, size_t num_blocks) {
shun-iwasawa 7bad49
  offset_data.offsets.resize(1);
shun-iwasawa 7bad49
  offset_data.offsets[0].resize(1);
shun-iwasawa 7bad49
  offset_data.offsets[0][0].resize(num_blocks);
shun-iwasawa 7bad49
  offset_data.num_x_levels = 1;
shun-iwasawa 7bad49
  offset_data.num_y_levels = 1;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// Return sum of tile blocks.
shun-iwasawa 7bad49
static int InitTileOffsets(OffsetData& offset_data,
shun-iwasawa 7bad49
  const EXRHeader* exr_header,
shun-iwasawa 7bad49
  const std::vector<int>& num_x_tiles,
shun-iwasawa 7bad49
  const std::vector<int>& num_y_tiles) {
shun-iwasawa 7bad49
  int num_tile_blocks = 0;
shun-iwasawa 7bad49
  offset_data.num_x_levels = static_cast<int>(num_x_tiles.size());
shun-iwasawa 7bad49
  offset_data.num_y_levels = static_cast<int>(num_y_tiles.size());
shun-iwasawa 7bad49
  switch (exr_header->tile_level_mode) {
shun-iwasawa 7bad49
  case TINYEXR_TILE_ONE_LEVEL:
shun-iwasawa 7bad49
  case TINYEXR_TILE_MIPMAP_LEVELS:
shun-iwasawa 7bad49
    assert(offset_data.num_x_levels == offset_data.num_y_levels);
shun-iwasawa 7bad49
    offset_data.offsets.resize(offset_data.num_x_levels);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (unsigned int l = 0; l < offset_data.offsets.size(); ++l) {
shun-iwasawa 7bad49
      offset_data.offsets[l].resize(num_y_tiles[l]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      for (unsigned int dy = 0; dy < offset_data.offsets[l].size(); ++dy) {
shun-iwasawa 7bad49
        offset_data.offsets[l][dy].resize(num_x_tiles[l]);
shun-iwasawa 7bad49
        num_tile_blocks += num_x_tiles[l];
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  case TINYEXR_TILE_RIPMAP_LEVELS:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    offset_data.offsets.resize(static_cast<size_t>(offset_data.num_x_levels) * static_cast<size_t>(offset_data.num_y_levels));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (int ly = 0; ly < offset_data.num_y_levels; ++ly) {
shun-iwasawa 7bad49
      for (int lx = 0; lx < offset_data.num_x_levels; ++lx) {
shun-iwasawa 7bad49
        int l = ly * offset_data.num_x_levels + lx;
shun-iwasawa 7bad49
        offset_data.offsets[l].resize(num_y_tiles[ly]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        for (size_t dy = 0; dy < offset_data.offsets[l].size(); ++dy) {
shun-iwasawa 7bad49
          offset_data.offsets[l][dy].resize(num_x_tiles[lx]);
shun-iwasawa 7bad49
          num_tile_blocks += num_x_tiles[lx];
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  default:
shun-iwasawa 7bad49
    assert(false);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return num_tile_blocks;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool IsAnyOffsetsAreInvalid(const OffsetData& offset_data) {
shun-iwasawa 7bad49
  for (unsigned int l = 0; l < offset_data.offsets.size(); ++l)
shun-iwasawa 7bad49
    for (unsigned int dy = 0; dy < offset_data.offsets[l].size(); ++dy)
shun-iwasawa 7bad49
      for (unsigned int dx = 0; dx < offset_data.offsets[l][dy].size(); ++dx)
shun-iwasawa 7bad49
        if (reinterpret_cast<const tinyexr::tinyexr_int64&>(offset_data.offsets[l][dy][dx]) <= 0)
shun-iwasawa 7bad49
          return true;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return false;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static bool isValidTile(const EXRHeader* exr_header,
shun-iwasawa 7bad49
                        const OffsetData& offset_data,
shun-iwasawa 7bad49
                        int dx, int dy, int lx, int ly) {
shun-iwasawa 7bad49
  if (lx < 0 || ly < 0 || dx < 0 || dy < 0) return false;
shun-iwasawa 7bad49
  int num_x_levels = offset_data.num_x_levels;
shun-iwasawa 7bad49
  int num_y_levels = offset_data.num_y_levels;
shun-iwasawa 7bad49
  switch (exr_header->tile_level_mode) {
shun-iwasawa 7bad49
  case TINYEXR_TILE_ONE_LEVEL:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (lx == 0 &&
shun-iwasawa 7bad49
        ly == 0 &&
shun-iwasawa 7bad49
        offset_data.offsets.size() > 0 &&
shun-iwasawa 7bad49
        offset_data.offsets[0].size() > static_cast<size_t>(dy) &&
shun-iwasawa 7bad49
        offset_data.offsets[0][dy].size() > static_cast<size_t>(dx)) {
shun-iwasawa 7bad49
      return true;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  case TINYEXR_TILE_MIPMAP_LEVELS:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (lx < num_x_levels &&
shun-iwasawa 7bad49
        ly < num_y_levels &&
shun-iwasawa 7bad49
        offset_data.offsets.size() > static_cast<size_t>(lx) &&
shun-iwasawa 7bad49
        offset_data.offsets[lx].size() > static_cast<size_t>(dy) &&
shun-iwasawa 7bad49
        offset_data.offsets[lx][dy].size() > static_cast<size_t>(dx)) {
shun-iwasawa 7bad49
      return true;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  case TINYEXR_TILE_RIPMAP_LEVELS:
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    size_t idx = static_cast<size_t>(lx) + static_cast<size_t>(ly)* static_cast<size_t>(num_x_levels);
shun-iwasawa 7bad49
    if (lx < num_x_levels &&
shun-iwasawa 7bad49
       ly < num_y_levels &&
shun-iwasawa 7bad49
       (offset_data.offsets.size() > idx) &&
shun-iwasawa 7bad49
       offset_data.offsets[idx].size() > static_cast<size_t>(dy) &&
shun-iwasawa 7bad49
       offset_data.offsets[idx][dy].size() > static_cast<size_t>(dx)) {
shun-iwasawa 7bad49
      return true;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    break;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  default:
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return false;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void ReconstructTileOffsets(OffsetData& offset_data,
shun-iwasawa 7bad49
                                   const EXRHeader* exr_header,
shun-iwasawa 7bad49
                                   const unsigned char* head, const unsigned char* marker, const size_t /*size*/,
shun-iwasawa 7bad49
                                   bool isMultiPartFile,
shun-iwasawa 7bad49
                                   bool isDeep) {
shun-iwasawa 7bad49
  int numXLevels = offset_data.num_x_levels;
shun-iwasawa 7bad49
  for (unsigned int l = 0; l < offset_data.offsets.size(); ++l) {
shun-iwasawa 7bad49
    for (unsigned int dy = 0; dy < offset_data.offsets[l].size(); ++dy) {
shun-iwasawa 7bad49
      for (unsigned int dx = 0; dx < offset_data.offsets[l][dy].size(); ++dx) {
shun-iwasawa 7bad49
        tinyexr::tinyexr_uint64 tileOffset = marker - head;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (isMultiPartFile) {
shun-iwasawa 7bad49
          //int partNumber;
shun-iwasawa 7bad49
          marker += sizeof(int);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        int tileX;
shun-iwasawa 7bad49
        memcpy(&tileX, marker, sizeof(int));
shun-iwasawa 7bad49
        tinyexr::swap4(&tileX);
shun-iwasawa 7bad49
        marker += sizeof(int);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        int tileY;
shun-iwasawa 7bad49
        memcpy(&tileY, marker, sizeof(int));
shun-iwasawa 7bad49
        tinyexr::swap4(&tileY);
shun-iwasawa 7bad49
        marker += sizeof(int);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        int levelX;
shun-iwasawa 7bad49
        memcpy(&levelX, marker, sizeof(int));
shun-iwasawa 7bad49
        tinyexr::swap4(&levelX);
shun-iwasawa 7bad49
        marker += sizeof(int);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        int levelY;
shun-iwasawa 7bad49
        memcpy(&levelY, marker, sizeof(int));
shun-iwasawa 7bad49
        tinyexr::swap4(&levelY);
shun-iwasawa 7bad49
        marker += sizeof(int);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (isDeep) {
shun-iwasawa 7bad49
          tinyexr::tinyexr_int64 packed_offset_table_size;
shun-iwasawa 7bad49
          memcpy(&packed_offset_table_size, marker, sizeof(tinyexr::tinyexr_int64));
shun-iwasawa 7bad49
          tinyexr::swap8(reinterpret_cast<tinyexr::tinyexr_uint64*>(&packed_offset_table_size));
shun-iwasawa 7bad49
          marker += sizeof(tinyexr::tinyexr_int64);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          tinyexr::tinyexr_int64 packed_sample_size;
shun-iwasawa 7bad49
          memcpy(&packed_sample_size, marker, sizeof(tinyexr::tinyexr_int64));
shun-iwasawa 7bad49
          tinyexr::swap8(reinterpret_cast<tinyexr::tinyexr_uint64*>(&packed_sample_size));
shun-iwasawa 7bad49
          marker += sizeof(tinyexr::tinyexr_int64);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          // next Int64 is unpacked sample size - skip that too
shun-iwasawa 7bad49
          marker += packed_offset_table_size + packed_sample_size + 8;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        } else {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          int dataSize;
shun-iwasawa 7bad49
          memcpy(&dataSize, marker, sizeof(int));
shun-iwasawa 7bad49
          tinyexr::swap4(&dataSize);
shun-iwasawa 7bad49
          marker += sizeof(int);
shun-iwasawa 7bad49
          marker += dataSize;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (!isValidTile(exr_header, offset_data,
shun-iwasawa 7bad49
          tileX, tileY, levelX, levelY))
shun-iwasawa 7bad49
          return;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        int level_idx = LevelIndex(levelX, levelY, exr_header->tile_level_mode, numXLevels);
shun-iwasawa 7bad49
        offset_data.offsets[level_idx][tileY][tileX] = tileOffset;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// marker output is also
shun-iwasawa 7bad49
static int ReadOffsets(OffsetData& offset_data,
shun-iwasawa 7bad49
                       const unsigned char* head,
shun-iwasawa 7bad49
                       const unsigned char*& marker,
shun-iwasawa 7bad49
                       const size_t size,
shun-iwasawa 7bad49
                       const char** err) {
shun-iwasawa 7bad49
  for (unsigned int l = 0; l < offset_data.offsets.size(); ++l) {
shun-iwasawa 7bad49
    for (unsigned int dy = 0; dy < offset_data.offsets[l].size(); ++dy) {
shun-iwasawa 7bad49
      for (unsigned int dx = 0; dx < offset_data.offsets[l][dy].size(); ++dx) {
shun-iwasawa 7bad49
        tinyexr::tinyexr_uint64 offset;
shun-iwasawa 7bad49
        if ((marker + sizeof(tinyexr_uint64)) >= (head + size)) {
shun-iwasawa 7bad49
          tinyexr::SetErrorMessage("Insufficient data size in offset table.", err);
shun-iwasawa 7bad49
          return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        memcpy(&offset, marker, sizeof(tinyexr::tinyexr_uint64));
shun-iwasawa 7bad49
        tinyexr::swap8(&offset);
shun-iwasawa 7bad49
        if (offset >= size) {
shun-iwasawa 7bad49
          tinyexr::SetErrorMessage("Invalid offset value in DecodeEXRImage.", err);
shun-iwasawa 7bad49
          return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        marker += sizeof(tinyexr::tinyexr_uint64);  // = 8
shun-iwasawa 7bad49
        offset_data.offsets[l][dy][dx] = offset;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int DecodeEXRImage(EXRImage *exr_image, const EXRHeader *exr_header,
shun-iwasawa 7bad49
                          const unsigned char *head,
shun-iwasawa 7bad49
                          const unsigned char *marker, const size_t size,
shun-iwasawa 7bad49
                          const char **err) {
shun-iwasawa 7bad49
  if (exr_image == NULL || exr_header == NULL || head == NULL ||
shun-iwasawa 7bad49
      marker == NULL || (size <= tinyexr::kEXRVersionSize)) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for DecodeEXRImage().", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int num_scanline_blocks = 1;
shun-iwasawa 7bad49
  if (exr_header->compression_type == TINYEXR_COMPRESSIONTYPE_ZIP) {
shun-iwasawa 7bad49
    num_scanline_blocks = 16;
shun-iwasawa 7bad49
  } else if (exr_header->compression_type == TINYEXR_COMPRESSIONTYPE_PIZ) {
shun-iwasawa 7bad49
    num_scanline_blocks = 32;
shun-iwasawa 7bad49
  } else if (exr_header->compression_type == TINYEXR_COMPRESSIONTYPE_ZFP) {
shun-iwasawa 7bad49
    num_scanline_blocks = 16;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->data_window.max_x < exr_header->data_window.min_x ||
shun-iwasawa 7bad49
      exr_header->data_window.max_x - exr_header->data_window.min_x ==
shun-iwasawa 7bad49
          std::numeric_limits<int>::max()) {
shun-iwasawa 7bad49
    // Issue 63
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid data width value", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  int data_width =
shun-iwasawa 7bad49
      exr_header->data_window.max_x - exr_header->data_window.min_x + 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->data_window.max_y < exr_header->data_window.min_y ||
shun-iwasawa 7bad49
      exr_header->data_window.max_y - exr_header->data_window.min_y ==
shun-iwasawa 7bad49
          std::numeric_limits<int>::max()) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid data height value", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  int data_height =
shun-iwasawa 7bad49
      exr_header->data_window.max_y - exr_header->data_window.min_y + 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Do not allow too large data_width and data_height. header invalid?
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    if (data_width > TINYEXR_DIMENSION_THRESHOLD) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("data width too large.", err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    if (data_height > TINYEXR_DIMENSION_THRESHOLD) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("data height too large.", err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->tiled) {
shun-iwasawa 7bad49
    if (exr_header->tile_size_x > TINYEXR_DIMENSION_THRESHOLD) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("tile width too large.", err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    if (exr_header->tile_size_y > TINYEXR_DIMENSION_THRESHOLD) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("tile height too large.", err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Read offset tables.
shun-iwasawa 7bad49
  OffsetData offset_data;
shun-iwasawa 7bad49
  size_t num_blocks = 0;
shun-iwasawa 7bad49
  // For a multi-resolution image, the size of the offset table will be calculated from the other attributes of the header.
shun-iwasawa 7bad49
  // If chunk_count > 0 then chunk_count must be equal to the calculated tile count.
shun-iwasawa 7bad49
  if (exr_header->tiled) {
shun-iwasawa 7bad49
    {
shun-iwasawa 7bad49
      std::vector<int> num_x_tiles, num_y_tiles;
shun-iwasawa 7bad49
      PrecalculateTileInfo(num_x_tiles, num_y_tiles, exr_header);
shun-iwasawa 7bad49
      num_blocks = InitTileOffsets(offset_data, exr_header, num_x_tiles, num_y_tiles);
shun-iwasawa 7bad49
      if (exr_header->chunk_count > 0) {
shun-iwasawa 7bad49
        if (exr_header->chunk_count != static_cast<int>(num_blocks)) {
shun-iwasawa 7bad49
          tinyexr::SetErrorMessage("Invalid offset table size.", err);
shun-iwasawa 7bad49
          return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int ret = ReadOffsets(offset_data, head, marker, size, err);
shun-iwasawa 7bad49
    if (ret != TINYEXR_SUCCESS) return ret;
shun-iwasawa 7bad49
    if (IsAnyOffsetsAreInvalid(offset_data)) {
shun-iwasawa 7bad49
      ReconstructTileOffsets(offset_data, exr_header,
shun-iwasawa 7bad49
        head, marker, size,
shun-iwasawa 7bad49
        exr_header->multipart, exr_header->non_image);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  } else if (exr_header->chunk_count > 0) {
shun-iwasawa 7bad49
    // Use `chunkCount` attribute.
shun-iwasawa 7bad49
    num_blocks = static_cast<size_t>(exr_header->chunk_count);
shun-iwasawa 7bad49
    InitSingleResolutionOffsets(offset_data, num_blocks);
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    num_blocks = static_cast<size_t>(data_height) /
shun-iwasawa 7bad49
      static_cast<size_t>(num_scanline_blocks);
shun-iwasawa 7bad49
    if (num_blocks * static_cast<size_t>(num_scanline_blocks) <
shun-iwasawa 7bad49
      static_cast<size_t>(data_height)) {
shun-iwasawa 7bad49
      num_blocks++;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    InitSingleResolutionOffsets(offset_data, num_blocks);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (!exr_header->tiled) {
shun-iwasawa 7bad49
    std::vector<tinyexr::tinyexr_uint64>& offsets = offset_data.offsets[0][0];
shun-iwasawa 7bad49
    for (size_t y = 0; y < num_blocks; y++) {
shun-iwasawa 7bad49
      tinyexr::tinyexr_uint64 offset;
shun-iwasawa 7bad49
      // Issue #81
shun-iwasawa 7bad49
      if ((marker + sizeof(tinyexr_uint64)) >= (head + size)) {
shun-iwasawa 7bad49
        tinyexr::SetErrorMessage("Insufficient data size in offset table.", err);
shun-iwasawa 7bad49
        return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      memcpy(&offset, marker, sizeof(tinyexr::tinyexr_uint64));
shun-iwasawa 7bad49
      tinyexr::swap8(&offset);
shun-iwasawa 7bad49
      if (offset >= size) {
shun-iwasawa 7bad49
        tinyexr::SetErrorMessage("Invalid offset value in DecodeEXRImage.", err);
shun-iwasawa 7bad49
        return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      marker += sizeof(tinyexr::tinyexr_uint64);  // = 8
shun-iwasawa 7bad49
      offsets[y] = offset;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // If line offsets are invalid, we try to reconstruct it.
shun-iwasawa 7bad49
    // See OpenEXR/IlmImf/ImfScanLineInputFile.cpp::readLineOffsets() for details.
shun-iwasawa 7bad49
    for (size_t y = 0; y < num_blocks; y++) {
shun-iwasawa 7bad49
      if (offsets[y] <= 0) {
shun-iwasawa 7bad49
        // TODO(syoyo) Report as warning?
shun-iwasawa 7bad49
        // if (err) {
shun-iwasawa 7bad49
        //  stringstream ss;
shun-iwasawa 7bad49
        //  ss << "Incomplete lineOffsets." << std::endl;
shun-iwasawa 7bad49
        //  (*err) += ss.str();
shun-iwasawa 7bad49
        //}
shun-iwasawa 7bad49
        bool ret =
shun-iwasawa 7bad49
          ReconstructLineOffsets(&offsets, num_blocks, head, marker, size);
shun-iwasawa 7bad49
        if (ret) {
shun-iwasawa 7bad49
          // OK
shun-iwasawa 7bad49
          break;
shun-iwasawa 7bad49
        } else {
shun-iwasawa 7bad49
          tinyexr::SetErrorMessage(
shun-iwasawa 7bad49
            "Cannot reconstruct lineOffset table in DecodeEXRImage.", err);
shun-iwasawa 7bad49
          return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    std::string e;
shun-iwasawa 7bad49
    int ret = DecodeChunk(exr_image, exr_header, offset_data, head, size, &e);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
      if (!e.empty()) {
shun-iwasawa 7bad49
        tinyexr::SetErrorMessage(e, err);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if 1
shun-iwasawa 7bad49
      FreeEXRImage(exr_image);
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
      // release memory(if exists)
shun-iwasawa 7bad49
      if ((exr_header->num_channels > 0) && exr_image && exr_image->images) {
shun-iwasawa 7bad49
        for (size_t c = 0; c < size_t(exr_header->num_channels); c++) {
shun-iwasawa 7bad49
          if (exr_image->images[c]) {
shun-iwasawa 7bad49
            free(exr_image->images[c]);
shun-iwasawa 7bad49
            exr_image->images[c] = NULL;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        free(exr_image->images);
shun-iwasawa 7bad49
        exr_image->images = NULL;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    return ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void GetLayers(const EXRHeader &exr_header,
shun-iwasawa 7bad49
                      std::vector<std::string> &layer_names) {
shun-iwasawa 7bad49
  // Naive implementation
shun-iwasawa 7bad49
  // Group channels by layers
shun-iwasawa 7bad49
  // go over all channel names, split by periods
shun-iwasawa 7bad49
  // collect unique names
shun-iwasawa 7bad49
  layer_names.clear();
shun-iwasawa 7bad49
  for (int c = 0; c < exr_header.num_channels; c++) {
shun-iwasawa 7bad49
    std::string full_name(exr_header.channels[c].name);
shun-iwasawa 7bad49
    const size_t pos = full_name.find_last_of('.');
shun-iwasawa 7bad49
    if (pos != std::string::npos && pos != 0 && pos + 1 < full_name.size()) {
shun-iwasawa 7bad49
      full_name.erase(pos);
shun-iwasawa 7bad49
      if (std::find(layer_names.begin(), layer_names.end(), full_name) ==
shun-iwasawa 7bad49
          layer_names.end())
shun-iwasawa 7bad49
        layer_names.push_back(full_name);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
struct LayerChannel {
shun-iwasawa 7bad49
  explicit LayerChannel(size_t i, std::string n) : index(i), name(n) {}
shun-iwasawa 7bad49
  size_t index;
shun-iwasawa 7bad49
  std::string name;
shun-iwasawa 7bad49
};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static void ChannelsInLayer(const EXRHeader &exr_header,
shun-iwasawa 7bad49
                            const std::string &layer_name,
shun-iwasawa 7bad49
                            std::vector<LayerChannel> &channels) {
shun-iwasawa 7bad49
  channels.clear();
shun-iwasawa 7bad49
  for (int c = 0; c < exr_header.num_channels; c++) {
shun-iwasawa 7bad49
    std::string ch_name(exr_header.channels[c].name);
shun-iwasawa 7bad49
    if (layer_name.empty()) {
shun-iwasawa 7bad49
      const size_t pos = ch_name.find_last_of('.');
shun-iwasawa 7bad49
      if (pos != std::string::npos && pos < ch_name.size()) {
shun-iwasawa 7bad49
        ch_name = ch_name.substr(pos + 1);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      const size_t pos = ch_name.find(layer_name + '.');
shun-iwasawa 7bad49
      if (pos == std::string::npos) continue;
shun-iwasawa 7bad49
      if (pos == 0) {
shun-iwasawa 7bad49
        ch_name = ch_name.substr(layer_name.size() + 1);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    LayerChannel ch(size_t(c), ch_name);
shun-iwasawa 7bad49
    channels.push_back(ch);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
}  // namespace tinyexr
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int EXRLayers(const char *filename, const char **layer_names[], int *num_layers,
shun-iwasawa 7bad49
              const char **err) {
shun-iwasawa 7bad49
  EXRVersion exr_version;
shun-iwasawa 7bad49
  EXRHeader exr_header;
shun-iwasawa 7bad49
  InitEXRHeader(&exr_header);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    int ret = ParseEXRVersionFromFile(&exr_version, filename);
shun-iwasawa 7bad49
    if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("Invalid EXR header.", err);
shun-iwasawa 7bad49
      return ret;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (exr_version.multipart || exr_version.non_image) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage(
shun-iwasawa 7bad49
          "Loading multipart or DeepImage is not supported  in LoadEXR() API",
shun-iwasawa 7bad49
          err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;  // @fixme.
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int ret = ParseEXRHeaderFromFile(&exr_header, &exr_version, filename, err);
shun-iwasawa 7bad49
  if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
    FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
    return ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<std::string> layer_vec;
shun-iwasawa 7bad49
  tinyexr::GetLayers(exr_header, layer_vec);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  (*num_layers) = int(layer_vec.size());
shun-iwasawa 7bad49
  (*layer_names) = static_cast<const char **>(
shun-iwasawa 7bad49
      malloc(sizeof(const char *) * static_cast<size_t>(layer_vec.size())));
shun-iwasawa 7bad49
  for (size_t c = 0; c < static_cast<size_t>(layer_vec.size()); c++) {
shun-iwasawa 7bad49
#ifdef _MSC_VER
shun-iwasawa 7bad49
    (*layer_names)[c] = _strdup(layer_vec[c].c_str());
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    (*layer_names)[c] = strdup(layer_vec[c].c_str());
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int LoadEXR(float **out_rgba, int *width, int *height, const char *filename,
shun-iwasawa 7bad49
            const char **err) {
shun-iwasawa 7bad49
  return LoadEXRWithLayer(out_rgba, width, height, filename,
shun-iwasawa 7bad49
                          /* layername */ NULL, err);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int LoadEXRWithLayer(float **out_rgba, int *width, int *height,
shun-iwasawa 7bad49
                     const char *filename, const char *layername,
shun-iwasawa 7bad49
                     const char **err) {
shun-iwasawa 7bad49
  if (out_rgba == NULL) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for LoadEXR()", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  EXRVersion exr_version;
shun-iwasawa 7bad49
  EXRImage exr_image;
shun-iwasawa 7bad49
  EXRHeader exr_header;
shun-iwasawa 7bad49
  InitEXRHeader(&exr_header);
shun-iwasawa 7bad49
  InitEXRImage(&exr_image);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    int ret = ParseEXRVersionFromFile(&exr_version, filename);
shun-iwasawa 7bad49
    if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
      std::stringstream ss;
shun-iwasawa 7bad49
      ss << "Failed to open EXR file or read version info from EXR file. code("
shun-iwasawa 7bad49
         << ret << ")";
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage(ss.str(), err);
shun-iwasawa 7bad49
      return ret;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (exr_version.multipart || exr_version.non_image) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage(
shun-iwasawa 7bad49
          "Loading multipart or DeepImage is not supported  in LoadEXR() API",
shun-iwasawa 7bad49
          err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;  // @fixme.
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    int ret = ParseEXRHeaderFromFile(&exr_header, &exr_version, filename, err);
shun-iwasawa 7bad49
    if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
      FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
      return ret;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Read HALF channel as FLOAT.
shun-iwasawa 7bad49
  for (int i = 0; i < exr_header.num_channels; i++) {
shun-iwasawa 7bad49
    if (exr_header.pixel_types[i] == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
      exr_header.requested_pixel_types[i] = TINYEXR_PIXELTYPE_FLOAT;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // TODO: Probably limit loading to layers (channels) selected by layer index
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    int ret = LoadEXRImageFromFile(&exr_image, &exr_header, filename, err);
shun-iwasawa 7bad49
    if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
      FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
      return ret;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // RGBA
shun-iwasawa 7bad49
  int idxR = -1;
shun-iwasawa 7bad49
  int idxG = -1;
shun-iwasawa 7bad49
  int idxB = -1;
shun-iwasawa 7bad49
  int idxA = -1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<std::string> layer_names;
shun-iwasawa 7bad49
  tinyexr::GetLayers(exr_header, layer_names);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<tinyexr::LayerChannel> channels;
shun-iwasawa 7bad49
  tinyexr::ChannelsInLayer(
shun-iwasawa 7bad49
      exr_header, layername == NULL ? "" : std::string(layername), channels);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (channels.size() < 1) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Layer Not Found", err);
shun-iwasawa 7bad49
    FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
    FreeEXRImage(&exr_image);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_LAYER_NOT_FOUND;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t ch_count = channels.size() < 4 ? channels.size() : 4;
shun-iwasawa 7bad49
  for (size_t c = 0; c < ch_count; c++) {
shun-iwasawa 7bad49
    const tinyexr::LayerChannel &ch = channels[c];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (ch.name == "R") {
shun-iwasawa 7bad49
      idxR = int(ch.index);
shun-iwasawa 7bad49
    } else if (ch.name == "G") {
shun-iwasawa 7bad49
      idxG = int(ch.index);
shun-iwasawa 7bad49
    } else if (ch.name == "B") {
shun-iwasawa 7bad49
      idxB = int(ch.index);
shun-iwasawa 7bad49
    } else if (ch.name == "A") {
shun-iwasawa 7bad49
      idxA = int(ch.index);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (channels.size() == 1) {
shun-iwasawa 7bad49
    int chIdx = int(channels.front().index);
shun-iwasawa 7bad49
    // Grayscale channel only.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    (*out_rgba) = reinterpret_cast<float *>(
shun-iwasawa 7bad49
        malloc(4 * sizeof(float) * static_cast<size_t>(exr_image.width) *
shun-iwasawa 7bad49
               static_cast<size_t>(exr_image.height)));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (exr_header.tiled) {
shun-iwasawa 7bad49
      for (int it = 0; it < exr_image.num_tiles; it++) {
shun-iwasawa 7bad49
        for (int j = 0; j < exr_header.tile_size_y; j++) {
shun-iwasawa 7bad49
          for (int i = 0; i < exr_header.tile_size_x; i++) {
shun-iwasawa 7bad49
            const int ii = exr_image.tiles[it].offset_x *
shun-iwasawa 7bad49
                               static_cast<int>(exr_header.tile_size_x) +
shun-iwasawa 7bad49
                           i;
shun-iwasawa 7bad49
            const int jj = exr_image.tiles[it].offset_y *
shun-iwasawa 7bad49
                               static_cast<int>(exr_header.tile_size_y) +
shun-iwasawa 7bad49
                           j;
shun-iwasawa 7bad49
            const int idx = ii + jj * static_cast<int>(exr_image.width);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // out of region check.
shun-iwasawa 7bad49
            if (ii >= exr_image.width) {
shun-iwasawa 7bad49
              continue;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            if (jj >= exr_image.height) {
shun-iwasawa 7bad49
              continue;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            const int srcIdx = i + j * exr_header.tile_size_x;
shun-iwasawa 7bad49
            unsigned char **src = exr_image.tiles[it].images;
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 0] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[chIdx][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 1] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[chIdx][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 2] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[chIdx][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 3] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[chIdx][srcIdx];
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      for (int i = 0; i < exr_image.width * exr_image.height; i++) {
shun-iwasawa 7bad49
        const float val =
shun-iwasawa 7bad49
            reinterpret_cast<float **>(exr_image.images)[chIdx][i];
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 0] = val;
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 1] = val;
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 2] = val;
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 3] = val;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    // Assume RGB(A)
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (idxR == -1) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("R channel not found", err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
      FreeEXRImage(&exr_image);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (idxG == -1) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("G channel not found", err);
shun-iwasawa 7bad49
      FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
      FreeEXRImage(&exr_image);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (idxB == -1) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("B channel not found", err);
shun-iwasawa 7bad49
      FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
      FreeEXRImage(&exr_image);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    (*out_rgba) = reinterpret_cast<float *>(
shun-iwasawa 7bad49
        malloc(4 * sizeof(float) * static_cast<size_t>(exr_image.width) *
shun-iwasawa 7bad49
               static_cast<size_t>(exr_image.height)));
shun-iwasawa 7bad49
    if (exr_header.tiled) {
shun-iwasawa 7bad49
      for (int it = 0; it < exr_image.num_tiles; it++) {
shun-iwasawa 7bad49
        for (int j = 0; j < exr_header.tile_size_y; j++) {
shun-iwasawa 7bad49
          for (int i = 0; i < exr_header.tile_size_x; i++) {
shun-iwasawa 7bad49
            const int ii =
shun-iwasawa 7bad49
                exr_image.tiles[it].offset_x * exr_header.tile_size_x + i;
shun-iwasawa 7bad49
            const int jj =
shun-iwasawa 7bad49
                exr_image.tiles[it].offset_y * exr_header.tile_size_y + j;
shun-iwasawa 7bad49
            const int idx = ii + jj * exr_image.width;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // out of region check.
shun-iwasawa 7bad49
            if (ii >= exr_image.width) {
shun-iwasawa 7bad49
              continue;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            if (jj >= exr_image.height) {
shun-iwasawa 7bad49
              continue;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            const int srcIdx = i + j * exr_header.tile_size_x;
shun-iwasawa 7bad49
            unsigned char **src = exr_image.tiles[it].images;
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 0] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[idxR][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 1] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[idxG][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 2] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[idxB][srcIdx];
shun-iwasawa 7bad49
            if (idxA != -1) {
shun-iwasawa 7bad49
              (*out_rgba)[4 * idx + 3] =
shun-iwasawa 7bad49
                  reinterpret_cast<float **>(src)[idxA][srcIdx];
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              (*out_rgba)[4 * idx + 3] = 1.0;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      for (int i = 0; i < exr_image.width * exr_image.height; i++) {
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 0] =
shun-iwasawa 7bad49
            reinterpret_cast<float **>(exr_image.images)[idxR][i];
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 1] =
shun-iwasawa 7bad49
            reinterpret_cast<float **>(exr_image.images)[idxG][i];
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 2] =
shun-iwasawa 7bad49
            reinterpret_cast<float **>(exr_image.images)[idxB][i];
shun-iwasawa 7bad49
        if (idxA != -1) {
shun-iwasawa 7bad49
          (*out_rgba)[4 * i + 3] =
shun-iwasawa 7bad49
              reinterpret_cast<float **>(exr_image.images)[idxA][i];
shun-iwasawa 7bad49
        } else {
shun-iwasawa 7bad49
          (*out_rgba)[4 * i + 3] = 1.0;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  (*width) = exr_image.width;
shun-iwasawa 7bad49
  (*height) = exr_image.height;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
  FreeEXRImage(&exr_image);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int IsEXR(const char *filename) {
shun-iwasawa 7bad49
  EXRVersion exr_version;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int ret = ParseEXRVersionFromFile(&exr_version, filename);
shun-iwasawa 7bad49
  if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
    return ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int ParseEXRHeaderFromMemory(EXRHeader *exr_header, const EXRVersion *version,
shun-iwasawa 7bad49
                             const unsigned char *memory, size_t size,
shun-iwasawa 7bad49
                             const char **err) {
shun-iwasawa 7bad49
  if (memory == NULL || exr_header == NULL) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage(
shun-iwasawa 7bad49
        "Invalid argument. `memory` or `exr_header` argument is null in "
shun-iwasawa 7bad49
        "ParseEXRHeaderFromMemory()",
shun-iwasawa 7bad49
        err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // Invalid argument
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (size < tinyexr::kEXRVersionSize) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Insufficient header/data size.\n", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const unsigned char *marker = memory + tinyexr::kEXRVersionSize;
shun-iwasawa 7bad49
  size_t marker_size = size - tinyexr::kEXRVersionSize;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  tinyexr::HeaderInfo info;
shun-iwasawa 7bad49
  info.clear();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::string err_str;
shun-iwasawa 7bad49
  int ret = ParseEXRHeader(&info, NULL, version, &err_str, marker, marker_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
    if (err && !err_str.empty()) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage(err_str, err);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  ConvertHeader(exr_header, info);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  exr_header->multipart = version->multipart ? 1 : 0;
shun-iwasawa 7bad49
  exr_header->non_image = version->non_image ? 1 : 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return ret;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int LoadEXRFromMemory(float **out_rgba, int *width, int *height,
shun-iwasawa 7bad49
                      const unsigned char *memory, size_t size,
shun-iwasawa 7bad49
                      const char **err) {
shun-iwasawa 7bad49
  if (out_rgba == NULL || memory == NULL) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for LoadEXRFromMemory", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  EXRVersion exr_version;
shun-iwasawa 7bad49
  EXRImage exr_image;
shun-iwasawa 7bad49
  EXRHeader exr_header;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  InitEXRHeader(&exr_header);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int ret = ParseEXRVersionFromMemory(&exr_version, memory, size);
shun-iwasawa 7bad49
  if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
    std::stringstream ss;
shun-iwasawa 7bad49
    ss << "Failed to parse EXR version. code(" << ret << ")";
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage(ss.str(), err);
shun-iwasawa 7bad49
    return ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  ret = ParseEXRHeaderFromMemory(&exr_header, &exr_version, memory, size, err);
shun-iwasawa 7bad49
  if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
    return ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Read HALF channel as FLOAT.
shun-iwasawa 7bad49
  for (int i = 0; i < exr_header.num_channels; i++) {
shun-iwasawa 7bad49
    if (exr_header.pixel_types[i] == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
      exr_header.requested_pixel_types[i] = TINYEXR_PIXELTYPE_FLOAT;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  InitEXRImage(&exr_image);
shun-iwasawa 7bad49
  ret = LoadEXRImageFromMemory(&exr_image, &exr_header, memory, size, err);
shun-iwasawa 7bad49
  if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
    return ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // RGBA
shun-iwasawa 7bad49
  int idxR = -1;
shun-iwasawa 7bad49
  int idxG = -1;
shun-iwasawa 7bad49
  int idxB = -1;
shun-iwasawa 7bad49
  int idxA = -1;
shun-iwasawa 7bad49
  for (int c = 0; c < exr_header.num_channels; c++) {
shun-iwasawa 7bad49
    if (strcmp(exr_header.channels[c].name, "R") == 0) {
shun-iwasawa 7bad49
      idxR = c;
shun-iwasawa 7bad49
    } else if (strcmp(exr_header.channels[c].name, "G") == 0) {
shun-iwasawa 7bad49
      idxG = c;
shun-iwasawa 7bad49
    } else if (strcmp(exr_header.channels[c].name, "B") == 0) {
shun-iwasawa 7bad49
      idxB = c;
shun-iwasawa 7bad49
    } else if (strcmp(exr_header.channels[c].name, "A") == 0) {
shun-iwasawa 7bad49
      idxA = c;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // TODO(syoyo): Refactor removing same code as used in LoadEXR().
shun-iwasawa 7bad49
  if (exr_header.num_channels == 1) {
shun-iwasawa 7bad49
    // Grayscale channel only.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    (*out_rgba) = reinterpret_cast<float *>(
shun-iwasawa 7bad49
        malloc(4 * sizeof(float) * static_cast<size_t>(exr_image.width) *
shun-iwasawa 7bad49
               static_cast<size_t>(exr_image.height)));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (exr_header.tiled) {
shun-iwasawa 7bad49
      for (int it = 0; it < exr_image.num_tiles; it++) {
shun-iwasawa 7bad49
        for (int j = 0; j < exr_header.tile_size_y; j++) {
shun-iwasawa 7bad49
          for (int i = 0; i < exr_header.tile_size_x; i++) {
shun-iwasawa 7bad49
            const int ii =
shun-iwasawa 7bad49
                exr_image.tiles[it].offset_x * exr_header.tile_size_x + i;
shun-iwasawa 7bad49
            const int jj =
shun-iwasawa 7bad49
                exr_image.tiles[it].offset_y * exr_header.tile_size_y + j;
shun-iwasawa 7bad49
            const int idx = ii + jj * exr_image.width;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // out of region check.
shun-iwasawa 7bad49
            if (ii >= exr_image.width) {
shun-iwasawa 7bad49
              continue;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            if (jj >= exr_image.height) {
shun-iwasawa 7bad49
              continue;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            const int srcIdx = i + j * exr_header.tile_size_x;
shun-iwasawa 7bad49
            unsigned char **src = exr_image.tiles[it].images;
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 0] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[0][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 1] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[0][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 2] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[0][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 3] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[0][srcIdx];
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      for (int i = 0; i < exr_image.width * exr_image.height; i++) {
shun-iwasawa 7bad49
        const float val = reinterpret_cast<float **>(exr_image.images)[0][i];
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 0] = val;
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 1] = val;
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 2] = val;
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 3] = val;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    // TODO(syoyo): Support non RGBA image.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (idxR == -1) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("R channel not found", err);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      // @todo { free exr_image }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (idxG == -1) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("G channel not found", err);
shun-iwasawa 7bad49
      // @todo { free exr_image }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (idxB == -1) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("B channel not found", err);
shun-iwasawa 7bad49
      // @todo { free exr_image }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    (*out_rgba) = reinterpret_cast<float *>(
shun-iwasawa 7bad49
        malloc(4 * sizeof(float) * static_cast<size_t>(exr_image.width) *
shun-iwasawa 7bad49
               static_cast<size_t>(exr_image.height)));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (exr_header.tiled) {
shun-iwasawa 7bad49
      for (int it = 0; it < exr_image.num_tiles; it++) {
shun-iwasawa 7bad49
        for (int j = 0; j < exr_header.tile_size_y; j++)
shun-iwasawa 7bad49
          for (int i = 0; i < exr_header.tile_size_x; i++) {
shun-iwasawa 7bad49
            const int ii =
shun-iwasawa 7bad49
                exr_image.tiles[it].offset_x * exr_header.tile_size_x + i;
shun-iwasawa 7bad49
            const int jj =
shun-iwasawa 7bad49
                exr_image.tiles[it].offset_y * exr_header.tile_size_y + j;
shun-iwasawa 7bad49
            const int idx = ii + jj * exr_image.width;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // out of region check.
shun-iwasawa 7bad49
            if (ii >= exr_image.width) {
shun-iwasawa 7bad49
              continue;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            if (jj >= exr_image.height) {
shun-iwasawa 7bad49
              continue;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            const int srcIdx = i + j * exr_header.tile_size_x;
shun-iwasawa 7bad49
            unsigned char **src = exr_image.tiles[it].images;
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 0] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[idxR][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 1] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[idxG][srcIdx];
shun-iwasawa 7bad49
            (*out_rgba)[4 * idx + 2] =
shun-iwasawa 7bad49
                reinterpret_cast<float **>(src)[idxB][srcIdx];
shun-iwasawa 7bad49
            if (idxA != -1) {
shun-iwasawa 7bad49
              (*out_rgba)[4 * idx + 3] =
shun-iwasawa 7bad49
                  reinterpret_cast<float **>(src)[idxA][srcIdx];
shun-iwasawa 7bad49
            } else {
shun-iwasawa 7bad49
              (*out_rgba)[4 * idx + 3] = 1.0;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      for (int i = 0; i < exr_image.width * exr_image.height; i++) {
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 0] =
shun-iwasawa 7bad49
            reinterpret_cast<float **>(exr_image.images)[idxR][i];
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 1] =
shun-iwasawa 7bad49
            reinterpret_cast<float **>(exr_image.images)[idxG][i];
shun-iwasawa 7bad49
        (*out_rgba)[4 * i + 2] =
shun-iwasawa 7bad49
            reinterpret_cast<float **>(exr_image.images)[idxB][i];
shun-iwasawa 7bad49
        if (idxA != -1) {
shun-iwasawa 7bad49
          (*out_rgba)[4 * i + 3] =
shun-iwasawa 7bad49
              reinterpret_cast<float **>(exr_image.images)[idxA][i];
shun-iwasawa 7bad49
        } else {
shun-iwasawa 7bad49
          (*out_rgba)[4 * i + 3] = 1.0;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  (*width) = exr_image.width;
shun-iwasawa 7bad49
  (*height) = exr_image.height;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FreeEXRHeader(&exr_header);
shun-iwasawa 7bad49
  FreeEXRImage(&exr_image);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int LoadEXRImageFromFile(EXRImage *exr_image, const EXRHeader *exr_header,
shun-iwasawa 7bad49
                         const char *filename, const char **err) {
shun-iwasawa 7bad49
  if (exr_image == NULL) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for LoadEXRImageFromFile", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FILE *fp = NULL;
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
#if defined(_MSC_VER) || (defined(MINGW_HAS_SECURE_API) && MINGW_HAS_SECURE_API) // MSVC, MinGW GCC, or Clang.
shun-iwasawa 7bad49
  errno_t errcode =
shun-iwasawa 7bad49
      _wfopen_s(&fp, tinyexr::UTF8ToWchar(filename).c_str(), L"rb");
shun-iwasawa 7bad49
  if (errcode != 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read file " + std::string(filename), err);
shun-iwasawa 7bad49
    // TODO(syoyo): return wfopen_s erro code
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  // Unknown compiler or MinGW without MINGW_HAS_SECURE_API.
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  if (!fp) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read file " + std::string(filename), err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t filesize;
shun-iwasawa 7bad49
  // Compute size
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_END);
shun-iwasawa 7bad49
  filesize = static_cast<size_t>(ftell(fp));
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_SET);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (filesize < 16) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("File size too short " + std::string(filename),
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned char> buf(filesize);  // @todo { use mmap }
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    size_t ret;
shun-iwasawa 7bad49
    ret = fread(&buf[0], 1, filesize, fp);
shun-iwasawa 7bad49
    assert(ret == filesize);
shun-iwasawa 7bad49
    fclose(fp);
shun-iwasawa 7bad49
    (void)ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return LoadEXRImageFromMemory(exr_image, exr_header, &buf.at(0), filesize,
shun-iwasawa 7bad49
                                err);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int LoadEXRImageFromMemory(EXRImage *exr_image, const EXRHeader *exr_header,
shun-iwasawa 7bad49
                           const unsigned char *memory, const size_t size,
shun-iwasawa 7bad49
                           const char **err) {
shun-iwasawa 7bad49
  if (exr_image == NULL || memory == NULL ||
shun-iwasawa 7bad49
      (size < tinyexr::kEXRVersionSize)) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for LoadEXRImageFromMemory",
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->header_len == 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("EXRHeader variable is not initialized.", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const unsigned char *head = memory;
shun-iwasawa 7bad49
  const unsigned char *marker = reinterpret_cast<const unsigned char *>(
shun-iwasawa 7bad49
      memory + exr_header->header_len +
shun-iwasawa 7bad49
      8);  // +8 for magic number + version header.
shun-iwasawa 7bad49
  return tinyexr::DecodeEXRImage(exr_image, exr_header, head, marker, size,
shun-iwasawa 7bad49
                                 err);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
namespace tinyexr
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// out_data must be allocated initially with the block-header size
shun-iwasawa 7bad49
// of the current image(-part) type
shun-iwasawa 7bad49
static bool EncodePixelData(/* out */ std::vector<unsigned char>& out_data,                         
shun-iwasawa 7bad49
                            const unsigned char* const* images,
shun-iwasawa 7bad49
                            int compression_type,
shun-iwasawa 7bad49
                            int /*line_order*/,
shun-iwasawa 7bad49
                            int width, // for tiled : tile.width
shun-iwasawa 7bad49
                            int /*height*/, // for tiled : header.tile_size_y
shun-iwasawa 7bad49
                            int x_stride, // for tiled : header.tile_size_x
shun-iwasawa 7bad49
                            int line_no, // for tiled : 0
shun-iwasawa 7bad49
                            int num_lines, // for tiled : tile.height
shun-iwasawa 7bad49
                            size_t pixel_data_size,
shun-iwasawa 7bad49
                            const std::vector<ChannelInfo>& channels,
shun-iwasawa 7bad49
                            const std::vector<size_t>& channel_offset_list,
shun-iwasawa 7bad49
                            const void* compression_param = 0) // zfp compression param
shun-iwasawa 7bad49
{
shun-iwasawa 7bad49
  size_t buf_size = static_cast<size_t>(width) *
shun-iwasawa 7bad49
                  static_cast<size_t>(num_lines) *
shun-iwasawa 7bad49
                  static_cast<size_t>(pixel_data_size);
shun-iwasawa 7bad49
  //int last2bit = (buf_size & 3);
shun-iwasawa 7bad49
  // buf_size must be multiple of four
shun-iwasawa 7bad49
  //if(last2bit) buf_size += 4 - last2bit;
shun-iwasawa 7bad49
  std::vector<unsigned char> buf(buf_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t start_y = static_cast<size_t>(line_no);
shun-iwasawa 7bad49
  for (size_t c = 0; c < channels.size(); c++) {
shun-iwasawa 7bad49
    if (channels[c].pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
      if (channels[c].requested_pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
        for (int y = 0; y < num_lines; y++) {
shun-iwasawa 7bad49
          // Assume increasing Y
shun-iwasawa 7bad49
          float *line_ptr = reinterpret_cast<float *>(&buf.at(
shun-iwasawa 7bad49
            static_cast<size_t>(pixel_data_size * y * width) +
shun-iwasawa 7bad49
            channel_offset_list[c] *
shun-iwasawa 7bad49
            static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (int x = 0; x < width; x++) {
shun-iwasawa 7bad49
            tinyexr::FP16 h16;
shun-iwasawa 7bad49
            h16.u = reinterpret_cast<const unsigned short * const *>(
shun-iwasawa 7bad49
              images)[c][(y + start_y) * x_stride + x];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::FP32 f32 = half_to_float(h16);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(&f32.f);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // line_ptr[x] = f32.f;
shun-iwasawa 7bad49
            tinyexr::cpy4(line_ptr + x, &(f32.f));
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else if (channels[c].requested_pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
        for (int y = 0; y < num_lines; y++) {
shun-iwasawa 7bad49
          // Assume increasing Y
shun-iwasawa 7bad49
          unsigned short *line_ptr = reinterpret_cast<unsigned short *>(
shun-iwasawa 7bad49
            &buf.at(static_cast<size_t>(pixel_data_size * y *
shun-iwasawa 7bad49
                                        width) +
shun-iwasawa 7bad49
                    channel_offset_list[c] *
shun-iwasawa 7bad49
                    static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (int x = 0; x < width; x++) {
shun-iwasawa 7bad49
            unsigned short val = reinterpret_cast<const unsigned short * const *>(
shun-iwasawa 7bad49
              images)[c][(y + start_y) * x_stride + x];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap2(&val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // line_ptr[x] = val;
shun-iwasawa 7bad49
            tinyexr::cpy2(line_ptr + x, &val);
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        assert(0);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
      if (channels[c].requested_pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
        for (int y = 0; y < num_lines; y++) {
shun-iwasawa 7bad49
          // Assume increasing Y
shun-iwasawa 7bad49
          unsigned short *line_ptr = reinterpret_cast<unsigned short *>(
shun-iwasawa 7bad49
            &buf.at(static_cast<size_t>(pixel_data_size * y *
shun-iwasawa 7bad49
                                        width) +
shun-iwasawa 7bad49
                    channel_offset_list[c] *
shun-iwasawa 7bad49
                    static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (int x = 0; x < width; x++) {
shun-iwasawa 7bad49
            tinyexr::FP32 f32;
shun-iwasawa 7bad49
            f32.f = reinterpret_cast<const float * const *>(
shun-iwasawa 7bad49
              images)[c][(y + start_y) * x_stride + x];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::FP16 h16;
shun-iwasawa 7bad49
            h16 = float_to_half_full(f32);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap2(reinterpret_cast<unsigned short *>(&h16.u));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // line_ptr[x] = h16.u;
shun-iwasawa 7bad49
            tinyexr::cpy2(line_ptr + x, &(h16.u));
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else if (channels[c].requested_pixel_type == TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
        for (int y = 0; y < num_lines; y++) {
shun-iwasawa 7bad49
          // Assume increasing Y
shun-iwasawa 7bad49
          float *line_ptr = reinterpret_cast<float *>(&buf.at(
shun-iwasawa 7bad49
            static_cast<size_t>(pixel_data_size * y * width) +
shun-iwasawa 7bad49
            channel_offset_list[c] *
shun-iwasawa 7bad49
            static_cast<size_t>(width)));
shun-iwasawa 7bad49
          for (int x = 0; x < width; x++) {
shun-iwasawa 7bad49
            float val = reinterpret_cast<const float * const *>(
shun-iwasawa 7bad49
              images)[c][(y + start_y) * x_stride + x];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            tinyexr::swap4(&val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
            // line_ptr[x] = val;
shun-iwasawa 7bad49
            tinyexr::cpy4(line_ptr + x, &val);
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        assert(0);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else if (channels[c].pixel_type == TINYEXR_PIXELTYPE_UINT) {
shun-iwasawa 7bad49
      for (int y = 0; y < num_lines; y++) {
shun-iwasawa 7bad49
        // Assume increasing Y
shun-iwasawa 7bad49
        unsigned int *line_ptr = reinterpret_cast<unsigned int *>(&buf.at(
shun-iwasawa 7bad49
          static_cast<size_t>(pixel_data_size * y * width) +
shun-iwasawa 7bad49
          channel_offset_list[c] * static_cast<size_t>(width)));
shun-iwasawa 7bad49
        for (int x = 0; x < width; x++) {
shun-iwasawa 7bad49
          unsigned int val = reinterpret_cast<const unsigned int * const *>(
shun-iwasawa 7bad49
            images)[c][(y + start_y) * x_stride + x];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          tinyexr::swap4(&val);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          // line_ptr[x] = val;
shun-iwasawa 7bad49
          tinyexr::cpy4(line_ptr + x, &val);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (compression_type == TINYEXR_COMPRESSIONTYPE_NONE) {
shun-iwasawa 7bad49
    // 4 byte: scan line
shun-iwasawa 7bad49
    // 4 byte: data size
shun-iwasawa 7bad49
    // ~     : pixel data(uncompressed)
shun-iwasawa 7bad49
    out_data.insert(out_data.end(), buf.begin(), buf.end());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  } else if ((compression_type == TINYEXR_COMPRESSIONTYPE_ZIPS) ||
shun-iwasawa 7bad49
    (compression_type == TINYEXR_COMPRESSIONTYPE_ZIP)) {
shun-iwasawa 7bad49
#if TINYEXR_USE_MINIZ
shun-iwasawa 7bad49
    std::vector<unsigned char> block(mz_compressBound(
shun-iwasawa 7bad49
      static_cast<unsigned long>(buf.size())));
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    std::vector<unsigned char> block(
shun-iwasawa 7bad49
      compressBound(static_cast<uLong>(buf.size())));
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    tinyexr::tinyexr_uint64 outSize = block.size();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tinyexr::CompressZip(&block.at(0), outSize,
shun-iwasawa 7bad49
                         reinterpret_cast<const unsigned char *>(&buf.at(0)),
shun-iwasawa 7bad49
                         static_cast<unsigned long>(buf.size()));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // 4 byte: scan line
shun-iwasawa 7bad49
    // 4 byte: data size
shun-iwasawa 7bad49
    // ~     : pixel data(compressed)
shun-iwasawa 7bad49
    unsigned int data_len = static_cast<unsigned int>(outSize);  // truncate
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    out_data.insert(out_data.end(), block.begin(), block.begin() + data_len);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  } else if (compression_type == TINYEXR_COMPRESSIONTYPE_RLE) {
shun-iwasawa 7bad49
    // (buf.size() * 3) / 2 would be enough.
shun-iwasawa 7bad49
    std::vector<unsigned char> block((buf.size() * 3) / 2);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tinyexr::tinyexr_uint64 outSize = block.size();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tinyexr::CompressRle(&block.at(0), outSize,
shun-iwasawa 7bad49
                         reinterpret_cast<const unsigned char *>(&buf.at(0)),
shun-iwasawa 7bad49
                         static_cast<unsigned long>(buf.size()));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // 4 byte: scan line
shun-iwasawa 7bad49
    // 4 byte: data size
shun-iwasawa 7bad49
    // ~     : pixel data(compressed)
shun-iwasawa 7bad49
    unsigned int data_len = static_cast<unsigned int>(outSize);  // truncate
shun-iwasawa 7bad49
    out_data.insert(out_data.end(), block.begin(), block.begin() + data_len);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  } else if (compression_type == TINYEXR_COMPRESSIONTYPE_PIZ) {
shun-iwasawa 7bad49
#if TINYEXR_USE_PIZ
shun-iwasawa 7bad49
    unsigned int bufLen =
shun-iwasawa 7bad49
      8192 + static_cast<unsigned int>(
shun-iwasawa 7bad49
        2 * static_cast<unsigned int>(
shun-iwasawa 7bad49
          buf.size()));  // @fixme { compute good bound. }
shun-iwasawa 7bad49
    std::vector<unsigned char> block(bufLen);
shun-iwasawa 7bad49
    unsigned int outSize = static_cast<unsigned int>(block.size());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    CompressPiz(&block.at(0), &outSize,
shun-iwasawa 7bad49
                reinterpret_cast<const unsigned char *>(&buf.at(0)),
shun-iwasawa 7bad49
                buf.size(), channels, width, num_lines);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // 4 byte: scan line
shun-iwasawa 7bad49
    // 4 byte: data size
shun-iwasawa 7bad49
    // ~     : pixel data(compressed)
shun-iwasawa 7bad49
    unsigned int data_len = outSize;
shun-iwasawa 7bad49
    out_data.insert(out_data.end(), block.begin(), block.begin() + data_len);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    assert(0);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  } else if (compression_type == TINYEXR_COMPRESSIONTYPE_ZFP) {
shun-iwasawa 7bad49
#if TINYEXR_USE_ZFP
shun-iwasawa 7bad49
    const ZFPCompressionParam* zfp_compression_param = reinterpret_cast<const ZFPCompressionParam*>(compression_param);
shun-iwasawa 7bad49
    std::vector<unsigned char> block;
shun-iwasawa 7bad49
    unsigned int outSize;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tinyexr::CompressZfp(
shun-iwasawa 7bad49
      &block, &outSize, reinterpret_cast<const float *>(&buf.at(0)),
shun-iwasawa 7bad49
      width, num_lines, static_cast<int>(channels.size()), *zfp_compression_param);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // 4 byte: scan line
shun-iwasawa 7bad49
    // 4 byte: data size
shun-iwasawa 7bad49
    // ~     : pixel data(compressed)
shun-iwasawa 7bad49
    unsigned int data_len = outSize;
shun-iwasawa 7bad49
    out_data.insert(out_data.end(), block.begin(), block.begin() + data_len);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    (void)compression_param;
shun-iwasawa 7bad49
    assert(0);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    assert(0);
shun-iwasawa 7bad49
    return false;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return true;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int EncodeTiledLevel(const EXRImage* level_image, const EXRHeader* exr_header,
shun-iwasawa 7bad49
                            const std::vector<tinyexr::ChannelInfo>& channels,
shun-iwasawa 7bad49
                            std::vector<std::vector<unsigned char> >& data_list,
shun-iwasawa 7bad49
                            size_t start_index, // for data_list
shun-iwasawa 7bad49
                            int num_x_tiles, int num_y_tiles,
shun-iwasawa 7bad49
                            const std::vector<size_t>& channel_offset_list,
shun-iwasawa 7bad49
                            int pixel_data_size,
shun-iwasawa 7bad49
                            const void* compression_param, // must be set if zfp compression is enabled
shun-iwasawa 7bad49
                            std::string* err) {
shun-iwasawa 7bad49
  int num_tiles = num_x_tiles * num_y_tiles;
shun-iwasawa 7bad49
  assert(num_tiles == level_image->num_tiles);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if ((exr_header->tile_size_x > level_image->width || exr_header->tile_size_y > level_image->height) &&
shun-iwasawa 7bad49
      level_image->level_x == 0 && level_image->level_y == 0) {
shun-iwasawa 7bad49
      if (err) {
shun-iwasawa 7bad49
        (*err) += "Failed to encode tile data.\n";
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
  std::atomic<bool> invalid_data(false);
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  bool invalid_data(false);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
  std::vector<std::thread> workers;
shun-iwasawa 7bad49
  std::atomic<int> tile_count(0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int num_threads = std::max(1, int(std::thread::hardware_concurrency()));
shun-iwasawa 7bad49
  if (num_threads > int(num_tiles)) {
shun-iwasawa 7bad49
    num_threads = int(num_tiles);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int t = 0; t < num_threads; t++) {
shun-iwasawa 7bad49
    workers.emplace_back(std::thread([&]() {
shun-iwasawa 7bad49
      int i = 0;
shun-iwasawa 7bad49
      while ((i = tile_count++) < num_tiles) {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  // Use signed int since some OpenMP compiler doesn't allow unsigned type for
shun-iwasawa 7bad49
  // `parallel for`
shun-iwasawa 7bad49
#if TINYEXR_USE_OPENMP
shun-iwasawa 7bad49
#pragma omp parallel for
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  for (int i = 0; i < num_tiles; i++) {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    size_t tile_idx = static_cast<size_t>(i);
shun-iwasawa 7bad49
    size_t data_idx = tile_idx + start_index;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int x_tile = i % num_x_tiles;
shun-iwasawa 7bad49
    int y_tile = i / num_x_tiles;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    EXRTile& tile = level_image->tiles[tile_idx];
shun-iwasawa 7bad49
 
shun-iwasawa 7bad49
    const unsigned char* const* images =
shun-iwasawa 7bad49
      static_cast<const unsigned char* const*>(tile.images);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    data_list[data_idx].resize(5*sizeof(int));
shun-iwasawa 7bad49
    size_t data_header_size = data_list[data_idx].size();
shun-iwasawa 7bad49
    bool ret = EncodePixelData(data_list[data_idx],                         
shun-iwasawa 7bad49
                               images,
shun-iwasawa 7bad49
                               exr_header->compression_type,
shun-iwasawa 7bad49
                               0, // increasing y
shun-iwasawa 7bad49
                               tile.width,
shun-iwasawa 7bad49
                               exr_header->tile_size_y,
shun-iwasawa 7bad49
                               exr_header->tile_size_x,
shun-iwasawa 7bad49
                               0,
shun-iwasawa 7bad49
                               tile.height,
shun-iwasawa 7bad49
                               pixel_data_size,
shun-iwasawa 7bad49
                               channels,
shun-iwasawa 7bad49
                               channel_offset_list,
shun-iwasawa 7bad49
                               compression_param);
shun-iwasawa 7bad49
    if (!ret) {
shun-iwasawa 7bad49
      invalid_data = true;
shun-iwasawa 7bad49
      continue;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    assert(data_list[data_idx].size() > data_header_size);
shun-iwasawa 7bad49
    int data_len = static_cast<int>(data_list[data_idx].size() - data_header_size);
shun-iwasawa 7bad49
    //tileX, tileY, levelX, levelY // pixel_data_size(int)
shun-iwasawa 7bad49
    memcpy(&data_list[data_idx][0], &x_tile, sizeof(int));
shun-iwasawa 7bad49
    memcpy(&data_list[data_idx][4], &y_tile, sizeof(int));
shun-iwasawa 7bad49
    memcpy(&data_list[data_idx][8], &level_image->level_x, sizeof(int));
shun-iwasawa 7bad49
    memcpy(&data_list[data_idx][12], &level_image->level_y, sizeof(int));
shun-iwasawa 7bad49
    memcpy(&data_list[data_idx][16], &data_len, sizeof(int));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    swap4(reinterpret_cast<int*>(&data_list[data_idx][0]));
shun-iwasawa 7bad49
    swap4(reinterpret_cast<int*>(&data_list[data_idx][4]));
shun-iwasawa 7bad49
    swap4(reinterpret_cast<int*>(&data_list[data_idx][8]));
shun-iwasawa 7bad49
    swap4(reinterpret_cast<int*>(&data_list[data_idx][12]));
shun-iwasawa 7bad49
    swap4(reinterpret_cast<int*>(&data_list[data_idx][16]));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}));
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (auto &t : workers) {
shun-iwasawa 7bad49
      t.join();
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    }  // omp parallel
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (invalid_data) {
shun-iwasawa 7bad49
    if (err) {
shun-iwasawa 7bad49
      (*err) += "Failed to encode tile data.\n";  
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int NumScanlines(int compression_type) {
shun-iwasawa 7bad49
  int num_scanlines = 1;
shun-iwasawa 7bad49
  if (compression_type == TINYEXR_COMPRESSIONTYPE_ZIP) {
shun-iwasawa 7bad49
    num_scanlines = 16;
shun-iwasawa 7bad49
  } else if (compression_type == TINYEXR_COMPRESSIONTYPE_PIZ) {
shun-iwasawa 7bad49
    num_scanlines = 32;
shun-iwasawa 7bad49
  } else if (compression_type == TINYEXR_COMPRESSIONTYPE_ZFP) {
shun-iwasawa 7bad49
    num_scanlines = 16;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return num_scanlines;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
static int EncodeChunk(const EXRImage* exr_image, const EXRHeader* exr_header,
shun-iwasawa 7bad49
                       const std::vector<ChannelInfo>& channels,
shun-iwasawa 7bad49
                       int num_blocks,
shun-iwasawa 7bad49
                       tinyexr_uint64 chunk_offset, // starting offset of current chunk
shun-iwasawa 7bad49
                       bool is_multipart,
shun-iwasawa 7bad49
                       OffsetData& offset_data, // output block offsets, must be initialized
shun-iwasawa 7bad49
                       std::vector<std::vector<unsigned char> >& data_list, // output
shun-iwasawa 7bad49
                       tinyexr_uint64& total_size, // output: ending offset of current chunk
shun-iwasawa 7bad49
                       std::string* err) {
shun-iwasawa 7bad49
  int num_scanlines = NumScanlines(exr_header->compression_type);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  data_list.resize(num_blocks);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<size_t> channel_offset_list(
shun-iwasawa 7bad49
    static_cast<size_t>(exr_header->num_channels));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int pixel_data_size = 0;
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    size_t channel_offset = 0;
shun-iwasawa 7bad49
    for (size_t c = 0; c < static_cast<size_t>(exr_header->num_channels); c++) {
shun-iwasawa 7bad49
      channel_offset_list[c] = channel_offset;
shun-iwasawa 7bad49
      if (channels[c].requested_pixel_type == TINYEXR_PIXELTYPE_HALF) {
shun-iwasawa 7bad49
        pixel_data_size += sizeof(unsigned short);
shun-iwasawa 7bad49
        channel_offset += sizeof(unsigned short);
shun-iwasawa 7bad49
      } else if (channels[c].requested_pixel_type ==
shun-iwasawa 7bad49
                 TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
        pixel_data_size += sizeof(float);
shun-iwasawa 7bad49
        channel_offset += sizeof(float);
shun-iwasawa 7bad49
      } else if (channels[c].requested_pixel_type == TINYEXR_PIXELTYPE_UINT) {
shun-iwasawa 7bad49
        pixel_data_size += sizeof(unsigned int);
shun-iwasawa 7bad49
        channel_offset += sizeof(unsigned int);
shun-iwasawa 7bad49
      } else {
shun-iwasawa 7bad49
        assert(0);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const void* compression_param = 0;
shun-iwasawa 7bad49
#if TINYEXR_USE_ZFP
shun-iwasawa 7bad49
  tinyexr::ZFPCompressionParam zfp_compression_param;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Use ZFP compression parameter from custom attributes(if such a parameter
shun-iwasawa 7bad49
  // exists)
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    std::string e;
shun-iwasawa 7bad49
    bool ret = tinyexr::FindZFPCompressionParam(
shun-iwasawa 7bad49
      &zfp_compression_param, exr_header->custom_attributes,
shun-iwasawa 7bad49
      exr_header->num_custom_attributes, &e);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (!ret) {
shun-iwasawa 7bad49
      // Use predefined compression parameter.
shun-iwasawa 7bad49
      zfp_compression_param.type = 0;
shun-iwasawa 7bad49
      zfp_compression_param.rate = 2;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    compression_param = &zfp_compression_param;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  tinyexr_uint64 offset = chunk_offset;
shun-iwasawa 7bad49
  tinyexr_uint64 doffset = is_multipart ? 4u : 0u;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_image->tiles) {
shun-iwasawa 7bad49
    const EXRImage* level_image = exr_image;
shun-iwasawa 7bad49
    size_t block_idx = 0;
shun-iwasawa 7bad49
    tinyexr::tinyexr_uint64 block_data_size = 0;
shun-iwasawa 7bad49
    int num_levels = (exr_header->tile_level_mode != TINYEXR_TILE_RIPMAP_LEVELS) ?
shun-iwasawa 7bad49
      offset_data.num_x_levels : (offset_data.num_x_levels * offset_data.num_y_levels);
shun-iwasawa 7bad49
    for (int level_index = 0; level_index < num_levels; ++level_index) {
shun-iwasawa 7bad49
      if (!level_image) {
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          (*err) += "Invalid number of tiled levels for EncodeChunk\n";
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      int level_index_from_image = LevelIndex(level_image->level_x, level_image->level_y,
shun-iwasawa 7bad49
                                    exr_header->tile_level_mode, offset_data.num_x_levels);
shun-iwasawa 7bad49
      if (level_index_from_image != level_index) {
shun-iwasawa 7bad49
        if (err) {
shun-iwasawa 7bad49
          (*err) += "Incorrect level ordering in tiled image\n";
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      int num_y_tiles = (int)offset_data.offsets[level_index].size();
shun-iwasawa 7bad49
      assert(num_y_tiles);
shun-iwasawa 7bad49
      int num_x_tiles = (int)offset_data.offsets[level_index][0].size();
shun-iwasawa 7bad49
      assert(num_x_tiles);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      std::string e;
shun-iwasawa 7bad49
      int ret = EncodeTiledLevel(level_image,
shun-iwasawa 7bad49
                                  exr_header,
shun-iwasawa 7bad49
                                  channels,
shun-iwasawa 7bad49
                                  data_list,
shun-iwasawa 7bad49
                                  block_idx,
shun-iwasawa 7bad49
                                  num_x_tiles,
shun-iwasawa 7bad49
                                  num_y_tiles,
shun-iwasawa 7bad49
                                  channel_offset_list,
shun-iwasawa 7bad49
                                  pixel_data_size,
shun-iwasawa 7bad49
                                  compression_param,
shun-iwasawa 7bad49
                                  &e);
shun-iwasawa 7bad49
      if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
        if (!e.empty() && err) {
shun-iwasawa 7bad49
          (*err) += e;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        return ret;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      for (size_t j = 0; j < static_cast<size_t>(num_y_tiles); ++j)
shun-iwasawa 7bad49
        for (size_t i = 0; i < static_cast<size_t>(num_x_tiles); ++i) {
shun-iwasawa 7bad49
          offset_data.offsets[level_index][j][i] = offset;
shun-iwasawa 7bad49
          swap8(reinterpret_cast<tinyexr_uint64*>(&offset_data.offsets[level_index][j][i]));
shun-iwasawa 7bad49
          offset += data_list[block_idx].size() + doffset;
shun-iwasawa 7bad49
          block_data_size += data_list[block_idx].size();
shun-iwasawa 7bad49
          ++block_idx;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      level_image = level_image->next_level;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    assert(static_cast<int>(block_idx) == num_blocks);
shun-iwasawa 7bad49
    total_size = offset;
shun-iwasawa 7bad49
  } else { // scanlines
shun-iwasawa 7bad49
    std::vector<tinyexr::tinyexr_uint64>& offsets = offset_data.offsets[0][0];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
    std::atomic<bool> invalid_data(false);
shun-iwasawa 7bad49
    std::vector<std::thread> workers;
shun-iwasawa 7bad49
    std::atomic<int> block_count(0);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    int num_threads = std::min(std::max(1, int(std::thread::hardware_concurrency())), num_blocks);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (int t = 0; t < num_threads; t++) {
shun-iwasawa 7bad49
      workers.emplace_back(std::thread([&]() {
shun-iwasawa 7bad49
        int i = 0;
shun-iwasawa 7bad49
        while ((i = block_count++) < num_blocks) {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    bool invalid_data(false);
shun-iwasawa 7bad49
#if TINYEXR_USE_OPENMP
shun-iwasawa 7bad49
#pragma omp parallel for
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    for (int i = 0; i < num_blocks; i++) {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
      int start_y = num_scanlines * i;
shun-iwasawa 7bad49
      int end_Y = (std::min)(num_scanlines * (i + 1), exr_image->height);
shun-iwasawa 7bad49
      int num_lines = end_Y - start_y;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      const unsigned char* const* images =
shun-iwasawa 7bad49
        static_cast<const unsigned char* const*>(exr_image->images);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      data_list[i].resize(2*sizeof(int));
shun-iwasawa 7bad49
      size_t data_header_size = data_list[i].size();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      bool ret = EncodePixelData(data_list[i],                         
shun-iwasawa 7bad49
                                 images,
shun-iwasawa 7bad49
                                 exr_header->compression_type,
shun-iwasawa 7bad49
                                 0, // increasing y
shun-iwasawa 7bad49
                                 exr_image->width,
shun-iwasawa 7bad49
                                 exr_image->height,
shun-iwasawa 7bad49
                                 exr_image->width,
shun-iwasawa 7bad49
                                 start_y,
shun-iwasawa 7bad49
                                 num_lines,
shun-iwasawa 7bad49
                                 pixel_data_size,
shun-iwasawa 7bad49
                                 channels,
shun-iwasawa 7bad49
                                 channel_offset_list,
shun-iwasawa 7bad49
                                 compression_param);
shun-iwasawa 7bad49
      if (!ret) {
shun-iwasawa 7bad49
        invalid_data = true;
shun-iwasawa 7bad49
        continue; // "break" cannot be used with OpenMP 
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      assert(data_list[i].size() > data_header_size);
shun-iwasawa 7bad49
      int data_len = static_cast<int>(data_list[i].size() - data_header_size);
shun-iwasawa 7bad49
      memcpy(&data_list[i][0], &start_y, sizeof(int));
shun-iwasawa 7bad49
      memcpy(&data_list[i][4], &data_len, sizeof(int));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      swap4(reinterpret_cast<int*>(&data_list[i][0]));
shun-iwasawa 7bad49
      swap4(reinterpret_cast<int*>(&data_list[i][4]));
shun-iwasawa 7bad49
#if TINYEXR_HAS_CXX11 && (TINYEXR_USE_THREAD > 0)
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
                                       }));
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (auto &t : workers) {
shun-iwasawa 7bad49
      t.join();
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    }  // omp parallel
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (invalid_data) {
shun-iwasawa 7bad49
      if (err) {
shun-iwasawa 7bad49
        (*err) += "Failed to encode scanline data.\n";
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    for (size_t i = 0; i < static_cast<size_t>(num_blocks); i++) {
shun-iwasawa 7bad49
      offsets[i] = offset;
shun-iwasawa 7bad49
      tinyexr::swap8(reinterpret_cast<tinyexr::tinyexr_uint64 *>(&offsets[i]));
shun-iwasawa 7bad49
      offset += data_list[i].size() + doffset;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    total_size = static_cast<size_t>(offset);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
// can save a single or multi-part image (no deep* formats)
shun-iwasawa 7bad49
static size_t SaveEXRNPartImageToMemory(const EXRImage* exr_images,
shun-iwasawa 7bad49
                                        const EXRHeader** exr_headers,
shun-iwasawa 7bad49
                                        unsigned int num_parts,
shun-iwasawa 7bad49
                                        unsigned char** memory_out, const char** err) {
shun-iwasawa 7bad49
  if (exr_images == NULL || exr_headers == NULL || num_parts == 0 ||
shun-iwasawa 7bad49
      memory_out == NULL) {
shun-iwasawa 7bad49
    SetErrorMessage("Invalid argument for SaveEXRNPartImageToMemory",
shun-iwasawa 7bad49
                    err);
shun-iwasawa 7bad49
    return 0;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    for (unsigned int i = 0; i < num_parts; ++i) {
shun-iwasawa 7bad49
      if (exr_headers[i]->compression_type < 0) {
shun-iwasawa 7bad49
        SetErrorMessage("Invalid argument for SaveEXRNPartImageToMemory",
shun-iwasawa 7bad49
                        err);
shun-iwasawa 7bad49
        return 0;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
#if !TINYEXR_USE_PIZ
shun-iwasawa 7bad49
      if (exr_headers[i]->compression_type == TINYEXR_COMPRESSIONTYPE_PIZ) {
shun-iwasawa 7bad49
        SetErrorMessage("PIZ compression is not supported in this build",
shun-iwasawa 7bad49
                        err);
shun-iwasawa 7bad49
        return 0;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#if !TINYEXR_USE_ZFP
shun-iwasawa 7bad49
      if (exr_headers[i]->compression_type == TINYEXR_COMPRESSIONTYPE_ZFP) {
shun-iwasawa 7bad49
        SetErrorMessage("ZFP compression is not supported in this build",
shun-iwasawa 7bad49
                        err);
shun-iwasawa 7bad49
        return 0;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
      for (int c = 0; c < exr_header->num_channels; ++c) {
shun-iwasawa 7bad49
        if (exr_headers[i]->requested_pixel_types[c] != TINYEXR_PIXELTYPE_FLOAT) {
shun-iwasawa 7bad49
          SetErrorMessage("Pixel type must be FLOAT for ZFP compression",
shun-iwasawa 7bad49
                          err);
shun-iwasawa 7bad49
          return 0;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned char> memory;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Header
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    const char header[] = { 0x76, 0x2f, 0x31, 0x01 };
shun-iwasawa 7bad49
    memory.insert(memory.end(), header, header + 4);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Version
shun-iwasawa 7bad49
  // using value from the first header
shun-iwasawa 7bad49
  int long_name = exr_headers[0]->long_name;
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    char marker[] = { 2, 0, 0, 0 };
shun-iwasawa 7bad49
    /* @todo
shun-iwasawa 7bad49
    if (exr_header->non_image) {
shun-iwasawa 7bad49
    marker[1] |= 0x8;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    */
shun-iwasawa 7bad49
    // tiled
shun-iwasawa 7bad49
    if (num_parts == 1 && exr_images[0].tiles) {
shun-iwasawa 7bad49
      marker[1] |= 0x2;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    // long_name
shun-iwasawa 7bad49
    if (long_name) {
shun-iwasawa 7bad49
      marker[1] |= 0x4;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    // multipart
shun-iwasawa 7bad49
    if (num_parts > 1) {
shun-iwasawa 7bad49
      marker[1] |= 0x10;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    memory.insert(memory.end(), marker, marker + 4);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int total_chunk_count = 0;
shun-iwasawa 7bad49
  std::vector<int> chunk_count(num_parts);
shun-iwasawa 7bad49
  std::vector<OffsetData> offset_data(num_parts);
shun-iwasawa 7bad49
  for (unsigned int i = 0; i < num_parts; ++i) {
shun-iwasawa 7bad49
    if (!exr_images[i].tiles) {
shun-iwasawa 7bad49
      int num_scanlines = NumScanlines(exr_headers[i]->compression_type);
shun-iwasawa 7bad49
      chunk_count[i] =
shun-iwasawa 7bad49
        (exr_images[i].height + num_scanlines - 1) / num_scanlines;
shun-iwasawa 7bad49
      InitSingleResolutionOffsets(offset_data[i], chunk_count[i]);
shun-iwasawa 7bad49
      total_chunk_count += chunk_count[i];
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        std::vector<int> num_x_tiles, num_y_tiles;
shun-iwasawa 7bad49
        PrecalculateTileInfo(num_x_tiles, num_y_tiles, exr_headers[i]);
shun-iwasawa 7bad49
        chunk_count[i] =
shun-iwasawa 7bad49
          InitTileOffsets(offset_data[i], exr_headers[i], num_x_tiles, num_y_tiles);
shun-iwasawa 7bad49
        total_chunk_count += chunk_count[i];
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  // Write attributes to memory buffer.
shun-iwasawa 7bad49
  std::vector< std::vector<tinyexr::ChannelInfo> > channels(num_parts);
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    std::set<std::string> partnames;
shun-iwasawa 7bad49
    for (unsigned int i = 0; i < num_parts; ++i) {
shun-iwasawa 7bad49
      //channels
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        std::vector<unsigned char> data;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        for (int c = 0; c < exr_headers[i]->num_channels; c++) {
shun-iwasawa 7bad49
          tinyexr::ChannelInfo info;
shun-iwasawa 7bad49
          info.p_linear = 0;
shun-iwasawa 7bad49
          info.pixel_type = exr_headers[i]->pixel_types[c];
shun-iwasawa 7bad49
          info.requested_pixel_type = exr_headers[i]->requested_pixel_types[c];
shun-iwasawa 7bad49
          info.x_sampling = 1;
shun-iwasawa 7bad49
          info.y_sampling = 1;
shun-iwasawa 7bad49
          info.name = std::string(exr_headers[i]->channels[c].name);
shun-iwasawa 7bad49
          channels[i].push_back(info);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        tinyexr::WriteChannelInfo(data, channels[i]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        tinyexr::WriteAttributeToMemory(&memory, "channels", "chlist", &data.at(0),
shun-iwasawa 7bad49
                                        static_cast<int>(data.size()));
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        int comp = exr_headers[i]->compression_type;
shun-iwasawa 7bad49
        swap4(&comp);
shun-iwasawa 7bad49
        WriteAttributeToMemory(
shun-iwasawa 7bad49
          &memory, "compression", "compression",
shun-iwasawa 7bad49
          reinterpret_cast<const unsigned char*>(&comp), 1);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        int data[4] = { 0, 0, exr_images[i].width - 1, exr_images[i].height - 1 };
shun-iwasawa 7bad49
        swap4(&data[0]);
shun-iwasawa 7bad49
        swap4(&data[1]);
shun-iwasawa 7bad49
        swap4(&data[2]);
shun-iwasawa 7bad49
        swap4(&data[3]);
shun-iwasawa 7bad49
        WriteAttributeToMemory(
shun-iwasawa 7bad49
          &memory, "dataWindow", "box2i",
shun-iwasawa 7bad49
          reinterpret_cast<const unsigned char*>(data), sizeof(int) * 4);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        int data0[4] = { 0, 0, exr_images[0].width - 1, exr_images[0].height - 1 };
shun-iwasawa 7bad49
        swap4(&data0[0]);
shun-iwasawa 7bad49
        swap4(&data0[1]);
shun-iwasawa 7bad49
        swap4(&data0[2]);
shun-iwasawa 7bad49
        swap4(&data0[3]);
shun-iwasawa 7bad49
        // Note: must be the same across parts (currently, using value from the first header)
shun-iwasawa 7bad49
        WriteAttributeToMemory(
shun-iwasawa 7bad49
          &memory, "displayWindow", "box2i",
shun-iwasawa 7bad49
          reinterpret_cast<const unsigned char*>(data0), sizeof(int) * 4);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        unsigned char line_order = 0;  // @fixme { read line_order from EXRHeader }
shun-iwasawa 7bad49
        WriteAttributeToMemory(&memory, "lineOrder", "lineOrder",
shun-iwasawa 7bad49
                               &line_order, 1);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        // Note: must be the same across parts
shun-iwasawa 7bad49
        float aspectRatio = 1.0f;
shun-iwasawa 7bad49
        swap4(&aspectRatio);
shun-iwasawa 7bad49
        WriteAttributeToMemory(
shun-iwasawa 7bad49
          &memory, "pixelAspectRatio", "float",
shun-iwasawa 7bad49
          reinterpret_cast<const unsigned char*>(&aspectRatio), sizeof(float));
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        float center[2] = { 0.0f, 0.0f };
shun-iwasawa 7bad49
        swap4(&center[0]);
shun-iwasawa 7bad49
        swap4(&center[1]);
shun-iwasawa 7bad49
        WriteAttributeToMemory(
shun-iwasawa 7bad49
          &memory, "screenWindowCenter", "v2f",
shun-iwasawa 7bad49
          reinterpret_cast<const unsigned char*>(center), 2 * sizeof(float));
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        float w = 1.0f;
shun-iwasawa 7bad49
        swap4(&w);
shun-iwasawa 7bad49
        WriteAttributeToMemory(&memory, "screenWindowWidth", "float",
shun-iwasawa 7bad49
                               reinterpret_cast<const unsigned char*>(&w),
shun-iwasawa 7bad49
                               sizeof(float));
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (exr_images[i].tiles) {
shun-iwasawa 7bad49
        unsigned char tile_mode = static_cast<unsigned char>(exr_headers[i]->tile_level_mode & 0x3);
shun-iwasawa 7bad49
        if (exr_headers[i]->tile_rounding_mode) tile_mode |= (1u << 4u);
shun-iwasawa 7bad49
        //unsigned char data[9] = { 0, 0, 0, 0, 0, 0, 0, 0, 0 };
shun-iwasawa 7bad49
        unsigned int datai[3] = { 0, 0, 0 };
shun-iwasawa 7bad49
        unsigned char* data = reinterpret_cast<unsigned char*>(&datai[0]);
shun-iwasawa 7bad49
        datai[0] = static_cast<unsigned int>(exr_headers[i]->tile_size_x);
shun-iwasawa 7bad49
        datai[1] = static_cast<unsigned int>(exr_headers[i]->tile_size_y);
shun-iwasawa 7bad49
        data[8] = tile_mode;
shun-iwasawa 7bad49
        swap4(reinterpret_cast<unsigned int*>(&data[0]));
shun-iwasawa 7bad49
        swap4(reinterpret_cast<unsigned int*>(&data[4]));
shun-iwasawa 7bad49
        WriteAttributeToMemory(
shun-iwasawa 7bad49
          &memory, "tiles", "tiledesc",
shun-iwasawa 7bad49
          reinterpret_cast<const unsigned char*>(data), 9);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      // must be present for multi-part files - according to spec.
shun-iwasawa 7bad49
      if (num_parts > 1) {
shun-iwasawa 7bad49
        // name
shun-iwasawa 7bad49
        {
shun-iwasawa 7bad49
          size_t len = 0;
shun-iwasawa 7bad49
          if ((len = strlen(exr_headers[i]->name)) > 0) {
shun-iwasawa 7bad49
            partnames.emplace(exr_headers[i]->name);
shun-iwasawa 7bad49
            if (partnames.size() != i + 1) {
shun-iwasawa 7bad49
              SetErrorMessage("'name' attributes must be unique for a multi-part file", err);
shun-iwasawa 7bad49
              return 0;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            WriteAttributeToMemory(
shun-iwasawa 7bad49
              &memory, "name", "string",
shun-iwasawa 7bad49
              reinterpret_cast<const unsigned char*>(exr_headers[i]->name),
shun-iwasawa 7bad49
              static_cast<int>(len));
shun-iwasawa 7bad49
          } else {
shun-iwasawa 7bad49
            SetErrorMessage("Invalid 'name' attribute for a multi-part file", err);
shun-iwasawa 7bad49
            return 0;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        // type
shun-iwasawa 7bad49
        {
shun-iwasawa 7bad49
          const char* type = "scanlineimage";
shun-iwasawa 7bad49
          if (exr_images[i].tiles) type = "tiledimage";
shun-iwasawa 7bad49
          WriteAttributeToMemory(
shun-iwasawa 7bad49
            &memory, "type", "string",
shun-iwasawa 7bad49
            reinterpret_cast<const unsigned char*>(type),
shun-iwasawa 7bad49
            static_cast<int>(strlen(type)));
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        // chunkCount
shun-iwasawa 7bad49
        {
shun-iwasawa 7bad49
          WriteAttributeToMemory(
shun-iwasawa 7bad49
            &memory, "chunkCount", "int",
shun-iwasawa 7bad49
            reinterpret_cast<const unsigned char*>(&chunk_count[i]),
shun-iwasawa 7bad49
            4);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      // Custom attributes
shun-iwasawa 7bad49
      if (exr_headers[i]->num_custom_attributes > 0) {
shun-iwasawa 7bad49
        for (int j = 0; j < exr_headers[i]->num_custom_attributes; j++) {
shun-iwasawa 7bad49
          tinyexr::WriteAttributeToMemory(
shun-iwasawa 7bad49
            &memory, exr_headers[i]->custom_attributes[j].name,
shun-iwasawa 7bad49
            exr_headers[i]->custom_attributes[j].type,
shun-iwasawa 7bad49
            reinterpret_cast<const unsigned char*>(
shun-iwasawa 7bad49
              exr_headers[i]->custom_attributes[j].value),
shun-iwasawa 7bad49
            exr_headers[i]->custom_attributes[j].size);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      {  // end of header
shun-iwasawa 7bad49
        memory.push_back(0);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  if (num_parts > 1) {
shun-iwasawa 7bad49
    // end of header list
shun-iwasawa 7bad49
    memory.push_back(0);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  tinyexr_uint64 chunk_offset = memory.size() + size_t(total_chunk_count) * sizeof(tinyexr_uint64);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  tinyexr_uint64 total_size = 0;
shun-iwasawa 7bad49
  std::vector< std::vector< std::vector<unsigned char> > > data_lists(num_parts);
shun-iwasawa 7bad49
  for (unsigned int i = 0; i < num_parts; ++i) {
shun-iwasawa 7bad49
    std::string e;
shun-iwasawa 7bad49
    int ret = EncodeChunk(&exr_images[i], exr_headers[i],
shun-iwasawa 7bad49
                          channels[i],
shun-iwasawa 7bad49
                          chunk_count[i],
shun-iwasawa 7bad49
                          // starting offset of current chunk after part-number
shun-iwasawa 7bad49
                          chunk_offset,
shun-iwasawa 7bad49
                          num_parts > 1,
shun-iwasawa 7bad49
                          offset_data[i], // output: block offsets, must be initialized
shun-iwasawa 7bad49
                          data_lists[i], // output
shun-iwasawa 7bad49
                          total_size, // output
shun-iwasawa 7bad49
                          &e);
shun-iwasawa 7bad49
    if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
      if (!e.empty()) {
shun-iwasawa 7bad49
        tinyexr::SetErrorMessage(e, err);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return 0;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    chunk_offset = total_size;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Allocating required memory
shun-iwasawa 7bad49
  if (total_size == 0) { // something went wrong
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Output memory size is zero", err);
shun-iwasawa 7bad49
    return 0;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  (*memory_out) = static_cast<unsigned char*>(malloc(total_size));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Writing header
shun-iwasawa 7bad49
  memcpy((*memory_out), &memory[0], memory.size());
shun-iwasawa 7bad49
  unsigned char* memory_ptr = *memory_out + memory.size();
shun-iwasawa 7bad49
  size_t sum = memory.size();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Writing offset data for chunks
shun-iwasawa 7bad49
  for (unsigned int i = 0; i < num_parts; ++i) {
shun-iwasawa 7bad49
    if (exr_images[i].tiles) {
shun-iwasawa 7bad49
      const EXRImage* level_image = &exr_images[i];
shun-iwasawa 7bad49
      int num_levels = (exr_headers[i]->tile_level_mode != TINYEXR_TILE_RIPMAP_LEVELS) ?
shun-iwasawa 7bad49
        offset_data[i].num_x_levels : (offset_data[i].num_x_levels * offset_data[i].num_y_levels);
shun-iwasawa 7bad49
      for (int level_index = 0; level_index < num_levels; ++level_index) {
shun-iwasawa 7bad49
        for (size_t j = 0; j < offset_data[i].offsets[level_index].size(); ++j) {
shun-iwasawa 7bad49
          size_t num_bytes = sizeof(tinyexr_uint64) * offset_data[i].offsets[level_index][j].size();
shun-iwasawa 7bad49
          sum += num_bytes;
shun-iwasawa 7bad49
          assert(sum <= total_size);
shun-iwasawa 7bad49
          memcpy(memory_ptr,
shun-iwasawa 7bad49
                 reinterpret_cast<unsigned char*>(&offset_data[i].offsets[level_index][j][0]),
shun-iwasawa 7bad49
                 num_bytes);
shun-iwasawa 7bad49
          memory_ptr += num_bytes;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        level_image = level_image->next_level;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      size_t num_bytes = sizeof(tinyexr::tinyexr_uint64) * static_cast<size_t>(chunk_count[i]);
shun-iwasawa 7bad49
      sum += num_bytes;
shun-iwasawa 7bad49
      assert(sum <= total_size);
shun-iwasawa 7bad49
      std::vector<tinyexr::tinyexr_uint64>& offsets = offset_data[i].offsets[0][0];
shun-iwasawa 7bad49
      memcpy(memory_ptr, reinterpret_cast<unsigned char*>(&offsets[0]), num_bytes);
shun-iwasawa 7bad49
      memory_ptr += num_bytes;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Writing chunk data
shun-iwasawa 7bad49
  for (unsigned int i = 0; i < num_parts; ++i) {
shun-iwasawa 7bad49
    for (size_t j = 0; j < static_cast<size_t>(chunk_count[i]); ++j) {
shun-iwasawa 7bad49
      if (num_parts > 1) {
shun-iwasawa 7bad49
        sum += 4;
shun-iwasawa 7bad49
        assert(sum <= total_size);
shun-iwasawa 7bad49
        unsigned int part_number = i;
shun-iwasawa 7bad49
        swap4(&part_number);
shun-iwasawa 7bad49
        memcpy(memory_ptr, &part_number, 4);
shun-iwasawa 7bad49
        memory_ptr += 4;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      sum += data_lists[i][j].size();
shun-iwasawa 7bad49
      assert(sum <= total_size);
shun-iwasawa 7bad49
      memcpy(memory_ptr, &data_lists[i][j][0], data_lists[i][j].size());
shun-iwasawa 7bad49
      memory_ptr += data_lists[i][j].size();
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  assert(sum == total_size);
shun-iwasawa 7bad49
  return total_size;  // OK
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
} // tinyexr
shun-iwasawa 7bad49
shun-iwasawa 7bad49
size_t SaveEXRImageToMemory(const EXRImage* exr_image,
shun-iwasawa 7bad49
                             const EXRHeader* exr_header,
shun-iwasawa 7bad49
                             unsigned char** memory_out, const char** err) {
shun-iwasawa 7bad49
  return tinyexr::SaveEXRNPartImageToMemory(exr_image, &exr_header, 1, memory_out, err);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int SaveEXRImageToFile(const EXRImage *exr_image, const EXRHeader *exr_header,
shun-iwasawa 7bad49
                       const char *filename, const char **err) {
shun-iwasawa 7bad49
  if (exr_image == NULL || filename == NULL ||
shun-iwasawa 7bad49
      exr_header->compression_type < 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for SaveEXRImageToFile", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if !TINYEXR_USE_PIZ
shun-iwasawa 7bad49
  if (exr_header->compression_type == TINYEXR_COMPRESSIONTYPE_PIZ) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("PIZ compression is not supported in this build",
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_UNSUPPORTED_FEATURE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if !TINYEXR_USE_ZFP
shun-iwasawa 7bad49
  if (exr_header->compression_type == TINYEXR_COMPRESSIONTYPE_ZFP) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("ZFP compression is not supported in this build",
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_UNSUPPORTED_FEATURE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FILE *fp = NULL;
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
#if defined(_MSC_VER) || (defined(MINGW_HAS_SECURE_API) && MINGW_HAS_SECURE_API) // MSVC, MinGW GCC, or Clang
shun-iwasawa 7bad49
  errno_t errcode =
shun-iwasawa 7bad49
      _wfopen_s(&fp, tinyexr::UTF8ToWchar(filename).c_str(), L"wb");
shun-iwasawa 7bad49
  if (errcode != 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot write a file: " + std::string(filename),
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_WRITE_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  // Unknown compiler or MinGW without MINGW_HAS_SECURE_API.
shun-iwasawa 7bad49
  fp = fopen(filename, "wb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  fp = fopen(filename, "wb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  if (!fp) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot write a file: " + std::string(filename),
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_WRITE_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  unsigned char *mem = NULL;
shun-iwasawa 7bad49
  size_t mem_size = SaveEXRImageToMemory(exr_image, exr_header, &mem, err);
shun-iwasawa 7bad49
  if (mem_size == 0) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_SERIALZATION_FAILED;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t written_size = 0;
shun-iwasawa 7bad49
  if ((mem_size > 0) && mem) {
shun-iwasawa 7bad49
    written_size = fwrite(mem, 1, mem_size, fp);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  free(mem);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  fclose(fp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (written_size != mem_size) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot write a file", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_WRITE_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
size_t SaveEXRMultipartImageToMemory(const EXRImage* exr_images,
shun-iwasawa 7bad49
                                     const EXRHeader** exr_headers,
shun-iwasawa 7bad49
                                     unsigned int num_parts,
shun-iwasawa 7bad49
                                     unsigned char** memory_out, const char** err) {
shun-iwasawa 7bad49
  if (exr_images == NULL || exr_headers == NULL || num_parts < 2 ||
shun-iwasawa 7bad49
      memory_out == NULL) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for SaveEXRNPartImageToMemory",
shun-iwasawa 7bad49
                              err);
shun-iwasawa 7bad49
    return 0;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return tinyexr::SaveEXRNPartImageToMemory(exr_images, exr_headers, num_parts, memory_out, err);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int SaveEXRMultipartImageToFile(const EXRImage* exr_images,
shun-iwasawa 7bad49
                                const EXRHeader** exr_headers,
shun-iwasawa 7bad49
                                unsigned int num_parts,
shun-iwasawa 7bad49
                                const char* filename,
shun-iwasawa 7bad49
                                const char** err) {
shun-iwasawa 7bad49
  if (exr_images == NULL || exr_headers == NULL || num_parts < 2) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for SaveEXRMultipartImageToFile",
shun-iwasawa 7bad49
                              err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FILE *fp = NULL;
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
#if defined(_MSC_VER) || (defined(MINGW_HAS_SECURE_API) && MINGW_HAS_SECURE_API) // MSVC, MinGW GCC, or Clang.
shun-iwasawa 7bad49
  errno_t errcode =
shun-iwasawa 7bad49
    _wfopen_s(&fp, tinyexr::UTF8ToWchar(filename).c_str(), L"wb");
shun-iwasawa 7bad49
  if (errcode != 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot write a file: " + std::string(filename),
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_WRITE_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  // Unknown compiler or MinGW without MINGW_HAS_SECURE_API.
shun-iwasawa 7bad49
  fp = fopen(filename, "wb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  fp = fopen(filename, "wb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  if (!fp) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot write a file: " + std::string(filename),
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_WRITE_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  unsigned char *mem = NULL;
shun-iwasawa 7bad49
  size_t mem_size = SaveEXRMultipartImageToMemory(exr_images, exr_headers, num_parts, &mem, err);
shun-iwasawa 7bad49
  if (mem_size == 0) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_SERIALZATION_FAILED;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t written_size = 0;
shun-iwasawa 7bad49
  if ((mem_size > 0) && mem) {
shun-iwasawa 7bad49
    written_size = fwrite(mem, 1, mem_size, fp);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  free(mem);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  fclose(fp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (written_size != mem_size) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot write a file", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_WRITE_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int LoadDeepEXR(DeepImage *deep_image, const char *filename, const char **err) {
shun-iwasawa 7bad49
  if (deep_image == NULL) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for LoadDeepEXR", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
  FILE *fp = NULL;
shun-iwasawa 7bad49
#if defined(_MSC_VER) || (defined(MINGW_HAS_SECURE_API) && MINGW_HAS_SECURE_API) // MSVC, MinGW GCC, or Clang.
shun-iwasawa 7bad49
  errno_t errcode =
shun-iwasawa 7bad49
      _wfopen_s(&fp, tinyexr::UTF8ToWchar(filename).c_str(), L"rb");
shun-iwasawa 7bad49
  if (errcode != 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read a file " + std::string(filename),
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  // Unknown compiler or MinGW without MINGW_HAS_SECURE_API.
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  if (!fp) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read a file " + std::string(filename),
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  FILE *fp = fopen(filename, "rb");
shun-iwasawa 7bad49
  if (!fp) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read a file " + std::string(filename),
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t filesize;
shun-iwasawa 7bad49
  // Compute size
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_END);
shun-iwasawa 7bad49
  filesize = static_cast<size_t>(ftell(fp));
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_SET);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (filesize == 0) {
shun-iwasawa 7bad49
    fclose(fp);
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("File size is zero : " + std::string(filename),
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<char> buf(filesize);  // @todo { use mmap }
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    size_t ret;
shun-iwasawa 7bad49
    ret = fread(&buf[0], 1, filesize, fp);
shun-iwasawa 7bad49
    assert(ret == filesize);
shun-iwasawa 7bad49
    (void)ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  fclose(fp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const char *head = &buf[0];
shun-iwasawa 7bad49
  const char *marker = &buf[0];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Header check.
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    const char header[] = {0x76, 0x2f, 0x31, 0x01};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (memcmp(marker, header, 4) != 0) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("Invalid magic number", err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_MAGIC_NUMBER;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    marker += 4;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Version, scanline.
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    // ver 2.0, scanline, deep bit on(0x800)
shun-iwasawa 7bad49
    // must be [2, 0, 0, 0]
shun-iwasawa 7bad49
    if (marker[0] != 2 || marker[1] != 8 || marker[2] != 0 || marker[3] != 0) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("Unsupported version or scanline", err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_UNSUPPORTED_FORMAT;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    marker += 4;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int dx = -1;
shun-iwasawa 7bad49
  int dy = -1;
shun-iwasawa 7bad49
  int dw = -1;
shun-iwasawa 7bad49
  int dh = -1;
shun-iwasawa 7bad49
  int num_scanline_blocks = 1;  // 16 for ZIP compression.
shun-iwasawa 7bad49
  int compression_type = -1;
shun-iwasawa 7bad49
  int num_channels = -1;
shun-iwasawa 7bad49
  std::vector<tinyexr::ChannelInfo> channels;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Read attributes
shun-iwasawa 7bad49
  size_t size = filesize - tinyexr::kEXRVersionSize;
shun-iwasawa 7bad49
  for (;;) {
shun-iwasawa 7bad49
    if (0 == size) {
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    } else if (marker[0] == '\0') {
shun-iwasawa 7bad49
      marker++;
shun-iwasawa 7bad49
      size--;
shun-iwasawa 7bad49
      break;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    std::string attr_name;
shun-iwasawa 7bad49
    std::string attr_type;
shun-iwasawa 7bad49
    std::vector<unsigned char> data;
shun-iwasawa 7bad49
    size_t marker_size;
shun-iwasawa 7bad49
    if (!tinyexr::ReadAttribute(&attr_name, &attr_type, &data, &marker_size,
shun-iwasawa 7bad49
                                marker, size)) {
shun-iwasawa 7bad49
      std::stringstream ss;
shun-iwasawa 7bad49
      ss << "Failed to parse attribute\n";
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage(ss.str(), err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    marker += marker_size;
shun-iwasawa 7bad49
    size -= marker_size;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (attr_name.compare("compression") == 0) {
shun-iwasawa 7bad49
      compression_type = data[0];
shun-iwasawa 7bad49
      if (compression_type > TINYEXR_COMPRESSIONTYPE_PIZ) {
shun-iwasawa 7bad49
        std::stringstream ss;
shun-iwasawa 7bad49
        ss << "Unsupported compression type : " << compression_type;
shun-iwasawa 7bad49
        tinyexr::SetErrorMessage(ss.str(), err);
shun-iwasawa 7bad49
        return TINYEXR_ERROR_UNSUPPORTED_FORMAT;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (compression_type == TINYEXR_COMPRESSIONTYPE_ZIP) {
shun-iwasawa 7bad49
        num_scanline_blocks = 16;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    } else if (attr_name.compare("channels") == 0) {
shun-iwasawa 7bad49
      // name: zero-terminated string, from 1 to 255 bytes long
shun-iwasawa 7bad49
      // pixel type: int, possible values are: UINT = 0 HALF = 1 FLOAT = 2
shun-iwasawa 7bad49
      // pLinear: unsigned char, possible values are 0 and 1
shun-iwasawa 7bad49
      // reserved: three chars, should be zero
shun-iwasawa 7bad49
      // xSampling: int
shun-iwasawa 7bad49
      // ySampling: int
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (!tinyexr::ReadChannelInfo(channels, data)) {
shun-iwasawa 7bad49
        tinyexr::SetErrorMessage("Failed to parse channel info", err);
shun-iwasawa 7bad49
        return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      num_channels = static_cast<int>(channels.size());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      if (num_channels < 1) {
shun-iwasawa 7bad49
        tinyexr::SetErrorMessage("Invalid channels format", err);
shun-iwasawa 7bad49
        return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    } else if (attr_name.compare("dataWindow") == 0) {
shun-iwasawa 7bad49
      memcpy(&dx, &data.at(0), sizeof(int));
shun-iwasawa 7bad49
      memcpy(&dy, &data.at(4), sizeof(int));
shun-iwasawa 7bad49
      memcpy(&dw, &data.at(8), sizeof(int));
shun-iwasawa 7bad49
      memcpy(&dh, &data.at(12), sizeof(int));
shun-iwasawa 7bad49
      tinyexr::swap4(&dx);
shun-iwasawa 7bad49
      tinyexr::swap4(&dy);
shun-iwasawa 7bad49
      tinyexr::swap4(&dw);
shun-iwasawa 7bad49
      tinyexr::swap4(&dh);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    } else if (attr_name.compare("displayWindow") == 0) {
shun-iwasawa 7bad49
      int x;
shun-iwasawa 7bad49
      int y;
shun-iwasawa 7bad49
      int w;
shun-iwasawa 7bad49
      int h;
shun-iwasawa 7bad49
      memcpy(&x, &data.at(0), sizeof(int));
shun-iwasawa 7bad49
      memcpy(&y, &data.at(4), sizeof(int));
shun-iwasawa 7bad49
      memcpy(&w, &data.at(8), sizeof(int));
shun-iwasawa 7bad49
      memcpy(&h, &data.at(12), sizeof(int));
shun-iwasawa 7bad49
      tinyexr::swap4(&x);
shun-iwasawa 7bad49
      tinyexr::swap4(&y);
shun-iwasawa 7bad49
      tinyexr::swap4(&w);
shun-iwasawa 7bad49
      tinyexr::swap4(&h);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  assert(dx >= 0);
shun-iwasawa 7bad49
  assert(dy >= 0);
shun-iwasawa 7bad49
  assert(dw >= 0);
shun-iwasawa 7bad49
  assert(dh >= 0);
shun-iwasawa 7bad49
  assert(num_channels >= 1);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int data_width = dw - dx + 1;
shun-iwasawa 7bad49
  int data_height = dh - dy + 1;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Read offset tables.
shun-iwasawa 7bad49
  int num_blocks = data_height / num_scanline_blocks;
shun-iwasawa 7bad49
  if (num_blocks * num_scanline_blocks < data_height) {
shun-iwasawa 7bad49
    num_blocks++;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<tinyexr::tinyexr_int64> offsets(static_cast<size_t>(num_blocks));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t y = 0; y < static_cast<size_t>(num_blocks); y++) {
shun-iwasawa 7bad49
    tinyexr::tinyexr_int64 offset;
shun-iwasawa 7bad49
    memcpy(&offset, marker, sizeof(tinyexr::tinyexr_int64));
shun-iwasawa 7bad49
    tinyexr::swap8(reinterpret_cast<tinyexr::tinyexr_uint64 *>(&offset));
shun-iwasawa 7bad49
    marker += sizeof(tinyexr::tinyexr_int64);  // = 8
shun-iwasawa 7bad49
    offsets[y] = offset;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#if TINYEXR_USE_PIZ
shun-iwasawa 7bad49
  if ((compression_type == TINYEXR_COMPRESSIONTYPE_NONE) ||
shun-iwasawa 7bad49
      (compression_type == TINYEXR_COMPRESSIONTYPE_RLE) ||
shun-iwasawa 7bad49
      (compression_type == TINYEXR_COMPRESSIONTYPE_ZIPS) ||
shun-iwasawa 7bad49
      (compression_type == TINYEXR_COMPRESSIONTYPE_ZIP) ||
shun-iwasawa 7bad49
      (compression_type == TINYEXR_COMPRESSIONTYPE_PIZ)) {
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  if ((compression_type == TINYEXR_COMPRESSIONTYPE_NONE) ||
shun-iwasawa 7bad49
      (compression_type == TINYEXR_COMPRESSIONTYPE_RLE) ||
shun-iwasawa 7bad49
      (compression_type == TINYEXR_COMPRESSIONTYPE_ZIPS) ||
shun-iwasawa 7bad49
      (compression_type == TINYEXR_COMPRESSIONTYPE_ZIP)) {
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    // OK
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Unsupported compression format", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_UNSUPPORTED_FORMAT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  deep_image->image = static_cast<float ***>(
shun-iwasawa 7bad49
      malloc(sizeof(float **) * static_cast<size_t>(num_channels)));
shun-iwasawa 7bad49
  for (int c = 0; c < num_channels; c++) {
shun-iwasawa 7bad49
    deep_image->image[c] = static_cast<float **>(
shun-iwasawa 7bad49
        malloc(sizeof(float *) * static_cast<size_t>(data_height)));
shun-iwasawa 7bad49
    for (int y = 0; y < data_height; y++) {
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  deep_image->offset_table = static_cast<int **>(
shun-iwasawa 7bad49
      malloc(sizeof(int *) * static_cast<size_t>(data_height)));
shun-iwasawa 7bad49
  for (int y = 0; y < data_height; y++) {
shun-iwasawa 7bad49
    deep_image->offset_table[y] = static_cast<int *>(
shun-iwasawa 7bad49
        malloc(sizeof(int) * static_cast<size_t>(data_width)));
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (size_t y = 0; y < static_cast<size_t>(num_blocks); y++) {
shun-iwasawa 7bad49
    const unsigned char *data_ptr =
shun-iwasawa 7bad49
        reinterpret_cast<const unsigned char *>(head + offsets[y]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // int: y coordinate
shun-iwasawa 7bad49
    // int64: packed size of pixel offset table
shun-iwasawa 7bad49
    // int64: packed size of sample data
shun-iwasawa 7bad49
    // int64: unpacked size of sample data
shun-iwasawa 7bad49
    // compressed pixel offset table
shun-iwasawa 7bad49
    // compressed sample data
shun-iwasawa 7bad49
    int line_no;
shun-iwasawa 7bad49
    tinyexr::tinyexr_int64 packedOffsetTableSize;
shun-iwasawa 7bad49
    tinyexr::tinyexr_int64 packedSampleDataSize;
shun-iwasawa 7bad49
    tinyexr::tinyexr_int64 unpackedSampleDataSize;
shun-iwasawa 7bad49
    memcpy(&line_no, data_ptr, sizeof(int));
shun-iwasawa 7bad49
    memcpy(&packedOffsetTableSize, data_ptr + 4,
shun-iwasawa 7bad49
           sizeof(tinyexr::tinyexr_int64));
shun-iwasawa 7bad49
    memcpy(&packedSampleDataSize, data_ptr + 12,
shun-iwasawa 7bad49
           sizeof(tinyexr::tinyexr_int64));
shun-iwasawa 7bad49
    memcpy(&unpackedSampleDataSize, data_ptr + 20,
shun-iwasawa 7bad49
           sizeof(tinyexr::tinyexr_int64));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tinyexr::swap4(&line_no);
shun-iwasawa 7bad49
    tinyexr::swap8(
shun-iwasawa 7bad49
        reinterpret_cast<tinyexr::tinyexr_uint64 *>(&packedOffsetTableSize));
shun-iwasawa 7bad49
    tinyexr::swap8(
shun-iwasawa 7bad49
        reinterpret_cast<tinyexr::tinyexr_uint64 *>(&packedSampleDataSize));
shun-iwasawa 7bad49
    tinyexr::swap8(
shun-iwasawa 7bad49
        reinterpret_cast<tinyexr::tinyexr_uint64 *>(&unpackedSampleDataSize));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    std::vector<int> pixelOffsetTable(static_cast<size_t>(data_width));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // decode pixel offset table.
shun-iwasawa 7bad49
    {
shun-iwasawa 7bad49
      unsigned long dstLen =
shun-iwasawa 7bad49
          static_cast<unsigned long>(pixelOffsetTable.size() * sizeof(int));
shun-iwasawa 7bad49
      if (!tinyexr::DecompressZip(
shun-iwasawa 7bad49
              reinterpret_cast<unsigned char *>(&pixelOffsetTable.at(0)),
shun-iwasawa 7bad49
              &dstLen, data_ptr + 28,
shun-iwasawa 7bad49
              static_cast<unsigned long>(packedOffsetTableSize))) {
shun-iwasawa 7bad49
        return false;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      assert(dstLen == pixelOffsetTable.size() * sizeof(int));
shun-iwasawa 7bad49
      for (size_t i = 0; i < static_cast<size_t>(data_width); i++) {
shun-iwasawa 7bad49
        deep_image->offset_table[y][i] = pixelOffsetTable[i];
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    std::vector<unsigned char> sample_data(
shun-iwasawa 7bad49
        static_cast<size_t>(unpackedSampleDataSize));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // decode sample data.
shun-iwasawa 7bad49
    {
shun-iwasawa 7bad49
      unsigned long dstLen = static_cast<unsigned long>(unpackedSampleDataSize);
shun-iwasawa 7bad49
      if (dstLen) {
shun-iwasawa 7bad49
        if (!tinyexr::DecompressZip(
shun-iwasawa 7bad49
                reinterpret_cast<unsigned char *>(&sample_data.at(0)), &dstLen,
shun-iwasawa 7bad49
                data_ptr + 28 + packedOffsetTableSize,
shun-iwasawa 7bad49
                static_cast<unsigned long>(packedSampleDataSize))) {
shun-iwasawa 7bad49
          return false;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
        assert(dstLen == static_cast<unsigned long>(unpackedSampleDataSize));
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // decode sample
shun-iwasawa 7bad49
    int sampleSize = -1;
shun-iwasawa 7bad49
    std::vector<int> channel_offset_list(static_cast<size_t>(num_channels));
shun-iwasawa 7bad49
    {
shun-iwasawa 7bad49
      int channel_offset = 0;
shun-iwasawa 7bad49
      for (size_t i = 0; i < static_cast<size_t>(num_channels); i++) {
shun-iwasawa 7bad49
        channel_offset_list[i] = channel_offset;
shun-iwasawa 7bad49
        if (channels[i].pixel_type == TINYEXR_PIXELTYPE_UINT) {  // UINT
shun-iwasawa 7bad49
          channel_offset += 4;
shun-iwasawa 7bad49
        } else if (channels[i].pixel_type == TINYEXR_PIXELTYPE_HALF) {  // half
shun-iwasawa 7bad49
          channel_offset += 2;
shun-iwasawa 7bad49
        } else if (channels[i].pixel_type ==
shun-iwasawa 7bad49
                   TINYEXR_PIXELTYPE_FLOAT) {  // float
shun-iwasawa 7bad49
          channel_offset += 4;
shun-iwasawa 7bad49
        } else {
shun-iwasawa 7bad49
          assert(0);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      sampleSize = channel_offset;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    assert(sampleSize >= 2);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    assert(static_cast<size_t>(
shun-iwasawa 7bad49
               pixelOffsetTable[static_cast<size_t>(data_width - 1)] *
shun-iwasawa 7bad49
               sampleSize) == sample_data.size());
shun-iwasawa 7bad49
    int samples_per_line = static_cast<int>(sample_data.size()) / sampleSize;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // Alloc memory
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    // pixel data is stored as image[channels][pixel_samples]
shun-iwasawa 7bad49
    //
shun-iwasawa 7bad49
    {
shun-iwasawa 7bad49
      tinyexr::tinyexr_uint64 data_offset = 0;
shun-iwasawa 7bad49
      for (size_t c = 0; c < static_cast<size_t>(num_channels); c++) {
shun-iwasawa 7bad49
        deep_image->image[c][y] = static_cast<float *>(
shun-iwasawa 7bad49
            malloc(sizeof(float) * static_cast<size_t>(samples_per_line)));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (channels[c].pixel_type == 0) {  // UINT
shun-iwasawa 7bad49
          for (size_t x = 0; x < static_cast<size_t>(samples_per_line); x++) {
shun-iwasawa 7bad49
            unsigned int ui;
shun-iwasawa 7bad49
            unsigned int *src_ptr = reinterpret_cast<unsigned int *>(
shun-iwasawa 7bad49
                &sample_data.at(size_t(data_offset) + x * sizeof(int)));
shun-iwasawa 7bad49
            tinyexr::cpy4(&ui, src_ptr);
shun-iwasawa 7bad49
            deep_image->image[c][y][x] = static_cast<float>(ui);  // @fixme
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
          data_offset +=
shun-iwasawa 7bad49
              sizeof(unsigned int) * static_cast<size_t>(samples_per_line);
shun-iwasawa 7bad49
        } else if (channels[c].pixel_type == 1) {  // half
shun-iwasawa 7bad49
          for (size_t x = 0; x < static_cast<size_t>(samples_per_line); x++) {
shun-iwasawa 7bad49
            tinyexr::FP16 f16;
shun-iwasawa 7bad49
            const unsigned short *src_ptr = reinterpret_cast<unsigned short *>(
shun-iwasawa 7bad49
                &sample_data.at(size_t(data_offset) + x * sizeof(short)));
shun-iwasawa 7bad49
            tinyexr::cpy2(&(f16.u), src_ptr);
shun-iwasawa 7bad49
            tinyexr::FP32 f32 = half_to_float(f16);
shun-iwasawa 7bad49
            deep_image->image[c][y][x] = f32.f;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
          data_offset += sizeof(short) * static_cast<size_t>(samples_per_line);
shun-iwasawa 7bad49
        } else {  // float
shun-iwasawa 7bad49
          for (size_t x = 0; x < static_cast<size_t>(samples_per_line); x++) {
shun-iwasawa 7bad49
            float f;
shun-iwasawa 7bad49
            const float *src_ptr = reinterpret_cast<float *>(
shun-iwasawa 7bad49
                &sample_data.at(size_t(data_offset) + x * sizeof(float)));
shun-iwasawa 7bad49
            tinyexr::cpy4(&f, src_ptr);
shun-iwasawa 7bad49
            deep_image->image[c][y][x] = f;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
          data_offset += sizeof(float) * static_cast<size_t>(samples_per_line);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }  // y
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  deep_image->width = data_width;
shun-iwasawa 7bad49
  deep_image->height = data_height;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  deep_image->channel_names = static_cast<const char **>(
shun-iwasawa 7bad49
      malloc(sizeof(const char *) * static_cast<size_t>(num_channels)));
shun-iwasawa 7bad49
  for (size_t c = 0; c < static_cast<size_t>(num_channels); c++) {
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
    deep_image->channel_names[c] = _strdup(channels[c].name.c_str());
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    deep_image->channel_names[c] = strdup(channels[c].name.c_str());
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  deep_image->num_channels = num_channels;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
void InitEXRImage(EXRImage *exr_image) {
shun-iwasawa 7bad49
  if (exr_image == NULL) {
shun-iwasawa 7bad49
    return;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  exr_image->width = 0;
shun-iwasawa 7bad49
  exr_image->height = 0;
shun-iwasawa 7bad49
  exr_image->num_channels = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  exr_image->images = NULL;
shun-iwasawa 7bad49
  exr_image->tiles = NULL;
shun-iwasawa 7bad49
  exr_image->next_level = NULL;
shun-iwasawa 7bad49
  exr_image->level_x = 0;
shun-iwasawa 7bad49
  exr_image->level_y = 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  exr_image->num_tiles = 0;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
void FreeEXRErrorMessage(const char *msg) {
shun-iwasawa 7bad49
  if (msg) {
shun-iwasawa 7bad49
    free(reinterpret_cast<void *>(const_cast<char *>(msg)));
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  return;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
void InitEXRHeader(EXRHeader *exr_header) {
shun-iwasawa 7bad49
  if (exr_header == NULL) {
shun-iwasawa 7bad49
    return;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  memset(exr_header, 0, sizeof(EXRHeader));
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int FreeEXRHeader(EXRHeader *exr_header) {
shun-iwasawa 7bad49
  if (exr_header == NULL) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->channels) {
shun-iwasawa 7bad49
    free(exr_header->channels);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->pixel_types) {
shun-iwasawa 7bad49
    free(exr_header->pixel_types);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->requested_pixel_types) {
shun-iwasawa 7bad49
    free(exr_header->requested_pixel_types);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < exr_header->num_custom_attributes; i++) {
shun-iwasawa 7bad49
    if (exr_header->custom_attributes[i].value) {
shun-iwasawa 7bad49
      free(exr_header->custom_attributes[i].value);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_header->custom_attributes) {
shun-iwasawa 7bad49
    free(exr_header->custom_attributes);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  EXRSetNameAttr(exr_header, NULL);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
void EXRSetNameAttr(EXRHeader* exr_header, const char* name) {
shun-iwasawa 7bad49
  if (exr_header == NULL) {
shun-iwasawa 7bad49
    return;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
  memset(exr_header->name, 0, 256);
shun-iwasawa 7bad49
  if (name != NULL) {
shun-iwasawa 7bad49
    size_t len = std::min(strlen(name), (size_t)255);
shun-iwasawa 7bad49
    if (len) {
shun-iwasawa 7bad49
      memcpy(exr_header->name, name, len);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int EXRNumLevels(const EXRImage* exr_image) {
shun-iwasawa 7bad49
  if (exr_image == NULL) return 0;
shun-iwasawa 7bad49
  if(exr_image->images) return 1; // scanlines
shun-iwasawa 7bad49
  int levels = 1;
shun-iwasawa 7bad49
  const EXRImage* level_image = exr_image;
shun-iwasawa 7bad49
  while((level_image = level_image->next_level)) ++levels;
shun-iwasawa 7bad49
  return levels;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int FreeEXRImage(EXRImage *exr_image) {
shun-iwasawa 7bad49
  if (exr_image == NULL) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_image->next_level) {
shun-iwasawa 7bad49
    FreeEXRImage(exr_image->next_level);
shun-iwasawa 7bad49
    delete exr_image->next_level;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (int i = 0; i < exr_image->num_channels; i++) {
shun-iwasawa 7bad49
    if (exr_image->images && exr_image->images[i]) {
shun-iwasawa 7bad49
      free(exr_image->images[i]);
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_image->images) {
shun-iwasawa 7bad49
    free(exr_image->images);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (exr_image->tiles) {
shun-iwasawa 7bad49
    for (int tid = 0; tid < exr_image->num_tiles; tid++) {
shun-iwasawa 7bad49
      for (int i = 0; i < exr_image->num_channels; i++) {
shun-iwasawa 7bad49
        if (exr_image->tiles[tid].images && exr_image->tiles[tid].images[i]) {
shun-iwasawa 7bad49
          free(exr_image->tiles[tid].images[i]);
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      if (exr_image->tiles[tid].images) {
shun-iwasawa 7bad49
        free(exr_image->tiles[tid].images);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    free(exr_image->tiles);
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int ParseEXRHeaderFromFile(EXRHeader *exr_header, const EXRVersion *exr_version,
shun-iwasawa 7bad49
                           const char *filename, const char **err) {
shun-iwasawa 7bad49
  if (exr_header == NULL || exr_version == NULL || filename == NULL) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Invalid argument for ParseEXRHeaderFromFile",
shun-iwasawa 7bad49
                             err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FILE *fp = NULL;
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
#if defined(_MSC_VER) || (defined(MINGW_HAS_SECURE_API) && MINGW_HAS_SECURE_API) // MSVC, MinGW GCC, or Clang.
shun-iwasawa 7bad49
  errno_t errcode =
shun-iwasawa 7bad49
      _wfopen_s(&fp, tinyexr::UTF8ToWchar(filename).c_str(), L"rb");
shun-iwasawa 7bad49
  if (errcode != 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read file " + std::string(filename), err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  // Unknown compiler or MinGW without MINGW_HAS_SECURE_API.
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  if (!fp) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read file " + std::string(filename), err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t filesize;
shun-iwasawa 7bad49
  // Compute size
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_END);
shun-iwasawa 7bad49
  filesize = static_cast<size_t>(ftell(fp));
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_SET);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned char> buf(filesize);  // @todo { use mmap }
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    size_t ret;
shun-iwasawa 7bad49
    ret = fread(&buf[0], 1, filesize, fp);
shun-iwasawa 7bad49
    assert(ret == filesize);
shun-iwasawa 7bad49
    fclose(fp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (ret != filesize) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("fread() error on " + std::string(filename),
shun-iwasawa 7bad49
                               err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_FILE;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return ParseEXRHeaderFromMemory(exr_header, exr_version, &buf.at(0), filesize,
shun-iwasawa 7bad49
                                  err);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int ParseEXRMultipartHeaderFromMemory(EXRHeader ***exr_headers,
shun-iwasawa 7bad49
                                      int *num_headers,
shun-iwasawa 7bad49
                                      const EXRVersion *exr_version,
shun-iwasawa 7bad49
                                      const unsigned char *memory, size_t size,
shun-iwasawa 7bad49
                                      const char **err) {
shun-iwasawa 7bad49
  if (memory == NULL || exr_headers == NULL || num_headers == NULL ||
shun-iwasawa 7bad49
      exr_version == NULL) {
shun-iwasawa 7bad49
    // Invalid argument
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage(
shun-iwasawa 7bad49
        "Invalid argument for ParseEXRMultipartHeaderFromMemory", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (size < tinyexr::kEXRVersionSize) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Data size too short", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const unsigned char *marker = memory + tinyexr::kEXRVersionSize;
shun-iwasawa 7bad49
  size_t marker_size = size - tinyexr::kEXRVersionSize;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<tinyexr::HeaderInfo> infos;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  for (;;) {
shun-iwasawa 7bad49
    tinyexr::HeaderInfo info;
shun-iwasawa 7bad49
    info.clear();
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    std::string err_str;
shun-iwasawa 7bad49
    bool empty_header = false;
shun-iwasawa 7bad49
    int ret = ParseEXRHeader(&info, &empty_header, exr_version, &err_str,
shun-iwasawa 7bad49
                             marker, marker_size);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage(err_str, err);
shun-iwasawa 7bad49
      return ret;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (empty_header) {
shun-iwasawa 7bad49
      marker += 1;  // skip '\0'
shun-iwasawa 7bad49
      break;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // `chunkCount` must exist in the header.
shun-iwasawa 7bad49
    if (info.chunk_count == 0) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage(
shun-iwasawa 7bad49
          "`chunkCount' attribute is not found in the header.", err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    infos.push_back(info);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // move to next header.
shun-iwasawa 7bad49
    marker += info.header_len;
shun-iwasawa 7bad49
    size -= info.header_len;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // allocate memory for EXRHeader and create array of EXRHeader pointers.
shun-iwasawa 7bad49
  (*exr_headers) =
shun-iwasawa 7bad49
      static_cast<EXRHeader **>(malloc(sizeof(EXRHeader *) * infos.size()));
shun-iwasawa 7bad49
  for (size_t i = 0; i < infos.size(); i++) {
shun-iwasawa 7bad49
    EXRHeader *exr_header = static_cast<EXRHeader *>(malloc(sizeof(EXRHeader)));
shun-iwasawa 7bad49
    memset(exr_header, 0, sizeof(EXRHeader));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    ConvertHeader(exr_header, infos[i]);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    exr_header->multipart = exr_version->multipart ? 1 : 0;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    (*exr_headers)[i] = exr_header;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  (*num_headers) = static_cast<int>(infos.size());
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int ParseEXRMultipartHeaderFromFile(EXRHeader ***exr_headers, int *num_headers,
shun-iwasawa 7bad49
                                    const EXRVersion *exr_version,
shun-iwasawa 7bad49
                                    const char *filename, const char **err) {
shun-iwasawa 7bad49
  if (exr_headers == NULL || num_headers == NULL || exr_version == NULL ||
shun-iwasawa 7bad49
      filename == NULL) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage(
shun-iwasawa 7bad49
        "Invalid argument for ParseEXRMultipartHeaderFromFile()", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FILE *fp = NULL;
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
#if defined(_MSC_VER) || (defined(MINGW_HAS_SECURE_API) && MINGW_HAS_SECURE_API) // MSVC, MinGW GCC, or Clang.
shun-iwasawa 7bad49
  errno_t errcode =
shun-iwasawa 7bad49
      _wfopen_s(&fp, tinyexr::UTF8ToWchar(filename).c_str(), L"rb");
shun-iwasawa 7bad49
  if (errcode != 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read file " + std::string(filename), err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  // Unknown compiler or MinGW without MINGW_HAS_SECURE_API.
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  if (!fp) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read file " + std::string(filename), err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t filesize;
shun-iwasawa 7bad49
  // Compute size
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_END);
shun-iwasawa 7bad49
  filesize = static_cast<size_t>(ftell(fp));
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_SET);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned char> buf(filesize);  // @todo { use mmap }
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    size_t ret;
shun-iwasawa 7bad49
    ret = fread(&buf[0], 1, filesize, fp);
shun-iwasawa 7bad49
    assert(ret == filesize);
shun-iwasawa 7bad49
    fclose(fp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (ret != filesize) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("`fread' error. file may be corrupted.", err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_FILE;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return ParseEXRMultipartHeaderFromMemory(
shun-iwasawa 7bad49
      exr_headers, num_headers, exr_version, &buf.at(0), filesize, err);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int ParseEXRVersionFromMemory(EXRVersion *version, const unsigned char *memory,
shun-iwasawa 7bad49
                              size_t size) {
shun-iwasawa 7bad49
  if (version == NULL || memory == NULL) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (size < tinyexr::kEXRVersionSize) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const unsigned char *marker = memory;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Header check.
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    const char header[] = {0x76, 0x2f, 0x31, 0x01};
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (memcmp(marker, header, 4) != 0) {
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_MAGIC_NUMBER;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    marker += 4;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  version->tiled = false;
shun-iwasawa 7bad49
  version->long_name = false;
shun-iwasawa 7bad49
  version->non_image = false;
shun-iwasawa 7bad49
  version->multipart = false;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Parse version header.
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    // must be 2
shun-iwasawa 7bad49
    if (marker[0] != 2) {
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_EXR_VERSION;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (version == NULL) {
shun-iwasawa 7bad49
      return TINYEXR_SUCCESS;  // May OK
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    version->version = 2;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (marker[1] & 0x2) {  // 9th bit
shun-iwasawa 7bad49
      version->tiled = true;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    if (marker[1] & 0x4) {  // 10th bit
shun-iwasawa 7bad49
      version->long_name = true;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    if (marker[1] & 0x8) {        // 11th bit
shun-iwasawa 7bad49
      version->non_image = true;  // (deep image)
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
    if (marker[1] & 0x10) {  // 12th bit
shun-iwasawa 7bad49
      version->multipart = true;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int ParseEXRVersionFromFile(EXRVersion *version, const char *filename) {
shun-iwasawa 7bad49
  if (filename == NULL) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FILE *fp = NULL;
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
#if defined(_MSC_VER) || (defined(MINGW_HAS_SECURE_API) && MINGW_HAS_SECURE_API) // MSVC, MinGW GCC, or Clang.
shun-iwasawa 7bad49
  errno_t err = _wfopen_s(&fp, tinyexr::UTF8ToWchar(filename).c_str(), L"rb");
shun-iwasawa 7bad49
  if (err != 0) {
shun-iwasawa 7bad49
    // TODO(syoyo): return wfopen_s erro code
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  // Unknown compiler or MinGW without MINGW_HAS_SECURE_API.
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  if (!fp) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t file_size;
shun-iwasawa 7bad49
  // Compute size
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_END);
shun-iwasawa 7bad49
  file_size = static_cast<size_t>(ftell(fp));
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_SET);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (file_size < tinyexr::kEXRVersionSize) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  unsigned char buf[tinyexr::kEXRVersionSize];
shun-iwasawa 7bad49
  size_t ret = fread(&buf[0], 1, tinyexr::kEXRVersionSize, fp);
shun-iwasawa 7bad49
  fclose(fp);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (ret != tinyexr::kEXRVersionSize) {
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return ParseEXRVersionFromMemory(version, buf, tinyexr::kEXRVersionSize);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int LoadEXRMultipartImageFromMemory(EXRImage *exr_images,
shun-iwasawa 7bad49
                                    const EXRHeader **exr_headers,
shun-iwasawa 7bad49
                                    unsigned int num_parts,
shun-iwasawa 7bad49
                                    const unsigned char *memory,
shun-iwasawa 7bad49
                                    const size_t size, const char **err) {
shun-iwasawa 7bad49
  if (exr_images == NULL || exr_headers == NULL || num_parts == 0 ||
shun-iwasawa 7bad49
      memory == NULL || (size <= tinyexr::kEXRVersionSize)) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage(
shun-iwasawa 7bad49
        "Invalid argument for LoadEXRMultipartImageFromMemory()", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // compute total header size.
shun-iwasawa 7bad49
  size_t total_header_size = 0;
shun-iwasawa 7bad49
  for (unsigned int i = 0; i < num_parts; i++) {
shun-iwasawa 7bad49
    if (exr_headers[i]->header_len == 0) {
shun-iwasawa 7bad49
      tinyexr::SetErrorMessage("EXRHeader variable is not initialized.", err);
shun-iwasawa 7bad49
      return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    total_header_size += exr_headers[i]->header_len;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  const char *marker = reinterpret_cast<const char *>(
shun-iwasawa 7bad49
      memory + total_header_size + 4 +
shun-iwasawa 7bad49
      4);  // +8 for magic number and version header.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  marker += 1;  // Skip empty header.
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // NOTE 1:
shun-iwasawa 7bad49
  //   In multipart image, There is 'part number' before chunk data.
shun-iwasawa 7bad49
  //   4 byte : part number
shun-iwasawa 7bad49
  //   4+     : chunk
shun-iwasawa 7bad49
  //
shun-iwasawa 7bad49
  // NOTE 2:
shun-iwasawa 7bad49
  //   EXR spec says 'part number' is 'unsigned long' but actually this is
shun-iwasawa 7bad49
  //   'unsigned int(4 bytes)' in OpenEXR implementation...
shun-iwasawa 7bad49
  //   http://www.openexr.com/openexrfilelayout.pdf
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Load chunk offset table.
shun-iwasawa 7bad49
  std::vector<tinyexr::OffsetData> chunk_offset_table_list;
shun-iwasawa 7bad49
  chunk_offset_table_list.reserve(num_parts);
shun-iwasawa 7bad49
  for (size_t i = 0; i < static_cast<size_t>(num_parts); i++) {
shun-iwasawa 7bad49
    chunk_offset_table_list.resize(chunk_offset_table_list.size() + 1);
shun-iwasawa 7bad49
    tinyexr::OffsetData& offset_data = chunk_offset_table_list.back();
shun-iwasawa 7bad49
    if (!exr_headers[i]->tiled || exr_headers[i]->tile_level_mode == TINYEXR_TILE_ONE_LEVEL) {
shun-iwasawa 7bad49
      tinyexr::InitSingleResolutionOffsets(offset_data, exr_headers[i]->chunk_count);
shun-iwasawa 7bad49
      std::vector<tinyexr::tinyexr_uint64>& offset_table = offset_data.offsets[0][0];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
      for (size_t c = 0; c < offset_table.size(); c++) {
shun-iwasawa 7bad49
        tinyexr::tinyexr_uint64 offset;
shun-iwasawa 7bad49
        memcpy(&offset, marker, 8);
shun-iwasawa 7bad49
        tinyexr::swap8(&offset);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        if (offset >= size) {
shun-iwasawa 7bad49
          tinyexr::SetErrorMessage("Invalid offset size in EXR header chunks.",
shun-iwasawa 7bad49
                                   err);
shun-iwasawa 7bad49
          return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
        offset_table[c] = offset + 4;  // +4 to skip 'part number'
shun-iwasawa 7bad49
        marker += 8;
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      {
shun-iwasawa 7bad49
        std::vector<int> num_x_tiles, num_y_tiles;
shun-iwasawa 7bad49
        tinyexr::PrecalculateTileInfo(num_x_tiles, num_y_tiles, exr_headers[i]);
shun-iwasawa 7bad49
        int num_blocks = InitTileOffsets(offset_data, exr_headers[i], num_x_tiles, num_y_tiles);
shun-iwasawa 7bad49
        if (num_blocks != exr_headers[i]->chunk_count) {
shun-iwasawa 7bad49
          tinyexr::SetErrorMessage("Invalid offset table size.", err);
shun-iwasawa 7bad49
          return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      for (unsigned int l = 0; l < offset_data.offsets.size(); ++l) {
shun-iwasawa 7bad49
        for (unsigned int dy = 0; dy < offset_data.offsets[l].size(); ++dy) {
shun-iwasawa 7bad49
          for (unsigned int dx = 0; dx < offset_data.offsets[l][dy].size(); ++dx) {
shun-iwasawa 7bad49
            tinyexr::tinyexr_uint64 offset;
shun-iwasawa 7bad49
            memcpy(&offset, marker, sizeof(tinyexr::tinyexr_uint64));
shun-iwasawa 7bad49
            tinyexr::swap8(&offset);
shun-iwasawa 7bad49
            if (offset >= size) {
shun-iwasawa 7bad49
              tinyexr::SetErrorMessage("Invalid offset size in EXR header chunks.",
shun-iwasawa 7bad49
                err);
shun-iwasawa 7bad49
              return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
            }
shun-iwasawa 7bad49
            offset_data.offsets[l][dy][dx] = offset + 4; // +4 to skip 'part number'
shun-iwasawa 7bad49
            marker += sizeof(tinyexr::tinyexr_uint64);  // = 8 
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  // Decode image.
shun-iwasawa 7bad49
  for (size_t i = 0; i < static_cast<size_t>(num_parts); i++) {
shun-iwasawa 7bad49
    tinyexr::OffsetData &offset_data = chunk_offset_table_list[i];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // First check 'part number' is identitical to 'i'
shun-iwasawa 7bad49
    for (unsigned int l = 0; l < offset_data.offsets.size(); ++l)
shun-iwasawa 7bad49
      for (unsigned int dy = 0; dy < offset_data.offsets[l].size(); ++dy)
shun-iwasawa 7bad49
        for (unsigned int dx = 0; dx < offset_data.offsets[l][dy].size(); ++dx) {
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          const unsigned char *part_number_addr =
shun-iwasawa 7bad49
              memory + offset_data.offsets[l][dy][dx] - 4;  // -4 to move to 'part number' field.
shun-iwasawa 7bad49
          unsigned int part_no;
shun-iwasawa 7bad49
          memcpy(&part_no, part_number_addr, sizeof(unsigned int));  // 4
shun-iwasawa 7bad49
          tinyexr::swap4(&part_no);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
          if (part_no != i) {
shun-iwasawa 7bad49
            tinyexr::SetErrorMessage("Invalid `part number' in EXR header chunks.",
shun-iwasawa 7bad49
                                     err);
shun-iwasawa 7bad49
            return TINYEXR_ERROR_INVALID_DATA;
shun-iwasawa 7bad49
          }
shun-iwasawa 7bad49
        }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    std::string e;
shun-iwasawa 7bad49
    int ret = tinyexr::DecodeChunk(&exr_images[i], exr_headers[i], offset_data,
shun-iwasawa 7bad49
                                   memory, size, &e);
shun-iwasawa 7bad49
    if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
      if (!e.empty()) {
shun-iwasawa 7bad49
        tinyexr::SetErrorMessage(e, err);
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
      return ret;
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return TINYEXR_SUCCESS;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int LoadEXRMultipartImageFromFile(EXRImage *exr_images,
shun-iwasawa 7bad49
                                  const EXRHeader **exr_headers,
shun-iwasawa 7bad49
                                  unsigned int num_parts, const char *filename,
shun-iwasawa 7bad49
                                  const char **err) {
shun-iwasawa 7bad49
  if (exr_images == NULL || exr_headers == NULL || num_parts == 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage(
shun-iwasawa 7bad49
        "Invalid argument for LoadEXRMultipartImageFromFile", err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  FILE *fp = NULL;
shun-iwasawa 7bad49
#ifdef _WIN32
shun-iwasawa 7bad49
#if defined(_MSC_VER) || (defined(MINGW_HAS_SECURE_API) && MINGW_HAS_SECURE_API) // MSVC, MinGW GCC, or Clang.
shun-iwasawa 7bad49
  errno_t errcode =
shun-iwasawa 7bad49
      _wfopen_s(&fp, tinyexr::UTF8ToWchar(filename).c_str(), L"rb");
shun-iwasawa 7bad49
  if (errcode != 0) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read file " + std::string(filename), err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  // Unknown compiler or MinGW without MINGW_HAS_SECURE_API.
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
  fp = fopen(filename, "rb");
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
  if (!fp) {
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage("Cannot read file " + std::string(filename), err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_CANT_OPEN_FILE;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  size_t filesize;
shun-iwasawa 7bad49
  // Compute size
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_END);
shun-iwasawa 7bad49
  filesize = static_cast<size_t>(ftell(fp));
shun-iwasawa 7bad49
  fseek(fp, 0, SEEK_SET);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<unsigned char> buf(filesize);  //  @todo { use mmap }
shun-iwasawa 7bad49
  {
shun-iwasawa 7bad49
    size_t ret;
shun-iwasawa 7bad49
    ret = fread(&buf[0], 1, filesize, fp);
shun-iwasawa 7bad49
    assert(ret == filesize);
shun-iwasawa 7bad49
    fclose(fp);
shun-iwasawa 7bad49
    (void)ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return LoadEXRMultipartImageFromMemory(exr_images, exr_headers, num_parts,
shun-iwasawa 7bad49
                                         &buf.at(0), filesize, err);
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
int SaveEXR(const float *data, int width, int height, int components,
shun-iwasawa 7bad49
            const int save_as_fp16, const char *outfilename, const char **err) {
shun-iwasawa 7bad49
  if ((components == 1) || components == 3 || components == 4) {
shun-iwasawa 7bad49
    // OK
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    std::stringstream ss;
shun-iwasawa 7bad49
    ss << "Unsupported component value : " << components << std::endl;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    tinyexr::SetErrorMessage(ss.str(), err);
shun-iwasawa 7bad49
    return TINYEXR_ERROR_INVALID_ARGUMENT;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  EXRHeader header;
shun-iwasawa 7bad49
  InitEXRHeader(&header);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if ((width < 16) && (height < 16)) {
shun-iwasawa 7bad49
    // No compression for small image.
shun-iwasawa 7bad49
    header.compression_type = TINYEXR_COMPRESSIONTYPE_NONE;
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    header.compression_type = TINYEXR_COMPRESSIONTYPE_ZIP;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  EXRImage image;
shun-iwasawa 7bad49
  InitEXRImage(&image);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  image.num_channels = components;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  std::vector<float> images[4];
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  if (components == 1) {
shun-iwasawa 7bad49
    images[0].resize(static_cast<size_t>(width * height));
shun-iwasawa 7bad49
    memcpy(images[0].data(), data, sizeof(float) * size_t(width * height));
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
    images[0].resize(static_cast<size_t>(width * height));
shun-iwasawa 7bad49
    images[1].resize(static_cast<size_t>(width * height));
shun-iwasawa 7bad49
    images[2].resize(static_cast<size_t>(width * height));
shun-iwasawa 7bad49
    images[3].resize(static_cast<size_t>(width * height));
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    // Split RGB(A)RGB(A)RGB(A)... into R, G and B(and A) layers
shun-iwasawa 7bad49
    for (size_t i = 0; i < static_cast<size_t>(width * height); i++) {
shun-iwasawa 7bad49
      images[0][i] = data[static_cast<size_t>(components) * i + 0];
shun-iwasawa 7bad49
      images[1][i] = data[static_cast<size_t>(components) * i + 1];
shun-iwasawa 7bad49
      images[2][i] = data[static_cast<size_t>(components) * i + 2];
shun-iwasawa 7bad49
      if (components == 4) {
shun-iwasawa 7bad49
        images[3][i] = data[static_cast<size_t>(components) * i + 3];
shun-iwasawa 7bad49
      }
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  float *image_ptr[4] = {0, 0, 0, 0};
shun-iwasawa 7bad49
  if (components == 4) {
shun-iwasawa 7bad49
    image_ptr[0] = &(images[3].at(0));  // A
shun-iwasawa 7bad49
    image_ptr[1] = &(images[2].at(0));  // B
shun-iwasawa 7bad49
    image_ptr[2] = &(images[1].at(0));  // G
shun-iwasawa 7bad49
    image_ptr[3] = &(images[0].at(0));  // R
shun-iwasawa 7bad49
  } else if (components == 3) {
shun-iwasawa 7bad49
    image_ptr[0] = &(images[2].at(0));  // B
shun-iwasawa 7bad49
    image_ptr[1] = &(images[1].at(0));  // G
shun-iwasawa 7bad49
    image_ptr[2] = &(images[0].at(0));  // R
shun-iwasawa 7bad49
  } else if (components == 1) {
shun-iwasawa 7bad49
    image_ptr[0] = &(images[0].at(0));  // A
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  image.images = reinterpret_cast<unsigned char **>(image_ptr);
shun-iwasawa 7bad49
  image.width = width;
shun-iwasawa 7bad49
  image.height = height;
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  header.num_channels = components;
shun-iwasawa 7bad49
  header.channels = static_cast<EXRChannelInfo *>(malloc(
shun-iwasawa 7bad49
      sizeof(EXRChannelInfo) * static_cast<size_t>(header.num_channels)));
shun-iwasawa 7bad49
  // Must be (A)BGR order, since most of EXR viewers expect this channel order.
shun-iwasawa 7bad49
  if (components == 4) {
shun-iwasawa 7bad49
#ifdef _MSC_VER
shun-iwasawa 7bad49
    strncpy_s(header.channels[0].name, "A", 255);
shun-iwasawa 7bad49
    strncpy_s(header.channels[1].name, "B", 255);
shun-iwasawa 7bad49
    strncpy_s(header.channels[2].name, "G", 255);
shun-iwasawa 7bad49
    strncpy_s(header.channels[3].name, "R", 255);
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    strncpy(header.channels[0].name, "A", 255);
shun-iwasawa 7bad49
    strncpy(header.channels[1].name, "B", 255);
shun-iwasawa 7bad49
    strncpy(header.channels[2].name, "G", 255);
shun-iwasawa 7bad49
    strncpy(header.channels[3].name, "R", 255);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    header.channels[0].name[strlen("A")] = '\0';
shun-iwasawa 7bad49
    header.channels[1].name[strlen("B")] = '\0';
shun-iwasawa 7bad49
    header.channels[2].name[strlen("G")] = '\0';
shun-iwasawa 7bad49
    header.channels[3].name[strlen("R")] = '\0';
shun-iwasawa 7bad49
  } else if (components == 3) {
shun-iwasawa 7bad49
#ifdef _MSC_VER
shun-iwasawa 7bad49
    strncpy_s(header.channels[0].name, "B", 255);
shun-iwasawa 7bad49
    strncpy_s(header.channels[1].name, "G", 255);
shun-iwasawa 7bad49
    strncpy_s(header.channels[2].name, "R", 255);
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    strncpy(header.channels[0].name, "B", 255);
shun-iwasawa 7bad49
    strncpy(header.channels[1].name, "G", 255);
shun-iwasawa 7bad49
    strncpy(header.channels[2].name, "R", 255);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    header.channels[0].name[strlen("B")] = '\0';
shun-iwasawa 7bad49
    header.channels[1].name[strlen("G")] = '\0';
shun-iwasawa 7bad49
    header.channels[2].name[strlen("R")] = '\0';
shun-iwasawa 7bad49
  } else {
shun-iwasawa 7bad49
#ifdef _MSC_VER
shun-iwasawa 7bad49
    strncpy_s(header.channels[0].name, "A", 255);
shun-iwasawa 7bad49
#else
shun-iwasawa 7bad49
    strncpy(header.channels[0].name, "A", 255);
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
    header.channels[0].name[strlen("A")] = '\0';
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  header.pixel_types = static_cast<int *>(
shun-iwasawa 7bad49
      malloc(sizeof(int) * static_cast<size_t>(header.num_channels)));
shun-iwasawa 7bad49
  header.requested_pixel_types = static_cast<int *>(
shun-iwasawa 7bad49
      malloc(sizeof(int) * static_cast<size_t>(header.num_channels)));
shun-iwasawa 7bad49
  for (int i = 0; i < header.num_channels; i++) {
shun-iwasawa 7bad49
    header.pixel_types[i] =
shun-iwasawa 7bad49
        TINYEXR_PIXELTYPE_FLOAT;  // pixel type of input image
shun-iwasawa 7bad49
shun-iwasawa 7bad49
    if (save_as_fp16 > 0) {
shun-iwasawa 7bad49
      header.requested_pixel_types[i] =
shun-iwasawa 7bad49
          TINYEXR_PIXELTYPE_HALF;  // save with half(fp16) pixel format
shun-iwasawa 7bad49
    } else {
shun-iwasawa 7bad49
      header.requested_pixel_types[i] =
shun-iwasawa 7bad49
          TINYEXR_PIXELTYPE_FLOAT;  // save with float(fp32) pixel format(i.e.
shun-iwasawa 7bad49
                                    // no precision reduction)
shun-iwasawa 7bad49
    }
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  int ret = SaveEXRImageToFile(&image, &header, outfilename, err);
shun-iwasawa 7bad49
  if (ret != TINYEXR_SUCCESS) {
shun-iwasawa 7bad49
    return ret;
shun-iwasawa 7bad49
  }
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  free(header.channels);
shun-iwasawa 7bad49
  free(header.pixel_types);
shun-iwasawa 7bad49
  free(header.requested_pixel_types);
shun-iwasawa 7bad49
shun-iwasawa 7bad49
  return ret;
shun-iwasawa 7bad49
}
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#ifdef __clang__
shun-iwasawa 7bad49
// zero-as-null-ppinter-constant
shun-iwasawa 7bad49
#pragma clang diagnostic pop
shun-iwasawa 7bad49
#endif
shun-iwasawa 7bad49
shun-iwasawa 7bad49
#endif  // TINYEXR_IMPLEMENTATION_DEFINED
shun-iwasawa 7bad49
#endif  // TINYEXR_IMPLEMENTATION