Add SymbolData for array elements and struct fields.

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@59618 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 35 insertions, 0 deletions

diff --git a/lib/Analysis/SymbolManager.cpp b/lib/Analysis/SymbolManager.cpp
index 7829e3ace3..aee74a4a14 100644
--- a/lib/Analysis/SymbolManager.cpp
+++ b/lib/Analysis/SymbolManager.cpp
@@ -51,6 +51,41 @@ SymbolID SymbolManager::getSymbol(VarDecl* D) {
   DataMap[SymbolCounter] = SD;
   return SymbolCounter++;
 }  
+
+SymbolID SymbolManager::getElementSymbol(const MemRegion* R, 
+                                         const llvm::APSInt* Idx){
+  llvm::FoldingSetNodeID ID;
+  SymbolDataElement::Profile(ID, R, Idx);
+  void* InsertPos;
+  SymbolData* SD = DataSet.FindNodeOrInsertPos(ID, InsertPos);
+
+  if (SD)
+    return SD->getSymbol();
+
+  SD = (SymbolData*) BPAlloc.Allocate<SymbolDataElement>();
+  new (SD) SymbolDataElement(SymbolCounter, R, Idx);
+
+  DataSet.InsertNode(SD, InsertPos);
+  DataMap[SymbolCounter] = SD;
+  return SymbolCounter++;
+}
+
+SymbolID SymbolManager::getFieldSymbol(const MemRegion* R, const FieldDecl* D) {
+  llvm::FoldingSetNodeID ID;
+  SymbolDataField::Profile(ID, R, D);
+  void* InsertPos;
+  SymbolData* SD = DataSet.FindNodeOrInsertPos(ID, InsertPos);
+
+  if (SD)
+    return SD->getSymbol();
+
+  SD = (SymbolData*) BPAlloc.Allocate<SymbolDataField>();
+  new (SD) SymbolDataField(SymbolCounter, R, D);
+
+  DataSet.InsertNode(SD, InsertPos);
+  DataMap[SymbolCounter] = SD;
+  return SymbolCounter++;
+}
  
 SymbolID SymbolManager::getContentsOfSymbol(SymbolID sym) {