source: trunk/packages/vizservers/nanovis/ParticleSystemFactory.cpp @ 2798

/* -*- mode: c++; c-basic-offset: 4; indent-tabs-mode: nil -*- */
#include "ParticleSystemFactory.h"
#include "ParticleSystem.h"
#include <vrutil/vrFilePath.h>
#include <stdio.h>
#include "Trace.h"

#ifdef _WIN32
#include <expat/expat.h>
#else
#include <expat.h>
#endif

#include <string.h>

#ifdef WIN32
#pragma warning (disable : 4996)
#endif

#define BUFSIZE 4096

void ParticleSystemFactory::text(void *data, const XML_Char *txt, int len)
{
}  

void ParticleSystemFactory::startElement(void *userData, const char *elementName, const char **attrs)
{
        ParticleSystemFactory* This = reinterpret_cast<ParticleSystemFactory*>(userData);

        if (!strcmp(elementName, "particle-sys-info"))
        {
                This->parseParticleSysInfo(attrs);
        }
        else if (!strcmp(elementName, "emitter"))
        {
                This->parseEmitterInfo(attrs);
        }
}

void ParticleSystemFactory::endElement(void *userData, const char *name)
{
}

ParticleSystemFactory::ParticleSystemFactory()
: _newParticleSystem(0)
{
}

ParticleSystemFactory::~ParticleSystemFactory()
{
}

ParticleSystem* ParticleSystemFactory::create(const std::string& fileName)
{
        FILE* fp = fopen(fileName.c_str(), "rb");

        if (fp == 0) {
                return 0;
        }
        XML_Parser parser = XML_ParserCreate(NULL);

        XML_SetUserData(parser, this);
        XML_SetElementHandler(parser, startElement, endElement);

        size_t stat;
        size_t cnt;

        while(! feof(fp))
        {
                void *buff = XML_GetBuffer(parser, BUFSIZE);
                if (! buff)
                {
                        break;
                }
                cnt = fread(buff, 1, BUFSIZE, fp);
                stat = XML_ParseBuffer(parser, (int) cnt, 0);
                if (!stat)
                {
                        //TRACE("Parse error at line %d\n", XML_GetCurrentLineNumber(parser));
                        break;
                }
        }

        fclose(fp);

        XML_ParserFree(parser);

        return _newParticleSystem;
}

