You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdepim/indexlib/stringset.cpp

107 lines
3.7 KiB

/* This file is part of indexlib.
* Copyright (C) 2005 Luís Pedro Coelho <luis@luispedro.org>
*
* Indexlib is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License, version 2, as
* published by the Free Software Foundation and available as file
* GPL_V2 which is distributed along with indexlib.
*
* Indexlib 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
*
* In addition, as a special exception, the copyright holders give
* permission to link the code of this program with any edition of
* the TQt library by Trolltech AS, Norway (or with modified versions
* of TQt that use the same license as TQt), and distribute linked
* combinations including the two. You must obey the GNU General
* Public License in all respects for all of the code used other than
* TQt. If you modify this file, you may extend this exception to
* your version of the file, but you are not obligated to do so. If
* you do not wish to do so, delete this exception statement from
* your version.
*/
#include "stringset.h"
#include "path.h"
#include <cstring>
using std::strcmp;
stringset::stringset( std::string name ):
strings_( path_concat( name, "strings-set" ) ),
ordered_( path_concat( name, "ordered-set" ) ),
trie_( path_concat( name, "trie" ) )
{
if ( trie_.empty() ) {
trie_.resize( 256 );
if ( !ordered_.empty() ) {
unsigned char last = 0;
for ( unsigned i = 0; i != ordered_.size(); ++i ) {
unsigned char cur = static_cast<unsigned char>( strings_.get_cstr( ordered_[ i ] )[ 0 ] );
if ( cur != last ) {
trie_[ cur ] = i;
last = cur;
}
}
if ( last < 255 ) trie_[ last + 1 ] = ordered_.size();
}
}
}
void stringset::remove( std::string name ) {
stringarray::remove( path_concat( name, "strings-set" ) );
memvector<stringarray::index_type>::remove( path_concat( name, "ordered-set" ) );
memvector<stringarray::index_type>::remove( path_concat( name, "trie" ) );
}
std::pair<stringset::const_iterator, stringset::const_iterator> stringset::upper_lower( const char* str ) const {
const_iterator first = lower_bound( str );
const_iterator second = lower_bound( ( std::string( str ) + char( 254 ) ).c_str() );
return std::make_pair( first, second );
}
stringset::const_iterator stringset::lower_bound( const char* str ) const {
const_iterator top = begin() + trie_[ ( unsigned )str[ 0 ] ];
const_iterator bottom = begin() + trie_[ ( unsigned )str[ 0 ] + 1 ];
while ( top < bottom ) {
const_iterator middle = top + ( bottom - top ) / 2;
int c = strcmp( *middle, str );
if ( c == 0 ) return middle;
if ( c > 0 ) bottom = middle;
else top = middle + 1;
}
return top;
}
unsigned stringset::order_of( const char* str ) const {
const_iterator where = lower_bound( str );
return where == end() || strcmp( *where, str ) ? unsigned( -1 ) : where.idx_;
}
stringarray::index_type stringset::add( const char* str ) {
const_iterator where = lower_bound( str );
if ( where != end() && !strcmp( *where, str ) ) return where.id();
stringarray::index_type res = strings_.add( str );
ordered_.insert( ordered_.begin() + where.order(), res );
assert( ordered_.size() == strings_.size() );
for ( unsigned next = ( unsigned )str[ 0 ] + 1; next != 256; ++next ) {
++trie_[ next ];
}
return res;
}
void stringset::clear() {
strings_.clear();
ordered_.clear();
}