| 22 | namespace { |
| 23 | |
| 24 | const OpNode* OpNodeOnEdge(TaskEdge* edge, TaskNode* (TaskEdge::*GetNode)() const, |
| 25 | void (TaskNode::*ForEachDataEdge)(const std::function<void(TaskEdge*)>&) |
| 26 | const) { |
| 27 | CompTaskNode* target_node = nullptr; |
| 28 | do { |
| 29 | TaskNode* tmp_node = (edge->*GetNode)(); |
| 30 | target_node = dynamic_cast<CompTaskNode*>(tmp_node); |
| 31 | edge = nullptr; |
| 32 | (tmp_node->*ForEachDataEdge)([&](TaskEdge* e) { |
| 33 | if (edge == nullptr) { edge = e; } |
| 34 | }); |
| 35 | } while (!target_node && edge); |
| 36 | if (target_node) { return target_node->op_node(); } |
| 37 | return nullptr; |
| 38 | } |
| 39 | |
| 40 | std::vector<CompTaskNode*> GetCompTaskNodesOnEdge( |
| 41 | TaskEdge* edge, TaskNode* (TaskEdge::*GetNode)() const, |
no test coverage detected