bool
OpenEXRInput::read_native_tile (int x, int y, int z, void *data)
{
ASSERT (m_input_tiled != NULL);
// Compute where OpenEXR needs to think the full buffers starts.
// OpenImageIO requires that 'data' points to where the client wants
// to put the pixels being read, but OpenEXR's frameBuffer.insert()
// wants where the address of the "virtual framebuffer" for the
// whole image.
size_t pixelbytes = m_spec.pixel_bytes (true);
char *buf = (char *)data
- x * pixelbytes
- y * pixelbytes * m_spec.tile_width;
try {
Imf::FrameBuffer frameBuffer;
size_t chanoffset = 0;
for (int c = 0; c < m_spec.nchannels; ++c) {
size_t chanbytes = m_spec.channelformats.size()
? m_spec.channelformats[c].size()
: m_spec.format.size();
frameBuffer.insert (m_spec.channelnames[c].c_str(),
Imf::Slice (m_pixeltype[c],
buf + chanoffset, pixelbytes,
pixelbytes*m_spec.tile_width));
chanoffset += chanbytes;
}
m_input_tiled->setFrameBuffer (frameBuffer);
m_input_tiled->readTile ((x - m_spec.x) / m_spec.tile_width,
(y - m_spec.y) / m_spec.tile_height,
m_miplevel, m_miplevel);
}
catch (const std::exception &e) {
error ("Filed OpenEXR read: %s", e.what());
return false;
}
return true;
}
bool
OpenEXRInput::seek_subimage (int subimage, int miplevel, ImageSpec &newspec)
{
if (subimage < 0 || subimage >= m_nsubimages) // out of range
return false;
if (subimage == m_subimage && miplevel == m_miplevel) { // no change
newspec = m_spec;
return true;
}
PartInfo &part (m_parts[subimage]);
if (! part.initialized) {
const Imf::Header *header = NULL;
#ifdef USE_OPENEXR_VERSION2
if (m_input_multipart)
header = &(m_input_multipart->header(subimage));
#else
if (m_input_tiled)
header = &(m_input_tiled->header());
if (m_input_scanline)
header = &(m_input_scanline->header());
#endif
part.parse_header (header);
part.initialized = true;
}
#ifdef USE_OPENEXR_VERSION2
if (subimage != m_subimage) {
delete m_scanline_input_part; m_scanline_input_part = NULL;
delete m_tiled_input_part; m_tiled_input_part = NULL;
delete m_deep_scanline_input_part; m_deep_scanline_input_part = NULL;
delete m_deep_tiled_input_part; m_deep_tiled_input_part = NULL;
try {
if (part.spec.deep) {
if (part.spec.tile_width)
m_deep_tiled_input_part = new Imf::DeepTiledInputPart (*m_input_multipart, subimage);
else
m_deep_scanline_input_part = new Imf::DeepScanLineInputPart (*m_input_multipart, subimage);
} else {
if (part.spec.tile_width)
m_tiled_input_part = new Imf::TiledInputPart (*m_input_multipart, subimage);
else
m_scanline_input_part = new Imf::InputPart (*m_input_multipart, subimage);
}
}
catch (const std::exception &e) {
error ("OpenEXR exception: %s", e.what());
m_scanline_input_part = NULL;
m_tiled_input_part = NULL;
m_deep_scanline_input_part = NULL;
m_deep_tiled_input_part = NULL;
ASSERT(0);
return false;
}
}
#endif
m_subimage = subimage;
if (miplevel < 0 || miplevel >= part.nmiplevels) // out of range
return false;
m_miplevel = miplevel;
m_spec = part.spec;
if (miplevel == 0 && part.levelmode == Imf::ONE_LEVEL) {
newspec = m_spec;
return true;
}
// Compute the resolution of the requested mip level.
int w = part.topwidth, h = part.topheight;
if (part.levelmode == Imf::MIPMAP_LEVELS) {
while (miplevel--) {
if (part.roundingmode == Imf::ROUND_DOWN) {
w = w / 2;
h = h / 2;
} else {
w = (w + 1) / 2;
h = (h + 1) / 2;
}
w = std::max (1, w);
h = std::max (1, h);
}
} else if (part.levelmode == Imf::RIPMAP_LEVELS) {
// FIXME
} else {
ASSERT(0);
}
m_spec.width = w;
m_spec.height = h;
// N.B. OpenEXR doesn't support data and display windows per MIPmap
// level. So always take from the top level.
Imath::Box2i datawindow = part.top_datawindow;
Imath::Box2i displaywindow = part.top_displaywindow;
m_spec.x = datawindow.min.x;
m_spec.y = datawindow.min.y;
if (m_miplevel == 0) {
m_spec.full_x = displaywindow.min.x;
m_spec.full_y = displaywindow.min.y;
m_spec.full_width = displaywindow.max.x - displaywindow.min.x + 1;
m_spec.full_height = displaywindow.max.y - displaywindow.min.y + 1;
} else {
m_spec.full_x = m_spec.x;
m_spec.full_y = m_spec.y;
m_spec.full_width = m_spec.width;
m_spec.full_height = m_spec.height;
}
if (part.cubeface) {
m_spec.full_width = w;
m_spec.full_height = w;
}
newspec = m_spec;
return true;
}
bool
OpenEXRInput::read_native_tiles (int xbegin, int xend, int ybegin, int yend,
int zbegin, int zend,
int chbegin, int chend, void *data)
{
chend = clamp (chend, chbegin+1, m_spec.nchannels);
#if 0
std::cerr << "openexr rnt " << xbegin << ' ' << xend << ' ' << ybegin
<< ' ' << yend << ", chans " << chbegin
<< "-" << (chend-1) << "\n";
#endif
if (! (m_input_tiled || m_tiled_input_part) ||
! m_spec.valid_tile_range (xbegin, xend, ybegin, yend, zbegin, zend)) {
error ("called OpenEXRInput::read_native_tiles without an open file");
return false;
}
// Compute where OpenEXR needs to think the full buffers starts.
// OpenImageIO requires that 'data' points to where the client wants
// to put the pixels being read, but OpenEXR's frameBuffer.insert()
// wants where the address of the "virtual framebuffer" for the
// whole image.
const PartInfo &part (m_parts[m_subimage]);
size_t pixelbytes = m_spec.pixel_bytes (chbegin, chend, true);
int firstxtile = (xbegin-m_spec.x) / m_spec.tile_width;
int firstytile = (ybegin-m_spec.y) / m_spec.tile_height;
// clamp to the image edge
xend = std::min (xend, m_spec.x+m_spec.width);
yend = std::min (yend, m_spec.y+m_spec.height);
zend = std::min (zend, m_spec.z+m_spec.depth);
// figure out how many tiles we need
int nxtiles = (xend - xbegin + m_spec.tile_width - 1) / m_spec.tile_width;
int nytiles = (yend - ybegin + m_spec.tile_height - 1) / m_spec.tile_height;
int whole_width = nxtiles * m_spec.tile_width;
int whole_height = nytiles * m_spec.tile_height;
boost::scoped_array<char> tmpbuf;
void *origdata = data;
if (whole_width != (xend-xbegin) || whole_height != (yend-ybegin)) {
// Deal with the case of reading not a whole number of tiles --
// OpenEXR will happily overwrite user memory in this case.
tmpbuf.reset (new char [nxtiles * nytiles * m_spec.tile_bytes(true)]);
data = &tmpbuf[0];
}
char *buf = (char *)data
- xbegin * pixelbytes
- ybegin * pixelbytes * m_spec.tile_width * nxtiles;
try {
Imf::FrameBuffer frameBuffer;
size_t chanoffset = 0;
for (int c = chbegin; c < chend; ++c) {
size_t chanbytes = m_spec.channelformat(c).size();
frameBuffer.insert (m_spec.channelnames[c].c_str(),
Imf::Slice (part.pixeltype[c],
buf + chanoffset, pixelbytes,
pixelbytes*m_spec.tile_width*nxtiles));
chanoffset += chanbytes;
}
if (m_input_tiled) {
m_input_tiled->setFrameBuffer (frameBuffer);
m_input_tiled->readTiles (firstxtile, firstxtile+nxtiles-1,
firstytile, firstytile+nytiles-1,
m_miplevel, m_miplevel);
#ifdef USE_OPENEXR_VERSION2
} else if (m_tiled_input_part) {
m_tiled_input_part->setFrameBuffer (frameBuffer);
m_tiled_input_part->readTiles (firstxtile, firstxtile+nxtiles-1,
firstytile, firstytile+nytiles-1,
m_miplevel, m_miplevel);
#endif
} else {
ASSERT (0);
}
if (data != origdata) {
stride_t user_scanline_bytes = (xend-xbegin) * pixelbytes;
stride_t scanline_stride = nxtiles*m_spec.tile_width*pixelbytes;
for (int y = ybegin; y < yend; ++y)
memcpy ((char *)origdata+(y-ybegin)*scanline_stride,
(char *)data+(y-ybegin)*scanline_stride,
user_scanline_bytes);
}
}
catch (const std::exception &e) {
error ("Failed OpenEXR read: %s", e.what());
return false;
}
return true;
}
bool
OpenEXRInput::open (const std::string &name, ImageSpec &newspec)
{
// Quick check to reject non-exr files
bool tiled;
if (! Imf::isOpenExrFile (name.c_str(), tiled))
return false;
m_spec = ImageSpec(); // Clear everything with default constructor
// Unless otherwise specified, exr files are assumed to be linear.
m_spec.attribute ("oiio:ColorSpace", "Linear");
try {
if (tiled) {
m_input_tiled = new Imf::TiledInputFile (name.c_str());
m_header = &(m_input_tiled->header());
} else {
m_input_scanline = new Imf::InputFile (name.c_str());
m_header = &(m_input_scanline->header());
}
}
catch (const std::exception &e) {
error ("OpenEXR exception: %s", e.what());
return false;
}
if (! m_input_scanline && ! m_input_tiled) {
error ("Unknown error opening EXR file");
return false;
}
Imath::Box2i datawindow = m_header->dataWindow();
m_spec.x = datawindow.min.x;
m_spec.y = datawindow.min.y;
m_spec.z = 0;
m_spec.width = datawindow.max.x - datawindow.min.x + 1;
m_spec.height = datawindow.max.y - datawindow.min.y + 1;
m_spec.depth = 1;
m_topwidth = m_spec.width; // Save top-level mipmap dimensions
m_topheight = m_spec.height;
Imath::Box2i displaywindow = m_header->displayWindow();
m_spec.full_x = displaywindow.min.x;
m_spec.full_y = displaywindow.min.y;
m_spec.full_z = 0;
m_spec.full_width = displaywindow.max.x - displaywindow.min.x + 1;
m_spec.full_height = displaywindow.max.y - displaywindow.min.y + 1;
m_spec.full_depth = 1;
if (tiled) {
m_spec.tile_width = m_input_tiled->tileXSize();
m_spec.tile_height = m_input_tiled->tileYSize();
} else {
m_spec.tile_width = 0;
m_spec.tile_height = 0;
}
m_spec.tile_depth = 1;
query_channels (); // also sets format
m_nsubimages = 1;
if (tiled) {
// FIXME: levelmode
m_levelmode = m_input_tiled->levelMode();
m_roundingmode = m_input_tiled->levelRoundingMode();
if (m_levelmode == Imf::MIPMAP_LEVELS) {
m_nmiplevels = m_input_tiled->numLevels();
m_spec.attribute ("openexr:roundingmode", m_roundingmode);
} else if (m_levelmode == Imf::RIPMAP_LEVELS) {
m_nmiplevels = std::max (m_input_tiled->numXLevels(),
m_input_tiled->numYLevels());
m_spec.attribute ("openexr:roundingmode", m_roundingmode);
} else {
m_nmiplevels = 1;
}
} else {
m_levelmode = Imf::ONE_LEVEL;
m_nmiplevels = 1;
}
const Imf::EnvmapAttribute *envmap;
envmap = m_header->findTypedAttribute<Imf::EnvmapAttribute>("envmap");
if (envmap) {
m_cubeface = (envmap->value() == Imf::ENVMAP_CUBE);
m_spec.attribute ("textureformat", m_cubeface ? "CubeFace Environment" : "LatLong Environment");
// OpenEXR conventions for env maps
if (! m_cubeface)
m_spec.attribute ("oiio:updirection", "y");
m_spec.attribute ("oiio:sampleborder", 1);
// FIXME - detect CubeFace Shadow?
} else {
m_cubeface = false;
if (tiled && m_levelmode == Imf::MIPMAP_LEVELS)
m_spec.attribute ("textureformat", "Plain Texture");
// FIXME - detect Shadow
}
const Imf::CompressionAttribute *compressattr;
compressattr = m_header->findTypedAttribute<Imf::CompressionAttribute>("compression");
if (compressattr) {
const char *comp = NULL;
switch (compressattr->value()) {
case Imf::NO_COMPRESSION : comp = "none"; break;
case Imf::RLE_COMPRESSION : comp = "rle"; break;
case Imf::ZIPS_COMPRESSION : comp = "zip"; break;
case Imf::ZIP_COMPRESSION : comp = "zip"; break;
case Imf::PIZ_COMPRESSION : comp = "piz"; break;
case Imf::PXR24_COMPRESSION : comp = "pxr24"; break;
#ifdef IMF_B44_COMPRESSION
// The enum Imf::B44_COMPRESSION is not defined in older versions
// of OpenEXR, and there are no explicit version numbers in the
// headers. BUT this other related #define is present only in
// the newer version.
case Imf::B44_COMPRESSION : comp = "b44"; break;
case Imf::B44A_COMPRESSION : comp = "b44a"; break;
#endif
default:
break;
}
if (comp)
m_spec.attribute ("compression", comp);
}
for (Imf::Header::ConstIterator hit = m_header->begin();
hit != m_header->end(); ++hit) {
const Imf::IntAttribute *iattr;
const Imf::FloatAttribute *fattr;
const Imf::StringAttribute *sattr;
const Imf::M44fAttribute *mattr;
const Imf::V3fAttribute *vattr;
const char *name = hit.name();
std::string oname = exr_tag_to_ooio_std[name];
if (oname.empty()) // Empty string means skip this attrib
continue;
// if (oname == name)
// oname = std::string(format_name()) + "_" + oname;
const Imf::Attribute &attrib = hit.attribute();
std::string type = attrib.typeName();
if (type == "string" &&
(sattr = m_header->findTypedAttribute<Imf::StringAttribute> (name)))
m_spec.attribute (oname, sattr->value().c_str());
else if (type == "int" &&
(iattr = m_header->findTypedAttribute<Imf::IntAttribute> (name)))
m_spec.attribute (oname, iattr->value());
else if (type == "float" &&
(fattr = m_header->findTypedAttribute<Imf::FloatAttribute> (name)))
m_spec.attribute (oname, fattr->value());
else if (type == "m44f" &&
(mattr = m_header->findTypedAttribute<Imf::M44fAttribute> (name)))
m_spec.attribute (oname, TypeDesc::TypeMatrix, &(mattr->value()));
else if (type == "v3f" &&
(vattr = m_header->findTypedAttribute<Imf::V3fAttribute> (name)))
m_spec.attribute (oname, TypeDesc::TypeVector, &(vattr->value()));
else {
#if 0
std::cerr << " unknown attribute " << type << ' ' << name << "\n";
#endif
}
}
m_subimage = 0;
m_miplevel = 0;
newspec = m_spec;
return true;
}