test_entsetFB.cpp

Go to the documentation of this file.

#include "moab/MOABConfig.h"
#include "FBiGeom.h"
#include "iMesh.h"
#include "iRel.h"
 
#include <cstdio>
#include <cstdlib>
#include <cstring>
 
#define DEFAULT_TEST_FILE shell.h5m
#define DEFAULT_TEST_FILE1 shellQuad.h5m
 
// clang-format off
 
#define STRINGIFY_( X ) #X
#define STRINGIFY( X ) STRINGIFY_( X )
#if defined( MESHDIR ) && defined( MOAB_HAVE_HDF5 )
#define DEFAULT_INPUT_FILE STRINGIFY( MESHDIR/fbigeom/DEFAULT_TEST_FILE )
#define DEFAULT_INPUT_FILE1 STRINGIFY( MESHDIR/fbigeom/DEFAULT_TEST_FILE1 )
#else
#error Specify MESHDIR to compile test
#endif
 
#define CHECK_SIZE_C( type, array, allocated_size, size ) \
 if( NULL == *(array) || *(allocated_size) == 0 ) \
 { \
 *(array) = (type*)malloc( sizeof( type ) * (size) ); \
 *(allocated_size) = size; \
 } \
 else if( *(allocated_size) < (size) ) \
 { \
 printf( " Array passed in is non-zero but too short.\n" ); \
 }
 
// clang-format on
 
typedef void* iRel_EntityHandle;
 
/*!
 prints out a result string based on the value of error_code
*/
void handle_error_code( const int result, int* number_failed, int* number_not_implemented, int* number_successful )
{
 if( result )
 {
 printf( "Success" );
 ( *number_successful )++;
 }
 else
 {
 printf( "Failure" );
 ( *number_failed )++;
 }
}
 
int print_geom_info( FBiGeom_Instance geom, iBase_EntityHandle gent )
{
 /* print information about this entity */
 int ent_type;
 int result;
 const char* type_names[] = { "Vertex", "Edge", "Face", "Region" };
 
 FBiGeom_getEntType( geom, gent, &ent_type, &result );
 
 if( iBase_SUCCESS != result )
 {
 printf( "Trouble getting entity adjacencies or types." );
 return 0;
 }
 
 printf( "%s 0x%lx\n", type_names[ent_type], (unsigned long)gent );
 
 return 1;
}
 
