VectorSet.h

Go to the documentation of this file.

/*
===========================================================================

Doom 3 GPL Source Code
Copyright (C) 1999-2011 id Software LLC, a ZeniMax Media company. 

This file is part of the Doom 3 GPL Source Code (?Doom 3 Source Code?).  

Doom 3 Source Code 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 3 of the License, or
(at your option) any later version.

Doom 3 Source Code 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 Doom 3 Source Code.  If not, see <http://www.gnu.org/licenses/>.

In addition, the Doom 3 Source Code is also subject to certain additional terms. You should have received a copy of these additional terms immediately following the terms and conditions of the GNU General Public License which accompanied the Doom 3 Source Code.  If not, please request a copy in writing from id Software at the address below.

If you have questions concerning this license or the applicable additional terms, you may contact in writing id Software LLC, c/o ZeniMax Media Inc., Suite 120, Rockville, Maryland 20850 USA.

===========================================================================
*/

#ifndef __VECTORSET_H__
#define __VECTORSET_H__

/*
===============================================================================

        Vector Set

        Creates a set of vectors without duplicates.

===============================================================================
*/

template< class type, int dimension >
class idVectorSet : public idList<type> {
public:
                                                        idVectorSet( void );
                                                        idVectorSet( const type &mins, const type &maxs, const int boxHashSize, const int initialSize );

                                                        // returns total size of allocated memory
        size_t                                  Allocated( void ) const { return idList<type>::Allocated() + hash.Allocated(); }
                                                        // returns total size of allocated memory including size of type
        size_t                                  Size( void ) const { return sizeof( *this ) + Allocated(); }

        void                                    Init( const type &mins, const type &maxs, const int boxHashSize, const int initialSize );
        void                                    ResizeIndex( const int newSize );
        void                                    Clear( void );

        int                                             FindVector( const type &v, const float epsilon );

private:
        idHashIndex                             hash;
        type                                    mins;
        type                                    maxs;
        int                                             boxHashSize;
        float                                   boxInvSize[dimension];
        float                                   boxHalfSize[dimension];
};

template< class type, int dimension >
ID_INLINE idVectorSet<type,dimension>::idVectorSet( void ) {
        hash.Clear( idMath::IPow( boxHashSize, dimension ), 128 );
        boxHashSize = 16;
        memset( boxInvSize, 0, dimension * sizeof( boxInvSize[0] ) );
        memset( boxHalfSize, 0, dimension * sizeof( boxHalfSize[0] ) );
}

template< class type, int dimension >
ID_INLINE idVectorSet<type,dimension>::idVectorSet( const type &mins, const type &maxs, const int boxHashSize, const int initialSize ) {
        Init( mins, maxs, boxHashSize, initialSize );
}

template< class type, int dimension >
ID_INLINE void idVectorSet<type,dimension>::Init( const type &mins, const type &maxs, const int boxHashSize, const int initialSize ) {
        int i;
        float boxSize;

        idList<type>::AssureSize( initialSize );
        idList<type>::SetNum( 0, false );

        hash.Clear( idMath::IPow( boxHashSize, dimension ), initialSize );

        this->mins = mins;
        this->maxs = maxs;
        this->boxHashSize = boxHashSize;

        for ( i = 0; i < dimension; i++ ) {
                boxSize = ( maxs[i] - mins[i] ) / (float) boxHashSize;
                boxInvSize[i] = 1.0f / boxSize;
                boxHalfSize[i] = boxSize * 0.5f;
        }
}

template< class type, int dimension >
ID_INLINE void idVectorSet<type,dimension>::ResizeIndex( const int newSize ) {
        idList<type>::Resize( newSize );
        hash.ResizeIndex( newSize );
}

template< class type, int dimension >
ID_INLINE void idVectorSet<type,dimension>::Clear( void ) {
        idList<type>::Clear();
        hash.Clear();
}

template< class type, int dimension >
ID_INLINE int idVectorSet<type,dimension>::FindVector( const type &v, const float epsilon ) {
        int i, j, k, hashKey, partialHashKey[dimension];

        for ( i = 0; i < dimension; i++ ) {
                assert( epsilon <= boxHalfSize[i] );
                partialHashKey[i] = (int) ( ( v[i] - mins[i] - boxHalfSize[i] ) * boxInvSize[i] );
        }

        for ( i = 0; i < ( 1 << dimension ); i++ ) {

                hashKey = 0;
                for ( j = 0; j < dimension; j++ ) {
                        hashKey *= boxHashSize;
                        hashKey += partialHashKey[j] + ( ( i >> j ) & 1 );
                }

                for ( j = hash.First( hashKey ); j >= 0; j = hash.Next( j ) ) {
                        const type &lv = (*this)[j];
                        for ( k = 0; k < dimension; k++ ) {
                                if ( idMath::Fabs( lv[k] - v[k] ) > epsilon ) {
                                        break;
                                }
                        }
                        if ( k >= dimension ) {
                                return j;
                        }
                }
        }

        hashKey = 0;
        for ( i = 0; i < dimension; i++ ) {
                hashKey *= boxHashSize;
                hashKey += (int) ( ( v[i] - mins[i] ) * boxInvSize[i] );
        }

        hash.Add( hashKey, idList<type>::Num() );
        this->Append( v );
        return idList<type>::Num()-1;
}


