FixPolys.cpp

Go to the documentation of this file.

/** \brief This test shows how to fix polygons that have duplicated vertices
 *
 * We sometimes use padded vertices option, to reduce the number of data sequences
 * We identify first the polygons that have padded vertices
 * then we set the new ones with reduced number of vertices, but with the same global id tag
 */
 
#include <iostream>
#include <cstdlib>
#include <cstdio>
 
#include "moab/Core.hpp"
#include "moab/Interface.hpp"
#include "moab/Range.hpp"
#include "moab/ProgOptions.hpp"
 
using namespace moab;
using namespace std;
 
#ifdef MOAB_HAVE_HDF5
string test_file_name = string( MESH_DIR ) + string( "/64bricks_512hex_256part.h5m" );
#endif
int main( int argc, char** argv )
{
 ProgOptions opts;
 
 string inputFile = test_file_name;
 opts.addOpt< string >( "inFile,i", "Specify the input file name string ", &inputFile );
 
 string outFile = "out.h5m";
 opts.addOpt< string >( "outFile,o", "Specify the output file name string ", &outFile );
 
 opts.parseCommandLine( argc, argv );
 
 // Instantiate
 Core mb;
 
 ErrorCode rval = mb.load_file( inputFile.c_str() );MB_CHK_SET_ERR( rval, "Error loading file" );
 
 cout << " reading file " << inputFile << "\n";
 
 // Get all 2d elements in the file set
 Range elems;
 rval = mb.get_entities_by_dimension( 0, 2, elems );MB_CHK_SET_ERR( rval, "Error getting 2d elements" );
 
 cout << "number of cells: " << elems.size() << "\n";
 
 Tag gidTag = mb.globalId_tag();
 Range OldCells;
 for( Range::iterator it = elems.begin(); it != elems.end(); ++it )
 {
 EntityHandle cell = *it;
 const EntityHandle* conn;
 int number_nodes;
 rval = mb.get_connectivity( cell, conn, number_nodes );MB_CHK_SET_ERR( rval, "Error getting connectivity" );
 // now check if we have consecutive duplicated vertices, and if so, create a new cell
 std::vector< EntityHandle > new_verts;
 // push to it, if we do not have duplicates
 
 EntityHandle current = conn[0];
 // new_verts.push_back(current);
 for( int i = 1; i <= number_nodes; i++ )
 {
 EntityHandle nextV;
 if( i < number_nodes )
 nextV = conn[i];
 else
 nextV = conn[0]; // first vertex
 if( current != nextV )
 {
 new_verts.push_back( current );
 current = nextV;
 }
 }
 if( number_nodes > (int)new_verts.size() )
 {
 // create a new poly, and put this in a list to be removed
 int gid;
 rval = mb.tag_get_data( gidTag, &cell, 1, &gid );MB_CHK_SET_ERR( rval, "Error getting global id tag" );
 EntityHandle newCell;
 rval = mb.create_element( MBPOLYGON, &new_verts[0], (int)new_verts.size(), newCell );MB_CHK_SET_ERR( rval, "Error creating new polygon " );
 rval = mb.tag_set_data( gidTag, &newCell, 1, &gid );MB_CHK_SET_ERR( rval, "Error setting global id tag" );
 cout << "delete old cell " << cell << " with num_nodes vertices: " << number_nodes
 << " and with global id: " << gid << "\n";
 for( int i = 0; i < number_nodes; i++ )
 {
 cout << " " << conn[i];
 }
 cout << "\n";
 OldCells.insert( cell );
 }
 mb.delete_entities( OldCells );
 }
 rval = mb.write_file( outFile.c_str() );MB_CHK_SET_ERR( rval, "Error writing file" );
 
 return 0;
}