BitArray.h

Go to the documentation of this file.

// -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 4; tab-width: 4; -*-
// vim: set shiftwidth=4 softtabstop=4 expandtab:
/*
 ********************************************************************
 ** NIDAS: NCAR In-situ Data Acquistion Software
 **
 ** 2006, Copyright University Corporation for Atmospheric Research
 **
 ** This program is free software; you can redistribute it and/or modify
 ** it under the terms of the GNU General Public License as published by
 ** the Free Software Foundation; either version 2 of the License, or
 ** (at your option) any later version.
 **
 ** This program is distributed in the hope that it will be useful,
 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 ** GNU General Public License for more details.
 **
 ** The LICENSE.txt file accompanying this software contains
 ** a copy of the GNU General Public License. If it is not found,
 ** write to the Free Software Foundation, Inc.,
 ** 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 **
 ********************************************************************
*/
 
#ifndef  NIDAS_UTIL_BITARRAY_H
#define  NIDAS_UTIL_BITARRAY_H
 
#include <string>
 
namespace nidas { namespace util {
 
class BitArray {
 
protected:
 
  unsigned char *bits;
  int lenBits;
  int lenBytes;
 
public:
 
    BitArray(int lenbits);
 
    BitArray(const BitArray& ba);
 
    BitArray& operator = (const BitArray& ba);
 
    ~BitArray();
 
    void setBits(bool value);
 
    void setBit(int num, bool value)
    {
    if (num < 0 || num >= getLength()) return;
    if (value) bits[num/8] |= (0x1 << (num%8));
    else bits[num/8] &= (~(0x1 << (num%8)));
    }
 
    void setBits(int begin,int end, unsigned int bitmask);
 
    void setBits64(int begin,int end, unsigned long long bitmask);
 
    bool getBit(int num) const
    {
    if (num >= getLength()) return false;
    return (bits[num/8] & (0x1 << (num%8))) != 0;
    }
 
    unsigned int getBits(int begin, int end);
 
    long long getBits64(int begin, int end);
 
    unsigned char* getPtr() { return bits; }
 
    const unsigned char* getConstPtr() const { return bits; }
 
    bool any() const
    {
        for (int i = 0; i < getLength(); i++)
        if (getBit(i) != 0) return true;
    return false;
    }
 
    bool all() const
    {
        for (int i = 0; i < getLength(); i++)
        if (getBit(i) == 0) return false;
    return true;
    }
 
    bool any(int begin,int end) const
    {
        for (int i = begin; i < std::min(end,getLength()); i++)
        if (getBit(i) != 0) return true;
    return false;
    }
 
    bool all(int begin,int end) const
    {
        for (int i = begin; i < std::min(end,getLength()); i++)
        if (getBit(i) == 0) return false;
    return true;
    }
 
    int getLength() const { return lenBits; }
 
    int getLengthInBytes() const { return lenBytes; }
 
    std::string toString() const;
 
    BitArray& operator |= (const BitArray& x);
 
    BitArray operator | (const BitArray& x);
 
    BitArray& operator &= (const BitArray& x);
 
    BitArray operator & (const BitArray& x);
 
    BitArray& operator ^= (const BitArray& x);
 
    BitArray operator ^ (const BitArray& x);
 
};
 
BitArray operator | (const BitArray& x,const BitArray& y);
 
BitArray operator & (const BitArray& x,const BitArray& y);
 
BitArray operator ^ (const BitArray& x,const BitArray& y);
 
bool operator == (const BitArray& x,const BitArray& y);
 
bool operator != (const BitArray& x,const BitArray& y);
 
} } // namespace nidas namespace util
 
#endif