VolumeGridEmitter2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: VolumeGridEmitter2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 2-D grid-based volumetric emitter.
> Created Time: 2017/08/06
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_VOLUME_GRID_EMITTER2_H
#define CUBBYFLOW_VOLUME_GRID_EMITTER2_H

#include <Core/Emitter/GridEmitter2.h>
#include <Core/Grid/ScalarGrid2.h>
#include <Core/Grid/VectorGrid2.h>
#include <Core/Surface/ImplicitSurface2.h>
#include <Core/Vector/Vector2.h>

#include <memory>

namespace CubbyFlow
{
    class VolumeGridEmitter2 final : public GridEmitter2
    {
    public:
        class Builder;

        using ScalarMapper = std::function<double(double, const Vector2D&, double)>;

        using VectorMapper = std::function<Vector2D(double, const Vector2D&, const Vector2D&)>;

        explicit VolumeGridEmitter2(const ImplicitSurface2Ptr& sourceRegion, bool isOneShot = true);

        virtual ~VolumeGridEmitter2();

        void AddSignedDistanceTarget(const ScalarGrid2Ptr& scalarGridTarget);

        void AddStepFunctionTarget(const ScalarGrid2Ptr& scalarGridTarget, double minValue, double maxValue);

        void AddTarget(const ScalarGrid2Ptr& scalarGridTarget, const ScalarMapper& customMapper);

        void AddTarget(const VectorGrid2Ptr& vectorGridTarget, const VectorMapper& customMapper);

        const ImplicitSurface2Ptr& GetSourceRegion() const;

        bool GetIsOneShot() const;

        static Builder GetBuilder();

    private:
        using ScalarTarget = std::tuple<ScalarGrid2Ptr, ScalarMapper>;
        using VectorTarget = std::tuple<VectorGrid2Ptr, VectorMapper>;

        ImplicitSurface2Ptr m_sourceRegion;
        bool m_isOneShot = true;
        bool m_hasEmitted = false;
        std::vector<ScalarTarget> m_customScalarTargets;
        std::vector<VectorTarget> m_customVectorTargets;

        void OnUpdate(double currentTimeInSeconds, double timeIntervalInSeconds) override;

        void Emit();
    };

    using VolumeGridEmitter2Ptr = std::shared_ptr<VolumeGridEmitter2>;
    
    class VolumeGridEmitter2::Builder final
    {
    public:
        Builder& WithSourceRegion(const Surface2Ptr& sourceRegion);

        Builder& WithIsOneShot(bool isOneShot);

        VolumeGridEmitter2 Build() const;

        VolumeGridEmitter2Ptr MakeShared() const;

    private:
        ImplicitSurface2Ptr m_sourceRegion;
        bool m_isOneShot = true;
    };
}

#endif