CellCenteredVectorGrid2.h

Go to the documentation of this file.

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

#include <Core/Grid/CollocatedVectorGrid2.h>

namespace CubbyFlow
{
    class CellCenteredVectorGrid2 final : public CollocatedVectorGrid2
    {
    public:
        CUBBYFLOW_GRID2_TYPE_NAME(CellCenteredVectorGrid2);

        class Builder;
        
        CellCenteredVectorGrid2();
        
        CellCenteredVectorGrid2(
            size_t resolutionX, size_t resolutionY,
            double gridSpacingX = 1.0, double gridSpacingY = 1.0,
            double originX = 0.0, double originY = 0.0,
            double initialValueU = 0.0, double initalValueV = 0.0);

        CellCenteredVectorGrid2(
            const Size2& resolution,
            const Vector2D& gridSpacing = Vector2D(1.0, 1.0),
            const Vector2D& origin = Vector2D(),
            const Vector2D& initalValue = Vector2D());
        
        CellCenteredVectorGrid2(const CellCenteredVectorGrid2& other);

        Size2 GetDataSize() const override;

        Vector2D GetDataOrigin() const override;

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

        void Swap(Grid2* other) override;

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

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

        void Set(const CellCenteredVectorGrid2& other);
        
        CellCenteredVectorGrid2& operator=(const CellCenteredVectorGrid2& other);
        
        static Builder GetBuilder();
    };

    using CellCenteredVectorGrid2Ptr = std::shared_ptr<CellCenteredVectorGrid2>;

    class CellCenteredVectorGrid2::Builder final : public VectorGridBuilder2
    {
    public:
        Builder& WithResolution(const Size2& resolution);
        
        Builder& WithResolution(size_t resolutionX, size_t resolutionY);

        Builder& WithGridSpacing(const Vector2D& gridSpacing);

        Builder& WithGridSpacing(double gridSpacingX, double gridSpacingY);
        
        Builder& WithOrigin(const Vector2D& gridOrigin);

        Builder& WithOrigin(double gridOriginX, double gridOriginY);

        Builder& WithInitialValue(const Vector2D& initalVal);

        Builder& WithInitialValue(double initialValX, double initialValY);

        CellCenteredVectorGrid2 Build() const;

        CellCenteredVectorGrid2Ptr MakeShared() const;

        VectorGrid2Ptr Build(
            const Size2& resolution,
            const Vector2D& gridSpacing,
            const Vector2D& gridOrigin,
            const Vector2D& initalVal) const override;
        
    private:
        Size2 m_resolution{ 1, 1 };
        Vector2D m_gridSpacing{ 1, 1 };
        Vector2D m_gridOrigin{ 0, 0 };
        Vector2D m_initialVal{ 0, 0 };
    };
}

#endif