5node_pyramid.c

/*****************************************************************************
 *
 *  Elmer, A Finite Element Software for Multiphysical Problems
 *
 *  Copyright 1st April 1995 - , CSC - IT Center for Science Ltd., Finland
 * 
 *  This program is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU General Public License
 *  as published by the Free Software Foundation; either version 2
 *  of the License, or (at your option) any later version.
 * 
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program (in file fem/GPL-2); if not, write to the 
 *  Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, 
 *  Boston, MA 02110-1301, USA.
 *
 *****************************************************************************/

/*******************************************************************************
 *
 * Definition of 6 node pyramid element.
 *
 *******************************************************************************
 *
 *                     Author:       Juha Ruokolainen
 *
 *                    Address: CSC - IT Center for Science Ltd.
 *                                Keilaranta 14, P.O. BOX 405
 *                                  02101 Espoo, Finland
 *                                  Tel. [phone]
 *                                Telefax: [phone]
 *                              EMail: [email protected]
 *
 *                       Date: 4 Oct 1995
 *
 ******************************************************************************/

#include "../elmerpost.h"
#include <elements.h>

/*
 * Five node pyramid volume element.
 *
 */

static double A[5][5],N[5][5];

static double NodeU[] = { -1.0, 1.0, 1.0, -1.0, 0.0 };
static double NodeV[] = { -1.0,-1.0, 1.0,  1.0, 0.0 };
static double NodeW[] = { 0.0, 0.0, 0.0, 0.0, 1.0};

/*******************************************************************************
 *
 *     Name:        elm_5node_pyramid_triangulate
 *
 *     Purpose:     Triangulate an element. The process also builds up an edge
 *                  table and adds new nodes to node table. The triangulation
 *                  and edge table is stored in geometry_t *geom-structure.
 *
 *     Parameters:
 *
 *         Input:   (geometry_t *) pointer to structure holding triangulation
 *                  (element_t  *) element to triangulate
 *
 *         Output:  (geometry_t *) structure is modified
 *
 *   Return value:  FALSE if malloc() fails, TRUE otherwise
 *
 ******************************************************************************/
static int elm_5node_pyramid_triangulate( geometry_t *geom,element_t *pyramid )
{
    element_t element;
    int i,j;

    int PyramidFace[5][4] = { { 0,1,2,3 }, { 0,1,4,0 }, { 1,2,4,0 }, { 2,3,4,0 }, { 3,0,4,0 } };
    
    int elm_4node_quad_triangulate();
    int geo_add_edge();
    int elm_3node_triangle_triangulate();

    if ( GlobalOptions.VolumeSides )
    {
       int topo[4];

       element.DisplayFlag = TRUE;
       element.Topology = topo;
       for( i=0; i<MAX_GROUP_IDS; i++ ) element.GroupIds[i] = pyramid->GroupIds[i];
   
       for( j=0; j<4; j++ )
       {
            element.Topology[j] = pyramid->Topology[PyramidFace[0][j]];
        }
        if ( !elm_4node_quad_triangulate( geom, &element, pyramid ) ) return FALSE;

        for( i=1; i<5; i++ )
        {
           for( j=0; j<3; j++ )
           {
               element.Topology[j] = pyramid->Topology[PyramidFace[i][j]];
           }
        }
           if ( !elm_3node_triangle_triangulate( geom, &element, pyramid ) ) return FALSE;
    } else {
       if ( !geo_add_edge( geom, pyramid->Topology[0], pyramid->Topology[1],pyramid ) ) return FALSE;
       if ( !geo_add_edge( geom, pyramid->Topology[1], pyramid->Topology[2],pyramid ) ) return FALSE;
       if ( !geo_add_edge( geom, pyramid->Topology[2], pyramid->Topology[3],pyramid ) ) return FALSE;
       if ( !geo_add_edge( geom, pyramid->Topology[3], pyramid->Topology[0],pyramid ) ) return FALSE;
       if ( !geo_add_edge( geom, pyramid->Topology[0], pyramid->Topology[4],pyramid ) ) return FALSE;
       if ( !geo_add_edge( geom, pyramid->Topology[1], pyramid->Topology[4],pyramid ) ) return FALSE;
       if ( !geo_add_edge( geom, pyramid->Topology[2], pyramid->Topology[4],pyramid ) ) return FALSE;
       if ( !geo_add_edge( geom, pyramid->Topology[3], pyramid->Topology[4],pyramid ) ) return FALSE;
    }

    return TRUE;
}

/*******************************************************************************
 *
 *     Name:        elm_5node_pyramid_shape_functions
 *
 *     Purpose:     Initialize element shape function array. Internal only.
 *
 *     Parameters:
 *
 *         Input:   Global (filewise) variables NodeU,NodeV,NodeW
 *
 *         Output:  Global (filewise) variable N[6][6], will contain
 *                  shape function coefficients
 *
 *   Return value:  void
 *   Return value:  void
 *
 ******************************************************************************/
static void elm_5node_pyramid_shape_functions()
{
     double u,v,w;
     int i,j;

#if 0
     for( i=0; i<5; i++ )
     {
         u = NodeU[i];
         v = NodeV[i];
         w = NodeW[i];

         A[i][0]   = 1;
         A[i][1]   = u;
         A[i][2]   = v;
         A[i][3]   = w;
         A[i][4]   = u*v;
     }

     lu_mtrinv( (double *)A,5 );

     for( i=0; i<5; i++ )
        for( j=0; j<5; j++ ) N[i][j] = A[j][i];
#endif
}

