FlowVis/Streamline.h

#pragma once

#include <vector>
#include "Framework/vec3.h"

using std::vector;

class Streamline
{
public:

        Streamline(vec3 seed);

        ~Streamline(void);

        vec3* addSamplePoint(vec3 p, bool backwards);

        void addSamplePointThickness(float val, bool backwards);

        vector<vec3> getSamplePoints(bool backwards) { if (backwards) return sample_points_b; return sample_points_f; };

        vector<vec3>* getSamplePointsPtr(bool backwards) { if (backwards) return &sample_points_b; return &sample_points_f; };

        vector<float> getSamplePointThicknesses(bool backwards) { if (backwards) return sample_point_thickness_b; return sample_point_thickness_f; };

        vec3 getSeedPoint() { return seed_point; };

        vec3* getSeedPointPtr() { return &seed_point; };
        
        void setSeedPointThickness(float val) { seed_point_thickness = val; };

        float getSeedPointThickness() { return seed_point_thickness; };

private:
        vector<vec3> sample_points_f;
        vector<vec3> sample_points_b;
        vector<float> sample_point_thickness_f;
        vector<float> sample_point_thickness_b;
        vec3 seed_point;
        float seed_point_thickness;
};