VertexCenteredVectorGrid3.h

Go to the documentation of this file.

/*************************************************************************
> File Name: VertexCenteredVectorGrid3.h
> Project Name: CubbyFlow
> Author: Dongmin Kim
> Purpose: 3-D Vertex-centered vector grid structure.
> Created Time: 2017/08/04
> Copyright (c) 2018, Dongmin Kim
*************************************************************************/
#ifndef CUBBYFLOW_VERTEX_CENTERED_VECTOR_GRID3_H
#define CUBBYFLOW_VERTEX_CENTERED_VECTOR_GRID3_H

#include <Core/Grid/CollocatedVectorGrid3.h>

namespace CubbyFlow
{
    class VertexCenteredVectorGrid3 final : public CollocatedVectorGrid3
    {
    public:
        CUBBYFLOW_GRID3_TYPE_NAME(VertexCenteredVectorGrid3)

        class Builder;

        VertexCenteredVectorGrid3();

        VertexCenteredVectorGrid3(
            size_t resolutionX, size_t resolutionY, size_t resolutionZ,
            double gridSpacingX = 1.0, double gridSpacingY = 1.0, double gridSpacingZ = 1.0,
            double originX = 0.0, double originY = 0.0, double originZ = 0.0,
            double initialValueU = 0.0, double initialValueV = 0.0, double initialValueW = 0.0);

        VertexCenteredVectorGrid3(
            const Size3& resolution,
            const Vector3D& gridSpacing = Vector3D(1.0, 1.0, 1.0),
            const Vector3D& origin = Vector3D(),
            const Vector3D& initialValue = Vector3D());

        VertexCenteredVectorGrid3(const VertexCenteredVectorGrid3& other);

        Size3 GetDataSize() const override;

        Vector3D GetDataOrigin() const override;

        void Swap(Grid3* other) override;

        void Set(const VertexCenteredVectorGrid3& other);

        VertexCenteredVectorGrid3& operator=(const VertexCenteredVectorGrid3& other);

        void Fill(const Vector3D& value, ExecutionPolicy policy = ExecutionPolicy::Parallel) override;

        void Fill(const std::function<Vector3D(const Vector3D&)>& func, ExecutionPolicy policy = ExecutionPolicy::Parallel) override;

        std::shared_ptr<VectorGrid3> Clone() const override;

        static Builder GetBuilder();
    };

    using VertexCenteredVectorGrid3Ptr = std::shared_ptr<VertexCenteredVectorGrid3>;

    class VertexCenteredVectorGrid3::Builder final : public VectorGridBuilder3
    {
    public:
        Builder& WithResolution(const Size3& resolution);

        Builder& WithResolution(size_t resolutionX, size_t resolutionY, size_t resolutionZ);

        Builder& WithGridSpacing(const Vector3D& gridSpacing);

        Builder& WithGridSpacing(double gridSpacingX, double gridSpacingY, double gridSpacingZ);

        Builder& WithOrigin(const Vector3D& gridOrigin);

        Builder& WithOrigin(double gridOriginX, double gridOriginY, double gridOriginZ);

        Builder& WithInitialValue(double initialValU, double initialValV, double initialValW);

        VertexCenteredVectorGrid3 Build() const;

        VertexCenteredVectorGrid3Ptr MakeShared() const;

        VectorGrid3Ptr Build(
            const Size3& resolution,
            const Vector3D& gridSpacing,
            const Vector3D& gridOrigin,
            const Vector3D& initialVal) const override;

    private:
        Size3 m_resolution{ 1, 1, 1 };
        Vector3D m_gridSpacing{ 1, 1, 1 };
        Vector3D m_gridOrigin{ 0, 0, 0 };
        Vector3D m_initialVal{ 0, 0, 0};
    };
}

#endif