/*
 * Copyright (c) 2014, Siemens AG. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef EMBB_BASE_INTERNAL_FUNCTION4_H_
#define EMBB_BASE_INTERNAL_FUNCTION4_H_

#include <new>
#include <cstddef>

#include <embb/base/internal/nil.h>
#include <embb/base/memory_allocation.h>
#include <embb/base/atomic.h>
#include <embb/base/internal/functionT.h>
#include <embb/base/internal/function0.h>
#include <embb/base/internal/function1.h>

namespace embb {
namespace base {

namespace internal {

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
class Function4 {
 public:
  virtual ~Function4() {}
  virtual R operator () (T1, T2, T3, T4) = 0;
  virtual void CopyTo(void* dst) = 0;
};

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
class FunctionPointer4
  : public Function4<R, T1, T2, T3, T4> {
 public:
  typedef R(*FuncPtrType)(T1, T2, T3, T4);
  explicit FunctionPointer4(FuncPtrType func) : function_(func) {}
  virtual R operator () (T1 p1, T2 p2, T3 p3, T4 p4) {
    return function_(p1, p2, p3, p4);
  }
  virtual void CopyTo(void* dst) {
    new(dst)FunctionPointer4(function_);
  }

 private:
  FuncPtrType function_;
};

template <typename T1, typename T2, typename T3, typename T4>
class FunctionPointer4<void, T1, T2, T3, T4>
  : public Function4<void, T1, T2, T3, T4> {
 public:
  typedef void(*FuncPtrType)(T1, T2, T3, T4);
  explicit FunctionPointer4(FuncPtrType func) : function_(func) {}
  virtual void operator () (T1 p1, T2 p2, T3 p3, T4 p4) {
    function_(p1, p2, p3, p4);
  }
  virtual void CopyTo(void* dst) {
    new(dst)FunctionPointer4(function_);
  }

 private:
  FuncPtrType function_;
};

template <class C, typename R,
  typename T1, typename T2, typename T3, typename T4>
class MemberFunctionPointer4
  : public Function4<R, T1, T2, T3, T4> {
 public:
  typedef R(C::*MemFuncPtrType)(T1, T2, T3, T4);
  typedef C & ClassRefType;
  MemberFunctionPointer4(ClassRefType obj, MemFuncPtrType func)
  : object_(obj), function_(func) {}
  explicit MemberFunctionPointer4(ClassRefType obj)
  : object_(obj), function_(&C::operator()) {}
  void operator = (MemberFunctionPointer4 const & memfunc) {
    object_ = memfunc.object_;
    function_ = memfunc.function_;
  }
  virtual R operator () (T1 p1, T2 p2, T3 p3, T4 p4) {
    return (object_.*function_)(p1, p2, p3, p4);
  }
  virtual void CopyTo(void* dst) {
    new(dst)MemberFunctionPointer4(object_, function_);
  }

 private:
  ClassRefType object_;
  MemFuncPtrType function_;
};

template <class C,
  typename T1, typename T2, typename T3, typename T4>
class MemberFunctionPointer4<C, void, T1, T2, T3, T4>
  : public Function4<void, T1, T2, T3, T4> {
 public:
  typedef void(C::*MemFuncPtrType)(T1, T2, T3, T4);
  typedef C & ClassRefType;
  MemberFunctionPointer4(ClassRefType obj, MemFuncPtrType func)
  : object_(obj), function_(func) {}
  explicit MemberFunctionPointer4(ClassRefType obj)
  : object_(obj), function_(&C::operator()) {}
  void operator = (MemberFunctionPointer4 const & memfunc) {
    object_ = memfunc.object_;
    function_ = memfunc.function_;
  }
  virtual void operator () (T1 p1, T2 p2, T3 p3, T4 p4) {
    (object_.*function_)(p1, p2, p3, p4);
  }
  virtual void CopyTo(void* dst) {
    new(dst)MemberFunctionPointer4(object_, function_);
  }

 private:
  ClassRefType object_;
  MemFuncPtrType function_;
};

template <class C, typename R,
  typename T1, typename T2, typename T3, typename T4>
class FunctorWrapper4
  : public Function4<R, T1, T2, T3, T4> {
 public:
  FunctorWrapper4() : object_(NULL), ref_count_(NULL) {}
  explicit FunctorWrapper4(C const & obj) {
    object_ = Allocation::New<C>(obj);
    ref_count_ = Allocation::New<Atomic<int> >(1);
  }
  explicit FunctorWrapper4(FunctorWrapper4 const & other) {
    object_ = other.object_;
    ref_count_ = other.ref_count_;
    ++*ref_count_;
  }
  virtual ~FunctorWrapper4() {
    if (0 == --*ref_count_) {
      Allocation::Delete(ref_count_);
      Allocation::Delete(object_);
    }
  }
  virtual R operator () (T1 p1, T2 p2, T3 p3, T4 p4) {
    return (*object_)(p1, p2, p3, p4);
  }
  virtual void CopyTo(void* dst) {
    new(dst)FunctorWrapper4(*this);
  }

 private:
  C * object_;
  Atomic<int> * ref_count_;
};

template <class C,
  typename T1, typename T2, typename T3, typename T4>
class FunctorWrapper4<C, void, T1, T2, T3, T4>
  : public Function4<void, T1, T2, T3, T4> {
 public:
  FunctorWrapper4() : object_(NULL), ref_count_(NULL) {}
  explicit FunctorWrapper4(C const & obj) {
    object_ = Allocation::New<C>(obj);
    ref_count_ = Allocation::New<Atomic<int> >(1);
  }
  explicit FunctorWrapper4(FunctorWrapper4 const & other) {
    object_ = other.object_;
    ref_count_ = other.ref_count_;
    ++*ref_count_;
  }
  virtual ~FunctorWrapper4() {
    if (0 == --*ref_count_) {
      Allocation::Delete(ref_count_);
      Allocation::Delete(object_);
    }
  }
  virtual void operator () (T1 p1, T2 p2, T3 p3, T4 p4) {
    (*object_)(p1, p2, p3, p4);
  }
  virtual void CopyTo(void* dst) {
    new(dst)FunctorWrapper4(*this);
  }

 private:
  C * object_;
  Atomic<int> * ref_count_;
};

// bind to function0
template <typename R,
  typename T1, typename T2, typename T3, typename T4>
class Bound4Functor0 {
 public:
  Bound4Functor0(Function<R, T1, T2, T3, T4> func,
    T1 p1, T2 p2, T3 p3, T4 p4) : function_(func)
      , p1_(p1), p2_(p2), p3_(p3), p4_(p4) {}
  Bound4Functor0(Bound4Functor0 const & func) : function_(func.function_)
    , p1_(func.p1_), p2_(func.p2_), p3_(func.p3_), p4_(func.p4_)
     {}
  R operator() () {
    return function_(p1_, p2_, p3_, p4_);
  }

 private:
  Function<R, T1, T2, T3, T4> function_;
  T1 p1_;
  T2 p2_;
  T3 p3_;
  T4 p4_;
};

// bind to function1
template <typename R,
  typename T1, typename T2, typename T3, typename T4>
class Bound4Functor1_Arg1 {
 public:
  Bound4Functor1_Arg1(Function<R, T1, T2, T3, T4> func
    , T2 p2, T3 p3, T4 p4) : function_(func)
      , p2_(p2), p3_(p3), p4_(p4) {}
  Bound4Functor1_Arg1(Bound4Functor1_Arg1 const & func)
    : function_(func.function_)
    , p2_(func.p2_), p3_(func.p3_), p4_(func.p4_) {}
  R operator() (T1 p1) {
    return function_(p1, p2_, p3_, p4_);
  }

 private:
  Function<R, T1, T2, T3, T4> function_;
  T2 p2_;
  T3 p3_;
  T4 p4_;
};

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
class Bound4Functor1_Arg2 {
 public:
  Bound4Functor1_Arg2(Function<R, T1, T2, T3, T4> func
    , T1 p1, T3 p3, T4 p4) : function_(func)
      , p1_(p1), p3_(p3), p4_(p4) {}
  Bound4Functor1_Arg2(Bound4Functor1_Arg2 const & func)
    : function_(func.function_)
    , p1_(func.p1_), p3_(func.p3_), p4_(func.p4_) {}
  R operator() (T2 p2) {
    return function_(p1_, p2, p3_, p4_);
  }

 private:
  Function<R, T1, T2, T3, T4> function_;
  T1 p1_;
  T3 p3_;
  T4 p4_;
};

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
class Bound4Functor1_Arg3 {
 public:
  Bound4Functor1_Arg3(Function<R, T1, T2, T3, T4> func
    , T1 p1, T2 p2, T4 p4) : function_(func)
      , p1_(p1), p2_(p2), p4_(p4) {}
  Bound4Functor1_Arg3(Bound4Functor1_Arg3 const & func)
    : function_(func.function_)
    , p1_(func.p1_), p2_(func.p2_), p4_(func.p4_) {}
  R operator() (T3 p3) {
    return function_(p1_, p2_, p3, p4_);
  }

 private:
  Function<R, T1, T2, T3, T4> function_;
  T1 p1_;
  T2 p2_;
  T4 p4_;
};

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
class Bound4Functor1_Arg4 {
 public:
  Bound4Functor1_Arg4(Function<R, T1, T2, T3, T4> func
    , T1 p1, T2 p2, T3 p3) : function_(func)
      , p1_(p1), p2_(p2), p3_(p3) {}
  Bound4Functor1_Arg4(Bound4Functor1_Arg4 const & func)
    : function_(func.function_)
    , p1_(func.p1_), p2_(func.p2_), p3_(func.p3_) {}
  R operator() (T4 p4) {
    return function_(p1_, p2_, p3_, p4);
  }

 private:
  Function<R, T1, T2, T3, T4> function_;
  T1 p1_;
  T2 p2_;
  T3 p3_;
};

} // namespace internal


using embb::base::internal::Nil;

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
class Function<R, T1, T2, T3, T4, Nil> {
 public:
  typedef internal::Function4<R, T1, T2, T3, T4> * FuncPtrType;
  Function() : function_(NULL) {}
  template <class C>
  explicit Function(C const & obj) {
    function_ = new(storage_)
      internal::FunctorWrapper4<C, R, T1, T2, T3, T4>(obj);
  }
  Function(Function const & func) {
    func.function_->CopyTo(&storage_[0]);
    function_ = reinterpret_cast<FuncPtrType>(&storage_[0]);
  }
  ~Function() {
    Free();
  }
  void operator = (R(*func)(T1, T2, T3, T4)) {
    Free();
    function_ = new(storage_)
      internal::FunctionPointer4<R, T1, T2, T3, T4>(func);
  }
  void operator = (Function & func) {
    Free();
    func.function_->CopyTo(&storage_[0]);
    function_ = reinterpret_cast<FuncPtrType>(&storage_[0]);
  }
  template <class C>
  void operator = (C const & obj) {
    Free();
    function_ = new(storage_)
      internal::FunctorWrapper4<C, R, T1, T2, T3, T4>(obj);
  }
  explicit Function(R(*func)(T1, T2, T3, T4)) {
    function_ = new(storage_)
      internal::FunctionPointer4<R, T1, T2, T3, T4>(func);
  }
  template <class C>
  Function(C & obj, R(C::*func)(T1, T2, T3, T4)) {
    function_ = new(storage_)
      internal::MemberFunctionPointer4<C, R, T1, T2, T3, T4>(obj, func);
  }
  R operator () (T1 p1, T2 p2, T3 p3, T4 p4) {
    return (*function_)(p1, p2, p3, p4);
  }

 private:
  char storage_[sizeof(
    internal::MemberFunctionPointer4<Nil, R, T1, T2, T3, T4>)];
  FuncPtrType function_;
  void Free() {
    if (NULL != function_) {
      function_->~Function4();
      function_ = NULL;
    }
  }
};

// wrap member function
template <class C, typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T1, T2, T3, T4> MakeFunction(C & obj,
  R(C::*func)(T1, T2, T3, T4)) {
  return Function<R, T1, T2, T3, T4>(obj, func);
}

// wrap function pointer
template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T1, T2, T3, T4> MakeFunction(
  R(*func)(T1, T2, T3, T4)) {
  return Function<R, T1, T2, T3, T4>(func);
}

// bind to function0
template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R> Bind(Function<R, T1, T2, T3, T4> func,
  T1 p1, T2 p2, T3 p3, T4 p4) {
  return Function<R>(
  internal::Bound4Functor0<R, T1, T2, T3, T4>(
    func, p1, p2, p3, p4));
}

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R> Bind(R(*func)(T1, T2, T3, T4),
  T1 p1, T2 p2, T3 p3, T4 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(func), p1, p2, p3, p4);
}

template <class C, typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R> Bind(C & obj, R(C::*func)(T1, T2, T3, T4),
  T1 p1, T2 p2, T3 p3, T4 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(obj, func), p1, p2, p3, p4);
}

// bind to function1
template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T1> Bind(Function<R, T1, T2, T3, T4> func
  , Placeholder::Arg_1, T2 p2, T3 p3, T4 p4) {
  return Function<R, T1>(
  internal::Bound4Functor1_Arg1<R, T1, T2, T3, T4>(func
    , p2, p3, p4));
}

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T1> Bind(R(*func)(T1, T2, T3, T4)
  , Placeholder::Arg_1 p1, T2 p2, T3 p3, T4 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(func), p1, p2, p3, p4);
}

template <class C, typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T1> Bind(C & obj, R(C::*func)(T1, T2, T3, T4)
  , Placeholder::Arg_1 p1, T2 p2, T3 p3, T4 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(obj, func), p1, p2, p3, p4);
}

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T2> Bind(Function<R, T1, T2, T3, T4> func
  , T1 p1, Placeholder::Arg_1, T3 p3, T4 p4) {
  return Function<R, T2>(
  internal::Bound4Functor1_Arg2<R, T1, T2, T3, T4>(func
    , p1, p3, p4));
}

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T2> Bind(R(*func)(T1, T2, T3, T4)
  , T1 p1, Placeholder::Arg_1 p2, T3 p3, T4 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(func), p1, p2, p3, p4);
}

template <class C, typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T2> Bind(C & obj, R(C::*func)(T1, T2, T3, T4)
  , T1 p1, Placeholder::Arg_1 p2, T3 p3, T4 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(obj, func), p1, p2, p3, p4);
}

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T3> Bind(Function<R, T1, T2, T3, T4> func
  , T1 p1, T2 p2, Placeholder::Arg_1, T4 p4) {
  return Function<R, T3>(
  internal::Bound4Functor1_Arg3<R, T1, T2, T3, T4>(func
    , p1, p2, p4));
}

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T3> Bind(R(*func)(T1, T2, T3, T4)
  , T1 p1, T2 p2, Placeholder::Arg_1 p3, T4 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(func), p1, p2, p3, p4);
}

template <class C, typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T3> Bind(C & obj, R(C::*func)(T1, T2, T3, T4)
  , T1 p1, T2 p2, Placeholder::Arg_1 p3, T4 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(obj, func), p1, p2, p3, p4);
}

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T4> Bind(Function<R, T1, T2, T3, T4> func
  , T1 p1, T2 p2, T3 p3, Placeholder::Arg_1) {
  return Function<R, T4>(
  internal::Bound4Functor1_Arg4<R, T1, T2, T3, T4>(func
    , p1, p2, p3));
}

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T4> Bind(R(*func)(T1, T2, T3, T4)
  , T1 p1, T2 p2, T3 p3, Placeholder::Arg_1 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(func), p1, p2, p3, p4);
}

template <class C, typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T4> Bind(C & obj, R(C::*func)(T1, T2, T3, T4)
  , T1 p1, T2 p2, T3 p3, Placeholder::Arg_1 p4) {
  return Bind(Function<R, T1, T2, T3, T4>(obj, func), p1, p2, p3, p4);
}

// bind to Function4
template <class C, typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T1, T2, T3, T4> Bind(
  C & obj,
  R(C::*func)(T1, T2, T3, T4),
  Placeholder::Arg_1 p1,
  Placeholder::Arg_2 p2,
  Placeholder::Arg_3 p3,
  Placeholder::Arg_4 p4) {
  return Function<R, T1, T2, T3, T4>(obj, func);
}

template <typename R,
  typename T1, typename T2, typename T3, typename T4>
Function<R, T1, T2, T3, T4> Bind(
  R(*func)(T1, T2, T3, T4),
  Placeholder::Arg_1 p1,
  Placeholder::Arg_2 p2,
  Placeholder::Arg_3 p3,
  Placeholder::Arg_4 p4) {
  return Function<R, T1, T2, T3, T4>(func);
}

} // namespace base
} // namespace embb

#endif // EMBB_BASE_INTERNAL_FUNCTION4_H_