Files
OpenSceneGraph/src/osgPlugins/logo/ReaderWriterLOGO.cpp
2016-06-10 16:36:13 +01:00

333 lines
11 KiB
C++

#include <stdio.h>
#include <osg/Geode>
#include <osg/Drawable>
#include <osg/BlendFunc>
#include <osg/StateSet>
#include <osg/Notify>
#include <osg/Viewport>
#include <osgDB/ReadFile>
#include <osgDB/FileUtils>
#include <osgDB/FileNameUtils>
#include <osgDB/Registry>
#include <osgDB/Input>
#include <osgDB/Output>
#include <osgUtil/CullVisitor>
using namespace osg;
using namespace osgDB;
class Logos: public osg::Drawable
{
public:
enum RelativePosition{
Center,
UpperLeft,
UpperRight,
LowerLeft,
LowerRight,
UpperCenter,
LowerCenter,
last_position
};
struct logosCullCallback : public osg::DrawableCullCallback
{
virtual bool cull(osg::NodeVisitor *visitor, osg::Drawable* drawable, osg::State*) const
{
Logos *logos = dynamic_cast <Logos *>(drawable);
osgUtil::CullVisitor *cv = visitor->asCullVisitor();
if (!cv) return true;
unsigned int contextID = cv->getState()!=0 ? cv->getState()->getContextID() : 0;
if(contextID != logos->getContextID())
{
// logo not appropriate for window assigned to the cull visitor so cull it.
return true;
}
if( logos != NULL && cv != NULL )
{
osg::Viewport *vp = cv->getViewport();
if( vp != NULL )
{
if( vp->width() != logos->getViewport()->width() ||
vp->height() != logos->getViewport()->height() )
{
logos->getViewport()->setViewport( vp->x(), vp->y(), vp->width(), vp->height() );
logos->dirtyDisplayList();
}
}
}
return false;
}
};
Logos()
{
osg::StateSet *sset = new osg::StateSet;
osg::BlendFunc *transp = new osg::BlendFunc;
transp->setFunction(GL_SRC_ALPHA,GL_ONE_MINUS_SRC_ALPHA);
sset->setAttribute( transp );
sset->setMode( GL_BLEND, osg::StateAttribute::ON );
sset->setMode( GL_DEPTH_TEST, osg::StateAttribute::OFF );
sset->setTextureMode( 0, GL_TEXTURE_2D, osg::StateAttribute::OFF );
#if 1
// for now we'll crudely set the bin number to 100 to force it to draw later and ontop of the scene
sset->setRenderBinDetails( 100 , "RenderBin" );
#else
sset->setRenderBinDetails( StateSet::TRANSPARENT_BIN + 1 , "RenderBin" );
#endif
setStateSet( sset );
_viewport = new osg::Viewport;
setCullCallback( new logosCullCallback );
_contextID = 0;
}
Logos(const Logos& logo, const CopyOp& copyop=CopyOp::SHALLOW_COPY) :Drawable( logo, copyop ) {}
virtual Object* cloneType() const { return new Logos(); }
virtual Object* clone( const CopyOp& copyop) const { return new Logos(*this, copyop ); }
virtual bool isSameKindAs(const Object* obj) const { return dynamic_cast<const Logos*>(obj)!=NULL; }
virtual const char* className() const { return "Logos"; }
virtual void drawImplementation(osg::RenderInfo& renderInfo) const
{
#if !defined(OSG_GLES1_AVAILABLE) && !defined(OSG_GLES2_AVAILABLE) && !defined(OSG_GL3_AVAILABLE)
if( renderInfo.getContextID() != _contextID )
return;
float vx = 0.0f;
float vy = 0.0f;
float vw = 1.0f;
float vh = 1.0f;
if (_viewport.valid())
{
vx = _viewport->x();
vy = _viewport->y();
vw = _viewport->width();
vh = _viewport->height();
}
glMatrixMode( GL_PROJECTION );
glPushMatrix();
glLoadIdentity();
glOrtho( 0.0, vw, 0.0, vh, -1.0, 1.0 );
glMatrixMode( GL_MODELVIEW );
glPushMatrix();
glLoadIdentity();
glColor4f( 1.0f, 1.0f, 1.0f, 1.0f );
Images::const_iterator p;
float th = 0.0;
for( p = _logos[Center].begin(); p != _logos[Center].end(); p++ )
{
th += (*p)->t();
}
float place[][4] = {
{ vw*0.5f, ((vh*0.5f) + th*0.5f), -0.5f, -1.0f },
{ vx, vh, 0.0f, -1.0f },
{ vw, vh, -1.0f, -1.0f },
{ vx, vy, 0.0f, 1.0f },
{ vw, vy, -1.0f, 1.0f },
{ vw*0.5f, vh , -0.5f, -1.0f },
{ vw*0.5f, 0.0f , -0.5f, 1.0f },
};
for( int i = Center; i < last_position; i++ )
{
if( _logos[i].size() != 0 )
{
float x = place[i][0];
float y = place[i][1];
float xi = place[i][2];
float yi = place[i][3];
for( p = _logos[i].begin(); p != _logos[i].end(); p++ )
{
osg::Image *img = (*p).get();
glPixelStorei(GL_UNPACK_ALIGNMENT, img->getPacking());
glPixelStorei(GL_UNPACK_ROW_LENGTH, img->getRowLength());
x = place[i][0] + xi * img->s();
if( i == Center || i == UpperLeft || i == UpperRight || i == UpperCenter)
y += yi * img->t();
glRasterPos2f( x, y );
glDrawPixels( img->s(), img->t(), img->getPixelFormat(), img->getDataType(), img->data() );
if( i == LowerLeft || i == LowerRight || i == LowerCenter)
y += yi * img->t();
}
}
}
glPopMatrix();
glMatrixMode( GL_PROJECTION );
glPopMatrix();
glMatrixMode( GL_MODELVIEW );
#else
OSG_NOTICE<<"Warning: Logos::drawImplementation(..) not supported."<<std::endl;
#endif
}
void addLogo( RelativePosition pos, std::string name )
{
osg::ref_ptr<osg::Image> image = osgDB::readRefImageFile( name.c_str() );
if( image.valid())
{
_logos[pos].push_back( image );
}
else
{
OSG_WARN<< "Logos::addLogo image file not found : " << name << ".\n";
}
}
osg::Viewport *getViewport() { return _viewport.get(); }
void setContextID( unsigned int id ) { _contextID = id; }
unsigned int getContextID() { return _contextID; }
bool hasLogos()
{
int n = 0;
for( int i = Center; i < last_position; i++ )
n += _logos[i].size();
return (n != 0);
}
virtual osg::BoundingBox computeBoundingBox() const
{
return osg::BoundingBox( -1, -1, -1, 1, 1, 1);
}
protected:
Logos& operator = (const Logos&) { return *this;}
virtual ~Logos() {}
private :
typedef std::vector < osg::ref_ptr<osg::Image> > Images;
Images _logos[last_position];
osg::ref_ptr<osg::Viewport> _viewport;
unsigned int _contextID;
};
class LOGOReaderWriter : public osgDB::ReaderWriter
{
public:
LOGOReaderWriter()
{
supportsExtension("logo","Ascii logo placement format");
}
virtual const char* className() const { return "Logo Database Reader/Writer"; }
virtual ReadResult readNode(const std::string& file, const osgDB::ReaderWriter::Options* options) const
{
std::string ext = osgDB::getLowerCaseFileExtension(file);
if (!acceptsExtension(ext)) return ReadResult::FILE_NOT_HANDLED;
std::string fileName = osgDB::findDataFile( file, options );
if (fileName.empty())
return ReadResult::FILE_NOT_FOUND;
OSG_INFO<< "ReaderWriterLOGO::readNode( "<<fileName.c_str()<<" )\n";
std::string filePath = osgDB::getFilePath(fileName);
if (!filePath.empty()) {
OSG_DEBUG<< "Adding : " << filePath << " to the file data path\n";
osgDB::getDataFilePathList().push_back(filePath);
}
osg::ref_ptr<osg::Geode> geode = new osg::Geode;
unsigned int screen = 0;
Logos* ld = new Logos;
ld->setContextID( screen );
Logos::RelativePosition pos = Logos::LowerRight;
FILE *fp;
if( (fp = osgDB::fopen( fileName.c_str(), "r")) == NULL )
return NULL;
while( !feof(fp))
{
char buff[128];
if( fscanf( fp, "%s", buff ) != 1 )
break;
std::string str(buff);
if( str == "Center" )
pos = Logos::Center;
else if( str == "UpperLeft" )
pos = Logos::UpperLeft;
else if( str == "UpperRight" )
pos = Logos::UpperRight;
else if( str == "LowerLeft" )
pos = Logos::LowerLeft;
else if( str == "LowerRight" )
pos = Logos::LowerRight;
else if( str == "UpperCenter" )
pos = Logos::UpperCenter;
else if( str == "LowerCenter" )
pos = Logos::LowerCenter;
else if( str == "Camera" )
{
int tn;
if( (fscanf( fp, "%d", &tn )) != 1 )
{
OSG_WARN << "Error... Camera requires an integer argument\n";
break;
}
if (tn < 0)
{
OSG_WARN << "Error... Camera requires an positive or null value argument\n";
break;
}
unsigned int n = static_cast<unsigned int>(tn);
if( screen != n )
{
screen = n;
if( ld->hasLogos() )
{
geode->addDrawable( ld );
ld = new Logos;
ld->setContextID( screen );
}
else
ld->setContextID( screen );
}
}
else
{
if( str.length() )
ld->addLogo( pos, str );
}
}
fclose( fp );
if( ld->hasLogos() )
geode->addDrawable( ld );
geode->setCullingActive(false);
return geode;
}
};
// now register with Registry to instantiate the above
// reader/writer.
REGISTER_OSGPLUGIN(logo, LOGOReaderWriter)