GteFluid3InitializeSource.h

Go to the documentation of this file.

// David Eberly, Geometric Tools, Redmond WA 98052
// Copyright (c) 1998-2017
// Distributed under the Boost Software License, Version 1.0.
// http://www.boost.org/LICENSE_1_0.txt
// http://www.geometrictools.com/License/Boost/LICENSE_1_0.txt
// File Version: 3.0.0 (2016/06/19)

#pragma once

#include <Physics/GteFluid3Parameters.h>
#include <Graphics/GteConstantBuffer.h>
#include <Graphics/GteProgramFactory.h>
#include <Graphics/GteTexture3.h>

namespace gte
{

class GraphicsEngine;

class GTE_IMPEXP Fluid3InitializeSource
{
public:
    // Construction.  The source density consists of a density producer
    // (increase density) and a density consumer (decrease density).  Each
    // has a location, a variance from that location, and an amplitude.
    // The source velocity (impulse) is generated from gravity, a single
    // wind source, and randomly generated vortices.  Each vortex is
    // selected with a random location, a variance from that location, and
    // an amplitude for the impulse.
    Fluid3InitializeSource(std::shared_ptr<ProgramFactory> const& factory,
        int xSize, int ySize, int zSize, int numXThreads, int numYThreads, int numZThreads,
        std::shared_ptr<ConstantBuffer> const& parameters);

    // Member access.  The texels are (velocity.xyz, density).
    inline std::shared_ptr<Texture3> const& GetSource() const;

    // Compute the source density and source velocity for the fluid
    // simulation.
    void Execute(std::shared_ptr<GraphicsEngine> const& engine);

private:
    enum { NUM_VORTICES = 1024 };

    struct Vortex
    {
        gte::Vector4<float> position;
        gte::Vector4<float> normal;
        gte::Vector4<float> data;
    };

    struct External
    {
        gte::Vector4<float> densityProducer;  // (x, y, z, *)
        gte::Vector4<float> densityPData;     // (variance, amplitude, *, *)
        gte::Vector4<float> densityConsumer;  // (x, y, z, *)
        gte::Vector4<float> densityCData;     // (variance, amplitude, *, *)
        gte::Vector4<float> gravity;
        gte::Vector4<float> windData;
    };

    int mNumXGroups, mNumYGroups, mNumZGroups;
    std::shared_ptr<ComputeProgram> mGenerateVortex;
    std::shared_ptr<ComputeProgram> mInitializeSource;
    std::shared_ptr<ConstantBuffer> mVortex;
    std::shared_ptr<ConstantBuffer> mExternal;
    std::shared_ptr<Texture3> mVelocity0;
    std::shared_ptr<Texture3> mVelocity1;
    std::shared_ptr<Texture3> mSource;

    // Shader source code as strings.
    static std::string const msGLSLGenerateSource;
    static std::string const msGLSLInitializeSource;
    static std::string const msHLSLGenerateSource;
    static std::string const msHLSLInitializeSource;
    static std::string const* msGenerateSource[ProgramFactory::PF_NUM_API];
    static std::string const* msInitializeSource[ProgramFactory::PF_NUM_API];
};

inline std::shared_ptr<Texture3> const& Fluid3InitializeSource::GetSource() const
{
    return mSource;
}

}