CellCenteredVectorGrid3.h

Go to the documentation of this file.

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

#include <Core/Grid/CollocatedVectorGrid3.h>

namespace CubbyFlow
{
    class CellCenteredVectorGrid3 final : public CollocatedVectorGrid3
    {
    public:
        CUBBYFLOW_GRID3_TYPE_NAME(CellCenteredVectorGrid3);

        class Builder;

        CellCenteredVectorGrid3();

        CellCenteredVectorGrid3(
            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 initalValueV = 0.0, double initalValueZ = 0.0);

        CellCenteredVectorGrid3(
            const Size3& resolution,
            const Vector3D& gridSpacing = Vector3D(1.0, 1.0, 1.0),
            const Vector3D& origin = Vector3D(),
            const Vector3D& initalValue = Vector3D());

        CellCenteredVectorGrid3(const CellCenteredVectorGrid3& other);

        Size3 GetDataSize() const override;

        Vector3D GetDataOrigin() const override;

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

        void Swap(Grid3* other) override;

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

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

        void Set(const CellCenteredVectorGrid3& other);

        CellCenteredVectorGrid3& operator=(const CellCenteredVectorGrid3& other);

        static Builder GetBuilder();
    };

    using CellCenteredVectorGrid3Ptr = std::shared_ptr<CellCenteredVectorGrid3>;

    class CellCenteredVectorGrid3::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(const Vector3D& initalVal);

        Builder& WithInitialValue(double initialValX, double initalValY, double initalValZ);

        CellCenteredVectorGrid3 Build() const;

        CellCenteredVectorGrid3Ptr MakeShared() const;

        VectorGrid3Ptr Build(
            const Size3& resolution,
            const Vector3D& gridSpacing,
            const Vector3D& gridOrigin,
            const Vector3D& initalVal) 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