15460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//===- UniqueGCFactory.h --------------------------------------------------===// 25460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// 35460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// The MCLinker Project 45460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// 55460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// This file is distributed under the University of Illinois Open Source 65460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// License. See LICENSE.TXT for details. 75460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// 85460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//===----------------------------------------------------------------------===// 987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines#ifndef MCLD_SUPPORT_UNIQUEGCFACTORY_H 1087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines#define MCLD_SUPPORT_UNIQUEGCFACTORY_H 115460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 125460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include "mcld/Support/GCFactory.h" 135460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <map> 145460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <utility> 155460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 165460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaonamespace mcld 175460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 185460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 195460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/** \class UniqueGCFactoryBase 205460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao * \brief UniqueGCFactories are unique associative factories, meaning that 215460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao * no two elements have the same key. 225460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao */ 235460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaotemplate<typename KeyType, typename DataType, size_t ChunkSize> 245460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoclass UniqueGCFactoryBase : public GCFactoryBase<LinearAllocator<DataType, ChunkSize> > 255460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 265460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoprotected: 275460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typedef GCFactoryBase<LinearAllocator<DataType, ChunkSize> > Alloc; 285460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typedef std::map<KeyType, DataType*> KeyMap; 295460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 305460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoprotected: 315460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao UniqueGCFactoryBase() 325460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao : GCFactoryBase<LinearAllocator<DataType, ChunkSize> >() 335460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao { } 345460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 355460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao UniqueGCFactoryBase(size_t pNum) 365460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao : GCFactoryBase<LinearAllocator<DataType, ChunkSize> >(pNum) 375460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao { } 385460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 395460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaopublic: 405460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao virtual ~UniqueGCFactoryBase() 415460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao { f_KeyMap.clear(); } 425460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 435460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* find(const KeyType& pKey) { 445460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typename KeyMap::iterator dataIter = f_KeyMap.find(pKey); 455460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (dataIter != f_KeyMap.end()) 465460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return dataIter->second; 475460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return 0; 485460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 495460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 505460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao const DataType* find(const KeyType& pKey) const { 515460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typename KeyMap::const_iterator dataIter = f_KeyMap.find(pKey); 525460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (dataIter != f_KeyMap.end()) 535460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return dataIter->second; 545460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return 0; 555460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 565460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 575460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* produce(const KeyType& pKey, bool& pExist) { 585460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typename KeyMap::iterator dataIter = f_KeyMap.find(pKey); 595460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (dataIter != f_KeyMap.end()) { 605460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = true; 615460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return dataIter->second; 625460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 635460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* data = Alloc::allocate(); 645460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao construct(data); 655460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao f_KeyMap.insert(std::make_pair(pKey, data)); 665460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = false; 675460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return data; 685460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 695460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 705460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* produce(const KeyType& pKey, const DataType& pValue, bool& pExist) { 715460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typename KeyMap::iterator dataIter = f_KeyMap.find(pKey); 725460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (dataIter != f_KeyMap.end()) { 735460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = true; 745460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return dataIter->second; 755460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 765460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* data = Alloc::allocate(); 775460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao construct(data, pValue); 785460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao f_KeyMap.insert(std::make_pair(pKey, data)); 795460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = false; 805460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return data; 815460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 825460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 835460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoprotected: 845460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao KeyMap f_KeyMap; 855460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 865460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao}; 875460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 885460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} // namespace of mcld 895460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 905460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#endif 915460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 92