AddFieldtoPG2.cpp

Go to the documentation of this file.

#include "moab/ProgOptions.hpp"
#include "moab/Core.hpp"
#include <iostream>

using namespace moab;
using namespace std;

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

    ProgOptions opts;

    std::string inputfile, outfile( "out.h5m" ), physgridfile, variable_name;

    opts.addOpt< std::string >( "input,i", "input mesh filename", &inputfile );
    opts.addOpt< std::string >( "output,o", "output mesh filename", &outfile );
    opts.addOpt< std::string >( "phys,p", "phys grid solution filename", &physgridfile );
    opts.addOpt< std::string >( "var,v", "variable to extract and add to output file", &variable_name );

    opts.parseCommandLine( argc, argv );

    if( inputfile.empty() )
    {
        opts.printHelp();
        return 0;
    }
    ErrorCode rval;
    Core* mb = new Core();

    EntityHandle fset1, fset2;
    rval = mb->create_meshset( MESHSET_SET, fset1 );MB_CHK_SET_ERR( rval, "can't create mesh set" );
    rval = mb->load_file( inputfile.c_str(), &fset1 );MB_CHK_SET_ERR( rval, "can't load input file" );

    cout << " opened " << inputfile << " with initial h5m data.\n";

    rval = mb->create_meshset( MESHSET_SET, fset2 );MB_CHK_SET_ERR( rval, "can't create mesh set" );
    rval = mb->load_file( physgridfile.c_str(), &fset2 );MB_CHK_SET_ERR( rval, "can't load phys grid file" );

    Tag tagv;
    rval = mb->tag_get_handle( variable_name.c_str(), tagv );MB_CHK_SET_ERR( rval, "can't get tag handle" );

    Tag gitag = mb->globalId_tag();

    Range verts;  // from phys grid
    rval = mb->get_entities_by_dimension( fset2, 0, verts );MB_CHK_SET_ERR( rval, "can't get vertices" );
    std::vector< int > gids;
    gids.resize( verts.size() );
    rval = mb->tag_get_data( gitag, verts, &gids[0] );MB_CHK_SET_ERR( rval, "can't get gi tag values" );
    std::vector< double > valsTag;
    valsTag.resize( verts.size() );
    rval = mb->tag_get_data( tagv, verts, &valsTag[0] );MB_CHK_SET_ERR( rval, "can't get tag vals" );
    Range cells;

    rval = mb->get_entities_by_dimension( fset1, 2, cells );MB_CHK_SET_ERR( rval, "can't get cells" );

    std::map< int, double > valsByID;
    for( int i = 0; i < (int)gids.size(); i++ )
        valsByID[gids[i]] = valsTag[i];

    // set now cells values
    std::vector< int > cellsIds;
    cellsIds.resize( cells.size() );
    rval = mb->tag_get_data( gitag, cells, &cellsIds[0] );MB_CHK_SET_ERR( rval, "can't get cells ids" );
    for( int i = 0; i < (int)cells.size(); i++ )
    {
        valsTag[i] = valsByID[cellsIds[i]];
    }
    rval = mb->tag_set_data( tagv, cells, &valsTag[0] );MB_CHK_SET_ERR( rval, "can't set  cells tags" );

    rval = mb->write_file( outfile.c_str(), 0, 0, &fset1, 1 );MB_CHK_SET_ERR( rval, "can't write file" );

    return 0;
}