diff options
Diffstat (limited to 'lib/Transforms/Hello/Hello.cpp')
| -rw-r--r-- | lib/Transforms/Hello/Hello.cpp | 67 | 
1 files changed, 67 insertions, 0 deletions
diff --git a/lib/Transforms/Hello/Hello.cpp b/lib/Transforms/Hello/Hello.cpp new file mode 100644 index 000000000000..d07f6135257f --- /dev/null +++ b/lib/Transforms/Hello/Hello.cpp @@ -0,0 +1,67 @@ +//===- Hello.cpp - Example code from "Writing an LLVM Pass" ---------------===// +// +//                     The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +// +// This file implements two versions of the LLVM "Hello World" pass described +// in docs/WritingAnLLVMPass.html +// +//===----------------------------------------------------------------------===// + +#define DEBUG_TYPE "hello" +#include "llvm/Pass.h" +#include "llvm/Function.h" +#include "llvm/ADT/StringExtras.h" +#include "llvm/Support/Streams.h" +#include "llvm/ADT/Statistic.h" +using namespace llvm; + +STATISTIC(HelloCounter, "Counts number of functions greeted"); + +namespace { +  // Hello - The first implementation, without getAnalysisUsage. +  struct Hello : public FunctionPass { +    static char ID; // Pass identification, replacement for typeid +    Hello() : FunctionPass(&ID) {} + +    virtual bool runOnFunction(Function &F) { +      HelloCounter++; +      std::string fname = F.getName(); +      EscapeString(fname); +      cerr << "Hello: " << fname << "\n"; +      return false; +    } +  }; +} + +char Hello::ID = 0; +static RegisterPass<Hello> X("hello", "Hello World Pass"); + +namespace { +  // Hello2 - The second implementation with getAnalysisUsage implemented. +  struct Hello2 : public FunctionPass { +    static char ID; // Pass identification, replacement for typeid +    Hello2() : FunctionPass(&ID) {} + +    virtual bool runOnFunction(Function &F) { +      HelloCounter++; +      std::string fname = F.getName(); +      EscapeString(fname); +      cerr << "Hello: " << fname << "\n"; +      return false; +    } + +    // We don't modify the program, so we preserve all analyses +    virtual void getAnalysisUsage(AnalysisUsage &AU) const { +      AU.setPreservesAll(); +    }; +  }; +} + +char Hello2::ID = 0; +static RegisterPass<Hello2> +Y("hello2", "Hello World Pass (with getAnalysisUsage implemented)");  | 
