// This example shows how to compose a taskflow
#include <taskflow/taskflow.hpp>

void composition_example_1() {

 std::cout << "Composition example 1\n";

 tf::Executor executor;

 // f1 has three independent tasks
 tf::Taskflow f1("F1");
 auto f1A = f1.emplace([](){ std::cout << "F1 TaskA\n"; });
 auto f1B = f1.emplace([](){ std::cout << "F1 TaskB\n"; });
 auto f1C = f1.emplace([](){ std::cout << "F1 TaskC\n"; });
 f1A.name("f1A");
 f1B.name("f1B");
 f1C.name("f1C");
 f1A.precede(f1C);
 f1B.precede(f1C);

 // f2A ---
 // |----> f2C ----> f1_module_task ----> f2D
 // f2B ---
 tf::Taskflow f2("F2");
 auto f2A = f2.emplace([](){ std::cout << " F2 TaskA\n"; });
 auto f2B = f2.emplace([](){ std::cout << " F2 TaskB\n"; });
 auto f2C = f2.emplace([](){ std::cout << " F2 TaskC\n"; });
 auto f2D = f2.emplace([](){ std::cout << " F2 TaskD\n"; });
 f2A.name("f2A");
 f2B.name("f2B");
 f2C.name("f2C");
 f2D.name("f2D");

 f2A.precede(f2C);
 f2B.precede(f2C);

 tf::Task f1_module_task = f2.composed_of(f1);
 f1_module_task.name("module");
 f2C.precede(f1_module_task);
 f1_module_task.precede(f2D);

 f2.dump(std::cout);

 executor.run_n(f2, 3).get();
}

void composition_example_2() {

 std::cout << "Composition example 2\n";

 tf::Executor executor;

 // f1 has two independent tasks
 tf::Taskflow f1("F1");
 auto f1A = f1.emplace([&](){ std::cout << "F1 TaskA\n"; });
 auto f1B = f1.emplace([&](){ std::cout << "F1 TaskB\n"; });
 f1A.name("f1A");
 f1B.name("f1B");

 // f2A ---
 // |----> f2C
 // f2B ---
 //
 // f1_module_task
 tf::Taskflow f2("F2");
 auto f2A = f2.emplace([&](){ std::cout << " F2 TaskA\n"; });
 auto f2B = f2.emplace([&](){ std::cout << " F2 TaskB\n"; });
 auto f2C = f2.emplace([&](){ std::cout << " F2 TaskC\n"; });
 f2A.name("f2A");
 f2B.name("f2B");
 f2C.name("f2C");

 f2A.precede(f2C);
 f2B.precede(f2C);
 f2.composed_of(f1).name("module_of_f1");

 // f3 has a module task (f2) and a regular task
 tf::Taskflow f3("F3");
 f3.composed_of(f2).name("module_of_f2");
 f3.emplace([](){ std::cout << " F3 TaskA\n"; }).name("f3A");

 // f4: f3_module_task -> f2_module_task
 tf::Taskflow f4;
 f4.name("F4");
 auto f3_module_task = f4.composed_of(f3).name("module_of_f3");
 auto f2_module_task = f4.composed_of(f2).name("module_of_f2");
 f3_module_task.precede(f2_module_task);

 f4.dump(std::cout);

 executor.run_until(
 f4,
 [iter = 1] () mutable { std::cout << '\n'; return iter-- == 0; },
 [](){ std::cout << "First run_until finished\n"; }
 ).get();

 executor.run_until(
 f4,
 [iter = 2] () mutable { std::cout << '\n'; return iter-- == 0; },
 [](){ std::cout << "Second run_until finished\n"; }
 );

 executor.run_until(
 f4,
 [iter = 3] () mutable { std::cout << '\n'; return iter-- == 0; },
 [](){ std::cout << "Third run_until finished\n"; }
 ).get();
}

int main(){
 composition_example_1();
 composition_example_2();
 return 0;
}