An Object Finite Element Library

UserData.h

Go to the documentation of this file.

/*==============================================================================

                                    O  F  E  L  I

                           Object  Finite  Element  Library

  ==============================================================================

   Copyright (C) 1998 - 2008 Rachid Touzani

   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; Version 2 of the License.

   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; if not, write to the :

   Free Software Foundation
   Inc., 59 Temple Place - Suite 330
   Boston, MA  02111-1307, USA

  ==============================================================================

              Template Class 'UserData' for user data prescription

  ==============================================================================*/


#ifndef __USERDATA_H
#define __USERDATA_H


#include <float.h>
#include <stdlib.h>
#include <math.h>

#include "OFELI_Config.h"
#include "Mesh.h"
#include "Vect.h"
#include "BCVect.h"
#include "Point.h"

namespace OFELI {

template <class T_> class UserData
{

public :

    UserData()
    {
       _time = 0.;
#ifdef _OFELI_DEBUG
       std::clog << "An instance of class UserData is constructed.\n";
       std::clog << "File : " << __FILE__ << ", Line : " << __LINE__ << endl;
#endif
    }

    UserData(const class Mesh &mesh)
    {
       _theMesh = &mesh;
       _time = 0.;
#ifdef _OFELI_DEBUG
       std::clog << "An instance of class UserData is constructed.\n";
       std::clog << "File : " << __FILE__ << ", Line : " << __LINE__ << endl;
#endif
    }

    virtual ~UserData()
    {
#ifdef _OFELI_DEBUG
       std::clog << "An instance of class UserData is destructed.\n";
       std::clog << "File : " << __FILE__ << ", Line : " << __LINE__ << endl;
#endif
    }

    void setTime(double time) { _time = time; }

    void setDBC(Vect<T_> &b)
    {
       const Node *nd;
       size_t k = 1;
       for (_theMesh->topNode(); (nd=_theMesh->getNode());) {
          Point<double> x(nd->getCoord(1),nd->getCoord(2),nd->getCoord(3));
          for (size_t i=1; i<=nd->getNbDOF(); i++)
             b(k++) = BoundaryCondition(x,nd->getCode(i),_time,i);
       }
    }

    void setInitialData(Vect<T_> &b)
    {
       const Node *nd;
       for (_theMesh->topNode(); (nd=_theMesh->getNode());) {
          Point<double> x(nd->getCoord(1),nd->getCoord(2),nd->getCoord(3));
          for (size_t j=1; j<=nd->getNbDOF(); j++)
             b(nd->getDOF(j)) = InitialData(x,j);
       }
    }

    void setBodyForce(Vect<T_> &b)
    {
       const Node *nd;
       for (_theMesh->topNode(); nd=_theMesh->getNode();) {
          Point<double> x(nd->getCoord(1),nd->getCoord(2),nd->getCoord(3));
          for (size_t j=1; j<=nd->getNbDOF(); j++)
             b(nd->getDOF(j)) = BodyForce(x,time,j);
       }
    }

    void setBodyForce(NodeVect<T_> &b)
    {
       const Node *nd;
       for (_theMesh->topNode(); nd=_theMesh->getNode();) {
          Point<double> x(nd->getCoord(1),nd->getCoord(2),nd->getCoord(3));
          for (size_t j=1; j<=nd->getNbDOF(); j++)
             b(nd->getDOF(j)) = BodyForce(x,time,j);
       }
    }

    void setBodyForce(ElementVect<T_> &b)
    {
       Point<double> c;
       const Element *el;
       size_t dim = _theMesh->getDim();
       for (_theMesh->topElement(); el=_theMesh->getElement();) {
          if (dim==1 && el->getShape()==Mesh::LINE) {
             class Line2 ln(el);
             c = ln.getCenter();
          }
          else if (dim==2 && el->getShape()==Mesh::TRIANGLE) {
             class Triang3 tr(el);
             c = tr.getCenter();
          }
          else if (dim==2 && el->getShape()==Mesh::QUADRILATERAL) {
             class Quad qd(el);
             c = qd.getCenter();
          }
          else if (dim==3 && el->getShape()==Mesh::TETRAHEDRON) {
             class Tetra4 tt(el);
             c = tt.getCenter();
          }
          else if (dim==3 && el->getShape()==Mesh::HEXAHEDRON) {
             class Hexa8 hx(el);
             c = hx.getCenter();
          }
          else
             ;
          b(el->getLabel()) = BodyForce(c,time,1);
       }
    }

    void setSurfaceForce(Vect<T_> &b)
    {
       const Node *nd;
       for (_theMesh->topNode(); nd=_theMesh->getNode();) {
          Point<double> coord(nd->getCoord(1),nd->getCoord(2),nd->getCoord(3));
          for (size_t j=1; j<=nd->getNbDOF(); j++)
             b(nd->getDOF(j)) = SurfaceForce(coord,nd->getCode(j),_time,j);
       }
    }

    virtual T_ BoundaryCondition(const Point<double> &x, int code, double time=0., size_t dof=1)
    {
       if (&x) { }
       time = 0;
       dof = 1;
       code = 0;
       return T_(0);
    }

    virtual T_ BodyForce(const Point<double> &x, double time=0., size_t dof=1)
    {
       if (&x) { }
       time = 0;
       dof = 1;
       return T_(0);
    }

    virtual T_ SurfaceForce(const Point<double> &x, int code, double time=0., size_t dof=1)
    {
       if (&x) { }
       time = 0;
       dof = 1;
       code = 0;
       return T_(0);
    }

    virtual T_ InitialData(const Point<double> &x, size_t dof=1)
    {
       if (&x){ }
       dof = 1;
       return T_(0.);
    }


protected :

#ifndef DOXYGEN_SHOULD_SKIP_THIS
   double      _time;
   const Mesh *_theMesh;
#endif /* DOXYGEN_SHOULD_SKIP_THIS */
};

} /* namespace OFELI */

#endif

Copyright © 1998-2007 Rachid Touzani