#ifndef TCG_AUTO_H
#define TCG_AUTO_H
#include "base.h"
#include "traits.h"
/* \file auto.h
\brief This file contains template classes able to perform special operations upon
instance destruction.
\details These classes can be useful to enforce block-scoped operations at a block's
entry point, considering that a block end can be far away, or the function
could return abruptly at several different points.
*/
namespace tcg
{
//*******************************************************************************
// tcg::auto_type definition
//*******************************************************************************
struct _auto_type {
mutable bool m_destruct;
public:
_auto_type(bool destruct) : m_destruct(destruct) {}
_auto_type(const _auto_type &other) : m_destruct(other.m_destruct)
{
other.m_destruct = false;
}
_auto_type &operator=(const _auto_type &other)
{
m_destruct = other.m_destruct, other.m_destruct = false;
return *this;
}
};
typedef const _auto_type &auto_type;
//*******************************************************************************
// tcg::auto_func definition
//*******************************************************************************
template <typename Op>
struct auto_zerary : public _auto_type {
Op m_op;
public:
auto_zerary(bool destruct = true) : _auto_type(destruct) {}
~auto_zerary()
{
if (this->m_destruct)
m_op();
}
};
//--------------------------------------------------------------------
template <typename Op, typename T = typename function_traits<Op>::arg1_type>
struct auto_unary : public _auto_type {
T m_arg1;
Op m_op;
public:
auto_unary(bool destruct = true) : _auto_type(destruct) {}
auto_unary(Op op, T arg, bool destruct = true)
: _auto_type(destruct), m_arg1(arg), m_op(op) {}
~auto_unary()
{
if (this->m_destruct)
m_op(m_arg1);
}
};
//--------------------------------------------------------------------
template <typename Op, typename T1 = typename function_traits<Op>::arg1_type,
typename T2 = typename function_traits<Op>::arg2_type>
struct auto_binary : public _auto_type {
T1 m_arg1;
T2 m_arg2;
Op m_op;
public:
auto_binary(bool destruct = true) : _auto_type(destruct) {}
auto_binary(Op op, T1 arg1, T2 arg2, bool destruct = true)
: _auto_type(destruct), m_arg1(arg1), m_arg2(arg2), m_op(op) {}
~auto_binary()
{
if (this->m_destruct)
m_op(m_arg1, m_arg2);
}
};
//*******************************************************************************
// Helper functions
//*******************************************************************************
template <typename Op>
auto_zerary<Op> make_auto(Op op, bool destruct = true)
{
return auto_zerary<Op>(op, destruct);
}
template <typename Op, typename T>
auto_unary<Op> make_auto(Op op, T &arg1, bool destruct = true)
{
return auto_unary<Op>(op, arg1, destruct);
}
template <typename Op, typename T>
auto_unary<Op> make_auto(Op op, const T &arg1, bool destruct = true)
{
return auto_unary<Op>(op, arg1, destruct);
}
template <typename Op, typename T1, typename T2>
auto_binary<Op> make_auto(Op op, T1 &arg1, T2 &arg2, bool destruct = true)
{
return auto_binary<Op>(op, arg1, arg2, destruct);
}
template <typename Op, typename T1, typename T2>
auto_binary<Op> make_auto(Op op, const T1 &arg1, T2 &arg2, bool destruct = true)
{
return auto_binary<Op>(op, arg1, arg2, destruct);
}
template <typename Op, typename T1, typename T2>
auto_binary<Op> make_auto(Op op, T1 &arg1, const T2 &arg2, bool destruct = true)
{
return auto_binary<Op>(op, arg1, arg2, destruct);
}
template <typename Op, typename T1, typename T2>
auto_binary<Op> make_auto(Op op, const T1 &arg1, const T2 &arg2, bool destruct = true)
{
return auto_binary<Op>(op, arg1, arg2, destruct);
}
//*******************************************************************************
// tcg::auto_reset definition
//*******************************************************************************
template <typename T, T val>
class auto_reset
{
typedef T var_type;
public:
var_type &m_var;
public:
auto_reset(var_type &var) : m_var(var) {}
~auto_reset() { m_var = val; }
private:
auto_reset(const auto_reset &);
auto_reset &operator=(const auto_reset &);
};
//*******************************************************************************
// tcg::auto_backup definition
//*******************************************************************************
template <typename T>
struct auto_backup {
typedef T var_type;
public:
var_type m_backup;
var_type *m_original;
public:
auto_backup() : m_original() {}
auto_backup(var_type &original) : m_original(&original), m_backup(original) {}
auto_backup(var_type *original) : m_original(original)
{
if (m_original)
m_backup = *m_original;
}
~auto_backup()
{
if (m_original)
*m_original = m_backup;
}
void reset(T &original)
{
m_original = &original;
m_backup = original;
}
void reset(T *original)
{
m_original = original;
if (m_original)
m_backup = *original;
}
T *release()
{
T *original = m_original;
m_original = 0;
return original;
}
private:
auto_backup(const auto_backup &);
auto_backup &operator=(const auto_backup &);
};
} // namespace tcg
#endif //TCG_AUTO_H