examples/python/activityset-waitany/activityset-waitany.py

   1 # Copyright (c) 2017-2023. The SimGrid Team. All rights reserved.
   2 #
   3 # This program is free software you can redistribute it and/or modify it
   4 # under the terms of the license (GNU LGPL) which comes with this package.
   5
   6 """
   7 Usage: activityset-waitany.py platform_file [other parameters]
   8 """
   9
  10 import sys
  11 from simgrid import Actor, ActivitySet, Engine, Comm, Exec, Io, Host, Mailbox, this_actor
  12
  13 def bob():
  14   mbox = Mailbox.by_name("mbox")
  15   disk = Host.current().get_disks()[0]
  16
  17   this_actor.info("Create my asynchronous activities")
  18   exec = this_actor.exec_async(5e9)
  19   comm, payload = mbox.get_async()
  20   io   = disk.read_async(300000000)
  21
  22   pending_activities = ActivitySet([exec, comm])
  23   pending_activities.push(io) # Activities can be pushed after creation, too
  24
  25   this_actor.info("Wait for asynchronous activities to complete")
  26   while not pending_activities.empty():
  27     completed_one = pending_activities.wait_any()
  28
  29     if isinstance(completed_one, Comm):
  30       this_actor.info("Completed a Comm")
  31     elif isinstance(completed_one, Exec):
  32       this_actor.info("Completed an Exec")
  33     elif isinstance(completed_one, Io):
  34       this_actor.info("Completed an I/O")
  35
  36   this_actor.info("Last activity is complete")
  37
  38 def alice():
  39   this_actor.info("Send 'Message'")
  40   Mailbox.by_name("mbox").put("Message", 600000000)
  41
  42 if __name__ == '__main__':
  43   e = Engine(sys.argv)
  44   e.set_log_control("root.fmt:[%4.6r]%e[%5a]%e%m%n")
  45
  46   # Load the platform description
  47   e.load_platform(sys.argv[1])
  48
  49   Actor.create("bob",   Host.by_name("bob"), bob)
  50   Actor.create("alice", Host.by_name("alice"), alice)
  51
  52   e.run()