int print_mesh_info( iMesh_Instance mesh, iBase_EntityHandle ment )
{
 /* print information about this entity */
 
 /* get adjacencies first; assume not more than 50 */
 iBase_EntityHandle adj_ents[50], *adj_ents_ptr = adj_ents;
 int ent_types[50], *ent_types_ptr = ent_types;
 int adj_ents_alloc = 50, adj_ents_size, ent_types_size, ent_types_allocated = 50;
 int result;
 iBase_TagHandle* ment_tags = NULL;
 int ment_tags_size, ment_tags_alloc;
 char** tag_names;
 int i;
 const char* type_names[] = { "Vertex", "Edge", "Face", "Region" };
 int tag_type;
 
 char* dum_handle = NULL;
 int dum_handle_alloc = 0, dum_handle_size = 0;
 int int_data;
 double dbl_data;
 iBase_EntityHandle eh_data;
 
 iMesh_getEntAdj( mesh, ment, iBase_ALL_TYPES, &adj_ents_ptr, &adj_ents_alloc, &adj_ents_size, &result );
 
 if( iBase_SUCCESS != result ) return 0;
 
 /* put this ent on the end, then get types */
 adj_ents[adj_ents_size] = ment;
 iMesh_getEntArrType( mesh, adj_ents, adj_ents_size + 1, &ent_types_ptr, &ent_types_allocated, &ent_types_size,
 &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Trouble getting entity adjacencies or types." );
 return 0;
 }
 
 /* get tags on ment */
 iMesh_getAllTags( mesh, ment, &ment_tags, &ment_tags_alloc, &ment_tags_size, &result );
 
 printf( "Trouble getting tags on an entity or their names." );
 
 /* while we're at it, get all the tag names */
 
 tag_names = (char**)malloc( ment_tags_size * sizeof( char* ) );
 
 for( i = 0; i < ment_tags_size; i++ )
 {
 tag_names[i] = (char*)malloc( 120 * sizeof( char ) );
 iMesh_getTagName( mesh, ment_tags[i], tag_names[i], &result, 120 );
 }
 
 /* now print the information */
 printf( "%s %ld:\n", type_names[ent_types[ent_types_size - 1]], (long)ment );
 printf( "Adjacencies:" );
 for( i = 0; i < adj_ents_size; i++ )
 {
 if( i > 0 ) printf( ", " );
 printf( "%s %ld", type_names[ent_types[i]], (long)adj_ents[i] );
 }
 printf( "\nTags: \n" );
 for( i = 0; i < ment_tags_size; i++ )
 {
 printf( "%s ", tag_names[i] );
 iMesh_getTagType( mesh, ment_tags[i], &tag_type, &result );
 if( iBase_SUCCESS != result )
 printf( "(trouble getting type...)\n" );
 else
 {
 dum_handle = NULL;
 dum_handle_alloc = 0, dum_handle_size = 0;
 
 switch( tag_type )
 {
 case iBase_INTEGER:
 iMesh_getIntData( mesh, ment, ment_tags[i], &int_data, &result );
 printf( "(Int value=%d)", int_data );
 break;
 case iBase_DOUBLE:
 iMesh_getDblData( mesh, ment, ment_tags[i], &dbl_data, &result );
 printf( "(Dbl value=%f)", dbl_data );
 break;
 case iBase_ENTITY_HANDLE:
 iMesh_getEHData( mesh, ment, ment_tags[i], &eh_data, &result );
 printf( "(EH value=%ld)", (long)eh_data );
 break;
 case iBase_BYTES:
 iMesh_getData( mesh, ment, ment_tags[i], (void**)&dum_handle, &dum_handle_alloc, &dum_handle_size,
 &result );
 if( NULL != dum_handle && dum_handle_size > 0 ) printf( "(Opaque value=%c)", dum_handle[0] );
 break;
 }
 }
 
 printf( "\n" );
 }
 printf( "(end tags)\n\n" );
 free( ment_tags );
 return 1;
}
 
/*!
 @test
 Load Mesh
 @li Load a geom and a mesh file
*/
int load_geom_mesh_test( const char* geom_filename,
 const char* mesh_filename,
 FBiGeom_Instance geom,
 iMesh_Instance mesh )
{
 /* load a geom */
 int result;
 FBiGeom_load( geom, geom_filename, 0, &result, strlen( geom_filename ), 0 );
 if( iBase_SUCCESS != result )
 {
 printf( "ERROR : can not load a geometry\n" );
 return 0;
 }
 
 /* load a mesh */
 iMesh_load( mesh, 0, mesh_filename, 0, &result, strlen( mesh_filename ), 0 );
 if( iBase_SUCCESS != result )
 {
 printf( "ERROR : can not load a mesh\n" );
 return 0;
 }
 
 return 1;
}
 
/*!
 @test
 TSTTLasso create relation Test
 @li Create relation between geom and mesh
*/
int create_relation_test( iRel_Instance assoc, FBiGeom_Instance geom, iMesh_Instance mesh, iRel_PairHandle* pair )
{
 int result;
 
 iBase_Instance iface1, iface2;
 int type1, type2;
 int ent_or_set1, ent_or_set2;
 int status1, status2;
 
 iRel_PairHandle tmp_pair;
 iRel_PairHandle* pair_ptr = &tmp_pair;
 int pairs_alloc = 1, pairs_size;
 
 /* create an relation, entity to set */
 iRel_createPair( assoc, geom, iRel_ENTITY, iRel_FBIGEOM_IFACE, iRel_ACTIVE, mesh, iRel_SET, iRel_IMESH_IFACE,
 iRel_ACTIVE, pair, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Couldn't create a new relation.\n" );
 return 0;
 }
 
 iRel_getPairInfo( assoc, *pair, &iface1, &ent_or_set1, &type1, &status1, &iface2, &ent_or_set2, &type2, &status2,
 &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Couldn't retrieve relation info.\n" );
 return 0;
 }
 if( iface1 != geom || ent_or_set1 != iRel_ENTITY || type1 != iRel_FBIGEOM_IFACE || iface2 != mesh ||
 ent_or_set2 != iRel_SET || type2 != iRel_IMESH_IFACE )
 {
 printf( "Unexpected relation info returned.\n" );
 return 0;
 }
 
 iRel_findPairs( assoc, geom, &pair_ptr, &pairs_alloc, &pairs_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Couldn't find relation pair when querying geom.\n" );
 return 0;
 }
 if( pairs_size != 1 || tmp_pair != *pair )
 {
 printf( "Unexpected relation pairs returned when querying geom.\n" );
 return 0;
 }
 
 iRel_findPairs( assoc, mesh, &pair_ptr, &pairs_alloc, &pairs_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Couldn't find relation pair when querying mesh.\n" );
 return 0;
 }
 if( pairs_size != 1 || tmp_pair != *pair )
 {
 printf( "Unexpected relation pairs returned when querying mesh.\n" );
 return 0;
 }
 
 return 1;
}
 
