#include <OpenThreads/ScopedLock>
-#include <osg/observer_ptr>
#include <osg/ref_ptr>
#include <osg/Group>
#include <osg/NodeCallback>
#include <simgear/props/condition.hxx>
#include "BoundingVolumeBuildVisitor.hxx"
+#include "model.hxx"
+#include "SGReaderWriterXMLOptions.hxx"
using namespace std;
using namespace osg;
using namespace osgDB;
using namespace simgear;
-using OpenThreads::ReentrantMutex;
-using OpenThreads::ScopedLock;
-
// Little helper class that holds an extra reference to a
// loaded 3d model.
// Since we clone all structural nodes from our 3d models,
}
};
-// Change the StateSets of a model to hold different textures based on
-// a livery path.
-
-class TextureUpdateVisitor : public NodeAndDrawableVisitor {
-public:
- TextureUpdateVisitor(const FilePathList& pathList) :
- NodeAndDrawableVisitor(NodeVisitor::TRAVERSE_ALL_CHILDREN),
- _pathList(pathList)
- {
- }
-
- virtual void apply(Node& node)
- {
- StateSet* stateSet = cloneStateSet(node.getStateSet());
- if (stateSet)
- node.setStateSet(stateSet);
- traverse(node);
- }
-
- virtual void apply(Drawable& drawable)
- {
- StateSet* stateSet = cloneStateSet(drawable.getStateSet());
- if (stateSet)
- drawable.setStateSet(stateSet);
- }
- // Copied from Mathias' earlier SGTextureUpdateVisitor
-protected:
- Texture2D* textureReplace(int unit, const StateAttribute* attr)
- {
- const Texture2D* texture = dynamic_cast<const Texture2D*>(attr);
-
- if (!texture)
- return 0;
-
- const Image* image = texture->getImage();
- const string* fullFilePath = 0;
- if (image) {
- // The currently loaded file name
- fullFilePath = &image->getFileName();
-
- } else {
- fullFilePath = &texture->getName();
- }
- // The short name
- string fileName = getSimpleFileName(*fullFilePath);
- if (fileName.empty())
- return 0;
- // The name that should be found with the current database path
- string fullLiveryFile = findFileInPath(fileName, _pathList);
- // If it is empty or they are identical then there is nothing to do
- if (fullLiveryFile.empty() || fullLiveryFile == *fullFilePath)
- return 0;
- Image* newImage = readImageFile(fullLiveryFile);
- if (!newImage)
- return 0;
- CopyOp copyOp(CopyOp::DEEP_COPY_ALL & ~CopyOp::DEEP_COPY_IMAGES);
- Texture2D* newTexture = static_cast<Texture2D*>(copyOp(texture));
- if (!newTexture) {
- return 0;
- } else {
- newTexture->setImage(newImage);
- return newTexture;
- }
- }
-
- StateSet* cloneStateSet(const StateSet* stateSet)
- {
- typedef pair<int, Texture2D*> Tex2D;
- vector<Tex2D> newTextures;
- StateSet* result = 0;
-
- if (!stateSet)
- return 0;
- int numUnits = stateSet->getTextureAttributeList().size();
- if (numUnits > 0) {
- for (int i = 0; i < numUnits; ++i) {
- const StateAttribute* attr
- = stateSet->getTextureAttribute(i, StateAttribute::TEXTURE);
- Texture2D* newTexture = textureReplace(i, attr);
- if (newTexture)
- newTextures.push_back(Tex2D(i, newTexture));
- }
- if (!newTextures.empty()) {
- result = static_cast<StateSet*>(stateSet->clone(CopyOp()));
- for (vector<Tex2D>::iterator i = newTextures.begin();
- i != newTextures.end();
- ++i) {
- result->setTextureAttribute(i->first, i->second);
- }
- }
- }
- return result;
- }
-private:
- FilePathList _pathList;
-};
-
-// Create new userdata structs in a copied model.
-// The BVH trees are shared with the original model, but the velocity fields
-// should usually be distinct fields for distinct models.
-class UserDataCopyVisitor : public osg::NodeVisitor {
-public:
- UserDataCopyVisitor() :
- osg::NodeVisitor(osg::NodeVisitor::NODE_VISITOR,
- osg::NodeVisitor::TRAVERSE_ALL_CHILDREN)
- {
- }
- virtual void apply(osg::Node& node)
- {
- osg::ref_ptr<SGSceneUserData> userData;
- userData = SGSceneUserData::getSceneUserData(&node);
- if (userData.valid()) {
- SGSceneUserData* newUserData = new SGSceneUserData(*userData);
- newUserData->setVelocity(0);
- node.setUserData(newUserData);
- }
- node.traverse(*this);
- }
-};
class SGTexCompressionVisitor : public SGTextureStateAttributeVisitor {
public:
ModelRegistry::readImage(const string& fileName,
const ReaderWriter::Options* opt)
{
- ScopedLock<ReentrantMutex> lock(readerMutex);
CallbackMap::iterator iter
= imageCallbackMap.find(getFileExtension(fileName));
- // XXX Workaround for OSG plugin bug
{
if (iter != imageCallbackMap.end() && iter->second.valid())
return iter->second->readImage(fileName, opt);
- string absFileName = findDataFile(fileName, opt);
+ string absFileName = SGModelLib::findDataFile(fileName, opt);
if (!fileExists(absFileName)) {
SG_LOG(SG_IO, SG_ALERT, "Cannot find image file \""
<< fileName << "\"");
}
if (res.loadedFromCache())
- SG_LOG(SG_IO, SG_INFO, "Returning cached image \""
+ SG_LOG(SG_IO, SG_BULK, "Returning cached image \""
<< res.getImage()->getFileName() << "\"");
else
- SG_LOG(SG_IO, SG_INFO, "Reading image \""
+ SG_LOG(SG_IO, SG_BULK, "Reading image \""
<< res.getImage()->getFileName() << "\"");
return res;
osg::Node* cached
= dynamic_cast<Node*>(registry->getFromObjectCache(fileName));
if (cached)
- SG_LOG(SG_IO, SG_INFO, "Got cached model \""
+ SG_LOG(SG_IO, SG_BULK, "Got cached model \""
<< fileName << "\"");
else
- SG_LOG(SG_IO, SG_INFO, "Reading model \""
+ SG_LOG(SG_IO, SG_BULK, "Reading model \""
<< fileName << "\"");
return cached;
}
return node;
}
-osg::Node* DefaultCopyPolicy::copy(osg::Node* model, const string& fileName,
- const osgDB::ReaderWriter::Options* opt)
-{
- // Add an extra reference to the model stored in the database.
- // That is to avoid expiring the object from the cache even if it is still
- // in use. Note that the object cache will think that a model is unused
- // if the reference count is 1. If we clone all structural nodes here
- // we need that extra reference to the original object
- SGDatabaseReference* databaseReference;
- databaseReference = new SGDatabaseReference(model);
- CopyOp::CopyFlags flags = CopyOp::DEEP_COPY_ALL;
- flags &= ~CopyOp::DEEP_COPY_TEXTURES;
- flags &= ~CopyOp::DEEP_COPY_IMAGES;
- flags &= ~CopyOp::DEEP_COPY_STATESETS;
- flags &= ~CopyOp::DEEP_COPY_STATEATTRIBUTES;
- flags &= ~CopyOp::DEEP_COPY_ARRAYS;
- flags &= ~CopyOp::DEEP_COPY_PRIMITIVES;
- // This will safe display lists ...
- flags &= ~CopyOp::DEEP_COPY_DRAWABLES;
- flags &= ~CopyOp::DEEP_COPY_SHAPES;
- osg::Node* res = CopyOp(flags)(model);
- res->addObserver(databaseReference);
-
- // Update liveries
- TextureUpdateVisitor liveryUpdate(opt->getDatabasePathList());
- res->accept(liveryUpdate);
-
- // Copy the userdata fields, still sharing the boundingvolumes,
- // but introducing new data for velocities.
- UserDataCopyVisitor userDataCopyVisitor;
- res->accept(userDataCopyVisitor);
-
- return res;
-}
-
string OSGSubstitutePolicy::substitute(const string& name,
const ReaderWriter::Options* opt)
{
string fileSansExtension = getNameLessExtension(name);
string osgFileName = fileSansExtension + ".osg";
- string absFileName = findDataFile(osgFileName, opt);
+ string absFileName = SGModelLib::findDataFile(osgFileName, opt);
return absFileName;
}
void
BuildLeafBVHPolicy::buildBVH(const std::string& fileName, osg::Node* node)
{
- SG_LOG(SG_IO, SG_INFO, "Building leaf attached boundingvolume tree for \""
+ SG_LOG(SG_IO, SG_BULK, "Building leaf attached boundingvolume tree for \""
<< fileName << "\".");
BoundingVolumeBuildVisitor bvBuilder(true);
node->accept(bvBuilder);
void
BuildGroupBVHPolicy::buildBVH(const std::string& fileName, osg::Node* node)
{
- SG_LOG(SG_IO, SG_INFO, "Building group attached boundingvolume tree for \""
+ SG_LOG(SG_IO, SG_BULK, "Building group attached boundingvolume tree for \""
<< fileName << "\".");
BoundingVolumeBuildVisitor bvBuilder(false);
node->accept(bvBuilder);
void
NoBuildBVHPolicy::buildBVH(const std::string& fileName, osg::Node*)
{
- SG_LOG(SG_IO, SG_INFO, "Omitting boundingvolume tree for \""
+ SG_LOG(SG_IO, SG_BULK, "Omitting boundingvolume tree for \""
<< fileName << "\".");
}
ModelRegistry::readNode(const string& fileName,
const ReaderWriter::Options* opt)
{
- ScopedLock<ReentrantMutex> lock(readerMutex);
-
- // XXX Workaround for OSG plugin bug.
- Registry* registry = Registry::instance();
ReaderWriter::ReadResult res;
CallbackMap::iterator iter
= nodeCallbackMap.find(getFileExtension(fileName));
static SGReadCallbackInstaller readCallbackInstaller;
-// we get optimal geometry from the loader.
+// we get optimal geometry from the loader (Hah!).
struct ACOptimizePolicy : public OptimizeModelPolicy {
ACOptimizePolicy(const string& extension) :
OptimizeModelPolicy(extension)
&& group->getNumChildren() == 1)
optimized = static_cast<Node*>(group->getChild(0));
}
+ const SGReaderWriterXMLOptions* sgopt
+ = dynamic_cast<const SGReaderWriterXMLOptions*>(opt);
+ if (sgopt && sgopt->getInstantiateEffects())
+ optimized = instantiateEffects(optimized.get(), sgopt);
return optimized.release();
}
};
};
typedef ModelRegistryCallback<ACProcessPolicy, DefaultCachePolicy,
- ACOptimizePolicy, DefaultCopyPolicy,
+ ACOptimizePolicy,
OSGSubstitutePolicy, BuildLeafBVHPolicy>
ACCallback;