Angelscript/angelscript/source/as_symboltable.h

568 lines
12 KiB
C
Raw Normal View History

2021-04-12 18:25:02 +00:00
/*
AngelCode Scripting Library
Copyright (c) 2012-2015 Andreas Jonsson
This software is provided 'as-is', without any express or implied
warranty. In no event will the authors be held liable for any
damages arising from the use of this software.
Permission is granted to anyone to use this software for any
purpose, including commercial applications, and to alter it and
redistribute it freely, subject to the following restrictions:
1. The origin of this software must not be misrepresented; you
must not claim that you wrote the original software. If you use
this software in a product, an acknowledgment in the product
documentation would be appreciated but is not required.
2. Altered source versions must be plainly marked as such, and
must not be misrepresented as being the original software.
3. This notice may not be removed or altered from any source
distribution.
The original version of this library can be located at:
http://www.angelcode.com/angelscript/
Andreas Jonsson
andreas@angelcode.com
*/
//
// as_symboltable.h
//
// Created on: Jun 19, 2012
// Author: Markus Lenger, a.k.a. mlengerx
//
// This class is used for fast symbol lookups while parsing or loading bytecode
//
#ifndef AS_SYMBOLTABLE_H
#define AS_SYMBOLTABLE_H
#include "as_config.h"
#include "as_memory.h"
#include "as_string.h"
#include "as_map.h"
#include "as_datatype.h"
#include "as_namespace.h"
BEGIN_AS_NAMESPACE
// Interface to avoid nested templates which is not well supported by older compilers, e.g. MSVC6
struct asIFilter
{
virtual bool operator()(const void*) const = 0;
virtual ~asIFilter() {};
};
// forward declaration
template<class T>
class asCSymbolTable;
// Iterator that allows iterating in index order
template<class T, class T2 = T>
class asCSymbolTableIterator
{
public:
T2* operator*() const;
T2* operator->() const;
asCSymbolTableIterator<T, T2>& operator++(int);
asCSymbolTableIterator<T, T2>& operator--(int);
operator bool() const;
int GetIndex() const { return m_idx; }
private:
friend class asCSymbolTable<T>;
2021-06-26 09:23:50 +00:00
asCSymbolTableIterator(asCSymbolTable<T> *table);
2021-04-12 18:25:02 +00:00
void Next();
void Previous();
asCSymbolTable<T>* m_table;
unsigned int m_idx;
};
// Symbol table mapping namespace + name to symbols
// The structure keeps the entries indexed in an array so the indices will not change
// There is also a map for a quick lookup. The map supports multiple entries with the same name
template<class T>
class asCSymbolTable
{
public:
typedef asCSymbolTableIterator<T, T> iterator;
typedef asCSymbolTableIterator<T, const T> const_iterator;
asCSymbolTable(asUINT initialCapacity = 0);
int GetFirstIndex(const asSNameSpace *ns, const asCString &name, const asIFilter &comparator) const;
int GetFirstIndex(const asSNameSpace *ns, const asCString &name) const;
int GetLastIndex() const;
int GetIndex(const T*) const;
T* GetFirst(const asSNameSpace *ns, const asCString &name, const asIFilter &comparator) const;
T* GetFirst(const asSNameSpace *ns, const asCString &name);
const T* GetFirst(const asSNameSpace *ns, const asCString &name) const;
T* Get(asUINT index);
const T* Get(asUINT index) const;
T* GetLast();
const T* GetLast() const;
const asCArray<asUINT> &GetIndexes(const asSNameSpace *ns, const asCString &name) const;
asUINT Put(T* entry);
asUINT GetSize() const;
void SwapWith(asCSymbolTable<T> &other);
void Clear();
bool Erase(asUINT idx);
void Allocate(asUINT elem_cnt, bool keep_data);
iterator List();
const_iterator List() const;
private:
// Don't allow assignment
asCSymbolTable<T>& operator=(const asCSymbolTable<T> &other) { return *this; }
friend class asCSymbolTableIterator<T, T>;
friend class asCSymbolTableIterator<T, const T>;
void GetKey(const T *entry, asSNameSpaceNamePair &key) const;
bool CheckIdx(asUINT idx) const;
asCMap<asSNameSpaceNamePair, asCArray<asUINT> > m_map;
asCArray<T*> m_entries;
unsigned int m_size;
};
template<class T>
void asCSymbolTable<T>::SwapWith(asCSymbolTable<T> &other)
{
m_map.SwapWith(other.m_map);
m_entries.SwapWith(other.m_entries);
asUINT tmp = m_size;
m_size = other.m_size;
other.m_size = tmp;
}
// Constructor
// initialCapacity gives the number of entries to allocate in advance
template<class T>
asCSymbolTable<T>::asCSymbolTable(asUINT initialCapacity) : m_entries(initialCapacity)
{
m_size = 0;
}
template<class T>
int asCSymbolTable<T>::GetFirstIndex(
const asSNameSpace *ns,
const asCString &name,
const asIFilter &filter) const
{
asSNameSpaceNamePair key(ns, name);
asSMapNode<asSNameSpaceNamePair, asCArray<asUINT> > *cursor;
if( m_map.MoveTo(&cursor, key) )
{
const asCArray<asUINT> &arr = m_map.GetValue(cursor);
for( asUINT n = 0; n < arr.GetLength(); n++ )
{
T *entry = m_entries[arr[n]];
if( entry && filter(entry) )
return arr[n];
}
}
return -1;
}
template<class T>
const asCArray<asUINT> &asCSymbolTable<T>::GetIndexes(const asSNameSpace *ns, const asCString &name) const
{
asSNameSpaceNamePair key(ns, name);
asSMapNode<asSNameSpaceNamePair, asCArray<asUINT> > *cursor;
if( m_map.MoveTo(&cursor, key) )
return m_map.GetValue(cursor);
static asCArray<asUINT> dummy;
return dummy;
}
template<class T>
T* asCSymbolTable<T>::GetFirst(const asSNameSpace *ns, const asCString &name, const asIFilter &comp) const
{
int idx = GetFirstIndex(ns, name, comp);
if (idx != -1) return m_entries[idx];
return 0;
}
template<class T>
int asCSymbolTable<T>::GetFirstIndex(const asSNameSpace *ns, const asCString &name) const
{
asSNameSpaceNamePair key(ns, name);
asSMapNode<asSNameSpaceNamePair, asCArray<asUINT> > *cursor;
if( m_map.MoveTo(&cursor, key) )
return m_map.GetValue(cursor)[0];
return -1;
}
// Find the index of a certain symbol
// ATTENTION: this function has linear runtime complexity O(n)!!
template<class T>
int asCSymbolTable<T>::GetIndex(const T* entry) const
{
for( asUINT n = 0; n < m_entries.GetLength(); n++ )
if( m_entries[n] == entry )
return n;
return -1;
}
template<class T>
T* asCSymbolTable<T>::Get(asUINT idx)
{
if( !CheckIdx(idx) )
return 0;
return m_entries[idx];
}
template<class T>
const T* asCSymbolTable<T>::Get(asUINT idx) const
{
return const_cast< asCSymbolTable<T>* >(this)->Get(idx);
}
template<class T>
T* asCSymbolTable<T>::GetFirst(const asSNameSpace *ns, const asCString &name)
{
int idx = GetFirstIndex(ns, name);
return Get(idx);
}
template<class T>
const T* asCSymbolTable<T>::GetFirst(const asSNameSpace *ns, const asCString &name) const
{
return const_cast< asCSymbolTable<T>* >(this)->GetFirst(ns, name);
}
template<class T>
T* asCSymbolTable<T>::GetLast()
{
return Get(GetLastIndex());
}
template<class T>
const T* asCSymbolTable<T>::GetLast() const
{
return const_cast< asCSymbolTable<T>* >(this)->GetLast();
}
// Clear the symbol table
// ATTENTION: The contained symbols are not rleased. This is up to the client
template<class T>
void asCSymbolTable<T>::Clear()
{
m_entries.SetLength(0);
m_map.EraseAll();
m_size = 0;
}
// Pre-allocate slots for elemCnt entries
template<class T>
void asCSymbolTable<T>::Allocate(asUINT elemCnt, bool keepData)
{
asASSERT( elemCnt >= m_entries.GetLength() );
m_entries.Allocate(elemCnt, keepData);
if( !keepData )
m_map.EraseAll();
}
template<class T>
bool asCSymbolTable<T>::Erase(asUINT idx)
{
if( !CheckIdx(idx) )
{
asASSERT(false);
return false;
}
T *entry = m_entries[idx];
asASSERT(entry);
if( !entry )
return false;
// Remove the symbol from the lookup map
asSNameSpaceNamePair key;
GetKey(entry, key);
asSMapNode<asSNameSpaceNamePair, asCArray<asUINT> > *cursor;
if( m_map.MoveTo(&cursor, key) )
{
asCArray<asUINT> &arr = m_map.GetValue(cursor);
arr.RemoveValue(idx);
if( arr.GetLength() == 0 )
m_map.Erase(cursor);
}
else
asASSERT(false);
// Remove the symbol from the indexed array
if( idx == m_entries.GetLength() - 1 )
m_entries.PopLast();
else
{
// Must keep the array packed
int prevIdx = int(m_entries.GetLength()-1);
m_entries[idx] = m_entries.PopLast();
// Update the index in the lookup map
entry = m_entries[idx];
GetKey(entry, key);
if( m_map.MoveTo(&cursor, key) )
{
asCArray<asUINT> &arr = m_map.GetValue(cursor);
arr[arr.IndexOf(prevIdx)] = idx;
}
else
asASSERT(false);
}
m_size--;
return true;
}
template<class T>
asUINT asCSymbolTable<T>::Put(T *entry)
{
asUINT idx = m_entries.GetLength();
asSNameSpaceNamePair key;
GetKey(entry, key);
asSMapNode<asSNameSpaceNamePair, asCArray<asUINT> > *cursor;
if( m_map.MoveTo(&cursor, key) )
m_map.GetValue(cursor).PushLast(idx);
else
{
asCArray<asUINT> arr(1);
arr.PushLast(idx);
m_map.Insert(key, arr);
}
m_entries.PushLast(entry);
m_size++;
return idx;
}
// Return key for specified symbol (namespace and name are used to generate the key)
template<class T>
void asCSymbolTable<T>::GetKey(const T *entry, asSNameSpaceNamePair &key) const
{
key = asSNameSpaceNamePair(entry->nameSpace, entry->name);
}
template<class T>
asUINT asCSymbolTable<T>::GetSize() const
{
return m_size;
}
template<class T>
bool asCSymbolTable<T>::CheckIdx(asUINT idx) const
{
return idx < m_entries.GetLength();
}
template<class T>
int asCSymbolTable<T>::GetLastIndex() const
{
int idx = int(m_entries.GetLength()) - 1;
asASSERT( idx == -1 || m_entries[idx] );
return idx;
}
template<class T>
asCSymbolTableIterator<T, T> asCSymbolTable<T>::List()
{
return asCSymbolTableIterator<T, T>(this);
}
template<class T>
typename asCSymbolTable<T>::const_iterator asCSymbolTable<T>::List() const
{
return asCSymbolTableIterator<T, const T>(const_cast< asCSymbolTable<T> *>(this));
}
/////////////////////////////////////////////////////////////////////////////////////////////////
// Iterator
template<class T, class T2>
asCSymbolTableIterator<T, T2>::asCSymbolTableIterator(asCSymbolTable<T> *table) : m_table(table), m_idx(0)
{
asUINT sz = m_table->m_entries.GetLength();
while( m_idx < sz && m_table->m_entries[m_idx] == 0 )
m_idx++;
}
template<class T, class T2>
T2* asCSymbolTableIterator<T, T2>::operator*() const
{
asASSERT(m_table->CheckIdx(m_idx));
return m_table->m_entries[m_idx];
}
template<class T, class T2>
T2* asCSymbolTableIterator<T, T2>::operator->() const
{
asASSERT(m_table->CheckIdx(m_idx));
return m_table->m_entries[m_idx];
}
template<class T, class T2>
asCSymbolTableIterator<T, T2>& asCSymbolTableIterator<T, T2>::operator++(int)
{
Next();
return *this;
}
// Return true if more elements are following
// ATTENTION: When deleting the object currently pointed to by this iterator this
// method returns false even though there might be more elements in the list
template<class T, class T2>
asCSymbolTableIterator<T, T2>::operator bool() const
{
return m_idx < m_table->m_entries.GetLength() && m_table->m_entries[m_idx] != 0;
}
template<class T, class T2>
void asCSymbolTableIterator<T, T2>::Next()
{
asUINT sz = m_table->m_entries.GetLength();
m_idx++;
while( m_idx < sz && m_table->m_entries[m_idx] == 0 )
m_idx++;
}
template<class T, class T2>
void asCSymbolTableIterator<T, T2>::Previous()
{
// overflow on stepping over first element
asUINT sz = m_table->m_entries.GetLength();
m_idx--;
while( m_idx < sz && m_table->m_entries[m_idx] == 0 )
m_idx--;
}
template<class T, class T2>
asCSymbolTableIterator<T, T2>& asCSymbolTableIterator<T, T2>::operator--(int)
{
Previous();
return *this;
}
END_AS_NAMESPACE
#endif // AS_SYMBOLTABLE_H