TPZMatElasticity2D.h

Go to the documentation of this file.

//
//  TPZMatElasticity2D.h
//  PZ
//
//  Created by Omar on 10/27/14.
//
//

#ifndef __PZ__TPZMatElasticity2D__
#define __PZ__TPZMatElasticity2D__

#include <stdio.h>
#include "TPZMaterial.h"
#include "pzvec.h"
#include <iostream>


class TPZMatElasticity2D : public TPZMaterial {
    
protected:
    
    TPZManVector<STATE,2>  m_f;
    
    REAL m_E;
    
    REAL m_nu;
    
    REAL m_lambda;
    
    REAL m_mu;
    
    REAL m_s0_xx;
    REAL m_s0_xy;
    REAL m_s0_yy;
    REAL m_s0_zz;
    
    int m_plane_stress;
    
    
public:
    
virtual int ClassId() const override;

    TPZMatElasticity2D();
    
    TPZMatElasticity2D(int matid, REAL E, REAL nu, REAL fx, REAL fy, int plainstress = 1);
    
    TPZMatElasticity2D(int matid);
    
    TPZMatElasticity2D &operator=(const TPZMatElasticity2D &copy);
    
    virtual ~TPZMatElasticity2D();
    
    TPZMatElasticity2D(const TPZMatElasticity2D &cp);    
    
    virtual TPZMaterial *NewMaterial() override
    {
        return new TPZMatElasticity2D(*this);
    }
    
    virtual void Print(std::ostream & out) override;
    
    virtual std::string Name()  override { return "TPZMatElasticity2D"; }
    
    int Dimension() const  override {return 2;}
    
    virtual int NStateVariables() const override;

    void SetParameters(REAL Lambda, REAL mu, REAL fx, REAL fy)
    {
        m_E = (mu*(3.0*Lambda+2.0*mu))/(Lambda+mu);
        m_nu = (Lambda)/(2*(Lambda+mu));
        
        m_lambda = Lambda;
        m_mu = mu;
        m_f[0] = fx;
        m_f[1] = fy;
    }
    
    void SetElasticParameters(REAL Eyoung, REAL nu)
    {
        this->SetElasticity(Eyoung,nu);
    }
    
    void SetElasticity(REAL Ey, REAL nu)
    {
        m_E = Ey;
        m_nu = nu;
        m_lambda = (Ey*nu)/((1+nu)*(1-2*nu));
        m_mu = Ey/(2*(1+nu));

    }    
    
    void SetPlaneStress()
    {
        m_plane_stress = 1;
    }
    
    void SetPlaneStrain()
    {
        m_plane_stress = 0;
    }    
    
    void SetPreStress(REAL SigmaXX, REAL SigmaXY, REAL SigmaYY, REAL SigmaZZ)
    {
        m_s0_xx = SigmaXX;
        m_s0_xy = SigmaXY;
        m_s0_yy = SigmaYY;
        m_s0_zz = SigmaZZ;
    }

    void ComputeSigma(const TPZFMatrix<STATE> &dudx, TPZFMatrix<STATE> &sigma);
    
    // Get Elastic Materials Parameters
    void GetElasticParameters(REAL &Ey, REAL &nu, REAL &Lambda, REAL &G)
    {
        Ey = m_E;
        nu =  m_nu;
        Lambda =  m_lambda;
        G = m_mu;
    }
    
    STATE GetEyoung() {return m_E;}
    
    STATE GetNu() {return m_nu;}
    
    STATE GetLambda() {return m_lambda;}
    
    STATE GetMu() {return m_mu;}

    
    virtual void FillDataRequirements(TPZMaterialData &data) override;
    
    virtual void FillBoundaryConditionDataRequirement(int type, TPZMaterialData &data) 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;
    void ContributeVec(TPZMaterialData &data, REAL weight, TPZFMatrix<STATE> &ek, TPZFMatrix<STATE> &ef);
    void ContributeVec(TPZMaterialData &data, REAL weight, TPZFMatrix<STATE> &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;
    
    virtual int VariableIndex(const std::string &name) override;
    
    virtual int NSolutionVariables(int var) override;
    
    //public:
    virtual void Solution(TPZMaterialData &data, int var, TPZVec<STATE> &Solout) override;
    virtual void Solution(TPZMaterialData &data, TPZVec<TPZMaterialData> &dataleftvec, TPZVec<TPZMaterialData> &datarightvec, int var, TPZVec<STATE> &Solout, TPZCompEl * Left, TPZCompEl * Right) override {
        DebugStop();
    }
    
    virtual void Errors(TPZVec<REAL> &x, TPZVec<STATE> &sol, TPZFMatrix<STATE> &dsol,
                        TPZFMatrix<REAL> &axes, TPZVec<STATE> &flux,
                        TPZVec<STATE> &uexact, TPZFMatrix<STATE> &duexact,
                        TPZVec<REAL> &val) override;

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

};

#endif /* defined(__PZ__TPZMatElasticity2D__) */