include/comma/basic/IdentifierPool.h

Go to the documentation of this file.

//===-- basic/IdentifierPool.h -------------------------------- -*- C++ -*-===//
//
// This file is distributed under the MIT license.  See LICENSE.txt for details.
//
// Copyright (C) 2008-2009, Stephen Wilson
//
//===----------------------------------------------------------------------===//

#ifndef COMMA_BASIC_IDENTIFIERPOOL_HDR_GUARD
#define COMMA_BASIC_IDENTIFIERPOOL_HDR_GUARD

#include "comma/basic/IdentifierInfo.h"
#include "llvm/ADT/StringMap.h"

namespace comma {

class IdentifierPool {

    typedef llvm::StringMap<IdentifierInfo> PoolType;
    PoolType pool;

public:
    IdentifierInfo &getIdentifierInfo(const char *name, size_t len) {
        return pool.GetOrCreateValue(name, name + len).getValue();
    }

    IdentifierInfo &getIdentifierInfo(const char *name) {
        return pool.GetOrCreateValue(name, name + strlen(name)).getValue();
    }

    IdentifierInfo &getIdentifierInfo(const std::string& name) {
        // We do not use c_str() here since the we do not require null
        // termination.
        const char *rep = &name[0];
        return getIdentifierInfo(rep, name.size());
    }

    typedef PoolType::const_iterator iterator;

    iterator begin() const { return pool.begin(); }

    iterator end()   const { return pool.end(); }

    unsigned size() const { return pool.size(); }
};

} // End comma namespace

#endif

---