ExportVRML.cpp

Go to the documentation of this file.

#include "ExportVRML.h"
 
#include <ArmarXCore/core/logging/Logging.h>
 
#include <Inventor/VRMLnodes/SoVRMLGroup.h>
#include <Inventor/actions/SoToVRML2Action.h>
#include <Inventor/actions/SoWriteAction.h>
#include <Inventor/nodes/SoFile.h>
#include <Inventor/nodes/SoSeparator.h>
 
namespace armarx::viz::coin
{
 
    static SoGroup*
    convertSoFileChildren(SoGroup* orig)
    {
        if (!orig)
        {
            return new SoGroup;
        }
 
        SoGroup* storeResult;
 
        if (orig->getTypeId() == SoSeparator::getClassTypeId())
        {
            storeResult = new SoSeparator;
        }
        else
        {
            storeResult = new SoGroup;
        }
 
        storeResult->ref();
 
        if (orig->getTypeId().isDerivedFrom(SoGroup::getClassTypeId()))
        {
            // process group node
            for (int i = 0; i < orig->getNumChildren(); i++)
            {
                SoNode* n1 = orig->getChild(i);
 
                if (n1->getTypeId().isDerivedFrom(SoGroup::getClassTypeId()))
                {
                    // convert group
                    SoGroup* n2 = (SoGroup*)n1;
                    SoGroup* gr1 = convertSoFileChildren(n2);
                    storeResult->addChild(gr1);
                }
                else if (n1->getTypeId() == SoFile::getClassTypeId())
                {
                    // really load file!!
                    SoFile* fn = (SoFile*)n1;
                    SoGroup* fileChildren;
                    fileChildren = fn->copyChildren();
                    storeResult->addChild(fileChildren);
                }
                else
                {
                    // just copy child node
                    storeResult->addChild(n1);
                }
            }
        }
 
        storeResult->unrefNoDelete();
        return storeResult;
    }
 
    void
    exportToVRML(SoNode* node, std::string const& exportFilePath)
    {
        SoOutput* so = new SoOutput();
        if (!so->openFile(exportFilePath.c_str()))
        {
            ARMARX_ERROR << "Could not open file " << exportFilePath << " for writing.";
            return;
        }
 
        so->setHeaderString("#VRML V2.0 utf8");
 
        SoGroup* n = new SoGroup;
        n->ref();
        n->addChild(node);
        SoGroup* newVisu = convertSoFileChildren(n);
        newVisu->ref();
 
        SoToVRML2Action tovrml2;
        tovrml2.apply(newVisu);
        SoVRMLGroup* newroot = tovrml2.getVRML2SceneGraph();
        newroot->ref();
        SoWriteAction wra(so);
        wra.apply(newroot);
        newroot->unref();
 
        so->closeFile();
 
        newVisu->unref();
        n->unref();
    }
 
} // namespace armarx::viz::coin