An Object Finite Element Library

MeshExtract.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

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

               Definition of classes for extracting submeshes

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

#ifndef __MESH_EXTRACT_H
#define __MESH_EXTRACT_H

#include "Mesh.h"

namespace OFELI {

class NodeList {

 public:

    NodeList(Mesh &ms);

    ~NodeList() { }

    void selectCode(int code, int dof=1);
            
    void selectCoordinate(double x, double y=ANY, double z=ANY);

    size_t getNbNodes() const { return _nb_nodes; }

    void top() { _it = 0; }

    void top() const { _const_it = 0; }

    Node *get()
    {
       if (_it==_nb_nodes)
          return NULL;
       else
          return _theList[_it++];
    }

    Node *get() const
    {
       if (_const_it==_nb_nodes)
          return NULL;
       else
          return _theList[_const_it++];
    }

 private:

   Mesh *_ms;
            NodeSet _theList;
            size_t _it, _nb_nodes, _nbn;
   mutable size_t _const_it;
};


class ElementList {

 public:

    ElementList(Mesh &ms);

    ~ElementList() { }

    void selectCode(int code);

    size_t getNbElements() const { return _nb_elements; }

    void top() { _it = 0; }

    void top() const { _const_it = 0; }

    Element *get()
    {
       if (_it==_nb_elements)
          return NULL;
       else
          return _theList[_it++];
    }

    Element *get() const
    {
       if (_const_it==_nb_elements)
          return NULL;
       else
          return _theList[_const_it++];
    }

 private:

   Mesh *_ms;
   ElementSet _theList;
   size_t _it, _nb_elements, _nbn;
   mutable size_t _const_it;
};


class SideList {

 public:

    SideList(Mesh &ms);

    ~SideList() { }

    void selectCode(int code, int dof=1);

    size_t getNbSides() const { return _nb_sides; }

    void top() { _it = 0; }

    void top() const { _const_it = 0; }

    Side *get()
    {
       if (_it==_nb_sides)
          return NULL;
       else
          return _theList[_it++];
    }

    Side *get() const
    {
       if (_const_it==_nb_sides)
          return NULL;
       else
          return _theList[_const_it++];
    }

 private:

   Mesh *_ms;
   SideSet _theList;
            size_t _it, _nb_sides, _nbn;
   mutable size_t _const_it;
};


    ostream& operator<<(ostream& s, const NodeList &nl);

    ostream& operator<<(ostream& s, const ElementList &el);

    ostream& operator<<(ostream& s, const SideList &sl);

} /* namespace OFELI */

#endif

Copyright © 1998-2007 Rachid Touzani