void ParticleSystemFactory::parseParticleSysInfo(const char** attrs)
{
        int width = 2, height = 2; // default
        std::string fileName = "J-wire-vec.dx";
        float pointSize = -1.0f;
        int numOfUsedParticles = -1;
        bool sortEnabled = false;
        bool glypEnabled = false;
        bool bboxVisible = false;
        bool advectionEnabled = false;
        bool streamlineEnabled = false;
        bool timeVaryingData = false;
        int fieldWidth = 1;
        int fieldHeight = 1;
        int fieldDepth = 1;
        // TBD..
        //float timeSeries_vel_mag_min;
        //float timeSeries_vel_mag_max;
        //float timeSeriesVelMagMax;
        int startIndex = -1, endIndex = -1;
        for (int i = 0; attrs[i]; i += 2) {
                if (!strcmp(attrs[i], "rendertarget-width"))
                {
                        width = atoi(attrs[i + 1]);
                }
                else if (!strcmp(attrs[i], "rendertarget-height"))
                {
                        height = atoi(attrs[i + 1]);
                }
                else if (!strcmp(attrs[i], "particle-point-size"))
                {
                        pointSize = (float) atof(attrs[i + 1]);
                }
                else if (!strcmp(attrs[i], "vector-field-x"))
                {
                        fieldWidth = (float) atof(attrs[i + 1]);
                }
                else if (!strcmp(attrs[i], "vector-field-y"))
                {
                        fieldHeight = (float) atof(attrs[i + 1]);
                }
                else if (!strcmp(attrs[i], "vector-field-z"))
                {
                        fieldDepth = (float) atof(attrs[i + 1]);
                }
                else if (!strcmp(attrs[i], "sort-enabled"))
                {
                        if (!strcmp(attrs[i + 1], "true"))
                                sortEnabled = true;
                }
                else if (!strcmp(attrs[i], "glyp-enabled"))
                {
                        if (!strcmp(attrs[i + 1], "true"))
                                glypEnabled = true;
                }
                else if (!strcmp(attrs[i], "bbox-draw-enabled"))
                {
                        if (!strcmp(attrs[i + 1], "true"))
                                bboxVisible = true;
                }
                else if (!strcmp(attrs[i], "advection-enabled"))
                {
                        if (!strcmp(attrs[i + 1], "true"))
                                advectionEnabled = true;
                }
                else if (!strcmp(attrs[i], "stream-line-enabled"))
                {
                        if (!strcmp(attrs[i + 1], "true"))
                                streamlineEnabled = true;
                }
                else if (!strcmp(attrs[i], "vector-field"))
                {
                        fileName = attrs[i + 1];
                }
                else if (!strcmp(attrs[i], "vector-field"))
                {
                        if (!strcmp(attrs[i + 1], "true"))
                                timeVaryingData = true;
                }
                else if (!strcmp(attrs[i], "particle-user-num"))
                {
                        numOfUsedParticles = atoi(attrs[i + 1]);
                }
                else if (!strcmp(attrs[i], "time-series-start-index"))
                {
                        startIndex = atoi(attrs[i + 1]);
                }
                else if (!strcmp(attrs[i], "time-series-end-index"))
                {
                        endIndex = atoi(attrs[i + 1]);
                }
                else if (!strcmp(attrs[i], "time-varying"))
                {
                        if (!strcmp(attrs[i + 1], "true"))
                                timeVaryingData = true;
                }
                
        }

        if (timeVaryingData)
        {
                char buff[256];
                sprintf(buff, fileName.c_str(), startIndex);
                std::string path = vrFilePath::getInstance()->getPath(buff);
                if (path.size())
                {
                        std::string dir;
                        int index = path.rfind('/');
                        if (index == -1) {
                                index = path.rfind('\\');
                                if (index == -1)
                                        TRACE("file not found\n");
                        }

                        dir = path.substr(0, index + 1);
                        path = dir + fileName;
                
                        _newParticleSystem = new ParticleSystem(width, height, path.c_str(), fieldWidth, fieldHeight, fieldDepth, timeVaryingData, startIndex, endIndex);
                }
                else
                {
                        _newParticleSystem = new ParticleSystem(width, height, fileName.c_str(), fieldWidth, fieldHeight, fieldDepth, timeVaryingData, startIndex, endIndex);
                }
        }
        else
        {
                std::string path = vrFilePath::getInstance()->getPath(fileName.c_str());
                _newParticleSystem = new ParticleSystem(width, height, path.c_str(), fieldWidth, fieldHeight, fieldDepth, timeVaryingData, startIndex, endIndex);
        }

        if (pointSize != -1.0f) _newParticleSystem->setDefaultPointSize(pointSize);
        if (sortEnabled) _newParticleSystem->enable(ParticleSystem::PS_SORT);
        if (glypEnabled) _newParticleSystem->enable(ParticleSystem::PS_GLYPE);
        if (bboxVisible) _newParticleSystem->enable(ParticleSystem::PS_DRAW_BBOX);
        if (advectionEnabled) _newParticleSystem->enable(ParticleSystem::PS_ADVECTION);
        if (streamlineEnabled) _newParticleSystem->enable(ParticleSystem::PS_STREAMLINE);
        if (numOfUsedParticles != -1) _newParticleSystem->setUserDefinedNumOfParticles(numOfUsedParticles);
}

void ParticleSystemFactory::parseEmitterInfo(const char** attrs)
{
        ParticleEmitter* emitter = new ParticleEmitter;
        for (int i = 0; attrs[i]; i += 2) {
                if (!strcmp(attrs[i], "max-position-offset"))
                {
                        float x = 0, y = 0, z = 0;
                        sscanf(attrs[i+1], "%f%f%f",&x, &y, &z);
                        emitter->setMaxPositionOffset(x, y, z);
                }
                else if (!strcmp(attrs[i], "position"))
                {
                        float x = 0, y = 0, z = 0;
                        sscanf(attrs[i+1], "%f%f%f",&x, &y, &z);
                        emitter->setPosition(x, y, z);
                }
                else if (!strcmp(attrs[i], "min-max-life-time"))
                {
                        float min = 0, max = 0;
                        sscanf(attrs[i+1], "%f%f",&min, &max);
                        emitter->setMinMaxLifeTime(min, max);
                }
                else if (!strcmp(attrs[i], "min-max-new-particles"))
                {
                        int min = 0, max = 0;
                        sscanf(attrs[i+1], "%d%d",&min, &max);
                        emitter->setMinMaxNumOfNewParticles(min, max);
                }
                else if (!strcmp(attrs[i], "enabled"))
                {
                        if (!strcmp(attrs[i + 1], "true"))
                                emitter->setEnabled(true);
                        else
                                emitter->setEnabled(false);
                }
        }
        _newParticleSystem->addEmitter(emitter);
}