MeshConversion.hpp

//---- GPL ---------------------------------------------------------------------
//
// Copyright (C)  Stichting Deltares, 2011-2023.
//
// This program 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 version 3.
//
// This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
//
// contact: delft3d.support@deltares.nl
// Stichting Deltares
// P.O. Box 177
// 2600 MH Delft, The Netherlands
//
// All indications and logos of, and references to, "Delft3D" and "Deltares"
// are registered trademarks of Stichting Deltares, and remain the property of
// Stichting Deltares. All rights reserved.
//
//------------------------------------------------------------------------------
 
#pragma once
 
#include <concepts>
 
#include "MeshKernel/Definitions.hpp"
#include "MeshKernel/Exceptions.hpp"
#include "MeshKernel/Mesh.hpp"
#include "MeshKernel/Point.hpp"
#include "MeshKernel/UndoActions/MeshConversionAction.hpp"
 
namespace meshkernel
{
 
    template <typename Operation>
    concept HasTransformationOperation = requires(Operation op, Point p) {{ op(p)} -> std::same_as<Point>; };
 
    template <typename Operation>
    concept HasConversionProjection = requires(Operation op) {{ op.SourceProjection()} -> std::same_as<Projection>;
                                                              { op.TargetProjection()} -> std::same_as<Projection>; };
 
    template <typename Operation>
    concept ConversionFunctor = HasTransformationOperation<Operation> && HasConversionProjection<Operation>;
 
    class MeshConversion
    {
    public:
        template <ConversionFunctor Conversion>
        [[nodiscard]] static std::unique_ptr<UndoAction> Compute(const Mesh& sourceMesh, Mesh& targetMesh, const Conversion& conversion)
        {
            if (sourceMesh.m_projection != conversion.SourceProjection())
            {
                throw MeshKernelError("Incorrect source mesh coordinate system, expecting '{}', found '{}'",
                                      ProjectionToString(conversion.SourceProjection()), ProjectionToString(sourceMesh.m_projection));
            }
 
            if (targetMesh.m_projection != conversion.TargetProjection())
            {
                throw MeshKernelError("Incorrect target mesh coordinate system, expecting '{}', found '{}'",
                                      ProjectionToString(conversion.TargetProjection()), ProjectionToString(targetMesh.m_projection));
            }
 
            if (sourceMesh.GetNumNodes() != targetMesh.GetNumNodes())
            {
                throw MeshKernelError("Source and target meshes have different numbers of nodes, source = '{}', target = '{}'",
                                      sourceMesh.GetNumNodes(), targetMesh.GetNumNodes());
            }
 
            std::vector<Point> targetNodes(targetMesh.Nodes());
            std::unique_ptr<MeshConversionAction> undoAction = MeshConversionAction::Create(targetMesh);
 
#pragma omp parallel for
            for (int i = 0; i < static_cast<int>(sourceMesh.GetNumNodes()); ++i)
            {
                if (sourceMesh.Node(i).IsValid())
                {
                    targetNodes[i] = conversion(sourceMesh.Node(i));
                }
                else
                {
                    targetNodes[i] = sourceMesh.Node(i);
                }
 
#if 0
                if (sourceMesh.Node(i).IsValid())
                {
                    targetMesh.SetNode(i, conversion(sourceMesh.Node(i)));
                }
                else
                {
                    targetMesh.SetNode(i, sourceMesh.Node(i));
                }
#endif
            }
 
            targetMesh.SetNodes(targetNodes);
            targetMesh.Administrate();
            return undoAction;
        }
 
        template <ConversionFunctor Conversion>
        [[nodiscard]] static std::unique_ptr<UndoAction> Compute(Mesh& mesh, const Conversion& conversion)
        {
            if (mesh.m_projection != conversion.SourceProjection())
            {
                throw MeshKernelError("Incorrect mesh coordinate system, expecting '{}', found '{}'",
                                      ProjectionToString(conversion.SourceProjection()), ProjectionToString(mesh.m_projection));
            }
 
            std::vector<Point> nodes(mesh.Nodes());
            std::unique_ptr<MeshConversionAction> undoAction = MeshConversionAction::Create(mesh);
 
#pragma omp parallel for
            for (int i = 0; i < static_cast<int>(mesh.GetNumNodes()); ++i)
            {
                if (nodes[i].IsValid())
                {
                    nodes[i] = conversion(nodes[i]);
                }
#if 0
                if (mesh.Node(i).IsValid())
                {
                    mesh.SetNode(i, conversion(mesh.Node(i)));
                }
#endif
            }
 
            mesh.SetNodes(nodes);
            mesh.m_projection = conversion.TargetProjection();
            mesh.Administrate();
            return undoAction;
        }
    };
 
} // namespace meshkernel