/*******************************************************************************
 *
 *     Name:        elm_5node_pyramid_fvalue
 *
 *     Purpose:     return value of a quantity given on nodes at point (u,v)
 *                  Use trough (element_type_t *) structure.
 *
 *     Parameters:
 *
 *         Input:  (double *) quantity values at nodes 
 *                 (double u,double v) point where values are evaluated
 *
 *         Output:  none
 *
 *   Return value:  quantity value
 *
 ******************************************************************************/
static double elm_5node_pyramid_fvalue(double *F,double u,double v,double w)
{
     double R=0.0, s;
     int i;

     if ( w == 1 ) w = 1-1.0e-12;
     s = 1.0 / (1-w);

     R = R + F[0] * ( (1-u) * (1-v) - w + u*v*w * s ) / 4;
     R = R + F[1] * ( (1+u) * (1-v) - w - u*v*w * s ) / 4;
     R = R + F[2] * ( (1+u) * (1+v) - w + u*v*w * s ) / 4;
     R = R + F[3] * ( (1-u) * (1+v) - w - u*v*w * s ) / 4;
     R = R + F[4] * w;

     return R;
}

/*******************************************************************************
 *
 *     Name:        elm_5node_pyramid_dndu_fvalue
 *
 *     Purpose:     return value of a first partial derivate in (v) of a
 *                  quantity given on nodes at point (u,v).
 *                  Use trough (element_type_t *) structure.
 *                 
 *
 *     Parameters:
 *
 *         Input:  (double *) quantity values at nodes 
 *                 (double u,double v,double w) point where values are evaluated
 *
 *         Output:  none
 *
 *   Return value:  quantity value
 *
 ******************************************************************************/
static double elm_5node_pyramid_dndu_fvalue(double *F,double u,double v,double w)
{
     double R=0.0,s;
     int i;

     if ( w == 1 ) w = 1-1.0e-12;
     s = 1.0 / (1-w);

     R = R + F[0] * ( -(1-v) + v*w * s ) / 4;
     R = R + F[1] * (  (1-v) - v*w * s ) / 4;
     R = R + F[2] * (  (1+v) + v*w * s ) / 4;
     R = R + F[3] * ( -(1+v) - v*w * s ) / 4;

     return R;
}

/*******************************************************************************
 *
 *     Name:        elm_5node_pyramid_dndv_fvalue
 *
 *     Purpose:     return value of a first partial derivate in (v) of a
 *                  quantity given on nodes at point (u,v,w).
 *                  Use trough (element_type_t *) structure.
 *                 
 *
 *     Parameters:
 *
 *         Input:  (double *) quantity values at nodes 
 *                 (double u,double v,double w) point where values are evaluated
 *
 *         Output:  none
 *
 *   Return value:  quantity value
 *
 ******************************************************************************/
static double elm_5node_pyramid_dndv_fvalue(double *F,double u,double v,double w)
{
     double R=0.0,s;
     int i;

     if ( w == 1 ) w = 1-1.0e-12;
     s = 1.0 / (1-w);

     R = R + F[0] * ( -(1-u) + u*w * s ) / 4;
     R = R + F[1] * ( -(1+u) - u*w * s ) / 4;
     R = R + F[2] * (  (1+u) + u*w * s ) / 4;
     R = R + F[3] * (  (1-u) - u*w * s ) / 4;

     return R;
}

/*******************************************************************************
 *
 *     Name:        elm_5node_pyramid_dndw_fvalue
 *
 *     Purpose:     return value of a first partial derivate in (w) of a
 *                  quantity given on nodes at point (u,v,w)
 *                  Use trough (element_type_t *) structure.
 *                 
 *
 *     Parameters:
 *
 *         Input:  (double *) quantity values at nodes 
 *                 (double u,double v,double w) point where values are evaluated
 *
 *         Output:  none
 *
 *   Return value:  quantity value
 *
 ******************************************************************************/
static double elm_5node_pyramid_dndw_fvalue(double *F,double u,double v,double w)
{
     double R=0.0,s;
     int i;

     if ( w == 1 ) w = 1.0-1.0e-12;
     s = 1.0 / (1-w);

     R = R + F[0] * ( -1 + u*v*(2-w) * s*s ) / 4;
     R = R + F[1] * ( -1 - u*v*(2-w) * s*s ) / 4;
     R = R + F[2] * ( -1 + u*v*(2-w) * s*s ) / 4;
     R = R + F[3] * ( -1 - u*v*(2-w) * s*s ) / 4;
     R = R + F[4];

     return R;
}

/******************************************************************************
 *
 *     Name:        elm_5node_pyramid_initialize
 *
 *     Purpose:     Register the element type
 *                  
 *     Parameters:
 *
 *         Input:  (char *) description of the element
 *                 (int)    numeric code for the element
 *
 *         Output:  Global list of element types is modified
 *
 *   Return value:  malloc() success
 *
 ******************************************************************************/
int elm_5node_pyramid_initialize()
{
     static char *Name = "ELM_5NODE_PYRAMID";

     element_type_t ElementDef;
     int elm_add_element_type();

     elm_5node_pyramid_shape_functions();

     ElementDef.ElementName = Name;
     ElementDef.ElementCode = 605;

     ElementDef.NumberOfNodes = 5;

     ElementDef.NodeU = NodeU;
     ElementDef.NodeV = NodeV;
     ElementDef.NodeW = NodeW;

     ElementDef.PartialU = (double (*)())elm_5node_pyramid_dndu_fvalue;
     ElementDef.PartialV = (double (*)())elm_5node_pyramid_dndv_fvalue;
     ElementDef.PartialW = (double (*)())elm_5node_pyramid_dndw_fvalue;

     ElementDef.FunctionValue = (double (*)())elm_5node_pyramid_fvalue;
     ElementDef.Triangulate   = (int (*)())elm_5node_pyramid_triangulate;
     ElementDef.PointInside   = (int (*)())NULL;

     return elm_add_element_type( &ElementDef );
}