Subversion Repositories slepc-dev

/*

     Routines for setting the bilinear form

   - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

   SLEPc - Scalable Library for Eigenvalue Problem Computations

   Copyright (c) 2002-2010, Universidad Politecnica de Valencia, Spain

   This file is part of SLEPc.

   SLEPc is free software: you can redistribute it and/or modify it under  the

   terms of version 3 of the GNU Lesser General Public License as published by

   the Free Software Foundation.

   SLEPc  is  distributed in the hope that it will be useful, but WITHOUT  ANY

   WARRANTY;  without even the implied warranty of MERCHANTABILITY or  FITNESS

   FOR  A  PARTICULAR PURPOSE. See the GNU Lesser General Public  License  for

   more details.

   You  should have received a copy of the GNU Lesser General  Public  License

   along with SLEPc. If not, see <http://www.gnu.org/licenses/>.

   - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

*/

#include "private/ipimpl.h"      /*I "slepcip.h" I*/

#undef __FUNCT__  

#define __FUNCT__ "IPSetBilinearForm"

/*@

   IPSetBilinearForm - Specifies the bilinear form to be used for

   inner products.

   Collective on IP

   Input Parameters:

+  ip    - the inner product context

.  mat   - the matrix of the bilinear form (may be PETSC_NULL)

-  form  - the type of bilinear form

   Note:

   This function is called by EPSSetProblemType() and usually need not be

   called by the user.

   Level: developer

.seealso: IPGetBilinearForm(), IPInnerProduct(), IPNorm(), EPSSetProblemType(),

          IPBilinearForm

@*/

PetscErrorCode IPSetBilinearForm(IP ip,Mat mat,IPBilinearForm form)

{

  PetscErrorCode ierr;

  PetscFunctionBegin;

  PetscValidHeaderSpecific(ip,IP_CLASSID,1);

  if (mat) {

    PetscValidHeaderSpecific(mat,MAT_CLASSID,1);

    PetscObjectReference((PetscObject)mat);

  }

  if (ip->matrix) {

    ierr = MatDestroy(ip->matrix);CHKERRQ(ierr);

    ierr = VecDestroy(ip->Bx);CHKERRQ(ierr);

  }

  ip->matrix = mat;

  ip->bilinear_form = form;

  ip->xid = ip->xstate = 0;

  if (mat) { ierr = MatGetVecs(mat,&ip->Bx,PETSC_NULL);CHKERRQ(ierr); }

  PetscFunctionReturn(0);

}

#undef __FUNCT__  

#define __FUNCT__ "IPGetBilinearForm"

/*@C

   IPGetBilinearForm - Retrieves the bilinear form to be used for

   inner products.

   Input Parameter:

.  ip    - the inner product context

   Output Parameter:

+  mat   - the matrix of the bilinear form (may be PETSC_NULL)

-  form  - the type of bilinear form

   Level: developer

.seealso: IPSetBilinearForm(), IPInnerProduct(), IPNorm(), EPSSetProblemType(),

          IPBilinearForm

@*/

PetscErrorCode IPGetBilinearForm(IP ip,Mat* mat,IPBilinearForm* form)

{

  PetscFunctionBegin;

  PetscValidHeaderSpecific(ip,IP_CLASSID,1);

  if (mat) *mat = ip->matrix;

  if (form) *form = ip->bilinear_form;

  PetscFunctionReturn(0);

}

#undef __FUNCT__  

#define __FUNCT__ "IPApplyMatrix_Private"

PetscErrorCode IPApplyMatrix_Private(IP ip,Vec x)

{

  PetscErrorCode ierr;

  PetscFunctionBegin;

  if (((PetscObject)x)->id != ip->xid || ((PetscObject)x)->state != ip->xstate) {

    ierr = PetscLogEventBegin(IP_ApplyMatrix,ip,0,0,0);CHKERRQ(ierr);

    ierr = MatMult(ip->matrix,x,ip->Bx);CHKERRQ(ierr);

    ip->xid = ((PetscObject)x)->id;

    ip->xstate = ((PetscObject)x)->state;

    ierr = PetscLogEventEnd(IP_ApplyMatrix,ip,0,0,0);CHKERRQ(ierr);

  }

  PetscFunctionReturn(0);  

}

#undef __FUNCT__  

#define __FUNCT__ "IPApplyMatrix"

/*@

   IPApplyMatrix - Multiplies a vector with the matrix associated to the

                   bilinear form.

   Collective on IP

   Input Parameters:

+  ip    - the inner product context

-  x     - the vector

   Output Parameter:

.  y     - the result  

   Note:

   If the bilinear form has no associated matrix this function copies the vector.

   Level: developer

.seealso: IPSetBilinearForm(), IPInnerProduct(), IPNorm(), EPSSetProblemType()

@*/

PetscErrorCode IPApplyMatrix(IP ip,Vec x,Vec y)

{

  PetscErrorCode ierr;

  PetscFunctionBegin;

  PetscValidHeaderSpecific(ip,IP_CLASSID,1);

  if (ip->matrix) {

    ierr = IPApplyMatrix_Private(ip,x);CHKERRQ(ierr);

    ierr = VecCopy(ip->Bx,y);CHKERRQ(ierr);

  } else {

    ierr = VecCopy(x,y);CHKERRQ(ierr);

  }

  PetscFunctionReturn(0);  

}