/*!
 @test
 TSTTLasso relate geom and mesh Test
 @li Check relation between geom and mesh
*/
int relate_geom_mesh_test( iRel_Instance assoc, FBiGeom_Instance geom, iMesh_Instance mesh, iRel_PairHandle pair )
{
 /* relate geometry entities with coresponding mesh entity sets */
 iBase_EntityHandle* gentities = NULL;
 int gentities_size = 0, gentities_alloc = 0;
 int result;
 
 iBase_EntitySetHandle* mentity_handles = NULL;
 int mentity_handles_size = 0, mentity_handles_alloc = 0;
 
 const char* dim_tag_name = "GEOM_DIMENSION";
 iBase_TagHandle dim_tag_mesh;
 
 iBase_EntitySetHandle* mentities_vec;
 int mentities_vec_size;
 int i;
 
 iBase_EntitySetHandle* out_mentities = NULL;
 int out_mentities_size = 0, out_mentities_alloc = 0;
 
 iBase_EntitySetHandle* out_mentities2 = NULL;
 int out_mentities2_size = 0, out_mentities2_alloc = 0;
 
 iBase_EntityHandle* out_gentities = NULL;
 int out_gentities_size = 0, out_gentities_alloc = 0;
 
 FBiGeom_getEntities( geom, NULL, iBase_VERTEX, &gentities, &gentities_alloc, &gentities_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to get gentities by type in relate_geom_mesh_test.\n" );
 return 0;
 }
 
 iRel_inferEntArrRelations( assoc, pair, gentities, gentities_size, 0, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to relate geom entities in relate_geom_mesh_test.\n" );
 return 0;
 }
 
 /* relate coresponding mesh entity sets for geometry entities */
 /* get 1-dimensional mesh entitysets */
 
 iMesh_getEntSets( mesh, NULL, 1, &mentity_handles, &mentity_handles_alloc, &mentity_handles_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Problem to get all entity sets.\n" );
 return 0;
 }
 
 /* get geom dimension tags for mesh entitysets */
 iMesh_createTag( mesh, dim_tag_name, 1, iBase_INTEGER, &dim_tag_mesh, &result, 15 );
 if( iBase_SUCCESS != result && result != iBase_TAG_ALREADY_EXISTS )
 {
 printf( "Couldn't create geom dim tag for mesh entities.\n" );
 return 0;
 }
 
 /* get 1-dimensional mesh entitysets */
 mentities_vec = (iBase_EntitySetHandle*)malloc( mentity_handles_size * sizeof( iBase_EntitySetHandle ) );
 mentities_vec_size = 0;
 
 for( i = 0; i < mentity_handles_size; i++ )
 {
 int dim;
 iMesh_getEntSetIntData( mesh, mentity_handles[i], dim_tag_mesh, &dim, &result );
 if( iBase_SUCCESS != result ) continue;
 
 if( dim == 1 ) mentities_vec[mentities_vec_size++] = mentity_handles[i];
 }
 
 iRel_inferSetArrRelations( assoc, pair, mentities_vec, mentities_vec_size, 1, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to relate mesh entities in relate_geom_mesh_test.\n" );
 return 0;
 }
 
 /* relate all geometry and mesh entities */
 iRel_inferAllRelations( assoc, pair, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to relate all geom and mesh entities in relate_geom_mesh_test.\n" );
 return 0;
 }
 
 /* reset geom entities list and get all geom entities (prev
 only vertices) */
 free( gentities );
 gentities = NULL;
 gentities_alloc = 0;
 FBiGeom_getEntities( geom, NULL, iBase_ALL_TYPES, &gentities, &gentities_alloc, &gentities_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to get gentities by type in relate_geom_mesh_test.\n" );
 return 0;
 }
 
 /* get related mesh entity sets for geometry entities */
 iRel_getEntArrSetArrRelation( assoc, pair, gentities, gentities_size, 0, &out_mentities, &out_mentities_alloc,
 &out_mentities_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to get geom entities in relate_geom_mesh_test.\n" );
 return 0;
 }
 
 if( out_mentities_size != gentities_size )
 {
 printf( "Number of input geom entities and output mesh entity sets should be same\n" );
 return 0;
 }
 
 /* now try deleting this relation */
 iRel_rmvEntArrRelation( assoc, pair, gentities, gentities_size, 0, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to remove relation in relate_geom_mesh_test.\n" );
 return 0;
 }
 iRel_getEntArrSetArrRelation( assoc, pair, gentities, gentities_size, 0, &out_mentities2, &out_mentities2_alloc,
 &out_mentities2_size, &result );
 if( iBase_SUCCESS == result )
 {
 printf( "Shouldn't have gotten mesh sets in relate_geom_mesh_test.\n" );
 return 0;
 }
 
 /* restore the relation, since we need it later */
 iRel_setEntArrSetArrRelation( assoc, pair, gentities, gentities_size, out_mentities, out_mentities_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to restore relation in relate_geom_mesh_test.\n" );
 return 0;
 }
 
 /* get related geometry entities for mesh entity sets */
 iRel_getSetArrEntArrRelation( assoc, pair, out_mentities, out_mentities_size, 1, &out_gentities,
 &out_gentities_alloc, &out_gentities_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to get mesh entities in relate_geom_mesh_test.\n" );
 return 0;
 }
 
 if( out_mentities_size != out_gentities_size )
 {
 printf( "Number of input mesh entity sets and output geom entities should be same\n" );
 return 0;
 }
 free( mentity_handles );
 mentity_handles = NULL;
 free( gentities );
 gentities = NULL;
 free( mentity_handles );
 mentity_handles = NULL;
 free( out_mentities );
 out_mentities = NULL;
 free( mentities_vec );
 mentities_vec = NULL;
 free( out_gentities );
 out_gentities = NULL;
 return 1;
}
 
