cleanTags.cpp

Go to the documentation of this file.

/*
 * cleanTags.cpp
 * this tool will remove a list of tags from a file, or keep only the tags from a given list
 * will use just the names of the tags
 *
 * example of usage:
 * ./mbcleantags -i input_file.h5m -o outfile.h5m -d list_tags_separated_by':' -k list_tags separated by ':'
 *
 */
#include "moab/MOABConfig.h"

#include "moab/ProgOptions.hpp"
#include "moab/Core.hpp"

using namespace moab;
using namespace std;

vector< string > split( const string& i_str, const string& i_delim )
{
    vector< string > result;

    size_t found      = i_str.find( i_delim );
    size_t startIndex = 0;

    while( found != string::npos )
    {
        result.push_back( string( i_str.begin() + startIndex, i_str.begin() + found ) );
        startIndex = found + i_delim.size();
        found      = i_str.find( i_delim, startIndex );
    }
    if( startIndex != i_str.size() ) result.push_back( string( i_str.begin() + startIndex, i_str.end() ) );
    return result;
}

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

    ProgOptions opts;

    string inputfile, outputfile, deleteTags, keepTags;
    opts.addOpt< string >( "input,i", "input filename ", &inputfile );
    opts.addOpt< string >( "output,o", "output file", &outputfile );

    opts.addOpt< string >( "deleteTags,d", "delete tags ", &deleteTags );
    opts.addOpt< string >( "keepTags,k", "keep tags ", &keepTags );
    opts.parseCommandLine( argc, argv );

    Core core;
    Interface* mb = &core;
    ErrorCode rval;
    rval = mb->load_file( inputfile.c_str() );MB_CHK_ERR( rval );
    vector< Tag > existingTags;
    rval = mb->tag_get_tags( existingTags );MB_CHK_ERR( rval );
    vector< string > tagsToDelete;
    if( !keepTags.empty() )
    {
        vector< string > tagsToKeep = split( keepTags, string( ":" ) );
        for( size_t i = 0; i < existingTags.size(); i++ )
        {
            string tname;
            rval = mb->tag_get_name( existingTags[i], tname );MB_CHK_ERR( rval );
            bool deleteTag = false;
            for( size_t k = 0; k < tagsToKeep.size() && !deleteTag; k++ )
            {
                if( tname.compare( tagsToKeep[k] ) == 0 ) deleteTag = true;
            }
            if( !deleteTag ) tagsToDelete.push_back( tname );
        }
    }
    if( !deleteTags.empty() )
    {
        tagsToDelete = split( deleteTags, string( ":" ) );
    }
    for( size_t i = 0; i < tagsToDelete.size(); i++ )
    {
        Tag tag;
        rval = mb->tag_get_handle( tagsToDelete[i].c_str(), tag );
        if( rval == MB_SUCCESS && tag != NULL )
        {
            rval = mb->tag_delete( tag );MB_CHK_ERR( rval );
        }
    }
    cout << "write file " << outputfile << endl;
    rval = mb->write_file( outputfile.c_str() );MB_CHK_ERR( rval );

    return 0;
}