DirichletBC.cc

//
// Software License for AMDiS
//
// Copyright (c) 2010 Dresden University of Technology 
// All rights reserved.
// Authors: Simon Vey, Thomas Witkowski et al.
//
// This file is part of AMDiS
//
// See also license.opensource.txt in the distribution.


#include "DirichletBC.h"
#include "ElInfo.h"
#include "BasisFunction.h"
#include "DOFVector.h"
#include "DOFMatrix.h"

namespace AMDiS {

  DirichletBC::DirichletBC(BoundaryType type,
			   AbstractFunction<double, WorldVector<double> > *fct,
			   const FiniteElemSpace *rowFeSpace,
			   const FiniteElemSpace *colFeSpace,
			   bool apply)
    : BoundaryCondition(type, rowFeSpace, colFeSpace), 
      f(fct), 
      dofVec(NULL),
      applyBC(apply)
  {}


  DirichletBC::DirichletBC(BoundaryType type,
			   DOFVectorBase<double> *vec,
			   bool apply)
    : BoundaryCondition(type, vec->getFeSpace(), vec->getFeSpace()), 
      f(NULL), 
      dofVec(vec),
      applyBC(apply)
  {}


  void DirichletBC::fillBoundaryCondition(DOFMatrix* matrix,
					  ElInfo* elInfo,
					  const DegreeOfFreedom* dofIndices,
					  const BoundaryType* localBound,
					  int nBasFcts)
  {
    FUNCNAME("DirichletBC::fillBoundaryCondition()");
    TEST_EXIT_DBG(matrix->getRowFeSpace() == rowFeSpace)("invalid row fe space\n");
  }


  void DirichletBC::fillBoundaryCondition(DOFVectorBase<double>* vector,
					  ElInfo* elInfo,
					  const DegreeOfFreedom* dofIndices,
					  const BoundaryType* localBound,
					  int nBasFcts)
  {
    FUNCNAME("DirichletBC::fillBoundaryCondition()");

    TEST_EXIT_DBG(vector->getFeSpace() == rowFeSpace)("invalid row fe space\n");

    const BasisFunction *basFcts = rowFeSpace->getBasisFcts();

    for (int i = 0; i < nBasFcts; i++) {
      if (localBound[i] == boundaryType) {
	double value = 0.0;
	if (f) {
	  elInfo->coordToWorld(*(basFcts->getCoords(i)), worldCoords);
	  value = (*f)(worldCoords);
	} else {
	  if (dofVec)
	    value = (*dofVec)[dofIndices[i]];
	  else
	    ERROR_EXIT("There is something wrong!\n");
	}
	vector->setDirichletDofValue(dofIndices[i], value);
	(*vector)[dofIndices[i]] = value;
      }
    }
  }

}