jitcraft/StringPool.h
changeset 0 c455ebb9a5dc
child 54 8e22792a3921
     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/jitcraft/StringPool.h	Mon Jun 25 02:57:28 2012 -0700
     1.3 @@ -0,0 +1,56 @@
     1.4 +/*
     1.5 + * Copyright (C) 2012 David Anderson
     1.6 + *
     1.7 + * This file is part of JITCraft.
     1.8 + *
     1.9 + * JITCraft is free software: you can redistribute it and/or modify it under
    1.10 + * the terms of the GNU General Public License as published by the Free
    1.11 + * Software Foundation, either version 3 of the License, or (at your option)
    1.12 + * any later version.
    1.13 + * 
    1.14 + * Foobar is distributed in the hope that it will be useful, but WITHOUT ANY
    1.15 + * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
    1.16 + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
    1.17 + *
    1.18 + * You should have received a copy of the GNU General Public License along with
    1.19 + * JITCraft. If not, see http://www.gnu.org/licenses/.
    1.20 + */
    1.21 +#ifndef _include_jitcraft_string_pool_h_
    1.22 +#define _include_jitcraft_string_pool_h_
    1.23 +
    1.24 +#include <vector>
    1.25 +#include <string.h>
    1.26 +#include "PoolAllocator.h"
    1.27 +
    1.28 +namespace jc {
    1.29 +
    1.30 +typedef const char Name;
    1.31 +
    1.32 +// Guarantee that strings with the same contents will map to one pointer. The
    1.33 +// strings are pool allocated, and search is linear for simplicity.
    1.34 +class StringPool
    1.35 +{
    1.36 +    PoolAllocator &pool_;
    1.37 +    std::vector<Name *> strings_;
    1.38 +
    1.39 +  public:
    1.40 +    StringPool(PoolAllocator &pool)
    1.41 +      : pool_(pool)
    1.42 +    {
    1.43 +    }
    1.44 +
    1.45 +    Name *add(const char *string) {
    1.46 +        for (size_t i = 0; i < strings_.size(); i++) {
    1.47 +            if (strcmp(strings_[i], string) == 0)
    1.48 +                return strings_[i];
    1.49 +        }
    1.50 +        char *name = (char *)pool_.allocate(strlen(string) + 1);
    1.51 +        strcpy(name, string);
    1.52 +        strings_.push_back(name);
    1.53 +        return name;
    1.54 +    }
    1.55 +};
    1.56 +
    1.57 +}
    1.58 +
    1.59 +#endif // _include_jitcraft_string_pool_h_