pzgeotetrahedra.h

Go to the documentation of this file.

#ifndef TPZGEOTETRAHEDRAH
#define TPZGEOTETRAHEDRAH

#include "pznoderep.h"
#include "pzvec.h"
#include "pzeltype.h"
#include "tpztetrahedron.h"
#include "pzfmatrix.h"

#include <string>

class TPZGeoEl;
class TPZGeoMesh;

namespace pzgeom {
                
                class TPZGeoTetrahedra  : public TPZNodeRep<4,pztopology::TPZTetrahedron>
                {
                public:
        typedef pztopology::TPZTetrahedron Top;
                                enum {NNodes = 4};

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

                
                                TPZGeoTetrahedra(TPZVec<int64_t> &nodeindexes) : TPZRegisterClassId(&TPZGeoTetrahedra::ClassId),
        TPZNodeRep<NNodes,pztopology::TPZTetrahedron>(nodeindexes)
                                {
                                }
                                
                                TPZGeoTetrahedra() : TPZRegisterClassId(&TPZGeoTetrahedra::ClassId),
        TPZNodeRep<NNodes,pztopology::TPZTetrahedron>()
                                {
                                }
                                
                                TPZGeoTetrahedra(const TPZGeoTetrahedra &cp,
                                                                                                 std::map<int64_t,int64_t> & gl2lcNdMap) : TPZRegisterClassId(&TPZGeoTetrahedra::ClassId),
        TPZNodeRep<NNodes,pztopology::TPZTetrahedron>(cp,gl2lcNdMap)
                                {
                                }
                                
                                TPZGeoTetrahedra(const TPZGeoTetrahedra &cp) : TPZRegisterClassId(&TPZGeoTetrahedra::ClassId),
        TPZNodeRep<NNodes,pztopology::TPZTetrahedron>(cp)
                                {
                                }
                                
                                TPZGeoTetrahedra(const TPZGeoTetrahedra &cp, TPZGeoMesh &) : TPZRegisterClassId(&TPZGeoTetrahedra::ClassId),
        TPZNodeRep<NNodes,pztopology::TPZTetrahedron>(cp)
                                {
                                }
                                
        static bool IsLinearMapping(int side)
        {
            return true;
        }

                                static std::string TypeName() { return "Tetrahedron";}
        
        template<class T>
        static void X(const TPZFMatrix<REAL> &nodes,TPZVec<T> &loc,TPZVec<T> &x);
        
        template<class T>
        static void GradX(const TPZFMatrix<REAL> &nodes,TPZVec<T> &loc, TPZFMatrix<T> &gradx);

                public:
        
        /* @param gmesh mesh in which the element should be inserted
         @param matid material id of the element
         @param lowercorner (in/out) on input lower corner o the cube where the element should be created, on exit position of the next cube
         @param size (in) size of space where the element should be created
         */
        static void InsertExampleElement(TPZGeoMesh &gmesh, int matid, TPZVec<REAL> &lowercorner, TPZVec<REAL> &size);

                                // static TPZGeoEl *CreateGeoElement(TPZGeoMesh &mesh, MElementType type,
                                //                                                                                                                                TPZVec<int64_t>& nodeindexes,
                                //                                                                                                                                int matid,
                                //                                                                                                                                int64_t& index);
                };
    

    
    template<class T>
    inline void TPZGeoTetrahedra::X(const TPZFMatrix<REAL> &nodes,TPZVec<T> &loc,TPZVec<T> &x){
        
        TPZFNMatrix<4,T> phi(NNodes,1);
        TPZFNMatrix<12,T> dphi(3,NNodes);
        TShape(loc,phi,dphi);
        int space = nodes.Rows();
        
        for(int i = 0; i < space; i++) {
            x[i] = 0.0;
            for(int j = 0; j < NNodes; j++) {
                x[i] += phi(j,0)*nodes.GetVal(i,j);
            }
        }
    }
    
    
    template<class T>
    inline void TPZGeoTetrahedra::GradX(const TPZFMatrix<REAL> &nodes,TPZVec<T> &loc, TPZFMatrix<T> &gradx){
        
        gradx.Resize(3,3);
        gradx.Zero();
        int nrow = nodes.Rows();
        int ncol = nodes.Cols();
#ifdef PZDEBUG
        if(nrow != 3 || ncol  != 4){
            std::cout << "Objects of incompatible lengths, gradient cannot be computed." << std::endl;
            std::cout << "nodes matrix must be 3x4." << std::endl;
            DebugStop();
        }
        
#endif
        TPZFNMatrix<4,T> phi(NNodes,1);
        TPZFNMatrix<12,T> dphi(3,NNodes);
        TShape(loc,phi,dphi);
        for(int i = 0; i < 4; i++)
        {
            for(int j = 0; j < 3; j++)
            {
                gradx(j,0) += nodes.GetVal(j,i)*dphi(0,i);
                gradx(j,1) += nodes.GetVal(j,i)*dphi(1,i);
                gradx(j,2) += nodes.GetVal(j,i)*dphi(2,i);
            }
        }
        
    }
                
};

#endif