NodeDecisionMasks.hpp

#pragma once
 
#include <cstddef>
#include <utility>
#include <vector>
 
namespace mmcfilters {
 
class NodePreservationMask {
  private:
    std::vector<bool> decisions_; 
 
  public:
    NodePreservationMask() = default;
 
    explicit NodePreservationMask(std::vector<bool> decisions) : decisions_(std::move(decisions)) {}
 
    NodePreservationMask(std::size_t size, bool preserveNode) : decisions_(size, preserveNode) {}
 
    [[nodiscard]] std::size_t size() const noexcept { return decisions_.size(); }
 
    [[nodiscard]] bool empty() const noexcept { return decisions_.empty(); }
 
    [[nodiscard]] bool operator[](std::size_t index) const { return decisions_[index]; }
 
    [[nodiscard]] const std::vector<bool>& decisions() const noexcept { return decisions_; }
};
 
class NodePruningMask {
  private:
    std::vector<bool> decisions_; 
 
  public:
    NodePruningMask() = default;
 
    explicit NodePruningMask(std::vector<bool> decisions) : decisions_(std::move(decisions)) {}
 
    NodePruningMask(std::size_t size, bool pruneNode) : decisions_(size, pruneNode) {}
 
    [[nodiscard]] std::size_t size() const noexcept { return decisions_.size(); }
 
    [[nodiscard]] bool empty() const noexcept { return decisions_.empty(); }
 
    [[nodiscard]] bool operator[](std::size_t index) const { return decisions_[index]; }
 
    [[nodiscard]] const std::vector<bool>& decisions() const noexcept { return decisions_; }
};
 
[[nodiscard]] inline NodePruningMask toNodePruningMask(const NodePreservationMask& nodePreservationMask) {
    std::vector<bool> nodePruningDecisions(nodePreservationMask.size(), false);
    for (std::size_t index = 0; index < nodePreservationMask.size(); ++index) {
        nodePruningDecisions[index] = !nodePreservationMask[index];
    }
    return NodePruningMask(std::move(nodePruningDecisions));
}
 
[[nodiscard]] inline NodePreservationMask toNodePreservationMask(const NodePruningMask& nodePruningMask) {
    std::vector<bool> nodePreservationDecisions(nodePruningMask.size(), false);
    for (std::size_t index = 0; index < nodePruningMask.size(); ++index) {
        nodePreservationDecisions[index] = !nodePruningMask[index];
    }
    return NodePreservationMask(std::move(nodePreservationDecisions));
}
 
} // namespace mmcfilters