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 >= getLengthInBytes() * 8) 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