source: branches/1.2/packages/vizservers/nanovis/NvVectorField.h @ 4160

/* -*- mode: c++; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
 * Copyright (c) 2004-2013  HUBzero Foundation, LLC
 *
 */
#ifndef NV_VECTOR_FIELD_H
#define NV_VECTOR_FIELD_H

#include <string>
#include <map>

#include <vrmath/Vector3f.h>
#include <vrmath/Vector4f.h>

#include "Volume.h"
#include "NvParticleRenderer.h"

class NvDeviceShape
{
public:
    NvDeviceShape() :
        visible(true)
    {
    }

    vrmath::Vector3f min;
    vrmath::Vector3f max;
    vrmath::Vector4f color;
    bool visible;
};

class NvVectorField
{
public :
    NvVectorField();

    ~NvVectorField();

    void setVectorField(Volume *vol, const vrmath::Vector3f& origin,
                        float scaleX, float scaleY, float scaleZ, float max);

    bool active()
    {
        return _activated;
    }

    void active(bool state)
    {
        _activated = state;
    }

    void activateDeviceShape()
    {
        _deviceVisible = true;
    }

    void deactivateDeviceShape()
    {
        _deviceVisible = false;
    }

    void addDeviceShape(const std::string& name, const NvDeviceShape& shape);

    void removeDeviceShape(const std::string& name);

    void activateDeviceShape(const std::string& name);

    void deactivateDeviceShape(const std::string& name);

    void initialize();

    void reset();
    
    void addPlane(const std::string& name);

    void removePlane(const std::string& name);
    
    void advect();

    void render();
    
    void drawDeviceShape();

    void activatePlane(const std::string& name);

    void deactivatePlane(const std::string& name);

    void setPlaneAxis(const std::string& name, int axis);

    void setPlanePos(const std::string& name, float pos);

    void setParticleColor(const std::string& name, float r, float g, float b, float a);

    void setParticleColor(const std::string& name, const vrmath::Vector4f& color);

private:
    GLuint _vectorFieldId;
    Volume *_volPtr;
    std::map<std::string, NvParticleRenderer *> _particleRendererMap;
    std::map<std::string, NvDeviceShape> _shapeMap;

    /**
     * @brief Specify the visibility
     */
    bool _activated;

    vrmath::Vector3f _origin;
    float _scaleX;
    float _scaleY;
    float _scaleZ;
    float _max;

    bool _deviceVisible;
};

#endif