include/Neighbour.h

Go to the documentation of this file.

#ifndef NEIGHBOUR_H
#define NEIGHBOUR_H

#include "map"

#include "FluidParticle.h"

typedef std::multimap<int, FluidParticle> hashmap;


class Neighbour
{
public:
    Neighbour
            (
                const int _numParticles,
                const float _smoothingLength
            );

    ~Neighbour();

    void reInitialise
            (
                const int _numParticles,
                const float _smoothingLength
            );

    void refreshHashmap(std::vector<FluidParticle>& _particleList);

    void clearHashmap();

    std::vector<FluidParticle> determineNeighbours(const ngl::Vector _centerPosition);

    inline void setCellSize(const ngl::Real _size) { m_cellSize = _size; }


private:
    hashmap m_hashMap;

    float m_cellSize;

    int m_mapSize;

    int m_largePrime1;

    int m_largePrime2;

    int m_largePrime3;

    int m_initialNeighbourListSize;

private:
    int modulus
            (
                const int _a,
                const int _b
            );

    int getNextPrimeNumber(const int _start);

    std::vector<int> discretize(const ngl::Vector _position);

    int hashPosition(const ngl::Vector _position);

    void getCellNeighbours
                    (
                        const ngl::Vector _centerPosition,
                        const bool _checkWithinSmoothingLength,
                        std::vector<FluidParticle>& o_neighbourList
                    );

    bool isParticleAlreadyInList
                    (
                        const int _testId,
                        std::vector<FluidParticle>& _neighbourList
                    );


};

#endif // NEIGHBOUR_H