fei_DirichletBCManager.hpp

#ifndef _fei_DirichletBCManager_hpp_
#define _fei_DirichletBCManager_hpp_
 
/*--------------------------------------------------------------------*/
/*    Copyright 2005 Sandia Corporation.                              */
/*    Under the terms of Contract DE-AC04-94AL85000, there is a       */
/*    non-exclusive license for use of this work by or on behalf      */
/*    of the U.S. Government.  Export of this program may require     */
/*    a license from the United States Government.                    */
/*--------------------------------------------------------------------*/
 
#include <fei_DirichletBCRecord.hpp>
#include <SNL_FEI_Structure.hpp>
#include <fei_VectorSpace.hpp>
 
#include <fei_Pool_alloc.hpp>
#include <map>
 
class NodeDatabase;
class EqnBuffer;
 
namespace fei {
class Matrix;
 
class DirichletBCManager {
 public:
  DirichletBCManager(SNL_FEI_Structure* structure)
   : structure_(structure), vecSpace_() {}
 
  DirichletBCManager(fei::SharedPtr<fei::VectorSpace> vecspace)
   : structure_(NULL), vecSpace_(vecspace) {}
 
  ~DirichletBCManager(){}
 
  void addBCRecords(int numBCs,
                    int IDType,
                    int fieldID,
                    int offsetIntoField,
                    const int* IDs,
                    const double* prescribedValues);
 
  void addBCRecords(int numBCs,
                    int IDType,
                    int fieldID,
                    const int* IDs,
                    const int* offsetsIntoField,
                    const double* prescribedValues);
 
  int finalizeBCEqns(fei::Matrix& matrix,
                     bool throw_if_bc_slave_conflict=false);
 
  int finalizeBCEqns(EqnBuffer& bcEqns);
 
  size_t getNumBCRecords() const;
 
  void clearAllBCs();
 
 private:
  int getEqnNumber(int IDType, int ID, int fieldID, int offsetIntoField);
 
  SNL_FEI_Structure* structure_;
  fei::SharedPtr<fei::VectorSpace> vecSpace_;
 
  typedef std::map<int,double,std::less<int>,
            fei_Pool_alloc<std::pair<const int, double> > > bc_map;
  bc_map bcs_;
};//class DirichletBCManager
}//namespace fei
#endif