/* Copyright (c) 2007-2020. The SimGrid Team. All rights reserved.          */

/* This program is free software; you can redistribute it and/or modify it
 * under the terms of the license (GNU LGPL) which comes with this package. */

#ifndef SIMGRID_KERNEL_ACTOR_ACTORIMPL_HPP
#define SIMGRID_KERNEL_ACTOR_ACTORIMPL_HPP

#include "simgrid/s4u/Actor.hpp"
#include "src/simix/popping_private.hpp"
#include "xbt/PropertyHolder.hpp"
#include <boost/intrusive/list.hpp>
#include <functional>
#include <list>
#include <map>
#include <memory>

namespace simgrid {
namespace kernel {
namespace actor {

class XBT_PUBLIC ActorImpl : public xbt::PropertyHolder {
  s4u::Host* host_   = nullptr; /* the host on which the actor is running */
  // XBT_DEPRECATED_v329
  void* userdata_    = nullptr; /* kept for compatibility, it should be replaced with moddata */
  aid_t pid_         = 0;
  aid_t ppid_        = -1;
  bool daemon_       = false; /* Daemon actors are automatically killed when the last non-daemon leaves */
  bool auto_restart_ = false;

public:
  xbt::string name_;
  ActorImpl(xbt::string name, s4u::Host* host);
  ActorImpl(const ActorImpl&) = delete;
  ActorImpl& operator=(const ActorImpl&) = delete;
  ~ActorImpl();

  static ActorImpl* self();
  double get_kill_time();
  void set_kill_time(double kill_time);
  boost::intrusive::list_member_hook<> host_actor_list_hook;   /* simgrid::simix::Host::process_list */
  boost::intrusive::list_member_hook<> smx_destroy_list_hook;  /* simix_global->actors_to_destroy */
  boost::intrusive::list_member_hook<> smx_synchro_hook;       /* {mutex,cond,sem}->sleeping */

  const xbt::string& get_name() const { return name_; }
  const char* get_cname() const { return name_.c_str(); }

  // Accessors to private fields
  s4u::Host* get_host() { return host_; }
  void set_host(s4u::Host* dest);
  // XBT_DEPRECATED_v329
  void* get_user_data() { return userdata_; }
  // XBT_DEPRECATED_v329
  void set_user_data(void* data) { userdata_ = data; }
  aid_t get_pid() const { return pid_; }
  aid_t get_ppid() const { return ppid_; }
  void set_ppid(aid_t ppid) { ppid_ = ppid; }
  bool is_daemon() { return daemon_; } /** Whether this actor has been daemonized */
  bool has_to_auto_restart() { return auto_restart_; }
  void set_auto_restart(bool autorestart) { auto_restart_ = autorestart; }

  std::unique_ptr<context::Context> context_; /* the context (uctx/raw/thread) that executes the user function */

  std::exception_ptr exception_;
  bool finished_  = false;
  bool suspended_ = false;

  activity::ActivityImplPtr waiting_synchro = nullptr; /* the current blocking synchro if any */
  std::list<activity::ActivityImplPtr> comms;          /* the current non-blocking communication synchros */
  s_smx_simcall simcall;
  /* list of functions executed when the process dies */
  std::shared_ptr<std::vector<std::function<void(bool)>>> on_exit =
      std::make_shared<std::vector<std::function<void(bool)>>>();

  std::function<void()> code_;
  simix::Timer* kill_timer = nullptr;

private:
  /* Refcounting */
  std::atomic_int_fast32_t refcount_{0};

public:
  int get_refcount() { return refcount_; }
  friend void intrusive_ptr_add_ref(ActorImpl* actor)
  {
    // This whole memory consistency semantic drives me nuts.
    // std::memory_order_relaxed proves to not be enough: There is a threading issue when actors commit suicide.
    //   My guess is that the maestro context wants to propagate changes to the actor's fields after the
    //   actor context frees that memory area or something. But I'm not 100% certain of what's going on.
    // std::memory_order_seq_cst works but that's rather demanding.
    // AFAIK, std::memory_order_acq_rel works on all tested platforms, so let's stick to it.
    // Reducing the requirements to _relaxed would require to fix our suicide procedure, which is a messy piece of code.
    actor->refcount_.fetch_add(1, std::memory_order_acq_rel);
  }
  friend void intrusive_ptr_release(ActorImpl* actor)
  {
    // inspired from http://www.boost.org/doc/libs/1_55_0/doc/html/atomic/usage_examples.html
    if (actor->refcount_.fetch_sub(1, std::memory_order_release) == 1) {
      // Make sure that any changes done on other threads before their acquire are committed before our delete
      // http://stackoverflow.com/questions/27751025/why-is-an-acquire-barrier-needed-before-deleting-the-data-in-an-atomically-refer
      std::atomic_thread_fence(std::memory_order_acquire);
      delete actor;
    }
  }

