pzmat2dlin.h

Go to the documentation of this file.

#ifndef MAT2DLINHPP
#define MAT2DLINHPP

#include <iostream>
#include "TPZMaterial.h"
#include "pzfmatrix.h"

struct TPZElementMatrix;
class TPZBndCond;
template<class T>
class TPZVec;

class TPZMat2dLin : public TPZMaterial {
                
                TPZFMatrix<STATE>    fKxx, fKxy, fKyx, fKyy, fKx0, fK0x, fKy0, fK0y, fK00, fXf;
                public :
                
    TPZMat2dLin(int num = 1) : TPZRegisterClassId(&TPZMat2dLin::ClassId),
    TPZMaterial(num), fKxx(), fKxy(),
                fKyx() , fKyy(), fKx0(), fK0x(), fKy0(), fK0y(), fK00(), fXf() {
    }
                
                TPZMat2dLin(TPZMat2dLin &copy) : TPZRegisterClassId(&TPZMat2dLin::ClassId),
    TPZMaterial(copy),
                fKxx(copy.fKxx), fKxy(copy.fKxy), fKyx(copy.fKyx), fKyy(copy.fKyy),
                fKx0(copy.fKx0), fK0x(copy.fK0x), fKy0(copy.fKy0),
                fK0y(copy.fK0y), fK00(copy.fK00), fXf(copy.fXf){}
                
                virtual int NStateVariables() const override { return fKxx.Rows(); }
                
                virtual int Dimension() const  override { return 2; }
                
                void Print(std::ostream & out = std::cout) override;
                
                void SetMaterial(TPZFMatrix<STATE> &xkin,TPZFMatrix<STATE> &xcin,TPZFMatrix<STATE> &xfin){
                                int r = xkin.Rows();
                                fKxx = xkin;
                                fKyy = xkin;
                                fK00 = xcin;
                                fXf = xfin;
                                fKxy.Redim(r,r);
                                fKyx.Redim(r,r);
                                fKx0.Redim(r,r);
                                fK0x.Redim(r,r);
                                fKy0.Redim(r,r);
                                fK0y.Redim(r,r);
                }
                
                void ConvectionDiffusion(REAL angle,REAL diff);
                
                TPZFMatrix<STATE> &Xk() {return fKxx;}
                TPZFMatrix<STATE> &Ck() {return fK00;}
                TPZFMatrix<STATE> &Xf() {return fXf;}
                
                virtual std::string Name()  override { return "TPZMat2dLin"; }
                
                virtual void Contribute(TPZMaterialData &data,REAL weight,
                                                                                                                TPZFMatrix<STATE> &ek,TPZFMatrix<STATE> &ef) override;
                
                virtual void Contribute(TPZMaterialData &data,REAL weight,
                                                                                                                TPZFMatrix<STATE> &ef) override
                {
                                TPZMaterial::Contribute(data,weight,ef);
                }
                
                virtual void ContributeBC(TPZMaterialData &data, REAL weight,
                                                                                                                  TPZFMatrix<STATE> &ek,TPZFMatrix<STATE> &ef,TPZBndCond &bc) override;
                
                virtual void ContributeBC(TPZMaterialData &data, REAL weight,
                                                                                                                  TPZFMatrix<STATE> &ef,TPZBndCond &bc) override
                {
                                TPZMaterial::ContributeBC(data,weight,ef,bc);
                }
                
                virtual int NFluxes() override;
                
                virtual void Flux(TPZVec<REAL> &x, TPZVec<STATE> &u, TPZFMatrix<STATE> &dudx, TPZFMatrix<REAL> &axes, TPZVec<STATE> &fl) override;
                
                void Errors(TPZVec<REAL> &x,TPZVec<STATE> &u,TPZFMatrix<STATE> &dudx,TPZFMatrix<REAL> &axes,TPZVec<STATE> &flux,
                                                                TPZVec<STATE> &u_exact,TPZFMatrix<STATE> &du_exact,TPZVec<REAL> &values) override;
                
                virtual int VariableIndex(const std::string &name) override;
                
                virtual int NSolutionVariables(int index) override;
                
protected:
                void Solution(TPZVec<STATE> &Sol,TPZFMatrix<STATE> &DSol,TPZFMatrix<REAL> &axes, int var,TPZVec<STATE> &Solout) override;
public:
                virtual void Solution(TPZMaterialData &data, int var, TPZVec<STATE> &Solout) override
                {
                                TPZMaterial::Solution(data,var,Solout);
                }
                virtual TPZMaterial * NewMaterial() override;
                
                TPZBndCond *OutflowFlux(TPZMaterial * &reference, int bc);
                
                public:
virtual int ClassId() const override;

    
                virtual void Write(TPZStream &buf, int withclassid) const override;
                
                virtual void Read(TPZStream &buf, void *context) override;
                
};

#endif