ParticleSystemData3.h

Go to the documentation of this file.

/*************************************************************************
> File Name: ParticleSystemData3.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 3-D particle system data.
> Created Time: 2017/05/08
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_PARTICLE_SYSTEM_DATA3_H
#define CUBBYFLOW_PARTICLE_SYSTEM_DATA3_H

#include <Core/Array/Array1.h>
#include <Core/Searcher/PointNeighborSearcher3.h>
#include <Core/Utils/Serialization.h>
#include <Core/Vector/Vector3.h>

#include <memory>
#include <vector>

#ifndef CUBBYFLOW_DOXYGEN

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

namespace CubbyFlow
{
    namespace fbs
    {
        struct ParticleSystemData3;
    }
}

#endif

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

        using VectorData = Array1<Vector3D>;

        ParticleSystemData3();

        explicit ParticleSystemData3(size_t numberOfParticles);

        ParticleSystemData3(const ParticleSystemData3& other);

        virtual ~ParticleSystemData3();

        void Resize(size_t newNumberOfParticles);

        size_t GetNumberOfParticles() const;

        size_t AddScalarData(double initialVal = 0.0);

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

        double GetRadius() const;

        virtual void SetRadius(double newRadius);

        double GetMass() const;

        virtual void SetMass(double newMass);

        ConstArrayAccessor1<Vector3D> GetPositions() const;

        ArrayAccessor1<Vector3D> GetPositions();

        ConstArrayAccessor1<Vector3D> GetVelocities() const;

        ArrayAccessor1<Vector3D> GetVelocities();

        ConstArrayAccessor1<Vector3D> GetForces() const;

        ArrayAccessor1<Vector3D> GetForces();

        ConstArrayAccessor1<double> ScalarDataAt(size_t idx) const;

        ArrayAccessor1<double> ScalarDataAt(size_t idx);

        ConstArrayAccessor1<Vector3D> VectorDataAt(size_t idx) const;

        ArrayAccessor1<Vector3D> VectorDataAt(size_t idx);

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

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

        const PointNeighborSearcher3Ptr& GetNeighborSearcher() const;

        void SetNeighborSearcher(const PointNeighborSearcher3Ptr& 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 ParticleSystemData3& other);

        ParticleSystemData3& operator=(const ParticleSystemData3& other);

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

        void DeserializeParticleSystemData(
            const fbs::ParticleSystemData3* 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;

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

    using ParticleSystemData3Ptr = std::shared_ptr<ParticleSystemData3>;
}

#endif