diff options
Diffstat (limited to 'source/blender/imbuf/intern/openexr/openexr_api.cpp')
| -rw-r--r-- | source/blender/imbuf/intern/openexr/openexr_api.cpp | 733 |
1 files changed, 371 insertions, 362 deletions
diff --git a/source/blender/imbuf/intern/openexr/openexr_api.cpp b/source/blender/imbuf/intern/openexr/openexr_api.cpp index baea18d4898..18957cb8260 100644 --- a/source/blender/imbuf/intern/openexr/openexr_api.cpp +++ b/source/blender/imbuf/intern/openexr/openexr_api.cpp @@ -4,7 +4,7 @@ * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License * as published by the Free Software Foundation; either version 2 - * of the License, or (at your option) any later version. + * of the License, or (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of @@ -64,7 +64,7 @@ _CRTIMP void __cdecl _invalid_parameter_noinfo(void) #include <iostream> -#if defined (_WIN32) && !defined(FREE_WINDOWS) +#if defined(_WIN32) && !defined(FREE_WINDOWS) #include <half.h> #include <IlmImf/ImfVersion.h> #include <IlmImf/ImfArray.h> @@ -95,19 +95,21 @@ _CRTIMP void __cdecl _invalid_parameter_noinfo(void) using namespace Imf; using namespace Imath; -class Mem_IStream: public Imf::IStream +class Mem_IStream : public Imf::IStream { public: - - Mem_IStream (unsigned char *exrbuf, size_t exrsize): - IStream("dummy"), _exrpos (0), _exrsize(exrsize) { _exrbuf = exrbuf; } - - virtual bool read (char c[], int n); - virtual Int64 tellg (); - virtual void seekg (Int64 pos); - virtual void clear (); + + Mem_IStream (unsigned char *exrbuf, size_t exrsize) : + IStream("dummy"), _exrpos(0), _exrsize(exrsize) { + _exrbuf = exrbuf; + } + + virtual bool read(char c[], int n); + virtual Int64 tellg(); + virtual void seekg(Int64 pos); + virtual void clear(); //virtual ~Mem_IStream() {}; // unused - + private: Int64 _exrpos; @@ -115,7 +117,7 @@ private: unsigned char *_exrbuf; }; -bool Mem_IStream::read (char c[], int n) +bool Mem_IStream::read(char c[], int n) { if (n + _exrpos <= _exrsize) { memcpy(c, (void *)(&_exrbuf[_exrpos]), n); @@ -126,22 +128,21 @@ bool Mem_IStream::read (char c[], int n) return false; } -Int64 Mem_IStream::tellg () +Int64 Mem_IStream::tellg() { return _exrpos; } -void Mem_IStream::seekg (Int64 pos) +void Mem_IStream::seekg(Int64 pos) { _exrpos = pos; } -void Mem_IStream::clear () -{ +void Mem_IStream::clear() +{ } -struct _RGBAZ -{ +struct _RGBAZ { half r; half g; half b; @@ -153,10 +154,10 @@ typedef struct _RGBAZ RGBAZ; extern "C" { - + int imb_is_a_openexr(unsigned char *mem) { - return Imf::isImfMagic ((const char *)mem); + return Imf::isImfMagic((const char *)mem); } static void openexr_header_compression(Header *header, int compression) @@ -179,69 +180,70 @@ static void openexr_header_compression(Header *header, int compression) break; default: header->compression() = ZIP_COMPRESSION; - break; + break; } } static void openexr_header_metadata(Header *header, struct ImBuf *ibuf) { - ImMetaData* info; + ImMetaData *info; - for (info= ibuf->metadata; info; info= info->next) + for (info = ibuf->metadata; info; info = info->next) header->insert(info->key, StringAttribute(info->value)); } static int imb_save_openexr_half(struct ImBuf *ibuf, const char *name, int flags) { - int channels = ibuf->channels; - int width = ibuf->x; - int height = ibuf->y; - int write_zbuf = (flags & IB_zbuffloat) && ibuf->zbuf_float != NULL; // summarize - + const int channels = ibuf->channels; + const int is_alpha = (channels >= 4) && (ibuf->planes == 32); + const int is_zbuf = (flags & IB_zbuffloat) && ibuf->zbuf_float != NULL; /* summarize */ + const int width = ibuf->x; + const int height = ibuf->y; + try { - Header header (width, height); - + Header header(width, height); + openexr_header_compression(&header, ibuf->ftype & OPENEXR_COMPRESS); openexr_header_metadata(&header, ibuf); - - header.channels().insert ("R", Channel (HALF)); - header.channels().insert ("G", Channel (HALF)); - header.channels().insert ("B", Channel (HALF)); - if (ibuf->planes==32 && channels >= 4) - header.channels().insert ("A", Channel (HALF)); - if (write_zbuf) // z we do as float always - header.channels().insert ("Z", Channel (Imf::FLOAT)); - - FrameBuffer frameBuffer; - OutputFile *file = new OutputFile(name, header); - + + header.channels().insert("R", Channel(HALF)); + header.channels().insert("G", Channel(HALF)); + header.channels().insert("B", Channel(HALF)); + if (is_alpha) + header.channels().insert("A", Channel(HALF)); + if (is_zbuf) // z we do as float always + header.channels().insert("Z", Channel(Imf::FLOAT)); + + FrameBuffer frameBuffer; + OutputFile *file = new OutputFile(name, header); + /* we store first everything in half array */ RGBAZ *pixels = new RGBAZ[height * width]; RGBAZ *to = pixels; - int xstride= sizeof (RGBAZ); - int ystride= xstride*width; + int xstride = sizeof(RGBAZ); + int ystride = xstride * width; /* indicate used buffers */ - frameBuffer.insert ("R", Slice (HALF, (char *) &pixels[0].r, xstride, ystride)); - frameBuffer.insert ("G", Slice (HALF, (char *) &pixels[0].g, xstride, ystride)); - frameBuffer.insert ("B", Slice (HALF, (char *) &pixels[0].b, xstride, ystride)); - if (ibuf->planes==32 && channels >= 4) - frameBuffer.insert ("A", Slice (HALF, (char *) &pixels[0].a, xstride, ystride)); - if (write_zbuf) - frameBuffer.insert ("Z", Slice (Imf::FLOAT, (char *)(ibuf->zbuf_float + (height-1)*width), - sizeof(float), sizeof(float) * -width)); + frameBuffer.insert("R", Slice(HALF, (char *) &pixels[0].r, xstride, ystride)); + frameBuffer.insert("G", Slice(HALF, (char *) &pixels[0].g, xstride, ystride)); + frameBuffer.insert("B", Slice(HALF, (char *) &pixels[0].b, xstride, ystride)); + if (is_alpha) + frameBuffer.insert("A", Slice(HALF, (char *) &pixels[0].a, xstride, ystride)); + if (is_zbuf) + frameBuffer.insert("Z", Slice(Imf::FLOAT, (char *)(ibuf->zbuf_float + (height - 1) * width), + sizeof(float), sizeof(float) * -width)); if (ibuf->rect_float) { float *from; - for (int i = ibuf->y-1; i >= 0; i--) { - from= ibuf->rect_float + channels*i*width; + for (int i = ibuf->y - 1; i >= 0; i--) { + from = ibuf->rect_float + channels * i * width; for (int j = ibuf->x; j > 0; j--) { to->r = from[0]; to->g = from[1]; to->b = from[2]; - to->a = (channels >= 4)? from[3]: 1.0f; + to->a = (channels >= 4) ? from[3] : 1.0f; to++; from += 4; } } @@ -250,105 +252,106 @@ static int imb_save_openexr_half(struct ImBuf *ibuf, const char *name, int flags unsigned char *from; if (ibuf->profile == IB_PROFILE_LINEAR_RGB) { - for (int i = ibuf->y-1; i >= 0; i--) { - from= (unsigned char *)ibuf->rect + channels*i*width; + for (int i = ibuf->y - 1; i >= 0; i--) { + from = (unsigned char *)ibuf->rect + channels * i * width; for (int j = ibuf->x; j > 0; j--) { - to->r = (float)(from[0])/255.0; - to->g = (float)(from[1])/255.0; - to->b = (float)(from[2])/255.0; - to->a = (float)(channels >= 4) ? from[3]/255.0 : 1.0f; + to->r = (float)(from[0]) / 255.0; + to->g = (float)(from[1]) / 255.0; + to->b = (float)(from[2]) / 255.0; + to->a = (float)(channels >= 4) ? from[3] / 255.0 : 1.0f; to++; from += 4; } } } else { - for (int i = ibuf->y-1; i >= 0; i--) { - from= (unsigned char *)ibuf->rect + channels*i*width; + for (int i = ibuf->y - 1; i >= 0; i--) { + from = (unsigned char *)ibuf->rect + channels * i * width; for (int j = ibuf->x; j > 0; j--) { to->r = srgb_to_linearrgb((float)from[0] / 255.0); to->g = srgb_to_linearrgb((float)from[1] / 255.0); to->b = srgb_to_linearrgb((float)from[2] / 255.0); - to->a = channels >= 4 ? (float)from[3]/255.0 : 1.0f; + to->a = channels >= 4 ? (float)from[3] / 255.0 : 1.0f; to++; from += 4; } } } } - + // printf("OpenEXR-save: Writing OpenEXR file of height %d.\n", height); - - file->setFrameBuffer (frameBuffer); - file->writePixels (height); + + file->setFrameBuffer(frameBuffer); + file->writePixels(height); delete file; - delete [] pixels; + delete[] pixels; } catch (const std::exception &exc) { printf("OpenEXR-save: ERROR: %s\n", exc.what()); if (ibuf) IMB_freeImBuf(ibuf); - + return (0); } - + return (1); } static int imb_save_openexr_float(struct ImBuf *ibuf, const char *name, int flags) { - int channels = ibuf->channels; - int width = ibuf->x; - int height = ibuf->y; - int write_zbuf = (flags & IB_zbuffloat) && ibuf->zbuf_float != NULL; // summarize + const int channels = ibuf->channels; + const int is_alpha = (channels >= 4) && (ibuf->planes == 32); + const int is_zbuf = (flags & IB_zbuffloat) && ibuf->zbuf_float != NULL; /* summarize */ + const int width = ibuf->x; + const int height = ibuf->y; try { - Header header (width, height); - + Header header(width, height); + openexr_header_compression(&header, ibuf->ftype & OPENEXR_COMPRESS); openexr_header_metadata(&header, ibuf); - - header.channels().insert ("R", Channel (Imf::FLOAT)); - header.channels().insert ("G", Channel (Imf::FLOAT)); - header.channels().insert ("B", Channel (Imf::FLOAT)); - if (ibuf->planes==32 && channels >= 4) - header.channels().insert ("A", Channel (Imf::FLOAT)); - if (write_zbuf) - header.channels().insert ("Z", Channel (Imf::FLOAT)); - - FrameBuffer frameBuffer; - OutputFile *file = new OutputFile(name, header); + + header.channels().insert("R", Channel(Imf::FLOAT)); + header.channels().insert("G", Channel(Imf::FLOAT)); + header.channels().insert("B", Channel(Imf::FLOAT)); + if (is_alpha) + header.channels().insert("A", Channel(Imf::FLOAT)); + if (is_zbuf) + header.channels().insert("Z", Channel(Imf::FLOAT)); + + FrameBuffer frameBuffer; + OutputFile *file = new OutputFile(name, header); int xstride = sizeof(float) * channels; - int ystride = - xstride*width; + int ystride = -xstride * width; float *rect[4] = {NULL, NULL, NULL, NULL}; /* last scanline, stride negative */ - rect[0]= ibuf->rect_float + channels*(height-1)*width; - rect[1]= rect[0]+1; - rect[2]= rect[0]+2; - rect[3]= (channels >= 4)? rect[0]+3:rect[0]; /* red as alpha, is this needed since alpha isn't written? */ - - frameBuffer.insert ("R", Slice (Imf::FLOAT, (char *)rect[0], xstride, ystride)); - frameBuffer.insert ("G", Slice (Imf::FLOAT, (char *)rect[1], xstride, ystride)); - frameBuffer.insert ("B", Slice (Imf::FLOAT, (char *)rect[2], xstride, ystride)); - if (ibuf->planes==32 && channels >= 4) - frameBuffer.insert ("A", Slice (Imf::FLOAT, (char *)rect[3], xstride, ystride)); - if (write_zbuf) - frameBuffer.insert ("Z", Slice (Imf::FLOAT, (char *) (ibuf->zbuf_float + (height-1)*width), - sizeof(float), sizeof(float) * -width)); - file->setFrameBuffer (frameBuffer); - file->writePixels (height); + rect[0] = ibuf->rect_float + channels * (height - 1) * width; + rect[1] = rect[0] + 1; + rect[2] = rect[0] + 2; + rect[3] = (channels >= 4) ? rect[0] + 3 : rect[0]; /* red as alpha, is this needed since alpha isn't written? */ + + frameBuffer.insert("R", Slice(Imf::FLOAT, (char *)rect[0], xstride, ystride)); + frameBuffer.insert("G", Slice(Imf::FLOAT, (char *)rect[1], xstride, ystride)); + frameBuffer.insert("B", Slice(Imf::FLOAT, (char *)rect[2], xstride, ystride)); + if (is_alpha) + frameBuffer.insert("A", Slice(Imf::FLOAT, (char *)rect[3], xstride, ystride)); + if (is_zbuf) + frameBuffer.insert("Z", Slice(Imf::FLOAT, (char *) (ibuf->zbuf_float + (height - 1) * width), + sizeof(float), sizeof(float) * -width)); + file->setFrameBuffer(frameBuffer); + file->writePixels(height); delete file; } catch (const std::exception &exc) { printf("OpenEXR-save: ERROR: %s\n", exc.what()); if (ibuf) IMB_freeImBuf(ibuf); - + return (0); } - + return (1); // printf("OpenEXR-save: Done.\n"); } @@ -359,15 +362,15 @@ int imb_save_openexr(struct ImBuf *ibuf, const char *name, int flags) if (flags & IB_mem) { printf("OpenEXR-save: Create EXR in memory CURRENTLY NOT SUPPORTED !\n"); imb_addencodedbufferImBuf(ibuf); - ibuf->encodedsize = 0; + ibuf->encodedsize = 0; return(0); - } - - if (ibuf->ftype & OPENEXR_HALF) + } + + if (ibuf->ftype & OPENEXR_HALF) return imb_save_openexr_half(ibuf, name, flags); else { /* when no float rect, we save as half (16 bits is sufficient) */ - if (ibuf->rect_float==NULL) + if (ibuf->rect_float == NULL) return imb_save_openexr_half(ibuf, name, flags); else return imb_save_openexr_float(ibuf, name, flags); @@ -383,30 +386,30 @@ int imb_save_openexr(struct ImBuf *ibuf, const char *name, int flags) * - separated with a dot: the Layer name (like "Lamp1" or "Walls" or "Characters") */ -static ListBase exrhandles= {NULL, NULL}; +static ListBase exrhandles = {NULL, NULL}; typedef struct ExrHandle { struct ExrHandle *next, *prev; - + InputFile *ifile; TiledOutputFile *tofile; OutputFile *ofile; int tilex, tiley; int width, height; int mipmap; - - ListBase channels; /* flattened out, ExrChannel */ - ListBase layers; /* hierarchical, pointing in end to ExrChannel */ + + ListBase channels; /* flattened out, ExrChannel */ + ListBase layers; /* hierarchical, pointing in end to ExrChannel */ } ExrHandle; /* flattened out channel */ typedef struct ExrChannel { struct ExrChannel *next, *prev; - - char name[EXR_TOT_MAXNAME+1]; /* full name of layer+pass */ - int xstride, ystride; /* step to next pixel, to next scanline */ - float *rect; /* first pointer to write in */ - char chan_id; /* quick lookup of channel char */ + + char name[EXR_TOT_MAXNAME + 1]; /* full name of layer+pass */ + int xstride, ystride; /* step to next pixel, to next scanline */ + float *rect; /* first pointer to write in */ + char chan_id; /* quick lookup of channel char */ } ExrChannel; @@ -422,7 +425,7 @@ typedef struct ExrPass { typedef struct ExrLayer { struct ExrLayer *next, *prev; - char name[EXR_LAY_MAXNAME+1]; + char name[EXR_LAY_MAXNAME + 1]; ListBase passes; } ExrLayer; @@ -430,7 +433,7 @@ typedef struct ExrLayer { void *IMB_exr_get_handle(void) { - ExrHandle *data= (ExrHandle *)MEM_callocN(sizeof(ExrHandle), "exr handle"); + ExrHandle *data = (ExrHandle *)MEM_callocN(sizeof(ExrHandle), "exr handle"); BLI_addtail(&exrhandles, data); return data; } @@ -439,25 +442,26 @@ void *IMB_exr_get_handle(void) /* xstride, ystride and rect can be done in set_channel too, for tile writing */ void IMB_exr_add_channel(void *handle, const char *layname, const char *passname, int xstride, int ystride, float *rect) { - ExrHandle *data= (ExrHandle *)handle; + ExrHandle *data = (ExrHandle *)handle; ExrChannel *echan; - - echan= (ExrChannel *)MEM_callocN(sizeof(ExrChannel), "exr tile channel"); - + + echan = (ExrChannel *)MEM_callocN(sizeof(ExrChannel), "exr tile channel"); + if (layname) { - char lay[EXR_LAY_MAXNAME+1], pass[EXR_PASS_MAXNAME+1]; + char lay[EXR_LAY_MAXNAME + 1], pass[EXR_PASS_MAXNAME + 1]; BLI_strncpy(lay, layname, EXR_LAY_MAXNAME); BLI_strncpy(pass, passname, EXR_PASS_MAXNAME); - sprintf(echan->name, "%s.%s", lay, pass); + BLI_snprintf(echan->name, sizeof(echan->name), "%s.%s", lay, pass); } - else - BLI_strncpy(echan->name, passname, EXR_TOT_MAXNAME-1); - - echan->xstride= xstride; - echan->ystride= ystride; - echan->rect= rect; - + else { + BLI_strncpy(echan->name, passname, EXR_TOT_MAXNAME - 1); + } + + echan->xstride = xstride; + echan->ystride = ystride; + echan->rect = rect; + // printf("added channel %s\n", echan->name); BLI_addtail(&data->channels, echan); } @@ -465,21 +469,21 @@ void IMB_exr_add_channel(void *handle, const char *layname, const char *passname /* only used for writing temp. render results (not image files) */ int IMB_exr_begin_write(void *handle, const char *filename, int width, int height, int compress) { - ExrHandle *data= (ExrHandle *)handle; - Header header (width, height); + ExrHandle *data = (ExrHandle *)handle; + Header header(width, height); ExrChannel *echan; - - data->width= width; - data->height= height; - - for (echan= (ExrChannel *)data->channels.first; echan; echan= echan->next) - header.channels().insert (echan->name, Channel (Imf::FLOAT)); - + + data->width = width; + data->height = height; + + for (echan = (ExrChannel *)data->channels.first; echan; echan = echan->next) + header.channels().insert(echan->name, Channel(Imf::FLOAT)); + openexr_header_compression(&header, compress); // openexr_header_metadata(&header, ibuf); // no imbuf. cant write /* header.lineOrder() = DECREASING_Y; this crashes in windows for file read! */ - - header.insert ("BlenderMultiChannel", StringAttribute ("Blender V2.55.1 and newer")); + + header.insert("BlenderMultiChannel", StringAttribute("Blender V2.55.1 and newer")); /* avoid crash/abort when we don't have permission to write here */ try { @@ -495,45 +499,45 @@ int IMB_exr_begin_write(void *handle, const char *filename, int width, int heigh void IMB_exrtile_begin_write(void *handle, const char *filename, int mipmap, int width, int height, int tilex, int tiley) { - ExrHandle *data= (ExrHandle *)handle; - Header header (width, height); + ExrHandle *data = (ExrHandle *)handle; + Header header(width, height); ExrChannel *echan; - - data->tilex= tilex; - data->tiley= tiley; - data->width= width; - data->height= height; - data->mipmap= mipmap; - - for (echan= (ExrChannel *)data->channels.first; echan; echan= echan->next) - header.channels().insert (echan->name, Channel (Imf::FLOAT)); - - header.setTileDescription (TileDescription (tilex, tiley, (mipmap)? MIPMAP_LEVELS: ONE_LEVEL)); + + data->tilex = tilex; + data->tiley = tiley; + data->width = width; + data->height = height; + data->mipmap = mipmap; + + for (echan = (ExrChannel *)data->channels.first; echan; echan = echan->next) + header.channels().insert(echan->name, Channel(Imf::FLOAT)); + + header.setTileDescription(TileDescription(tilex, tiley, (mipmap) ? MIPMAP_LEVELS : ONE_LEVEL)); header.lineOrder() = RANDOM_Y; header.compression() = RLE_COMPRESSION; - - header.insert ("BlenderMultiChannel", StringAttribute ("Blender V2.43")); - + + header.insert("BlenderMultiChannel", StringAttribute("Blender V2.43")); + data->tofile = new TiledOutputFile(filename, header); } /* read from file */ int IMB_exr_begin_read(void *handle, const char *filename, int *width, int *height) { - ExrHandle *data= (ExrHandle *)handle; - - if (BLI_exists(filename) && BLI_file_size(filename)>32) { /* 32 is arbitrary, but zero length files crashes exr */ + ExrHandle *data = (ExrHandle *)handle; + + if (BLI_exists(filename) && BLI_file_size(filename) > 32) { /* 32 is arbitrary, but zero length files crashes exr */ data->ifile = new InputFile(filename); if (data->ifile) { Box2i dw = data->ifile->header().dataWindow(); - data->width= *width = dw.max.x - dw.min.x + 1; - data->height= *height = dw.max.y - dw.min.y + 1; - + data->width = *width = dw.max.x - dw.min.x + 1; + data->height = *height = dw.max.y - dw.min.y + 1; + const ChannelList &channels = data->ifile->header().channels(); - + for (ChannelList::ConstIterator i = channels.begin(); i != channels.end(); ++i) IMB_exr_add_channel(data, NULL, i.name(), 0, 0, NULL); - + return 1; } } @@ -543,26 +547,26 @@ int IMB_exr_begin_read(void *handle, const char *filename, int *width, int *heig /* still clumsy name handling, layers/channels can be ordered as list in list later */ void IMB_exr_set_channel(void *handle, const char *layname, const char *passname, int xstride, int ystride, float *rect) { - ExrHandle *data= (ExrHandle *)handle; + ExrHandle *data = (ExrHandle *)handle; ExrChannel *echan; char name[EXR_TOT_MAXNAME + 1]; - + if (layname) { - char lay[EXR_LAY_MAXNAME+1], pass[EXR_PASS_MAXNAME+1]; + char lay[EXR_LAY_MAXNAME + 1], pass[EXR_PASS_MAXNAME + 1]; BLI_strncpy(lay, layname, EXR_LAY_MAXNAME); BLI_strncpy(pass, passname, EXR_PASS_MAXNAME); - - sprintf(name, "%s.%s", lay, pass); + + BLI_snprintf(name, sizeof(name), "%s.%s", lay, pass); } else - BLI_strncpy(name, passname, EXR_TOT_MAXNAME-1); + BLI_strncpy(name, passname, EXR_TOT_MAXNAME - 1); - echan= (ExrChannel *)BLI_findstring(&data->channels, name, offsetof(ExrChannel, name)); + echan = (ExrChannel *)BLI_findstring(&data->channels, name, offsetof(ExrChannel, name)); if (echan) { - echan->xstride= xstride; - echan->ystride= ystride; - echan->rect= rect; + echan->xstride = xstride; + echan->ystride = ystride; + echan->rect = rect; } else printf("IMB_exrtile_set_channel error %s\n", name); @@ -570,28 +574,28 @@ void IMB_exr_set_channel(void *handle, const char *layname, const char *passname void IMB_exrtile_clear_channels(void *handle) { - ExrHandle *data= (ExrHandle *)handle; + ExrHandle *data = (ExrHandle *)handle; BLI_freelistN(&data->channels); } void IMB_exrtile_write_channels(void *handle, int partx, int party, int level) { - ExrHandle *data= (ExrHandle *)handle; + ExrHandle *data = (ExrHandle *)handle; FrameBuffer frameBuffer; ExrChannel *echan; - - for (echan= (ExrChannel *)data->channels.first; echan; echan= echan->next) { - float *rect= echan->rect - echan->xstride*partx - echan->ystride*party; - frameBuffer.insert (echan->name, Slice (Imf::FLOAT, (char *)rect, - echan->xstride*sizeof(float), echan->ystride*sizeof(float))); + for (echan = (ExrChannel *)data->channels.first; echan; echan = echan->next) { + float *rect = echan->rect - echan->xstride * partx - echan->ystride * party; + + frameBuffer.insert(echan->name, Slice(Imf::FLOAT, (char *)rect, + echan->xstride * sizeof(float), echan->ystride * sizeof(float))); } - - data->tofile->setFrameBuffer (frameBuffer); + + data->tofile->setFrameBuffer(frameBuffer); try { // printf("write tile %d %d\n", partx/data->tilex, party/data->tiley); - data->tofile->writeTile (partx/data->tilex, party/data->tiley, level); + data->tofile->writeTile(partx / data->tilex, party / data->tiley, level); } catch (const std::exception &exc) { std::cerr << "OpenEXR-writeTile: ERROR: " << exc.what() << std::endl; @@ -600,22 +604,22 @@ void IMB_exrtile_write_channels(void *handle, int partx, int party, int level) void IMB_exr_write_channels(void *handle) { - ExrHandle *data= (ExrHandle *)handle; + ExrHandle *data = (ExrHandle *)handle; FrameBuffer frameBuffer; ExrChannel *echan; - + if (data->channels.first) { - for (echan= (ExrChannel *)data->channels.first; echan; echan= echan->next) { + for (echan = (ExrChannel *)data->channels.first; echan; echan = echan->next) { /* last scanline, stride negative */ - float *rect = echan->rect + echan->xstride*(data->height-1)*data->width; - - frameBuffer.insert (echan->name, Slice (Imf::FLOAT, (char *)rect, - echan->xstride*sizeof(float), -echan->ystride*sizeof(float))); + float *rect = echan->rect + echan->xstride * (data->height - 1) * data->width; + + frameBuffer.insert(echan->name, Slice(Imf::FLOAT, (char *)rect, + echan->xstride * sizeof(float), -echan->ystride * sizeof(float))); } - - data->ofile->setFrameBuffer (frameBuffer); + + data->ofile->setFrameBuffer(frameBuffer); try { - data->ofile->writePixels (data->height); + data->ofile->writePixels(data->height); } catch (const std::exception &exc) { std::cerr << "OpenEXR-writePixels: ERROR: " << exc.what() << std::endl; @@ -628,58 +632,58 @@ void IMB_exr_write_channels(void *handle) void IMB_exr_read_channels(void *handle) { - ExrHandle *data= (ExrHandle *)handle; + ExrHandle *data = (ExrHandle *)handle; FrameBuffer frameBuffer; ExrChannel *echan; - + /* check if exr was saved with previous versions of blender which flipped images */ const StringAttribute *ta = data->ifile->header().findTypedAttribute <StringAttribute> ("BlenderMultiChannel"); - short flip = (ta && strncmp(ta->value().c_str(), "Blender V2.43", 13)==0); /* 'previous multilayer attribute, flipped */ - - for (echan= (ExrChannel *)data->channels.first; echan; echan= echan->next) { - + short flip = (ta && strncmp(ta->value().c_str(), "Blender V2.43", 13) == 0); /* 'previous multilayer attribute, flipped */ + + for (echan = (ExrChannel *)data->channels.first; echan; echan = echan->next) { + if (echan->rect) { if (flip) - frameBuffer.insert (echan->name, Slice (Imf::FLOAT, (char *)echan->rect, - echan->xstride*sizeof(float), echan->ystride*sizeof(float))); + frameBuffer.insert(echan->name, Slice(Imf::FLOAT, (char *)echan->rect, + echan->xstride * sizeof(float), echan->ystride * sizeof(float))); else - frameBuffer.insert (echan->name, Slice (Imf::FLOAT, (char *)(echan->rect + echan->xstride*(data->height-1)*data->width), - echan->xstride*sizeof(float), -echan->ystride*sizeof(float))); + frameBuffer.insert(echan->name, Slice(Imf::FLOAT, (char *)(echan->rect + echan->xstride * (data->height - 1) * data->width), + echan->xstride * sizeof(float), -echan->ystride * sizeof(float))); } - else + else printf("warning, channel with no rect set %s\n", echan->name); } - - data->ifile->setFrameBuffer (frameBuffer); + + data->ifile->setFrameBuffer(frameBuffer); try { - data->ifile->readPixels (0, data->height-1); + data->ifile->readPixels(0, data->height - 1); } catch (const std::exception &exc) { std::cerr << "OpenEXR-readPixels: ERROR: " << exc.what() << std::endl; } } -void IMB_exr_multilayer_convert(void *handle, void *base, - void * (*addlayer)(void *base, char *str), - void (*addpass)(void *base, void *lay, char *str, - float *rect, int totchan, char *chan_id)) +void IMB_exr_multilayer_convert(void *handle, void *base, + void * (*addlayer)(void *base, char *str), + void (*addpass)(void *base, void *lay, char *str, + float *rect, int totchan, char *chan_id)) { - ExrHandle *data= (ExrHandle *)handle; + ExrHandle *data = (ExrHandle *)handle; ExrLayer *lay; ExrPass *pass; - if (data->layers.first==NULL) { + if (data->layers.first == NULL) { printf("cannot convert multilayer, no layers in handle\n"); return; } - for (lay= (ExrLayer *)data->layers.first; lay; lay= lay->next) { - void *laybase= addlayer(base, lay->name); + for (lay = (ExrLayer *)data->layers.first; lay; lay = lay->next) { + void *laybase = addlayer(base, lay->name); if (laybase) { - for (pass= (ExrPass *)lay->passes.first; pass; pass= pass->next) { + for (pass = (ExrPass *)lay->passes.first; pass; pass = pass->next) { addpass(base, laybase, pass->name, pass->rect, pass->totchan, pass->chan_id); - pass->rect= NULL; + pass->rect = NULL; } } } @@ -688,31 +692,31 @@ void IMB_exr_multilayer_convert(void *handle, void *base, void IMB_exr_close(void *handle) { - ExrHandle *data= (ExrHandle *)handle; + ExrHandle *data = (ExrHandle *)handle; ExrLayer *lay; ExrPass *pass; - + if (data->ifile) delete data->ifile; else if (data->ofile) delete data->ofile; else if (data->tofile) delete data->tofile; - - data->ifile= NULL; - data->ofile= NULL; - data->tofile= NULL; - + + data->ifile = NULL; + data->ofile = NULL; + data->tofile = NULL; + BLI_freelistN(&data->channels); - - for (lay= (ExrLayer *)data->layers.first; lay; lay= lay->next) { - for (pass= (ExrPass *)lay->passes.first; pass; pass= pass->next) + + for (lay = (ExrLayer *)data->layers.first; lay; lay = lay->next) { + for (pass = (ExrPass *)lay->passes.first; pass; pass = pass->next) if (pass->rect) MEM_freeN(pass->rect); BLI_freelistN(&lay->passes); } BLI_freelistN(&data->layers); - + BLI_remlink(&exrhandles, data); MEM_freeN(data); } @@ -722,12 +726,12 @@ void IMB_exr_close(void *handle) /* get a substring from the end of the name, separated by '.' */ static int imb_exr_split_token(const char *str, const char *end, const char **token) { - int maxlen = end-str; + int maxlen = end - str; int len = 0; - while (len < maxlen && *(end-len-1) != '.') + while (len < maxlen && *(end - len - 1) != '.') ++len; - - *token = end-len; + + *token = end - len; return len; } @@ -738,7 +742,7 @@ static int imb_exr_split_channel_name(ExrChannel *echan, char *layname, char *pa const char *token; char tokenbuf[EXR_TOT_MAXNAME]; int len; - + /* last token is single character channel identifier */ len = imb_exr_split_token(name, end, &token); if (len == 0) { @@ -751,32 +755,32 @@ static int imb_exr_split_channel_name(ExrChannel *echan, char *layname, char *pa return 0; } echan->chan_id = token[0]; - end -= len + 1; /* +1 to skip '.' separator */ - + end -= len + 1; /* +1 to skip '.' separator */ + /* second token is pass name */ len = imb_exr_split_token(name, end, &token); if (len == 0) { printf("multilayer read: bad channel name: %s\n", name); return 0; } - BLI_strncpy(passname, token, len+1); - end -= len + 1; /* +1 to skip '.' separator */ - + BLI_strncpy(passname, token, len + 1); + end -= len + 1; /* +1 to skip '.' separator */ + /* all preceding tokens combined as layer name */ if (end > name) - BLI_strncpy(layname, name, (int)(end-name)+1); + BLI_strncpy(layname, name, (int)(end - name) + 1); else layname[0] = '\0'; - + return 1; } static ExrLayer *imb_exr_get_layer(ListBase *lb, char *layname) { - ExrLayer *lay= (ExrLayer *)BLI_findstring(lb, layname, offsetof(ExrLayer, name)); + ExrLayer *lay = (ExrLayer *)BLI_findstring(lb, layname, offsetof(ExrLayer, name)); - if (lay==NULL) { - lay= (ExrLayer *)MEM_callocN(sizeof(ExrLayer), "exr layer"); + if (lay == NULL) { + lay = (ExrLayer *)MEM_callocN(sizeof(ExrLayer), "exr layer"); BLI_addtail(lb, lay); BLI_strncpy(lay->name, layname, EXR_LAY_MAXNAME); } @@ -786,19 +790,19 @@ static ExrLayer *imb_exr_get_layer(ListBase *lb, char *layname) static ExrPass *imb_exr_get_pass(ListBase *lb, char *passname) { - ExrPass *pass= (ExrPass *)BLI_findstring(lb, passname, offsetof(ExrPass, name)); - - if (pass==NULL) { - pass= (ExrPass *)MEM_callocN(sizeof(ExrPass), "exr pass"); + ExrPass *pass = (ExrPass *)BLI_findstring(lb, passname, offsetof(ExrPass, name)); - if (strcmp(passname, "Combined")==0) + if (pass == NULL) { + pass = (ExrPass *)MEM_callocN(sizeof(ExrPass), "exr pass"); + + if (strcmp(passname, "Combined") == 0) BLI_addhead(lb, pass); else BLI_addtail(lb, pass); } BLI_strncpy(pass->name, passname, EXR_LAY_MAXNAME); - + return pass; } @@ -808,29 +812,29 @@ static ExrHandle *imb_exr_begin_read_mem(InputFile *file, int width, int height) ExrLayer *lay; ExrPass *pass; ExrChannel *echan; - ExrHandle *data= (ExrHandle *)IMB_exr_get_handle(); + ExrHandle *data = (ExrHandle *)IMB_exr_get_handle(); int a; char layname[EXR_TOT_MAXNAME], passname[EXR_TOT_MAXNAME]; - - data->ifile= file; - data->width= width; - data->height= height; - + + data->ifile = file; + data->width = width; + data->height = height; + const ChannelList &channels = data->ifile->header().channels(); for (ChannelList::ConstIterator i = channels.begin(); i != channels.end(); ++i) IMB_exr_add_channel(data, NULL, i.name(), 0, 0, NULL); - + /* now try to sort out how to assign memory to the channels */ /* first build hierarchical layer list */ - for (echan= (ExrChannel *)data->channels.first; echan; echan= echan->next) { - if ( imb_exr_split_channel_name(echan, layname, passname) ) { - ExrLayer *lay= imb_exr_get_layer(&data->layers, layname); - ExrPass *pass= imb_exr_get_pass(&lay->passes, passname); - - pass->chan[pass->totchan]= echan; + for (echan = (ExrChannel *)data->channels.first; echan; echan = echan->next) { + if (imb_exr_split_channel_name(echan, layname, passname) ) { + ExrLayer *lay = imb_exr_get_layer(&data->layers, layname); + ExrPass *pass = imb_exr_get_pass(&lay->passes, passname); + + pass->chan[pass->totchan] = echan; pass->totchan++; - if (pass->totchan>=EXR_PASS_MAXCHAN) + if (pass->totchan >= EXR_PASS_MAXCHAN) break; } } @@ -839,65 +843,65 @@ static ExrHandle *imb_exr_begin_read_mem(InputFile *file, int width, int height) IMB_exr_close(data); return NULL; } - + /* with some heuristics, try to merge the channels in buffers */ - for (lay= (ExrLayer *)data->layers.first; lay; lay= lay->next) { - for (pass= (ExrPass *)lay->passes.first; pass; pass= pass->next) { + for (lay = (ExrLayer *)data->layers.first; lay; lay = lay->next) { + for (pass = (ExrPass *)lay->passes.first; pass; pass = pass->next) { if (pass->totchan) { - pass->rect= (float *)MEM_mapallocN(width*height*pass->totchan*sizeof(float), "pass rect"); - if (pass->totchan==1) { - echan= pass->chan[0]; - echan->rect= pass->rect; - echan->xstride= 1; - echan->ystride= width; - pass->chan_id[0]= echan->chan_id; + pass->rect = (float *)MEM_mapallocN(width * height * pass->totchan * sizeof(float), "pass rect"); + if (pass->totchan == 1) { + echan = pass->chan[0]; + echan->rect = pass->rect; + echan->xstride = 1; + echan->ystride = width; + pass->chan_id[0] = echan->chan_id; } else { char lookup[256]; - + memset(lookup, 0, sizeof(lookup)); - + /* we can have RGB(A), XYZ(W), UVA */ - if (pass->totchan==3 || pass->totchan==4) { - if (pass->chan[0]->chan_id=='B' || pass->chan[1]->chan_id=='B' || pass->chan[2]->chan_id=='B') { - lookup[(unsigned int)'R']= 0; - lookup[(unsigned int)'G']= 1; - lookup[(unsigned int)'B']= 2; - lookup[(unsigned int)'A']= 3; + if (pass->totchan == 3 || pass->totchan == 4) { + if (pass->chan[0]->chan_id == 'B' || pass->chan[1]->chan_id == 'B' || pass->chan[2]->chan_id == 'B') { + lookup[(unsigned int)'R'] = 0; + lookup[(unsigned int)'G'] = 1; + lookup[(unsigned int)'B'] = 2; + lookup[(unsigned int)'A'] = 3; } - else if (pass->chan[0]->chan_id=='Y' || pass->chan[1]->chan_id=='Y' || pass->chan[2]->chan_id=='Y') { - lookup[(unsigned int)'X']= 0; - lookup[(unsigned int)'Y']= 1; - lookup[(unsigned int)'Z']= 2; - lookup[(unsigned int)'W']= 3; + else if (pass->chan[0]->chan_id == 'Y' || pass->chan[1]->chan_id == 'Y' || pass->chan[2]->chan_id == 'Y') { + lookup[(unsigned int)'X'] = 0; + lookup[(unsigned int)'Y'] = 1; + lookup[(unsigned int)'Z'] = 2; + lookup[(unsigned int)'W'] = 3; } else { - lookup[(unsigned int)'U']= 0; - lookup[(unsigned int)'V']= 1; - lookup[(unsigned int)'A']= 2; + lookup[(unsigned int)'U'] = 0; + lookup[(unsigned int)'V'] = 1; + lookup[(unsigned int)'A'] = 2; } - for (a=0; a<pass->totchan; a++) { - echan= pass->chan[a]; - echan->rect= pass->rect + lookup[(unsigned int)echan->chan_id]; - echan->xstride= pass->totchan; - echan->ystride= width*pass->totchan; - pass->chan_id[ (unsigned int)lookup[(unsigned int)echan->chan_id] ]= echan->chan_id; + for (a = 0; a < pass->totchan; a++) { + echan = pass->chan[a]; + echan->rect = pass->rect + lookup[(unsigned int)echan->chan_id]; + echan->xstride = pass->totchan; + echan->ystride = width * pass->totchan; + pass->chan_id[(unsigned int)lookup[(unsigned int)echan->chan_id]] = echan->chan_id; } } else { /* unknown */ - for (a=0; a<pass->totchan; a++) { - echan= pass->chan[a]; - echan->rect= pass->rect + a; - echan->xstride= pass->totchan; - echan->ystride= width*pass->totchan; - pass->chan_id[a]= echan->chan_id; + for (a = 0; a < pass->totchan; a++) { + echan = pass->chan[a]; + echan->rect = pass->rect + a; + echan->xstride = pass->totchan; + echan->ystride = width * pass->totchan; + pass->chan_id[a] = echan->chan_id; } } } } } } - + return data; } @@ -916,7 +920,7 @@ typedef struct RGBA { static void exr_print_filecontents(InputFile *file) { const ChannelList &channels = file->header().channels(); - + for (ChannelList::ConstIterator i = channels.begin(); i != channels.end(); ++i) { const Channel &channel = i.channel(); printf("OpenEXR-load: Found channel %s of type %d\n", i.name(), channel.type); @@ -927,13 +931,13 @@ static void exr_print_filecontents(InputFile *file) static const char *exr_rgba_channelname(InputFile *file, const char *chan) { const ChannelList &channels = file->header().channels(); - + for (ChannelList::ConstIterator i = channels.begin(); i != channels.end(); ++i) { /* const Channel &channel = i.channel(); */ /* Not used yet */ - const char *str= i.name(); - int len= strlen(str); + const char *str = i.name(); + int len = strlen(str); if (len) { - if (BLI_strcasecmp(chan, str+len-1)==0) { + if (BLI_strcasecmp(chan, str + len - 1) == 0) { return str; } } @@ -941,23 +945,26 @@ static const char *exr_rgba_channelname(InputFile *file, const char *chan) return chan; } - - static int exr_has_zbuffer(InputFile *file) { return !(file->header().channels().findChannel("Z") == NULL); } +static int exr_has_alpha(InputFile *file) +{ + return !(file->header().channels().findChannel("A") == NULL); +} + static int exr_is_multilayer(InputFile *file) { - const StringAttribute *comments= file->header().findTypedAttribute<StringAttribute>("BlenderMultiChannel"); + const StringAttribute *comments = file->header().findTypedAttribute<StringAttribute>("BlenderMultiChannel"); const ChannelList &channels = file->header().channels(); std::set <std::string> layerNames; channels.layers(layerNames); - if (comments || layerNames.size()>1) - return 1; + if (comments || layerNames.size() > 1) + return 1; return 0; } @@ -966,82 +973,84 @@ struct ImBuf *imb_load_openexr(unsigned char *mem, size_t size, int flags) { struct ImBuf *ibuf = NULL; InputFile *file = NULL; - + if (imb_is_a_openexr(mem) == 0) return(NULL); - + try { - Mem_IStream *membuf = new Mem_IStream(mem, size); + Mem_IStream *membuf = new Mem_IStream(mem, size); int is_multi; file = new InputFile(*membuf); - + Box2i dw = file->header().dataWindow(); - int width = dw.max.x - dw.min.x + 1; - int height = dw.max.y - dw.min.y + 1; - - //printf("OpenEXR-load: image data window %d %d %d %d\n", + const int width = dw.max.x - dw.min.x + 1; + const int height = dw.max.y - dw.min.y + 1; + + //printf("OpenEXR-load: image data window %d %d %d %d\n", // dw.min.x, dw.min.y, dw.max.x, dw.max.y); if (0) // debug exr_print_filecontents(file); - - is_multi= exr_is_multilayer(file); - + + is_multi = exr_is_multilayer(file); + /* do not make an ibuf when */ if (is_multi && !(flags & IB_test) && !(flags & IB_multilayer)) { printf("Error: can't process EXR multilayer file\n"); } else { - - ibuf = IMB_allocImBuf(width, height, 32, 0); + const int is_alpha = exr_has_alpha(file); + + ibuf = IMB_allocImBuf(width, height, is_alpha ? 32 : 24, 0); ibuf->ftype = OPENEXR; /* openEXR is linear as per EXR spec */ ibuf->profile = IB_PROFILE_LINEAR_RGB; - + if (!(flags & IB_test)) { if (is_multi) { /* only enters with IB_multilayer flag set */ /* constructs channels for reading, allocates memory in channels */ - ExrHandle *handle= imb_exr_begin_read_mem(file, width, height); + ExrHandle *handle = imb_exr_begin_read_mem(file, width, height); if (handle) { IMB_exr_read_channels(handle); - ibuf->userdata= handle; /* potential danger, the caller has to check for this! */ + ibuf->userdata = handle; /* potential danger, the caller has to check for this! */ } } else { FrameBuffer frameBuffer; float *first; int xstride = sizeof(float) * 4; - int ystride = - xstride*width; - + int ystride = -xstride * width; + imb_addrectfloatImBuf(ibuf); - + /* inverse correct first pixel for datawindow coordinates (- dw.min.y because of y flip) */ - first= ibuf->rect_float - 4*(dw.min.x - dw.min.y*width); + first = ibuf->rect_float - 4 * (dw.min.x - dw.min.y * width); /* but, since we read y-flipped (negative y stride) we move to last scanline */ - first+= 4*(height-1)*width; - - frameBuffer.insert ( exr_rgba_channelname(file, "R"), - Slice (Imf::FLOAT, (char *) first, xstride, ystride)); - frameBuffer.insert ( exr_rgba_channelname(file, "G"), - Slice (Imf::FLOAT, (char *) (first+1), xstride, ystride)); - frameBuffer.insert ( exr_rgba_channelname(file, "B"), - Slice (Imf::FLOAT, (char *) (first+2), xstride, ystride)); - - frameBuffer.insert ( exr_rgba_channelname(file, "A"), - Slice (Imf::FLOAT, (char *) (first+3), xstride, ystride, 1, 1, 1.0f)); /* 1.0 is fill value */ + first += 4 * (height - 1) * width; + + frameBuffer.insert(exr_rgba_channelname(file, "R"), + Slice(Imf::FLOAT, (char *) first, xstride, ystride)); + frameBuffer.insert(exr_rgba_channelname(file, "G"), + Slice(Imf::FLOAT, (char *) (first + 1), xstride, ystride)); + frameBuffer.insert(exr_rgba_channelname(file, "B"), + Slice(Imf::FLOAT, (char *) (first + 2), xstride, ystride)); + + /* 1.0 is fill value, this still neesd to be assigned even when (is_alpha == 0) */ + frameBuffer.insert(exr_rgba_channelname(file, "A"), + Slice(Imf::FLOAT, (char *) (first + 3), xstride, ystride, 1, 1, 1.0f)); if (exr_has_zbuffer(file)) { float *firstz; - + addzbuffloatImBuf(ibuf); - firstz= ibuf->zbuf_float - (dw.min.x - dw.min.y*width); - firstz+= (height-1)*width; - frameBuffer.insert("Z", Slice (Imf::FLOAT, (char *)firstz, sizeof(float), -width * sizeof(float))); + firstz = ibuf->zbuf_float - (dw.min.x - dw.min.y * width); + firstz += (height - 1) * width; + frameBuffer.insert("Z", Slice(Imf::FLOAT, (char *)firstz, sizeof(float), -width * sizeof(float))); } - - file->setFrameBuffer (frameBuffer); - file->readPixels (dw.min.y, dw.max.y); + + file->setFrameBuffer(frameBuffer); + file->readPixels(dw.min.y, dw.max.y); // XXX, ImBuf has no nice way to deal with this. // ideally IM_rect would be used when the caller wants a rect BUT @@ -1052,7 +1061,7 @@ struct ImBuf *imb_load_openexr(unsigned char *mem, size_t size, int flags) // // if (flag & IM_rect) // IMB_rect_from_float(ibuf); - + /* file is no longer needed */ delete file; } @@ -1065,10 +1074,10 @@ struct ImBuf *imb_load_openexr(unsigned char *mem, size_t size, int flags) std::cerr << exc.what() << std::endl; if (ibuf) IMB_freeImBuf(ibuf); delete file; - + return (0); } - + } void imb_initopenexr(void) |