dagmc_simple_test.cpp

Go to the documentation of this file.

#include <iostream>
#include "moab/Interface.hpp"
#ifndef IS_BUILDING_MB
#define IS_BUILDING_MB
#endif
#include "TestUtil.hpp"
#include "Internals.hpp"
#include "moab/Core.hpp"
 
#include "DagMC.hpp"
 
using namespace moab;
 
using moab::DagMC;
 
DagMC* DAG;
 
#define CHKERR( A ) \
 do \
 { \
 if( MB_SUCCESS != ( A ) ) \
 { \
 std::cerr << "Failure (error code " << ( A ) << ") at " __FILE__ ":" << __LINE__ << std::endl; \
 return A; \
 } \
 } while( false )
 
const std::string input_file = TestDir + "unittest/test_geom.h5m";
 
void dagmc_load_file()
{
 ErrorCode rval = DAG->load_file( input_file.c_str() ); // open the Dag file
 CHECK_ERR( rval );
}
 
void dagmc_load_file_dagmc()
{
 /* 1 - Test with external moab, load file in DAGMC*/
 // make new moab core
 Core* mbi = new moab::Core();
 // make new dagmc into that moab
 DagMC* dagmc = new moab::DagMC( mbi );
 
 ErrorCode rval;
 // load a file
 rval = dagmc->load_file( input_file.c_str() );CHECK_ERR( rval );
 
 // delete dagmc
 delete dagmc;
 delete mbi;
}
 
void dagmc_load_file_dagmc_via_moab()
{
 /* 2 - Test with external moab, load file in MOAB*/
 // load the file into moab rather than dagmc
 ErrorCode rval;
 
 moab::Core* mbi = new moab::Core();
 rval = mbi->load_file( input_file.c_str() );CHECK_ERR( rval );
 moab::DagMC* dagmc = new moab::DagMC( mbi );
 rval = dagmc->load_existing_contents();CHECK_ERR( rval );
 
 // delete dagmc;
 delete dagmc;
 delete mbi;
}
 
void dagmc_load_file_dagmc_internal()
{
 /* 3 - Test with internal moab, load file in DAG*/
 // make new dagmc into that moab
 ErrorCode rval;
 
 moab::DagMC* dagmc = new moab::DagMC();
 // load a file
 rval = dagmc->load_file( input_file.c_str() );CHECK_ERR( rval );
 delete dagmc;
}
 
void dagmc_load_file_dagmc_build_obb()
{
 /* 1 - Test with external moab, load file in DAGMC*/
 // make new moab core
 ErrorCode rval;
 
 moab::Core* mbi = new moab::Core();
 // make new dagmc into that moab
 DagMC* dagmc = new moab::DagMC( mbi );
 
 // load a file
 rval = dagmc->load_file( input_file.c_str() );CHECK_ERR( rval );
 rval = dagmc->init_OBBTree();CHECK_ERR( rval );
 // delete dagmc
 delete dagmc;
 delete mbi;
}
 
void dagmc_load_file_dagmc_via_moab_build_obb()
{
 /* 2 - Test with external moab, load file in MOAB*/
 // load the file into moab rather than dagmc
 ErrorCode rval;
 
 moab::Core* mbi = new moab::Core();
 rval = mbi->load_file( input_file.c_str() );CHECK_ERR( rval );
 moab::DagMC* dagmc = new moab::DagMC( mbi );
 rval = dagmc->load_existing_contents();CHECK_ERR( rval );
 rval = dagmc->init_OBBTree();CHECK_ERR( rval );
 
 // delete dagmc;
 delete dagmc;
 delete mbi;
}
 
void dagmc_load_file_dagmc_internal_build_obb()
{
 /* 3 - Test with internal moab, load file in DAG*/
 // make new dagmc into that moab
 ErrorCode rval;
 
 moab::DagMC* dagmc = new moab::DagMC();
 // load a file
 rval = dagmc->load_file( input_file.c_str() );CHECK_ERR( rval );
 rval = dagmc->init_OBBTree();CHECK_ERR( rval );
 delete dagmc;
}
 
void dagmc_test_obb_retreval()
{
 // make new dagmc
 std::cout << "test_obb_retreval" << std::endl;
 
 DagMC* dagmc = new moab::DagMC();
 
 ErrorCode rval;
 // load a file
 rval = dagmc->load_file( input_file.c_str() );CHECK_ERR( rval );
 rval = dagmc->init_OBBTree();CHECK_ERR( rval );
 
 // write the file
 rval = dagmc->write_mesh( "fcad", 4 );
 
 // now remove the dagmc instance a
 delete dagmc;
 
 dagmc = new moab::DagMC();
 rval = dagmc->load_file( "fcad" );CHECK_ERR( rval );
 rval = dagmc->init_OBBTree();CHECK_ERR( rval );
 
 // delete the fcad file
 remove( "fcad" );
 delete dagmc;
}
 
void dagmc_build_obb()
{
 ErrorCode rval = DAG->init_OBBTree();CHECK_ERR( rval );
}
 
void dagmc_num_vols()
{
 int expect_num_vols = 2;
 int num_vols = DAG->num_entities( 3 );
 CHECK_EQUAL( expect_num_vols, num_vols );
}
 
void dagmc_entity_handle()
{
 /*
 int num_vols = DAG->num_entities(3);
 EntityHandle vol_h;
 for (int i = 0; i < num_vols; i++)
 vol_h = DAG->entity_by_index(3, i);
 */
 // EntityHandle expect_vol_h = 12682136550675316765;
 // CHECK_EQUAL(expect_vol_h, vol_h);
}
 
