Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 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 | #include "src/compiler/source-position.h" |
| 6 | #include "src/compiler/graph.h" |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 7 | #include "src/compiler/node-aux-data.h" |
Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 8 | |
| 9 | namespace v8 { |
| 10 | namespace internal { |
| 11 | namespace compiler { |
| 12 | |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 13 | class SourcePositionTable::Decorator final : public GraphDecorator { |
Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 14 | public: |
| 15 | explicit Decorator(SourcePositionTable* source_positions) |
| 16 | : source_positions_(source_positions) {} |
| 17 | |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 18 | void Decorate(Node* node) final { |
Ben Murdoch | c561043 | 2016-08-08 18:44:38 +0100 | [diff] [blame^] | 19 | source_positions_->SetSourcePosition(node, |
| 20 | source_positions_->current_position_); |
Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 21 | } |
| 22 | |
| 23 | private: |
| 24 | SourcePositionTable* source_positions_; |
| 25 | }; |
| 26 | |
| 27 | |
| 28 | SourcePositionTable::SourcePositionTable(Graph* graph) |
| 29 | : graph_(graph), |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 30 | decorator_(nullptr), |
| 31 | current_position_(SourcePosition::Unknown()), |
Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 32 | table_(graph->zone()) {} |
| 33 | |
| 34 | |
| 35 | void SourcePositionTable::AddDecorator() { |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 36 | DCHECK_NULL(decorator_); |
Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 37 | decorator_ = new (graph_->zone()) Decorator(this); |
| 38 | graph_->AddDecorator(decorator_); |
| 39 | } |
| 40 | |
| 41 | |
| 42 | void SourcePositionTable::RemoveDecorator() { |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 43 | DCHECK_NOT_NULL(decorator_); |
Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 44 | graph_->RemoveDecorator(decorator_); |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 45 | decorator_ = nullptr; |
Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 46 | } |
| 47 | |
| 48 | |
Emily Bernier | d0a1eb7 | 2015-03-24 16:35:39 -0400 | [diff] [blame] | 49 | SourcePosition SourcePositionTable::GetSourcePosition(Node* node) const { |
Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 50 | return table_.Get(node); |
| 51 | } |
| 52 | |
Ben Murdoch | c561043 | 2016-08-08 18:44:38 +0100 | [diff] [blame^] | 53 | void SourcePositionTable::SetSourcePosition(Node* node, |
| 54 | SourcePosition position) { |
| 55 | table_.Set(node, position); |
| 56 | } |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 57 | |
| 58 | void SourcePositionTable::Print(std::ostream& os) const { |
| 59 | os << "{"; |
| 60 | bool needs_comma = false; |
| 61 | for (auto i : table_) { |
| 62 | SourcePosition pos = i.second; |
| 63 | if (pos.IsKnown()) { |
| 64 | if (needs_comma) { |
| 65 | os << ","; |
| 66 | } |
| 67 | os << "\"" << i.first << "\"" |
| 68 | << ":" << pos.raw(); |
| 69 | needs_comma = true; |
| 70 | } |
| 71 | } |
| 72 | os << "}"; |
| 73 | } |
| 74 | |
Ben Murdoch | b8a8cc1 | 2014-11-26 15:28:44 +0000 | [diff] [blame] | 75 | } // namespace compiler |
| 76 | } // namespace internal |
| 77 | } // namespace v8 |