//===-- DSGraphDrawer.cpp - DSGraph viewing ----------------------*- C++ -*-==//
//
// Classes for viewing DataStructure analysis graphs
//
//===----------------------------------------------------------------------===//

#include "DSAGraphDrawer.h"
#include "TVTreeItem.h"
#include "GraphPrinters.h"
#include "llvm/Function.h"
#include "llvm/Module.h"
#include "llvm/Pass.h"
#include "llvm/PassManager.h"
#include "llvm/Target/TargetData.h"
using namespace llvm;

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

// DSGraphDrawer implementation
wxImage *DSGraphDrawer::drawModuleGraph(Module *M) {
  PassManager PM;
  PM.add(new TargetData(M));
  PM.add(getModulePass());
  PM.run(*M);
  return buildwxImageFromDotFile(getFilename(M));
}

wxImage *DSGraphDrawer::drawFunctionGraph(Function *F) {
  Module *M = F->getParent();
  PassManager PM;
  PM.add(new TargetData(M));
  PM.add(getFunctionPass(F));
  PM.run(*M);
  return buildwxImageFromDotFile(getFilename(F));
}

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

// BUGraphDrawer implementation
Pass *BUGraphDrawer::getFunctionPass(Function *F) {
  return createBUDSFunctionPrinterPass(F);
}

Pass *BUGraphDrawer::getModulePass() {
  return createBUDSModulePrinterPass();
}

std::string BUGraphDrawer::getFilename(Function *F) {
  return "buds." + F->getName().str() + ".dot";
}

std::string BUGraphDrawer::getFilename(Module *M) {
  return "buds.dot";
}

std::string BUGraphDrawer::getDisplayTitle (TVTreeItemData *item) {
  if (item)
    return "Bottom-up data structure " + item->dsGraphName ();
  else
    return "Bottom-up DSGraph";
}

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

// TDGraphDrawer implementation
Pass *TDGraphDrawer::getFunctionPass(Function *F) {
  return createTDDSFunctionPrinterPass(F);
}

Pass *TDGraphDrawer::getModulePass() {
  return createTDDSModulePrinterPass();
}

std::string TDGraphDrawer::getFilename(Function *F) {
  return "tdds." + F->getName().str() + ".dot";
}

std::string TDGraphDrawer::getFilename(Module *M) {
  return "tdds.dot";
}

std::string TDGraphDrawer::getDisplayTitle (TVTreeItemData *item) {
  if (item)
    return "Top-down data structure " + item->dsGraphName ();
  else
    return "Top-down DSGraph";
}

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

// LocalGraphDrawer implementation
Pass *LocalGraphDrawer::getFunctionPass(Function *F) {
  return createLocalDSFunctionPrinterPass(F);
}

Pass *LocalGraphDrawer::getModulePass() {
  return createLocalDSModulePrinterPass();
}

std::string LocalGraphDrawer::getFilename(Function *F) {
  return "localds." + F->getName().str() + ".dot";
}

std::string LocalGraphDrawer::getFilename(Module *M) {
  return "localds.dot";
}

std::string LocalGraphDrawer::getDisplayTitle (TVTreeItemData *item) {
  if (item)
    return "Local data structure " + item->dsGraphName ();
  else 
    return "Local DSGraph";
}