SimpleArcLengthControl Struct Reference

#include "users_modules/basic_finite_elements/src/ArcLengthTools.hpp"

Inheritance diagram for SimpleArcLengthControl:

Collaboration diagram for SimpleArcLengthControl:

Public Member Functions
	SimpleArcLengthControl (boost::shared_ptr< ArcLengthCtx > &arc_ptr, const bool assemble=false)
	~SimpleArcLengthControl ()
MoFEMErrorCode	preProcess ()
MoFEMErrorCode	operator() ()
MoFEMErrorCode	postProcess ()
double	calculateLambdaInt ()
	Calculate internal lambda.
MoFEMErrorCode	calculateDb ()
	Calculate db.
MoFEMErrorCode	calculateDxAndDlambda (Vec x)

Public Attributes
boost::shared_ptr< ArcLengthCtx >	arcPtr
const bool	aSsemble

Detailed Description

|brief Simple arc-length control of force

This is added for testing, it simply control force, i.e.

\[ \lambda = s \]

Constructor takes one argument,

Parameters

arc_ptr

Pointer to arc-length CTX.

Definition at line 334 of file ArcLengthTools.hpp.

Constructor & Destructor Documentation

SimpleArcLengthControl()

SimpleArcLengthControl::SimpleArcLengthControl	(	boost::shared_ptr< ArcLengthCtx > &	arc_ptr,
		const bool	assemble = false )

Definition at line 521 of file ArcLengthTools.cpp.

    : FEMethod(), arcPtr(arc_ptr), aSsemble(assemble) {}

~SimpleArcLengthControl()

SimpleArcLengthControl::~SimpleArcLengthControl

(

)

Definition at line 525 of file ArcLengthTools.cpp.

{}

Member Function Documentation

calculateDb()

MoFEMErrorCode SimpleArcLengthControl::calculateDb

(

)

Calculate db.

Definition at line 597 of file ArcLengthTools.cpp.

                                                   {
  MoFEMFunctionBegin;
  CHKERR VecZeroEntries(arcPtr->db);
  CHKERR VecGhostUpdateBegin(arcPtr->db, INSERT_VALUES, SCATTER_FORWARD);
  CHKERR VecGhostUpdateEnd(arcPtr->db, INSERT_VALUES, SCATTER_FORWARD);
  MoFEMFunctionReturn(0);
}

calculateDxAndDlambda()

MoFEMErrorCode SimpleArcLengthControl::calculateDxAndDlambda

(

Vec

x

)

Definition at line 605 of file ArcLengthTools.cpp.

                                                                  {
  ArcFunctionBegin;
  // Calculate dx
  CHKERR VecGhostUpdateBegin(arcPtr->x0, INSERT_VALUES, SCATTER_FORWARD);
  CHKERR VecGhostUpdateEnd(arcPtr->x0, INSERT_VALUES, SCATTER_FORWARD);
 
  Vec l_x, l_x0, l_dx;
  CHKERR VecGhostGetLocalForm(x, &l_x);
  CHKERR VecGhostGetLocalForm(arcPtr->x0, &l_x0);
  CHKERR VecGhostGetLocalForm(arcPtr->dx, &l_dx);
  {
    double *x_array, *x0_array, *dx_array;
    CHKERR VecGetArray(l_x, &x_array);
    CHKERR VecGetArray(l_x0, &x0_array);
    CHKERR VecGetArray(l_dx, &dx_array);
    int size =
        problemPtr->getNbLocalDofsRow() + problemPtr->getNbGhostDofsRow();
    for (int i = 0; i != size; ++i) {
      dx_array[i] = x_array[i] - x0_array[i];
    }
    CHKERR VecRestoreArray(l_x, &x_array);
    CHKERR VecRestoreArray(l_x0, &x0_array);
    CHKERR VecRestoreArray(l_dx, &dx_array);
  }
  CHKERR VecGhostRestoreLocalForm(x, &l_x);
  CHKERR VecGhostRestoreLocalForm(arcPtr->x0, &l_x0);
  CHKERR VecGhostRestoreLocalForm(arcPtr->dx, &l_dx);
 
  // Calculate dlambda
  if (arcPtr->getPetscLocalDofIdx() != -1) {
    double *array;
    CHKERR VecGetArray(arcPtr->dx, &array);
    arcPtr->dLambda = array[arcPtr->getPetscLocalDofIdx()];
    array[arcPtr->getPetscLocalDofIdx()] = 0;
    CHKERR VecRestoreArray(arcPtr->dx, &array);
  }
  CHKERR VecGhostUpdateBegin(arcPtr->ghosTdLambda, INSERT_VALUES,
                             SCATTER_FORWARD);
  CHKERR VecGhostUpdateEnd(arcPtr->ghosTdLambda, INSERT_VALUES,
                           SCATTER_FORWARD);
 
  // Calculate dx2
  double x_nrm, x0_nrm;
  CHKERR VecNorm(x, NORM_2, &x_nrm);
  CHKERR VecNorm(arcPtr->x0, NORM_2, &x0_nrm);
  CHKERR VecDot(arcPtr->dx, arcPtr->dx, &arcPtr->dx2);
 
  MOFEM_LOG_C("WORLD", Sev::verbose,
              "\tx norm = %6.4e x0 norm = %6.4e dx2 = %6.4e", x_nrm, x0_nrm,
              arcPtr->dx2);
  MoFEMFunctionReturn(0);
}

