dof_updater.h

Go to the documentation of this file.

//    |  /           |
//    ' /   __| _` | __|  _ \   __|
//    . \  |   (   | |   (   |\__ `
//   _|\_\_|  \__,_|\__|\___/ ____/
//                   Multi-Physics
//
//  License:         BSD License
//                   Kratos default license: kratos/license.txt
//
//  Main authors:    Jordi Cotela
//
 
#if !defined(KRATOS_DOF_UPDATER_H_INCLUDED )
#define  KRATOS_DOF_UPDATER_H_INCLUDED
 
// Project includes
#include "includes/define.h"
#include "includes/model_part.h"
#include "utilities/parallel_utilities.h"
 
namespace Kratos
{
 
 
 
template< class TSparseSpace >
class DofUpdater
{
public:
 
    KRATOS_CLASS_POINTER_DEFINITION(DofUpdater);
 
    using DofType = Dof<typename TSparseSpace::DataType>;
    using DofsArrayType = PointerVectorSet<DofType>;
 
    using SystemVectorType = typename TSparseSpace::VectorType;
 
 
    DofUpdater(){}
 
    DofUpdater(DofUpdater const& rOther) = delete;
 
    virtual ~DofUpdater(){}
 
    DofUpdater& operator=(DofUpdater const& rOther) = delete;
 
 
 
    virtual typename DofUpdater::UniquePointer Create() const
    {
        return Kratos::make_unique<DofUpdater>();
    }
 
 
    virtual void Initialize(
        const DofsArrayType& rDofSet,
        const SystemVectorType& rDx)
    {}
 
 
    virtual void Clear() {}
 
 
    virtual void UpdateDofs(
        DofsArrayType& rDofSet,
        const SystemVectorType& rDx)
    {
        block_for_each(
            rDofSet,
            [&rDx](DofType& rDof)
            {
                if (rDof.IsFree()) {
                    rDof.GetSolutionStepValue() += TSparseSpace::GetValue(rDx, rDof.EquationId());
        }
            }
        );
    }
 
 
    virtual void AssignDofs(DofsArrayType& rDofSet, const SystemVectorType& rX)
    {
        block_for_each(
            rDofSet,
            [&rX](DofType& rDof)
            {
                if (rDof.IsFree()) {
                    rDof.GetSolutionStepValue() = TSparseSpace::GetValue(rX, rDof.EquationId());
        }
            }
        );
    }
 
 
    virtual std::string Info() const
    {
        std::stringstream buffer;
        buffer << "DofUpdater" ;
        return buffer.str();
    }
 
    virtual void PrintInfo(std::ostream& rOStream) const
    {
        rOStream << this->Info() << std::endl;
    }
 
    virtual void PrintData(std::ostream& rOStream) const
    {
        rOStream << this->Info() << std::endl;
    }
 
 
}; // Class DofUpdater
 
 
 
template< class TSparseSpace >
inline std::istream& operator >> (
    std::istream& rIStream,
    DofUpdater<TSparseSpace>& rThis)
{
    return rIStream;
}
 
template< class TSparseSpace >
inline std::ostream& operator << (
    std::ostream& rOStream,
    const DofUpdater<TSparseSpace>& rThis)
{
    rThis.PrintInfo(rOStream);
    rOStream << std::endl;
    rThis.PrintData(rOStream);
 
    return rOStream;
}
 
 
 
}  // namespace Kratos.
 
#endif // KRATOS_DOF_UPDATER_H_INCLUDED  defined