  /* S4U/implem interfaces */
private:
  s4u::Actor piface_; // Our interface is part of ourselves

  void cleanup_from_simix();
  void undaemonize();

public:
  s4u::ActorPtr iface() { return s4u::ActorPtr(&piface_); }
  s4u::Actor* ciface() { return &piface_; }

  ActorImplPtr init(const std::string& name, s4u::Host* host);
  ActorImpl* start(const simix::ActorCode& code);

  static ActorImplPtr create(const std::string& name, const simix::ActorCode& code, void* data, s4u::Host* host,
                             const std::unordered_map<std::string, std::string>* properties, ActorImpl* parent_actor);
  static ActorImplPtr attach(const std::string& name, void* data, s4u::Host* host,
                             const std::unordered_map<std::string, std::string>* properties);
  static void detach();
  void cleanup();
  void exit();
  void kill(ActorImpl* actor);
  void kill_all();

  void yield();
  void daemonize();
  bool is_suspended() { return suspended_; }
  s4u::Actor* restart();
  void suspend();
  void resume();
  activity::ActivityImplPtr join(const ActorImpl* actor, double timeout);
  activity::ActivityImplPtr sleep(double duration);
  /** Ask the actor to throw an exception right away */
  void throw_exception(std::exception_ptr e);

  /** execute the pending simcall -- must be called from the maestro context */
  void simcall_handle(int value);
  /** Terminates a simcall currently executed in maestro context. The actor will be restarted in the next scheduling
   * round */
  void simcall_answer();
};

class ProcessArg {
public:
  std::string name;
  std::function<void()> code;
  void* data                                                               = nullptr;
  s4u::Host* host                                                          = nullptr;
  double kill_time                                                         = 0.0;
  std::shared_ptr<const std::unordered_map<std::string, std::string>> properties = nullptr;
  bool auto_restart                                                        = false;
  bool daemon_                                                             = false;
  /* list of functions executed when the process dies */
  const std::shared_ptr<std::vector<std::function<void(bool)>>> on_exit;

  ProcessArg()                                                             = default;

  explicit ProcessArg(const std::string& name, const std::function<void()>& code, void* data, s4u::Host* host,
                      double kill_time, std::shared_ptr<std::unordered_map<std::string, std::string>> properties,
                      bool auto_restart)
      : name(name)
      , code(code)
      , data(data)
      , host(host)
      , kill_time(kill_time)
      , properties(properties)
      , auto_restart(auto_restart)
  {
  }

  explicit ProcessArg(s4u::Host* host, ActorImpl* actor)
      : name(actor->get_name())
      , code(actor->code_)
      , data(actor->get_user_data())
      , host(host)
      , kill_time(actor->get_kill_time())
      , auto_restart(actor->has_to_auto_restart())
      , daemon_(actor->is_daemon())
      , on_exit(actor->on_exit)
  {
    properties.reset(actor->get_properties(), [](decltype(actor->get_properties())) {});
  }
};

/* Used to keep the list of actors blocked on a synchro  */
typedef boost::intrusive::list<ActorImpl, boost::intrusive::member_hook<ActorImpl, boost::intrusive::list_member_hook<>,
                                                                        &ActorImpl::smx_synchro_hook>>
    SynchroList;

XBT_PUBLIC void create_maestro(const std::function<void()>& code);
XBT_PUBLIC int get_maxpid();
} // namespace actor
} // namespace kernel
} // namespace simgrid

extern void (*SMPI_switch_data_segment)(simgrid::s4u::ActorPtr actor);

#endif
