src/third_party/skia/include/gpu/GrBinHashKey.h

   1
   2 /*
   3  * Copyright 2011 Google Inc.
   4  *
   5  * Use of this source code is governed by a BSD-style license that can be
   6  * found in the LICENSE file.
   7  */
   8
   9
  10 #ifndef GrBinHashKey_DEFINED
  11 #define GrBinHashKey_DEFINED
  12
  13 #include "GrTypes.h"
  14
  15 /**
  16  *  GrBinHashKey is a hash key class that can take a data chunk of any predetermined
  17  *  length. The hash function used is the One-at-a-Time Hash
  18  *  (http://burtleburtle.net/bob/hash/doobs.html).
  19  */
  20 template<size_t KEY_SIZE>
  21 class GrBinHashKey {
  22 public:
  23     enum { kKeySize = KEY_SIZE };
  24
  25     GrBinHashKey() {
  26         this->reset();
  27     }
  28
  29     void reset() {
  30         fHash = 0;
  31 #ifdef SK_DEBUG
  32         fIsValid = false;
  33 #endif
  34     }
  35
  36     void setKeyData(const uint32_t* SK_RESTRICT data) {
  37         SK_COMPILE_ASSERT(KEY_SIZE % 4 == 0, key_size_mismatch);
  38         memcpy(&fData, data, KEY_SIZE);
  39
  40         uint32_t hash = 0;
  41         size_t len = KEY_SIZE;
  42         while (len >= 4) {
  43             hash += *data++;
  44             hash += (hash << 10);
  45             hash ^= (hash >> 6);
  46             len -= 4;
  47         }
  48         hash += (hash << 3);
  49         hash ^= (hash >> 11);
  50         hash += (hash << 15);
  51 #ifdef SK_DEBUG
  52         fIsValid = true;
  53 #endif
  54         fHash = hash;
  55     }
  56
  57     bool operator==(const GrBinHashKey<KEY_SIZE>& key) const {
  58         SkASSERT(fIsValid && key.fIsValid);
  59         if (fHash != key.fHash) {
  60             return false;
  61         }
  62         for (size_t i = 0; i < SK_ARRAY_COUNT(fData); ++i) {
  63             if (fData[i] != key.fData[i]) {
  64                 return false;
  65             }
  66         }
  67         return true;
  68     }
  69
  70     bool operator<(const GrBinHashKey<KEY_SIZE>& key) const {
  71         SkASSERT(fIsValid && key.fIsValid);
  72         for (size_t i = 0; i < SK_ARRAY_COUNT(fData); ++i) {
  73             if (fData[i] < key.fData[i]) {
  74                 return true;
  75             } else if (fData[i] > key.fData[i]) {
  76                 return false;
  77             }
  78         }
  79         return false;
  80     }
  81
  82     uint32_t getHash() const {
  83         SkASSERT(fIsValid);
  84         return fHash;
  85     }
  86
  87     const uint8_t* getData() const {
  88         SkASSERT(fIsValid);
  89         return reinterpret_cast<const uint8_t*>(fData);
  90     }
  91
  92 private:
  93     uint32_t            fHash;
  94     uint32_t            fData[KEY_SIZE / sizeof(uint32_t)];  // Buffer for key storage.
  95
  96 #ifdef SK_DEBUG
  97 public:
  98     bool                fIsValid;
  99 #endif
 100 };
 101
 102 #endif