#include "CallGraphDrawer.h"
#include "GraphPrinters.h"
#include "llvm/PassManager.h"
#include "TVTreeItem.h"
using namespace llvm;

//===----------------------------------------------------------------------===//

// CallGraphDrawer implementation

wxImage *CallGraphDrawer::drawModuleGraph (Module *module) {
  PassManager PM;
  PM.add (createCallGraphPrinterPass ());
  PM.run (*module);
  return buildwxImageFromDotFile ("callgraph.dot");
}

std::string CallGraphDrawer::getDisplayTitle (TVTreeItemData *item) {
  std::string title ("Call graph");
  if (item) title += " of " + item->getTitle ();
  return title;
}