Chris Lattner | cf3056d | 2003-10-13 03:32:08 +0000 | [diff] [blame] | 1 | //===- FindUsedTypes.cpp - Find all Types used by a module ----------------===// |
Misha Brukman | 2b37d7c | 2005-04-21 21:13:18 +0000 | [diff] [blame] | 2 | // |
John Criswell | b576c94 | 2003-10-20 19:43:21 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 4ee451d | 2007-12-29 20:36:04 +0000 | [diff] [blame^] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Misha Brukman | 2b37d7c | 2005-04-21 21:13:18 +0000 | [diff] [blame] | 7 | // |
John Criswell | b576c94 | 2003-10-20 19:43:21 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 9 | // |
Chris Lattner | ba12c23 | 2003-11-02 01:28:41 +0000 | [diff] [blame] | 10 | // This pass is used to seek out all of the types in use by the program. Note |
| 11 | // that this analysis explicitly does not include types only used by the symbol |
| 12 | // table. |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 13 | // |
| 14 | //===----------------------------------------------------------------------===// |
| 15 | |
| 16 | #include "llvm/Analysis/FindUsedTypes.h" |
Chris Lattner | c2bcde4 | 2003-10-28 23:09:45 +0000 | [diff] [blame] | 17 | #include "llvm/Constants.h" |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 18 | #include "llvm/DerivedTypes.h" |
Chris Lattner | 93193f8 | 2002-01-31 00:42:27 +0000 | [diff] [blame] | 19 | #include "llvm/Module.h" |
Chris Lattner | e538838 | 2006-12-06 06:35:25 +0000 | [diff] [blame] | 20 | #include "llvm/Assembly/Writer.h" |
Chris Lattner | 221d688 | 2002-02-12 21:07:25 +0000 | [diff] [blame] | 21 | #include "llvm/Support/InstIterator.h" |
Chris Lattner | 0555ed8 | 2004-05-09 06:22:29 +0000 | [diff] [blame] | 22 | using namespace llvm; |
Brian Gaeke | d0fde30 | 2003-11-11 22:41:34 +0000 | [diff] [blame] | 23 | |
Devang Patel | 1997473 | 2007-05-03 01:11:54 +0000 | [diff] [blame] | 24 | char FindUsedTypes::ID = 0; |
Chris Lattner | 5d8925c | 2006-08-27 22:30:17 +0000 | [diff] [blame] | 25 | static RegisterPass<FindUsedTypes> |
Chris Lattner | a6275cc | 2002-07-26 21:12:46 +0000 | [diff] [blame] | 26 | X("printusedtypes", "Find Used Types"); |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 27 | |
| 28 | // IncorporateType - Incorporate one type and all of its subtypes into the |
| 29 | // collection of used types. |
| 30 | // |
| 31 | void FindUsedTypes::IncorporateType(const Type *Ty) { |
Misha Brukman | 2b37d7c | 2005-04-21 21:13:18 +0000 | [diff] [blame] | 32 | // If ty doesn't already exist in the used types map, add it now, otherwise |
Chris Lattner | a5dcc4f | 2004-05-28 05:36:49 +0000 | [diff] [blame] | 33 | // return. |
| 34 | if (!UsedTypes.insert(Ty).second) return; // Already contain Ty. |
Misha Brukman | 2b37d7c | 2005-04-21 21:13:18 +0000 | [diff] [blame] | 35 | |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 36 | // Make sure to add any types this type references now. |
| 37 | // |
| 38 | for (Type::subtype_iterator I = Ty->subtype_begin(), E = Ty->subtype_end(); |
| 39 | I != E; ++I) |
| 40 | IncorporateType(*I); |
| 41 | } |
| 42 | |
Chris Lattner | c2bcde4 | 2003-10-28 23:09:45 +0000 | [diff] [blame] | 43 | void FindUsedTypes::IncorporateValue(const Value *V) { |
| 44 | IncorporateType(V->getType()); |
Misha Brukman | 2b37d7c | 2005-04-21 21:13:18 +0000 | [diff] [blame] | 45 | |
Chris Lattner | c2bcde4 | 2003-10-28 23:09:45 +0000 | [diff] [blame] | 46 | // If this is a constant, it could be using other types... |
| 47 | if (const Constant *C = dyn_cast<Constant>(V)) { |
Reid Spencer | e840434 | 2004-07-18 00:18:30 +0000 | [diff] [blame] | 48 | if (!isa<GlobalValue>(C)) |
| 49 | for (User::const_op_iterator OI = C->op_begin(), OE = C->op_end(); |
Misha Brukman | dedf2bd | 2005-04-22 04:01:18 +0000 | [diff] [blame] | 50 | OI != OE; ++OI) |
| 51 | IncorporateValue(*OI); |
Chris Lattner | c2bcde4 | 2003-10-28 23:09:45 +0000 | [diff] [blame] | 52 | } |
| 53 | } |
| 54 | |
| 55 | |
Chris Lattner | 2fbfdcf | 2002-04-07 20:49:59 +0000 | [diff] [blame] | 56 | // run - This incorporates all types used by the specified module |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 57 | // |
Chris Lattner | b12914b | 2004-09-20 04:48:05 +0000 | [diff] [blame] | 58 | bool FindUsedTypes::runOnModule(Module &m) { |
Chris Lattner | 93193f8 | 2002-01-31 00:42:27 +0000 | [diff] [blame] | 59 | UsedTypes.clear(); // reset if run multiple times... |
| 60 | |
Chris Lattner | 93193f8 | 2002-01-31 00:42:27 +0000 | [diff] [blame] | 61 | // Loop over global variables, incorporating their types |
Chris Lattner | e4d5c44 | 2005-03-15 04:54:21 +0000 | [diff] [blame] | 62 | for (Module::const_global_iterator I = m.global_begin(), E = m.global_end(); I != E; ++I) { |
Chris Lattner | 7e70829 | 2002-06-25 16:13:24 +0000 | [diff] [blame] | 63 | IncorporateType(I->getType()); |
Chris Lattner | c2bcde4 | 2003-10-28 23:09:45 +0000 | [diff] [blame] | 64 | if (I->hasInitializer()) |
| 65 | IncorporateValue(I->getInitializer()); |
| 66 | } |
Chris Lattner | 93193f8 | 2002-01-31 00:42:27 +0000 | [diff] [blame] | 67 | |
Chris Lattner | 7e70829 | 2002-06-25 16:13:24 +0000 | [diff] [blame] | 68 | for (Module::iterator MI = m.begin(), ME = m.end(); MI != ME; ++MI) { |
Chris Lattner | a92dc19 | 2003-05-31 23:30:52 +0000 | [diff] [blame] | 69 | IncorporateType(MI->getType()); |
Chris Lattner | 7e70829 | 2002-06-25 16:13:24 +0000 | [diff] [blame] | 70 | const Function &F = *MI; |
Misha Brukman | 2b37d7c | 2005-04-21 21:13:18 +0000 | [diff] [blame] | 71 | |
Chris Lattner | 2fbfdcf | 2002-04-07 20:49:59 +0000 | [diff] [blame] | 72 | // Loop over all of the instructions in the function, adding their return |
| 73 | // type as well as the types of their operands. |
Chris Lattner | 93193f8 | 2002-01-31 00:42:27 +0000 | [diff] [blame] | 74 | // |
Chris Lattner | 7e70829 | 2002-06-25 16:13:24 +0000 | [diff] [blame] | 75 | for (const_inst_iterator II = inst_begin(F), IE = inst_end(F); |
Chris Lattner | 93193f8 | 2002-01-31 00:42:27 +0000 | [diff] [blame] | 76 | II != IE; ++II) { |
Chris Lattner | 6ffe551 | 2004-04-27 15:13:33 +0000 | [diff] [blame] | 77 | const Instruction &I = *II; |
Misha Brukman | 2b37d7c | 2005-04-21 21:13:18 +0000 | [diff] [blame] | 78 | |
Chris Lattner | a5dcc4f | 2004-05-28 05:36:49 +0000 | [diff] [blame] | 79 | IncorporateType(I.getType()); // Incorporate the type of the instruction |
Chris Lattner | 6ffe551 | 2004-04-27 15:13:33 +0000 | [diff] [blame] | 80 | for (User::const_op_iterator OI = I.op_begin(), OE = I.op_end(); |
Chris Lattner | 93193f8 | 2002-01-31 00:42:27 +0000 | [diff] [blame] | 81 | OI != OE; ++OI) |
Chris Lattner | c2bcde4 | 2003-10-28 23:09:45 +0000 | [diff] [blame] | 82 | IncorporateValue(*OI); // Insert inst operand types as well |
Chris Lattner | 93193f8 | 2002-01-31 00:42:27 +0000 | [diff] [blame] | 83 | } |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 84 | } |
Misha Brukman | 2b37d7c | 2005-04-21 21:13:18 +0000 | [diff] [blame] | 85 | |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 86 | return false; |
| 87 | } |
| 88 | |
| 89 | // Print the types found in the module. If the optional Module parameter is |
| 90 | // passed in, then the types are printed symbolically if possible, using the |
| 91 | // symbol table from the module. |
| 92 | // |
Chris Lattner | a59cbb2 | 2002-07-27 01:12:17 +0000 | [diff] [blame] | 93 | void FindUsedTypes::print(std::ostream &o, const Module *M) const { |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 94 | o << "Types in use by this module:\n"; |
Chris Lattner | e538838 | 2006-12-06 06:35:25 +0000 | [diff] [blame] | 95 | for (std::set<const Type *>::const_iterator I = UsedTypes.begin(), |
| 96 | E = UsedTypes.end(); I != E; ++I) |
| 97 | WriteTypeSymbolic(o << " ", *I, M) << "\n"; |
Chris Lattner | b61107a | 2001-11-09 05:27:21 +0000 | [diff] [blame] | 98 | } |
Reid Spencer | 4f1bd9e | 2006-06-07 22:00:26 +0000 | [diff] [blame] | 99 | |
| 100 | // Ensure that this file gets linked in when FindUsedTypes.h is used. |
| 101 | DEFINING_FILE_FOR(FindUsedTypes) |