//==- GRStateTrait.h - Partial implementations of GRStateTrait -----*- C++ -*-//
//             
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
//  This file defines partial implementations of template specializations of
//  the class GRStateTrait<>.  GRStateTrait<> is used by GRState to implement
//  set/get methods for mapulating a GRState's generic data map.
//
//===----------------------------------------------------------------------===//


#ifndef LLVM_CLANG_ANALYSIS_GRSTATETRAIT_H
#define LLVM_CLANG_ANALYSIS_GRSTATETRAIT_H

namespace llvm {
  class BumpPtrAllocator;
  template <typename K, typename D, typename I> class ImmutableMap;
  template <typename K, typename I> class ImmutableSet;
  template <typename T> class ImmutableList;
  template <typename T> class ImmutableListImpl;
}

namespace clang {
  template <typename T> struct GRStatePartialTrait;
  
  // Partial-specialization for ImmutableMap.
  
  template <typename Key, typename Data, typename Info>
  struct GRStatePartialTrait< llvm::ImmutableMap<Key,Data,Info> > {
    typedef llvm::ImmutableMap<Key,Data,Info> data_type;
    typedef typename data_type::Factory&      context_type;  
    typedef Key                               key_type;
    typedef Data                              value_type;
    typedef const value_type*                 lookup_type;
    
    static inline data_type MakeData(void* const* p) {
      return p ? data_type((typename data_type::TreeTy*) *p) : data_type(0);
    }  
    static inline void* MakeVoidPtr(data_type B) {
      return B.getRoot();
    }  
    static lookup_type Lookup(data_type B, key_type K) {
      return B.lookup(K);
    }  
    static data_type Set(data_type B, key_type K, value_type E,context_type F){
      return F.Add(B, K, E);
    }
    
    static data_type Remove(data_type B, key_type K, context_type F) {
      return F.Remove(B, K);
    }
    
    static inline context_type MakeContext(void* p) {
      return *((typename data_type::Factory*) p);
    }
    
    static void* CreateContext(llvm::BumpPtrAllocator& Alloc) {
      return new typename data_type::Factory(Alloc);      
    }
    
    static void DeleteContext(void* Ctx) {
      delete (typename data_type::Factory*) Ctx;
    }      
  };
  
  
  // Partial-specialization for ImmutableSet.
  
  template <typename Key, typename Info>
  struct GRStatePartialTrait< llvm::ImmutableSet<Key,Info> > {
    typedef llvm::ImmutableSet<Key,Info>      data_type;
    typedef typename data_type::Factory&      context_type;  
    typedef Key                               key_type;
    
    static inline data_type MakeData(void* const* p) {
      return p ? data_type((typename data_type::TreeTy*) *p) : data_type(0);
    }  

    static inline void* MakeVoidPtr(data_type B) {
      return B.getRoot();
    }

    static data_type Add(data_type B, key_type K, context_type F) {
      return F.Add(B, K);
    }
    
    static data_type Remove(data_type B, key_type K, context_type F) {
      return F.Remove(B, K);
    }
    
    static bool Contains(data_type B, key_type K) {
      return B.contains(K);
    }
    
    static inline context_type MakeContext(void* p) {
      return *((typename data_type::Factory*) p);
    }
    
    static void* CreateContext(llvm::BumpPtrAllocator& Alloc) {
      return new typename data_type::Factory(Alloc);      
    }
    
    static void DeleteContext(void* Ctx) {
      delete (typename data_type::Factory*) Ctx;
    }      
  };
    
  // Partial-specialization for ImmutableList.
  
  template <typename T>
  struct GRStatePartialTrait< llvm::ImmutableList<T> > {
    typedef llvm::ImmutableList<T>            data_type;
    typedef T                                 key_type;
    typedef typename data_type::Factory&      context_type;  
    
    static data_type Add(data_type L, key_type K, context_type F) {
      return F.Add(K, L);
    }
    
    static inline data_type MakeData(void* const* p) {
      return p ? data_type((const llvm::ImmutableListImpl<T>*) *p) 
               : data_type(0);
    }  
    
    static inline void* MakeVoidPtr(data_type D) {
      return  (void*) D.getInternalPointer();
    }  
    
    static inline context_type MakeContext(void* p) {
      return *((typename data_type::Factory*) p);
    }
    
    static void* CreateContext(llvm::BumpPtrAllocator& Alloc) {
      return new typename data_type::Factory(Alloc);      
    }
    
    static void DeleteContext(void* Ctx) {
      delete (typename data_type::Factory*) Ctx;
    }      
  };
} // end clang namespace

#endif