/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   \\    /   O peration     |
    \\  /    A nd           | Copyright (C) 2004-2011 OpenCFD Ltd.
     \\/     M anipulation  |
-------------------------------------------------------------------------------
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/>.

Application
    globalMeshDataTest

Description
    Test global point communication

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

#include "globalMeshData.H"
#include "argList.H"
#include "polyMesh.H"
#include "Time.H"
#include "mapDistribute.H"

using namespace Foam;


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

// Main program:

int main(int argc, char *argv[])
{
#   include "setRootCase.H"
#   include "createTime.H"
#   include "createPolyMesh.H"

    const globalMeshData& globalData = mesh.globalData();
    const indirectPrimitivePatch& coupledPatch = globalData.coupledPatch();
    const globalIndexAndTransform& transforms = globalData.globalTransforms();


    // Test:print shared points
    {
        const mapDistribute& globalPointSlavesMap =
            globalData.globalPointSlavesMap();
        const labelListList& slaves =
            globalData.globalPointSlaves();
        const labelListList& transformedSlaves =
            globalData.globalPointTransformedSlaves();

        // Create field with my local data
        pointField coords(globalPointSlavesMap.constructSize());
        SubList<point>(coords, coupledPatch.nPoints()).assign
        (
            coupledPatch.localPoints()
        );

        // Exchange data. Apply positional transforms.
        globalPointSlavesMap.distribute(transforms, coords, true);

        // Print
        forAll(slaves, pointI)
        {
            const labelList& slavePoints = slaves[pointI];

            if (slavePoints.size() > 0)
            {
                Pout<< "Master point:" << pointI
                    << " coord:" << coords[pointI]
                    << " connected to untransformed slave points:" << endl;

                forAll(slavePoints, i)
                {
                    Pout<< "    " << coords[slavePoints[i]] << endl;
                }
            }

            const labelList& transformedSlavePoints = transformedSlaves[pointI];

            if (transformedSlavePoints.size() > 0)
            {
                Pout<< "Master point:" << pointI
                    << " coord:" << coords[pointI]
                    << " connected to transformed slave points:" << endl;

                forAll(transformedSlavePoints, i)
                {
                    Pout<< "    " << coords[transformedSlavePoints[i]]
                        << endl;
                }
            }
        }
    }


    // Test:print shared edges
    {
        const mapDistribute& globalEdgeSlavesMap =
            globalData.globalEdgeSlavesMap();
        const labelListList& slaves =
            globalData.globalEdgeSlaves();
        const labelListList& transformedSlaves =
            globalData.globalEdgeTransformedSlaves();

        // Test: distribute edge centres
        pointField ec(globalEdgeSlavesMap.constructSize());
        forAll(coupledPatch.edges(), edgeI)
        {
            ec[edgeI] = coupledPatch.edges()[edgeI].centre
            (
                coupledPatch.localPoints()
            );
        }

        // Exchange data Apply positional transforms.
        globalEdgeSlavesMap.distribute(transforms, ec, true);

        // Print
        forAll(slaves, edgeI)
        {
            const labelList& slaveEdges = slaves[edgeI];

            if (slaveEdges.size() > 0)
            {
                Pout<< "Master edge:" << edgeI
                    << " centre:" << ec[edgeI]
                    << " connected to slave edges:" << endl;

                forAll(slaveEdges, i)
                {
                    Pout<< "    " << ec[slaveEdges[i]] << endl;
                }
            }
            const labelList& transformedSlaveEdges = transformedSlaves[edgeI];

            if (transformedSlaveEdges.size() > 0)
            {
                Pout<< "Master edge:" << edgeI
                    << " centre:" << ec[edgeI]
                    << " connected to transformed slave edges:" << endl;

                forAll(transformedSlaveEdges, i)
                {
                    Pout<< "    " << ec[transformedSlaveEdges[i]]
                        << endl;
                }
            }
        }
    }


    //// Test: (collocated) point to faces addressing
    //{
    //    const labelListList& globalPointBoundaryFaces =
    //        globalData.globalPointBoundaryFaces();
    //    const mapDistribute& globalPointBoundaryFacesMap =
    //        globalData.globalPointBoundaryFacesMap();
    //
    //    label nBnd = mesh.nFaces()-mesh.nInternalFaces();
    //
    //    pointField fc(globalPointBoundaryFacesMap.constructSize());
    //    SubList<point>(fc, nBnd).assign
    //    (
    //        primitivePatch
    //        (
    //            SubList<face>
    //            (
    //                mesh.faces(),
    //                nBnd,
    //                mesh.nInternalFaces()
    //            ),
    //            mesh.points()
    //        ).faceCentres()
    //    );
    //
    //    // Exchange data
    //    globalPointBoundaryFacesMap.distribute(fc);
    //
    //    // Print
    //    forAll(globalPointBoundaryFaces, pointI)
    //    {
    //        const labelList& bFaces = globalPointBoundaryFaces[pointI];
    //
    //        Pout<< "Point:" << pointI
    //            << " at:" << coupledPatch.localPoints()[pointI]
    //            << " connected to faces:" << endl;
    //
    //        forAll(bFaces, i)
    //        {
    //            Pout<< "    " << fc[bFaces[i]] << endl;
    //        }
    //    }
    //}
    //
    //
    //// Test:(collocated) point to cells addressing
    //{
    //    const labelList& boundaryCells = globalData.boundaryCells();
    //    const labelListList& globalPointBoundaryCells =
    //        globalData.globalPointBoundaryCells();
    //    const mapDistribute& globalPointBoundaryCellsMap =
    //        globalData.globalPointBoundaryCellsMap();
    //
    //    pointField cc(globalPointBoundaryCellsMap.constructSize());
    //    forAll(boundaryCells, i)
    //    {
    //        cc[i] = mesh.cellCentres()[boundaryCells[i]];
    //    }
    //
    //    // Exchange data
    //    globalPointBoundaryCellsMap.distribute(cc);
    //
    //    // Print
    //    forAll(globalPointBoundaryCells, pointI)
    //    {
    //        const labelList& bCells = globalPointBoundaryCells[pointI];
    //
    //        Pout<< "Point:" << pointI
    //            << " at:" << coupledPatch.localPoints()[pointI]
    //            << " connected to cells:" << endl;
    //
    //        forAll(bCells, i)
    //        {
    //            Pout<< "    " << cc[bCells[i]] << endl;
    //        }
    //    }
    //}


    Info<< "End\n" << endl;

    return 0;
}


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