/*!
* get CallGraph edge via nodes
*/
PTACallGraphEdge* PTACallGraph::getGraphEdge(PTACallGraphNode* src, PTACallGraphNode* dst,PTACallGraphEdge::CEDGEK kind) {
for (PTACallGraphEdge::CallGraphEdgeSet::iterator iter = src->OutEdgeBegin();
iter != src->OutEdgeEnd(); ++iter) {
PTACallGraphEdge* edge = (*iter);
if (edge->getEdgeKind() == kind && edge->getDstID() == dst->getId())
return edge;
}
return NULL;
}
static std::string getEdgeAttributes(PTACallGraphNode *node, EdgeIter EI, PTACallGraph *PTACallGraph) {
//TODO: mark indirect call of Fork with different color
PTACallGraphEdge* edge = *(EI.getCurrent());
assert(edge && "No edge found!!");
std::string color;
if (edge->getEdgeKind() == PTACallGraphEdge::TDJoinEdge) {
color = "color=green";
} else if (edge->getEdgeKind() == PTACallGraphEdge::TDForkEdge) {
color = "color=blue";
} else {
color = "color=black";
}
if (0 != edge->getIndirectCalls().size()) {
color = "color=red";
}
return color;
}
