ParticleFactory.h

Go to the documentation of this file.

#pragma once

#include "Particle.h"

#include <memory>


class ParticleFactory
{
public:
        ParticleFactory(vl::ref<vl::Image> volume, vl::ref<vl::Image> gradient, vl::ref<vl::Image> curvature, float depth, float magThresh, int radius, int maxParticles );
        ~ParticleFactory(void);

        // setters
        void setDepth(float d) { _depth = d; }
        void setMagnitudeThreshold(float t) { _magnitudeThreshold = t; }
        void setRadius(int r) { _radius = r; }
        void setMaxParticles(int p) { _maxParticles = p; }

        // getters
        float getDepth() const { return _depth; }
        float getMagnitudeThreshold() const { return _magnitudeThreshold; }
        int getRadius() const { return _radius; }
        int getMaxParticles() const { return _maxParticles; }

        void build();

        std::vector<std::shared_ptr<Particle>> getParticleList() const { return _particles; }
private:
        bool containsParticle(int x, int y, int z) const;
        
        vl::ref<vl::Image> _volume;
        vl::ref<vl::Image> _gradient;
        vl::ref<vl::Image> _curvature;
        float _depth;
        float _magnitudeThreshold;
        int _radius;
        int _maxParticles;

        std::vector<std::shared_ptr<Particle>> _particles;
        vl::ref<vl::Image> _particleImage;
};