postLIE.cpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) OpenGeoSys Community (opengeosys.org)
// SPDX-License-Identifier: BSD-3-Clause
 
#include <tclap/CmdLine.h>
 
#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/xml_parser.hpp>
#include <map>
#include <memory>
#include <vector>
 
#include "BaseLib/FileTools.h"
#include "BaseLib/Logging.h"
#include "BaseLib/MPI.h"
#include "BaseLib/TCLAPArguments.h"
#include "InfoLib/GitInfo.h"
#include "MeshLib/IO/readMeshFromFile.h"
#include "MeshLib/IO/writeMeshToFile.h"
#include "MeshLib/Mesh.h"
#include "MeshToolsLib/MeshEditing/ConvertToLinearMesh.h"
#include "ProcessLib/LIE/Common/MeshUtils.h"
#include "ProcessLib/LIE/Common/PostUtils.h"
 
namespace
{
void postVTU(std::string const& int_vtu_filename,
             std::string const& out_vtu_filename)
{
    // read VTU with simulation results
    std::unique_ptr<MeshLib::Mesh const> mesh(
        MeshLib::IO::readMeshFromFile(int_vtu_filename));
    if (mesh->hasNonlinearElement())
    {
        mesh = MeshToolsLib::convertToLinearMesh(*mesh, mesh->getName());
    }
 
    // post-process
    std::vector<MeshLib::Element*> vec_matrix_elements;
    std::vector<int> vec_fracture_mat_IDs;
    std::vector<std::vector<MeshLib::Element*>> vec_fracture_elements;
    std::vector<std::vector<MeshLib::Element*>> vec_fracture_matrix_elements;
    std::vector<std::vector<MeshLib::Node*>> vec_fracture_nodes;
    std::vector<std::pair<std::size_t, std::vector<int>>>
        vec_branch_nodeID_matIDs;
    std::vector<std::pair<std::size_t, std::vector<int>>>
        vec_junction_nodeID_matIDs;
    ProcessLib::LIE::getFractureMatrixDataInMesh(
        *mesh, vec_matrix_elements, vec_fracture_mat_IDs, vec_fracture_elements,
        vec_fracture_matrix_elements, vec_fracture_nodes,
        vec_branch_nodeID_matIDs, vec_junction_nodeID_matIDs);
 
    ProcessLib::LIE::PostProcessTool post(*mesh,
                                          vec_fracture_mat_IDs,
                                          vec_fracture_nodes,
                                          vec_fracture_matrix_elements,
                                          vec_branch_nodeID_matIDs,
                                          vec_junction_nodeID_matIDs);
 
    // create a new VTU file
    INFO("create {:s}", out_vtu_filename);
    if (MeshLib::IO::writeMeshToFile(post.getOutputMesh(), out_vtu_filename) !=
        0)
    {
        return;
    }
}
 
void postPVD(std::string const& in_pvd_filename,
             std::string const& out_pvd_filename,
             bool const allow_overwrite)
{
    auto const in_pvd_file_dir = BaseLib::extractPath(in_pvd_filename);
    auto const out_pvd_file_dir = BaseLib::extractPath(out_pvd_filename);
    INFO("start reading the PVD file {:s}", in_pvd_filename);
    boost::property_tree::ptree pt;
    read_xml(in_pvd_filename, pt,
             boost::property_tree::xml_parser::trim_whitespace);
 
    for (auto& dataset : pt.get_child("VTKFile.Collection"))
    {
        if (dataset.first != "DataSet")
        {
            continue;
        }
 
        // get VTU file name
        auto const org_vtu_filename =
            dataset.second.get<std::string>("<xmlattr>.file");
        auto const org_vtu_filebasename =
            BaseLib::extractBaseName(org_vtu_filename);
        auto org_vtu_dir = BaseLib::extractPath(org_vtu_filename);
        if (org_vtu_dir.empty())
        {
            org_vtu_dir = in_pvd_file_dir;
        }
        auto const org_vtu_filepath =
            BaseLib::joinPaths(org_vtu_dir, org_vtu_filebasename);
        INFO("processing {:s}...", org_vtu_filepath);
 
        // post-process the VTU and save into the new file
        auto const dest_vtu_filename = "post_" + org_vtu_filebasename;
        auto const dest_vtu_filepath =
            BaseLib::joinPaths(out_pvd_file_dir, dest_vtu_filename);
        if (!allow_overwrite && BaseLib::IsFileExisting(dest_vtu_filepath))
        {
            INFO("The destination file already exists. Skip overwriting it.");
        }
        else
        {
            postVTU(org_vtu_filepath, dest_vtu_filepath);
        }
 
        // create a new VTU file and update XML
        dataset.second.put("<xmlattr>.file", dest_vtu_filename);
    }
 
    // save into the new PVD file
    INFO("save into the new PVD file {:s}", out_pvd_filename);
    boost::property_tree::xml_writer_settings<std::string> settings('\t', 1);
    write_xml(out_pvd_filename, pt, std::locale(), settings);
}
 
}  // unnamed namespace
 
int main(int argc, char* argv[])
{
    TCLAP::CmdLine cmd(
        "Post-process results of the LIE approach.\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<std::string> arg_out_file(
        "o", "output-file",
        "Output (.vtu | .pvd). The name of the new PVD or VTU file", true, "",
        "OUTPUT_PATH");
    cmd.add(arg_out_file);
    TCLAP::ValueArg<std::string> arg_in_file(
        "i", "input-file",
        "Input (.vtu | .pvd). The original PVD or VTU file name", true, "",
        "INPUT_FILE");
    cmd.add(arg_in_file);
    TCLAP::SwitchArg nooverwrite_arg(
        "",
        "no-overwrite",
        "don't overwrite existing post processed VTU files");
    cmd.add(nooverwrite_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());
 
    auto const in_file_ext = BaseLib::getFileExtension(arg_in_file.getValue());
    if (in_file_ext == ".pvd")
    {
        postPVD(arg_in_file.getValue(), arg_out_file.getValue(),
                !nooverwrite_arg.getValue());
    }
    else if (in_file_ext == ".vtu")
    {
        postVTU(arg_in_file.getValue(), arg_out_file.getValue());
    }
    else
    {
        OGS_FATAL("The given file type ({:s}) is not supported.", in_file_ext);
    }
 
    return EXIT_SUCCESS;
}