git.sesse.net Git - plocate/blob - db.h

   1 #ifndef DB_H
   2 #define DB_H 1
   3
   4 #include <stdint.h>
   5
   6 struct Header {
   7         char magic[8];  // "\0plocate";
   8         uint32_t version;  // 1.
   9         uint32_t hashtable_size;
  10         uint32_t extra_ht_slots;
  11         uint32_t num_docids;
  12         uint64_t hash_table_offset_bytes;
  13         uint64_t filename_index_offset_bytes;
  14
  15         // Version 1 and up only.
  16         uint32_t max_version;  // Nominally 1, but can be increased if more features are added in a backward-compatible way.
  17         uint32_t zstd_dictionary_length_bytes;
  18         uint64_t zstd_dictionary_offset_bytes;
  19 };
  20
  21 struct Trigram {
  22         uint32_t trgm;
  23         uint32_t num_docids;
  24         uint64_t offset;
  25
  26         bool operator==(const Trigram &other) const
  27         {
  28                 return trgm == other.trgm;
  29         }
  30         bool operator<(const Trigram &other) const
  31         {
  32                 return trgm < other.trgm;
  33         }
  34 };
  35
  36 inline uint32_t hash_trigram(uint32_t trgm, uint32_t ht_size)
  37 {
  38         // CRC-like computation.
  39         uint32_t crc = trgm;
  40         for (int i = 0; i < 32; i++) {
  41                 bool bit = crc & 0x80000000;
  42                 crc <<= 1;
  43                 if (bit) {
  44                         crc ^= 0x1edc6f41;
  45                 }
  46         }
  47         return crc % ht_size;
  48 }
  49
  50 #endif  // !defined(DB_H)