/*!
 @test
 TSTTAssoc move to test
 @li Move meshes onto the given geometry
*/
int query_relations_test( iRel_Instance assoc, FBiGeom_Instance geom, iMesh_Instance mesh, iRel_PairHandle pair )
{
 /* get all the geom entities, and find relation to some mesh entity */
 iBase_EntityHandle* gentities = NULL;
 int gentities_size = 0, gentities_alloc = 0;
 int result;
 
 iBase_EntitySetHandle* out_mentities = NULL;
 int out_mentities_size, out_mentities_alloc = 0;
 
 char descr[120];
 int i;
 int is_list;
 
 iBase_EntityHandle* out_gentities = NULL;
 int out_gentities_size, out_gentities_alloc = 0;
 
 FBiGeom_getEntities( geom, NULL, iBase_ALL_TYPES, &gentities, &gentities_alloc, &gentities_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Problem getting all geom entities.\n" );
 return 0;
 }
 
 iRel_getEntArrSetArrRelation( assoc, pair, gentities, gentities_size, 0, &out_mentities, &out_mentities_alloc,
 &out_mentities_size, &result );
 /* might not all be */
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to get mesh entities related to geom entities in query_relations_test.\n" );
 
 iRel_getDescription( assoc, descr, sizeof( descr ) - 1 );
 printf( "Entities missing relations: %s\n", descr );
 
 for( i = 0; i < gentities_size; i++ )
 {
 print_geom_info( geom, gentities[i] );
 }
 
 return 0;
 }
 
 /* check that they're all non-null */
 if( out_mentities_size != gentities_size )
 {
 printf( "Number of mesh & related geom entities don't match.\n" );
 return 0;
 }
 
 /* check to make sure they're mesh sets; how to do that? */
 for( i = 0; i < out_mentities_size; i++ )
 {
 iMesh_isList( mesh, (iBase_EntitySetHandle)out_mentities[i], &is_list, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Entity set returned from classification wasn't valid.\n" );
 return 0;
 }
 }
 
 /* now turn around and check classification of those mesh entities */
 iRel_getSetArrEntArrRelation( assoc, pair, out_mentities, out_mentities_size, 1, &out_gentities,
 &out_gentities_alloc, &out_gentities_size, &result );
 if( iBase_SUCCESS != result )
 {
 printf( "Failed to get geom entities related to mesh entities in query_relations_test.\n" );
 return 0;
 }
 
 /* check that they're all non-null */
 if( out_mentities_size != out_gentities_size )
 {
 printf( "Number of geom & related mesh entities don't match.\n" );
 return 0;
 }
 
 free( gentities );
 gentities = NULL;
 free( out_mentities );
 out_mentities = NULL;
 free( out_gentities );
 out_gentities = NULL;
 /* ok, we're done */
 return 1;
}
 
