| 25 | } |
| 26 | |
| 27 | llvm::BasicBlock *IRAccessHelper::getBasicBlock(std::string Name, |
| 28 | unsigned BIdx) { |
| 29 | auto *F = getFunction(Name); |
| 30 | if (!F) |
| 31 | return nullptr; |
| 32 | |
| 33 | auto Iter = BasicBlocks.find(F); |
| 34 | if (Iter == BasicBlocks.end()) |
| 35 | return nullptr; |
| 36 | |
| 37 | auto &BasicBlocks = Iter->second; |
| 38 | if (BasicBlocks.size() < BIdx) |
| 39 | return nullptr; |
| 40 | |
| 41 | auto *BB = BasicBlocks[BIdx]; |
| 42 | if (!BB) |
| 43 | return nullptr; |
| 44 | |
| 45 | return const_cast<llvm::BasicBlock *>(BB); |
| 46 | } |
| 47 | |
| 48 | llvm::Instruction * |
| 49 | IRAccessHelper::getInstruction(std::string Name, unsigned BIdx, unsigned IIdx) { |