2002-08-09 04:10:38 +08:00
|
|
|
//===- Hello.cpp - Example code from "Writing an LLVM Pass" ---------------===//
|
2005-04-22 07:48:37 +08:00
|
|
|
//
|
2003-10-21 03:43:21 +08:00
|
|
|
// The LLVM Compiler Infrastructure
|
|
|
|
//
|
2007-12-30 04:36:04 +08:00
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
2005-04-22 07:48:37 +08:00
|
|
|
//
|
2003-10-21 03:43:21 +08:00
|
|
|
//===----------------------------------------------------------------------===//
|
2002-08-09 04:10:38 +08:00
|
|
|
//
|
|
|
|
// This file implements two versions of the LLVM "Hello World" pass described
|
|
|
|
// in docs/WritingAnLLVMPass.html
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
2006-12-20 06:24:09 +08:00
|
|
|
#define DEBUG_TYPE "hello"
|
2002-08-09 04:10:38 +08:00
|
|
|
#include "llvm/Pass.h"
|
|
|
|
#include "llvm/Function.h"
|
2006-08-08 07:17:24 +08:00
|
|
|
#include "llvm/ADT/StringExtras.h"
|
2006-11-26 17:17:06 +08:00
|
|
|
#include "llvm/Support/Streams.h"
|
2006-08-08 07:17:24 +08:00
|
|
|
#include "llvm/ADT/Statistic.h"
|
2004-01-09 14:12:26 +08:00
|
|
|
using namespace llvm;
|
2003-11-12 06:41:34 +08:00
|
|
|
|
2006-12-20 06:24:09 +08:00
|
|
|
STATISTIC(HelloCounter, "Counts number of functions greeted");
|
|
|
|
|
2002-08-09 04:10:38 +08:00
|
|
|
namespace {
|
|
|
|
// Hello - The first implementation, without getAnalysisUsage.
|
|
|
|
struct Hello : public FunctionPass {
|
2007-05-06 21:37:16 +08:00
|
|
|
static char ID; // Pass identification, replacement for typeid
|
2008-09-05 01:05:41 +08:00
|
|
|
Hello() : FunctionPass(&ID) {}
|
2007-05-02 05:15:47 +08:00
|
|
|
|
2002-08-09 04:10:38 +08:00
|
|
|
virtual bool runOnFunction(Function &F) {
|
2006-08-08 07:17:24 +08:00
|
|
|
HelloCounter++;
|
|
|
|
std::string fname = F.getName();
|
|
|
|
EscapeString(fname);
|
2006-12-07 09:30:32 +08:00
|
|
|
cerr << "Hello: " << fname << "\n";
|
2002-08-09 04:10:38 +08:00
|
|
|
return false;
|
|
|
|
}
|
2005-04-22 07:48:37 +08:00
|
|
|
};
|
2008-05-13 08:00:25 +08:00
|
|
|
}
|
2007-05-02 05:15:47 +08:00
|
|
|
|
2008-05-13 08:00:25 +08:00
|
|
|
char Hello::ID = 0;
|
|
|
|
static RegisterPass<Hello> X("hello", "Hello World Pass");
|
2002-08-09 04:10:38 +08:00
|
|
|
|
2008-05-13 08:00:25 +08:00
|
|
|
namespace {
|
2002-08-09 04:10:38 +08:00
|
|
|
// Hello2 - The second implementation with getAnalysisUsage implemented.
|
|
|
|
struct Hello2 : public FunctionPass {
|
2007-05-06 21:37:16 +08:00
|
|
|
static char ID; // Pass identification, replacement for typeid
|
2008-09-05 01:05:41 +08:00
|
|
|
Hello2() : FunctionPass(&ID) {}
|
2007-05-02 05:15:47 +08:00
|
|
|
|
2002-08-09 04:10:38 +08:00
|
|
|
virtual bool runOnFunction(Function &F) {
|
2006-08-08 07:17:24 +08:00
|
|
|
HelloCounter++;
|
|
|
|
std::string fname = F.getName();
|
|
|
|
EscapeString(fname);
|
2006-12-07 09:30:32 +08:00
|
|
|
cerr << "Hello: " << fname << "\n";
|
2002-08-09 04:10:38 +08:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
// We don't modify the program, so we preserve all analyses
|
|
|
|
virtual void getAnalysisUsage(AnalysisUsage &AU) const {
|
|
|
|
AU.setPreservesAll();
|
|
|
|
};
|
2005-04-22 07:48:37 +08:00
|
|
|
};
|
2002-08-09 04:10:38 +08:00
|
|
|
}
|
2008-05-13 08:00:25 +08:00
|
|
|
|
|
|
|
char Hello2::ID = 0;
|
|
|
|
static RegisterPass<Hello2>
|
|
|
|
Y("hello2", "Hello World Pass (with getAnalysisUsage implemented)");
|