src/INTERP_KERNEL/MeshUtils.hxx

Go to the documentation of this file.

// Copyright (C) 2007-2011  CEA/DEN, EDF R&D
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License.
//
// This library 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
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
//
// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
//

#ifndef __MESHUTILS_HXX__
#define __MESHUTILS_HXX__

#include "InterpolationUtils.hxx"

namespace INTERP_KERNEL
{
  template<class MyMeshType>
  inline typename MyMeshType::MyConnType getGlobalNumberOfNode(typename MyMeshType::MyConnType node, typename MyMeshType::MyConnType element, const MyMeshType& mesh)
  {
    typedef typename MyMeshType::MyConnType ConnType;
    const NumberingPolicy numPol=MyMeshType::My_numPol;
    const ConnType elemIdx=OTT<ConnType,numPol>::conn2C(mesh.getConnectivityIndexPtr()[OTT<ConnType,numPol>::ind2C(element)]);
    if(mesh.getTypeOfElement(element)!=INTERP_KERNEL::NORM_POLYHED)
      return OTT<ConnType,numPol>::coo2C(mesh.getConnectivityPtr()[elemIdx + node]);
    else
      {
        const ConnType *startNodalConnOfElem=mesh.getConnectivityPtr()+elemIdx;
        ConnType ptr=0,ret=0;
        while(startNodalConnOfElem[ret]==-1 || ptr!=node)
          {
            ret++;
            if(startNodalConnOfElem[ret]!=-1)
              ptr++;
          }
        return OTT<ConnType,numPol>::coo2C(startNodalConnOfElem[ret]);
      }
  }

  template<class MyMeshType>
  inline const double* getCoordsOfNode(typename MyMeshType::MyConnType node, typename MyMeshType::MyConnType element, const MyMeshType& mesh)
  {
    typedef typename MyMeshType::MyConnType ConnType;
    const ConnType connIdx = getGlobalNumberOfNode(node, element, mesh);
    const double *ret=mesh.getCoordinatesPtr()+MyMeshType::MY_SPACEDIM*connIdx;
    return ret;
  }

  template<class MyMeshType>
  inline const double* getCoordsOfNode2(typename MyMeshType::MyConnType node, typename MyMeshType::MyConnType element, const MyMeshType& mesh, typename MyMeshType::MyConnType& nodeId)
  {
    typedef typename MyMeshType::MyConnType ConnType;
    nodeId= getGlobalNumberOfNode(node, element, mesh);
    return mesh.getCoordinatesPtr()+MyMeshType::MY_SPACEDIM*nodeId;
  }

  template<class MyMeshType, int NB_NODES>
  inline void getBarycentricCoordinates(const double*                   point,
                                        typename MyMeshType::MyConnType element,
                                        const MyMeshType&               mesh,
                                        double*                         barycentricCoords)
  {
    std::vector<const double*> nodes( NB_NODES );
    for ( int node = 0; node < NB_NODES; ++node )
      {
        nodes[ node ] = getCoordsOfNode( node, element, mesh );
      }
    barycentric_coords( nodes, point, barycentricCoords );
  }
    
}

#endif