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//===----------------------------------------------------------------------===// 95460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#ifndef MCLD_UNIQUE_GCFACTORY_H 105460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#define MCLD_UNIQUE_GCFACTORY_H 115460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#ifdef ENABLE_UNITTEST 125460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <gtest.h> 135460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#endif 145460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 155460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include "mcld/Support/GCFactory.h" 165460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <map> 175460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <utility> 185460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 195460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaonamespace mcld 205460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 215460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 225460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/** \class UniqueGCFactoryBase 235460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao * \brief UniqueGCFactories are unique associative factories, meaning that 245460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao * no two elements have the same key. 255460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao */ 265460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaotemplate<typename KeyType, typename DataType, size_t ChunkSize> 275460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoclass UniqueGCFactoryBase : public GCFactoryBase<LinearAllocator<DataType, ChunkSize> > 285460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 295460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoprotected: 305460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typedef GCFactoryBase<LinearAllocator<DataType, ChunkSize> > Alloc; 315460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typedef std::map<KeyType, DataType*> KeyMap; 325460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 335460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoprotected: 345460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao UniqueGCFactoryBase() 355460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao : GCFactoryBase<LinearAllocator<DataType, ChunkSize> >() 365460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao { } 375460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 385460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao UniqueGCFactoryBase(size_t pNum) 395460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao : GCFactoryBase<LinearAllocator<DataType, ChunkSize> >(pNum) 405460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao { } 415460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 425460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaopublic: 435460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao virtual ~UniqueGCFactoryBase() 445460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao { f_KeyMap.clear(); } 455460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 465460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* find(const KeyType& pKey) { 475460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typename KeyMap::iterator dataIter = f_KeyMap.find(pKey); 485460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (dataIter != f_KeyMap.end()) 495460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return dataIter->second; 505460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return 0; 515460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 525460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 535460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao const DataType* find(const KeyType& pKey) const { 545460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typename KeyMap::const_iterator dataIter = f_KeyMap.find(pKey); 555460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (dataIter != f_KeyMap.end()) 565460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return dataIter->second; 575460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return 0; 585460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 595460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 605460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* produce(const KeyType& pKey, bool& pExist) { 615460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typename KeyMap::iterator dataIter = f_KeyMap.find(pKey); 625460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (dataIter != f_KeyMap.end()) { 635460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = true; 645460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return dataIter->second; 655460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 665460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* data = Alloc::allocate(); 675460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao construct(data); 685460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao f_KeyMap.insert(std::make_pair(pKey, data)); 695460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = false; 705460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return data; 715460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 725460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 735460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* produce(const KeyType& pKey, const DataType& pValue, bool& pExist) { 745460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao typename KeyMap::iterator dataIter = f_KeyMap.find(pKey); 755460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (dataIter != f_KeyMap.end()) { 765460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = true; 775460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return dataIter->second; 785460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 795460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao DataType* data = Alloc::allocate(); 805460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao construct(data, pValue); 815460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao f_KeyMap.insert(std::make_pair(pKey, data)); 825460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = false; 835460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return data; 845460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 855460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 865460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoprotected: 875460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao KeyMap f_KeyMap; 885460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 895460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao}; 905460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 915460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} // namespace of mcld 925460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 935460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#endif 945460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 95