Loading [MathJax]/extensions/tex2jax.js
Mesh Oriented datABase  (version 5.5.1)
An array-based unstructured mesh library
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
KDTree.cpp File Reference
#include "moab/Core.hpp"
#include "moab/AdaptiveKDTree.hpp"
#include "moab/Range.hpp"
#include "moab/GeomUtil.hpp"
#include <iostream>
#include <string>
+ Include dependency graph for KDTree.cpp:

Go to the source code of this file.

Macros

#define CHKERR(err)
 

Functions

void print_error (moab::Interface &mb, moab::ErrorCode err)
 
moab::EntityHandle hex_containing_point (moab::Interface &mb, moab::EntityHandle set, const double point[3])
 
void print_hex (moab::Interface &mb, moab::EntityHandle hex)
 
int main ()
 

Variables

const double EPSILON = 1e-6
 

Macro Definition Documentation

◆ CHKERR

#define CHKERR (   err)
Value:
do \ { \ if( moab::MB_SUCCESS != ( err ) ) \ { \ print_error( mb, ( err ) ); \ std::cerr << "Unexpected failure at: " << __FILE__ << ":" << __LINE__ << std::endl; \ return 2; \ } \ } while( false )

Definition at line 30 of file KDTree.cpp.

Function Documentation

◆ hex_containing_point()

moab::EntityHandle hex_containing_point ( moab::Interface mb,
moab::EntityHandle  set,
const double  point[3] 
)

Definition at line 103 of file KDTree.cpp.

104 { 105  moab::ErrorCode rval; 106  moab::CartVect pt( point ); // input location 107  moab::CartVect coords[8]; // coordinates of corners of hexahedron 108  const moab::EntityHandle* conn; // hex connectivity 109  int conn_len; 110  111  // Get hexes in leaf 112  std::vector< moab::EntityHandle > hexes; 113  rval = mb.get_entities_by_type( set, moab::MBHEX, hexes );CHKERR( rval ); 114  115  // Check which hex the point is in 116  std::vector< moab::EntityHandle >::const_iterator i; 117  for( i = hexes.begin(); i != hexes.end(); ++i ) 118  { 119  rval = mb.get_connectivity( *i, conn, conn_len );CHKERR( rval ); 120  rval = mb.get_coords( conn, 8, &coords[0][0] );CHKERR( rval ); 121  if( moab::GeomUtil::point_in_trilinear_hex( coords, pt, EPSILON ) ) return *i; 122  } 123  124  // Return 0 if no hex contains point. 125  return 0; 126 }

References CHKERR, EPSILON, ErrorCode, mb, MBHEX, and moab::GeomUtil::point_in_trilinear_hex().

Referenced by main().

◆ main()

int main ( )

Definition at line 49 of file KDTree.cpp.

50 { 51  // Ask user for file to read 52  std::string filename; 53  std::cout << "Hex mesh file name: "; 54  std::cin >> filename; 55  56  // Read file into MOAB instance 57  moab::ErrorCode rval; 58  moab::Core moab; 59  moab::Interface& mb = moab; 60  rval = mb.load_file( filename.c_str() ); 61  if( moab::MB_SUCCESS != rval ) 62  { 63  print_error( mb, rval ); 64  std::cerr << filename << ": file load failed" << std::endl; 65  return 1; 66  } 67  68  // Get all hex elemeents 69  moab::Range elems; 70  rval = mb.get_entities_by_type( 0, moab::MBHEX, elems );CHKERR( rval ); 71  if( elems.empty() ) 72  { 73  std::cerr << filename << ": file containd no hexahedra" << std::endl; 74  return 1; 75  } 76  77  // Build a kD-tree from hex elements 78  moab::EntityHandle tree_root; 79  moab::AdaptiveKDTree tool( &mb ); 80  rval = tool.build_tree( elems, tree_root );CHKERR( rval ); 81  82  // Loop forever (or until EOF), asking user for a point 83  // to query and printing the hex element containing that 84  // point. 85  for( ;; ) 86  { 87  double point[3]; 88  std::cout << "Point coordinates: "; 89  if( !( std::cin >> point[0] >> point[1] >> point[2] ) ) break; 90  91  moab::EntityHandle leaf; 92  rval = tool.leaf_containing_point( tree_root, point, leaf );CHKERR( rval ); 93  moab::EntityHandle hex = hex_containing_point( mb, leaf, point ); 94  if( 0 == hex ) 95  std::cout << "Point is not contained in any hexahedron." << std::endl; 96  else 97  print_hex( mb, hex ); 98  } 99  100  return 0; 101 }

References moab::AdaptiveKDTree::build_tree(), CHKERR, moab::Range::empty(), ErrorCode, hex_containing_point(), mb, MB_SUCCESS, MBHEX, print_error(), and print_hex().

◆ print_error()

void print_error ( moab::Interface mb,
moab::ErrorCode  err 
)

Definition at line 19 of file KDTree.cpp.

20 { 21  std::string message; 22  std::string code; 23  if( moab::MB_SUCCESS != mb.get_last_error( message ) ) message.clear(); 24  code = mb.get_error_string( err ); 25  std::cerr << "Error: " << code << std::endl; 26  if( !message.empty() ) std::cerr << " " << message << std::endl; 27 }

References mb, and MB_SUCCESS.

Referenced by main(), and moab::FileTokenizer::match_token().

◆ print_hex()

void print_hex ( moab::Interface mb,
moab::EntityHandle  hex 
)

Definition at line 128 of file KDTree.cpp.

129 { 130  // Get MOAB's internal ID for hex element 131  int id = mb.id_from_handle( hex ); 132  133  // Get vertex handles for hex corners 134  const moab::EntityHandle* conn; // hex connectivity 135  int conn_len; 136  mb.get_connectivity( hex, conn, conn_len ); 137  138  // Get coordinates of vertices 139  double coords[3 * 8]; 140  mb.get_coords( conn, 8, coords ); 141  142  // Print 143  std::cout << " Point is in hex " << id << " with corners: " << std::endl; 144  for( int i = 0; i < 8; ++i ) 145  { 146  std::cout << " (" << coords[3 * i] << ", " << coords[3 * i + 1] << ", " << coords[3 * i + 2] << ")" 147  << std::endl; 148  } 149 }

References mb.

Referenced by main().

Variable Documentation

◆ EPSILON