|
MOAB: Mesh Oriented datABase
(version 5.4.1)
|
|
MOAB: Mesh Oriented datABase
(version 5.4.1)
|
#include "cgm2moab.hpp"#include "moab/ProgOptions.hpp"#include "moab/Core.hpp"#include "moab/Range.hpp"#include "moab/CartVect.hpp"#include "MBTagConventions.hpp"#include "moab/GeomQueryTool.hpp"#include "moab/GeomTopoTool.hpp"#include <sstream>#include <iomanip>#include <cstdlib>#include <algorithm>#include <cstdio>
Include dependency graph for cgm2moab.cpp:Go to the source code of this file.
Defines | |
| #define | OPTION_APPEND(X) |
Functions | |
| void | chkerr (Interface *mbi, ErrorCode code, int line, const char *file) |
| void | chkerr (Interface &mbi, ErrorCode code, int line, const char *file) |
| void | chkerr (GeomTopoTool >t, ErrorCode code, int line, const char *file) |
| static ErrorCode | get_signed_volume (Interface *MBI, const EntityHandle surf_set, const CartVect &offset, double &signed_volume) |
| static ErrorCode | replace_surface (Interface *mbi, EntityHandle old_surf, EntityHandle old_file_set, EntityHandle new_surf, const Tag &senseTag) |
| static ErrorCode | merge_input_surfs (Interface *mbi, const EntityHandle old_file_set, const EntityHandle new_file_set, const Tag &idTag, const Tag &dimTag, const Tag &senseTag) |
| int | main (int argc, char *argv[]) |
Variables | |
| bool | verbose = false |
| program-wide verbose output flag | |
| #define OPTION_APPEND | ( | X | ) |
{ \
if( options.length() ) options += ";"; \
options += ( X ); \
}
Referenced by main(), and read_file().
Generic halt-on-error error checking
Definition at line 21 of file cgm2moab.cpp.
References fname, moab::Interface::get_error_string(), moab::Interface::get_last_error(), and MB_SUCCESS.
Referenced by chkerr().
{
if( code != MB_SUCCESS )
{
std::cerr << "Error: " << mbi->get_error_string( code ) << " (" << code << ")" << std::endl;
std::string message;
if( MB_SUCCESS == mbi->get_last_error( message ) && !message.empty() )
std::cerr << "Error message: " << message << std::endl;
std::string fname = file;
size_t slash = fname.find_last_of( '/' );
if( slash != fname.npos )
{
fname = fname.substr( slash + 1 );
}
std::cerr << "At " << fname << " line: " << line << std::endl;
std::exit( EXIT_FAILURE );
}
}
Definition at line 41 of file cgm2moab.cpp.
References chkerr().
{
chkerr( &mbi, code, line, file );
}
| void chkerr | ( | GeomTopoTool & | gtt, |
| ErrorCode | code, | ||
| int | line, | ||
| const char * | file | ||
| ) |
Definition at line 46 of file cgm2moab.cpp.
References chkerr(), and moab::GeomTopoTool::get_moab_instance().
{
chkerr( gtt.get_moab_instance(), code, line, file );
}
| static ErrorCode get_signed_volume | ( | Interface * | MBI, |
| const EntityHandle | surf_set, | ||
| const CartVect & | offset, | ||
| double & | signed_volume | ||
| ) | [static] |
Estimate the volume of the surface (actually multiplied by a factor of six). See DagMC::measure_volume, from which this code is borrowed, for algorithmic details. For our purpose, all that matters is the signedness.
| offset | Offset to apply to surface to avoid a zero result. |
Definition at line 58 of file cgm2moab.cpp.
References moab::Range::begin(), moab::Range::end(), ErrorCode, moab::Interface::get_connectivity(), moab::Interface::get_coords(), moab::Interface::get_entities_by_type(), MB_INVALID_SIZE, MB_SUCCESS, and MBTRI.
Referenced by replace_surface().
{
ErrorCode rval;
Range tris;
rval = MBI->get_entities_by_type( surf_set, MBTRI, tris );
if( MB_SUCCESS != rval ) return rval;
signed_volume = 0.0;
const EntityHandle* conn;
int len;
CartVect coords[3];
for( Range::iterator j = tris.begin(); j != tris.end(); ++j )
{
rval = MBI->get_connectivity( *j, conn, len, true );
if( MB_SUCCESS != rval ) return rval;
if( 3 != len ) return MB_INVALID_SIZE;
rval = MBI->get_coords( conn, 3, coords[0].array() );
if( MB_SUCCESS != rval ) return rval;
// Apply offset to avoid calculating 0 for cases when the origin is in the
// plane of the surface.
for( unsigned int k = 0; k < 3; ++k )
{
coords[k] += offset;
}
coords[1] -= coords[0];
coords[2] -= coords[0];
signed_volume += ( coords[0] % ( coords[1] * coords[2] ) );
}
return MB_SUCCESS;
}
| int main | ( | int | argc, |
| char * | argv[] | ||
| ) |
Definition at line 260 of file cgm2moab.cpp.
References ProgOptions::add_cancel_opt, ProgOptions::addOpt(), ProgOptions::addOptionHelpHeading(), ProgOptions::addRequiredArg(), CHECKERR, moab::GeomTopoTool::construct_obb_trees(), moab::Core::create_meshset(), ProgOptions::error(), ErrorCode, moab::GeomTopoTool::find_geomsets(), GEOM_DIMENSION_TAG_NAME, ProgOptions::getOpt(), ProgOptions::getOptAllArgs(), moab::Core::globalId_tag(), idTag, input_file, moab::Core::load_file(), MB_TAG_CREAT, MB_TAG_SPARSE, MB_TYPE_HANDLE, MB_TYPE_INTEGER, MB_UNHANDLED_OPTION, merge_input_surfs(), ProgOptions::numOptSet(), OPTION_APPEND, ProgOptions::parseCommandLine(), moab::Core::tag_get_handle(), verbose, and moab::Core::write_file().
{
ProgOptions po( "cgm2moab: a tool for preprocessing CAD and mesh files for analysis" );
std::string input_file;
std::string output_file = "dagmc_preproc_out.h5m";
int grid = 50;
po.addOpt< void >( ",v", "Verbose output", &verbose );
po.addOpt< std::string >( "outmesh,o", "Specify output file name (default " + output_file + ")", &output_file );
po.addOpt< void >( "no-outmesh,", "Do not write an output mesh" );
po.addOpt< std::string >( ",m", "Specify alternate input mesh to override surfaces in input_file" );
po.addOpt< std::string >( "obb-vis,O", "Specify obb visualization output file (default none)" );
po.addOpt< int >( "obb-vis-divs", "Resolution of obb visualization grid (default 50)", &grid );
po.addOpt< void >( "obb-stats,S", "Print obb statistics. With -v, print verbose statistics." );
po.addOpt< std::vector< int > >( "vols,V",
"Specify a set of volumes (applies to --obb_vis and --obb_stats, default all)" );
po.addOptionHelpHeading( "Options for loading CAD files" );
po.addOpt< double >( "ftol,f", "Faceting distance tolerance", po.add_cancel_opt );
po.addOpt< double >( "ltol,l", "Faceting edge length tolerance", po.add_cancel_opt );
po.addOpt< int >( "atol,a", "Faceting normal angle tolerance (degrees)", po.add_cancel_opt );
po.addOpt< void >( "all-warnings", "Verbose warnings about attributes and curve tolerances" );
po.addOpt< void >( "no-attribs", "Do not actuate CGM attributes" );
po.addOpt< void >( "fatal_curves", "Fatal Error when curves cannot be faceted" );
po.addRequiredArg< std::string >( "input_file", "Path to input file for preprocessing", &input_file );
po.parseCommandLine( argc, argv );
/* Check that user has asked for at least one useful thing to be done */
bool obb_task = po.numOptSet( "obb-vis" ) || po.numOptSet( "obb-stats" );
if( po.numOptSet( "no-outmesh" ) && !obb_task )
{
po.error( "Nothing to do. Please specify an OBB-related option, or remove --no_outmesh." );
}
/* Load input file, with CAD processing options, if specified */
std::string options;
#define OPTION_APPEND( X ) \
{ \
if( options.length() ) options += ";"; \
options += ( X ); \
}
if( po.numOptSet( "no-attribs" ) )
{
OPTION_APPEND( "CGM_ATTRIBS=no" );
}
if( po.numOptSet( "fatal_curves" ) )
{
OPTION_APPEND( "FATAL_ON_CURVES" );
}
if( po.numOptSet( "all-warnings" ) )
{
OPTION_APPEND( "VERBOSE_CGM_WARNINGS" );
}
// This is more roundabout than necessary, but we don't want *any* of the CGM-specific options
// to appear in the option string unless they were explicitly requested
double tol;
static const int tol_prec = 12;
if( po.getOpt( "ftol", &tol ) )
{
std::stringstream s;
s << "FACET_DISTANCE_TOLERANCE=" << std::setprecision( tol_prec ) << tol;
OPTION_APPEND( s.str() );
}
if( po.getOpt( "ltol", &tol ) )
{
std::stringstream s;
s << "MAX_FACET_EDGE_LENGTH=" << std::setprecision( tol_prec ) << tol;
OPTION_APPEND( s.str() );
}
int atol;
if( po.getOpt( "atol", &atol ) )
{
std::stringstream s;
s << "FACET_NORMAL_TOLERANCE=" << atol;
OPTION_APPEND( s.str() );
}
#undef OPTION_APPEND
/* Load main input file */
if( verbose )
{
std::cout << "Loading file " << input_file << std::endl;
if( options.length() ) std::cout << "Option string: " << options << std::endl;
}
EntityHandle input_file_set;
ErrorCode ret;
Core mbi;
ret = mbi.create_meshset( 0, input_file_set );
CHECKERR( mbi, ret );
ret = mbi.load_file( input_file.c_str(), &input_file_set, options.c_str() );
if( ret == MB_UNHANDLED_OPTION )
{
std::cerr << "Warning: unhandled option while loading input_file, will proceed anyway" << std::endl;
}
else
{
CHECKERR( mbi, ret );
}
/* Iterate through any -m alternate mesh files and replace surfaces */
std::vector< std::string > m_list;
po.getOptAllArgs( ",m", m_list );
if( m_list.size() > 0 )
{
if( obb_task )
{
std::cerr << "Warning: using obb features in conjunction with -m may not work correctly!" << std::endl;
}
// Create tags
Tag dimTag, idTag, senseTag;
ret = mbi.tag_get_handle( GEOM_DIMENSION_TAG_NAME, 1, MB_TYPE_INTEGER, dimTag, MB_TAG_SPARSE | MB_TAG_CREAT );
CHECKERR( mbi, ret );
idTag = mbi.globalId_tag();
ret = mbi.tag_get_handle( "GEOM_SENSE_2", 2, MB_TYPE_HANDLE, senseTag, MB_TAG_SPARSE | MB_TAG_CREAT );
CHECKERR( mbi, ret );
for( std::vector< std::string >::iterator i = m_list.begin(); i != m_list.end(); ++i )
{
std::cout << "Loading alternate mesh file " << *i << std::endl;
EntityHandle alt_file_set;
ret = mbi.create_meshset( 0, alt_file_set );
CHECKERR( mbi, ret );
ret = mbi.load_file( ( *i ).c_str(), &alt_file_set );
CHECKERR( mbi, ret );
if( verbose ) std::cout << "Merging input surfaces..." << std::flush;
ret = merge_input_surfs( &mbi, input_file_set, alt_file_set, idTag, dimTag, senseTag );
CHECKERR( mbi, ret );
if( verbose ) std::cout << "done." << std::endl;
}
}
/* Write output file */
if( !po.numOptSet( "no-outmesh" ) )
{
if( verbose )
{
std::cout << "Writing " << output_file << std::endl;
}
ret = mbi.write_file( output_file.c_str(), NULL, NULL, &input_file_set, 1 );
CHECKERR( mbi, ret );
}
/* OBB statistics and visualization */
if( obb_task )
{
if( verbose )
{
std::cout << "Loading data into GeomTopoTool" << std::endl;
}
GeomTopoTool* gtt = new GeomTopoTool( &mbi, false );
ret = gtt->find_geomsets();
CHECKERR( *gtt, ret );
ret = gtt->construct_obb_trees();
CHECKERR( *gtt, ret );
}
return 0;
}
| static ErrorCode merge_input_surfs | ( | Interface * | mbi, |
| const EntityHandle | old_file_set, | ||
| const EntityHandle | new_file_set, | ||
| const Tag & | idTag, | ||
| const Tag & | dimTag, | ||
| const Tag & | senseTag | ||
| ) | [static] |
Given an "old" file and a "new" file, replace the facets in any surface of the old file with facets from the new file.
Definition at line 199 of file cgm2moab.cpp.
References moab::Range::begin(), moab::Range::end(), ErrorCode, moab::Interface::get_entities_by_type_and_tag(), MB_SUCCESS, MBENTITYSET, replace_surface(), moab::Range::size(), moab::Interface::tag_get_data(), and verbose.
Referenced by main().
{
ErrorCode rval;
const int two = 2;
const Tag tags[2] = { dimTag, idTag };
const void* tag_vals[2] = { &two, NULL };
Range old_surfs;
rval = mbi->get_entities_by_type_and_tag( old_file_set, MBENTITYSET, &dimTag, tag_vals, 1, old_surfs );
if( MB_SUCCESS != rval ) return rval;
int count = 0;
for( Range::iterator i = old_surfs.begin(); i != old_surfs.end(); ++i )
{
EntityHandle old_surf = *i;
int surf_id;
rval = mbi->tag_get_data( idTag, &old_surf, 1, &surf_id );
if( MB_SUCCESS != rval ) return rval;
Range new_surf_range;
tag_vals[1] = &surf_id;
rval = mbi->get_entities_by_type_and_tag( new_file_set, MBENTITYSET, tags, tag_vals, 2, new_surf_range );
if( MB_SUCCESS != rval ) return rval;
if( new_surf_range.size() != 1 )
{
if( new_surf_range.size() > 1 )
{
std::cerr << "Warning: surface " << surf_id << " has more than one representation in new file"
<< std::endl;
}
continue;
}
// Now we have found a surf in new_file_set to replace an old surf
EntityHandle new_surf = new_surf_range[0];
// TODO: check for quads and convert to triangles
if( verbose )
{
std::cout << "Surface " << surf_id << ": " << std::flush;
}
rval = replace_surface( mbi, old_surf, old_file_set, new_surf, senseTag );
if( MB_SUCCESS != rval ) return rval;
count++;
}
std::cout << "Replaced " << count << " surface" << ( count == 1 ? "." : "s." ) << std::endl;
return MB_SUCCESS;
}
| static ErrorCode replace_surface | ( | Interface * | mbi, |
| EntityHandle | old_surf, | ||
| EntityHandle | old_file_set, | ||
| EntityHandle | new_surf, | ||
| const Tag & | senseTag | ||
| ) | [static] |
Replace the triangles in an old surface with those in a new surface, ensuring that their surfaces senses match before the replacement
Definition at line 98 of file cgm2moab.cpp.
References moab::Interface::add_entities(), moab::Interface::delete_entities(), ErrorCode, moab::Interface::get_entities_by_type(), get_signed_volume(), MB_SUCCESS, MBTRI, moab::Interface::remove_entities(), moab::Range::size(), swap(), moab::Interface::tag_get_data(), moab::Interface::tag_set_data(), and verbose.
Referenced by merge_input_surfs().
{
ErrorCode rval;
// Get the signed volume for each surface representation. If a volume comes
// back as zero, it's probably because a planar surface passes through the
// origin. In that case, try applying random offsets until reasonable
// values are returned.
CartVect offset; // start with no offset
const double min_vol = 0.1; // try again if abs(vol) < this value
double old_vol = 0, new_vol = 0;
bool success = false;
int num_attempts = 100;
while( num_attempts-- > 0 )
{
rval = get_signed_volume( mbi, old_surf, offset, old_vol );
if( MB_SUCCESS != rval ) return rval;
rval = get_signed_volume( mbi, new_surf, offset, new_vol );
if( MB_SUCCESS != rval ) return rval;
if( std::fabs( old_vol ) >= min_vol && std::fabs( new_vol ) >= min_vol )
{
success = true;
break;
}
// haven't succeeded yet: randomize the offset vector
const int max_random = 10;
for( int i = 0; i < 3; ++i )
{
offset[i] = std::rand() % max_random;
}
}
if( !success )
{
std::cerr << "Error: could not calculate a surface volume" << std::endl;
return MB_FAILURE;
}
// If the sign is different, reverse the old surf senses so that both
// representations have the same signed volume.
if( ( old_vol < 0 && new_vol > 0 ) || ( old_vol > 0 && new_vol < 0 ) )
{
EntityHandle old_surf_volumes[2];
rval = mbi->tag_get_data( senseTag, &old_surf, 1, old_surf_volumes );
if( MB_SUCCESS != rval ) return rval;
std::swap( old_surf_volumes[0], old_surf_volumes[1] );
rval = mbi->tag_set_data( senseTag, &old_surf, 1, old_surf_volumes );
if( MB_SUCCESS != rval ) return rval;
}
int num_old_tris, num_new_tris;
// Remove the tris from the old surf. Also remove them from the
// old_file_set because it is not TRACKING.
Range old_tris;
rval = mbi->get_entities_by_type( old_surf, MBTRI, old_tris );
if( MB_SUCCESS != rval ) return rval;
num_old_tris = old_tris.size();
rval = mbi->remove_entities( old_surf, old_tris );
if( MB_SUCCESS != rval ) return rval;
rval = mbi->remove_entities( old_file_set, old_tris );
if( MB_SUCCESS != rval ) return rval;
rval = mbi->delete_entities( old_tris );
if( MB_SUCCESS != rval ) return rval;
// Add the new_surf's triangles to the old_surf
Range new_tris;
rval = mbi->get_entities_by_type( new_surf, MBTRI, new_tris );
if( MB_SUCCESS != rval ) return rval;
num_new_tris = new_tris.size();
rval = mbi->add_entities( old_surf, new_tris );
if( MB_SUCCESS != rval ) return rval;
if( verbose )
{
std::cout << num_old_tris << " tris -> " << num_new_tris << " tris" << std::endl;
}
return MB_SUCCESS;
}
| bool verbose = false |
program-wide verbose output flag
Definition at line 19 of file cgm2moab.cpp.
1.7.6.1