bitvector.H

Go to the documentation of this file.

// bitvector.H

#ifndef BITVECTOR_H
#define BITVECTOR_H

#include <stdio.h>

class bitvector
{
  unsigned *bits;
  int size, num;

  bitvector (void) { bits = NULL; size = num = 0; }
public:
  bitvector (int s)
  {
    size = s;
    num = s / (sizeof (unsigned) * 8) + 1;
    bits = new unsigned [num];
    Clear();
  }
  bitvector (const bitvector &orig)
  {
    size = orig.size;
    num = orig.num;
    bits = new unsigned [orig.num];
    Clear();
    for (int i = 0; i < num; i++) bits[i] = orig.bits[i];
  }
  bitvector & operator = (const bitvector &orig)
  {
    if (this != &orig)
      {
        delete[] bits;
        size = orig.size;
        num = orig.num;
        bits = new unsigned [orig.num];
        Clear();
        for (int i = 0; i < num; i++) bits[i] = orig.bits[i];
      } // end if
    return *this;
  }
  ~bitvector() { delete[] bits; }

  int GetSize (void) { return size; }

  void Clear (void)
  {
    for (int i = 0; i < num; i++) bits[i] = 0;
  }
  void Unclear (void)
  {
    for (int i = 0; i < num; i++) bits[i] = ~0;
  }
  void Set (int b)
  {
    int n = b / (sizeof (unsigned) * 8);
    if (n >= num || b < 0)
      {
        fprintf (stderr, "bitvector error: bit %d out of range.\n", b);
        return;
      }

    int bit = b % (sizeof (unsigned) * 8);
    bits[n] |= (1 << bit);
    return;
  }
  void Reset (int b)
  {
    int n = b / (sizeof (unsigned) * 8);
    if (n >= num || b < 0)
      {
        fprintf (stderr, "bitvector error: bit %d out of range.\n", b);
        return;
      }

    int bit = b % (sizeof (unsigned) * 8);
    bits[n] &= ~(1 << bit);
    return;
  }
  int Read (int b)
  {
    int n = b / (sizeof (unsigned) * 8);
    if (n >= num || b < 0)
      {
        fprintf (stderr, "bitvector error: bit %d out of range.\n", b);
        return 2;
      }

    int bit = b % (sizeof (unsigned) * 8);

    int value = ((bits[n] & (1 << bit)) != 0);
    return value;
  }

  bitvector operator ~ (void)
  {
    bitvector temp (size);
    for (int i = 0; i < num; i++) temp.bits[i] = ~(bits[i]);
    return temp;
  }
  bitvector operator | (bitvector mask)
  {
    bitvector temp (size);
    if (size != mask.size)
      {
        fprintf (stderr, "bitvector error: mask size mismatch.\n");
        return temp;
      }

    for (int i = 0; i < num; i++) temp.bits[i] = bits[i] | mask.bits[i];
    return temp;
  }
  bitvector operator & (bitvector mask)
  {
    bitvector temp (size);
    if (size != mask.size)
      {
        fprintf (stderr, "bitvector error: mask size mismatch.\n");
        return temp;
      }

    for (int i = 0; i < num; i++) temp.bits[i] = bits[i] & mask.bits[i];
    return temp;
  }
  int operator == (bitvector other)
  {
    if (size != other.size)
      return 0;

    for (int i = 0; i < num; i++)
      if (bits[i] != other.bits[i]) return 0;

    return 1;
  }
  int operator != (bitvector other)
  {
    return (!(*this == other));
  }
  void Show (void)
  {
    cerr << "\n";
    for (int i = (size - 1); i >= 0; i--) {
      if (Read(i))
        cerr << "1";
      else
        cerr << "0";
    }
    cerr << "\n";
  }
}; // end class bitvector

//int operator == (class bitvector first, class bitvector second)
//{
//  return (first.operator ==(second));
//}

#endif

---