point_locator_test.C

Go to the documentation of this file.

#include <libmesh/mesh.h>
#include <libmesh/mesh_generation.h>
#include <libmesh/elem.h>
#include <libmesh/node.h>
#include <libmesh/parallel.h>
 
#include "test_comm.h"
#include "libmesh_cppunit.h"
 
 
using namespace libMesh;
 
 
 
class PointLocatorTest : public CppUnit::TestCase {
public:
  CPPUNIT_TEST_SUITE( PointLocatorTest );
 
  CPPUNIT_TEST( testLocatorOnEdge3 );
#if LIBMESH_DIM > 1
  CPPUNIT_TEST( testLocatorOnQuad9 );
  CPPUNIT_TEST( testLocatorOnTri6 );
#endif
#if LIBMESH_DIM > 2
  CPPUNIT_TEST( testLocatorOnHex27 );
#endif
 
  CPPUNIT_TEST_SUITE_END();
 
private:
 
public:
  void setUp()
  {}
 
  void tearDown()
  {}
 
  void testLocator(const ElemType elem_type)
  {
    Mesh mesh(*TestCommWorld);
 
    const unsigned int n_elem_per_side = 5;
    const std::unique_ptr<Elem> test_elem = Elem::build(elem_type);
    const unsigned int ymax = test_elem->dim() > 1;
    const unsigned int zmax = test_elem->dim() > 2;
    const unsigned int ny = ymax * n_elem_per_side;
    const unsigned int nz = zmax * n_elem_per_side;
 
    MeshTools::Generation::build_cube (mesh,
                                       n_elem_per_side,
                                       ny,
                                       nz,
                                       0., 1.,
                                       0., ymax,
                                       0., zmax,
                                       elem_type);
 
    std::unique_ptr<PointLocatorBase> locator = mesh.sub_point_locator();
 
    if (!mesh.is_serial())
      locator->enable_out_of_mesh_mode();
 
    for (unsigned int i=0; i != n_elem_per_side+1; ++i)
      {
        for (unsigned int j=0; j != ny+1; ++j)
          {
            for (unsigned int k=0; k != nz+1; ++k)
              {
                const libMesh::Real h = libMesh::Real(1)/n_elem_per_side;
                Point p(i*h, j*h, k*h);
 
                const Elem *elem = locator->operator()(p);
 
                bool found_elem = elem;
                if (!mesh.is_serial())
                  mesh.comm().max(found_elem);
 
                CPPUNIT_ASSERT(found_elem);
                if (elem)
                  {
                    CPPUNIT_ASSERT(elem->contains_point(p));
                  }
 
                const Node *node = locator->locate_node(p);
 
                bool found_node = node;
                if (!mesh.is_serial())
                  mesh.comm().max(found_node);
 
                CPPUNIT_ASSERT(found_node);
 
                if (node)
                  {
                    LIBMESH_ASSERT_FP_EQUAL((*node)(0), i*h,
                                            TOLERANCE*TOLERANCE);
                    if (LIBMESH_DIM > 1)
                      LIBMESH_ASSERT_FP_EQUAL((*node)(1), j*h,
                                              TOLERANCE*TOLERANCE);
                    if (LIBMESH_DIM > 2)
                      LIBMESH_ASSERT_FP_EQUAL((*node)(2), k*h,
                                              TOLERANCE*TOLERANCE);
                  }
              }
          }
      }
  }
 
 
 
  void testLocatorOnEdge3() { testLocator(EDGE3); }
  void testLocatorOnQuad9() { testLocator(QUAD9); }
  void testLocatorOnTri6()  { testLocator(TRI6); }
  void testLocatorOnHex27() { testLocator(HEX27); }
 
};
 
CPPUNIT_TEST_SUITE_REGISTRATION( PointLocatorTest );