imstkRenderParticles.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkGeometry.h"
#include "imstkMath.h"
#include "imstkMacros.h"

#ifdef WIN32
#pragma warning( push )
#pragma warning( disable : 4201 )
#endif

#ifdef WIN32
#pragma warning( pop )
#endif

namespace imstk
{
struct RenderParticle
{
    Vec3f m_position     = Vec3f(0, 0, 0);
    Vec3f m_velocity     = Vec3f(0, 0, 0);
    Vec3f m_acceleration = Vec3f(0, 0, 0);
    Vec4d m_color    = Vec4d(1., 1., 1., 1.);
    float m_age      = 0;
    bool m_created   = false;
    float m_scale    = 1.0f;
    float m_rotation = 0;
    float m_rotationalVelocity     = 0;
    float m_rotationalAcceleration = 0;
};

class RenderParticles : public Geometry
{
public:
    RenderParticles(const unsigned int maxNumParticles = 128);
    ~RenderParticles() override = default;

    IMSTK_TYPE_NAME(RenderParticles)

    
    void setParticleSize(const float size);

    std::vector<std::unique_ptr<RenderParticle>>& getParticles();

    void reset();

    void incrementNumOfParticles();

    unsigned int getNumParticles();

    unsigned int getMaxNumParticles();

    std::unique_ptr<RenderParticles> clone()
    {
        return std::unique_ptr<RenderParticles>(cloneImplementation());
    }

protected:
    unsigned int m_numParticles    = 0;
    unsigned int m_maxNumParticles = 128; 
    float m_particleSize = 0.1f;

    std::vector<std::unique_ptr<RenderParticle>> m_particles; 
    Vec3d m_vertexPositions[4];
    Vec3d m_vertexNormals[4];
    Vec3d m_vertexTangents[4];
    Vec2d m_vertexUVs[4];
    Vec3i m_triangles[2];

    void applyTransform(const Mat4d& m) override;

    void updatePostTransformData() const override {}

private:
    RenderParticles* cloneImplementation() const;
};
} // namespace imstk