/*
===============================================================================

        Vector Subset

        Creates a subset without duplicates from an existing list with vectors.

===============================================================================
*/

template< class type, int dimension >
class idVectorSubset {
public:
                                                        idVectorSubset( void );
                                                        idVectorSubset( const type &mins, const type &maxs, const int boxHashSize, const int initialSize );

                                                        // returns total size of allocated memory
        size_t                                  Allocated( void ) const { return idList<type>::Allocated() + hash.Allocated(); }
                                                        // returns total size of allocated memory including size of type
        size_t                                  Size( void ) const { return sizeof( *this ) + Allocated(); }

        void                                    Init( const type &mins, const type &maxs, const int boxHashSize, const int initialSize );
        void                                    Clear( void );

                                                        // returns either vectorNum or an index to a previously found vector
        int                                             FindVector( const type *vectorList, const int vectorNum, const float epsilon );

private:
        idHashIndex                             hash;
        type                                    mins;
        type                                    maxs;
        int                                             boxHashSize;
        float                                   boxInvSize[dimension];
        float                                   boxHalfSize[dimension];
};

template< class type, int dimension >
ID_INLINE idVectorSubset<type,dimension>::idVectorSubset( void ) {
        hash.Clear( idMath::IPow( boxHashSize, dimension ), 128 );
        boxHashSize = 16;
        memset( boxInvSize, 0, dimension * sizeof( boxInvSize[0] ) );
        memset( boxHalfSize, 0, dimension * sizeof( boxHalfSize[0] ) );
}

template< class type, int dimension >
ID_INLINE idVectorSubset<type,dimension>::idVectorSubset( const type &mins, const type &maxs, const int boxHashSize, const int initialSize ) {
        Init( mins, maxs, boxHashSize, initialSize );
}

template< class type, int dimension >
ID_INLINE void idVectorSubset<type,dimension>::Init( const type &mins, const type &maxs, const int boxHashSize, const int initialSize ) {
        int i;
        float boxSize;

        hash.Clear( idMath::IPow( boxHashSize, dimension ), initialSize );

        this->mins = mins;
        this->maxs = maxs;
        this->boxHashSize = boxHashSize;

        for ( i = 0; i < dimension; i++ ) {
                boxSize = ( maxs[i] - mins[i] ) / (float) boxHashSize;
                boxInvSize[i] = 1.0f / boxSize;
                boxHalfSize[i] = boxSize * 0.5f;
        }
}

template< class type, int dimension >
ID_INLINE void idVectorSubset<type,dimension>::Clear( void ) {
        idList<type>::Clear();
        hash.Clear();
}

template< class type, int dimension >
ID_INLINE int idVectorSubset<type,dimension>::FindVector( const type *vectorList, const int vectorNum, const float epsilon ) {
        int i, j, k, hashKey, partialHashKey[dimension];
        const type &v = vectorList[vectorNum];

        for ( i = 0; i < dimension; i++ ) {
                assert( epsilon <= boxHalfSize[i] );
                partialHashKey[i] = (int) ( ( v[i] - mins[i] - boxHalfSize[i] ) * boxInvSize[i] );
        }

        for ( i = 0; i < ( 1 << dimension ); i++ ) {

                hashKey = 0;
                for ( j = 0; j < dimension; j++ ) {
                        hashKey *= boxHashSize;
                        hashKey += partialHashKey[j] + ( ( i >> j ) & 1 );
                }

                for ( j = hash.First( hashKey ); j >= 0; j = hash.Next( j ) ) {
                        const type &lv = vectorList[j];
                        for ( k = 0; k < dimension; k++ ) {
                                if ( idMath::Fabs( lv[k] - v[k] ) > epsilon ) {
                                        break;
                                }
                        }
                        if ( k >= dimension ) {
                                return j;
                        }
                }
        }

        hashKey = 0;
        for ( i = 0; i < dimension; i++ ) {
                hashKey *= boxHashSize;
                hashKey += (int) ( ( v[i] - mins[i] ) * boxInvSize[i] );
        }

        hash.Add( hashKey, vectorNum );
        return vectorNum;
}

#endif /* !__VECTORSET_H__ */