variable_expression_io.h

Go to the documentation of this file.

//    |  /           |
//    ' /   __| _` | __|  _ \   __|
//    . \  |   (   | |   (   |\__ `
//   _|\_\_|  \__,_|\__|\___/ ____/
//                   Multi-Physics
//
//  License:         BSD License
//                   Kratos default license: kratos/license.txt
//
//  Main authors:    Suneth Warnakulasuriya
//
 
#pragma once
 
// System incldues
#include <variant>
 
// Project includes
#include "containers/array_1d.h"
#include "expression/container_expression.h"
#include "expression/traits.h"
#include "containers/variable.h"
#include "expression_io.h"
#include "includes/define.h"
#include "includes/model_part.h"
#include "includes/global_variables.h"
 
namespace Kratos {
 
 
class KRATOS_API(KRATOS_CORE) VariableExpressionIO
{
public:
 
    using VariableType = std::variant<
                                const Variable<int>*,
                                const Variable<double>*,
                                const Variable<array_1d<double, 3>>*,
                                const Variable<array_1d<double, 4>>*,
                                const Variable<array_1d<double, 6>>*,
                                const Variable<array_1d<double, 9>>*,
                                const Variable<Vector>*,
                                const Variable<Matrix>*>;
 
 
    class KRATOS_API(KRATOS_CORE) Input : public ExpressionInput
    {
    public:
 
        KRATOS_CLASS_POINTER_DEFINITION(Input);
 
 
        Input(
            const ModelPart& rModelPart,
            const VariableType& rVariable,
            Globals::DataLocation CurrentLocation,
            MeshType CurrentMeshType = MeshType::Local);
 
        ~Input() override = default;
 
 
        Expression::Pointer Execute() const override;
 
 
    private:
 
        ModelPart const * mpModelPart;
 
        VariableType mpVariable;
 
        Globals::DataLocation mDataLocation;
 
        MeshType mMeshType;
 
 
    };
 
    class KRATOS_API(KRATOS_CORE) Output : public ExpressionOutput
    {
    public:
 
        KRATOS_CLASS_POINTER_DEFINITION(Output);
 
 
        Output(
            ModelPart& rModelPart,
            const VariableType& rVariable,
            Globals::DataLocation CurrentLocation,
            MeshType CurrentMeshType = MeshType::Local);
 
        ~Output() override = default;
 
 
        void Execute(const Expression& rExpression) override;
 
 
    private:
 
        ModelPart * mpModelPart;
 
        VariableType mpVariable;
 
        Globals::DataLocation mDataLocation;
 
        MeshType mMeshType;
 
 
    };
 
 
    template<MeshType TMeshType>
    KRATOS_API(KRATOS_CORE) static void Read(
        ContainerExpression<ModelPart::NodesContainerType, TMeshType>& rContainerExpression,
        const VariableType& rVariable,
        const bool IsHistorical);
 
    template<class TContainerType, MeshType TMeshType>
    KRATOS_API(KRATOS_CORE) static void Read(
        ContainerExpression<TContainerType, TMeshType>& rContainerExpression,
        const VariableType& rVariable);
 
    template<MeshType TMeshType>
    KRATOS_API(KRATOS_CORE) static void Write(
        const ContainerExpression<ModelPart::NodesContainerType, TMeshType>& rContainerExpression,
        const VariableType& rVariable,
        const bool IsHistorical);
 
    template<class TContainerType, MeshType TMeshType>
    KRATOS_API(KRATOS_CORE) static void Write(
        const ContainerExpression<TContainerType, TMeshType>& rContainerExpression,
        const VariableType& rVariable);
 
 
}; // class ExpressionIO
 
 
} // namespace Kratos