MatrixOutput.h

Go to the documentation of this file.

 
#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,
                    GlobalMatrix const* const Jac = nullptr);
 
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,
                    std::vector<double> const* const local_Jac_data = nullptr);
 
    ~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