// Copyright 2018-2022 David Robillard <d@drobilla.net>
// Copyright 2006-2007 Chris Hamilton <chamilton@cs.dal.ca>
// SPDX-License-Identifier: GPL-2.0-or-later

#ifndef CHILBERT_OPERATORS_HPP
#define CHILBERT_OPERATORS_HPP

#include "chilbert/detail/traits.hpp"

#include <iostream>
#include <type_traits>

namespace chilbert {

using detail::is_bitvec_v;

template<class T>
std::enable_if_t<is_bitvec_v<T>, T>
operator&(const T& lhs, const T& rhs)
{
  T r{lhs};
  r &= rhs;
  return r;
}

template<class T>
std::enable_if_t<is_bitvec_v<T>, T>
operator|(const T& lhs, const T& rhs)
{
  T r{lhs};
  r |= rhs;
  return r;
}

template<class T>
std::enable_if_t<is_bitvec_v<T>, T>
operator^(const T& lhs, const T& rhs)
{
  T r{lhs};
  r ^= rhs;
  return r;
}

template<class T>
std::enable_if_t<is_bitvec_v<T>, T>
operator~(const T& vec)
{
  T r{vec};
  r.flip();
  return r;
}

template<class T>
std::enable_if_t<is_bitvec_v<T>, T>
operator<<(const T& vec, const size_t bits)
{
  T r{vec};
  r <<= bits;
  return r;
}

template<class T>
std::enable_if_t<is_bitvec_v<T>, T>
operator>>(const T& vec, const size_t bits)
{
  T r{vec};
  r >>= bits;
  return r;
}

template<class T>
inline std::enable_if_t<is_bitvec_v<T>, std::ostream>&
operator<<(std::ostream& os, const T& vec)
{
  for (size_t i = 0; i < vec.size(); ++i) {
    os << vec.test(vec.size() - i - 1);
  }
  return os;
}

} // namespace chilbert

#endif