Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 1 | //===- Hello.cpp - Example code from "Writing an LLVM Pass" ---------------===// |
Misha Brukman | fd93908 | 2005-04-21 23:48:37 +0000 | [diff] [blame] | 2 | // |
John Criswell | b576c94 | 2003-10-20 19:43:21 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file was developed by the LLVM research group and is distributed under |
| 6 | // the University of Illinois Open Source License. See LICENSE.TXT for details. |
Misha Brukman | fd93908 | 2005-04-21 23:48:37 +0000 | [diff] [blame] | 7 | // |
John Criswell | b576c94 | 2003-10-20 19:43:21 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 9 | // |
| 10 | // This file implements two versions of the LLVM "Hello World" pass described |
| 11 | // in docs/WritingAnLLVMPass.html |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
Chris Lattner | cbfdd1f | 2006-12-19 22:24:09 +0000 | [diff] [blame] | 15 | #define DEBUG_TYPE "hello" |
Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 16 | #include "llvm/Pass.h" |
| 17 | #include "llvm/Function.h" |
Reid Spencer | 05445e3 | 2006-08-07 23:17:24 +0000 | [diff] [blame] | 18 | #include "llvm/ADT/StringExtras.h" |
Bill Wendling | 62c804a | 2006-11-26 09:17:06 +0000 | [diff] [blame] | 19 | #include "llvm/Support/Streams.h" |
Reid Spencer | 05445e3 | 2006-08-07 23:17:24 +0000 | [diff] [blame] | 20 | #include "llvm/ADT/Statistic.h" |
Chris Lattner | f7703df | 2004-01-09 06:12:26 +0000 | [diff] [blame] | 21 | using namespace llvm; |
Brian Gaeke | d0fde30 | 2003-11-11 22:41:34 +0000 | [diff] [blame] | 22 | |
Chris Lattner | cbfdd1f | 2006-12-19 22:24:09 +0000 | [diff] [blame] | 23 | STATISTIC(HelloCounter, "Counts number of functions greeted"); |
| 24 | |
Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 25 | namespace { |
| 26 | // Hello - The first implementation, without getAnalysisUsage. |
| 27 | struct Hello : public FunctionPass { |
Devang Patel | 1997473 | 2007-05-03 01:11:54 +0000 | [diff] [blame^] | 28 | static char ID; // Pass identifcation, replacement for typeid |
Devang Patel | 794fd75 | 2007-05-01 21:15:47 +0000 | [diff] [blame] | 29 | Hello() : FunctionPass((intptr_t)&ID) {} |
| 30 | |
Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 31 | virtual bool runOnFunction(Function &F) { |
Reid Spencer | 05445e3 | 2006-08-07 23:17:24 +0000 | [diff] [blame] | 32 | HelloCounter++; |
| 33 | std::string fname = F.getName(); |
| 34 | EscapeString(fname); |
Bill Wendling | e815619 | 2006-12-07 01:30:32 +0000 | [diff] [blame] | 35 | cerr << "Hello: " << fname << "\n"; |
Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 36 | return false; |
| 37 | } |
Misha Brukman | fd93908 | 2005-04-21 23:48:37 +0000 | [diff] [blame] | 38 | }; |
Devang Patel | 794fd75 | 2007-05-01 21:15:47 +0000 | [diff] [blame] | 39 | |
Devang Patel | 1997473 | 2007-05-03 01:11:54 +0000 | [diff] [blame^] | 40 | char Hello::ID = 0; |
Chris Lattner | 7f8897f | 2006-08-27 22:42:52 +0000 | [diff] [blame] | 41 | RegisterPass<Hello> X("hello", "Hello World Pass"); |
Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 42 | |
| 43 | // Hello2 - The second implementation with getAnalysisUsage implemented. |
| 44 | struct Hello2 : public FunctionPass { |
Devang Patel | 1997473 | 2007-05-03 01:11:54 +0000 | [diff] [blame^] | 45 | static char ID; // Pass identifcation, replacement for typeid |
Devang Patel | 794fd75 | 2007-05-01 21:15:47 +0000 | [diff] [blame] | 46 | Hello2() : FunctionPass((intptr_t)&ID) {} |
| 47 | |
Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 48 | virtual bool runOnFunction(Function &F) { |
Reid Spencer | 05445e3 | 2006-08-07 23:17:24 +0000 | [diff] [blame] | 49 | HelloCounter++; |
| 50 | std::string fname = F.getName(); |
| 51 | EscapeString(fname); |
Bill Wendling | e815619 | 2006-12-07 01:30:32 +0000 | [diff] [blame] | 52 | cerr << "Hello: " << fname << "\n"; |
Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 53 | return false; |
| 54 | } |
| 55 | |
| 56 | // We don't modify the program, so we preserve all analyses |
| 57 | virtual void getAnalysisUsage(AnalysisUsage &AU) const { |
| 58 | AU.setPreservesAll(); |
| 59 | }; |
Misha Brukman | fd93908 | 2005-04-21 23:48:37 +0000 | [diff] [blame] | 60 | }; |
Devang Patel | 1997473 | 2007-05-03 01:11:54 +0000 | [diff] [blame^] | 61 | char Hello2::ID = 0; |
Chris Lattner | 7f8897f | 2006-08-27 22:42:52 +0000 | [diff] [blame] | 62 | RegisterPass<Hello2> Y("hello2", |
| 63 | "Hello World Pass (with getAnalysisUsage implemented)"); |
Chris Lattner | 180e568 | 2002-08-08 20:10:38 +0000 | [diff] [blame] | 64 | } |