calculateLambdaInt()

double SimpleArcLengthControl::calculateLambdaInt

(

)

Calculate internal lambda.

Definition at line 593 of file ArcLengthTools.cpp.

                                                  {
  return arcPtr->beta * arcPtr->dLambda;
}

operator()()

MoFEMErrorCode SimpleArcLengthControl::operator()

(

)

Definition at line 550 of file ArcLengthTools.cpp.

                                                  {
  MoFEMFunctionBegin;
  switch (snes_ctx) {
  case CTX_SNESSETFUNCTION: {
    arcPtr->res_lambda = calculateLambdaInt() - arcPtr->s;
    CHKERR VecSetValue(snes_f, arcPtr->getPetscGlobalDofIdx(),
                       arcPtr->res_lambda, ADD_VALUES);
  } break;
  case CTX_SNESSETJACOBIAN: {
    arcPtr->dIag = arcPtr->beta;
    CHKERR MatSetValue(snes_B, arcPtr->getPetscGlobalDofIdx(),
                       arcPtr->getPetscGlobalDofIdx(), 1, ADD_VALUES);
  } break;
  default:
    break;
  }
  MoFEMFunctionReturn(0);
}

postProcess()

MoFEMErrorCode SimpleArcLengthControl::postProcess

(

)

Definition at line 569 of file ArcLengthTools.cpp.

                                                   {
  MoFEMFunctionBegin;
  switch (snes_ctx) {
  case CTX_SNESSETFUNCTION: {
    if (aSsemble) {
      CHKERR VecAssemblyBegin(snes_f);
      CHKERR VecAssemblyEnd(snes_f);
    }
  } break;
  case CTX_SNESSETJACOBIAN: {
    if (aSsemble) {
      CHKERR MatAssemblyBegin(snes_B, MAT_FLUSH_ASSEMBLY);
      CHKERR MatAssemblyEnd(snes_B, MAT_FLUSH_ASSEMBLY);
    }
    CHKERR VecGhostUpdateBegin(arcPtr->ghostDiag, INSERT_VALUES,
                               SCATTER_FORWARD);
    CHKERR VecGhostUpdateEnd(arcPtr->ghostDiag, INSERT_VALUES, SCATTER_FORWARD);
  } break;
  default:
    break;
  }
  MoFEMFunctionReturn(0);
}

preProcess()

MoFEMErrorCode SimpleArcLengthControl::preProcess

(

)

Definition at line 527 of file ArcLengthTools.cpp.

                                                  {
  MoFEMFunctionBegin;
  switch (snes_ctx) {
  case CTX_SNESSETFUNCTION: {
    if (aSsemble) {
      CHKERR VecAssemblyBegin(snes_f);
      CHKERR VecAssemblyEnd(snes_f);
    }
    CHKERR calculateDxAndDlambda(snes_x);
    CHKERR calculateDb();
  } break;
  case CTX_SNESSETJACOBIAN: {
    if (aSsemble) {
      CHKERR MatAssemblyBegin(snes_B, MAT_FLUSH_ASSEMBLY);
      CHKERR MatAssemblyEnd(snes_B, MAT_FLUSH_ASSEMBLY);
    }
  } break;
  default:
    break;
  }
  MoFEMFunctionReturn(0);
}

Member Data Documentation

arcPtr

boost::shared_ptr<ArcLengthCtx> SimpleArcLengthControl::arcPtr

Definition at line 336 of file ArcLengthTools.hpp.

aSsemble

const bool SimpleArcLengthControl::aSsemble

Definition at line 337 of file ArcLengthTools.hpp.

---

The documentation for this struct was generated from the following files:

• users_modules/basic_finite_elements/src/ArcLengthTools.hpp
• users_modules/basic_finite_elements/src/impl/ArcLengthTools.cpp