/cygdrive/d/src/svn/vrut/trunk/modules/iovrml/iovrml.cpp

Go to the documentation of this file.

/*
 *
 * Description  : Reads and writes VRML files
 * Author       : Michael Smutny smutnm2@fel.cvut.cz & Hana Truskova truskh1@fel.cvut.cz
 *
 * Purpose      :
 *     Imports and exports scenes in VRML files.
 */


#include <wx/stopwatch.h>
#include <wx/tokenzr.h>
#include <wx/zstream.h>
#include "iovrml.h"
#include "vrmlparser.h"
#include "vrmlwriter.h"

using namespace VRUT;


IOVrmlModule::IOVrmlModule(const MODULE_ID & _id, const wxString & _name, EventHandler * msgSink)
              : IOModule(_id, _name, 0, msgSink)
{
       teselate = true;
       REGISTER_PARAM_GUI_CHECKBOX(teselateParamID, wxT("teselate"), wxT("1"), wxT("teselate"));
       hFlag = true;
       REGISTER_PARAM_GUI_CHECKBOX(hierarchyFlagParamID, wxT("hFlag"), wxT("1"), wxT("Hierarchy flag."));
       extraCams = true;
       REGISTER_PARAM_GUI_CHECKBOX(extraCamsParamID, wxT("extraCams"), wxT("1"), wxT("Export extra cameras"));
}


IOVrmlModule::~IOVrmlModule()
{
}

void IOVrmlModule::processEvent(wxCommandEvent & evt)
{
       IOModule::processEvent(evt);
       switch (evt.GetEventType())
       {
       case Event::EVT_PARAM_SET:
              {
                     UPDATE_PARAM_FROM_EVENT_BOOL(teselateParamID, teselate, evt);
                     UPDATE_PARAM_FROM_EVENT_BOOL(hierarchyFlagParamID, hFlag, evt);
                     UPDATE_PARAM_FROM_EVENT_BOOL(extraCamsParamID, extraCams, evt);
              }
              break;
       default:
              break;
       }
}

wxString IOVrmlModule::GetDesc() const
{
       return wxT("Import and export VRML files with scene hierarchy and geometry data");
}


wxString IOVrmlModule::GetSupportedExts() const
{
       return wxT("WRL");
}


bool IOVrmlModule::ImportScene(const wxString & fname, SCENE_ID _sceneID, const wxString & rootUid)
{
       wxStopWatch sw;
       bool ret = false;
       wxInputStream * is = GetInputStream(fname);
       if (is)
       {
           unsigned char buffer[2];
           is -> Read(buffer, sizeof(unsigned char)*2);
           if ( buffer[0] == 31 && buffer[1] == 139){
            wxFileName wxFname(fname);
            delete is;
            is = new wxFileInputStream(wxFname.GetFullPath());
            is = new wxZlibInputStream(is,wxZLIB_GZIP);
            VRMLParser vrmlParser2(is, _sceneID, fname, this);
                      ret = vrmlParser2.Parse(rootUid, teselate, NULL);
                      delete is;
        }
           else {
             is -> Ungetch(buffer,  sizeof(unsigned char)*2);
        VRMLParser vrmlParser(is, _sceneID, fname, this);
        ret = vrmlParser.Parse(rootUid, teselate, NULL);
                  delete is;
           }
       }
       LOG(wxString::Format(wxT("<VRMLParser>Scene parsed in %.3f secs, ID %i"), 0.001f * sw.Time(), _sceneID));
       wxCommandEvent ev = Event::GET_EVT_IO_SCENE_IMPORT_DONE(_sceneID);
       PostToKernel(ev);
       return ret;
}


bool IOVrmlModule::ExportScene(const wxString & fname, const Scene * scene)
{
//     if(hFlag == 1) return false;
       LOG(wxT("<IOVrmlModule>Inside IOVrmlModule::ExportScene"));
       wxOutputStream * sceneStream = new wxFileOutputStream(fname);
       if (sceneStream && sceneStream->IsOk())
       {
              wxBufferedOutputStream *sceneStream2=new wxBufferedOutputStream(*sceneStream);
              VRMLWriter vrmlWriter(sceneStream2, scene, fname);
              vrmlWriter.Write(hFlag, extraCams);
              sceneStream2->Sync();
              SAFE_DELETE(sceneStream2);
              return true;
       }
       SAFE_DELETE(sceneStream);
       return false;
}

---