libcryfs/either.h

218 lines
6.4 KiB
C
Raw Normal View History

2015-06-25 15:46:53 +02:00
#pragma once
2015-10-15 12:58:16 +02:00
#ifndef MESSMER_CPPUTILS_EITHER_H
#define MESSMER_CPPUTILS_EITHER_H
2015-06-25 15:46:53 +02:00
#include <boost/optional.hpp>
2015-06-25 16:27:26 +02:00
2015-06-25 15:46:53 +02:00
namespace cpputils {
template<class Left, class Right>
2015-07-24 20:08:03 +02:00
class either final {
2015-06-25 15:46:53 +02:00
public:
2015-06-25 16:27:26 +02:00
//TODO Try allowing construction with any type that std::is_convertible to Left or Right.
2015-07-24 20:08:03 +02:00
either(const Left &left): _side(Side::left) {
2015-06-26 01:59:29 +02:00
_construct_left(left);
2015-06-25 15:46:53 +02:00
}
2015-07-24 20:08:03 +02:00
either(Left &&left): _side(Side::left) {
2015-06-26 01:59:29 +02:00
_construct_left(std::move(left));
2015-06-25 15:46:53 +02:00
}
2015-07-24 20:08:03 +02:00
either(const Right &right): _side(Side::right) {
2015-06-26 01:59:29 +02:00
_construct_right(right);
2015-06-25 15:46:53 +02:00
}
2015-07-24 20:08:03 +02:00
either(Right &&right): _side(Side::right) {
2015-06-26 01:59:29 +02:00
_construct_right(std::move(right));
2015-06-25 15:46:53 +02:00
}
2015-06-25 16:27:26 +02:00
//TODO Try allowing copy-construction when Left/Right types are std::is_convertible
2015-07-24 20:08:03 +02:00
either(const either<Left, Right> &rhs): _side(rhs._side) {
2015-06-25 15:46:53 +02:00
if(_side == Side::left) {
2015-06-26 01:59:29 +02:00
_construct_left(rhs._left);
2015-06-25 15:46:53 +02:00
} else {
2015-06-26 01:59:29 +02:00
_construct_right(rhs._right);
2015-06-25 15:46:53 +02:00
}
}
2015-07-24 20:08:03 +02:00
either(either<Left, Right> &&rhs): _side(rhs._side) {
2015-06-25 15:46:53 +02:00
if(_side == Side::left) {
2015-06-26 01:59:29 +02:00
_construct_left(std::move(rhs._left));
2015-06-25 15:46:53 +02:00
} else {
2015-06-26 01:59:29 +02:00
_construct_right(std::move(rhs._right));
2015-06-25 15:46:53 +02:00
}
}
2015-06-25 16:27:26 +02:00
2015-07-24 20:08:03 +02:00
~either() {
2015-06-26 01:59:29 +02:00
_destruct();
}
//TODO Try allowing copy-assignment when Left/Right types are std::is_convertible
2015-07-24 20:08:03 +02:00
either<Left, Right> &operator=(const either<Left, Right> &rhs) {
2015-06-26 01:59:29 +02:00
_destruct();
_side = rhs._side;
2015-06-25 16:27:26 +02:00
if (_side == Side::left) {
2015-06-26 01:59:29 +02:00
_construct_left(rhs._left);
2015-06-25 16:27:26 +02:00
} else {
2015-06-26 01:59:29 +02:00
_construct_right(rhs._right);
2015-06-25 16:27:26 +02:00
}
2015-06-26 01:59:29 +02:00
return *this;
2015-06-25 16:27:26 +02:00
}
2015-07-24 20:08:03 +02:00
either<Left, Right> &operator=(either<Left, Right> &&rhs) {
2015-06-26 01:59:29 +02:00
_destruct();
_side = rhs._side;
if (_side == Side::left) {
_construct_left(std::move(rhs._left));
} else {
_construct_right(std::move(rhs._right));
}
return *this;
}
2015-06-25 15:46:53 +02:00
2015-06-26 01:59:29 +02:00
//TODO fold, map_left, map_right, left_or_else(val), right_or_else(val), left_or_else(func), right_or_else(func)
2015-06-25 15:46:53 +02:00
bool is_left() const {
return _side == Side::left;
}
bool is_right() const {
return _side == Side::right;
}
const Left &left() const& {
2015-06-25 15:46:53 +02:00
return _left;
}
Left &left() & {
2015-07-24 20:08:03 +02:00
return const_cast<Left&>(const_cast<const either<Left, Right>*>(this)->left());
}
Left &&left() && {
return std::move(left());
}
2015-06-25 15:46:53 +02:00
const Right &right() const& {
2015-06-25 15:46:53 +02:00
return _right;
}
Right &right() & {
2015-07-24 20:08:03 +02:00
return const_cast<Right&>(const_cast<const either<Left, Right>*>(this)->right());
2015-06-25 15:46:53 +02:00
}
Right &&right() && {
return std::move(right());
}
boost::optional<const Left&> left_opt() const& {
if (_side == Side::left) {
return _left;
} else {
return boost::none;
}
}
boost::optional<Left&> left_opt() & {
if (_side == Side::left) {
return _left;
} else {
return boost::none;
}
}
boost::optional<Left> left_opt() && {
if (_side == Side::left) {
return std::move(_left);
} else {
return boost::none;
}
}
boost::optional<const Right&> right_opt() const& {
if (_side == Side::right) {
return _right;
} else {
return boost::none;
}
}
boost::optional<Right&> right_opt() & {
if (_side == Side::right) {
return _right;
} else {
return boost::none;
}
}
boost::optional<Right> right_opt() && {
if (_side == Side::right) {
return std::move(_right);
} else {
return boost::none;
}
}
2015-06-25 15:46:53 +02:00
private:
union {
Left _left;
Right _right;
};
enum class Side : unsigned char {left, right} _side;
2015-06-26 01:59:29 +02:00
2015-07-24 20:08:03 +02:00
either(Side side): _side(side) {}
2015-06-26 12:41:25 +02:00
template<typename... Args>
void _construct_left(Args&&... args) {
new(&_left)Left(std::forward<Args>(args)...);
2015-06-26 01:59:29 +02:00
}
2015-06-26 12:41:25 +02:00
template<typename... Args>
void _construct_right(Args&&... args) {
new(&_right)Right(std::forward<Args>(args)...);
2015-06-26 01:59:29 +02:00
}
void _destruct() {
if (_side == Side::left) {
_left.~Left();
} else {
_right.~Right();
}
}
2015-06-26 12:41:25 +02:00
template<typename Left_, typename Right_, typename... Args>
2015-07-24 20:08:03 +02:00
friend either<Left_, Right_> make_left(Args&&... args);
2015-06-26 12:41:25 +02:00
template<typename Left_, typename Right_, typename... Args>
2015-07-24 20:08:03 +02:00
friend either<Left_, Right_> make_right(Args&&... args);
2015-06-25 15:46:53 +02:00
};
2015-06-25 16:27:26 +02:00
template<class Left, class Right>
2015-07-24 20:08:03 +02:00
bool operator==(const either<Left, Right> &lhs, const either<Left, Right> &rhs) {
2015-06-25 16:27:26 +02:00
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>
2015-07-24 20:08:03 +02:00
bool operator!=(const either<Left, Right> &lhs, const either<Left, Right> &rhs) {
2015-06-25 16:27:26 +02:00
return !operator==(lhs, rhs);
}
2015-06-26 02:17:08 +02:00
template<class Left, class Right>
2015-07-24 20:08:03 +02:00
std::ostream &operator<<(std::ostream &stream, const either<Left, Right> &value) {
2015-06-26 02:17:08 +02:00
if (value.is_left()) {
stream << "Left(" << value.left() << ")";
} else {
stream << "Right(" << value.right() << ")";
}
return stream;
}
2015-06-26 12:41:25 +02:00
template<typename Left, typename Right, typename... Args>
2015-07-24 20:08:03 +02:00
either<Left, Right> make_left(Args&&... args) {
either<Left, Right> result(either<Left, Right>::Side::left);
2015-06-26 12:41:25 +02:00
result._construct_left(std::forward<Args>(args)...);
return result;
}
template<typename Left, typename Right, typename... Args>
2015-07-24 20:08:03 +02:00
either<Left, Right> make_right(Args&&... args) {
either<Left, Right> result(either<Left, Right>::Side::right);
2015-06-26 12:41:25 +02:00
result._construct_right(std::forward<Args>(args)...);
return result;
}
2015-06-25 15:46:53 +02:00
}
#endif