/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   \\    /   O peration     |
    \\  /    A nd           | www.openfoam.com
     \\/     M anipulation  |
-------------------------------------------------------------------------------
    Copyright (C) 2011-2017 OpenFOAM Foundation
-------------------------------------------------------------------------------
License
    This file is part of OpenFOAM.

    OpenFOAM is free software: you can redistribute it and/or modify it
    under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
    ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
    FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
    for more details.

    You should have received a copy of the GNU General Public License
    along with OpenFOAM.  If not, see <http://www.gnu.org/licenses/>.

\*---------------------------------------------------------------------------*/

#include "SlicedGeometricgpuField.H"
#include "processorgpuFvPatch.H"

// * * * * * * * * * * * * Private Member Functions * * * * * * * * * * * * * //

template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::tmp<Foam::FieldField<PatchField, Type>>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
slicedBoundaryField
(
    const Mesh& mesh,
    const gpuField<Type>& completeField,
    const bool preserveCouples,
    const bool preserveProcessorOnly
)
{
    tmp<FieldField<PatchField, Type>> tbf
    (
        new FieldField<PatchField, Type>(mesh.boundary().size())
    );
    FieldField<PatchField, Type>& bf = tbf.ref();

    forAll(mesh.boundary(), patchi)
    {
        if
        (
            preserveCouples
         && mesh.boundary()[patchi].coupled()
         && (
               !preserveProcessorOnly
            || isA<processorgpuFvPatch>(mesh.boundary()[patchi])
            )
        )
        {
            // For coupled patched construct the correct patch field type
            bf.set
            (
                patchi,
                PatchField<Type>::New
                (
                    mesh.boundary()[patchi].type(),
                    mesh.boundary()[patchi],
                    *this
                )
            );

            // Initialize the values on the coupled patch to those of the slice
            // of the given field.
            // Note: these will usually be over-ridden by the boundary field
            // evaluation e.g. in the case of processor and cyclic patches.
            bf[patchi] = SlicedPatchField<Type>
            (
                mesh.boundary()[patchi],
                DimensionedgpuField<Type, GeoMesh>::null(),
                completeField
            );
        }
        else
        {
            bf.set
            (
                patchi,
                new SlicedPatchField<Type>
                (
                    mesh.boundary()[patchi],
                    DimensionedgpuField<Type, GeoMesh>::null(),
                    completeField
                )
            );
        }
    }

    return tbf;
}


template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::tmp<Foam::FieldField<PatchField, Type>>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
slicedBoundaryField
(
    const Mesh& mesh,
    const FieldField<PatchField, Type>& bField,
    const bool preserveCouples
)
{
    tmp<FieldField<PatchField, Type>> tbf
    (
        new FieldField<PatchField, Type>(mesh.boundary().size())
    );
    FieldField<PatchField, Type>& bf = tbf.ref();

    forAll(mesh.boundary(), patchi)
    {
        if (preserveCouples && mesh.boundary()[patchi].coupled())
        {
            // For coupled patched construct the correct patch field type
            bf.set
            (
                patchi,
                PatchField<Type>::New
                (
                    mesh.boundary()[patchi].type(),
                    mesh.boundary()[patchi],
                    *this
                )
            );

            // Assign field
            bf[patchi] == bField[patchi];
        }
        else
        {
            // Create unallocated copy of patch field
            bf.set
            (
                patchi,
                new SlicedPatchField<Type>
                (
                    mesh.boundary()[patchi],
                    DimensionedgpuField<Type, GeoMesh>::null()
                )
            );
            bf[patchi].setDelegate(bField[patchi]);
        }
    }

    return tbf;
}


// * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //

template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
Internal::Internal
(
    const IOobject& io,
    const Mesh& mesh,
    const dimensionSet& ds,
    const gpuField<Type>& iField
)
:
    DimensionedgpuField<Type, GeoMesh>
    (
        io,
        mesh,
        ds,
        gpuField<Type>()
    )
{
    // Set the internalField to the slice of the complete field
    GeometricgpuField<Type, PatchField, GeoMesh>::Internal::field().setDelegate
    (
        const_cast<gpuField<Type>&>(iField), GeoMesh::size(mesh)
    ); 
}


