copyPartition.cpp File Reference

Example demonstrating partition information copying between mesh files. More...

#include "moab/ProgOptions.hpp"
#include "moab/Core.hpp"
#include <cmath>
#include <sstream>

Include dependency graph for copyPartition.cpp:

Go to the source code of this file.

Functions
int	main (int argc, char *argv[])

Detailed Description

Example demonstrating partition information copying between mesh files.

This example shows how to:

• Load partition information from point cloud files
• Copy partition data to PG2 mesh files
• Match entities between different mesh files using global IDs
• Create parallel partition sets for visualization
• Handle partition mapping for E3SM climate model meshes
• Write partition-enhanced mesh files for VisIt visualization

This tool is useful for climate model visualization where partition information needs to be transferred between different mesh representations (point cloud to structured mesh).

Author

MOAB Development Team

Date

2024

this tool will take an existing h5m phys grid partition file (point cloud) and copy the partition information on a pg2 mesh file, for better viewing with VisIt

example of usage: ./copyPartition -p AtmPhys.h5m -g wholeATM_PG2.h5m -o atm_PG2_part.h5m

the *PG2" style atm mesh is available in E3SM only for pg2 runs, something like –res ne30pg2_r05_oECv3_ICG –compset A_WCYCL1850S_CMIP6 or –res ne4pg2_ne4pg2 –compset FC5AV1C-L

Parameters

argc	Number of command line arguments
argv	Command line arguments array

Returns

0 on success, 1 on failure

Definition in file copyPartition.cpp.

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 44 of file copyPartition.cpp.

{
 
    ProgOptions opts;
 
    std::string physfile, pg2file, outfile;
 
    opts.addOpt< std::string >( "physgridfile,p", "phys grid filename", &physfile );
    opts.addOpt< std::string >( "pg2file,g", "pg2 mesh file", &pg2file );
    opts.addOpt< std::string >( "output,o", "output mesh filename", &outfile );
 
    opts.parseCommandLine( argc, argv );
 
    std::cout << "phys grid cloud file: " << physfile << "\n";
    std::cout << "pg2 mesh file: " << pg2file << "\n";
    std::cout << "output file: " << outfile << "\n";
 
    if( physfile.empty() )
    {
        opts.printHelp();
        return 0;
    }
    ErrorCode rval;
    Core* mb = new Core();
 
    MB_CHK_SET_ERR( mb->load_file( physfile.c_str() ), "can't load phys grid file" );
 
    Core* mb2 = new Core();
    MB_CHK_SET_ERR( mb2->load_file( pg2file.c_str() ), "can't load pg2 mesh file" );
 
    Tag globalIDTag1 = mb->globalId_tag();
    Tag parti;
    MB_CHK_SET_ERR( mb->tag_get_handle( "partition", parti ), "can't get partition tag phys grid mesh " );
 
    Tag globalIDTag2 = mb2->globalId_tag();
 
    Range verts1;
    MB_CHK_SET_ERR( mb->get_entities_by_dimension( 0, 0, verts1 ), "can't get vertices " );
 
    std::vector< int > partValues;
    partValues.resize( verts1.size() );
    MB_CHK_SET_ERR( mb->tag_get_data( parti, verts1, &partValues[0] ), "can't get parts values on vertices " );
 
    Range cells;
    MB_CHK_SET_ERR( mb2->get_entities_by_dimension( 0, 2, cells ), "can't get 2d cells " );
    std::vector< int > globalIdsCells;
    globalIdsCells.resize( cells.size() );
    MB_CHK_SET_ERR( mb2->tag_get_data( globalIDTag2, cells, &globalIdsCells[0] ), "can't get global ids cells " );
 
    std::vector< int > globalIdsVerts;
    globalIdsVerts.resize( verts1.size() );
    MB_CHK_SET_ERR( mb->tag_get_data( globalIDTag1, verts1, &globalIdsVerts[0] ), "can't get global ids cells " );
 
    Tag partTag;
    MB_CHK_SET_ERR( mb2->tag_get_handle( "PARALLEL_PARTITION", partTag ), "can't partition tag " );
 
    Range sets;
    MB_CHK_ERR( mb2->get_entities_by_type_and_tag( 0, MBENTITYSET, &partTag, NULL, 1, sets ) );
 
    std::vector< int > setValues;
    setValues.resize( sets.size() );
    MB_CHK_ERR( mb2->tag_get_data( partTag, sets, &setValues[0] ) );
 
    std::map< int, EntityHandle > valToSet;
    int i = 0;
    for( Range::iterator st = sets.begin(); st != sets.end(); ++st, i++ )
    {
        valToSet[setValues[i]] = *st;
    }
    // now, every cell will be put into one set, by looking at the global id of cell
 
    std::map< int, EntityHandle > gidToCell;
    i = 0;
    for( Range::iterator it = cells.begin(); it != cells.end(); ++it, i++ )
    {
        gidToCell[globalIdsCells[i]] = *it;
    }
    // empty all sets
    MB_CHK_ERR( mb2->clear_meshset( sets ) );
 
    // look now at parti values for vertices, and their global ids
    for( i = 0; i < (int)verts1.size(); i++ )
    {
        int part          = partValues[i];
        int gid           = globalIdsVerts[i];
        EntityHandle set1 = valToSet[part];
        EntityHandle cell = gidToCell[gid];
        MB_CHK_ERR( mb2->add_entities( set1, &cell, 1 ) );
    }
 
    MB_CHK_SET_ERR( mb2->write_file( outfile.c_str() ), "can't write file" );
 
    delete mb;
    delete mb2;
 
    return 0;
}

References moab::Core::add_entities(), ProgOptions::addOpt(), moab::Range::begin(), moab::Core::clear_meshset(), moab::Range::end(), ErrorCode, moab::Core::get_entities_by_dimension(), moab::Core::get_entities_by_type_and_tag(), moab::Core::globalId_tag(), moab::Core::load_file(), mb, MB_CHK_ERR, MB_CHK_SET_ERR, MBENTITYSET, ProgOptions::parseCommandLine(), ProgOptions::printHelp(), moab::Range::size(), moab::Core::tag_get_data(), moab::Core::tag_get_handle(), and moab::Core::write_file().