#pragma once

#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