1*0b57cec5SDimitry Andric //===- PtrUseVisitor.cpp - InstVisitors over a pointers uses --------------===//
2*0b57cec5SDimitry Andric //
3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*0b57cec5SDimitry Andric //
7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
8*0b57cec5SDimitry Andric //
9*0b57cec5SDimitry Andric /// \file
10*0b57cec5SDimitry Andric /// Implementation of the pointer use visitors.
11*0b57cec5SDimitry Andric //
12*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
13*0b57cec5SDimitry Andric 
14*0b57cec5SDimitry Andric #include "llvm/Analysis/PtrUseVisitor.h"
15*0b57cec5SDimitry Andric #include "llvm/IR/Instruction.h"
16*0b57cec5SDimitry Andric #include "llvm/IR/Instructions.h"
17*0b57cec5SDimitry Andric #include <algorithm>
18*0b57cec5SDimitry Andric 
19*0b57cec5SDimitry Andric using namespace llvm;
20*0b57cec5SDimitry Andric 
enqueueUsers(Instruction & I)21*0b57cec5SDimitry Andric void detail::PtrUseVisitorBase::enqueueUsers(Instruction &I) {
22*0b57cec5SDimitry Andric   for (Use &U : I.uses()) {
23*0b57cec5SDimitry Andric     if (VisitedUses.insert(&U).second) {
24*0b57cec5SDimitry Andric       UseToVisit NewU = {
25*0b57cec5SDimitry Andric         UseToVisit::UseAndIsOffsetKnownPair(&U, IsOffsetKnown),
26*0b57cec5SDimitry Andric         Offset
27*0b57cec5SDimitry Andric       };
28*0b57cec5SDimitry Andric       Worklist.push_back(std::move(NewU));
29*0b57cec5SDimitry Andric     }
30*0b57cec5SDimitry Andric   }
31*0b57cec5SDimitry Andric }
32*0b57cec5SDimitry Andric 
adjustOffsetForGEP(GetElementPtrInst & GEPI)33*0b57cec5SDimitry Andric bool detail::PtrUseVisitorBase::adjustOffsetForGEP(GetElementPtrInst &GEPI) {
34*0b57cec5SDimitry Andric   if (!IsOffsetKnown)
35*0b57cec5SDimitry Andric     return false;
36*0b57cec5SDimitry Andric 
37*0b57cec5SDimitry Andric   APInt TmpOffset(DL.getIndexTypeSizeInBits(GEPI.getType()), 0);
38*0b57cec5SDimitry Andric   if (GEPI.accumulateConstantOffset(DL, TmpOffset)) {
39*0b57cec5SDimitry Andric     Offset += TmpOffset.sextOrTrunc(Offset.getBitWidth());
40*0b57cec5SDimitry Andric     return true;
41*0b57cec5SDimitry Andric   }
42*0b57cec5SDimitry Andric 
43*0b57cec5SDimitry Andric   return false;
44*0b57cec5SDimitry Andric }
45