xref: /petsc/src/snes/linesearch/impls/shell/linesearchshell.c (revision 954494b2f105144292715050ea1b8966feeb7ce6)

#include <private/linesearchimpl.h>
#include <private/snesimpl.h>


typedef struct {
  PetscLineSearchUserFunc func;
  void               *ctx;
} PetscLineSearch_Shell;

#undef __FUNCT__
#define __FUNCT__ "PetscLineSearchShellSetUserFunc"
/*@C
   PetscLineSearchShellSetUserFunc - Sets the user function for the PetscLineSearch Shell implementation.

   Not Collective

   Level: advanced

   .keywords: PetscLineSearch, PetscLineSearchShell, Shell

   .seealso: PetscLineSearchShellGetUserFunc()
@*/
PetscErrorCode PetscLineSearchShellSetUserFunc(PetscLineSearch linesearch, PetscLineSearchUserFunc func, void *ctx) {

  PetscErrorCode   ierr;
  PetscBool        flg;
  PetscLineSearch_Shell *shell = (PetscLineSearch_Shell *)linesearch->data;
  PetscFunctionBegin;
  PetscValidHeaderSpecific(linesearch, PETSCLINESEARCH_CLASSID, 1);
  ierr = PetscTypeCompare((PetscObject)linesearch,PETSCLINESEARCHSHELL,&flg);CHKERRQ(ierr);
  if (flg) {
    shell->ctx = ctx;
    shell->func = func;
  }
  PetscFunctionReturn(0);
}


#undef __FUNCT__
#define __FUNCT__ "PetscLineSearchShellGetUserFunc"
/*@C
   PetscLineSearchShellGetUserFunc - Gets the user function and context for the shell implementation.

   Not Collective

   Level: advanced

   .keywords: PetscLineSearch, PetscLineSearchShell, Shell

   .seealso: PetscLineSearchShellSetUserFunc()
@*/
PetscErrorCode PetscLineSearchShellGetUserFunc(PetscLineSearch linesearch, PetscLineSearchUserFunc *func, void **ctx) {

  PetscErrorCode   ierr;
  PetscBool        flg;
  PetscLineSearch_Shell *shell = (PetscLineSearch_Shell *)linesearch->data;
  PetscFunctionBegin;
  PetscValidHeaderSpecific(linesearch, PETSCLINESEARCH_CLASSID, 1);
  if (func) PetscValidPointer(func,2);
  if (ctx)  PetscValidPointer(ctx,3);
  ierr = PetscTypeCompare((PetscObject)linesearch,PETSCLINESEARCHSHELL,&flg);CHKERRQ(ierr);
  if (flg) {
    *ctx  = shell->ctx;
    *func = shell->func;
  }
  PetscFunctionReturn(0);
}


#undef __FUNCT__
#define __FUNCT__ "PetscLineSearchApply_Shell"
static PetscErrorCode  PetscLineSearchApply_Shell(PetscLineSearch linesearch)
{
  PetscLineSearch_Shell *shell = (PetscLineSearch_Shell *)linesearch->data;
  PetscErrorCode   ierr;

  PetscFunctionBegin;

  /* apply the user function */
  if (shell->func) {
    ierr = (*shell->func)(linesearch, shell->ctx);CHKERRQ(ierr);
  } else {
    SETERRQ(((PetscObject)linesearch)->comm, PETSC_ERR_USER, "PetscLineSearchShell needs to have a shell function set with PetscLineSearchShellSetUserFunc");
  }
  PetscFunctionReturn(0);
}

#undef __FUNCT__
#define __FUNCT__ "PetscLineSearchDestroy_Shell"
static PetscErrorCode  PetscLineSearchDestroy_Shell(PetscLineSearch linesearch)
{
  PetscLineSearch_Shell *shell = (PetscLineSearch_Shell *)linesearch->data;
  PetscErrorCode   ierr;

  PetscFunctionBegin;
  ierr = PetscFree(shell);CHKERRQ(ierr);
  PetscFunctionReturn(0);
}

#undef __FUNCT__
#define __FUNCT__ "PetscLineSearchCreate_Shell"
/*MC

PETSCLINESEARCHSHELL - Provides context for a user-provided line search routine.

The user routine has one argument, the PetscLineSearch context.  The user uses the interface to
extract line search parameters and set them accordingly when the computation is finished.

Any of the other line searches may serve as a guide to how this is to be done.

Level: advanced

M*/
PETSC_EXTERN_C PetscErrorCode PetscLineSearchCreate_Shell(PetscLineSearch linesearch)
{

  PetscLineSearch_Shell     *shell;
  PetscErrorCode       ierr;

  PetscFunctionBegin;

  linesearch->ops->apply          = PetscLineSearchApply_Shell;
  linesearch->ops->destroy        = PetscLineSearchDestroy_Shell;
  linesearch->ops->setfromoptions = PETSC_NULL;
  linesearch->ops->reset          = PETSC_NULL;
  linesearch->ops->view           = PETSC_NULL;
  linesearch->ops->setup          = PETSC_NULL;

  ierr = PetscNewLog(linesearch, PetscLineSearch_Shell, &shell);CHKERRQ(ierr);
  linesearch->data = (void*) shell;
  PetscFunctionReturn(0);
}