summaryrefslogtreecommitdiff
path: root/libartbase/base/bit_vector.cc
diff options
context:
space:
mode:
author David Sehr <sehr@google.com> 2018-03-02 12:01:51 -0800
committer David Sehr <sehr@google.com> 2018-03-05 13:58:20 -0800
commitc431b9dc4b23cc950eb313695258df5d89f53b22 (patch)
tree422273559c3ae52caff0c6b1cf1a62a8312f0e26 /libartbase/base/bit_vector.cc
parentf46f46cf5bd32788d5252b7107628a66594a5e98 (diff)
Move most of runtime/base to libartbase/base
Enforce the layering that code in runtime/base should not depend on runtime by separating it into libartbase. Some of the code in runtime/base depends on the Runtime class, so it cannot be moved yet. Also, some of the tests depend on CommonRuntimeTest, which itself needs to be factored (in a subsequent CL). Bug: 22322814 Test: make -j 50 checkbuild make -j 50 test-art-host Change-Id: I8b096c1e2542f829eb456b4b057c71421b77d7e2
Diffstat (limited to 'libartbase/base/bit_vector.cc')
-rw-r--r--libartbase/base/bit_vector.cc375
1 files changed, 375 insertions, 0 deletions
diff --git a/libartbase/base/bit_vector.cc b/libartbase/base/bit_vector.cc
new file mode 100644
index 0000000000..c706c7ebf2
--- /dev/null
+++ b/libartbase/base/bit_vector.cc
@@ -0,0 +1,375 @@
+/*
+ * Copyright (C) 2011 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "bit_vector.h"
+
+#include <limits>
+#include <sstream>
+
+#include "allocator.h"
+#include "bit_vector-inl.h"
+
+namespace art {
+
+BitVector::BitVector(bool expandable,
+ Allocator* allocator,
+ uint32_t storage_size,
+ uint32_t* storage)
+ : storage_(storage),
+ storage_size_(storage_size),
+ allocator_(allocator),
+ expandable_(expandable) {
+ DCHECK(storage_ != nullptr);
+
+ static_assert(sizeof(*storage_) == kWordBytes, "word bytes");
+ static_assert(sizeof(*storage_) * 8u == kWordBits, "word bits");
+}
+
+BitVector::BitVector(uint32_t start_bits,
+ bool expandable,
+ Allocator* allocator)
+ : BitVector(expandable,
+ allocator,
+ BitsToWords(start_bits),
+ static_cast<uint32_t*>(allocator->Alloc(BitsToWords(start_bits) * kWordBytes))) {
+}
+
+
+BitVector::BitVector(const BitVector& src,
+ bool expandable,
+ Allocator* allocator)
+ : BitVector(expandable,
+ allocator,
+ src.storage_size_,
+ static_cast<uint32_t*>(allocator->Alloc(src.storage_size_ * kWordBytes))) {
+ // Direct memcpy would be faster, but this should be fine too and is cleaner.
+ Copy(&src);
+}
+
+BitVector::~BitVector() {
+ allocator_->Free(storage_);
+}
+
+bool BitVector::SameBitsSet(const BitVector *src) const {
+ int our_highest = GetHighestBitSet();
+ int src_highest = src->GetHighestBitSet();
+
+ // If the highest bit set is different, we are different.
+ if (our_highest != src_highest) {
+ return false;
+ }
+
+ // If the highest bit set is -1, both are cleared, we are the same.
+ // If the highest bit set is 0, both have a unique bit set, we are the same.
+ if (our_highest <= 0) {
+ return true;
+ }
+
+ // Get the highest bit set's cell's index
+ // No need of highest + 1 here because it can't be 0 so BitsToWords will work here.
+ int our_highest_index = BitsToWords(our_highest);
+
+ // This memcmp is enough: we know that the highest bit set is the same for both:
+ // - Therefore, min_size goes up to at least that, we are thus comparing at least what we need to, but not less.
+ // ie. we are comparing all storage cells that could have difference, if both vectors have cells above our_highest_index,
+ // they are automatically at 0.
+ return (memcmp(storage_, src->GetRawStorage(), our_highest_index * kWordBytes) == 0);
+}
+
+bool BitVector::IsSubsetOf(const BitVector *other) const {
+ int this_highest = GetHighestBitSet();
+ int other_highest = other->GetHighestBitSet();
+
+ // If the highest bit set is -1, this is empty and a trivial subset.
+ if (this_highest < 0) {
+ return true;
+ }
+
+ // If the highest bit set is higher, this cannot be a subset.
+ if (this_highest > other_highest) {
+ return false;
+ }
+
+ // Compare each 32-bit word.
+ size_t this_highest_index = BitsToWords(this_highest + 1);
+ for (size_t i = 0; i < this_highest_index; ++i) {
+ uint32_t this_storage = storage_[i];
+ uint32_t other_storage = other->storage_[i];
+ if ((this_storage | other_storage) != other_storage) {
+ return false;
+ }
+ }
+ return true;
+}
+
+void BitVector::Intersect(const BitVector* src) {
+ uint32_t src_storage_size = src->storage_size_;
+
+ // Get the minimum size between us and source.
+ uint32_t min_size = (storage_size_ < src_storage_size) ? storage_size_ : src_storage_size;
+
+ uint32_t idx;
+ for (idx = 0; idx < min_size; idx++) {
+ storage_[idx] &= src->GetRawStorageWord(idx);
+ }
+
+ // Now, due to this being an intersection, there are two possibilities:
+ // - Either src was larger than us: we don't care, all upper bits would thus be 0.
+ // - Either we are larger than src: we don't care, all upper bits would have been 0 too.
+ // So all we need to do is set all remaining bits to 0.
+ for (; idx < storage_size_; idx++) {
+ storage_[idx] = 0;
+ }
+}
+
+bool BitVector::Union(const BitVector* src) {
+ // Get the highest bit to determine how much we need to expand.
+ int highest_bit = src->GetHighestBitSet();
+ bool changed = false;
+
+ // If src has no bit set, we are done: there is no need for a union with src.
+ if (highest_bit == -1) {
+ return changed;
+ }
+
+ // Update src_size to how many cells we actually care about: where the bit is + 1.
+ uint32_t src_size = BitsToWords(highest_bit + 1);
+
+ // Is the storage size smaller than src's?
+ if (storage_size_ < src_size) {
+ changed = true;
+
+ EnsureSize(highest_bit);
+
+ // Paranoid: storage size should be big enough to hold this bit now.
+ DCHECK_LT(static_cast<uint32_t> (highest_bit), storage_size_ * kWordBits);
+ }
+
+ for (uint32_t idx = 0; idx < src_size; idx++) {
+ uint32_t existing = storage_[idx];
+ uint32_t update = existing | src->GetRawStorageWord(idx);
+ if (existing != update) {
+ changed = true;
+ storage_[idx] = update;
+ }
+ }
+ return changed;
+}
+
+bool BitVector::UnionIfNotIn(const BitVector* union_with, const BitVector* not_in) {
+ // Get the highest bit to determine how much we need to expand.
+ int highest_bit = union_with->GetHighestBitSet();
+ bool changed = false;
+
+ // If src has no bit set, we are done: there is no need for a union with src.
+ if (highest_bit == -1) {
+ return changed;
+ }
+
+ // Update union_with_size to how many cells we actually care about: where the bit is + 1.
+ uint32_t union_with_size = BitsToWords(highest_bit + 1);
+
+ // Is the storage size smaller than src's?
+ if (storage_size_ < union_with_size) {
+ EnsureSize(highest_bit);
+
+ // Paranoid: storage size should be big enough to hold this bit now.
+ DCHECK_LT(static_cast<uint32_t> (highest_bit), storage_size_ * kWordBits);
+ }
+
+ uint32_t not_in_size = not_in->GetStorageSize();
+
+ uint32_t idx = 0;
+ for (; idx < std::min(not_in_size, union_with_size); idx++) {
+ uint32_t existing = storage_[idx];
+ uint32_t update = existing |
+ (union_with->GetRawStorageWord(idx) & ~not_in->GetRawStorageWord(idx));
+ if (existing != update) {
+ changed = true;
+ storage_[idx] = update;
+ }
+ }
+
+ for (; idx < union_with_size; idx++) {
+ uint32_t existing = storage_[idx];
+ uint32_t update = existing | union_with->GetRawStorageWord(idx);
+ if (existing != update) {
+ changed = true;
+ storage_[idx] = update;
+ }
+ }
+ return changed;
+}
+
+void BitVector::Subtract(const BitVector *src) {
+ uint32_t src_size = src->storage_size_;
+
+ // We only need to operate on bytes up to the smaller of the sizes of the two operands.
+ unsigned int min_size = (storage_size_ > src_size) ? src_size : storage_size_;
+
+ // Difference until max, we know both accept it:
+ // There is no need to do more:
+ // If we are bigger than src, the upper bits are unchanged.
+ // If we are smaller than src, the nonexistent upper bits are 0 and thus can't get subtracted.
+ for (uint32_t idx = 0; idx < min_size; idx++) {
+ storage_[idx] &= (~(src->GetRawStorageWord(idx)));
+ }
+}
+
+uint32_t BitVector::NumSetBits() const {
+ uint32_t count = 0;
+ for (uint32_t word = 0; word < storage_size_; word++) {
+ count += POPCOUNT(storage_[word]);
+ }
+ return count;
+}
+
+uint32_t BitVector::NumSetBits(uint32_t end) const {
+ DCHECK_LE(end, storage_size_ * kWordBits);
+ return NumSetBits(storage_, end);
+}
+
+void BitVector::SetInitialBits(uint32_t num_bits) {
+ // If num_bits is 0, clear everything.
+ if (num_bits == 0) {
+ ClearAllBits();
+ return;
+ }
+
+ // Set the highest bit we want to set to get the BitVector allocated if need be.
+ SetBit(num_bits - 1);
+
+ uint32_t idx;
+ // We can set every storage element with -1.
+ for (idx = 0; idx < WordIndex(num_bits); idx++) {
+ storage_[idx] = std::numeric_limits<uint32_t>::max();
+ }
+
+ // Handle the potentially last few bits.
+ uint32_t rem_num_bits = num_bits & 0x1f;
+ if (rem_num_bits != 0) {
+ storage_[idx] = (1U << rem_num_bits) - 1;
+ ++idx;
+ }
+
+ // Now set the upper ones to 0.
+ for (; idx < storage_size_; idx++) {
+ storage_[idx] = 0;
+ }
+}
+
+int BitVector::GetHighestBitSet() const {
+ unsigned int max = storage_size_;
+ for (int idx = max - 1; idx >= 0; idx--) {
+ // If not 0, we have more work: check the bits.
+ uint32_t value = storage_[idx];
+
+ if (value != 0) {
+ // Return highest bit set in value plus bits from previous storage indexes.
+ return 31 - CLZ(value) + (idx * kWordBits);
+ }
+ }
+
+ // All zero, therefore return -1.
+ return -1;
+}
+
+void BitVector::Copy(const BitVector *src) {
+ // Get highest bit set, we only need to copy till then.
+ int highest_bit = src->GetHighestBitSet();
+
+ // If nothing is set, clear everything.
+ if (highest_bit == -1) {
+ ClearAllBits();
+ return;
+ }
+
+ // Set upper bit to ensure right size before copy.
+ SetBit(highest_bit);
+
+ // Now set until highest bit's storage.
+ uint32_t size = 1 + (highest_bit / kWordBits);
+ memcpy(storage_, src->GetRawStorage(), kWordBytes * size);
+
+ // Set upper bits to 0.
+ uint32_t left = storage_size_ - size;
+
+ if (left > 0) {
+ memset(storage_ + size, 0, kWordBytes * left);
+ }
+}
+
+uint32_t BitVector::NumSetBits(const uint32_t* storage, uint32_t end) {
+ uint32_t word_end = WordIndex(end);
+ uint32_t partial_word_bits = end & 0x1f;
+
+ uint32_t count = 0u;
+ for (uint32_t word = 0u; word < word_end; word++) {
+ count += POPCOUNT(storage[word]);
+ }
+ if (partial_word_bits != 0u) {
+ count += POPCOUNT(storage[word_end] & ~(0xffffffffu << partial_word_bits));
+ }
+ return count;
+}
+
+void BitVector::Dump(std::ostream& os, const char *prefix) const {
+ std::ostringstream buffer;
+ DumpHelper(prefix, buffer);
+ os << buffer.str() << std::endl;
+}
+
+void BitVector::DumpHelper(const char* prefix, std::ostringstream& buffer) const {
+ // Initialize it.
+ if (prefix != nullptr) {
+ buffer << prefix;
+ }
+
+ buffer << '(';
+ for (size_t i = 0; i < storage_size_ * kWordBits; i++) {
+ buffer << IsBitSet(i);
+ }
+ buffer << ')';
+}
+
+void BitVector::EnsureSize(uint32_t idx) {
+ if (idx >= storage_size_ * kWordBits) {
+ DCHECK(expandable_) << "Attempted to expand a non-expandable bitmap to position " << idx;
+
+ /* Round up to word boundaries for "idx+1" bits */
+ uint32_t new_size = BitsToWords(idx + 1);
+ DCHECK_GT(new_size, storage_size_);
+ uint32_t *new_storage =
+ static_cast<uint32_t*>(allocator_->Alloc(new_size * kWordBytes));
+ memcpy(new_storage, storage_, storage_size_ * kWordBytes);
+ // Zero out the new storage words.
+ memset(&new_storage[storage_size_], 0, (new_size - storage_size_) * kWordBytes);
+ // TODO: collect stats on space wasted because of resize.
+
+ // Free old storage.
+ allocator_->Free(storage_);
+
+ // Set fields.
+ storage_ = new_storage;
+ storage_size_ = new_size;
+ }
+}
+
+Allocator* BitVector::GetAllocator() const {
+ return allocator_;
+}
+
+} // namespace art