/*
 * Copyright 2013-2019 Software Radio Systems Limited
 *
 * This file is part of srsLTE.
 *
 * srsLTE is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * srsLTE is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * A copy of the GNU Affero General Public License can be found in
 * the LICENSE file in the top-level directory of this distribution
 * and at http://www.gnu.org/licenses/.
 *
 */

#include "srslte/common/stack_procedure.h"
#include <iostream>

#define TESTASSERT(cond)                                                                                               \
  {                                                                                                                    \
    if (!(cond)) {                                                                                                     \
      std::cout << "[" << __FUNCTION__ << "][Line " << __LINE__ << "]: FAIL at " << (#cond) << std::endl;              \
      return -1;                                                                                                       \
    }                                                                                                                  \
  }

using srslte::proc_outcome_t;

enum class obj_state_t { default_ctor, move_ctor, copy_ctor, from_move_ctor, from_move_assign };

const char* to_string(obj_state_t o)
{
  switch (o) {
    case obj_state_t::default_ctor:
      return "default_ctor";
    case obj_state_t::move_ctor:
      return "move_ctor";
    case obj_state_t::copy_ctor:
      return "copy_ctor";
    case obj_state_t::from_move_ctor:
      return "from_move_ctor";
    case obj_state_t::from_move_assign:
      return "from_move_assign";
  }
  return "";
}

class TestObj
{
public:
  TestObj() = default;
  TestObj(const TestObj& other)
  {
    printf("TestObj copy ctor called: {%s,%d} <- {%s,%d}!!!\n", to_string(state), id, to_string(other.state), other.id);
    id    = other.id;
    state = obj_state_t::copy_ctor;
    copy_counter++;
  }
  TestObj(TestObj&& other) noexcept
  {
    printf("TestObj move ctor called: {%s,%d} <- {%s,%d}!!!\n", to_string(state), id, to_string(other.state), other.id);
    id          = other.id;
    state       = obj_state_t::move_ctor;
    other.state = obj_state_t::from_move_ctor;
    move_counter++;
  }
  ~TestObj()
  {
    dtor_counter++;
    printf("TestObj {%s,%d} dtor called!!!\n", to_string(state), id);
  }
  TestObj& operator=(const TestObj& other)
  {
    printf("TestObj copy operator called: {%s,%d} <- {%s,%d}!!!\n",
           to_string(state),
           id,
           to_string(other.state),
           other.id);
    id    = other.id;
    state = other.state;
    copy_counter++;
    return *this;
  }
  TestObj& operator=(TestObj&& other) noexcept
  {
    printf("TestObj move operator called: {%s,%d} <- {%s,%d}!!!\n",
           to_string(state),
           id,
           to_string(other.state),
           other.id);
    if (&other != this) {
      id          = other.id;
      state       = other.state;
      other.state = obj_state_t::from_move_assign;
      move_counter++;
    }
    return *this;
  }

  obj_state_t state = obj_state_t::default_ctor;
  int         id    = 0;
  static int  copy_counter;
  static int  move_counter;
  static int  dtor_counter;
};

int TestObj::copy_counter = 0;
int TestObj::move_counter = 0;
int TestObj::dtor_counter = 0;

void new_test()
{
  TestObj::copy_counter = 0;
  TestObj::move_counter = 0;
  TestObj::dtor_counter = 0;
}

class custom_proc : public srslte::proc_impl_t
{
public:
  proc_outcome_t init(int a_)
  {
    if (a_ < 0) {
      printf("Failed to initiate custom_proc\n");
      return proc_outcome_t::error;
    }
    obj.id = a_;
    return proc_outcome_t::yield;
  }
  proc_outcome_t step() final
  {
    if (counter++ > 5) {
      return proc_outcome_t::success;
    }
    return proc_outcome_t::yield;
  }
  void stop() final { printf("TestObj %d stop() was called\n", obj.id); }
  const char* name() const { return "custom proc"; }

  TestObj obj;

private:
  int counter = 0;
};

int test_local_1()
{
  new_test();
  printf("\n--- Test %s ---\n", __func__);
  srslte::proc_t<custom_proc> proc;
  TESTASSERT(not proc.is_active());

  proc.launch(1);
  TESTASSERT(proc.is_active());
  TESTASSERT(not proc.is_complete());

  while (proc.run()) {
  }

  TESTASSERT(proc.is_active());
  TESTASSERT(proc.is_complete());

  printf("pop being called\n");
  custom_proc procobj = proc.pop();
  TESTASSERT(procobj.obj.id == 1);
  TESTASSERT(procobj.is_success());
  TESTASSERT(not proc.is_active());
  TESTASSERT(proc.get()->obj.id == 0); // Proc is ready to be reused

  printf("EXIT\n");
  TESTASSERT(TestObj::copy_counter == 0);
  TESTASSERT(TestObj::move_counter == 2); // pop() makes a swap which causes 2 moves
  TESTASSERT(TestObj::dtor_counter == 2); // 2 dtors inside pop() (handler and popped obj not yet destructed)
  return 0;
}

int test_callback_1()
{
  new_test();
  printf("\n--- Test %s ---\n", __func__);
  srslte::callback_list_t     callbacks;
  srslte::proc_t<custom_proc> proc;
  TESTASSERT(not proc.is_active());

  proc.launch(2);
  callbacks.add_proc(proc); // We have to call pop() explicitly to take the result
  TESTASSERT(callbacks.size() == 1);

  while (callbacks.size() > 0) {
    TESTASSERT(proc.is_active());
    TESTASSERT(not proc.is_complete());
    callbacks.run();
  }
  TESTASSERT(proc.is_active());
  TESTASSERT(proc.is_complete());

  printf("pop being called\n");
  custom_proc procobj = proc.pop();
  TESTASSERT(procobj.is_success());
  TESTASSERT(procobj.obj.id == 2);
  TESTASSERT(not proc.is_active());

  TESTASSERT(proc.get()->obj.id == 0); // Proc is ready to be reused
  printf("EXIT\n");
  TESTASSERT(TestObj::copy_counter == 0);
  TESTASSERT(TestObj::move_counter == 2); // pop makes two moves
  TESTASSERT(TestObj::dtor_counter == 2); // handler not yet destructed
  return 0;
}

int test_callback_2()
{
  new_test();
  printf("\n--- Test %s ---\n", __func__);
  srslte::callback_list_t     callbacks;
  srslte::proc_t<custom_proc> proc;
  TESTASSERT(not proc.is_active());

  proc.launch(3);
  TESTASSERT(proc.is_active());
  TESTASSERT(not proc.is_complete());
  callbacks.consume_proc(std::move(proc));
  TESTASSERT(callbacks.size() == 1);

  while (callbacks.size() > 0) {
    callbacks.run();
  }
  // since the proc was consumed, it is erased without the need for pop()

  printf("EXIT\n");
  TESTASSERT(TestObj::copy_counter == 0);
  TESTASSERT(TestObj::move_counter == 0); // no pop()
  TESTASSERT(TestObj::dtor_counter == 1); // handler not yet destructed
  return 0;
}

int test_callback_3()
{
  new_test();
  printf("\n--- Test %s ---\n", __func__);
  srslte::callback_list_t     callbacks;
  srslte::proc_t<custom_proc> proc;
  TESTASSERT(not proc.is_active());

  proc.launch(4);
  TESTASSERT(proc.is_active());
  TESTASSERT(not proc.is_complete());
  callbacks.defer_proc(proc); // we still have access to proc, but we do not need to call pop()
  TESTASSERT(callbacks.size() == 1);
  TESTASSERT(proc.is_active());
  TESTASSERT(not proc.is_complete());

  while (callbacks.size() > 0) {
    TESTASSERT(proc.is_active());
    TESTASSERT(not proc.is_complete());
    callbacks.run();
  }
  TESTASSERT(not proc.is_active());
  TESTASSERT(not proc.is_complete());
  TESTASSERT(proc.get()->obj.id == 0); // Proc is ready to be reused

  printf("EXIT\n");
  TESTASSERT(TestObj::copy_counter == 0);
  TESTASSERT(TestObj::move_counter == 0);
  TESTASSERT(TestObj::dtor_counter == 1); // handler not yet destructed
  return 0;
}

int test_callback_4()
{
  new_test();
  printf("\n--- Test %s ---\n", __func__);
  srslte::callback_list_t callbacks;
  int*                    counter = new int(5);

  {
    callbacks.defer_task([counter]() {
      printf("current counter=%d\n", *counter);
      if (--(*counter) == 0) {
        return proc_outcome_t::success;
      }
      return proc_outcome_t::yield;
    });
  }

  while (callbacks.size() > 0) {
    callbacks.run();
  }

  //  printf("counter=%d\n", counter);
  TESTASSERT(*counter == 0);
  delete counter;

  return 0;
}

int main()
{
  TESTASSERT(test_local_1() == 0);
  TESTASSERT(test_callback_1() == 0);
  TESTASSERT(test_callback_2() == 0);
  TESTASSERT(test_callback_3() == 0);
  TESTASSERT(test_callback_4() == 0);

  return 0;
}