ParticleSystemData2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: ParticleSystemData2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 2-D particle system data.
> Created Time: 2017/04/28
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_PARTICLE_SYSTEM_DATA2_H
#define CUBBYFLOW_PARTICLE_SYSTEM_DATA2_H

#include <Core/Array/Array1.h>
#include <Core/Searcher/PointNeighborSearcher2.h>
#include <Core/Utils/Serialization.h>
#include <Core/Vector/Vector2.h>

#include <memory>
#include <vector>

#ifndef CUBBYFLOW_DOXYGEN

namespace flatbuffers
{
    class FlatBufferBuilder;
    template <typename T> struct Offset;
}

namespace CubbyFlow
{
    namespace fbs
    {
        struct ParticleSystemData2;
    }
}

#endif

namespace CubbyFlow
{
    class ParticleSystemData2 : public Serializable
    {
    public:
        using ScalarData = Array1<double>;

        using VectorData = Array1<Vector2D>;

        ParticleSystemData2();

        explicit ParticleSystemData2(size_t numberOfParticles);

        ParticleSystemData2(const ParticleSystemData2& other);

        virtual ~ParticleSystemData2();

        void Resize(size_t newNumberOfParticles);

        size_t GetNumberOfParticles() const;

        size_t AddScalarData(double initialVal = 0.0);

        size_t AddVectorData(const Vector2D& initialVal = Vector2D());

        double GetRadius() const;

        virtual void SetRadius(double newRadius);

        double GetMass() const;

        virtual void SetMass(double newMass);

        ConstArrayAccessor1<Vector2D> GetPositions() const;

        ArrayAccessor1<Vector2D> GetPositions();

        ConstArrayAccessor1<Vector2D> GetVelocities() const;

        ArrayAccessor1<Vector2D> GetVelocities();

        ConstArrayAccessor1<Vector2D> GetForces() const;

        ArrayAccessor1<Vector2D> GetForces();

        ConstArrayAccessor1<double> ScalarDataAt(size_t idx) const;

        ArrayAccessor1<double> ScalarDataAt(size_t idx);

        ConstArrayAccessor1<Vector2D> VectorDataAt(size_t idx) const;

        ArrayAccessor1<Vector2D> VectorDataAt(size_t idx);

        void AddParticle(
            const Vector2D& newPosition,
            const Vector2D& newVelocity = Vector2D(),
            const Vector2D& newForce = Vector2D());

        void AddParticles(
            const ConstArrayAccessor1<Vector2D>& newPositions,
            const ConstArrayAccessor1<Vector2D>& newVelocities = ConstArrayAccessor1<Vector2D>(),
            const ConstArrayAccessor1<Vector2D>& newForces = ConstArrayAccessor1<Vector2D>());

        const PointNeighborSearcher2Ptr& GetNeighborSearcher() const;

        void SetNeighborSearcher(const PointNeighborSearcher2Ptr& newNeighborSearcher);

        const std::vector<std::vector<size_t>>& GetNeighborLists() const;

        void BuildNeighborSearcher(double maxSearchRadius);

        void BuildNeighborLists(double maxSearchRadius);

        void Serialize(std::vector<uint8_t>* buffer) const override;

        void Deserialize(const std::vector<uint8_t>& buffer) override;

        void Set(const ParticleSystemData2& other);

        ParticleSystemData2& operator=(const ParticleSystemData2& other);

    protected:
        void SerializeParticleSystemData(
            flatbuffers::FlatBufferBuilder* builder,
            flatbuffers::Offset<fbs::ParticleSystemData2>* fbsParticleSystemData)
            const;

        void DeserializeParticleSystemData(
            const fbs::ParticleSystemData2* fbsParticleSystemData);

    private:
        double m_radius = 1e-3;
        double m_mass = 1e-3;
        size_t m_numberOfParticles = 0;
        size_t m_positionIdx;
        size_t m_velocityIdx;
        size_t m_forceIdx;

        std::vector<ScalarData> m_scalarDataList;
        std::vector<VectorData> m_vectorDataList;

        PointNeighborSearcher2Ptr m_neighborSearcher;
        std::vector<std::vector<size_t>> m_neighborLists;
    };

    using ParticleSystemData2Ptr = std::shared_ptr<ParticleSystemData2>;
}

#endif