480 lines
16 KiB
C++
480 lines
16 KiB
C++
///////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (c) 2004, Industrial Light & Magic, a division of Lucas
|
|
// Digital Ltd. LLC
|
|
//
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are
|
|
// met:
|
|
// * Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above
|
|
// copyright notice, this list of conditions and the following disclaimer
|
|
// in the documentation and/or other materials provided with the
|
|
// distribution.
|
|
// * Neither the name of Industrial Light & Magic nor the names of
|
|
// its contributors may be used to endorse or promote products derived
|
|
// from this software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
#ifndef INCLUDED_IMF_TILED_RGBA_FILE_H
|
|
#define INCLUDED_IMF_TILED_RGBA_FILE_H
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
// Simplified RGBA image I/O for tiled files
|
|
//
|
|
// class TiledRgbaOutputFile
|
|
// class TiledRgbaInputFile
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
#include <ImfHeader.h>
|
|
#include <ImfFrameBuffer.h>
|
|
#include "ImathVec.h"
|
|
#include "ImathBox.h"
|
|
#include "half.h"
|
|
#include <ImfTileDescription.h>
|
|
#include <ImfRgba.h>
|
|
#include <ImfThreading.h>
|
|
#include <string>
|
|
|
|
namespace Imf {
|
|
|
|
class TiledOutputFile;
|
|
class TiledInputFile;
|
|
struct PreviewRgba;
|
|
|
|
|
|
//
|
|
// Tiled RGBA output file.
|
|
//
|
|
|
|
class TiledRgbaOutputFile
|
|
{
|
|
public:
|
|
|
|
//---------------------------------------------------
|
|
// Constructor -- rgbaChannels, tileXSize, tileYSize,
|
|
// levelMode, and levelRoundingMode overwrite the
|
|
// channel list and tile description attribute in the
|
|
// header that is passed as an argument to the
|
|
// constructor.
|
|
//---------------------------------------------------
|
|
|
|
TiledRgbaOutputFile (const char name[],
|
|
const Header &header,
|
|
RgbaChannels rgbaChannels,
|
|
int tileXSize,
|
|
int tileYSize,
|
|
LevelMode mode,
|
|
LevelRoundingMode rmode = ROUND_DOWN,
|
|
int numThreads = globalThreadCount ());
|
|
|
|
|
|
//---------------------------------------------------
|
|
// Constructor -- like the previous one, but the new
|
|
// TiledRgbaOutputFile is attached to a file that has
|
|
// already been opened by the caller. Destroying
|
|
// TiledRgbaOutputFileObjects constructed with this
|
|
// constructor does not automatically close the
|
|
// corresponding files.
|
|
//---------------------------------------------------
|
|
|
|
TiledRgbaOutputFile (OStream &os,
|
|
const Header &header,
|
|
RgbaChannels rgbaChannels,
|
|
int tileXSize,
|
|
int tileYSize,
|
|
LevelMode mode,
|
|
LevelRoundingMode rmode = ROUND_DOWN,
|
|
int numThreads = globalThreadCount ());
|
|
|
|
|
|
//------------------------------------------------------
|
|
// Constructor -- header data are explicitly specified
|
|
// as function call arguments (an empty dataWindow means
|
|
// "same as displayWindow")
|
|
//------------------------------------------------------
|
|
|
|
TiledRgbaOutputFile (const char name[],
|
|
int tileXSize,
|
|
int tileYSize,
|
|
LevelMode mode,
|
|
LevelRoundingMode rmode,
|
|
const Imath::Box2i &displayWindow,
|
|
const Imath::Box2i &dataWindow = Imath::Box2i(),
|
|
RgbaChannels rgbaChannels = WRITE_RGBA,
|
|
float pixelAspectRatio = 1,
|
|
const Imath::V2f screenWindowCenter =
|
|
Imath::V2f (0, 0),
|
|
float screenWindowWidth = 1,
|
|
LineOrder lineOrder = INCREASING_Y,
|
|
Compression compression = ZIP_COMPRESSION,
|
|
int numThreads = globalThreadCount ());
|
|
|
|
|
|
//-----------------------------------------------
|
|
// Constructor -- like the previous one, but both
|
|
// the display window and the data window are
|
|
// Box2i (V2i (0, 0), V2i (width - 1, height -1))
|
|
//-----------------------------------------------
|
|
|
|
TiledRgbaOutputFile (const char name[],
|
|
int width,
|
|
int height,
|
|
int tileXSize,
|
|
int tileYSize,
|
|
LevelMode mode,
|
|
LevelRoundingMode rmode = ROUND_DOWN,
|
|
RgbaChannels rgbaChannels = WRITE_RGBA,
|
|
float pixelAspectRatio = 1,
|
|
const Imath::V2f screenWindowCenter =
|
|
Imath::V2f (0, 0),
|
|
float screenWindowWidth = 1,
|
|
LineOrder lineOrder = INCREASING_Y,
|
|
Compression compression = ZIP_COMPRESSION,
|
|
int numThreads = globalThreadCount ());
|
|
|
|
|
|
virtual ~TiledRgbaOutputFile ();
|
|
|
|
|
|
//------------------------------------------------
|
|
// Define a frame buffer as the pixel data source:
|
|
// Pixel (x, y) is at address
|
|
//
|
|
// base + x * xStride + y * yStride
|
|
//
|
|
//------------------------------------------------
|
|
|
|
void setFrameBuffer (const Rgba *base,
|
|
size_t xStride,
|
|
size_t yStride);
|
|
|
|
//--------------------------
|
|
// Access to the file header
|
|
//--------------------------
|
|
|
|
const Header & header () const;
|
|
const FrameBuffer & frameBuffer () const;
|
|
const Imath::Box2i & displayWindow () const;
|
|
const Imath::Box2i & dataWindow () const;
|
|
float pixelAspectRatio () const;
|
|
const Imath::V2f screenWindowCenter () const;
|
|
float screenWindowWidth () const;
|
|
LineOrder lineOrder () const;
|
|
Compression compression () const;
|
|
RgbaChannels channels () const;
|
|
|
|
|
|
//----------------------------------------------------
|
|
// Utility functions (same as in Imf::TiledOutputFile)
|
|
//----------------------------------------------------
|
|
|
|
unsigned int tileXSize () const;
|
|
unsigned int tileYSize () const;
|
|
LevelMode levelMode () const;
|
|
LevelRoundingMode levelRoundingMode () const;
|
|
|
|
int numLevels () const;
|
|
int numXLevels () const;
|
|
int numYLevels () const;
|
|
bool isValidLevel (int lx, int ly) const;
|
|
|
|
int levelWidth (int lx) const;
|
|
int levelHeight (int ly) const;
|
|
|
|
int numXTiles (int lx = 0) const;
|
|
int numYTiles (int ly = 0) const;
|
|
|
|
Imath::Box2i dataWindowForLevel (int l = 0) const;
|
|
Imath::Box2i dataWindowForLevel (int lx, int ly) const;
|
|
|
|
Imath::Box2i dataWindowForTile (int dx, int dy,
|
|
int l = 0) const;
|
|
|
|
Imath::Box2i dataWindowForTile (int dx, int dy,
|
|
int lx, int ly) const;
|
|
|
|
//------------------------------------------------------------------
|
|
// Write pixel data:
|
|
//
|
|
// writeTile(dx, dy, lx, ly) writes the tile with tile
|
|
// coordinates (dx, dy), and level number (lx, ly) to
|
|
// the file.
|
|
//
|
|
// dx must lie in the interval [0, numXTiles(lx)-1]
|
|
// dy must lie in the interval [0, numYTiles(ly)-1]
|
|
//
|
|
// lx must lie in the interval [0, numXLevels()-1]
|
|
// ly must lie in the inverval [0, numYLevels()-1]
|
|
//
|
|
// writeTile(dx, dy, level) is a convenience function
|
|
// used for ONE_LEVEL and MIPMAP_LEVEL files. It calls
|
|
// writeTile(dx, dy, level, level).
|
|
//
|
|
// The two writeTiles(dx1, dx2, dy1, dy2, ...) functions allow
|
|
// writing multiple tiles at once. If multi-threading is used
|
|
// multiple tiles are written concurrently.
|
|
//
|
|
// Pixels that are outside the pixel coordinate range for the tile's
|
|
// level, are never accessed by writeTile().
|
|
//
|
|
// Each tile in the file must be written exactly once.
|
|
//
|
|
//------------------------------------------------------------------
|
|
|
|
void writeTile (int dx, int dy, int l = 0);
|
|
void writeTile (int dx, int dy, int lx, int ly);
|
|
|
|
void writeTiles (int dxMin, int dxMax, int dyMin, int dyMax,
|
|
int lx, int ly);
|
|
|
|
void writeTiles (int dxMin, int dxMax, int dyMin, int dyMax,
|
|
int l = 0);
|
|
|
|
|
|
// -------------------------------------------------------------------------
|
|
// Update the preview image (see Imf::TiledOutputFile::updatePreviewImage())
|
|
// -------------------------------------------------------------------------
|
|
|
|
void updatePreviewImage (const PreviewRgba[]);
|
|
|
|
|
|
//------------------------------------------------
|
|
// Break a tile -- for testing and debugging only
|
|
// (see Imf::TiledOutputFile::breakTile())
|
|
//
|
|
// Warning: Calling this function usually results
|
|
// in a broken image file. The file or parts of
|
|
// it may not be readable, or the file may contain
|
|
// bad data.
|
|
//
|
|
//------------------------------------------------
|
|
|
|
void breakTile (int dx, int dy,
|
|
int lx, int ly,
|
|
int offset,
|
|
int length,
|
|
char c);
|
|
private:
|
|
|
|
//
|
|
// Copy constructor and assignment are not implemented
|
|
//
|
|
|
|
TiledRgbaOutputFile (const TiledRgbaOutputFile &);
|
|
TiledRgbaOutputFile & operator = (const TiledRgbaOutputFile &);
|
|
|
|
class ToYa;
|
|
|
|
TiledOutputFile * _outputFile;
|
|
ToYa * _toYa;
|
|
};
|
|
|
|
|
|
|
|
//
|
|
// Tiled RGBA input file
|
|
//
|
|
|
|
class TiledRgbaInputFile
|
|
{
|
|
public:
|
|
|
|
//--------------------------------------------------------
|
|
// Constructor -- opens the file with the specified name.
|
|
// Destroying TiledRgbaInputFile objects constructed with
|
|
// this constructor automatically closes the corresponding
|
|
// files.
|
|
//--------------------------------------------------------
|
|
|
|
TiledRgbaInputFile (const char name[],
|
|
int numThreads = globalThreadCount ());
|
|
|
|
|
|
//-------------------------------------------------------
|
|
// Constructor -- attaches the new TiledRgbaInputFile
|
|
// object to a file that has already been opened by the
|
|
// caller.
|
|
// Destroying TiledRgbaInputFile objects constructed with
|
|
// this constructor does not automatically close the
|
|
// corresponding files.
|
|
//-------------------------------------------------------
|
|
|
|
TiledRgbaInputFile (IStream &is, int numThreads = globalThreadCount ());
|
|
|
|
|
|
//------------------------------------------------------------
|
|
// Constructors -- the same as the previous two, but the names
|
|
// of the red, green, blue, alpha, and luminance channels are
|
|
// expected to be layerName.R, layerName.G, etc.
|
|
//------------------------------------------------------------
|
|
|
|
TiledRgbaInputFile (const char name[],
|
|
const std::string &layerName,
|
|
int numThreads = globalThreadCount());
|
|
|
|
TiledRgbaInputFile (IStream &is,
|
|
const std::string &layerName,
|
|
int numThreads = globalThreadCount());
|
|
|
|
//-----------
|
|
// Destructor
|
|
//-----------
|
|
|
|
virtual ~TiledRgbaInputFile ();
|
|
|
|
|
|
//-----------------------------------------------------
|
|
// Define a frame buffer as the pixel data destination:
|
|
// Pixel (x, y) is at address
|
|
//
|
|
// base + x * xStride + y * yStride
|
|
//
|
|
//-----------------------------------------------------
|
|
|
|
void setFrameBuffer (Rgba *base,
|
|
size_t xStride,
|
|
size_t yStride);
|
|
|
|
//-------------------------------------------------------------------
|
|
// Switch to a different layer -- subsequent calls to readTile()
|
|
// and readTiles() will read channels layerName.R, layerName.G, etc.
|
|
// After each call to setLayerName(), setFrameBuffer() must be called
|
|
// at least once before the next call to readTile() or readTiles().
|
|
//-------------------------------------------------------------------
|
|
|
|
void setLayerName (const std::string &layerName);
|
|
|
|
|
|
//--------------------------
|
|
// Access to the file header
|
|
//--------------------------
|
|
|
|
const Header & header () const;
|
|
const FrameBuffer & frameBuffer () const;
|
|
const Imath::Box2i & displayWindow () const;
|
|
const Imath::Box2i & dataWindow () const;
|
|
float pixelAspectRatio () const;
|
|
const Imath::V2f screenWindowCenter () const;
|
|
float screenWindowWidth () const;
|
|
LineOrder lineOrder () const;
|
|
Compression compression () const;
|
|
RgbaChannels channels () const;
|
|
const char * fileName () const;
|
|
bool isComplete () const;
|
|
|
|
//----------------------------------
|
|
// Access to the file format version
|
|
//----------------------------------
|
|
|
|
int version () const;
|
|
|
|
|
|
//---------------------------------------------------
|
|
// Utility functions (same as in Imf::TiledInputFile)
|
|
//---------------------------------------------------
|
|
|
|
unsigned int tileXSize () const;
|
|
unsigned int tileYSize () const;
|
|
LevelMode levelMode () const;
|
|
LevelRoundingMode levelRoundingMode () const;
|
|
|
|
int numLevels () const;
|
|
int numXLevels () const;
|
|
int numYLevels () const;
|
|
bool isValidLevel (int lx, int ly) const;
|
|
|
|
int levelWidth (int lx) const;
|
|
int levelHeight (int ly) const;
|
|
|
|
int numXTiles (int lx = 0) const;
|
|
int numYTiles (int ly = 0) const;
|
|
|
|
Imath::Box2i dataWindowForLevel (int l = 0) const;
|
|
Imath::Box2i dataWindowForLevel (int lx, int ly) const;
|
|
|
|
Imath::Box2i dataWindowForTile (int dx, int dy,
|
|
int l = 0) const;
|
|
|
|
Imath::Box2i dataWindowForTile (int dx, int dy,
|
|
int lx, int ly) const;
|
|
|
|
|
|
//----------------------------------------------------------------
|
|
// Read pixel data:
|
|
//
|
|
// readTile(dx, dy, lx, ly) reads the tile with tile
|
|
// coordinates (dx, dy), and level number (lx, ly),
|
|
// and stores it in the current frame buffer.
|
|
//
|
|
// dx must lie in the interval [0, numXTiles(lx)-1]
|
|
// dy must lie in the interval [0, numYTiles(ly)-1]
|
|
//
|
|
// lx must lie in the interval [0, numXLevels()-1]
|
|
// ly must lie in the inverval [0, numYLevels()-1]
|
|
//
|
|
// readTile(dx, dy, level) is a convenience function used
|
|
// for ONE_LEVEL and MIPMAP_LEVELS files. It calls
|
|
// readTile(dx, dy, level, level).
|
|
//
|
|
// The two readTiles(dx1, dx2, dy1, dy2, ...) functions allow
|
|
// reading multiple tiles at once. If multi-threading is used
|
|
// multiple tiles are read concurrently.
|
|
//
|
|
// Pixels that are outside the pixel coordinate range for the
|
|
// tile's level, are never accessed by readTile().
|
|
//
|
|
// Attempting to access a tile that is not present in the file
|
|
// throws an InputExc exception.
|
|
//
|
|
//----------------------------------------------------------------
|
|
|
|
void readTile (int dx, int dy, int l = 0);
|
|
void readTile (int dx, int dy, int lx, int ly);
|
|
|
|
void readTiles (int dxMin, int dxMax,
|
|
int dyMin, int dyMax, int lx, int ly);
|
|
|
|
void readTiles (int dxMin, int dxMax,
|
|
int dyMin, int dyMax, int l = 0);
|
|
|
|
private:
|
|
|
|
//
|
|
// Copy constructor and assignment are not implemented
|
|
//
|
|
|
|
TiledRgbaInputFile (const TiledRgbaInputFile &);
|
|
TiledRgbaInputFile & operator = (const TiledRgbaInputFile &);
|
|
|
|
class FromYa;
|
|
|
|
TiledInputFile * _inputFile;
|
|
FromYa * _fromYa;
|
|
std::string _channelNamePrefix;
|
|
};
|
|
|
|
|
|
} // namespace Imf
|
|
|
|
#endif
|