TPZLinearConvection.h

Go to the documentation of this file.

#ifndef TPZLINEARCONVECTION_H
#define TPZLINEARCONVECTION_H

#include "TPZMaterial.h"
#include "pzfmatrix.h"

class TPZBndCond;
class TPZLinearConvection : public TPZMaterial {
public:
    
    virtual int ClassId() const override;

    TPZLinearConvection(TPZLinearConvection & copy);
    TPZLinearConvection(int id,TPZVec<STATE> &conv) ;

    virtual int Dimension() const override;
                
    virtual int NStateVariables() const override ;
                
    virtual int NFluxes()  override {return 2;}

    virtual void Contribute(TPZMaterialData &data, REAL weight,
                                                                                                                TPZFMatrix<STATE> &ek,TPZFMatrix<STATE> &ef) override;
                
                
    virtual void ContributeBC(TPZMaterialData &data,REAL weight,
                                                                                                                  TPZFMatrix<STATE> &ek,TPZFMatrix<STATE> &ef,TPZBndCond &bc) 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> &ef,TPZBndCond &bc) override
                {
                                TPZMaterial::ContributeBC(data,weight,ef,bc);
    }

    virtual void Print(std::ostream &out = std::cout) override;
                
    virtual int VariableIndex(const std::string &name) override;
                
    virtual int NSolutionVariables(int var) override;
                
protected:
                virtual 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
                {
        int numbersol = data.sol.size();
        if (numbersol != 1) {
            DebugStop();
        }

        Solution(data.sol[0],data.dsol[0],data.axes,var,Solout);
    }
                
    virtual void Flux(TPZVec<REAL> &x, TPZVec<STATE> &Sol, TPZFMatrix<STATE> &DSol, TPZFMatrix<REAL> &axes, TPZVec<STATE> &flux) override {}
                
    virtual TPZMaterial * NewMaterial() override;
                
    virtual void SetData(std::istream &data) override;
                
private:    
    STATE fConvect[2];

};

#endif //TPZLINEARCONVECTION_H