AssignRasterDataToMesh.cpp

Go to the documentation of this file.

 
#include <memory>
#include <string>
 
// ThirdParty
#include <tclap/CmdLine.h>
 
#ifdef USE_PETSC
#include <mpi.h>
#endif
 
#include "GeoLib/IO/AsciiRasterInterface.h"
#include "GeoLib/Raster.h"
#include "InfoLib/GitInfo.h"
#include "MeshLib/IO/VtkIO/VtuInterface.h"
#include "MeshLib/IO/readMeshFromFile.h"
#include "MeshLib/Mesh.h"
#include "MeshToolsLib/MeshEditing/RasterDataToMesh.h"
 
int main(int argc, char* argv[])
{
    TCLAP::CmdLine cmd(
        "Assigns the pixel information of a raster file to a scalar array of a "
        "specified 2D mesh. Data will be assigned to a node array by default. "
        "Adding information to cell arrays is also possible, pixel values at "
        "the centre of the cell will be used in this case. Note that large "
        "differences in resolution between cell size of the mesh and pixel "
        "size of the raster can give unexpected results. A no-data value will "
        "be added in case of missing or transparent values.\n\n"
        "OpenGeoSys-6 software, version " +
            GitInfoLib::GitInfo::ogs_version +
            ".\n"
            "Copyright (c) 2012-2024, OpenGeoSys Community "
            "(http://www.opengeosys.org)",
        ' ', GitInfoLib::GitInfo::ogs_version);
 
    TCLAP::ValueArg<double> nodata_arg(
        "e", "nodata", "The no data value used for missing values", false, 0,
        "a number");
    cmd.add(nodata_arg);
 
    TCLAP::SwitchArg set_cells_arg("c", "cell-array",
                                   "Assigns raster data to cell array");
    cmd.add(set_cells_arg);
 
    TCLAP::SwitchArg set_nodes_arg(
        "n", "node-array", "Assigns raster data to node array (default)");
    cmd.add(set_nodes_arg);
 
    TCLAP::ValueArg<std::string> array_name_arg(
        "s", "scalar-name", "The name of the newly created scalar array.", true,
        "", "scalar array name");
    cmd.add(array_name_arg);
    TCLAP::ValueArg<std::string> raster_arg("r", "raster",
                                            "Name of the input raster (*.asc)",
                                            true, "", "raster file name");
    cmd.add(raster_arg);
 
    TCLAP::ValueArg<std::string> output_arg("o", "output",
                                            "Name of the output mesh (*.vtu)",
                                            true, "", "output file name");
    cmd.add(output_arg);
    TCLAP::ValueArg<std::string> input_arg("i", "input",
                                           "Name of the input mesh (*.vtu)",
                                           true, "", "input file name");
    cmd.add(input_arg);
    cmd.parse(argc, argv);
 
#ifdef USE_PETSC
    MPI_Init(&argc, &argv);
#endif
 
    bool const create_cell_array(set_cells_arg.isSet());
    bool const create_node_array =
        (create_cell_array) ? set_nodes_arg.isSet() : true;
 
    std::string const& mesh_name = input_arg.getValue();
    std::string const& output_name = output_arg.getValue();
    std::string const& raster_name = raster_arg.getValue();
 
    std::unique_ptr<MeshLib::Mesh> const mesh(
        MeshLib::IO::readMeshFromFile(mesh_name));
    if (mesh->getDimension() > 2)
    {
        ERR("Method can not be applied to 3D meshes.");
#ifdef USE_PETSC
        MPI_Finalize();
#endif
        return EXIT_FAILURE;
    }
 
    std::unique_ptr<GeoLib::Raster> const raster(
        FileIO::AsciiRasterInterface::getRasterFromASCFile(raster_name));
 
    if (create_node_array)
    {
        bool const assigned = MeshToolsLib::RasterDataToMesh::projectToNodes(
            *mesh, *raster, nodata_arg.getValue(), array_name_arg.getValue());
 
        if (!assigned)
        {
            ERR("Error assigning raster data to scalar node array");
#ifdef USE_PETSC
            MPI_Finalize();
#endif
            return EXIT_FAILURE;
        }
        INFO("Created node array {:s}", array_name_arg.getValue());
    }
 
    if (create_cell_array)
    {
        bool const assigned = MeshToolsLib::RasterDataToMesh::projectToElements(
            *mesh, *raster, nodata_arg.getValue(), array_name_arg.getValue());
 
        if (!assigned)
        {
            ERR("Error assigning raster data to scalar cell array");
#ifdef USE_PETSC
            MPI_Finalize();
#endif
            return EXIT_FAILURE;
        }
        INFO("Created cell array {:s}", array_name_arg.getValue());
    }
 
    MeshLib::IO::VtuInterface vtu(mesh.get());
    vtu.writeToFile(output_name);
#ifdef USE_PETSC
    MPI_Finalize();
#endif
    return EXIT_SUCCESS;
}