variable_expression_data_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 includes
#include <vector>
#include <type_traits>
 
// Project includes
#include "expression/expression.h"
#include "expression/literal_flat_expression.h"
 
namespace Kratos {
 
 
template<class TDataType>
class KRATOS_API(KRATOS_CORE) VariableExpressionDataIO
{
public:
 
    using Pointer = std::shared_ptr<VariableExpressionDataIO<TDataType>>;
 
    using IndexType = std::size_t;
 
    using RawType = std::conditional_t<
                            std::disjunction_v<
                                std::is_same<TDataType, int>,
                                std::is_same<TDataType, std::vector<int>>
                            >, int, double>;
 
    using RawLiteralFlatExpression = LiteralFlatExpression<RawType>;
 
 
    VariableExpressionDataIO(const TDataType& SampleValue);
 
    VariableExpressionDataIO(const std::vector<IndexType>& rShape);
 
 
    static Pointer Create(const TDataType& SampleValue);
 
    static Pointer Create(const std::vector<IndexType>& rShape);
 
    void Assign(
        TDataType& rOutput,
        const Expression& rExpression,
        const IndexType EntityIndex) const;
 
    void Read(
        RawLiteralFlatExpression& rExpression,
        const IndexType EntityIndex,
        const TDataType& Value) const;
 
    const std::vector<IndexType> GetItemShape() const { return mShape; }
 
private:
 
    std::vector<IndexType> mShape;
 
};
 
} // namespace Kratos