MatrixOutput.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) OpenGeoSys Community (opengeosys.org)
// SPDX-License-Identifier: BSD-3-Clause
 
#pragma once
 
#include <fstream>
#include <functional>
#include <mutex>
#include <unordered_set>
 
#include "MathLib/LinAlg/GlobalMatrixVectorTypes.h"
 
namespace ProcessLib::Assembly
{
struct GlobalMatrixOutput
{
    GlobalMatrixOutput();
 
    void operator()(double const t, int const process_id, GlobalMatrix const& M,
                    GlobalMatrix const& K, GlobalVector const& b);
 
    void operator()(double const t, int const process_id, GlobalVector const& b,
                    GlobalMatrix const& Jac);
 
private:
    std::string filenamePrefix_;

    std::size_t counter_ = 0;
 
    bool do_output_ = false;
};

struct LocalMatrixOutput
{
    LocalMatrixOutput();
 
    void operator()(double const t, int const process_id,
                    std::size_t const element_id,
                    std::vector<double> const& local_M_data,
                    std::vector<double> const& local_K_data,
                    std::vector<double> const& local_b_data);
 
    void operator()(double const t, int const process_id,
                    std::size_t const element_id,
                    std::vector<double> const& local_b_data,
                    std::vector<double> const& local_Jac_data);
 
    ~LocalMatrixOutput();
 
private:
    bool isOutputRequested(std::size_t const element_id) const;
 
    std::mutex mutex_;
    std::ofstream outputFile_;
    std::function<bool(std::size_t)> output_element_predicate_;
};
 
namespace detail
{
// helper function, exposed for unit testing, only
std::function<bool(std::size_t)> createLocalMatrixOutputElementPredicate(
    std::string const& element_ids_str);
 
// helper function, exposed for unit testing, only
std::unordered_set<std::size_t> parseSetOfSizeT(std::string const& str,
                                                std::string const& warn_msg);
}  // namespace detail
}  // namespace ProcessLib::Assembly