1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
//
// Copyright (c) 2003-2008 greg Landrum and Rational Discovery LLC
//
// @@ All Rights Reserved @@
// This file is part of the RDKit.
// The contents are covered by the terms of the BSD license
// which is included in the file license.txt, found at the root
// of the RDKit source tree.
//
#include <RDGeneral/export.h>
#ifndef __RD_SPARSEBITVECTS_H__
#define __RD_SPARSEBITVECTS_H__
#include "BitVect.h"
#include <set>
using std::set;
#include <iterator>
#include <algorithm>
typedef set<int> IntSet;
typedef IntSet::iterator IntSetIter;
typedef IntSet::const_iterator IntSetConstIter;
//! a class for bit vectors that are sparsely occupied.
/*!
SparseBitVect objects store only their on bits, in an
std::set.
They are, as you might expect, quite memory efficient for sparsely populated
vectors but become rather a nightmare if they need to be negated.
*/
class RDKIT_DATASTRUCTS_EXPORT SparseBitVect : public BitVect {
public:
SparseBitVect() : dp_bits(0), d_size(0){};
//! initialize with a particular size;
explicit SparseBitVect(unsigned int size) : dp_bits(0), d_size(0) {
_initForSize(size);
};
//! copy constructor
SparseBitVect(const SparseBitVect &other) : BitVect(other) {
d_size = 0;
dp_bits = 0;
_initForSize(other.getNumBits());
IntSet *bv = other.dp_bits;
std::copy(bv->begin(), bv->end(), std::inserter(*dp_bits, dp_bits->end()));
}
//! construct from a string pickle
SparseBitVect(const std::string &);
//! construct from a text pickle
SparseBitVect(const char *data, const unsigned int dataLen);
SparseBitVect &operator=(const SparseBitVect &);
~SparseBitVect() { delete dp_bits; };
bool operator[](const unsigned int which) const;
SparseBitVect operator|(const SparseBitVect &) const;
SparseBitVect operator&(const SparseBitVect &) const;
SparseBitVect operator^(const SparseBitVect &) const;
SparseBitVect operator~() const;
//! returns a (const) pointer to our raw storage
const IntSet *getBitSet() const { return dp_bits; }
unsigned int getNumBits() const { return d_size; };
bool setBit(const unsigned int which);
bool setBit(const IntSetIter which);
bool unsetBit(const unsigned int which);
bool getBit(const unsigned int which) const;
bool getBit(const IntVectIter which) const;
bool getBit(const IntSetIter which) const;
unsigned int getNumOnBits() const { return static_cast<unsigned int>(dp_bits->size()); };
unsigned int getNumOffBits() const { return d_size - static_cast<unsigned int>(dp_bits->size()); };
std::string toString() const;
void getOnBits(IntVect &v) const;
void clearBits() { dp_bits->clear(); };
IntSet *dp_bits; //!< our raw data, exposed for the sake of efficiency
bool operator==(const SparseBitVect &o) const {
return *dp_bits == *o.dp_bits;
}
bool operator!=(const SparseBitVect &o) const {
return *dp_bits != *o.dp_bits;
}
private:
unsigned int d_size;
void _initForSize(const unsigned int size);
};
#endif
|
