aboutsummaryrefslogtreecommitdiffstats
path: root/chilbert/DynamicBitVec.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'chilbert/DynamicBitVec.hpp')
-rw-r--r--chilbert/DynamicBitVec.hpp148
1 files changed, 148 insertions, 0 deletions
diff --git a/chilbert/DynamicBitVec.hpp b/chilbert/DynamicBitVec.hpp
new file mode 100644
index 0000000..ab7d00b
--- /dev/null
+++ b/chilbert/DynamicBitVec.hpp
@@ -0,0 +1,148 @@
+/*
+ Copyright (C) 2018 David Robillard <d@drobilla.net>
+ Copyright (C) 2006-2007 Chris Hamilton <chamilton@cs.dal.ca>
+
+ 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.
+
+ You should have received a copy of the GNU General Public License along with
+ this program. If not, see <https://www.gnu.org/licenses/>.
+*/
+
+#ifndef CHILBERT_DYNAMICBITVEC_HPP
+#define CHILBERT_DYNAMICBITVEC_HPP
+
+#include "chilbert/BitVecIndex.hpp"
+#include "chilbert/BitVecIterator.hpp"
+#include "chilbert/BitVecMask.hpp"
+#include "chilbert/MultiBitVec.hpp"
+#include "chilbert/Operations.hpp"
+
+#include <algorithm>
+#include <cstddef>
+#include <cstdlib>
+#include <cstring>
+#include <memory>
+
+namespace chilbert {
+
+class DynamicBitVec : public MultiBitVec<DynamicBitVec>
+{
+public:
+ struct RacksDeleter
+ {
+ void operator()(Rack* const racks) { free(racks); }
+ };
+
+ struct NullDeleter
+ {
+ void operator()(const Rack* const) {}
+ };
+
+ using RacksPtr = std::unique_ptr<Rack[], RacksDeleter>;
+ using ConstRacksPtr = std::unique_ptr<const Rack[], NullDeleter>;
+
+ explicit DynamicBitVec(const size_t bits)
+ : m_racks{make_racks(calculate_num_racks(bits))}
+ , m_size{bits}
+ {
+ }
+
+ DynamicBitVec(const size_t bits, const Rack value)
+ : DynamicBitVec{bits}
+ {
+ m_racks[0] = value;
+ }
+
+ DynamicBitVec(const DynamicBitVec& vec)
+ : m_racks{make_racks(vec.num_racks())}
+ , m_size{vec.m_size}
+ {
+ if (vec.data()) {
+ memcpy(data(), vec.data(), data_size());
+ }
+ }
+
+ DynamicBitVec(DynamicBitVec&& vec) = default;
+
+ DynamicBitVec& operator=(const DynamicBitVec& vec)
+ {
+ if (num_racks() < vec.num_racks()) {
+ m_racks = make_racks(vec.num_racks());
+ m_size = vec.m_size;
+ memcpy(data(), vec.data(), data_size());
+ } else if (vec.num_racks() > 0) {
+ m_size = vec.m_size;
+ memcpy(data(), vec.data(), data_size());
+ } else {
+ m_size = 0;
+ m_racks.reset();
+ }
+
+ return *this;
+ }
+
+ DynamicBitVec& operator=(DynamicBitVec&& vec) = default;
+
+ /// Return the size in bits
+ size_t size() const { return m_size; }
+
+ /// Return a reference to the `index`th rack
+ const Rack& rack(const size_t index) const { return m_racks[index]; }
+ Rack& rack(const size_t index) { return m_racks[index]; }
+
+ /// Return a raw pointer to the racks
+ Rack* data() { return m_racks.get(); }
+ const Rack* data() const { return m_racks.get(); }
+
+ /// Return the total size of all racks in bytes
+ size_t data_size() const { return num_racks() * sizeof(Rack); }
+
+ /// Return the number of racks
+ size_t num_racks() const { return calculate_num_racks(m_size); }
+
+private:
+ static size_t calculate_num_racks(const size_t bits)
+ {
+ return (std::max(bits, size_t(1)) + bits_per_rack - 1) / bits_per_rack;
+ }
+
+ static RacksPtr make_racks(const size_t n)
+ {
+ return RacksPtr{static_cast<Rack*>(calloc(n, sizeof(Rack)))};
+ }
+
+ RacksPtr m_racks;
+ size_t m_size;
+};
+
+template <>
+struct is_bitvec<DynamicBitVec>
+{
+ constexpr static bool value = true;
+};
+
+template <>
+void
+gray_code(DynamicBitVec& value)
+{
+ gray_code(static_cast<MultiBitVec<DynamicBitVec>&>(value));
+}
+
+template <>
+void
+gray_code_inv(DynamicBitVec& value)
+{
+ gray_code_inv(static_cast<MultiBitVec<DynamicBitVec>&>(value));
+}
+
+} // namespace chilbert
+
+#endif