sql_bitmap.h
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:5k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2003 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. /*
  14.   Implementation of a bitmap type.
  15.   The idea with this is to be able to handle any constant number of bits but
  16.   also be able to use 32 or 64 bits bitmaps very efficiently
  17. */
  18. #include <my_bitmap.h>
  19. template <uint default_width> class Bitmap
  20. {
  21.   MY_BITMAP map;
  22.   uchar buffer[(default_width+7)/8];
  23. public:
  24.   Bitmap() { init(); }
  25.   Bitmap(Bitmap& from) { *this=from; }
  26.   explicit Bitmap(uint prefix_to_set) { init(prefix_to_set); }
  27.   void init() { bitmap_init(&map, buffer, default_width, 0); }
  28.   void init(uint prefix_to_set) { init(); set_prefix(prefix_to_set); }
  29.   uint length() const { return default_width; }
  30.   Bitmap& operator=(const Bitmap& map2)
  31.   {
  32.     init();
  33.     memcpy(buffer, map2.buffer, sizeof(buffer));
  34.     return *this;
  35.   }
  36.   void set_bit(uint n) { bitmap_set_bit(&map, n); }
  37.   void clear_bit(uint n) { bitmap_clear_bit(&map, n); }
  38.   void set_prefix(uint n) { bitmap_set_prefix(&map, n); }
  39.   void set_all() { bitmap_set_all(&map); }
  40.   void clear_all() { bitmap_clear_all(&map); }
  41.   void intersect(Bitmap& map2) { bitmap_intersect(&map, &map2.map); }
  42.   void intersect(ulonglong map2buff)
  43.   {
  44.     MY_BITMAP map2;
  45.     bitmap_init(&map2, (uchar *)&map2buff, sizeof(ulonglong)*8, 0);
  46.     bitmap_intersect(&map, &map2);
  47.   }
  48.   void subtract(Bitmap& map2) { bitmap_subtract(&map, &map2.map); }
  49.   void merge(Bitmap& map2) { bitmap_union(&map, &map2.map); }
  50.   my_bool is_set(uint n) const { return bitmap_is_set(&map, n); }
  51.   my_bool is_prefix(uint n) const { return bitmap_is_prefix(&map, n); }
  52.   my_bool is_clear_all() const { return bitmap_is_clear_all(&map); }
  53.   my_bool is_set_all() const { return bitmap_is_set_all(&map); }
  54.   my_bool is_subset(const Bitmap& map2) const { return bitmap_is_subset(&map, &map2.map); }
  55.   my_bool operator==(const Bitmap& map2) const { return bitmap_cmp(&map, &map2.map); }
  56.   char *print(char *buf) const
  57.   {
  58.     char *s=buf; int i;
  59.     for (i=sizeof(buffer)-1; i>=0 ; i--)
  60.     {
  61.       if ((*s=_dig_vec_upper[buffer[i] >> 4]) != '0')
  62.         break;
  63.       if ((*s=_dig_vec_upper[buffer[i] & 15]) != '0')
  64.         break;
  65.     }
  66.     for (s++, i-- ; i>=0 ; i--)
  67.     {
  68.       *s++=_dig_vec_upper[buffer[i] >> 4];
  69.       *s++=_dig_vec_upper[buffer[i] & 15];
  70.     }
  71.     *s=0;
  72.     return buf;
  73.   }
  74.   ulonglong to_ulonglong() const
  75.   {
  76.     if (sizeof(buffer) >= 8)
  77.       return uint8korr(buffer);
  78.     DBUG_ASSERT(sizeof(buffer) >= 4);
  79.     uint4korr(buffer);
  80.   }
  81. };
  82. template <> class Bitmap<64>
  83. {
  84.   ulonglong map;
  85. public:
  86.   Bitmap<64>() { }
  87. #if defined(__NETWARE__) || defined(__MWERKS__)
  88.   /*
  89.     Metwork compiler gives error on Bitmap<64>
  90.     Changed to Bitmap, since in this case also it will proper construct
  91.     this class
  92.   */
  93.   explicit Bitmap(uint prefix_to_set) { set_prefix(prefix_to_set); }
  94. #else
  95.   explicit Bitmap<64>(uint prefix_to_set) { set_prefix(prefix_to_set); }
  96. #endif
  97.   void init() { }
  98.   void init(uint prefix_to_set) { set_prefix(prefix_to_set); }
  99.   uint length() const { return 64; }
  100.   void set_bit(uint n) { map|= ((ulonglong)1) << n; }
  101.   void clear_bit(uint n) { map&= ~(((ulonglong)1) << n); }
  102.   void set_prefix(uint n)
  103.   {
  104.     if (n >= length())
  105.       set_all();
  106.     else
  107.       map= (((ulonglong)1) << n)-1;
  108.   }
  109.   void set_all() { map=~(ulonglong)0; }
  110.   void clear_all() { map=(ulonglong)0; }
  111.   void intersect(Bitmap<64>& map2) { map&= map2.map; }
  112.   void intersect(ulonglong map2) { map&= map2; }
  113.   void subtract(Bitmap<64>& map2) { map&= ~map2.map; }
  114.   void merge(Bitmap<64>& map2) { map|= map2.map; }
  115.   my_bool is_set(uint n) const { return test(map & (((ulonglong)1) << n)); }
  116.   my_bool is_prefix(uint n) const { return map == (((ulonglong)1) << n)-1; }
  117.   my_bool is_clear_all() const { return map == (ulonglong)0; }
  118.   my_bool is_set_all() const { return map == ~(ulonglong)0; }
  119.   my_bool is_subset(const Bitmap<64>& map2) const { return !(map & ~map2.map); }
  120.   my_bool operator==(const Bitmap<64>& map2) const { return map == map2.map; }
  121.   char *print(char *buf) const { longlong2str(map,buf,16); return buf; }
  122.   ulonglong to_ulonglong() const { return map; }
  123. };