template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
SlicedGeometricgpuField
(
    const IOobject& io,
    const Mesh& mesh,
    const dimensionSet& ds,
    const gpuField<Type>& completeField,
    const bool preserveCouples
)
:
    GeometricgpuField<Type, PatchField, GeoMesh>
    (
        io,
        mesh,
        ds,
        gpuField<Type>(),
        slicedBoundaryField(mesh, completeField, preserveCouples)
    )
{
    // Set the internalField to the slice of the complete field
    GeometricgpuField<Type, PatchField, GeoMesh>::Internal::field().setDelegate
    (
        const_cast<gpuField<Type>&>(completeField), GeoMesh::size(mesh)
    ); 

    correctBoundaryConditions();
}


template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
SlicedGeometricgpuField
(
    const IOobject& io,
    const Mesh& mesh,
    const dimensionSet& ds,
    const gpuField<Type>& completeIField,
    const gpuField<Type>& completeBField,
    const bool preserveCouples,
    const bool preserveProcessorOnly
)
:
    GeometricgpuField<Type, PatchField, GeoMesh>
    (
        io,
        mesh,
        ds,
        gpuField<Type>(),
        slicedBoundaryField
        (
            mesh,
            completeBField,
            preserveCouples,
            preserveProcessorOnly
        )
    )
{
    // Set the internalField to the slice of the complete field
    GeometricgpuField<Type, PatchField, GeoMesh>::Internal::field().setDelegate
    (
        const_cast<gpuField<Type>&>(completeIField), GeoMesh::size(mesh)
    ); 

    correctBoundaryConditions();
}


template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
SlicedGeometricgpuField
(
    const IOobject& io,
    const GeometricgpuField<Type, PatchField, GeoMesh>& gf,
    const bool preserveCouples
)
:
    GeometricgpuField<Type, PatchField, GeoMesh>
    (
        io,
        gf.mesh(),
        gf.dimensions(),
        gpuField<Type>(),
        slicedBoundaryField(gf.mesh(), gf.boundaryField(), preserveCouples)
    )
{
    // Set the internalField to the supplied internal field
    GeometricgpuField<Type, PatchField, GeoMesh>::primitiveFieldRef().setDelegate(gf.primitiveField());  

    correctBoundaryConditions();
}


template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
SlicedGeometricgpuField
(
    const SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>& gf
)
:
    GeometricgpuField<Type, PatchField, GeoMesh>
    (
        gf,
        gf.mesh(),
        gf.dimensions(),
        gpuField<Type>(),
        slicedBoundaryField(gf.mesh(), gf.boundaryField(), true)
    )
{
    // Set the internalField to the supplied internal field
    GeometricgpuField<Type, PatchField, GeoMesh>::Internal::field().setDelegate(gf.primitiveField());
}


template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::tmp
<
    Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>
>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
clone() const
{
    return tmp
    <
        SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>
    >
    (
        new SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>
        (
            *this
        )
    );
}


// * * * * * * * * * * * * * * * * Destructor  * * * * * * * * * * * * * * * //

template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
~SlicedGeometricgpuField()
{
    // Set the internalField storage pointer to nullptr before its destruction
    // to protect the field it a slice of.
//    UList<Type>::shallowCopy(UList<Type>(nullptr, 0));
}


template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
Internal::~Internal()
{
    // Set the internalField storage pointer to nullptr before its destruction
    // to protect the field it a slice of.
//    UList<Type>::shallowCopy(UList<Type>(nullptr, 0));
}


// * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //

template
<
    class Type,
    template<class> class PatchField,
    template<class> class SlicedPatchField,
    class GeoMesh
>
void Foam::SlicedGeometricgpuField<Type, PatchField, SlicedPatchField, GeoMesh>::
correctBoundaryConditions()
{
    GeometricgpuField<Type, PatchField, GeoMesh>::correctBoundaryConditions();
}


// ************************************************************************* //
