#pragma once

#ifndef TCG_FUNCTIONAL_H
#define TCG_FUNCTIONAL_H

#include "traits.h"

// std includes
#include <functional>

//**********************************************************************************
//    Logical functor combinators
//**********************************************************************************

namespace tcg
{

template <typename Fn1, typename Fn2>
class unary_and : public std::unary_function<typename function_traits<Fn1>::arg_type, bool>
{
	Fn1 m_fn1;
	Fn2 m_fn2;

public:
	unary_and(const Fn1 &fn1, const Fn2 &fn2)
		: m_fn1(fn1), m_fn2(fn2) {}

	bool operator()(const typename function_traits<Fn1>::arg_type &t) const
	{
		return m_fn1(t) && m_fn2(t);
	}
};

template <typename Fn1, typename Fn2>
unary_and<Fn1, Fn2> and1(const Fn1 &fn1, const Fn2 &fn2)
{
	return unary_and<Fn1, Fn2>(fn1, fn2);
}

//----------------------------------------------------------------------------------

template <typename Fn1, typename Fn2>
class unary_or : public std::unary_function<typename function_traits<Fn1>::arg_type, bool>
{
	Fn1 m_fn1;
	Fn2 m_fn2;

public:
	unary_or(const Fn1 &fn1, const Fn2 &fn2)
		: m_fn1(fn1), m_fn2(fn2) {}

	bool operator()(const typename function_traits<Fn1>::arg_type &t) const
	{
		return m_fn1(t) || m_fn2(t);
	}
};

template <typename Fn1, typename Fn2>
unary_or<Fn1, Fn2> or1(const Fn1 &fn1, const Fn2 &fn2)
{
	return unary_or<Fn1, Fn2>(fn1, fn2);
}

} // namespace tcg

#endif // TCG_FUNCTIONAL_H