components.hpp

// Copyright (c) Lawrence Livermore National Security, LLC and
// other Smith Project Developers. See the top-level LICENSE file for
// details.
//
// SPDX-License-Identifier: (BSD-3-Clause)
 
#pragma once
 
#include <bitset>
 
namespace smith {
 
enum class Component : size_t
{
  X = 0b001,
  Y = 0b010,
  Z = 0b100,
  ALL = 0b111
};
 
class Components {
 public:
  Components(Component i) : flags_{size_t(i)} {};
 
  bool operator[](size_t i) const { return flags_[i]; };
 
  friend Components operator+(Component i, Component j);
 
  Components operator+(Component i)
  {
    flags_ |= size_t(i);
    return *this;
  };
 
  friend Components operator+(Component i, Components j);
 
 private:
  std::bitset<3> flags_;
};
 
inline Components operator+(Component i, Component j) { return Components(i) + j; };
 
inline Components operator+(Component i, Components c) { return c + i; }
 
}  // namespace smith