Fix tiled EXR crash, update tinyexr to head to fix corrupted uncompressed EXR loading

(cherry picked from commit 4b7885fb1e)
This commit is contained in:
elasota 2018-08-27 01:52:13 -04:00 committed by Rémi Verschelde
parent c9fdaa412a
commit 0921e68913
3 changed files with 876 additions and 489 deletions

View File

@ -129,15 +129,45 @@ Error ImageLoaderTinyEXR::load_image(Ref<Image> p_image, FileAccess *f, bool p_f
PoolVector<uint8_t> imgdata; PoolVector<uint8_t> imgdata;
Image::Format format; Image::Format format;
int output_channels = 0;
if (idxA > 0) { if (idxA > 0) {
imgdata.resize(exr_image.width * exr_image.height * 8); //RGBA16 imgdata.resize(exr_image.width * exr_image.height * 8); //RGBA16
format = Image::FORMAT_RGBAH; format = Image::FORMAT_RGBAH;
output_channels = 4;
} else { } else {
imgdata.resize(exr_image.width * exr_image.height * 6); //RGB16 imgdata.resize(exr_image.width * exr_image.height * 6); //RGB16
format = Image::FORMAT_RGBH; format = Image::FORMAT_RGBH;
output_channels = 3;
}
EXRTile single_image_tile;
int num_tiles;
int tile_width = 0;
int tile_height = 0;
const EXRTile *exr_tiles;
if (!exr_header.tiled) {
single_image_tile.images = exr_image.images;
single_image_tile.width = exr_image.width;
single_image_tile.height = exr_image.height;
single_image_tile.level_x = exr_image.width;
single_image_tile.level_y = exr_image.height;
single_image_tile.offset_x = 0;
single_image_tile.offset_y = 0;
exr_tiles = &single_image_tile;
num_tiles = 1;
tile_width = exr_image.width;
tile_height = exr_image.height;
} else {
tile_width = exr_header.tile_size_x;
tile_height = exr_header.tile_size_y;
num_tiles = exr_image.num_tiles;
exr_tiles = exr_image.tiles;
} }
{ {
@ -145,22 +175,51 @@ Error ImageLoaderTinyEXR::load_image(Ref<Image> p_image, FileAccess *f, bool p_f
uint16_t *iw = (uint16_t *)wd.ptr(); uint16_t *iw = (uint16_t *)wd.ptr();
// Assume `out_rgba` have enough memory allocated. // Assume `out_rgba` have enough memory allocated.
for (int i = 0; i < exr_image.width * exr_image.height; i++) { for (int tile_index = 0; tile_index < num_tiles; tile_index++) {
Color color( const EXRTile &tile = exr_tiles[tile_index];
reinterpret_cast<float **>(exr_image.images)[idxR][i],
reinterpret_cast<float **>(exr_image.images)[idxG][i], int tw = tile.width;
reinterpret_cast<float **>(exr_image.images)[idxB][i]); int th = tile.height;
const float *r_channel_start = reinterpret_cast<const float *>(tile.images[idxR]);
const float *g_channel_start = reinterpret_cast<const float *>(tile.images[idxG]);
const float *b_channel_start = reinterpret_cast<const float *>(tile.images[idxB]);
const float *a_channel_start = NULL;
if (idxA > 0) {
a_channel_start = reinterpret_cast<const float *>(tile.images[idxA]);
}
uint16_t *first_row_w = iw + (tile.offset_y * tile_height * exr_image.width + tile.offset_x * tile_width) * output_channels;
for (int y = 0; y < th; y++) {
const float *r_channel = r_channel_start + y * tile_width;
const float *g_channel = g_channel_start + y * tile_width;
const float *b_channel = b_channel_start + y * tile_width;
const float *a_channel = NULL;
if (a_channel_start) {
a_channel = a_channel_start + y * tile_width;
}
uint16_t *row_w = first_row_w + (y * exr_image.width * output_channels);
for (int x = 0; x < tw; x++) {
Color color(*r_channel++, *g_channel++, *b_channel++);
if (p_force_linear) if (p_force_linear)
color = color.to_linear(); color = color.to_linear();
*iw++ = Math::make_half_float(color.r); *row_w++ = Math::make_half_float(color.r);
*iw++ = Math::make_half_float(color.g); *row_w++ = Math::make_half_float(color.g);
*iw++ = Math::make_half_float(color.b); *row_w++ = Math::make_half_float(color.b);
if (idxA > 0) { if (idxA > 0) {
*iw++ = Math::make_half_float(reinterpret_cast<float **>(exr_image.images)[idxA][i]); *row_w++ = Math::make_half_float(*a_channel++);
}
}
} }
} }
} }

View File

@ -450,7 +450,7 @@ are provided to reapply them.
## tinyexr ## tinyexr
- Upstream: https://github.com/syoyo/tinyexr - Upstream: https://github.com/syoyo/tinyexr
- Version: git (e385dad, 2018) - Version: git (2d5375f, 2018)
- License: BSD-3-Clause - License: BSD-3-Clause
Files extracted from upstream source: Files extracted from upstream source:

File diff suppressed because it is too large Load Diff