assign_scalar_field_to_entities_process.h

Go to the documentation of this file.

//
//    |  /           |
//    ' /   __| _` | __|  _ \   __|
//    . \  |   (   | |   (   |\__ `
//   _|\_\_|  \__,_|\__|\___/ ____/
//                   Multi-Physics
//
//  License:         BSD License
//                   Kratos default license: kratos/license.txt
//
//  Main authors:    Riccardo Rossi
//                   Josep Maria Carbonell
//                   Vicente Mataix Ferrandiz
//
 
#if !defined(KRATOS_ASSIGN_SCALAR_FIELD_TO_ENTITIES_PROCESS_H_INCLUDED )
#define  KRATOS_ASSIGN_SCALAR_FIELD_TO_ENTITIES_PROCESS_H_INCLUDED
 
// System includes
 
// External includes
 
// Project includes
#include "utilities/function_parser_utility.h"
#include "includes/model_part.h"
#include "includes/kratos_parameters.h"
#include "processes/process.h"
 
namespace Kratos
{
 
 
template<class TEntity>
class KRATOS_API(KRATOS_CORE) AssignScalarFieldToEntitiesProcess
    : public Process
{
public:
 
    typedef Node NodeType;
 
    typedef Geometry<NodeType> GeometryType;
 
    typedef std::size_t IndexType;
 
    typedef std::size_t SizeType;
 
    typedef PointerVectorSet<TEntity, IndexedObject> EntityContainerType;
 
    KRATOS_CLASS_POINTER_DEFINITION(AssignScalarFieldToEntitiesProcess);
 
 
    AssignScalarFieldToEntitiesProcess(
        ModelPart& rModelPart,
        Parameters rParameters
        );
 
    ~AssignScalarFieldToEntitiesProcess() override {}
 
 
    void operator()()
    {
        Execute();
    }
 
 
    void Execute() override;
 
    void ExecuteInitializeSolutionStep() override
    {
        Execute();
    }
 
    const Parameters GetDefaultParameters() const override;
 
 
 
 
 
 
    std::string Info() const override
    {
        return "AssignScalarFieldToEntitiesProcess";
    }
 
    void PrintInfo(std::ostream& rOStream) const override
    {
        rOStream << "AssignScalarFieldToEntitiesProcess";
    }
 
    void PrintData(std::ostream& rOStream) const override
    {
    }
 
 
protected:
 
 
 
private:
 
 
    ModelPart& mrModelPart;                     
 
    Kratos::unique_ptr<GenericFunctionUtility> mpFunction; 
 
    std::string mVariableName;                  
 
    std::size_t mMeshId = 0;                    
 
 
    void CallFunction(
        const typename TEntity::Pointer& pEntity,
        const double Time,
        Vector& rValue
        );
 
    void CallFunctionComponents(
        const typename TEntity::Pointer& pEntity,
        const double Time,
        double& rValue
        );
 
    void CallFunctionLocalSystem(
        const typename TEntity::Pointer& pEntity,
        const double Time,
        Vector& rValue
        );
 
    void CallFunctionLocalSystemComponents(
        const typename TEntity::Pointer& pEntity,
        const double Time,
        double& rValue
        );
 
    void AssignTimeDependentValue(
        const typename TEntity::Pointer& pEntity,
        const double Time,
        Vector& rValue,
        const double Value
        );
 
    template< class TVarType >
    void InternalAssignValueVector(
        TVarType& rVar,
        const double Time
        )
    {
        auto& r_entities_array = GetEntitiesContainer();
        const SizeType number_of_entities = r_entities_array.size();
 
        Vector Value;
 
        if(number_of_entities != 0) {
            auto it_begin = r_entities_array.begin();
 
            if(mpFunction->DependsOnSpace()) {
                if(mpFunction->UseLocalSystem()) {
                    // WARNING: do not parallelize with openmp. python GIL prevents it
                    for(IndexType i = 0; i<number_of_entities; ++i) {
                        auto it_entity = it_begin + i;
                        this->CallFunctionLocalSystem(*(it_entity.base()), Time, Value);
                        it_entity->SetValue(rVar, Value);
                    }
                } else {
                    // WARNING: do not parallelize with openmp. python GIL prevents it
                    for(IndexType i = 0; i<number_of_entities; ++i) {
                        auto it_entity = it_begin + i;
                        this->CallFunction(*(it_entity.base()), Time, Value);
                        it_entity->SetValue(rVar, Value);
                    }
                }
            } else { // only varies in time
                const double TimeValue = mpFunction->CallFunction(0.0, 0.0, 0.0,  Time);
                // WARNING: do not parallelize with openmp. python GIL prevents it
                for(IndexType i = 0; i<number_of_entities; ++i) {
                    auto it_entity = it_begin + i;
                    this->AssignTimeDependentValue(*(it_entity.base()), Time, Value,  TimeValue);
                    it_entity->SetValue(rVar, Value);
                }
            }
        }
    }
 
    template< class TVarType >
    void InternalAssignValueScalar(
        TVarType& rVar,
        const double Time
        )
    {
        auto& r_entities_array = GetEntitiesContainer();
        const SizeType number_of_entities = r_entities_array.size();
 
        if(number_of_entities != 0) {
            auto it_begin = r_entities_array.begin();
 
            if(mpFunction->DependsOnSpace()) {
                double Value;
 
                if(mpFunction->UseLocalSystem()) {
                    // WARNING: do not parallelize with openmp. python GIL prevents it
                    for(IndexType i = 0; i<number_of_entities; ++i) {
                        auto it_entity = it_begin + i;
                        this->CallFunctionLocalSystemComponents(*(it_entity.base()), Time, Value);
                        it_entity->SetValue(rVar, Value);
                    }
                } else {
                    // WARNING: do not parallelize with openmp. python GIL prevents it
                    for(IndexType i = 0; i<number_of_entities; ++i) {
                        auto it_entity = it_begin + i;
                        this->CallFunctionComponents(*(it_entity.base()), Time, Value);
                        it_entity->SetValue(rVar, Value);
                    }
                }
            } else { // only varies in time
                const double TimeValue = mpFunction->CallFunction(0.0, 0.0, 0.0,  Time);
                // WARNING: do not parallelize with openmp. python GIL prevents it
                for(IndexType i = 0; i<number_of_entities; ++i) {
                    auto it_entity = it_begin + i;
                    it_entity->SetValue(rVar, TimeValue);
                }
 
            }
        }
    }
 
    EntityContainerType& GetEntitiesContainer();
 
 
    AssignScalarFieldToEntitiesProcess& operator=(AssignScalarFieldToEntitiesProcess const& rOther);
 
 
 
}; // Class AssignScalarFieldToEntitiesProcess
 
 
 
 
 
 
 
template<class TEntity>
inline std::istream& operator >> (std::istream& rIStream,
                                  AssignScalarFieldToEntitiesProcess<TEntity>& rThis);
 
template<class TEntity>
inline std::ostream& operator << (std::ostream& rOStream,
                                  const AssignScalarFieldToEntitiesProcess<TEntity>& rThis)
{
    rThis.PrintInfo(rOStream);
    rOStream << std::endl;
    rThis.PrintData(rOStream);
 
    return rOStream;
}
 
 
}  // namespace Kratos.
 
#endif // KRATOS_ASSIGN_SCALAR_FIELD_TO_ENTITIES_PROCESS_H_INCLUDED  defined