fix_scalar_dof_process.hpp

Go to the documentation of this file.

//
//   Project Name:        KratosSolidMechanicsApplication $
//   Created by:          $Author:            JMCarbonell $
//   Last modified by:    $Co-Author:                     $
//   Date:                $Date:              August 2016 $
//   Revision:            $Revision:                  0.0 $
//
//
 
#if !defined(KRATOS_FIX_SCALAR_DOF_PROCESS_H_INCLUDED)
#define  KRATOS_FIX_SCALAR_DOF_PROCESS_H_INCLUDED
 
 
 
// System includes
 
// External includes
 
// Project includes
#include "includes/model_part.h"
#include "includes/kratos_parameters.h"
#include "processes/process.h"
 
namespace Kratos
{
 
 
 
class FixScalarDofProcess : public Process
{
public:
 
    KRATOS_CLASS_POINTER_DEFINITION(FixScalarDofProcess);
 
    FixScalarDofProcess(ModelPart& model_part,
            Parameters rParameters
            ) : Process() , mrModelPart(model_part)
    {
        KRATOS_TRY
 
        Parameters default_parameters( R"(
            {
                "model_part_name":"PLEASE_CHOOSE_MODEL_PART_NAME",
                "variable_name": "PLEASE_PRESCRIBE_VARIABLE_NAME"
            }  )" );
 
 
        // Validate against defaults -- this ensures no type mismatch
        rParameters.ValidateAndAssignDefaults(default_parameters);
 
        mvariable_name = rParameters["variable_name"].GetString();
 
        if( KratosComponents< Variable<double> >::Has( mvariable_name ) ) //case of double variable
        {
            if( model_part.GetNodalSolutionStepVariablesList().Has( KratosComponents< Variable<double> >::Get( mvariable_name ) ) == false )
            {
                KRATOS_THROW_ERROR(std::runtime_error,"trying to set a variable that is not in the model_part - variable name is ",mvariable_name);
            }
 
        }
        else if( KratosComponents< Variable<int> >::Has( mvariable_name ) ) //case of int variable
        {
            if( model_part.GetNodalSolutionStepVariablesList().Has( KratosComponents< Variable<int> >::Get( mvariable_name ) ) == false )
            {
                KRATOS_THROW_ERROR(std::runtime_error,"trying to set a variable that is not in the model_part - variable name is ",mvariable_name);
            }
 
        }
        else if( KratosComponents< Variable<bool> >::Has( mvariable_name ) ) //case of bool variable
        {
      if( model_part.GetNodalSolutionStepVariablesList().Has(KratosComponents< Variable<bool> >::Get( mvariable_name ) ) == false )
            {
                KRATOS_THROW_ERROR(std::runtime_error,"trying to set a variable that is not in the model_part - variable name is ",mvariable_name);
            }
        }
 
        KRATOS_CATCH("");
    }
 
    FixScalarDofProcess(ModelPart& model_part,
            const Variable<double>& rVariable) : Process(), mrModelPart(model_part)
    {
        KRATOS_TRY;
 
 
        if( model_part.GetNodalSolutionStepVariablesList().Has( rVariable ) == false )
        {
                KRATOS_THROW_ERROR(std::runtime_error,"trying to set a variable that is not in the model_part - variable name is ",rVariable);
        }
 
        mvariable_name = rVariable.Name();
 
        KRATOS_CATCH("");
    }
 
 
    FixScalarDofProcess(ModelPart& model_part,
            const Variable< int >& rVariable) : Process(), mrModelPart(model_part)
    {
        KRATOS_TRY;
 
        if( model_part.GetNodalSolutionStepVariablesList().Has( rVariable ) == false )
        {
                KRATOS_THROW_ERROR(std::runtime_error,"Trying to set a variable that is not in the model_part - variable name is ",rVariable);
        }
 
    mvariable_name = rVariable.Name();
 
        KRATOS_CATCH("");
    }
 
    FixScalarDofProcess(ModelPart& model_part,
            const Variable< bool >& rVariable) : Process(), mrModelPart(model_part)
    {
        KRATOS_TRY;
 
 
        if( model_part.GetNodalSolutionStepVariablesList().Has( rVariable ) == false )
        {
                KRATOS_THROW_ERROR(std::runtime_error,"Trying to set a variable that is not in the model_part - variable name is ",rVariable);
        }
 
        mvariable_name = rVariable.Name();
 
        KRATOS_CATCH("");
    }
 
 
    ~FixScalarDofProcess() override {}
 
 
 
    void operator()()
    {
        Execute();
    }
 
 
 
 
    void Execute()  override
    {
 
        KRATOS_TRY;
 
        if( KratosComponents< Variable<double> >::Has( mvariable_name ) ) //case of double variable
        {
            InternalFixDof<>(KratosComponents< Variable<double> >::Get(mvariable_name));
        }
        else
        {
            KRATOS_THROW_ERROR(std::logic_error, "Not able to set the variable. Attempting to set variable:",mvariable_name);
        }
 
        KRATOS_CATCH("");
 
    }
 
    void ExecuteInitialize() override
    {
    }
 
    void ExecuteBeforeSolutionLoop() override
    {
    }
 
 
    void ExecuteInitializeSolutionStep() override
    {
    }
 
    void ExecuteFinalizeSolutionStep() override
    {
    }
 
 
    void ExecuteBeforeOutputStep() override
    {
    }
 
 
    void ExecuteAfterOutputStep() override
    {
    }
 
 
    void ExecuteFinalize() override
    {
    }
 
 
 
 
 
 
 
    std::string Info() const override
    {
        return "FixScalarDofProcess";
    }
 
    void PrintInfo(std::ostream& rOStream) const override
    {
        rOStream << "FixScalarDofProcess";
    }
 
    void PrintData(std::ostream& rOStream) const override
    {
    }
 
 
 
protected:
 
 
    FixScalarDofProcess(FixScalarDofProcess const& rOther);
 
 
private:
 
 
    ModelPart& mrModelPart;
    std::string mvariable_name;
 
 
    template< class TVarType >
    void InternalFixDof(TVarType& rVar)
    {
        const int nnodes = mrModelPart.GetMesh().Nodes().size();
 
        if(nnodes != 0)
        {
            ModelPart::NodesContainerType::iterator it_begin = mrModelPart.GetMesh().NodesBegin();
 
            #pragma omp parallel for
            for(int i = 0; i<nnodes; i++)
            {
                ModelPart::NodesContainerType::iterator it = it_begin + i;
        //it->pAddDof(rVar)->FixDof();
        it->Fix(rVar);
 
           }
        }
    }
 
 
    FixScalarDofProcess& operator=(FixScalarDofProcess const& rOther);
 
 
 
}; // Class FixScalarDofProcess
 
 
 
 
 
 
inline std::istream& operator >> (std::istream& rIStream,
                                  FixScalarDofProcess& rThis);
 
inline std::ostream& operator << (std::ostream& rOStream,
                                  const FixScalarDofProcess& rThis)
{
    rThis.PrintInfo(rOStream);
    rOStream << std::endl;
    rThis.PrintData(rOStream);
 
    return rOStream;
}
 
 
}  // namespace Kratos.
 
#endif // KRATOS_FIX_SCALAR_DOF_PROCESS_H_INCLUDED  defined