int main( int argc, char* argv[] )
{
 /* Check command line arg */
 const char* geom_filename = DEFAULT_INPUT_FILE;
 const char* mesh_filename = DEFAULT_INPUT_FILE1;
 
 int result;
 int number_tests = 0;
 int number_tests_successful = 0;
 int number_tests_not_implemented = 0;
 int number_tests_failed = 0;
 
 FBiGeom_Instance geom;
 iMesh_Instance mesh;
 iRel_Instance assoc;
 iRel_PairHandle pair;
 
 if( argc == 2 && !strcmp( argv[1], "-h" ) )
 {
 printf( "Usage: %s <geom_filename> <mesh_filename>\n", argv[0] );
 return 1;
 }
 else if( argc == 2 )
 {
 geom_filename = argv[1];
 mesh_filename = argv[1];
 }
 else if( argc == 3 )
 {
 geom_filename = argv[1];
 mesh_filename = argv[2];
 }
 
 /* initialize the Geometry */
 FBiGeom_newGeom( 0, &geom, &result, 0 );
 
 /* initialize the Mesh */
 iMesh_newMesh( 0, &mesh, &result, 0 );
 
 /* initialize the Associate */
 iRel_create( 0, &assoc, &result, 0 );
 
 /* Print out Header information */
 printf( "\n\niRel TEST PROGRAM:\n\n" );
 
 /* load_geom_mesh test */
 printf( " load_geom_mesh: " );
 result = load_geom_mesh_test( geom_filename, mesh_filename, geom, mesh );
 handle_error_code( result, &number_tests_failed, &number_tests_not_implemented, &number_tests_successful );
 number_tests++;
 printf( "\n" );
 
 /* create_relation test */
 printf( " create_relation: " );
 result = create_relation_test( assoc, geom, mesh, &pair );
 handle_error_code( result, &number_tests_failed, &number_tests_not_implemented, &number_tests_successful );
 number_tests++;
 printf( "\n" );
 
 /* relate_geom_mesh test */
 printf( " relate_geom_mesh: " );
 result = relate_geom_mesh_test( assoc, geom, mesh, pair );
 handle_error_code( result, &number_tests_failed, &number_tests_not_implemented, &number_tests_successful );
 number_tests++;
 printf( "\n" );
 
 /* query_relations test */
 printf( " query_relations: " );
 result = query_relations_test( assoc, geom, mesh, pair );
 handle_error_code( result, &number_tests_failed, &number_tests_not_implemented, &number_tests_successful );
 number_tests++;
 printf( "\n" );
 
 /* summary */
 
 printf( "\niRel TEST SUMMARY: \n" );
 printf( " Number Tests: %d\n", number_tests );
 printf( " Number Successful: %d\n", number_tests_successful );
 printf( " Number Not Implemented: %d\n", number_tests_not_implemented );
 printf( " Number Failed: %d\n", number_tests_failed );
 printf( "\n\n" );
 
 iRel_destroy( assoc, &result );
 iMesh_dtor( mesh, &result );
 FBiGeom_dtor( geom, &result );
 
 return number_tests_failed != 0;
}