Applications/Utils/MeshEdit/AddFaultToVoxelGrid.cpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) OpenGeoSys Community (opengeosys.org)
// SPDX-License-Identifier: BSD-3-Clause
 
#include "MeshToolsLib/MeshGenerators/AddFaultToVoxelGrid.h"
 
#include <tclap/CmdLine.h>
 
#include <algorithm>
#include <memory>
#include <string>
#include <vector>
 
#include "BaseLib/Logging.h"
#include "BaseLib/MPI.h"
#include "BaseLib/TCLAPArguments.h"
#include "InfoLib/GitInfo.h"
#include "MeshLib/IO/VtkIO/VtuInterface.h"
#include "MeshLib/IO/readMeshFromFile.h"
#include "MeshLib/Mesh.h"
 
int main(int argc, char* argv[])
{
    constexpr int mat_not_set = std::numeric_limits<int>::max();
 
    TCLAP::CmdLine cmd(
        "Marks all elements in a voxel grid (i.e. a structured hex grid, for "
        "instance created with Layers2Grid or Vtu2Grid) that are intersected "
        "by a triangulated 2D mesh representing a fault or some other "
        "significant structure. The material group for those intersected "
        "elements can be explicitly specified, otherwise the largest existing "
        "MaterialID will be increased by one.\n\n"
        "OpenGeoSys-6 software, version " +
            GitInfoLib::GitInfo::ogs_version +
            ".\n"
            "Copyright (c) 2012-2026, OpenGeoSys Community "
            "(http://www.opengeosys.org)",
        ' ', GitInfoLib::GitInfo::ogs_version);
    TCLAP::ValueArg<int> id_arg(
        "m", "material",
        "material id for cells intersected by fault, (min = 0)", false,
        mat_not_set, "MATERIAL_ID");
    cmd.add(id_arg);
 
    TCLAP::ValueArg<std::string> output_arg(
        "o", "output", "Output (.vtu). Name of output mesh file", true, "",
        "OUTPUT_FILE");
    cmd.add(output_arg);
 
    TCLAP::ValueArg<std::string> fault_arg(
        "f", "fault", "Input (.vtu). Name of mesh file representing fault",
        true, "", "INPUT_FILE");
    cmd.add(fault_arg);
 
    TCLAP::ValueArg<std::string> input_arg(
        "i", "input",
        "Input (.vtu). Name of the input file containing the paths the "
        "all input "
        "in correct order from top to bottom",
        true, "", "INPUT_FILE");
    cmd.add(input_arg);
 
    auto log_level_arg = BaseLib::makeLogLevelArg();
    cmd.add(log_level_arg);
    cmd.parse(argc, argv);
 
    BaseLib::MPI::Setup mpi_setup(argc, argv);
    BaseLib::initOGSLogger(log_level_arg.getValue());
 
    std::string const input_name = input_arg.getValue();
    std::string const fault_name = fault_arg.getValue();
    std::string const output_name = output_arg.getValue();
 
    using namespace MeshToolsLib::MeshGenerator::AddFaultToVoxelGrid;
 
    std::unique_ptr<MeshLib::Mesh> mesh(
        MeshLib::IO::readMeshFromFile(input_name));
    std::unique_ptr<MeshLib::Mesh> fault(
        MeshLib::IO::readMeshFromFile(fault_name));
    if (mesh == nullptr)
    {
        ERR("Input mesh not found...");
        return EXIT_FAILURE;
    }
    auto const& mat_ids = MeshLib::materialIDs(*mesh);
    if (!mat_ids)
    {
        ERR("Input mesh has no material IDs");
        return EXIT_FAILURE;
    }
    int fault_id = id_arg.getValue();
    if (!id_arg.isSet())
    {
        auto it = std::max_element(mat_ids->cbegin(), mat_ids->cend());
        fault_id = *it + 1;
    }
    if (addFaultToVoxelGrid(mesh.get(), fault.get(), fault_id))
    {
        MeshLib::IO::VtuInterface vtu(mesh.get());
        vtu.writeToFile(output_name);
        INFO("The fault was successfully added.");
        return EXIT_SUCCESS;
    }
    ERR("No fault could be added.");
    return EXIT_FAILURE;
}