colselect.cc

// --8<--8<--8<--8<--
//
// Copyright (C) 2006 Smithsonian Astrophysical Observatory
//
// This file is part of suplibxx
//
// suplibxx is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// suplibxx is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the 
//       Free Software Foundation, Inc. 
//       51 Franklin Street, Fifth Floor
//       Boston, MA  02110-1301, USA
//
// -->8-->8-->8-->8--

#include "colselect.h"

namespace suplib {
void colselect( 
  const vector<string>& icolumns,
  const vector<string>& exact_add, 
  const vector<string>& regex_add,
  const vector<string>& exact_del,
  const vector<string>& regex_del,
  vector<string>& ocolumns
  ) throw ( Exception ) {
  
  if ( 0 == (exact_add.size( ) + regex_add.size( ) + 
             exact_del.size( ) + regex_del.size( )) ) {
    ocolumns = icolumns;

  } else {
    set<string> 
      add_set, exact_add_set, regex_add_set, 
      del_set, exact_del_set, regex_del_set, 
      column_set;

    exact_add_set.insert( exact_add.begin( ), exact_add.end( ) );
    regex_add_set.insert( regex_add.begin( ), regex_add.end( ) );
    exact_del_set.insert( exact_del.begin( ), exact_del.end( ) );
    regex_del_set.insert( regex_del.begin( ), regex_del.end( ) );
    column_set.insert(     icolumns.begin( ),  icolumns.end( ) );

    ocolumns.erase( ocolumns.begin( ), ocolumns.end( ) );

    // Add exact matches...
    set_intersection( column_set.begin( ), column_set.end( ),
                      exact_add_set.begin( ), exact_add_set.end( ),
                      inserter( add_set, add_set.begin( ) )
                      );

    // Add regex matches...
    for ( set<string>::iterator pat = regex_add_set.begin( );
          pat != regex_add_set.end( );
          pat++ )
      for ( 
        set<string>::iterator nam = column_set.begin( ); 
        nam != column_set.end( ); 
        nam++ 
        ) 
        if ( match( *nam, *pat ) )
          add_set.insert( *nam );

    // Remove exact matches...
    set_difference( column_set.begin( ), column_set.end( ),
                    exact_del_set.begin( ), exact_del_set.end( ),
                    inserter( del_set, del_set.begin( ) )
                    );
    
    // Remove regex matches...
    for ( set<string>::iterator pat = regex_del_set.begin( );
          pat != regex_del_set.end( );
          pat++ )
      for ( 
        set<string>::iterator nam = del_set.begin( ); 
        nam != del_set.end( ); 
        nam++ 
        ) 
        if ( match( *nam, *pat ) )
          del_set.erase( *nam );

    if ( exact_add.size( ) + regex_add.size( ) ) {
      set<string> ocolumn_set;
      set_intersection( del_set.begin( ), del_set.end( ),
                        add_set.begin( ), add_set.end( ),
                        inserter( ocolumn_set, ocolumn_set.begin( ) ) );
      
      for ( vector<string>::size_type idx = 0; idx < icolumns.size( ); idx++ ) {
        if ( ocolumn_set.end( ) != ocolumn_set.find( icolumns[idx] ) ) {
          ocolumns.push_back( icolumns[idx] );
          
        }
      }
    } else {
      for ( vector<string>::size_type idx = 0; idx != icolumns.size( ); idx++ ) {
        if ( del_set.end( ) != del_set.find( icolumns[idx] ) ) {
          ocolumns.push_back( icolumns[idx] );

        }
      }
    }
  }
}

} // namespace suplib