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.

{
 moab::ErrorCode rval;
 moab::CartVect pt( point ); // input location
 moab::CartVect coords[8]; // coordinates of corners of hexahedron
 const moab::EntityHandle* conn; // hex connectivity
 int conn_len;
 
 // Get hexes in leaf
 std::vector< moab::EntityHandle > hexes;
 rval = mb.get_entities_by_type( set, moab::MBHEX, hexes );CHKERR( rval );
 
 // Check which hex the point is in
 std::vector< moab::EntityHandle >::const_iterator i;
 for( i = hexes.begin(); i != hexes.end(); ++i )
 {
 rval = mb.get_connectivity( *i, conn, conn_len );CHKERR( rval );
 rval = mb.get_coords( conn, 8, &coords[0][0] );CHKERR( rval );
 if( moab::GeomUtil::point_in_trilinear_hex( coords, pt, EPSILON ) ) return *i;
 }
 
 // Return 0 if no hex contains point.
 return 0;
}

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.

{
 // Ask user for file to read
 std::string filename;
 std::cout << "Hex mesh file name: ";
 std::cin >> filename;
 
 // Read file into MOAB instance
 moab::ErrorCode rval;
 moab::Core moab;
 moab::Interface& mb = moab;
 rval = mb.load_file( filename.c_str() );
 if( moab::MB_SUCCESS != rval )
 {
 print_error( mb, rval );
 std::cerr << filename << ": file load failed" << std::endl;
 return 1;
 }
 
 // Get all hex elemeents
 moab::Range elems;
 rval = mb.get_entities_by_type( 0, moab::MBHEX, elems );CHKERR( rval );
 if( elems.empty() )
 {
 std::cerr << filename << ": file containd no hexahedra" << std::endl;
 return 1;
 }
 
 // Build a kD-tree from hex elements
 moab::EntityHandle tree_root;
 moab::AdaptiveKDTree tool( &mb );
 rval = tool.build_tree( elems, tree_root );CHKERR( rval );
 
 // Loop forever (or until EOF), asking user for a point
 // to query and printing the hex element containing that
 // point.
 for( ;; )
 {
 double point[3];
 std::cout << "Point coordinates: ";
 if( !( std::cin >> point[0] >> point[1] >> point[2] ) ) break;
 
 moab::EntityHandle leaf;
 rval = tool.leaf_containing_point( tree_root, point, leaf );CHKERR( rval );
 moab::EntityHandle hex = hex_containing_point( mb, leaf, point );
 if( 0 == hex )
 std::cout << "Point is not contained in any hexahedron." << std::endl;
 else
 print_hex( mb, hex );
 }
 
 return 0;
}

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.

{
 std::string message;
 std::string code;
 if( moab::MB_SUCCESS != mb.get_last_error( message ) ) message.clear();
 code = mb.get_error_string( err );
 std::cerr << "Error: " << code << std::endl;
 if( !message.empty() ) std::cerr << " " << message << std::endl;
}

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.

{
 // Get MOAB's internal ID for hex element
 int id = mb.id_from_handle( hex );
 
 // Get vertex handles for hex corners
 const moab::EntityHandle* conn; // hex connectivity
 int conn_len;
 mb.get_connectivity( hex, conn, conn_len );
 
 // Get coordinates of vertices
 double coords[3 * 8];
 mb.get_coords( conn, 8, coords );
 
 // Print
 std::cout << " Point is in hex " << id << " with corners: " << std::endl;
 for( int i = 0; i < 8; ++i )
 {
 std::cout << " (" << coords[3 * i] << ", " << coords[3 * i + 1] << ", " << coords[3 * i + 2] << ")"
 << std::endl;
 }
}

References mb.

Referenced by main().

Variable Documentation

EPSILON

const double EPSILON = 1e-6

Definition at line 15 of file KDTree.cpp.

Referenced by moab::BSPTreePoly::cut_polyhedron(), hex_containing_point(), and moab::BSPTreePoly::is_point_contained().