110 lines
3.5 KiB
Plaintext
110 lines
3.5 KiB
Plaintext
#ifndef OSGUTIL_CUBEMAPGENERATOR_
|
|
#define OSGUTIL_CUBEMAPGENERATOR_
|
|
|
|
#include <osgUtil/Export>
|
|
|
|
#include <osg/Vec3>
|
|
#include <osg/Vec4>
|
|
#include <osg/CopyOp>
|
|
#include <osg/Referenced>
|
|
#include <osg/TextureCubeMap>
|
|
#include <osg/Image>
|
|
#include <osg/Notify>
|
|
|
|
#include <vector>
|
|
|
|
namespace osgUtil
|
|
{
|
|
|
|
/** This is the base class for cube map generators.
|
|
It exposes the necessary interface to access the six generated images;
|
|
descendants should only override the compute_color() method.
|
|
*/
|
|
class OSGUTIL_EXPORT CubeMapGenerator: public osg::Referenced {
|
|
public:
|
|
explicit CubeMapGenerator(int texture_size = 64);
|
|
CubeMapGenerator(const CubeMapGenerator ©, const osg::CopyOp ©op = osg::CopyOp::SHALLOW_COPY);
|
|
|
|
inline osg::Image *getImage(osg::TextureCubeMap::Face face);
|
|
inline const osg::Image *getImage(osg::TextureCubeMap::Face face) const;
|
|
|
|
/** generate the six cube images.
|
|
If use_osg_system is true, then the OSG's coordinate system is used instead
|
|
of the default OpenGL one.
|
|
*/
|
|
void generateMap(bool use_osg_system = true);
|
|
|
|
protected:
|
|
virtual ~CubeMapGenerator() {}
|
|
CubeMapGenerator &operator=(const CubeMapGenerator &) { return *this; }
|
|
|
|
inline void set_pixel(int index, int c, int r, const osg::Vec4 &color);
|
|
inline static osg::Vec4 vector_to_color(const osg::Vec3 &vec);
|
|
|
|
/** override this method to define how colors are computed.
|
|
The parameter R is the reflection vector, pointing from the center of the cube.
|
|
The return value should be the RGBA color associated with that reflection ray.
|
|
*/
|
|
virtual osg::Vec4 compute_color(const osg::Vec3 &R) const = 0;
|
|
|
|
private:
|
|
int texture_size_;
|
|
|
|
typedef std::vector<osg::ref_ptr<osg::Image> > Image_list;
|
|
Image_list images_;
|
|
};
|
|
|
|
// INLINE METHODS
|
|
|
|
inline osg::Image *CubeMapGenerator::getImage(osg::TextureCubeMap::Face face)
|
|
{
|
|
switch (face) {
|
|
case osg::TextureCubeMap::POSITIVE_X: return images_[0].get();
|
|
case osg::TextureCubeMap::NEGATIVE_X: return images_[1].get();
|
|
case osg::TextureCubeMap::POSITIVE_Y: return images_[2].get();
|
|
case osg::TextureCubeMap::NEGATIVE_Y: return images_[3].get();
|
|
case osg::TextureCubeMap::POSITIVE_Z: return images_[4].get();
|
|
case osg::TextureCubeMap::NEGATIVE_Z: return images_[5].get();
|
|
default: return 0;
|
|
}
|
|
}
|
|
|
|
inline const osg::Image *CubeMapGenerator::getImage(osg::TextureCubeMap::Face face) const
|
|
{
|
|
switch (face) {
|
|
case osg::TextureCubeMap::POSITIVE_X: return images_[0].get();
|
|
case osg::TextureCubeMap::NEGATIVE_X: return images_[1].get();
|
|
case osg::TextureCubeMap::POSITIVE_Y: return images_[2].get();
|
|
case osg::TextureCubeMap::NEGATIVE_Y: return images_[3].get();
|
|
case osg::TextureCubeMap::POSITIVE_Z: return images_[4].get();
|
|
case osg::TextureCubeMap::NEGATIVE_Z: return images_[5].get();
|
|
default: return 0;
|
|
}
|
|
}
|
|
|
|
inline void CubeMapGenerator::set_pixel(int index, int c, int r, const osg::Vec4 &color)
|
|
{
|
|
osg::Image *i = images_[index].get();
|
|
if (i) {
|
|
*(i->data(c, r)+0) = static_cast<unsigned char>(color.x() * 255);
|
|
*(i->data(c, r)+1) = static_cast<unsigned char>(color.y() * 255);
|
|
*(i->data(c, r)+2) = static_cast<unsigned char>(color.z() * 255);
|
|
*(i->data(c, r)+3) = static_cast<unsigned char>(color.w() * 255);
|
|
} else {
|
|
osg::notify(osg::WARN) << "Warning: CubeMapGenerator::set_pixel(): invalid image index\n";
|
|
}
|
|
}
|
|
|
|
inline osg::Vec4 CubeMapGenerator::vector_to_color(const osg::Vec3 &vec)
|
|
{
|
|
return osg::Vec4(
|
|
vec.x() / vec.length() / 2 + 0.5f,
|
|
vec.y() / vec.length() / 2 + 0.5f,
|
|
vec.z() / vec.length() / 2 + 0.5f,
|
|
1);
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|