237 lines
8.5 KiB
C
237 lines
8.5 KiB
C
![]() |
#pragma once
|
||
|
#ifndef MESSMER_CPPUTILS_EITHER_H
|
||
|
#define MESSMER_CPPUTILS_EITHER_H
|
||
|
|
||
|
#include <boost/optional.hpp>
|
||
|
#include <iostream>
|
||
|
#include "assert/assert.h"
|
||
|
|
||
|
namespace cpputils {
|
||
|
|
||
|
template<class Left, class Right>
|
||
|
class either final {
|
||
|
public:
|
||
|
//TODO Try allowing construction with any type that std::is_convertible to Left or Right.
|
||
|
either(const Left &left) noexcept(noexcept(std::declval<either<Left, Right>>()._construct_left(left)))
|
||
|
: _side(Side::left) {
|
||
|
_construct_left(left);
|
||
|
}
|
||
|
|
||
|
either(Left &&left) noexcept(noexcept(std::declval<either<Left, Right>>()._construct_left(std::move(left))))
|
||
|
: _side(Side::left) {
|
||
|
_construct_left(std::move(left));
|
||
|
}
|
||
|
|
||
|
either(const Right &right) noexcept(noexcept(std::declval<either<Left, Right>>()._construct_right(right)))
|
||
|
: _side(Side::right) {
|
||
|
_construct_right(right);
|
||
|
}
|
||
|
|
||
|
either(Right &&right) noexcept(noexcept(std::declval<either<Left, Right>>()._construct_right(std::move(right))))
|
||
|
: _side(Side::right) {
|
||
|
_construct_right(std::move(right));
|
||
|
}
|
||
|
|
||
|
//TODO Try allowing copy-construction when Left/Right types are std::is_convertible
|
||
|
either(const either<Left, Right> &rhs) noexcept(noexcept(std::declval<either<Left, Right>>()._construct_left(rhs._left)) && noexcept(std::declval<either<Left, Right>>()._construct_right(rhs._right)))
|
||
|
: _side(rhs._side) {
|
||
|
if(_side == Side::left) {
|
||
|
_construct_left(rhs._left);
|
||
|
} else {
|
||
|
_construct_right(rhs._right);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
either(either<Left, Right> &&rhs) noexcept(noexcept(_construct_left(std::move(rhs._left))) && noexcept(_construct_right(std::move(rhs._right))))
|
||
|
: _side(rhs._side) {
|
||
|
if(_side == Side::left) {
|
||
|
_construct_left(std::move(rhs._left));
|
||
|
} else {
|
||
|
_construct_right(std::move(rhs._right));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
~either() {
|
||
|
_destruct();
|
||
|
}
|
||
|
|
||
|
//TODO Try allowing copy-assignment when Left/Right types are std::is_convertible
|
||
|
either<Left, Right> &operator=(const either<Left, Right> &rhs) noexcept(noexcept(_construct_left(rhs._left)) && noexcept(_construct_right(rhs._right))) {
|
||
|
_destruct();
|
||
|
_side = rhs._side;
|
||
|
if (_side == Side::left) {
|
||
|
_construct_left(rhs._left);
|
||
|
} else {
|
||
|
_construct_right(rhs._right);
|
||
|
}
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
either<Left, Right> &operator=(either<Left, Right> &&rhs) noexcept(noexcept(_construct_left(std::move(rhs._left))) && noexcept(_construct_right(std::move(rhs._right)))) {
|
||
|
_destruct();
|
||
|
_side = rhs._side;
|
||
|
if (_side == Side::left) {
|
||
|
_construct_left(std::move(rhs._left));
|
||
|
} else {
|
||
|
_construct_right(std::move(rhs._right));
|
||
|
}
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
//TODO fold, map_left, map_right, left_or_else(val), right_or_else(val), left_or_else(func), right_or_else(func)
|
||
|
|
||
|
bool is_left() const noexcept {
|
||
|
return _side == Side::left;
|
||
|
}
|
||
|
|
||
|
bool is_right() const noexcept {
|
||
|
return _side == Side::right;
|
||
|
}
|
||
|
|
||
|
const Left &left() const& {
|
||
|
ASSERT(is_left(), "Tried to get left side of an either which is right.");
|
||
|
return _left;
|
||
|
}
|
||
|
Left &left() & {
|
||
|
return const_cast<Left&>(const_cast<const either<Left, Right>*>(this)->left());
|
||
|
}
|
||
|
Left &&left() && {
|
||
|
return std::move(left());
|
||
|
}
|
||
|
|
||
|
const Right &right() const& {
|
||
|
ASSERT(is_right(), "Tried to get right side of an either which is left.");
|
||
|
return _right;
|
||
|
}
|
||
|
Right &right() & {
|
||
|
return const_cast<Right&>(const_cast<const either<Left, Right>*>(this)->right());
|
||
|
}
|
||
|
Right &&right() && {
|
||
|
return std::move(right());
|
||
|
}
|
||
|
|
||
|
boost::optional<const Left&> left_opt() const& noexcept {
|
||
|
if (_side == Side::left) {
|
||
|
return _left;
|
||
|
} else {
|
||
|
return boost::none;
|
||
|
}
|
||
|
}
|
||
|
boost::optional<Left&> left_opt() & noexcept {
|
||
|
if (_side == Side::left) {
|
||
|
return _left;
|
||
|
} else {
|
||
|
return boost::none;
|
||
|
}
|
||
|
}
|
||
|
boost::optional<Left> left_opt() && noexcept {
|
||
|
if (_side == Side::left) {
|
||
|
return std::move(_left);
|
||
|
} else {
|
||
|
return boost::none;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
boost::optional<const Right&> right_opt() const& noexcept {
|
||
|
if (_side == Side::right) {
|
||
|
return _right;
|
||
|
} else {
|
||
|
return boost::none;
|
||
|
}
|
||
|
}
|
||
|
boost::optional<Right&> right_opt() & noexcept {
|
||
|
if (_side == Side::right) {
|
||
|
return _right;
|
||
|
} else {
|
||
|
return boost::none;
|
||
|
}
|
||
|
}
|
||
|
boost::optional<Right> right_opt() && noexcept {
|
||
|
if (_side == Side::right) {
|
||
|
return std::move(_right);
|
||
|
} else {
|
||
|
return boost::none;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
union {
|
||
|
Left _left;
|
||
|
Right _right;
|
||
|
};
|
||
|
enum class Side : uint8_t {left, right} _side;
|
||
|
|
||
|
explicit either(Side side) noexcept : _side(side) {}
|
||
|
|
||
|
template<typename... Args>
|
||
|
void _construct_left(Args&&... args) noexcept(noexcept(new Left(std::forward<Args>(args)...))) {
|
||
|
new(&_left)Left(std::forward<Args>(args)...);
|
||
|
}
|
||
|
template<typename... Args>
|
||
|
void _construct_right(Args&&... args) noexcept(noexcept(new Right(std::forward<Args>(args)...))) {
|
||
|
new(&_right)Right(std::forward<Args>(args)...);
|
||
|
}
|
||
|
void _destruct() noexcept {
|
||
|
if (_side == Side::left) {
|
||
|
_left.~Left();
|
||
|
} else {
|
||
|
_right.~Right();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template<class... Args> static constexpr bool left_noexcept_constructible(Args&&... args) {
|
||
|
return noexcept(_construct_left(std::forward<Args>(args)...));
|
||
|
}
|
||
|
|
||
|
template<typename Left_, typename Right_, typename... Args>
|
||
|
friend either<Left_, Right_> make_left(Args&&... args) noexcept(noexcept(std::declval<either<Left, Right>>()._construct_left(std::forward<Args>(args)...)));
|
||
|
|
||
|
template<typename Left_, typename Right_, typename... Args>
|
||
|
friend either<Left_, Right_> make_right(Args&&... args) noexcept(noexcept(std::declval<either<Left, Right>>()._construct_right(std::forward<Args>(args)...)));
|
||
|
};
|
||
|
|
||
|
template<class Left, class Right>
|
||
|
bool operator==(const either<Left, Right> &lhs, const either<Left, Right> &rhs) noexcept(noexcept(std::declval<Left>() == std::declval<Left>()) && noexcept(std::declval<Right>() == std::declval<Right>())) {
|
||
|
if (lhs.is_left() != rhs.is_left()) {
|
||
|
return false;
|
||
|
}
|
||
|
if (lhs.is_left()) {
|
||
|
return lhs.left() == rhs.left();
|
||
|
} else {
|
||
|
return lhs.right() == rhs.right();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template<class Left, class Right>
|
||
|
bool operator!=(const either<Left, Right> &lhs, const either<Left, Right> &rhs) noexcept(noexcept(operator==(lhs, rhs))) {
|
||
|
return !operator==(lhs, rhs);
|
||
|
}
|
||
|
|
||
|
template<class Left, class Right>
|
||
|
std::ostream &operator<<(std::ostream &stream, const either<Left, Right> &value) {
|
||
|
if (value.is_left()) {
|
||
|
stream << "Left(" << value.left() << ")";
|
||
|
} else {
|
||
|
stream << "Right(" << value.right() << ")";
|
||
|
}
|
||
|
return stream;
|
||
|
}
|
||
|
|
||
|
template<typename Left, typename Right, typename... Args>
|
||
|
either<Left, Right> make_left(Args&&... args) noexcept(noexcept(std::declval<either<Left, Right>>()._construct_left(std::forward<Args>(args)...))) {
|
||
|
either<Left, Right> result(either<Left, Right>::Side::left);
|
||
|
result._construct_left(std::forward<Args>(args)...);
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
template<typename Left, typename Right, typename... Args>
|
||
|
either<Left, Right> make_right(Args&&... args) noexcept(noexcept(std::declval<either<Left, Right>>()._construct_right(std::forward<Args>(args)...))) {
|
||
|
either<Left, Right> result(either<Left, Right>::Side::right);
|
||
|
result._construct_right(std::forward<Args>(args)...);
|
||
|
return result;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
#endif
|