1 // Copyright 2014 the V8 project authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef V8_COMPILER_NODE_AUX_DATA_H_ 6 #define V8_COMPILER_NODE_AUX_DATA_H_ 7 8 #include "src/compiler/node.h" 9 #include "src/zone/zone-containers.h" 10 11 namespace v8 { 12 namespace internal { 13 namespace compiler { 14 15 // Forward declarations. 16 class Node; 17 18 template <class T, T def()> 19 class NodeAuxData { 20 public: 21 explicit NodeAuxData(Zone* zone) : aux_data_(zone) {} 22 23 void Set(Node* node, T const& data) { 24 size_t const id = node->id(); 25 if (id >= aux_data_.size()) aux_data_.resize(id + 1, def()); 26 aux_data_[id] = data; 27 } 28 29 T Get(Node* node) const { 30 size_t const id = node->id(); 31 return (id < aux_data_.size()) ? aux_data_[id] : def(); 32 } 33 34 class const_iterator; 35 friend class const_iterator; 36 37 const_iterator begin() const; 38 const_iterator end() const; 39 40 private: 41 ZoneVector<T> aux_data_; 42 }; 43 44 template <class T, T def()> 45 class NodeAuxData<T, def>::const_iterator { 46 public: 47 typedef std::forward_iterator_tag iterator_category; 48 typedef int difference_type; 49 typedef std::pair<size_t, T> value_type; 50 typedef value_type* pointer; 51 typedef value_type& reference; 52 53 const_iterator(const ZoneVector<T>* data, size_t current) 54 : data_(data), current_(current) {} 55 const_iterator(const const_iterator& other) 56 : data_(other.data_), current_(other.current_) {} 57 58 value_type operator*() const { 59 return std::make_pair(current_, (*data_)[current_]); 60 } 61 bool operator==(const const_iterator& other) const { 62 return current_ == other.current_ && data_ == other.data_; 63 } 64 bool operator!=(const const_iterator& other) const { 65 return !(*this == other); 66 } 67 const_iterator& operator++() { 68 ++current_; 69 return *this; 70 } 71 const_iterator operator++(int); 72 73 private: 74 const ZoneVector<T>* data_; 75 size_t current_; 76 }; 77 78 template <class T, T def()> 79 typename NodeAuxData<T, def>::const_iterator NodeAuxData<T, def>::begin() 80 const { 81 return typename NodeAuxData<T, def>::const_iterator(&aux_data_, 0); 82 } 83 84 template <class T, T def()> 85 typename NodeAuxData<T, def>::const_iterator NodeAuxData<T, def>::end() const { 86 return typename NodeAuxData<T, def>::const_iterator(&aux_data_, 87 aux_data_.size()); 88 } 89 90 } // namespace compiler 91 } // namespace internal 92 } // namespace v8 93 94 #endif // V8_COMPILER_NODE_AUX_DATA_H_ 95