void dagmc_point_in()
{
 int result = 0;
 int expect_result = 1;
 int vol_idx = 1;
 double xyz[3] = { 0.0, 0.0, 0.0 };
 EntityHandle vol_h = DAG->entity_by_index( 3, vol_idx );
 ErrorCode rval = DAG->point_in_volume( vol_h, xyz, result );CHECK_ERR( rval );
 CHECK_EQUAL( expect_result, result );
}
 
void dagmc_test_obb_retreval_rayfire()
{
 // make new dagmc
 std::cout << "test_obb_retreval and ray_fire" << std::endl;
 
 DagMC* dagmc = new moab::DagMC();
 
 ErrorCode rval;
 // load a file
 rval = dagmc->load_file( input_file.c_str() );CHECK_ERR( rval );
 rval = dagmc->init_OBBTree();CHECK_ERR( rval );
 
 // write the file
 rval = dagmc->write_mesh( "fcad", 4 );
 
 // now remove the dagmc instance a
 delete dagmc;
 
 // now create new DAGMC
 dagmc = new moab::DagMC();
 rval = dagmc->load_file( "fcad" );CHECK_ERR( rval );
 rval = dagmc->init_OBBTree();CHECK_ERR( rval );
 
 // delete the fcad file
 remove( "fcad" );
 
 // now perform full ray fire
 double eps = 1.e-6;
 int vol_idx = 1;
 // note model is cube of side 10, centred at 0,0,0, so ray fire along
 // any unit direction should be exactly 5.0
 double xyz[3] = { 0.0, 0.0, 0.0 };
 double dir[3] = { 0.0, 0.0, 1.0 };
 EntityHandle next_surf;
 double next_surf_dist;
 double expect_next_surf_dist = 5.0;
 EntityHandle vol_h = DAG->entity_by_index( 3, vol_idx );
 
 rval = DAG->ray_fire( vol_h, xyz, dir, next_surf, next_surf_dist );CHECK_ERR( rval );
 CHECK_REAL_EQUAL( expect_next_surf_dist, next_surf_dist, eps );
 delete dagmc;
}
 
void dagmc_rayfire()
{
 const double eps = 1e-6; // epsilon for test, faceting tol?
 
 int vol_idx = 1;
 // note model is cube of side 10, centred at 0,0,0, so ray fire along
 // any unit direction should be exactly 5.0
 double xyz[3] = { 0.0, 0.0, 0.0 };
 double dir[3] = { 0.0, 0.0, 1.0 };
 EntityHandle next_surf;
 double next_surf_dist;
 double expect_next_surf_dist = 5.0;
 EntityHandle vol_h = DAG->entity_by_index( 3, vol_idx );
 
 ErrorCode rval = DAG->ray_fire( vol_h, xyz, dir, next_surf, next_surf_dist );CHECK_ERR( rval );
 CHECK_REAL_EQUAL( expect_next_surf_dist, next_surf_dist, eps );
}
 
void dagmc_closest_to()
{
 const double eps = 1e-6; // epsilon for test, faceting tol?
 
 int vol_idx = 1;
 // note model is cube of side 10, centred at 0,0,0, so ray fire along
 // any unit direction should be exactly 5.0
 double xyz[3] = { -6.0, 0.0, 0.0 };
 double distance; // distance from point to nearest surface
 double expect_distance = 1.0;
 EntityHandle vol_h = DAG->entity_by_index( 3, vol_idx );
 
 ErrorCode rval = DAG->closest_to_location( vol_h, xyz, distance );CHECK_ERR( rval );
 // distance should be 1.0 cm
 CHECK_REAL_EQUAL( expect_distance, distance, eps );
}
 
void dagmc_test_boundary()
{
 int vol_idx = 1;
 EntityHandle vol_h = DAG->entity_by_index( 3, vol_idx );
 int surf_idx = 1;
 EntityHandle surf_h = DAG->entity_by_index( 2, surf_idx );
 
 double xyz[3] = { 0.0, 0.0, 5.0 };
 double dir[3] = { 0.0, 0.0, 1.0 };
 int result;
 int expect_result = 0;
 
 ErrorCode rval = DAG->test_volume_boundary( vol_h, surf_h, xyz, dir, result );CHECK_ERR( rval );
 // check ray leaving volume
 CHECK_EQUAL( expect_result, result );
}
 
int main( int /* argc */, char** /* argv */ )
{
 int result = 0;
 
 DAG = new moab::DagMC();
 
 result += RUN_TEST( dagmc_load_file ); // test ray fire
 result += RUN_TEST( dagmc_build_obb ); // build the obb
 result += RUN_TEST( dagmc_num_vols ); // make sure the num of vols correct
 result += RUN_TEST( dagmc_load_file_dagmc ); //
 result += RUN_TEST( dagmc_load_file_dagmc_via_moab ); //
 result += RUN_TEST( dagmc_load_file_dagmc_internal ); //
 result += RUN_TEST( dagmc_load_file_dagmc_build_obb ); //
 result += RUN_TEST( dagmc_load_file_dagmc_via_moab_build_obb ); //
 result += RUN_TEST( dagmc_load_file_dagmc_internal_build_obb ); //
 result += RUN_TEST( dagmc_test_obb_retreval ); // check that we are retreving loaded obbs
 result += RUN_TEST( dagmc_test_obb_retreval_rayfire ); // check that we can ray fire on loaded obbs
 result += RUN_TEST( dagmc_point_in ); // check entity by point
 result += RUN_TEST( dagmc_rayfire ); // ensure ray fire distance is correct
 result += RUN_TEST( dagmc_closest_to ); // check the distance to surface nearest point
 result += RUN_TEST( dagmc_test_boundary ); // check particle entering leaving
 
 delete DAG;
 
 return result;
}