---

FEDiscretizer.h

Go to the documentation of this file.

#ifndef FEDISCRETIZER_H
#define FEDISCRETIZER_H

#include "SundanceDefs.h"

#include "WeakForm.h"
#include "TSFRBTree.h"
#include "DistributedDOFMap.h"
#include "TSFSmartPtr.h"
#include "Element.h"
#include "CellType.h"
#include "Mesh.h"
#include "OldIntegral.h"
#include "EssentialBC.h"
#include "EquationSet.h"
#include "TSFVector.h"
#include "TSFMatrixOperator.h"
#include "TSFHashtable.h"


namespace Sundance
{

  using namespace TSF;
  using std::string;

  using std::ostream;

  /** \ingroup LowLevelFE
   * Distributed assembly of a matrix
   */

  class FEDiscretizer
    {
    public:
      /** */
      FEDiscretizer(const Mesh& mesh, const OldIntegral& eqn,
                    const Expr& test, const Expr& shape);
      /** */
      FEDiscretizer(const Mesh& mesh, const OldIntegral& eqn,
                    const EssentialBC& bc,
                    const Expr& test, const Expr& shape);

      /** */
      void buildMatrix(MatrixBase& mat) const ;
      /** */
      void buildVector(DistributedVector& vec) const ;

      /** */
      const TSFSmartPtr<DOFMapBase>& shapeDOFMap() const {return shapeDOFMap_;}

      /** */
      bool lookupReducedShapeID(int shapeID, int& reducedShapeID) const ;

      /** */
      void buildSoln(const TSFVector& solnVec,
                     Expr& soln) const ;

      /** */
      void print(ostream& os) const ;

      /** */
      static void silent() {monitorLocalMatrices_ = false;}
      /** */
      static void verbose() {monitorLocalMatrices_ = true;}

    private:

      void getGraph(TSFArray<TSFNonDupArray<int> >& columnIndices) const ;

      bool isUpdateRow(int globalRowIndex) const ;
      bool isBCRow(int globalRowIndex) const ;
      void setBCRow(int globalRowIndex) ;

      int getLocalRowIndex(int globalRowIndex) const ;

      void init(const Mesh& mesh, const EquationSet& eqn,
                const Expr& test, const Expr& shape);

      void insertLocalMatrix(int cellDim,
                             int cellIndex,
                             const WeakForm& wf,
                             const LocalMatrix& localMat,
                             MatrixBase& mat) const ;
      void insertLocalVector(int cellDim,
                             int cellIndex,
                             const WeakForm& wf,
                             const DenseSerialVector& localVec,
                             DistributedVector& vec) const ;

      TSFArray<CellSet> cellSets_;
      TSFArray<TSFArray<WeakForm> > wf_;
      TSFRBTree<int, int> testIDMap_;
      TSFRBTree<int, int> shapeIDMap_;
      TSFSmartPtr<DOFMapBase> testDOFMap_;
      TSFSmartPtr<DOFMapBase> shapeDOFMap_;
      TSFSmartPtr<TSFArray<int> > updateList_;
      TSFSmartPtr<TSFArray<int> > remoteDOFList_;
      int lowestLocalDOF_;
      int highestLocalDOF_;

      // the BCRowTables indicate whether a given row is to be replaced
      // by an essential boundary condition.
      TSFArray<bool> localBCRowTable_;
      TSFHashtable<int, Bool> remoteBCRowTable_;

      TSFArray<TSFArray<TSFRBTree<CellTopologyCode, Element> > > referenceElements_;
      TSFArray<TSFArray<Cell> > cells_;
      TSFArray<TSFNonDupArray<CellTopologyCode> > cellTypes_;
      Expr var_;
      Expr unk_;

      // verbosity flags
      static bool monitorLocalMatrices_;

    };

}
#endif

---

Contact:

Kevin Long (krlong@ca.sandia.gov)


Documentation generated by