feMat.h

Go to the documentation of this file.

#ifndef __FE_MAT_H_
#define __FE_MAT_H_

#include "petscda.h"
#include "oda.h"

namespace ot {
  namespace fem {

    class feMat {
      public:
        enum daType {
          PETSC, OCT
        }; 

        feMat() { };
        feMat(daType da) {
#ifdef __DEBUG__
          assert ( ( da == PETSC ) || ( da == OCT ) );
#endif
          m_daType = da;


        }
        ~feMat() {

        }

        void setDA (_p_DA* da) { m_DA = da; }
        void setDA (ot::DA* da) { m_octDA = da; }

        _p_DA* getDA() { return m_DA; }
        ot::DA* getOctDA() { return m_octDA; }
        virtual bool MatVec(Vec _in, Vec _out, double scale=1.0) = 0;
        virtual bool MatGetDiagonal(Vec _diag, double scale=1.0) = 0;


        void setProblemDimensions(double x, double y, double z) {
          m_dLx = x;
          m_dLy = y;
          m_dLz = z;
        }
      protected:

        daType          m_daType;

        _p_DA*              m_DA;
        ot::DA*         m_octDA;
        double m_dLx;
        double m_dLy;
        double m_dLz;
    };

  } // end namespace fem
} // end namespace ot

#endif

---