src/renderer/streamgrid.h

#ifndef STREAMGRID_H
#define STREAMGRID_H

#include "streamline.h"
#include "vec3.h"

#include "FlowGeometry.h"
#include "FlowChannel.h"

#include <queue>
#include <vector>

class StreamGrid
{
    friend class Streamline;

public:
    StreamGrid(int x, int y, int sep, float test, float step, FlowGeometry geo, FlowChannel* chanX, FlowChannel* chanY, bool rk);
    ~StreamGrid();

    void calculateGrid(void);

    std::vector<Streamline*> getAllStreamlines(void);

    void addStreamline(const vec3& seed);
    void addStreamline(Streamline* line);

    bool getStreamline(Streamline* line);

    void addSample(const vec3& point);

    vec3 getVelocity(const vec3& point);

    void clear(void);

    void setDSep(int sep);
    void setDTest(float test);
    void setStepSize(float step);

    int getDSep(void) { return this->dSep; }
    float getDTest(void) { return this->dTest; }
    float getStepSize(void) { return this->stepSize; }

    void setUseRungeKutta(bool rk) { this->rungeKutta = rk; }

    bool useRungeKutta(void) { return this->rungeKutta; }

    int getDimX(void) { return this->dimX; }
    int getDimY(void) { return this->dimY; }
    FlowChannel* getChannelX(void) { return this->channelX; }
    FlowChannel* getChannelY(void) { return this->channelY; }

    int getMaxX(void) { return this->geometry.getMaxX(); }
    int getMaxY(void) { return this->geometry.getMaxY(); }

    bool checkDistanceForStop(const vec3& point);
    bool checkDistanceForSeed(const vec3& point);
private:
    int dSep;
    float dTest;
    float stepSize;

    bool rungeKutta;

    FlowGeometry geometry;
    FlowChannel* channelX;
    FlowChannel* channelY;

    int dimX;
    int dimY;

    int cellsX;
    int cellsY;

    std::vector<std::vector<vec3>*> cells;
    std::vector<Streamline*> streamlines;
    std::queue<Streamline*> possibleStreamlines;

    bool checkDistance(const vec3& point, float dist);
};

#endif // STREAMGRID_H