my_bitmap.c
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:3k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  2.    
  3.    This library is free software; you can redistribute it and/or
  4.    modify it under the terms of the GNU Library General Public
  5.    License as published by the Free Software Foundation; either
  6.    version 2 of the License, or (at your option) any later version.
  7.    
  8.    This library is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11.    Library General Public License for more details.
  12.    
  13.    You should have received a copy of the GNU Library General Public
  14.    License along with this library; if not, write to the Free
  15.    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  16.    MA 02111-1307, USA */
  17. /*
  18.   Handling of uchar arrays as large bitmaps.
  19.   We assume that the size of the used bitmap is less than ~(uint) 0
  20.   TODO:
  21.   Make assembler THREAD safe versions of these using test-and-set instructions
  22. */
  23. #include "mysys_priv.h"
  24. #include <my_bitmap.h>
  25. #include <assert.h>
  26. my_bool bitmap_init(MY_BITMAP *map, uint bitmap_size)
  27. {
  28.   if (!(map->bitmap=(uchar*) my_malloc((bitmap_size+7)/8,
  29.        MYF(MY_WME | MY_ZEROFILL))))
  30.     return 1;
  31.   dbug_assert(bitmap_size != ~(uint) 0);
  32. #ifdef THREAD
  33.   pthread_mutex_init(&map->mutex, NULL);
  34. #endif
  35.   map->bitmap_size=bitmap_size;
  36.   return 0;
  37. }
  38. void bitmap_free(MY_BITMAP *map)
  39. {
  40.   if (map->bitmap)
  41.   {
  42.     my_free((char*) map->bitmap, MYF(0));
  43.     map->bitmap=0;
  44. #ifdef THREAD
  45.     pthread_mutex_destroy(&map->mutex);
  46. #endif
  47.   }
  48. }
  49. void bitmap_set_bit(MY_BITMAP *map, uint bitmap_bit)
  50. {
  51.   if (bitmap_bit < map->bitmap_size)
  52.   {
  53.     pthread_mutex_lock(&map->mutex);
  54.     map->bitmap[bitmap_bit / 8] |= (1 << (bitmap_bit & 7));
  55.     pthread_mutex_unlock(&map->mutex);
  56.   }
  57. }
  58. uint bitmap_set_next(MY_BITMAP *map)
  59. {
  60.   uchar *bitmap=map->bitmap;
  61.   uint bit_found = MY_BIT_NONE;
  62.   uint bitmap_size=map->bitmap_size;
  63.   uint i;
  64.   pthread_mutex_lock(&map->mutex);
  65.   for (i=0; i < bitmap_size ; i++, bitmap++)
  66.   {
  67.     if (*bitmap != 0xff)
  68.     { /* Found slot with free bit */
  69.       uint b;
  70.       for (b=0; ; b++)
  71.       {
  72. if (!(*bitmap & (1 << b)))
  73. {
  74.   *bitmap |= 1<<b;
  75.   bit_found = (i*8)+b;
  76.   break;
  77. }
  78.       }
  79.       break; /* Found bit */
  80.     }
  81.   }
  82.   pthread_mutex_unlock(&map->mutex);
  83.   return bit_found;
  84. }
  85. void bitmap_clear_bit(MY_BITMAP *map, uint bitmap_bit)
  86. {
  87.   if (bitmap_bit < map->bitmap_size)
  88.   {
  89.     pthread_mutex_lock(&map->mutex);
  90.     map->bitmap[bitmap_bit / 8] &= ~ (1 << (bitmap_bit & 7));
  91.     pthread_mutex_unlock(&map->mutex);
  92.   }
  93. }