PythonSourceTerm.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 "ProcessLib/BoundaryConditionAndSourceTerm/SourceTerm.h"
#include "PythonSourceTermLocalAssemblerInterface.h"
#include "PythonSourceTermPythonSideInterface.h"
#include "Utils/BcOrStData.h"
 
namespace ProcessLib
{
class ProcessVariable;
 
namespace SourceTerms::Python
{
struct PythonStData final
    : ProcessLib::BoundaryConditionAndSourceTerm::Python::BcOrStData<
          PythonSourceTermPythonSideInterface>
{
    ProcessLib::BoundaryConditionAndSourceTerm::Python::FlagAndFluxAndDFlux
    getFlagAndFluxAndDFlux(double const t, std::array<double, 3> const coords,
                           std::vector<double> const& prim_vars_data) const
    {
        auto [flux, dFlux] =
            bc_or_st_object->getFlux(t, coords, prim_vars_data);
 
        return {true, flux, std::move(dFlux)};
    }
};

class PythonSourceTerm final : public ProcessLib::SourceTerm
{
public:
    explicit PythonSourceTerm(
        std::unique_ptr<NumLib::LocalToGlobalIndexMap> source_term_dof_table,
        PythonStData&& source_term_data, unsigned const integration_order,
        unsigned const global_dim, bool const flush_stdout);
 
    void integrate(const double t, GlobalVector const& x, GlobalVector& b,
                   GlobalMatrix* jac) const override;
 
private:
    PythonStData _source_term_data;

    std::vector<std::unique_ptr<PythonSourceTermLocalAssemblerInterface>>
        _local_assemblers;

    bool const _flush_stdout;
};
 
}  // namespace SourceTerms::Python
}  // namespace ProcessLib