/**
 This program demonstrates how to traverse a graph using task visitor.

 We first create four tasks: A, B, C, and D, where task A runs before B and C,
 and task D runs after B and C. During the execution of B, it spawns another subflow
 graph of three tasks: B1, B2, and B3, where B3 runs after B1 and B2.
 Upon completion of the subflow, it joins its parent task B.
 
 By default, subflows are automatically cleaned up when they finish to avoid memory explosion. 
 In this example, since we would like to inspect the spawned subflow,
 we disable this behavior by calling `tf::Subflow::retain(true)`.
 
 Note that we must run the subflow once for it to be created.
*/

#include <taskflow/taskflow.hpp>

int main() {

  // Create a taskflow graph with three static tasks and one subflow task.
  tf::Taskflow taskflow("visitor");
  tf::Executor executor;

  auto A = taskflow.emplace([]() { std::cout << "TaskA\n"; });
  auto B = taskflow.emplace([](tf::Subflow& subflow) {
    std::cout << "TaskB is spawning B1, B2, and B3 ...\n";
    auto B1 = subflow.emplace([&](){ printf("  Subtask B1\n"); }).name("B1");
    auto B2 = subflow.emplace([&](){ printf("  Subtask B2\n"); }).name("B2");
    auto B3 = subflow.emplace([&](){ printf("  Subtask B3\n"); }).name("B3");
    B1.precede(B3);
    B2.precede(B3);
    subflow.retain(true);  // retains the subflow
  });

  auto C = taskflow.emplace([] () { std::cout << "TaskC\n"; });
  auto D = taskflow.emplace([] () { std::cout << "TaskD\n"; });
  A.name("A");
  B.name("B");
  C.name("C");
  D.name("D");

  A.precede(B);  // B runs after A
  A.precede(C);  // C runs after A
  B.precede(D);  // D runs after B
  C.precede(D);  // D runs after C

  executor.run(taskflow).wait();

  // examine the graph
  taskflow.dump(std::cout);

  // traverse all tasks in the taskflow
  taskflow.for_each_task([](tf::Task task){
    std::cout << "task " << task.name() << " [type=" << tf::to_string(task.type()) << "]\n";
    // traverse it's successor
    task.for_each_successor([](tf::Task successor) {
      std::cout << "  -> successor   task " << successor.name() << '\n'; 
    });
    // traverse it's predecessor
    task.for_each_predecessor([](tf::Task predecessor) {
      std::cout << "  <- predecessor task " << predecessor.name() << '\n'; 
    });

    // traverse the subflow (in our example, task B)
    task.for_each_subflow_task([](tf::Task stask){
      std::cout << "  subflow task " << stask.name() << '\n';
      // traverse it's successor
      stask.for_each_successor([](tf::Task successor) {
        std::cout << "    -> successor   task " << successor.name() << '\n'; 
      });
      // traverse it's predecessor
      stask.for_each_predecessor([](tf::Task predecessor) {
        std::cout << "    <- predecessor task " << predecessor.name() << '\n'; 
      });
    });
  });

  return 0;
}