X-Git-Url: http://bilbo.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/138a767165df006f5ae2a28a5d0db79f06c6a7fb..417ed3b671abe3a71fa4106d23d0a432084cc207:/include/simgrid/s4u/Io.hpp

diff --git a/include/simgrid/s4u/Io.hpp b/include/simgrid/s4u/Io.hpp
index 1625419e1f..8c626f774c 100644
--- a/include/simgrid/s4u/Io.hpp
+++ b/include/simgrid/s4u/Io.hpp
@@ -1,4 +1,4 @@
-/* Copyright (c) 2017-2018. The SimGrid Team. All rights reserved.          */
+/* Copyright (c) 2017-2023. 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. */
@@ -9,37 +9,62 @@
 #include <simgrid/forward.h>
 #include <simgrid/s4u/Activity.hpp>
 
-#include <atomic>
+#include <string>
 
 namespace simgrid {
 namespace s4u {
 
-class XBT_PUBLIC Io : public Activity {
-  Io() : Activity() {}
-public:
-  friend XBT_PUBLIC void intrusive_ptr_release(simgrid::s4u::Io* i);
-  friend XBT_PUBLIC void intrusive_ptr_add_ref(simgrid::s4u::Io* i);
-  friend Storage; // Factory of IOs
+/** I/O Activity, representing the asynchronous disk access.
+ *
+ * They are generated from Disk::io_init(), Disk::read() Disk::read_async(), Disk::write() and Disk::write_async().
+ */
+
+class XBT_PUBLIC Io : public Activity_T<Io> {
+#ifndef DOXYGEN
+  friend kernel::activity::IoImpl;
+  friend kernel::EngineImpl;
+#endif
+
+  static xbt::signal<void(Io const&)> on_start;
+
+protected:
+  explicit Io(kernel::activity::IoImplPtr pimpl);
+  Io* do_start() override;
 
+public:
   enum class OpType { READ, WRITE };
-  ~Io() = default;
-
-  Activity* start() override;
-  Activity* wait() override;
-  Activity* wait(double timeout) override;
-  Activity* cancel() override;
-
-  double get_remaining() override;
-  IoPtr set_io_type(OpType type);
-
-private:
-  sg_size_t size_       = 0;
-  sg_storage_t storage_ = nullptr;
-  std::string name_     = "";
-  OpType type_          = OpType::READ;
-  std::atomic_int_fast32_t refcount_{0};
-}; // class
-}
-}; // Namespace simgrid::s4u
+
+  static void on_start_cb(const std::function<void(Io const&)>& cb) { on_start.connect(cb); }
+
+  static IoPtr init();
+  /*! take a vector of s4u::IoPtr and return when one of them is finished.
+   * The return value is the rank of the first finished IoPtr. */
+  static ssize_t wait_any(const std::vector<IoPtr>& ios) { return wait_any_for(ios, -1); }
+  /*! Same as wait_any, but with a timeout. If the timeout occurs, parameter last is returned.*/
+  static ssize_t wait_any_for(const std::vector<IoPtr>& ios, double timeout);
+
+  double get_remaining() const override;
+  sg_size_t get_performed_ioops() const;
+  IoPtr set_disk(const_sg_disk_t disk);
+  IoPtr set_priority(double priority);
+  IoPtr set_size(sg_size_t size);
+  IoPtr set_op_type(OpType type);
+
+  static IoPtr streamto_init(Host* from, const Disk* from_disk, Host* to, const Disk* to_disk);
+  static IoPtr streamto_async(Host* from, const Disk* from_disk, Host* to, const Disk* to_disk,
+                              uint64_t simulated_size_in_bytes);
+  static void streamto(Host* from, const Disk* from_disk, Host* to, const Disk* to_disk,
+                       uint64_t simulated_size_in_bytes);
+
+  IoPtr set_source(Host* from, const Disk* from_disk);
+  IoPtr set_destination(Host* to, const Disk* to_disk);
+
+  IoPtr update_priority(double priority);
+
+  bool is_assigned() const override;
+};
+
+} // namespace s4u
+} // namespace simgrid
 
 #endif /* SIMGRID_S4U_IO_HPP */