pzpostprocmat.h

Go to the documentation of this file.

#ifndef PZPOSTPROCMAT_H
#define PZPOSTPROCMAT_H

#include "TPZMaterial.h"
#include "pzdiscgal.h"
#include "tpzautopointer.h"
#include "pzvec.h"

#include <string>

class TPZPostProcVar : public TPZSavable
{
public:
                                
                  TPZPostProcVar():fIndex(-1), fName(""), fNumEq(-1) {}

                  TPZPostProcVar(const TPZPostProcVar & source)
                  {
                      this->operator=(source);
                  }
                
                  TPZPostProcVar& operator=(const TPZPostProcVar & source)
                  {
                                  fIndex = source.fIndex;
                                  fName = source.fName;
                                  fNumEq  = source.fNumEq;
                                  return *this;
                  }
                                
                  ~TPZPostProcVar(){}           int ClassId() const override;
        void Read(TPZStream &buf, void *context) override;
        void Write(TPZStream &buf, int withclassid) const override;

public: //members
                
                  int64_t fIndex;
                
                  std::string fName;
                
                  int64_t fNumEq;
};

class  TPZPostProcMat : public TPZDiscontinuousGalerkin
{
   public:
                                
      TPZPostProcMat();                         
                                
      TPZPostProcMat(int64_t id);

      TPZPostProcMat(const TPZPostProcMat &mat);

      virtual ~TPZPostProcMat();
                
      virtual std::string Name() override;

      virtual int Dimension() const override;

      virtual int NStateVariables() const override;

      virtual void Print(std::ostream &out = std::cout) override;

      virtual int VariableIndex(const std::string &name) override;

      virtual int NSolutionVariables(int var) override;

      virtual void Solution(TPZMaterialData &data, int var, TPZVec<STATE> &Solout) override;
                
      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;


      virtual void ContributeBC(TPZMaterialData &data, REAL weight, TPZFMatrix<STATE> &ek, TPZFMatrix<STATE> &ef, TPZBndCond &bc) override;
                
      virtual void ContributeInterface(TPZMaterialData &data, TPZMaterialData &dataleft, TPZMaterialData &dataright, 
                                       REAL weight, TPZFMatrix<STATE> &ek, TPZFMatrix<STATE> &ef) override;

      virtual void ContributeBCInterface(TPZMaterialData &data, TPZMaterialData &dataleft, 
                                         REAL weight, TPZFMatrix<STATE> &ek,TPZFMatrix<STATE> &ef,TPZBndCond &bc) override;

      public:
        virtual int ClassId() const override;


      virtual void Write(TPZStream &buf, int withclassid) const override;

      virtual void Read(TPZStream &buf, void *context) override;
                
                  virtual void FillDataRequirements(TPZMaterialData &data) override;
                
                  void GetPostProcessVarIndexList(TPZVec<int> & varIndexList);
    
    void GetPostProcVarName(int64_t index, std::string &varname)
    {
        varname = fVars[index].fName;
        
    }

                  void SetPostProcessVarIndexList(TPZVec<std::string> & varIndexNames, TPZMaterial * pRefMat);
                
protected:
                                
                                TPZManVector<TPZPostProcVar,20> fVars;
                                int fDimension;
};

#endif