/Users/jack/Code/basso_dev/inc/Tria3Basis.h

Go to the documentation of this file.

#ifndef _TRIA3_BASIS_BASSO_H_
#define _TRIA3_BASIS_BASSO_H_

#include "basso.h"
#include "Basis.h"

namespace basso {

class Tria3Basis : public Basis {

public:
        
        // constructors
        Tria3Basis() { }
        
        // destructors
        virtual ~Tria3Basis() { }
        
        // member functions
        virtual BasisType       Type() const { return kTRIA3; }
        virtual BasisShape Shape() const { return kTRIANGLE; }
        virtual int                             NumNodes() const { return 3; }     
        virtual int                             NumEdges() const { return 3; }      
        virtual int                             NumFaces() const { return 0; }      
        virtual int                             Dimension() const { return 2; }
        virtual int                             Order() const { return 1; }
        
        virtual void            ParentCoord( Array< Point > &xi )       const;
        virtual Point   Centroid() const { return Point( 0.33333333333333333, 0.33333333333333333 ); }
        virtual void    NumNodesOnFaces( iArray &nn_face )      const;
        virtual void            NumNodesOnEdges( iArray &nn_edge )      const;
        virtual void    NodesOnFace ( int f, iArray &face_nodeids )             const;
        virtual void            NodesOnEdge ( int e, iArray &edge_nodeids )                     const;
        virtual void    FaceBasisType( Array<BasisType> &face_basis )   const;
        virtual void            EdgeBasisType( Array< BasisType > &edge_basis ) const;
        
        virtual void            Quadrature( int order, QuadratureRule &qrule ) const;
        
        virtual void            Na( const Point &p, nArray &shapefunct )  const { shape_tria3(shapefunct,p.x(),p.y()); }
        virtual void            DNa( const Point &p, nMatrix &grad_shape )      const { dshape_tria3(grad_shape,p.x(),p.y()); }
        virtual void    Na( nArray &shapefunct )    const { Na( Centroid(), shapefunct ); }
        virtual void            DNa( nMatrix &grad_shape ) const { DNa( Centroid(), grad_shape ); }

private:
        
};

void Tria3Basis::ParentCoord( Array< Point > &xi ) const
{
#ifdef ALLOW_DYNAMIC_RESIZE 
        if ( xi.size()!=NumNodes() )
                xi.resize(NumNodes());
#endif
        xi[0]=Point(0.0,0.0);
        xi[1]=Point(1.0,0.0);
        xi[2]=Point(0.0,1.0);
}

void    Tria3Basis::NumNodesOnFaces( iArray &nn_face )  const
{
#ifdef ALLOW_DYNAMIC_RESIZE 
        if ( nn_face.size()!=NumEdges() )
                nn_face.resize( NumEdges() );
#endif
}

void    Tria3Basis::NumNodesOnEdges( iArray &nn_edge )  const
{
#ifdef ALLOW_DYNAMIC_RESIZE 
        if ( nn_edge.size()!=NumEdges() )
                nn_edge.resize( NumEdges() );
#endif
        for ( int i=0; i<NumEdges(); ++i )
                nn_edge[i]=2; 
}

void Tria3Basis::NodesOnFace ( int e, iArray &face_nodeids ) const
{
#ifdef ALLOW_DYNAMIC_RESIZE 
        if ( face_nodeids.size()!=0 )
                face_nodeids.resize(0);
#endif
}

void Tria3Basis::NodesOnEdge ( int e, iArray &edge_nodeids ) const
{
#ifdef ALLOW_DYNAMIC_RESIZE 
        if ( edge_nodeids.size()!=2 )
                edge_nodeids.resize(2);
#endif
        
        switch (e) {
        
                case 0:
                edge_nodeids[0]=1; edge_nodeids[1]=2;
                break;
                
                case 1:
                edge_nodeids[0]=2; edge_nodeids[1]=0;
                break;
                
                case 2:
                edge_nodeids[0]=0; edge_nodeids[1]=1;
                break;
                
                default:
                WarningMessage("Tria3Basis::NodesOnFace","face id out of range");
                break;
                
        }
}

void Tria3Basis::FaceBasisType( Array< BasisType > &face_basis ) const
{
#ifdef ALLOW_DYNAMIC_RESIZE 
        if ( face_basis.size()!=NumFaces() )
                face_basis.resize(NumFaces());
#endif
}

void Tria3Basis::EdgeBasisType( Array< BasisType > &edge_basis ) const
{
#ifdef ALLOW_DYNAMIC_RESIZE 
        if ( edge_basis.size()!=NumEdges() )
                edge_basis.resize(NumEdges());
#endif
        for ( int i=0; i<NumEdges(); ++i )
                edge_basis[i]=kLINE2; 
}

void Tria3Basis::Quadrature( int order, QuadratureRule &qrule ) const
{
        
        int npts;
        switch (order)
        {
                case 0:
                npts=1;
                break;
                
                case 1:
                npts=1;
                break;
                
                case 2:
                npts=3;
                break;  
                
                case 3:
                npts=4;
                break;
                
                case 4:
                npts=6;
                break;
                
                case 5:
                npts=7;
                break;
                
                case 6:
                npts=12;
                break;
                
                case 7:
                npts=13;
                break;  
                
                default:
                WarningMessage("Tria3Basis::Quadrature( int order )","quadrature order >7");
                npts=13;
                break;
                
        }

        if ( vect_size(qrule)!=npts )
                resize(qrule,npts);

        quadrature_tria( qrule );
        
}
/*
 void Tria3Basis::Na( const Point &p, nArray &shapefunct ) const
{
#ifdef ALLOW_DYNAMIC_RESIZE 
        if ( shapefunct.size()!=NumNodes() )
                shapefunct.resize(NumNodes());  
#endif
        shapefunct[0]=1-p.x()-p.y();
        shapefunct[1]=p.x();
        shapefunct[2]=p.y();
}

void Tria3Basis::DNa( const Point &p, nMatrix &grad_shape ) const
{
#ifdef ALLOW_DYNAMIC_RESIZE 
        if ( mat_nrows(grad_shape)!=NumNodes() || mat_ncols(grad_shape)!=Dimension() )
                resize( grad_shape, NumNodes(), Dimension() );  
#endif
        grad_shape(0,0)=-1.0; grad_shape(0,1)=-1.0; 
        grad_shape(1,0)= 1.0; grad_shape(1,1)= 0.0;
        grad_shape(2,0)= 0.0; grad_shape(2,1)= 1.0;
}
*/
} // end namespace

#endif

---