+static s4u::VirtualMachine* get_vm_from_task(kernel::activity::ActivityImpl const& task)
+{
+ auto* exec = dynamic_cast<kernel::activity::ExecImpl const*>(&task);
+ return exec != nullptr ? dynamic_cast<s4u::VirtualMachine*>(exec->host_) : nullptr;
+}
+
+static void add_active_task(kernel::activity::ActivityImpl const& task)
+{
+ s4u::VirtualMachine* vm = get_vm_from_task(task);
+ if (vm != nullptr) {
+ VirtualMachineImpl *vm_impl = vm->get_impl();
+ vm_impl->active_tasks_ = vm_impl->active_tasks_ + 1;
+ vm_impl->update_action_weight();
+ }
+}
+
+static void remove_active_task(kernel::activity::ActivityImpl const& task)
+{
+ s4u::VirtualMachine* vm = get_vm_from_task(task);
+ if (vm != nullptr) {
+ VirtualMachineImpl *vm_impl = vm->get_impl();
+ vm_impl->active_tasks_ = vm_impl->active_tasks_ - 1;
+ vm_impl->update_action_weight();
+ }
+}
+